/***************************************************************************** * dts.c: parse DTS audio sync info and packetize the stream ***************************************************************************** * Copyright (C) 2003-2005 the VideoLAN team * $Id$ * * Authors: Jon Lech Johansen <jon-vl@nanocrew.net> *          Gildas Bazin <gbazin@netcourrier.com> * * 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 * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA. *****************************************************************************//***************************************************************************** * Preamble *****************************************************************************/#ifdef HAVE_CONFIG_H# include "config.h"#endif#include <vlc_common.h>#include <vlc_plugin.h>#include <vlc_codec.h>#include <vlc_aout.h>#include <vlc_block_helper.h>#define DTS_HEADER_SIZE 14/***************************************************************************** * decoder_sys_t : decoder descriptor *****************************************************************************/struct decoder_sys_t{    /* Module mode */    bool b_packetizer;    /*     * Input properties     */    int i_state;    block_bytestream_t bytestream;    /*     * Common properties     */    audio_date_t   end_date;    mtime_t i_pts;    unsigned int i_bit_rate;    unsigned int i_frame_size;    unsigned int i_frame_length;    unsigned int i_rate;    unsigned int i_channels;    unsigned int i_channels_conf;};enum {    STATE_NOSYNC,    STATE_SYNC,    STATE_HEADER,    STATE_NEXT_SYNC,    STATE_GET_DATA,    STATE_SEND_DATA};/**************************************************************************** * Local prototypes ****************************************************************************/static int  OpenDecoder   ( vlc_object_t * );static int  OpenPacketizer( vlc_object_t * );static void CloseDecoder  ( vlc_object_t * );static void *DecodeBlock  ( decoder_t *, block_t ** );static inline int SyncCode( const uint8_t * );static int  SyncInfo      ( const uint8_t *, unsigned int *, unsigned int *,                            unsigned int *, unsigned int *, unsigned int * );static uint8_t       *GetOutBuffer ( decoder_t *, void ** );static aout_buffer_t *GetAoutBuffer( decoder_t * );static block_t       *GetSoutBuffer( decoder_t * );/***************************************************************************** * Module descriptor *****************************************************************************/vlc_module_begin();    set_description( N_("DTS parser") );    set_capability( "decoder", 100 );    set_callbacks( OpenDecoder, CloseDecoder );    add_submodule();    set_description( N_("DTS audio packetizer") );    set_capability( "packetizer", 10 );    set_callbacks( OpenPacketizer, CloseDecoder );vlc_module_end();/***************************************************************************** * OpenDecoder: probe the decoder and return score *****************************************************************************/static int OpenDecoder( vlc_object_t *p_this ){    decoder_t *p_dec = (decoder_t*)p_this;    decoder_sys_t *p_sys;    if( p_dec->fmt_in.i_codec != VLC_FOURCC('d','t','s',' ')         && p_dec->fmt_in.i_codec != VLC_FOURCC('d','t','s','b') )    {        return VLC_EGENERIC;    }    /* Allocate the memory needed to store the decoder's structure */    if( ( p_dec->p_sys = p_sys =          (decoder_sys_t *)malloc(sizeof(decoder_sys_t)) ) == NULL )        return VLC_ENOMEM;    /* Misc init */    p_sys->b_packetizer = false;    p_sys->i_state = STATE_NOSYNC;    aout_DateSet( &p_sys->end_date, 0 );    p_sys->bytestream = block_BytestreamInit();    /* Set output properties */    p_dec->fmt_out.i_cat = AUDIO_ES;    p_dec->fmt_out.i_codec = VLC_FOURCC('d','t','s',' ');    p_dec->fmt_out.audio.i_rate = 0; /* So end_date gets initialized */    /* Set callback */    p_dec->pf_decode_audio = (aout_buffer_t *(*)(decoder_t *, block_t **))        DecodeBlock;    p_dec->pf_packetize    = (block_t *(*)(decoder_t *, block_t **))        DecodeBlock;    return VLC_SUCCESS;}static int OpenPacketizer( vlc_object_t *p_this ){    decoder_t *p_dec = (decoder_t*)p_this;    int i_ret = OpenDecoder( p_this );    if( i_ret == VLC_SUCCESS ) p_dec->p_sys->b_packetizer = true;    return i_ret;}/**************************************************************************** * DecodeBlock: the whole thing **************************************************************************** * This function is called just after the thread is launched. ****************************************************************************/static void *DecodeBlock( decoder_t *p_dec, block_t **pp_block ){    decoder_sys_t *p_sys = p_dec->p_sys;    uint8_t p_header[DTS_HEADER_SIZE];    uint8_t *p_buf;    void *p_out_buffer;    if( !pp_block || !*pp_block ) return NULL;    if( (*pp_block)->i_flags&(BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED) )    {        if( (*pp_block)->i_flags&BLOCK_FLAG_CORRUPTED )        {            p_sys->i_state = STATE_NOSYNC;            block_BytestreamFlush( &p_sys->bytestream );        }//        aout_DateSet( &p_sys->end_date, 0 );        block_Release( *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;    }    block_BytestreamPush( &p_sys->bytestream, *pp_block );    while( 1 )    {        switch( p_sys->i_state )        {        case STATE_NOSYNC:            /* Look for sync code - should be 0x7ffe8001 */            while( block_PeekBytes( &p_sys->bytestream, p_header, 6 )                   == VLC_SUCCESS )            {                if( SyncCode( p_header ) == VLC_SUCCESS )                {                    p_sys->i_state = STATE_SYNC;                    break;                }                block_SkipByte( &p_sys->bytestream );            }            if( p_sys->i_state != STATE_SYNC )            {                block_BytestreamFlush( &p_sys->bytestream );                /* Need more data */                return NULL;            }        case STATE_SYNC:            /* New frame, set the Presentation Time Stamp */            p_sys->i_pts = p_sys->bytestream.p_block->i_pts;            if( p_sys->i_pts != 0 &&                p_sys->i_pts != aout_DateGet( &p_sys->end_date ) )            {                aout_DateSet( &p_sys->end_date, p_sys->i_pts );            }            p_sys->i_state = STATE_HEADER;        case STATE_HEADER:            /* Get DTS frame header (DTS_HEADER_SIZE bytes) */            if( block_PeekBytes( &p_sys->bytestream, p_header,                                 DTS_HEADER_SIZE ) != VLC_SUCCESS )            {                /* Need more data */                return NULL;            }            /* Check if frame is valid and get frame info */            p_sys->i_frame_size = SyncInfo( p_header,                                            &p_sys->i_channels,                                            &p_sys->i_channels_conf,                                            &p_sys->i_rate,                                            &p_sys->i_bit_rate,                                            &p_sys->i_frame_length );            if( !p_sys->i_frame_size )            {                msg_Dbg( p_dec, "emulated sync word" );                block_SkipByte( &p_sys->bytestream );                p_sys->i_state = STATE_NOSYNC;                break;            }            p_sys->i_state = STATE_NEXT_SYNC;        case STATE_NEXT_SYNC:            /* TODO: If pp_block == NULL, flush the buffer without checking the             * next sync word */            /* Check if next expected frame contains the sync word */            if( block_PeekOffsetBytes( &p_sys->bytestream,                                       p_sys->i_frame_size, p_header, 6 )                != VLC_SUCCESS )            {                /* Need more data */                return NULL;            }            if( SyncCode( p_header ) != VLC_SUCCESS )            {                msg_Dbg( p_dec, "emulated sync word "                         "(no sync on following frame): %2.2x%2.2x%2.2x%2.2x",                         (int)p_header[0], (int)p_header[1],                         (int)p_header[2], (int)p_header[3] );                p_sys->i_state = STATE_NOSYNC;                block_SkipByte( &p_sys->bytestream );                break;            }            p_sys->i_state = STATE_SEND_DATA;            break;        case STATE_GET_DATA:            /* Make sure we have enough data.             * (Not useful if we went through NEXT_SYNC) */            if( block_WaitBytes( &p_sys->bytestream,                                 p_sys->i_frame_size ) != VLC_SUCCESS )            {                /* Need more data */                return NULL;            }            p_sys->i_state = STATE_SEND_DATA;        case STATE_SEND_DATA:            if( !(p_buf = GetOutBuffer( p_dec, &p_out_buffer )) )            {                //p_dec->b_error = true;                return NULL;            }            /* Copy the whole frame into the buffer. When we reach this point             * we already know we have enough data available. */            block_GetBytes( &p_sys->bytestream, p_buf, p_sys->i_frame_size );            /* Make sure we don't reuse the same pts twice */            if( p_sys->i_pts == p_sys->bytestream.p_block->i_pts )                p_sys->i_pts = p_sys->bytestream.p_block->i_pts = 0;            p_sys->i_state = STATE_NOSYNC;            /* So p_block doesn't get re-added several times */            *pp_block = block_BytestreamPop( &p_sys->bytestream );            return p_out_buffer;        }    }    return NULL;}/***************************************************************************** * CloseDecoder: clean up the decoder *****************************************************************************/static void CloseDecoder( vlc_object_t *p_this ){    decoder_t *p_dec = (decoder_t*)p_this;    decoder_sys_t *p_sys = p_dec->p_sys;    block_BytestreamRelease( &p_sys->bytestream );    free( p_sys );}/***************************************************************************** * GetOutBuffer: *****************************************************************************/static uint8_t *GetOutBuffer( decoder_t *p_dec, void **pp_out_buffer ){    decoder_sys_t *p_sys = p_dec->p_sys;    uint8_t *p_buf;    if( p_dec->fmt_out.audio.i_rate != p_sys->i_rate )    {        msg_Info( p_dec, "DTS channels:%d samplerate:%d bitrate:%d",                  p_sys->i_channels, p_sys->i_rate, p_sys->i_bit_rate );        aout_DateInit( &p_sys->end_date, p_sys->i_rate );        aout_DateSet( &p_sys->end_date, p_sys->i_pts );    }    p_dec->fmt_out.audio.i_rate     = p_sys->i_rate;    p_dec->fmt_out.audio.i_channels = p_sys->i_channels;    /* Hack for DTS S/PDIF filter which needs to pad the DTS frames */    p_dec->fmt_out.audio.i_bytes_per_frame =        __MAX( p_sys->i_frame_size, p_sys->i_frame_length * 4 );    p_dec->fmt_out.audio.i_frame_length = p_sys->i_frame_length;    p_dec->fmt_out.audio.i_original_channels = p_sys->i_channels_conf;