]> git.sesse.net Git - vlc/blobdiff - modules/codec/faad.c
* modules/codec/quicktime.c: fixed small mem leak.
[vlc] / modules / codec / faad.c
index 47046a030a1308a937baa83faf8182fd42a87c17..c1a5447662692879d73ac5c9b28f9e6f719953b5 100644 (file)
@@ -2,9 +2,10 @@
  * decoder.c: AAC decoder using libfaad2
  *****************************************************************************
  * Copyright (C) 2001, 2003 VideoLAN
- * $Id: faad.c,v 1.7 2003/12/03 12:19:55 fenrir Exp $
+ * $Id$
  *
  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
+ *          Gildas Bazin <gbazin@videolan.org>
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -39,11 +40,14 @@ vlc_module_begin();
     set_callbacks( Open, Close );
 vlc_module_end();
 
-
 /****************************************************************************
  * Local prototypes
  ****************************************************************************/
 static aout_buffer_t *DecodeBlock( decoder_t *, block_t ** );
+static void DoReordering( decoder_t *, uint32_t *, uint32_t *, int, int,
+                          uint32_t * );
+
+#define MAX_CHANNEL_POSITIONS 9
 
 struct decoder_sys_t
 {
@@ -57,23 +61,28 @@ struct decoder_sys_t
     uint8_t *p_buffer;
     int     i_buffer;
     int     i_buffer_size;
-};
 
-static unsigned int pi_channels_maps[7] =
-{
-    0,
-    AOUT_CHAN_CENTER,
-    AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT,
-    AOUT_CHAN_CENTER | AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT,
-    AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_REARLEFT |
-        AOUT_CHAN_REARRIGHT,
-    AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER | AOUT_CHAN_REARLEFT |
-        AOUT_CHAN_REARRIGHT,
-     /* FIXME */
-    AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER | AOUT_CHAN_REARLEFT |
-        AOUT_CHAN_REARRIGHT | AOUT_CHAN_REARCENTER
+    /* Channel positions of the current stream (for re-ordering) */
+    uint32_t pi_channel_positions[MAX_CHANNEL_POSITIONS];
 };
 
+static const uint32_t pi_channels_in[MAX_CHANNEL_POSITIONS] =
+    { FRONT_CHANNEL_CENTER, FRONT_CHANNEL_LEFT, FRONT_CHANNEL_RIGHT,
+      SIDE_CHANNEL_LEFT, SIDE_CHANNEL_RIGHT,
+      BACK_CHANNEL_LEFT, BACK_CHANNEL_RIGHT,
+      BACK_CHANNEL_CENTER, LFE_CHANNEL };
+static const uint32_t pi_channels_out[MAX_CHANNEL_POSITIONS] =
+    { AOUT_CHAN_CENTER, AOUT_CHAN_LEFT, AOUT_CHAN_RIGHT,
+      AOUT_CHAN_MIDDLELEFT, AOUT_CHAN_MIDDLERIGHT,
+      AOUT_CHAN_REARLEFT, AOUT_CHAN_REARRIGHT,
+      AOUT_CHAN_REARCENTER, AOUT_CHAN_LFE };
+static const uint32_t pi_channels_ordered[MAX_CHANNEL_POSITIONS] =
+    { AOUT_CHAN_LEFT, AOUT_CHAN_RIGHT,
+      AOUT_CHAN_MIDDLELEFT, AOUT_CHAN_MIDDLERIGHT,
+      AOUT_CHAN_REARLEFT, AOUT_CHAN_REARRIGHT,
+      AOUT_CHAN_CENTER, AOUT_CHAN_REARCENTER, AOUT_CHAN_LFE
+    };
+
 /*****************************************************************************
  * OpenDecoder: probe the decoder and return score
  *****************************************************************************/
@@ -99,16 +108,23 @@ static int Open( vlc_object_t *p_this )
     /* Open a faad context */
     if( ( p_sys->hfaad = faacDecOpen() ) == NULL )
     {
-        msg_Err( p_dec, "Cannot initialize faad" );
+        msg_Err( p_dec, "cannot initialize faad" );
         return VLC_EGENERIC;
     }
 
     /* Misc init */
     aout_DateSet( &p_sys->date, 0 );
     p_dec->fmt_out.i_cat = AUDIO_ES;
-    p_dec->fmt_out.i_codec = VLC_FOURCC('f','l','3','2');
+
+    if (p_this->p_libvlc->i_cpu & CPU_CAPABILITY_FPU)
+        p_dec->fmt_out.i_codec = VLC_FOURCC('f','l','3','2');
+    else
+        p_dec->fmt_out.i_codec = AOUT_FMT_S16_NE;
     p_dec->pf_decode_audio = DecodeBlock;
 
+    p_dec->fmt_out.audio.i_physical_channels =
+        p_dec->fmt_out.audio.i_original_channels = 0;
+
     if( p_dec->fmt_in.i_extra > 0 )
     {
         /* We have a decoder config so init the handle */
@@ -124,10 +140,6 @@ static int Open( vlc_object_t *p_this )
 
         p_dec->fmt_out.audio.i_rate = i_rate;
         p_dec->fmt_out.audio.i_channels = i_channels;
-        p_dec->fmt_out.audio.i_physical_channels =
-            p_dec->fmt_out.audio.i_original_channels =
-                pi_channels_maps[i_channels];
-
         aout_DateInit( &p_sys->date, i_rate );
     }
     else
@@ -135,19 +147,22 @@ static int Open( vlc_object_t *p_this )
         /* Will be initalised from first frame */
         p_dec->fmt_out.audio.i_rate = 0;
         p_dec->fmt_out.audio.i_channels = 0;
-        p_dec->fmt_out.audio.i_physical_channels =
-            p_dec->fmt_out.audio.i_original_channels = 0;
     }
 
     /* Set the faad config */
     cfg = faacDecGetCurrentConfiguration( p_sys->hfaad );
-    cfg->outputFormat = FAAD_FMT_FLOAT;
+    if (p_this->p_libvlc->i_cpu & CPU_CAPABILITY_FPU)
+        cfg->outputFormat = FAAD_FMT_FLOAT;
+    else
+        cfg->outputFormat = FAAD_FMT_16BIT;
     faacDecSetConfiguration( p_sys->hfaad, cfg );
 
     /* buffer */
-    p_sys->i_buffer = 0;
-    p_sys->i_buffer_size = 10000;
-    p_sys->p_buffer = malloc( p_sys->i_buffer_size );
+    p_sys->i_buffer = p_sys->i_buffer_size = 0;
+    p_sys->p_buffer = 0;
+
+    /* Faad2 can't deal with truncated data (eg. from MPEG TS) */
+    p_dec->b_need_packetized = VLC_TRUE;
 
     return VLC_SUCCESS;
 }
@@ -164,7 +179,7 @@ static aout_buffer_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
 
     p_block = *pp_block;
 
-    if( p_block->b_discontinuity )
+    if( p_block->i_flags&BLOCK_FLAG_DISCONTINUITY )
     {
         block_Release( p_block );
         return NULL;
@@ -185,6 +200,22 @@ static aout_buffer_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
         p_block->i_buffer = 0;
     }
 
+    if( p_dec->fmt_out.audio.i_rate == 0 && p_dec->fmt_in.i_extra > 0 )
+    {
+        /* We have a decoder config so init the handle */
+        unsigned long i_rate;
+        unsigned char i_channels;
+
+        if( faacDecInit2( p_sys->hfaad, p_dec->fmt_in.p_extra,
+                          p_dec->fmt_in.i_extra,
+                          &i_rate, &i_channels ) >= 0 )
+        {
+            p_dec->fmt_out.audio.i_rate = i_rate;
+            p_dec->fmt_out.audio.i_channels = i_channels;
+            aout_DateInit( &p_sys->date, i_rate );
+        }
+    }
+
     if( p_dec->fmt_out.audio.i_rate == 0 && p_sys->i_buffer )
     {
         unsigned long i_rate;
@@ -201,10 +232,6 @@ static aout_buffer_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
 
         p_dec->fmt_out.audio.i_rate = i_rate;
         p_dec->fmt_out.audio.i_channels = i_channels;
-        p_dec->fmt_out.audio.i_physical_channels =
-            p_dec->fmt_out.audio.i_original_channels =
-                pi_channels_maps[i_channels];
-
         aout_DateInit( &p_sys->date, i_rate );
     }
 
@@ -226,6 +253,7 @@ static aout_buffer_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
         void *samples;
         faacDecFrameInfo frame;
         aout_buffer_t *p_out;
+        int i, j;
 
         samples = faacDecDecode( p_sys->hfaad, &frame,
                                  p_sys->p_buffer, p_sys->i_buffer );
@@ -240,22 +268,32 @@ static aout_buffer_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
             return NULL;
         }
 
-        if( frame.channels <= 0 || frame.channels > 6 )
+        if( frame.channels <= 0 || frame.channels > 8 || frame.channels == 7 )
         {
-            msg_Warn( p_dec, "invalid channels count" );
+            msg_Warn( p_dec, "invalid channels count: %i", frame.channels );
 
             /* Flush the buffer */
-            p_sys->i_buffer = 0;
+            p_sys->i_buffer -= frame.bytesconsumed;
+            if( p_sys->i_buffer > 0 )
+            {
+                memmove( p_sys->p_buffer,&p_sys->p_buffer[frame.bytesconsumed],
+                         p_sys->i_buffer );
+            }
             block_Release( p_block );
             return NULL;
         }
 
         if( frame.samples <= 0 )
         {
-            msg_Warn( p_dec, "decoded zero samples" );
+            msg_Warn( p_dec, "decoded zero sample" );
 
             /* Flush the buffer */
-            p_sys->i_buffer = 0;
+            p_sys->i_buffer -= frame.bytesconsumed;
+            if( p_sys->i_buffer > 0 )
+            {
+                memmove( p_sys->p_buffer,&p_sys->p_buffer[frame.bytesconsumed],
+                         p_sys->i_buffer );
+            }
             block_Release( p_block );
             return NULL;
         }
@@ -270,12 +308,34 @@ static aout_buffer_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
 
         p_dec->fmt_out.audio.i_rate = frame.samplerate;
         p_dec->fmt_out.audio.i_channels = frame.channels;
-        p_dec->fmt_out.audio.i_physical_channels =
-            p_dec->fmt_out.audio.i_original_channels =
-                pi_channels_maps[frame.channels];
 
-        p_out = p_dec->pf_aout_buffer_new( p_dec,
-                                           frame.samples / frame.channels );
+
+        /* Convert frame.channel_position to our own channel values */
+        for( i = 0; i < frame.channels; i++ )
+        {
+            /* Find the channel code */
+            for( j = 0; j < MAX_CHANNEL_POSITIONS; j++ )
+            {
+                if( frame.channel_position[i] == pi_channels_in[j] )
+                {
+                    p_sys->pi_channel_positions[i] = pi_channels_out[j];
+                    p_dec->fmt_out.audio.i_physical_channels |=
+                        pi_channels_out[j];
+                    break;
+                }
+            }
+
+            if( j == MAX_CHANNEL_POSITIONS )
+            {
+                msg_Warn( p_dec, "unknown channel ordering" );
+                block_Release( p_block );
+                return NULL;
+            }
+        }
+        p_dec->fmt_out.audio.i_original_channels =
+            p_dec->fmt_out.audio.i_physical_channels;
+
+        p_out = p_dec->pf_aout_buffer_new(p_dec, frame.samples/frame.channels);
         if( p_out == NULL )
         {
             p_sys->i_buffer = 0;
@@ -287,7 +347,9 @@ static aout_buffer_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
         p_out->end_date = aout_DateIncrement( &p_sys->date,
                                               frame.samples / frame.channels );
 
-        memcpy( p_out->p_buffer, samples, p_out->i_nb_bytes );
+        DoReordering( p_dec, (uint32_t *)p_out->p_buffer, samples,
+                      frame.samples / frame.channels, frame.channels,
+                      p_sys->pi_channel_positions );
 
         p_sys->i_buffer -= frame.bytesconsumed;
         if( p_sys->i_buffer > 0 )
@@ -314,3 +376,40 @@ static void Close( vlc_object_t *p_this )
     faacDecClose( p_sys->hfaad );
     free( p_sys );
 }
+
+/*****************************************************************************
+ * DoReordering: do some channel re-ordering (the ac3 channel order is
+ *   different from the aac one).
+ *****************************************************************************/
+static void DoReordering( decoder_t *p_dec,
+                          uint32_t *p_out, uint32_t *p_in, int i_samples,
+                          int i_nb_channels, uint32_t *pi_chan_positions )
+{
+    int pi_chan_table[MAX_CHANNEL_POSITIONS];
+    int i, j, k;
+
+    /* Find the channels mapping */
+    for( i = 0, j = 0; i < MAX_CHANNEL_POSITIONS; i++ )
+    {
+        for( k = 0; k < i_nb_channels; k++ )
+        {
+            if( pi_channels_ordered[i] == pi_chan_positions[k] )
+            {
+                pi_chan_table[k] = j++;
+                break;
+            }
+        }
+    }
+
+    /* Do the actual reordering */
+    if( p_dec->p_libvlc->i_cpu & CPU_CAPABILITY_FPU )
+        for( i = 0; i < i_samples; i++ )
+            for( j = 0; j < i_nb_channels; j++ )
+                p_out[i * i_nb_channels + pi_chan_table[j]] =
+                    p_in[i * i_nb_channels + j];
+    else
+        for( i = 0; i < i_samples; i++ )
+            for( j = 0; j < i_nb_channels; j++ )
+                ((uint16_t *)p_out)[i * i_nb_channels + pi_chan_table[j]] =
+                    ((uint16_t *)p_in)[i * i_nb_channels + j];
+}