Mine

/***************************************************************************** * speex.c: speex decoder/packetizer/encoder module making use of libspeex. ***************************************************************************** * Copyright (C) 2003 the VideoLAN team * $Id: a5ccbbe358bde89b80c59bd2791d36efbc24d04c $ * * Authors: 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 * 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_input.h>#include <vlc_codec.h>#include <vlc_aout.h>#include <ogg/ogg.h>#include <speex/speex.h>#include <speex/speex_header.h>#include <speex/speex_stereo.h>#include <speex/speex_callbacks.h>#include <assert.h>/***************************************************************************** * The blue stuff means it doesn't count *****************************************************************************/static int  OpenDecoder   ( vlc_object_t * );static int  OpenPacketizer( vlc_object_t * );static void CloseDecoder  ( vlc_object_t * );static int OpenEncoder   ( vlc_object_t * );static void CloseEncoder ( vlc_object_t * );#define ENC_CFG_PREFIX "sout-speex-"#define ENC_MODE_TEXT N_("Mode" )#define ENC_MODE_LONGTEXT N_( \    "Enforce the mode of the encoder." )#define ENC_QUALITY_TEXT N_("Encoding quality")#define ENC_QUALITY_LONGTEXT N_( \    "Enforce a quality between 0 (low) and 10 (high)." )#define ENC_COMPLEXITY_TEXT N_("Encoding complexity" )#define ENC_COMPLEXITY_LONGTEXT N_( \    "Enforce the complexity of the encoder." )#define ENC_MAXBITRATE_TEXT N_( "Maximal bitrate" )#define ENC_MAXBITRATE_LONGTEXT N_( \    "Enforce the maximal VBR bitrate" )#define ENC_CBR_TEXT N_( "CBR encoding" )#define ENC_CBR_LONGTEXT N_( \    "Enforce a constant bitrate encoding (CBR) instead of default " \    "variable bitrate encoding (VBR)." )#define ENC_VAD_TEXT N_( "Voice activity detection" )#define ENC_VAD_LONGTEXT N_( \    "Enable voice activity detection (VAD). It is automatically " \    "activated in VBR mode." )#define ENC_DTX_TEXT N_( "Discontinuous Transmission" )#define ENC_DTX_LONGTEXT N_( \    "Enable discontinuous transmission (DTX)." )static const int pi_enc_mode_values[] = { 0, 1, 2 };static const char * const ppsz_enc_mode_descriptions[] = {    N_("Narrow-band (8kHz)"), N_("Wide-band (16kHz)"), N_("Ultra-wideband (32kHz)"), NULL};vlc_module_begin();    set_category( CAT_INPUT );    set_subcategory( SUBCAT_INPUT_ACODEC );    set_description( N_("Speex audio decoder") );    set_capability( "decoder", 100 );    set_callbacks( OpenDecoder, CloseDecoder );    add_submodule();    set_description( N_("Speex audio packetizer") );    set_capability( "packetizer", 100 );    set_callbacks( OpenPacketizer, CloseDecoder );    add_submodule();    set_description( N_("Speex audio encoder") );    set_capability( "encoder", 100 );    set_callbacks( OpenEncoder, CloseEncoder );    add_integer( ENC_CFG_PREFIX "mode", 0, NULL, ENC_MODE_TEXT,                 ENC_MODE_LONGTEXT, false );        change_integer_list( pi_enc_mode_values, ppsz_enc_mode_descriptions, NULL );    add_integer( ENC_CFG_PREFIX "complexity", 3, NULL, ENC_COMPLEXITY_TEXT,                 ENC_COMPLEXITY_LONGTEXT, false );        change_integer_range( 1, 10 );    add_bool( ENC_CFG_PREFIX "cbr", false, NULL, ENC_CBR_TEXT,                 ENC_CBR_LONGTEXT, false );    add_float( ENC_CFG_PREFIX "quality", 8.0, NULL, ENC_QUALITY_TEXT,               ENC_QUALITY_LONGTEXT, false );        change_float_range( 0.0, 10.0 );    add_integer( ENC_CFG_PREFIX "max-bitrate", 0, NULL, ENC_MAXBITRATE_TEXT,                 ENC_MAXBITRATE_LONGTEXT, false );    add_bool( ENC_CFG_PREFIX "vad", true, NULL, ENC_VAD_TEXT,                 ENC_VAD_LONGTEXT, false );    add_bool( ENC_CFG_PREFIX "dtx", false, NULL, ENC_DTX_TEXT,                 ENC_DTX_LONGTEXT, false );    /* TODO agc, noise suppression, */vlc_module_end();static const char *const ppsz_enc_options[] = {    "mode", "complexity", "cbr", "quality", "max-bitrate", "vad", "dtx", NULL};/***************************************************************************** * AH OOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOO! *****************************************************************************/struct decoder_sys_t{    /* Module mode */    bool b_packetizer;    /*     * Input properties     */    int i_headers;    int i_frame_in_packet;    /*     * Speex properties     */    SpeexBits bits;    SpeexHeader *p_header;    SpeexStereoState stereo;    void *p_state;    unsigned int rtp_rate;    /*     * Common properties     */    audio_date_t end_date;};static const int pi_channels_maps[6] ={    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};/**************************************************************************** * Local prototypes ****************************************************************************/static void *DecodeBlock  ( decoder_t *, block_t ** );static aout_buffer_t *DecodeRtpSpeexPacket( decoder_t *, block_t **);static int  ProcessHeaders( decoder_t * );static int  ProcessInitialHeader ( decoder_t *, ogg_packet * );static void *ProcessPacket( decoder_t *, ogg_packet *, block_t ** );static aout_buffer_t *DecodePacket( decoder_t *, ogg_packet * );static block_t *SendPacket( decoder_t *, block_t * );static void ParseSpeexComments( decoder_t *, ogg_packet * );static block_t *Encode   ( encoder_t *, aout_buffer_t * );/***************************************************************************** * 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 = p_dec->p_sys;    if( p_dec->fmt_in.i_codec != VLC_FOURCC('s','p','x',' ')         && p_dec->fmt_in.i_codec != VLC_FOURCC('s', 'p', 'x', 'r') )    {        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;    p_dec->p_sys->bits.buf_size = 0;    p_dec->p_sys->b_packetizer = false;    p_dec->p_sys->rtp_rate = p_dec->fmt_in.audio.i_rate;    aout_DateSet( &p_sys->end_date, 0 );    /* Set output properties */    p_dec->fmt_out.i_cat = AUDIO_ES;    p_dec->fmt_out.i_codec = AOUT_FMT_S16_NE;    /*      Set callbacks      If the codec is spxr then this decoder is       being invoked on a Speex stream arriving via RTP.       A special decoder callback is used.    */    if (p_dec->fmt_in.i_codec == VLC_FOURCC('s', 'p', 'x', 'r'))    {        msg_Dbg( p_dec, "Using RTP version of Speex decoder @ rate %d.", 	    p_dec->fmt_in.audio.i_rate );        p_dec->pf_decode_audio = (aout_buffer_t *(*)(decoder_t *, block_t **))            DecodeRtpSpeexPacket;    }    else    {        p_dec->pf_decode_audio = (aout_buffer_t *(*)(decoder_t *, block_t **))            DecodeBlock;    }    p_dec->pf_packetize    = (block_t *(*)(decoder_t *, block_t **))        DecodeBlock;    p_sys->i_headers = 0;    p_sys->p_state = NULL;    p_sys->p_header = NULL;    p_sys->i_frame_in_packet = 0;    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;        p_dec->fmt_out.i_codec = VLC_FOURCC('s','p','x',' ');    }    return i_ret;}/**************************************************************************** * DecodeBlock: the whole thing **************************************************************************** * This function must be fed with ogg packets. ****************************************************************************/static void *DecodeBlock( decoder_t *p_dec, block_t **pp_block ){    decoder_sys_t *p_sys = p_dec->p_sys;    ogg_packet oggpacket;    if( !pp_block ) return NULL;    if( *pp_block )    {        /* Block to Ogg packet */        oggpacket.packet = (*pp_block)->p_buffer;        oggpacket.bytes = (*pp_block)->i_buffer;    }    else    {        if( p_sys->b_packetizer ) return NULL;        /* Block to Ogg packet */        oggpacket.packet = NULL;        oggpacket.bytes = 0;    }    oggpacket.granulepos = -1;    oggpacket.b_o_s = 0;    oggpacket.e_o_s = 0;    oggpacket.packetno = 0;    /* Check for headers */    if( p_sys->i_headers == 0 && p_dec->fmt_in.i_extra )    {        p_sys->i_headers = 2;    }    else if( oggpacket.bytes && p_sys->i_headers < 2 )    {        uint8_t *p_extra;        p_dec->fmt_in.p_extra =            realloc( p_dec->fmt_in.p_extra, p_dec->fmt_in.i_extra +                     oggpacket.bytes + 2 );        p_extra = ((uint8_t *)p_dec->fmt_in.p_extra) + p_dec->fmt_in.i_extra;        *(p_extra++) = oggpacket.bytes >> 8;        *(p_extra++) = oggpacket.bytes & 0xFF;        memcpy( p_extra, oggpacket.packet, oggpacket.bytes );        p_dec->fmt_in.i_extra += oggpacket.bytes + 2;        block_Release( *pp_block );        p_sys->i_headers++;        return NULL;    }    if( p_sys->i_headers == 2 )    {        if( ProcessHeaders( p_dec ) != VLC_SUCCESS )        {            p_sys->i_headers = 0;            p_dec->fmt_in.i_extra = 0;            block_Release( *pp_block );            return NULL;        }        else p_sys->i_headers++;    }    return ProcessPacket( p_dec, &oggpacket, pp_block );}/***************************************************************************** * ProcessHeaders: process Speex headers. *****************************************************************************/static int ProcessHeaders( decoder_t *p_dec ){    decoder_sys_t *p_sys = p_dec->p_sys;    ogg_packet oggpacket;    uint8_t *p_extra;    int i_extra;    if( !p_dec->fmt_in.i_extra ) return VLC_EGENERIC;    oggpacket.granulepos = -1;    oggpacket.b_o_s = 1; /* yes this actually is a b_o_s packet :) */    oggpacket.e_o_s = 0;    oggpacket.packetno = 0;    p_extra = p_dec->fmt_in.p_extra;    i_extra = p_dec->fmt_in.i_extra;    /* Take care of the initial Vorbis header */    oggpacket.bytes = *(p_extra++) << 8;    oggpacket.bytes |= (*(p_extra++) & 0xFF);    oggpacket.packet = p_extra;    p_extra += oggpacket.bytes;    i_extra -= (oggpacket.bytes + 2);    if( i_extra < 0 )    {        msg_Err( p_dec, "header data corrupted");        return VLC_EGENERIC;    }    /* Take care of the initial Speex header */    if( ProcessInitialHeader( p_dec, &oggpacket ) != VLC_SUCCESS )    {        msg_Err( p_dec, "initial Speex header is corrupted" );        return VLC_EGENERIC;    }    /* The next packet in order is the comments header */    oggpacket.b_o_s = 0;    oggpacket.bytes = *(p_extra++) << 8;    oggpacket.bytes |= (*(p_extra++) & 0xFF);    oggpacket.packet = p_extra;    p_extra += oggpacket.bytes;    i_extra -= (oggpacket.bytes + 2);    if( i_extra < 0 )    {        msg_Err( p_dec, "header data corrupted");        return VLC_EGENERIC;    }    ParseSpeexComments( p_dec, &oggpacket );    if( p_sys->b_packetizer )    {        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 );    }    return VLC_SUCCESS;}/***************************************************************************** * heh *****************************************************************************//***************************************************************************** * ha *****************************************************************************/static aout_buffer_t *DecodePacket( decoder_t *p_dec, ogg_packet *p_oggpacket ){    decoder_sys_t *p_sys = p_dec->p_sys;    if( p_oggpacket->bytes )    {        /* Copy Ogg packet to Speex bitstream */        speex_bits_read_from( &p_sys->bits, (char *)p_oggpacket->packet,                              p_oggpacket->bytes );        p_sys->i_frame_in_packet = 0;    }    /* Decode one frame at a time */    if( p_sys->i_frame_in_packet < p_sys->p_header->frames_per_packet )    {        aout_buffer_t *p_aout_buffer;        if( p_sys->p_header->frame_size == 0 )            return NULL;        p_aout_buffer =            p_dec->pf_aout_buffer_new( p_dec, p_sys->p_header->frame_size );        if( !p_aout_buffer )        {            return NULL;        }        switch( speex_decode_int( p_sys->p_state, &p_sys->bits,                                  (int16_t *)p_aout_buffer->p_buffer ) )        {            case -2:                msg_Err( p_dec, "decoding error: corrupted stream?" );            case -1: /* End of stream */                return NULL;        }        if( speex_bits_remaining( &p_sys->bits ) < 0 )        {            msg_Err( p_dec, "decoding overflow: corrupted stream?" );        }        if( p_sys->p_header->nb_channels == 2 )            speex_decode_stereo_int( (int16_t *)p_aout_buffer->p_buffer,                                     p_sys->p_header->frame_size,                                     &p_sys->stereo );        /* Date management */        p_aout_buffer->start_date = aout_DateGet( &p_sys->end_date );        p_aout_buffer->end_date =            aout_DateIncrement( &p_sys->end_date, p_sys->p_header->frame_size );        p_sys->i_frame_in_packet++;        return p_aout_buffer;    }    else    {        return NULL;    }}/***************************************************************************** * SendPacket: send an ogg packet to the stream output. *****************************************************************************/static block_t *SendPacket( decoder_t *p_dec, block_t *p_block ){    decoder_sys_t *p_sys = p_dec->p_sys;    /* Date management */    p_block->i_dts = p_block->i_pts = aout_DateGet( &p_sys->end_date );    if( p_sys->i_headers >= p_sys->p_header->extra_headers + 2 )    {        p_block->i_length =            aout_DateIncrement( &p_sys->end_date,                                p_sys->p_header->frame_size ) -            p_block->i_pts;    }    else        p_block->i_length = 0;    return p_block;}/***************************************************************************** * ParseSpeexComments: FIXME should be done in demuxer *****************************************************************************/#define readint(buf, base) (((buf[base+3]<<24)&0xff000000)| \                           ((buf[base+2]<<16)&0xff0000)| \                           ((buf[base+1]<<8)&0xff00)| \                            (buf[base]&0xff))static void ParseSpeexComments( decoder_t *p_dec, ogg_packet *p_oggpacket ){    input_thread_t *p_input = (input_thread_t *)p_dec->p_parent;    decoder_sys_t *p_sys = p_dec->p_sys;    char *p_buf = (char *)p_oggpacket->packet;    const SpeexMode *p_mode;    int i_len;    if( p_input->i_object_type != VLC_OBJECT_INPUT ) return;    assert( p_sys->p_header->mode < SPEEX_NB_MODES );    p_mode = speex_mode_list[p_sys->p_header->mode];    assert( p_mode != NULL );    input_Control( p_input, INPUT_ADD_INFO, _("Speex comment"), _("Mode"),                   "%s%s", p_mode->modeName,                   p_sys->p_header->vbr ? " VBR" : "" );    if( p_oggpacket->bytes < 8 )    {        msg_Err( p_dec, "invalid/corrupted comments" );        return;    }    i_len = readint( p_buf, 0 ); p_buf += 4;    if( i_len > p_oggpacket->bytes - 4 )    {        msg_Err( p_dec, "invalid/corrupted comments" );        return;    }    input_Control( p_input, INPUT_ADD_INFO, _("Speex comment"), p_buf, "" );    /* TODO: finish comments parsing */}/***************************************************************************** * CloseDecoder: speex decoder destruction *****************************************************************************/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;    if( p_sys->p_state )    {        speex_decoder_destroy( p_sys->p_state );        speex_bits_destroy( &p_sys->bits );    }    free( p_sys->p_header );    free( p_sys );}/***************************************************************************** * encoder_sys_t: encoder descriptor *****************************************************************************/#define MAX_FRAME_SIZE  2000#define MAX_FRAME_BYTES 2000struct encoder_sys_t{    /*     * Input properties     */    char *p_buffer;    char p_buffer_out[MAX_FRAME_BYTES];    /*     * Speex properties     */    SpeexBits bits;    SpeexHeader header;    SpeexStereoState stereo;    void *p_state;    int i_frames_per_packet;    int i_frames_in_packet;    int i_frame_length;    int i_samples_delay;    int i_frame_size;    /*     * Common properties     */    mtime_t i_pts;};/***************************************************************************** * ahaha *****************************************************************************///GIANT FOO FOO!/**************************************************************************** * ahahaha **************************************************************************** * AH HEHEHEHE!!!! ****************************************************************************/static block_t *Encode( encoder_t *p_enc, aout_buffer_t *p_aout_buf ){    encoder_sys_t *p_sys = p_enc->p_sys;    block_t *p_block, *p_chain = NULL;    unsigned char *p_buffer = p_aout_buf->p_buffer;    int i_samples = p_aout_buf->i_nb_samples;    int i_samples_delay = p_sys->i_samples_delay;    p_sys->i_pts = p_aout_buf->start_date -                (mtime_t)1000000 * (mtime_t)p_sys->i_samples_delay /                (mtime_t)p_enc->fmt_in.audio.i_rate;    p_sys->i_samples_delay += i_samples;    while( p_sys->i_samples_delay >= p_sys->i_frame_length )    {        int16_t *p_samples;        int i_out;        if( i_samples_delay )        {            /* Take care of the left-over from last time */            int i_delay_size = i_samples_delay * 2 *                                 p_enc->fmt_in.audio.i_channels;            int i_size = p_sys->i_frame_size - i_delay_size;            p_samples = (int16_t *)p_sys->p_buffer;            memcpy( p_sys->p_buffer + i_delay_size, p_buffer, i_size );            p_buffer -= i_delay_size;            i_samples += i_samples_delay;            i_samples_delay = 0;        }        else        {            p_samples = (int16_t *)p_buffer;        }        /* Encode current frame */        if( p_enc->fmt_in.audio.i_channels == 2 )            speex_encode_stereo_int( p_samples, p_sys->i_frame_length,                                     &p_sys->bits );#if 0        if( p_sys->preprocess )            speex_preprocess( p_sys->preprocess, p_samples, NULL );#endif        speex_encode_int( p_sys->p_state, p_samples, &p_sys->bits );        p_buffer += p_sys->i_frame_size;        p_sys->i_samples_delay -= p_sys->i_frame_length;        i_samples -= p_sys->i_frame_length;        p_sys->i_frames_in_packet++;        if( p_sys->i_frames_in_packet < p_sys->header.frames_per_packet )            continue;        p_sys->i_frames_in_packet = 0;        speex_bits_insert_terminator( &p_sys->bits );        i_out = speex_bits_write( &p_sys->bits, p_sys->p_buffer_out,                                  MAX_FRAME_BYTES );        speex_bits_reset( &p_sys->bits );        p_block = block_New( p_enc, i_out );        memcpy( p_block->p_buffer, p_sys->p_buffer_out, i_out );        p_block->i_length = (mtime_t)1000000 *            (mtime_t)p_sys->i_frame_length * p_sys->header.frames_per_packet /            (mtime_t)p_enc->fmt_in.audio.i_rate;        p_block->i_dts = p_block->i_pts = p_sys->i_pts;        /* Update pts */        p_sys->i_pts += p_block->i_length;        block_ChainAppend( &p_chain, p_block );    }    /* Backup the remaining raw samples */    if( i_samples )    {        memcpy( p_sys->p_buffer + i_samples_delay * 2 *                p_enc->fmt_in.audio.i_channels, p_buffer,                i_samples * 2 * p_enc->fmt_in.audio.i_channels );    }    return p_chain;}/***************************************************************************** * CloseEncoder: encoder destruction *****************************************************************************/static void CloseEncoder( vlc_object_t *p_this ){    encoder_t *p_enc = (encoder_t *)p_this;    encoder_sys_t *p_sys = p_enc->p_sys;    speex_encoder_destroy( p_sys->p_state );    speex_bits_destroy( &p_sys->bits );    free( p_sys->p_buffer );    free( p_sys );}