]> git.sesse.net Git - vlc/blobdiff - modules/codec/flac.c
* block: added
[vlc] / modules / codec / flac.c
index 907445e2a6babc21bf2ab04349154bb463dc53b8..8c7b95faef637002ab70076037a6f1d01bbcb945 100644 (file)
@@ -2,16 +2,16 @@
  * flac.c: flac decoder/packetizer/encoder module making use of libflac
  *****************************************************************************
  * Copyright (C) 1999-2001 VideoLAN
- * $Id: flac.c,v 1.4 2003/11/22 12:41:32 gbazin Exp $
+ * $Id$
  *
- * Authors: Gildas Bazin <gbazin@netcourrier.com>
+ * Authors: Gildas Bazin <gbazin@videolan.org>
  *          Sigmund Augdal <sigmunau@idi.ntnu.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
  * the Free Software Foundation; either version 2 of the License, or
  * (at your option) any later version.
- * 
+ *
  * This program is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
@@ -25,9 +25,9 @@
 /*****************************************************************************
  * Preamble
  *****************************************************************************/
+
 #include <vlc/vlc.h>
 #include <vlc/decoder.h>
-#include <vlc/input.h>
 
 #include <FLAC/stream_decoder.h>
 #include <FLAC/stream_encoder.h>
@@ -41,9 +41,6 @@
  *****************************************************************************/
 struct decoder_sys_t
 {
-    /* Module mode */
-    vlc_bool_t b_packetizer;
-
     /*
      * Input properties
      */
@@ -145,6 +142,9 @@ static uint8_t flac_crc8( const uint8_t *data, unsigned len );
  *****************************************************************************/
 vlc_module_begin();
 
+    set_category( CAT_INPUT );
+    set_subcategory( SUBCAT_INPUT_ACODEC );
+
     set_description( _("Flac audio decoder") );
     set_capability( "decoder", 100 );
     set_callbacks( OpenDecoder, CloseDecoder );
@@ -159,6 +159,7 @@ vlc_module_begin();
     set_capability( "encoder", 100 );
     set_callbacks( OpenEncoder, CloseEncoder );
 
+    add_shortcut( "flac" );
 vlc_module_end();
 
 /*****************************************************************************
@@ -184,7 +185,6 @@ static int OpenDecoder( vlc_object_t *p_this )
 
     /* Misc init */
     aout_DateSet( &p_sys->end_date, 0 );
-    p_sys->b_packetizer = VLC_FALSE;
     p_sys->i_state = STATE_NOSYNC;
     p_sys->b_stream_info = VLC_FALSE;
 
@@ -218,32 +218,56 @@ static int OpenDecoder( vlc_object_t *p_this )
     p_dec->pf_decode_audio = DecodeBlock;
     p_dec->pf_packetize    = PacketizeBlock;
 
-    /* Decode STREAMINFO */
-    msg_Dbg( p_dec, "decode STREAMINFO" );
-    p_sys->p_block = block_New( p_dec, p_dec->fmt_in.i_extra );
-    memcpy( p_sys->p_block->p_buffer, p_dec->fmt_in.p_extra,
-            p_dec->fmt_in.i_extra );
-    FLAC__stream_decoder_process_until_end_of_metadata( p_sys->p_flac );
-    msg_Dbg( p_dec, "STREAMINFO decoded" );
-
     return VLC_SUCCESS;
 }
 
 static int OpenPacketizer( vlc_object_t *p_this )
 {
     decoder_t *p_dec = (decoder_t*)p_this;
+    int i_ret;
 
-    int i_ret = OpenDecoder( p_this );
+    /* Hmmm, mem leak ?*/
+    es_format_Copy( &p_dec->fmt_out, &p_dec->fmt_in );
 
-    if( i_ret != VLC_SUCCESS ) return i_ret;
+    i_ret = OpenDecoder( p_this );
 
-    p_dec->p_sys->b_packetizer = VLC_TRUE;
+    /* Set output properties */
+    p_dec->fmt_out.i_codec = VLC_FOURCC('f','l','a','c');
 
-    es_format_Copy( &p_dec->fmt_out, &p_dec->fmt_in );
+    if( i_ret != VLC_SUCCESS ) return i_ret;
 
     return i_ret;
 }
 
+/*****************************************************************************
+ * ProcessHeader: processe Flac header.
+ *****************************************************************************/
+static void ProcessHeader( decoder_t *p_dec )
+{
+    decoder_sys_t *p_sys = p_dec->p_sys;
+
+    if( !p_dec->fmt_in.i_extra ) return;
+
+    /* Decode STREAMINFO */
+    msg_Dbg( p_dec, "decode STREAMINFO" );
+    p_sys->p_block = block_New( p_dec, p_dec->fmt_in.i_extra );
+    memcpy( p_sys->p_block->p_buffer, p_dec->fmt_in.p_extra,
+            p_dec->fmt_in.i_extra );
+    FLAC__stream_decoder_process_until_end_of_metadata( p_sys->p_flac );
+    msg_Dbg( p_dec, "STREAMINFO decoded" );
+
+    if( !p_sys->b_stream_info ) return;
+
+    if( p_dec->fmt_out.i_codec == VLC_FOURCC('f','l','a','c') )
+    {
+        p_dec->fmt_out.i_extra = p_dec->fmt_in.i_extra;
+        p_dec->fmt_out.p_extra =
+            realloc( p_dec->fmt_out.p_extra, p_dec->fmt_out.i_extra );
+        memcpy( p_dec->fmt_out.p_extra,
+                p_dec->fmt_in.p_extra, p_dec->fmt_out.i_extra );
+    }
+}
+
 /****************************************************************************
  * PacketizeBlock: the whole thing
  ****************************************************************************
@@ -257,12 +281,24 @@ static block_t *PacketizeBlock( decoder_t *p_dec, block_t **pp_block )
 
     if( !pp_block || !*pp_block ) return NULL;
 
-#if 0
-    if( (*pp_block)->b_discontinuity )
+    if( !p_sys->b_stream_info ) ProcessHeader( p_dec );
+
+    if( !aout_DateGet( &p_sys->end_date ) && !(*pp_block)->i_pts )
+    {
+        /* We've just started the stream, wait for the first PTS. */
+        block_Release( *pp_block );
+        return NULL;
+    }
+    else if( !aout_DateGet( &p_sys->end_date ) )
+    {
+        /* The first PTS is as good as anything else. */
+        aout_DateSet( &p_sys->end_date, (*pp_block)->i_pts );
+    }
+
+    if( (*pp_block)->i_flags&(BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED) )
     {
         p_sys->i_state = STATE_NOSYNC;
     }
-#endif
 
     block_BytestreamPush( &p_sys->bytestream, *pp_block );
 
@@ -325,7 +361,6 @@ static block_t *PacketizeBlock( decoder_t *p_dec, block_t **pp_block )
             {
                 p_dec->fmt_out.audio.i_rate = p_sys->i_rate;
                 aout_DateInit( &p_sys->end_date, p_sys->i_rate );
-                p_dec->fmt_out.audio.i_rate = p_sys->i_rate;
             }
             p_sys->i_state = STATE_NEXT_SYNC;
             p_sys->i_frame_size = 1;
@@ -343,14 +378,14 @@ static block_t *PacketizeBlock( decoder_t *p_dec, block_t **pp_block )
                 if( p_header[0] == 0xFF && p_header[1] == 0xF8 )
                 {
                     /* Check if frame is valid and get frame info */
-                    p_sys->i_frame_length =
+                    int i_frame_length =
                         SyncInfo( p_dec, p_header,
                                   &p_sys->i_channels,
                                   &p_sys->i_channels_conf,
                                   &p_sys->i_rate,
                                   &p_sys->i_bits_per_sample );
 
-                    if( p_sys->i_frame_length )
+                    if( i_frame_length )
                     {
                         p_sys->i_state = STATE_SEND_DATA;
                         break;
@@ -364,7 +399,6 @@ static block_t *PacketizeBlock( decoder_t *p_dec, block_t **pp_block )
                 /* Need more data */
                 return NULL;
             }
-            break;
 
         case STATE_SEND_DATA:
             p_sout_block = block_New( p_dec, p_sys->i_frame_size );
@@ -386,8 +420,9 @@ static block_t *PacketizeBlock( decoder_t *p_dec, block_t **pp_block )
             /* Date management */
             p_sout_block->i_pts =
                 p_sout_block->i_dts = aout_DateGet( &p_sys->end_date );
+            aout_DateIncrement( &p_sys->end_date, p_sys->i_frame_length );
             p_sout_block->i_length =
-                aout_DateIncrement( &p_sys->end_date, p_sys->i_frame_length );
+                aout_DateGet( &p_sys->end_date ) - p_sout_block->i_pts;
 
             return p_sout_block;
         }
@@ -405,13 +440,6 @@ static aout_buffer_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
 
     if( !pp_block || !*pp_block ) return NULL;
 
-    if( !aout_DateGet( &p_sys->end_date ) && !(*pp_block)->i_pts )
-    {
-        /* We've just started the stream, wait for the first PTS. */
-        block_Release( *pp_block );
-        return NULL;
-    }
-
     p_sys->p_aout_buffer = 0;
     if( ( p_sys->p_block = PacketizeBlock( p_dec, pp_block ) ) )
     {
@@ -431,6 +459,7 @@ static aout_buffer_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
         }
 
         block_Release( p_sys->p_block );
+        p_sys->p_block = NULL;
     }
 
     return p_sys->p_aout_buffer;
@@ -446,6 +475,7 @@ static void CloseDecoder( vlc_object_t *p_this )
 
     FLAC__stream_decoder_finish( p_sys->p_flac );
     FLAC__stream_decoder_delete( p_sys->p_flac );
+    if( p_sys->p_block ) free( p_sys->p_block );
     free( p_sys );
 }
     
@@ -486,18 +516,6 @@ DecoderWriteCallback( const FLAC__StreamDecoder *decoder,
     decoder_t *p_dec = (decoder_t *)client_data;
     decoder_sys_t *p_sys = p_dec->p_sys;
 
-    if( p_sys->p_block && p_sys->p_block->i_pts != 0 &&
-        p_sys->p_block->i_pts != aout_DateGet( &p_sys->end_date ) )
-    {
-        aout_DateSet( &p_sys->end_date, p_sys->p_block->i_pts );
-    }
-    else if( !aout_DateGet( &p_sys->end_date ) )
-    {
-        return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE;
-    }
-
-    if( p_sys->p_block ) p_sys->p_block->i_pts = 0;
-
     p_sys->p_aout_buffer =
         p_dec->pf_aout_buffer_new( p_dec, frame->header.blocksize );
 
@@ -515,10 +533,10 @@ DecoderWriteCallback( const FLAC__StreamDecoder *decoder,
                       frame->header.channels, frame->header.blocksize );
     }
 
-    /* Date management */
-    p_sys->p_aout_buffer->start_date = aout_DateGet( &p_sys->end_date );
+    /* Date management (already done by packetizer) */
+    p_sys->p_aout_buffer->start_date = p_sys->p_block->i_pts;
     p_sys->p_aout_buffer->end_date =
-        aout_DateIncrement( &p_sys->end_date, frame->header.blocksize );
+        p_sys->p_block->i_pts + p_sys->p_block->i_length;
 
     return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE;
 }
@@ -542,7 +560,7 @@ static void DecoderMetadataCallback( const FLAC__StreamDecoder *decoder,
         p_dec->fmt_out.i_codec = AOUT_FMT_S16_NE;
         break;
     default:
-        msg_Dbg( p_dec, "strange bps %d",
+        msg_Dbg( p_dec, "strange bit/sample value: %d",
                  metadata->data.stream_info.bits_per_sample );
         p_dec->fmt_out.i_codec = VLC_FOURCC('f','i','3','2');
         break;
@@ -581,14 +599,14 @@ static void DecoderErrorCallback( const FLAC__StreamDecoder *decoder,
     switch( status )
     {
     case FLAC__STREAM_DECODER_ERROR_STATUS_LOST_SYNC:
-        msg_Err( p_dec, "An error in the stream caused the decoder to "
-                 "loose synchronization." );
+        msg_Err( p_dec, "an error in the stream caused the decoder to "
+                 "lose synchronization." );
         break;
     case FLAC__STREAM_DECODER_ERROR_STATUS_BAD_HEADER:
-        msg_Err( p_dec, "The decoder encountered a corrupted frame header." );
+        msg_Err( p_dec, "the decoder encountered a corrupted frame header." );
         break;
     case FLAC__STREAM_DECODER_ERROR_STATUS_FRAME_CRC_MISMATCH:
-        msg_Err( p_dec, "The frame's data did not match the CRC in the "
+        msg_Err( p_dec, "frame's data did not match the CRC in the "
                  "footer." );
         break;
     default:
@@ -636,32 +654,32 @@ static void decoder_state_error( decoder_t *p_dec,
     switch ( state )
     {
     case FLAC__STREAM_DECODER_SEARCH_FOR_METADATA:
-        msg_Err( p_dec, "The decoder is ready to search for metadata." );
+        msg_Err( p_dec, "the decoder is ready to search for metadata." );
         break;
     case FLAC__STREAM_DECODER_READ_METADATA:
-        msg_Err( p_dec, "The decoder is ready to or is in the process of "
+        msg_Err( p_dec, "the decoder is ready to or is in the process of "
                  "reading metadata." );
         break;
     case FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC:
-        msg_Err( p_dec, "The decoder is ready to or is in the process of "
+        msg_Err( p_dec, "the decoder is ready to or is in the process of "
                  "searching for the frame sync code." );
         break;
     case FLAC__STREAM_DECODER_READ_FRAME:
-        msg_Err( p_dec, "The decoder is ready to or is in the process of "
+        msg_Err( p_dec, "the decoder is ready to or is in the process of "
                  "reading a frame." );
         break;
     case FLAC__STREAM_DECODER_END_OF_STREAM:
-        msg_Err( p_dec, "The decoder has reached the end of the stream." );
+        msg_Err( p_dec, "the decoder has reached the end of the stream." );
         break;
     case FLAC__STREAM_DECODER_ABORTED:
-        msg_Err( p_dec, "The decoder was aborted by the read callback." );
+        msg_Err( p_dec, "the decoder was aborted by the read callback." );
         break;
     case FLAC__STREAM_DECODER_UNPARSEABLE_STREAM:
-        msg_Err( p_dec, "The decoder encountered reserved fields in use "
+        msg_Err( p_dec, "the decoder encountered reserved fields in use "
                  "in the stream." );
         break;
     case FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR:
-        msg_Err( p_dec, "An error occurred allocating memory." );
+        msg_Err( p_dec, "error when allocating memory." );
         break;
     case FLAC__STREAM_DECODER_ALREADY_INITIALIZED:
         msg_Err( p_dec, "FLAC__stream_decoder_init() was called when the "
@@ -673,7 +691,7 @@ static void decoder_state_error( decoder_t *p_dec,
                  "all callbacks being set." );
         break;
     case FLAC__STREAM_DECODER_UNINITIALIZED:
-        msg_Err( p_dec, "The decoder is in the uninitialized state." );
+        msg_Err( p_dec, "decoder in uninitialized state." );
         break;
     default:
         msg_Err(p_dec, "unknown error" );
@@ -1077,7 +1095,8 @@ 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_FOURCC('f','l','a','c') )
+    if( p_enc->fmt_out.i_codec != VLC_FOURCC('f','l','a','c') &&
+        !p_enc->b_force )
     {
         return VLC_EGENERIC;
     }
@@ -1090,9 +1109,12 @@ static int OpenEncoder( vlc_object_t *p_this )
     }
     p_enc->p_sys = p_sys;
     p_enc->pf_encode_audio = Encode;
+    p_enc->fmt_out.i_codec = VLC_FOURCC('f','l','a','c');
+
     p_sys->i_headers = 0;
     p_sys->p_buffer = 0;
     p_sys->i_buffer = 0;
+    p_sys->i_samples_delay = 0;
 
     /* Create flac encoder */
     p_sys->p_flac = FLAC__stream_encoder_new();
@@ -1197,18 +1219,18 @@ EncoderWriteCallback( const FLAC__StreamEncoder *encoder,
     encoder_sys_t *p_sys = p_enc->p_sys;
     block_t *p_block;
 
-    if( samples == 0 && p_sys->i_headers <= 1 )
+    if( samples == 0 )
     {
         if( p_sys->i_headers == 1 )
         {
-            msg_Err( p_enc, "Writing STREAMINFO: %i", bytes );
+            msg_Dbg( p_enc, "Writing STREAMINFO: %i", bytes );
 
             /* Backup the STREAMINFO metadata block */
             p_enc->fmt_out.i_extra = STREAMINFO_SIZE + 4;
             p_enc->fmt_out.p_extra = malloc( STREAMINFO_SIZE + 4 );
             memcpy( p_enc->fmt_out.p_extra, "fLaC", 4 );
             memcpy( ((uint8_t *)p_enc->fmt_out.p_extra) + 4, buffer,
-                    STREAMINFO_SIZE + 4 );
+                    STREAMINFO_SIZE );
 
             /* Fake this as the last metadata block */
             ((uint8_t*)p_enc->fmt_out.p_extra)[4] |= 0x80;
@@ -1220,7 +1242,15 @@ EncoderWriteCallback( const FLAC__StreamEncoder *encoder,
     p_block = block_New( p_enc, bytes );
     memcpy( p_block->p_buffer, buffer, bytes );
 
-    p_block->i_dts = p_block->i_pts = p_block->i_length = 0;
+    p_block->i_dts = p_block->i_pts = p_sys->i_pts;
+
+    p_sys->i_samples_delay -= samples;
+
+    p_block->i_length = (mtime_t)1000000 *
+        (mtime_t)samples / (mtime_t)p_enc->fmt_in.audio.i_rate;
+
+    /* Update pts */
+    p_sys->i_pts += p_block->i_length;
 
     block_ChainAppend( &p_sys->p_chain, p_block );