X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;ds=inline;f=modules%2Fcodec%2Favcodec%2Faudio.c;h=483c3067c84515774f7747c3b45705de54cfad2c;hb=6233041035593e0c83daf28992087c5a45c4cccd;hp=77413fd4155759aeef7c6b5aacb5d7f168d7ea9a;hpb=5300a9e008d3f0c477c14472faf52bfb52efb372;p=vlc diff --git a/modules/codec/avcodec/audio.c b/modules/codec/avcodec/audio.c index 77413fd415..483c3067c8 100644 --- a/modules/codec/avcodec/audio.c +++ b/modules/codec/avcodec/audio.c @@ -78,42 +78,12 @@ struct decoder_sys_t int64_t i_previous_layout; }; +#define BLOCK_FLAG_PRIVATE_REALLOCATED (1 << BLOCK_FLAG_PRIVATE_SHIFT) + static void SetupOutputFormat( decoder_t *p_dec, bool b_trust ); -/***************************************************************************** - * InitAudioDec: initialize audio decoder - ***************************************************************************** - * The ffmpeg codec will be opened, some memory allocated. - *****************************************************************************/ -int InitAudioDec( decoder_t *p_dec, AVCodecContext *p_context, - AVCodec *p_codec, int i_codec_id, const char *psz_namecodec ) +static void InitDecoderConfig( decoder_t *p_dec, AVCodecContext *p_context ) { - decoder_sys_t *p_sys; - - /* Allocate the memory needed to store the decoder's structure */ - if( ( p_dec->p_sys = p_sys = malloc(sizeof(*p_sys)) ) == NULL ) - { - return VLC_ENOMEM; - } - - p_sys->p_context = p_context; - p_sys->p_codec = p_codec; - p_sys->i_codec_id = i_codec_id; - p_sys->psz_namecodec = psz_namecodec; - p_sys->b_delayed_open = false; - - /* ***** Fill p_context with init values ***** */ - p_sys->p_context->sample_rate = p_dec->fmt_in.audio.i_rate; - p_sys->p_context->channels = p_dec->fmt_in.audio.i_channels; - - p_sys->p_context->block_align = p_dec->fmt_in.audio.i_blockalign; - p_sys->p_context->bit_rate = p_dec->fmt_in.i_bitrate; -#if LIBAVCODEC_VERSION_INT < ((52<<16)+(0<<8)+0) - p_sys->p_context->bits_per_sample = p_dec->fmt_in.audio.i_bitspersample; -#else - p_sys->p_context->bits_per_coded_sample = p_dec->fmt_in.audio.i_bitspersample; -#endif - if( p_dec->fmt_in.i_extra > 0 ) { const uint8_t * const p_src = p_dec->fmt_in.p_extra; @@ -146,13 +116,13 @@ int InitAudioDec( decoder_t *p_dec, AVCodecContext *p_context, if( i_size > 0 ) { - p_sys->p_context->extradata = + p_context->extradata = malloc( i_size + FF_INPUT_BUFFER_PADDING_SIZE ); - if( p_sys->p_context->extradata ) + if( p_context->extradata ) { - uint8_t *p_dst = p_sys->p_context->extradata; + uint8_t *p_dst = p_context->extradata; - p_sys->p_context->extradata_size = i_size; + p_context->extradata_size = i_size; memcpy( &p_dst[0], &p_src[i_offset], i_size ); memset( &p_dst[i_size], 0, FF_INPUT_BUFFER_PADDING_SIZE ); @@ -161,16 +131,41 @@ int InitAudioDec( decoder_t *p_dec, AVCodecContext *p_context, } else { - p_sys->p_context->extradata_size = 0; - p_sys->p_context->extradata = NULL; + p_context->extradata_size = 0; + p_context->extradata = NULL; + } +} + +/***************************************************************************** + * InitAudioDec: initialize audio decoder + ***************************************************************************** + * The ffmpeg codec will be opened, some memory allocated. + *****************************************************************************/ +int InitAudioDec( decoder_t *p_dec, AVCodecContext *p_context, + AVCodec *p_codec, int i_codec_id, const char *psz_namecodec ) +{ + decoder_sys_t *p_sys; + + /* Allocate the memory needed to store the decoder's structure */ + if( ( p_dec->p_sys = p_sys = malloc(sizeof(*p_sys)) ) == NULL ) + { + return VLC_ENOMEM; } + p_codec->type = CODEC_TYPE_AUDIO; + p_context->codec_type = CODEC_TYPE_AUDIO; + p_context->codec_id = i_codec_id; + p_sys->p_context = p_context; + p_sys->p_codec = p_codec; + p_sys->i_codec_id = i_codec_id; + p_sys->psz_namecodec = psz_namecodec; + p_sys->b_delayed_open = false; + + // Initialize decoder extradata + InitDecoderConfig( p_dec, p_context); + /* ***** Open the codec ***** */ - int ret; - vlc_avcodec_lock(); - ret = avcodec_open( p_sys->p_context, p_sys->p_codec ); - vlc_avcodec_unlock(); - if( ret < 0 ) + if( ffmpeg_OpenCodec( p_dec ) < 0 ) { msg_Err( p_dec, "cannot open codec (%s)", p_sys->psz_namecodec ); free( p_sys->p_context->extradata ); @@ -178,8 +173,6 @@ int InitAudioDec( decoder_t *p_dec, AVCodecContext *p_context, return VLC_EGENERIC; } - msg_Dbg( p_dec, "ffmpeg codec (%s) started", p_sys->psz_namecodec ); - switch( i_codec_id ) { case CODEC_ID_WAVPACK: @@ -191,9 +184,11 @@ int InitAudioDec( decoder_t *p_dec, AVCodecContext *p_context, case CODEC_ID_FLAC: p_sys->i_output_max = 8 * sizeof(int32_t) * 65535; break; +#if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT( 52, 35, 0 ) case CODEC_ID_WMAPRO: - p_sys->i_output_max = 8 * sizeof(float) * 6145; /* (1 << 12) * 3/2 */ + p_sys->i_output_max = 8 * sizeof(float) * 6144; /* (1 << 12) * 3/2 */ break; +#endif default: p_sys->i_output_max = 0; break; @@ -210,15 +205,17 @@ int InitAudioDec( decoder_t *p_dec, AVCodecContext *p_context, p_sys->i_previous_channels = 0; p_sys->i_previous_layout = 0; - date_Set( &p_sys->end_date, 0 ); - if( p_dec->fmt_in.audio.i_rate ) - date_Init( &p_sys->end_date, p_dec->fmt_in.audio.i_rate, 1 ); - /* */ p_dec->fmt_out.i_cat = AUDIO_ES; - /* Try to set as much informations as possible but do not trust it */ + /* Try to set as much information as possible but do not trust it */ SetupOutputFormat( p_dec, false ); + date_Set( &p_sys->end_date, 0 ); + if( p_dec->fmt_out.audio.i_rate ) + date_Init( &p_sys->end_date, p_dec->fmt_out.audio.i_rate, 1 ); + else if( p_dec->fmt_in.audio.i_rate ) + date_Init( &p_sys->end_date, p_dec->fmt_in.audio.i_rate, 1 ); + return VLC_SUCCESS; } @@ -237,15 +234,16 @@ static aout_buffer_t *SplitBuffer( decoder_t *p_dec ) if( ( p_buffer = decoder_NewAudioBuffer( p_dec, i_samples ) ) == NULL ) return NULL; - p_buffer->start_date = date_Get( &p_sys->end_date ); - p_buffer->end_date = date_Increment( &p_sys->end_date, i_samples ); + p_buffer->i_pts = date_Get( &p_sys->end_date ); + p_buffer->i_length = date_Increment( &p_sys->end_date, i_samples ) + - p_buffer->i_pts; if( p_sys->b_extract ) aout_ChannelExtract( p_buffer->p_buffer, p_dec->fmt_out.audio.i_channels, p_sys->p_samples, p_sys->p_context->channels, i_samples, p_sys->pi_extraction, p_dec->fmt_out.audio.i_bitspersample ); else - memcpy( p_buffer->p_buffer, p_sys->p_samples, p_buffer->i_nb_bytes ); + memcpy( p_buffer->p_buffer, p_sys->p_samples, p_buffer->i_buffer ); p_sys->p_samples += i_samples * p_sys->p_context->channels * ( p_dec->fmt_out.audio.i_bitspersample / 8 ); p_sys->i_samples -= i_samples; @@ -300,45 +298,48 @@ aout_buffer_t * DecodeAudio ( decoder_t *p_dec, block_t **pp_block ) return NULL; } - i_output = __MAX( p_block->i_buffer, p_sys->i_output_max ); - if( i_output > p_sys->i_output_max ) + if( (p_block->i_flags & BLOCK_FLAG_PRIVATE_REALLOCATED) == 0 ) { - /* Grow output buffer if necessary (eg. for PCM data) */ - p_sys->p_output = av_realloc( p_sys->p_output, i_output ); + *pp_block = p_block = block_Realloc( p_block, 0, p_block->i_buffer + FF_INPUT_BUFFER_PADDING_SIZE ); + if( !p_block ) + return NULL; + p_block->i_buffer -= FF_INPUT_BUFFER_PADDING_SIZE; + memset( &p_block->p_buffer[p_block->i_buffer], 0, FF_INPUT_BUFFER_PADDING_SIZE ); + + p_block->i_flags |= BLOCK_FLAG_PRIVATE_REALLOCATED; } - *pp_block = p_block = block_Realloc( p_block, 0, p_block->i_buffer + FF_INPUT_BUFFER_PADDING_SIZE ); - if( !p_block ) - return NULL; - p_block->i_buffer -= FF_INPUT_BUFFER_PADDING_SIZE; - memset( &p_block->p_buffer[p_block->i_buffer], 0, FF_INPUT_BUFFER_PADDING_SIZE ); + do + { + i_output = __MAX( p_block->i_buffer, p_sys->i_output_max ); + if( i_output > p_sys->i_output_max ) + { + /* Grow output buffer if necessary (eg. for PCM data) */ + p_sys->p_output = av_realloc( p_sys->p_output, i_output ); + } -#if LIBAVCODEC_VERSION_INT >= ((52<<16)+(0<<8)+0) - i_used = avcodec_decode_audio2( p_sys->p_context, - (int16_t*)p_sys->p_output, &i_output, - p_block->p_buffer, p_block->i_buffer ); -#else - i_used = avcodec_decode_audio( p_sys->p_context, - (int16_t*)p_sys->p_output, &i_output, - p_block->p_buffer, p_block->i_buffer ); -#endif + i_used = avcodec_decode_audio2( p_sys->p_context, + (int16_t*)p_sys->p_output, &i_output, + p_block->p_buffer, p_block->i_buffer ); - if( i_used < 0 || i_output < 0 ) - { - if( i_used < 0 ) - msg_Warn( p_dec, "cannot decode one frame (%zu bytes)", - p_block->i_buffer ); + if( i_used < 0 || i_output < 0 ) + { + if( i_used < 0 ) + msg_Warn( p_dec, "cannot decode one frame (%zu bytes)", + p_block->i_buffer ); - block_Release( p_block ); - return NULL; - } - else if( (size_t)i_used > p_block->i_buffer ) - { - i_used = p_block->i_buffer; - } + block_Release( p_block ); + return NULL; + } + else if( (size_t)i_used > p_block->i_buffer ) + { + i_used = p_block->i_buffer; + } + + p_block->i_buffer -= i_used; + p_block->p_buffer += i_used; - p_block->i_buffer -= i_used; - p_block->p_buffer += i_used; + } while( p_block->i_buffer > 0 && i_output <= 0 ); if( p_sys->p_context->channels <= 0 || p_sys->p_context->channels > 8 || p_sys->p_context->sample_rate <= 0 ) @@ -394,13 +395,36 @@ void EndAudioDec( decoder_t *p_dec ) /***************************************************************************** * *****************************************************************************/ -#if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT( 52, 2, 0 ) -# define LIBAVCODEC_AUDIO_LAYOUT -#else -# warning "Audio channel layout is unsupported by your avcodec version." -#endif -#if defined(LIBAVCODEC_AUDIO_LAYOUT) +void GetVlcAudioFormat( vlc_fourcc_t *pi_codec, unsigned *pi_bits, int i_sample_fmt ) +{ + switch( i_sample_fmt ) + { + case SAMPLE_FMT_U8: + *pi_codec = VLC_CODEC_U8; + *pi_bits = 8; + break; + case SAMPLE_FMT_S32: + *pi_codec = VLC_CODEC_S32N; + *pi_bits = 32; + break; + case SAMPLE_FMT_FLT: + *pi_codec = VLC_CODEC_FL32; + *pi_bits = 32; + break; + case SAMPLE_FMT_DBL: + *pi_codec = VLC_CODEC_FL64; + *pi_bits = 64; + break; + + case SAMPLE_FMT_S16: + default: + *pi_codec = VLC_CODEC_S16N; + *pi_bits = 16; + break; + } +} + static const uint64_t pi_channels_map[][2] = { { CH_FRONT_LEFT, AOUT_CHAN_LEFT }, @@ -424,97 +448,33 @@ static const uint64_t pi_channels_map[][2] = { CH_STEREO_LEFT, 0 }, { CH_STEREO_RIGHT, 0 }, }; -#else -static const uint64_t pi_channels_map[][2] = -{ - { 0, AOUT_CHAN_LEFT }, - { 0, AOUT_CHAN_RIGHT }, - { 0, AOUT_CHAN_CENTER }, - { 0, AOUT_CHAN_LFE }, - { 0, AOUT_CHAN_REARLEFT }, - { 0, AOUT_CHAN_REARRIGHT }, - { 0, 0 }, - { 0, 0 }, - { 0, AOUT_CHAN_REARCENTER }, - { 0, AOUT_CHAN_MIDDLELEFT }, - { 0, AOUT_CHAN_MIDDLERIGHT }, - { 0, 0 }, - { 0, 0 }, - { 0, 0 }, - { 0, 0 }, - { 0, 0 }, - { 0, 0 }, - { 0, 0 }, - { 0, 0 }, - { 0, 0 }, -}; -#endif static void SetupOutputFormat( decoder_t *p_dec, bool b_trust ) { decoder_sys_t *p_sys = p_dec->p_sys; -#if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT( 51, 65, 0 ) - switch( p_sys->p_context->sample_fmt ) - { - case SAMPLE_FMT_U8: - p_dec->fmt_out.i_codec = VLC_CODEC_U8; - p_dec->fmt_out.audio.i_bitspersample = 8; - break; - case SAMPLE_FMT_S32: - p_dec->fmt_out.i_codec = VLC_CODEC_S32N; - p_dec->fmt_out.audio.i_bitspersample = 32; - break; - case SAMPLE_FMT_FLT: - p_dec->fmt_out.i_codec = VLC_CODEC_FL32; - p_dec->fmt_out.audio.i_bitspersample = 32; - break; - case SAMPLE_FMT_DBL: - p_dec->fmt_out.i_codec = VLC_CODEC_FL64; - p_dec->fmt_out.audio.i_bitspersample = 64; - break; - - case SAMPLE_FMT_S16: - default: - p_dec->fmt_out.i_codec = VLC_CODEC_S16N; - p_dec->fmt_out.audio.i_bitspersample = 16; - break; - } -#else - p_dec->fmt_out.i_codec = VLC_CODEC_S16N; - p_dec->fmt_out.audio.i_bitspersample = 16; -#endif - p_dec->fmt_out.audio.i_rate = p_sys->p_context->sample_rate; + GetVlcAudioFormat( &p_dec->fmt_out.i_codec, + &p_dec->fmt_out.audio.i_bitspersample, + p_sys->p_context->sample_fmt ); + p_dec->fmt_out.audio.i_rate = p_sys->p_context->sample_rate; /* */ -#if defined(LIBAVCODEC_AUDIO_LAYOUT) if( p_sys->i_previous_channels == p_sys->p_context->channels && p_sys->i_previous_layout == p_sys->p_context->channel_layout ) return; -#else - if( p_sys->i_previous_channels == p_sys->p_context->channels ) - return; -#endif if( b_trust ) { p_sys->i_previous_channels = p_sys->p_context->channels; -#if defined(LIBAVCODEC_AUDIO_LAYOUT) p_sys->i_previous_layout = p_sys->p_context->channel_layout; -#endif } /* Specified order * FIXME should we use fmt_in.audio.i_physical_channels or not ? */ -#if defined(LIBAVCODEC_AUDIO_LAYOUT) const unsigned i_order_max = 8 * sizeof(p_sys->p_context->channel_layout); -#else - const unsigned i_order_max = 64; -#endif uint32_t pi_order_src[i_order_max]; int i_channels_src = 0; -#if defined(LIBAVCODEC_AUDIO_LAYOUT) if( p_sys->p_context->channel_layout ) { for( unsigned i = 0; i < sizeof(pi_channels_map)/sizeof(*pi_channels_map); i++ ) @@ -524,7 +484,6 @@ static void SetupOutputFormat( decoder_t *p_dec, bool b_trust ) } } else -#endif { /* Create default order */ if( b_trust )