git.sesse.net Git - vlc/blob - modules/codec/araw.c

   1 /*****************************************************************************
   2  * araw.c: Pseudo audio decoder; for raw pcm data
   3  *****************************************************************************
   4  * Copyright (C) 2001, 2002 VideoLAN
   5  * $Id: araw.c,v 1.2 2002/10/20 17:44:17 fenrir Exp $
   6  *
   7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  22  *****************************************************************************/
  23
  24 /*****************************************************************************
  25  * Preamble
  26  *****************************************************************************/
  27 #include <vlc/vlc.h>
  28 #include <vlc/aout.h>
  29 #include <vlc/decoder.h>
  30 #include <vlc/input.h>
  31
  32 #include <stdlib.h>                                      /* malloc(), free() */
  33 #include <string.h>                                              /* strdup() */
  34
  35 /*****************************************************************************
  36  * Local prototypes
  37  *****************************************************************************/
  38 typedef struct waveformatex_s
  39 {
  40     u16 i_formattag;
  41     u16 i_channels;
  42     u32 i_samplespersec;
  43     u32 i_avgbytespersec;
  44     u16 i_blockalign;
  45     u16 i_bitspersample;
  46     u16 i_size; /* the extra size in bytes */
  47     u8  *p_data; /* The extra data */
  48 } waveformatex_t;
  49
  50 typedef struct adec_thread_s
  51 {
  52     waveformatex_t  format;
  53
  54     /* The bit stream structure handles the PES stream at the bit level */
  55     bit_stream_t        bit_stream;
  56
  57     /* Input properties */
  58     decoder_fifo_t *p_fifo;
  59
  60     /* Output properties */
  61     aout_instance_t *   p_aout;       /* opaque */
  62     aout_input_t *      p_aout_input; /* opaque */
  63     audio_sample_format_t output_format;
  64
  65     audio_date_t        date;
  66     mtime_t             pts;
  67
  68 } adec_thread_t;
  69
  70 static int  OpenDecoder    ( vlc_object_t * );
  71
  72 static int  RunDecoder     ( decoder_fifo_t * );
  73 static int  InitThread     ( adec_thread_t * );
  74 static void DecodeThread   ( adec_thread_t * );
  75 static void EndThread      ( adec_thread_t * );
  76
  77 /*****************************************************************************
  78  * Module descriptor
  79  *****************************************************************************/
  80
  81 vlc_module_begin();
  82     set_description( _("Pseudo Raw Audio decoder") );
  83     set_capability( "decoder", 50 );
  84     set_callbacks( OpenDecoder, NULL );
  85 vlc_module_end();
  86
  87
  88 static int i_channels_maps[6] =
  89 {
  90     0,
  91     AOUT_CHAN_MONO,     AOUT_CHAN_STEREO,
  92     AOUT_CHAN_3F,       AOUT_CHAN_2F2R,
  93     AOUT_CHAN_3F2R
  94 };
  95
  96 /*****************************************************************************
  97  * OpenDecoder: probe the decoder and return score
  98  *****************************************************************************
  99  * Tries to launch a decoder and return score so that the interface is able
 100  * to choose.
 101  *****************************************************************************/
 102 static int OpenDecoder( vlc_object_t *p_this )
 103 {
 104     decoder_fifo_t *p_fifo = (decoder_fifo_t*) p_this;
 105
 106     switch( p_fifo->i_fourcc )
 107     {
 108         case VLC_FOURCC('a','r','a','w'):
 109         case VLC_FOURCC('t','w','o','s'): /* for mov file */
 110             p_fifo->pf_run = RunDecoder;
 111             return VLC_SUCCESS;
 112
 113         default:
 114             return VLC_EGENERIC;
 115     }
 116
 117 }
 118
 119 /*****************************************************************************
 120  * RunDecoder: this function is called just after the thread is created
 121  *****************************************************************************/
 122 static int RunDecoder( decoder_fifo_t *p_fifo )
 123 {
 124     adec_thread_t *p_adec;
 125     int b_error;
 126
 127     if( !( p_adec = malloc( sizeof( adec_thread_t ) ) ) )
 128     {
 129         msg_Err( p_fifo, "out of memory" );
 130         DecoderError( p_fifo );
 131         return( -1 );
 132     }
 133     memset( p_adec, 0, sizeof( adec_thread_t ) );
 134
 135     p_adec->p_fifo = p_fifo;
 136
 137     if( InitThread( p_adec ) != 0 )
 138     {
 139         DecoderError( p_fifo );
 140         return( -1 );
 141     }
 142
 143     while( ( !p_adec->p_fifo->b_die )&&( !p_adec->p_fifo->b_error ) )
 144     {
 145         DecodeThread( p_adec );
 146     }
 147
 148
 149     if( ( b_error = p_adec->p_fifo->b_error ) )
 150     {
 151         DecoderError( p_adec->p_fifo );
 152     }
 153
 154     EndThread( p_adec );
 155     if( b_error )
 156     {
 157         return( -1 );
 158     }
 159
 160     return( 0 );
 161 }
 162
 163
 164 #define FREE( p ) if( p ) free( p ); p = NULL
 165 #define GetWLE( p ) \
 166     ( *(u8*)(p) + ( *((u8*)(p)+1) << 8 ) )
 167
 168 #define GetDWLE( p ) \
 169     (  *(u8*)(p) + ( *((u8*)(p)+1) << 8 ) + \
 170         ( *((u8*)(p)+2) << 16 ) + ( *((u8*)(p)+3) << 24 ) )
 171
 172 static void GetWaveFormatEx( waveformatex_t *p_wh,
 173                              u8 *p_data )
 174 {
 175
 176     p_wh->i_formattag     = GetWLE( p_data );
 177     p_wh->i_channels      = GetWLE( p_data + 2 );
 178     p_wh->i_samplespersec = GetDWLE( p_data + 4 );
 179     p_wh->i_avgbytespersec= GetDWLE( p_data + 8 );
 180     p_wh->i_blockalign    = GetWLE( p_data + 12 );
 181     p_wh->i_bitspersample = GetWLE( p_data + 14 );
 182     if( p_wh->i_formattag != 1 )
 183     {
 184         p_wh->i_size          = GetWLE( p_data + 16 );
 185
 186         if( p_wh->i_size )
 187         {
 188             p_wh->p_data = malloc( p_wh->i_size );
 189             memcpy( p_wh->p_data, p_data + 18, p_wh->i_size );
 190         }
 191     }
 192 }
 193
 194 /* get the first pes from fifo */
 195 static pes_packet_t *PESGetFirst( decoder_fifo_t *p_fifo )
 196 {
 197     pes_packet_t *p_pes;
 198
 199     vlc_mutex_lock( &p_fifo->data_lock );
 200
 201     /* if fifo is empty wait */
 202     while( !p_fifo->p_first )
 203     {
 204         if( p_fifo->b_die )
 205         {
 206             vlc_mutex_unlock( &p_fifo->data_lock );
 207             return NULL;
 208         }
 209         vlc_cond_wait( &p_fifo->data_wait, &p_fifo->data_lock );
 210     }
 211     p_pes = p_fifo->p_first;
 212
 213     vlc_mutex_unlock( &p_fifo->data_lock );
 214
 215     return p_pes;
 216 }
 217 static int PESGetSize( pes_packet_t *p_pes )
 218 {
 219     data_packet_t *p_data;
 220     int i_size = 0;
 221
 222     if( !p_pes )
 223     {
 224         return( 0 );
 225     }
 226
 227     for( p_data = p_pes->p_first; p_data != NULL; p_data = p_data->p_next )
 228     {
 229         i_size += p_data->p_payload_end - p_data->p_payload_start;
 230     }
 231
 232     return( i_size );
 233 }
 234
 235
 236 /*****************************************************************************
 237  * InitThread: initialize data before entering main loop
 238  *****************************************************************************/
 239
 240 static int InitThread( adec_thread_t * p_adec )
 241 {
 242
 243     if( p_adec->p_fifo->p_demux_data )
 244     {
 245         GetWaveFormatEx( &p_adec->format,
 246                          (u8*)p_adec->p_fifo->p_demux_data );
 247         /* fixing some values */
 248         if( p_adec->format.i_formattag == 1 && !p_adec->format.i_blockalign )
 249         {
 250             p_adec->format.i_blockalign = p_adec->format.i_channels *
 251                     ( ( p_adec->format.i_bitspersample + 7 ) / 8 );
 252         }
 253     }
 254     else
 255     {
 256         msg_Err( p_adec->p_fifo, "unknown raw format" );
 257         return( -1 );
 258     }
 259
 260     msg_Dbg( p_adec->p_fifo,
 261              "raw format: samplerate:%dHz channels:%d bits/sample:%d blockalign:%d",
 262              p_adec->format.i_samplespersec,
 263              p_adec->format.i_channels,
 264              p_adec->format.i_bitspersample, p_adec->format.i_blockalign );
 265
 266     /* Initialize the thread properties */
 267     switch( ( p_adec->format.i_bitspersample + 7 ) / 8 )
 268     {
 269         case( 2 ):
 270             p_adec->output_format.i_format = VLC_FOURCC('s','1','6','l');
 271             break;
 272         case( 3 ):
 273             p_adec->output_format.i_format = VLC_FOURCC('s','2','4','l');
 274             break;
 275         case( 4 ):
 276             p_adec->output_format.i_format = VLC_FOURCC('s','3','2','l');
 277             break;
 278
 279         case( 1 ):
 280         default:
 281             msg_Err( p_adec->p_fifo, "bad parameters(bits/sample)" );
 282             return( -1 );
 283     }
 284     p_adec->output_format.i_rate = p_adec->format.i_samplespersec;
 285     if( p_adec->output_format.i_channels <= 0 ||
 286             p_adec->output_format.i_channels > 5 )
 287     {
 288         msg_Err( p_adec->p_fifo, "bad channels count(1-5)" );
 289         return( -1 );
 290     }
 291
 292     p_adec->output_format.i_channels =
 293             i_channels_maps[p_adec->format.i_channels];
 294     p_adec->p_aout = NULL;
 295     p_adec->p_aout_input = NULL;
 296
 297     /* **** Create a new audio output **** */
 298     aout_DateInit( &p_adec->date, p_adec->output_format.i_rate );
 299     p_adec->p_aout_input = aout_DecNew( p_adec->p_fifo,
 300                                         &p_adec->p_aout,
 301                                         &p_adec->output_format );
 302     if( !p_adec->p_aout_input )
 303     {
 304         msg_Err( p_adec->p_fifo, "cannot create aout" );
 305         return( -1 );
 306     }
 307
 308     /* Init the BitStream */
 309     InitBitstream( &p_adec->bit_stream, p_adec->p_fifo,
 310                    NULL, NULL );
 311
 312     return( 0 );
 313 }
 314
 315 /*****************************************************************************
 316  * DecodeThread: decodes a frame
 317  *****************************************************************************/
 318 static void DecodeThread( adec_thread_t *p_adec )
 319 {
 320     aout_buffer_t   *p_aout_buffer;
 321     int             i_samples; // per channels
 322     int             i_size;
 323     pes_packet_t    *p_pes;
 324
 325     /* **** get samples count **** */
 326     p_pes = PESGetFirst( p_adec->p_fifo );
 327
 328     i_size = PESGetSize( p_pes );
 329     if( p_adec->format.i_blockalign > 0 )
 330     {
 331         i_size -= i_size % p_adec->format.i_blockalign;
 332     }
 333     i_size = __MAX( i_size, p_adec->format.i_blockalign );
 334
 335     if( !i_size || !p_pes )
 336     {
 337         msg_Err( p_adec->p_fifo, "infinite loop..." );
 338         return;
 339     }
 340     i_samples = i_size /
 341                 ( ( p_adec->format.i_bitspersample + 7 ) / 8 ) /
 342                 p_adec->format.i_channels;
 343
 344 //    msg_Warn( p_adec->p_fifo, "got %d samples (%d bytes)", i_samples, i_size );
 345     p_adec->pts = p_pes->i_pts;
 346
 347     /* **** Now we can output these samples **** */
 348
 349     if( p_adec->pts != 0 && p_adec->pts != aout_DateGet( &p_adec->date ) )
 350     {
 351         aout_DateSet( &p_adec->date, p_adec->pts );
 352     }
 353     else if( !aout_DateGet( &p_adec->date ) )
 354     {
 355         return;
 356     }
 357
 358     p_aout_buffer = aout_DecNewBuffer( p_adec->p_aout,
 359                                        p_adec->p_aout_input,
 360                                        i_samples );
 361     if( !p_aout_buffer )
 362     {
 363         msg_Err( p_adec->p_fifo, "cannot get aout buffer" );
 364         p_adec->p_fifo->b_error = 1;
 365         return;
 366     }
 367
 368     p_aout_buffer->start_date = aout_DateGet( &p_adec->date );
 369     p_aout_buffer->end_date = aout_DateIncrement( &p_adec->date,
 370                                                   i_samples );
 371     GetChunk( &p_adec->bit_stream,
 372               p_aout_buffer->p_buffer,
 373               p_aout_buffer->i_nb_bytes );
 374
 375     aout_DecPlay( p_adec->p_aout, p_adec->p_aout_input, p_aout_buffer );
 376 }
 377
 378
 379 /*****************************************************************************
 380  * EndThread : faad decoder thread destruction
 381  *****************************************************************************/
 382 static void EndThread (adec_thread_t *p_adec)
 383 {
 384     if( p_adec->p_aout_input )
 385     {
 386         aout_DecDelete( p_adec->p_aout, p_adec->p_aout_input );
 387     }
 388
 389     FREE( p_adec->format.p_data );
 390
 391     msg_Dbg( p_adec->p_fifo, "raw audio decoder closed" );
 392
 393     free( p_adec );
 394 }
 395
 396