]> git.sesse.net Git - vlc/blobdiff - modules/codec/lpcm.c
Merge branch 'master' into lpcm_encoder
[vlc] / modules / codec / lpcm.c
index f323304c9e93637f34cd82a46e84052c6e659727..73b90780963d8fed23c889c48ba98d7eac7b6080 100644 (file)
@@ -9,6 +9,7 @@
  *          Christophe Massiot <massiot@via.ecp.fr>
  *          Gildas Bazin <gbazin@videolan.org>
  *          Lauren Aimar <fenrir _AT_ videolan _DOT_ org >
+ *          Steinar H. Gunderson <steinar+vlc@gunderson.no>
  *
  * 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
@@ -48,7 +49,7 @@ static void CloseCommon   ( vlc_object_t * );
 #ifdef ENABLE_SOUT
 static int  OpenEncoder   ( vlc_object_t * );
 static void CloseEncoder  ( vlc_object_t * );
-static block_t *EncoderEncode( encoder_t *, aout_buffer_t * );
+static block_t *EncodeFrames( encoder_t *, aout_buffer_t * );
 #endif
 
 vlc_module_begin ()
@@ -64,11 +65,13 @@ vlc_module_begin ()
     set_capability( "packetizer", 100 )
     set_callbacks( OpenPacketizer, CloseCommon )
 
+#ifdef ENABLE_SOUT
     add_submodule ()
     set_description( N_("Linear PCM audio encoder") )
     set_capability( "encoder", 100 )
     set_callbacks( OpenEncoder, CloseEncoder )
     add_shortcut( "lpcm" )
+#endif
 
 vlc_module_end ()
 
@@ -91,6 +94,7 @@ struct decoder_sys_t
     int      i_type;
 };
 
+#ifdef ENABLE_SOUT
 struct encoder_sys_t
 {
     int     i_channels;
@@ -101,6 +105,7 @@ struct encoder_sys_t
     int     i_buffer_used;
     int     i_frame_num;
 };
+#endif
 
 /*
  * LPCM DVD header :
@@ -422,21 +427,31 @@ static void CloseCommon( vlc_object_t *p_this )
     free( p_dec->p_sys );
 }
 
+#ifdef ENABLE_SOUT
 /*****************************************************************************
- * OpenEncoder:
+ * OpenEncoder: lpcm encoder construction
  *****************************************************************************/
 static int OpenEncoder( vlc_object_t *p_this )
 {
     encoder_t *p_enc = (encoder_t *)p_this;
     encoder_sys_t *p_sys;
 
-    if( p_enc->fmt_out.i_codec != VLC_CODEC_DVD_LPCM ||
-        ( p_enc->fmt_in.audio.i_rate != 48000 &&
-          p_enc->fmt_in.audio.i_rate != 96000 &&
-          p_enc->fmt_in.audio.i_rate != 44100 &&
-          p_enc->fmt_in.audio.i_rate != 32000 ) ||
-       p_enc->fmt_in.audio.i_channels > 8 )
+    /* We only support DVD LPCM yet. */
+    if( p_enc->fmt_out.i_codec != VLC_CODEC_DVD_LPCM )
+        return VLC_EGENERIC;
+
+    if( p_enc->fmt_in.audio.i_rate != 48000 &&
+        p_enc->fmt_in.audio.i_rate != 96000 &&
+        p_enc->fmt_in.audio.i_rate != 44100 &&
+        p_enc->fmt_in.audio.i_rate != 32000 )
     {
+        msg_Err( p_enc, "DVD LPCM supports only sample rates of 48, 96, 44.1 or 32 kHz" );
+        return VLC_EGENERIC;
+    }
+
+    if( p_enc->fmt_in.audio.i_channels > 8 )
+    {
+        msg_Err( p_enc, "DVD LPCM supports a maximum of eight channels" );
         return VLC_EGENERIC;
     }
 
@@ -445,20 +460,6 @@ static int OpenEncoder( vlc_object_t *p_this )
           (encoder_sys_t *)malloc(sizeof(encoder_sys_t)) ) == NULL )
         return VLC_ENOMEM;
 
-    p_enc->pf_encode_audio = EncoderEncode;
-    p_enc->fmt_in.i_codec = p_enc->fmt_out.i_codec;
-
-    p_enc->fmt_in.audio.i_bitspersample = 16;
-    p_enc->fmt_in.i_codec = VLC_CODEC_S16B;
-
-    p_enc->fmt_out.i_bitrate =
-        p_enc->fmt_in.audio.i_channels *
-        p_enc->fmt_in.audio.i_rate *
-        p_enc->fmt_in.audio.i_bitspersample;
-    
-    p_sys->i_channels = p_enc->fmt_in.audio.i_channels;
-    p_sys->i_rate = p_enc->fmt_in.audio.i_rate;
-    
     /* In DVD LCPM, a frame is always 150 PTS ticks. */
     p_sys->i_frame_samples = p_enc->fmt_in.audio.i_rate * 150 / 90000;
     p_sys->p_buffer = (uint8_t *)malloc(
@@ -468,11 +469,27 @@ static int OpenEncoder( vlc_object_t *p_this )
     p_sys->i_buffer_used = 0;
     p_sys->i_frame_num = 0;
 
+    p_sys->i_channels = p_enc->fmt_in.audio.i_channels;
+    p_sys->i_rate = p_enc->fmt_in.audio.i_rate;
+
+    p_enc->pf_encode_audio = EncodeFrames;
+    p_enc->fmt_in.i_codec = p_enc->fmt_out.i_codec;
+
+    p_enc->fmt_in.audio.i_bitspersample = 16;
+    p_enc->fmt_in.i_codec = VLC_CODEC_S16B;
+
+    p_enc->fmt_out.i_bitrate =
+        p_enc->fmt_in.audio.i_channels *
+        p_enc->fmt_in.audio.i_rate *
+        p_enc->fmt_in.audio.i_bitspersample *
+        (p_sys->i_frame_samples + LPCM_VOB_HEADER_LEN) /
+        p_sys->i_frame_samples;
+
     return VLC_SUCCESS;
 }
 
 /*****************************************************************************
- * CloseEncoder
+ * CloseEncoder: lpcm encoder destruction
  *****************************************************************************/
 static void CloseEncoder ( vlc_object_t *p_this )
 {
@@ -480,9 +497,9 @@ static void CloseEncoder ( vlc_object_t *p_this )
 }
 
 /*****************************************************************************
- * EncoderEncode:
+ * EncodeFrames: encode zero or more LCPM audio packets
  *****************************************************************************/
-static block_t *EncoderEncode( encoder_t *p_enc, aout_buffer_t *p_aout_buf )
+static block_t *EncodeFrames( encoder_t *p_enc, aout_buffer_t *p_aout_buf )
 {
     encoder_sys_t *p_sys = p_enc->p_sys;
     block_t *p_first_block = NULL, *p_last_block = NULL;
@@ -494,6 +511,7 @@ static block_t *EncoderEncode( encoder_t *p_enc, aout_buffer_t *p_aout_buf )
     const int i_leftover_samples = ( p_sys->i_buffer_used + p_aout_buf->i_nb_samples ) %
         p_sys->i_frame_samples;
     const int i_frame_size = p_sys->i_frame_samples * p_sys->i_channels * 2 + LPCM_VOB_HEADER_LEN;
+    const int i_start_offset = -p_sys->i_buffer_used;
 
     uint8_t i_freq_code = 0;
 
@@ -510,6 +528,8 @@ static block_t *EncoderEncode( encoder_t *p_enc, aout_buffer_t *p_aout_buf )
     case 32000:
         i_freq_code = 3;
         break;
+    default:
+        assert(0);
     }
 
     int i_bytes_consumed = 0;
@@ -534,22 +554,19 @@ static block_t *EncoderEncode( encoder_t *p_enc, aout_buffer_t *p_aout_buf )
 
         memcpy( frame + 6, p_sys->p_buffer, i_kept_bytes );
         memcpy( frame + 6 + i_kept_bytes, p_aout_buf->p_buffer + i_bytes_consumed, i_consume_bytes );
-     
+
         p_sys->i_frame_num++;
         p_sys->i_buffer_used = 0;
         i_bytes_consumed += i_consume_bytes;
 
-        /* 
-         * The pts is strictly not correct if we have samples kept from
-         * a previous buffer, but the frames are so small it should be OK.
-         */
-        p_block->i_dts = p_block->i_pts = p_aout_buf->i_pts + i * p_sys->i_frame_samples * CLOCK_FREQ / p_sys->i_rate;
+        p_block->i_dts = p_block->i_pts = p_aout_buf->i_pts +
+            (i * p_sys->i_frame_samples + i_start_offset) * CLOCK_FREQ / p_sys->i_rate;
         p_block->i_length = p_sys->i_frame_samples * CLOCK_FREQ / p_sys->i_rate;
-        
+
         if( !p_first_block )
             p_first_block = p_last_block = p_block;
         else
-            block_ChainLastAppend( &p_last_block, p_block );
+            p_last_block = p_last_block->p_next = p_block;
     }
 
     memcpy( p_sys->p_buffer,
@@ -559,6 +576,7 @@ static block_t *EncoderEncode( encoder_t *p_enc, aout_buffer_t *p_aout_buf )
 
     return p_first_block;
 }
+#endif
 
 /*****************************************************************************
  *