diff options
author | Michael Niedermayer <michaelni@gmx.at> | 2012-03-21 23:47:44 +0100 |
---|---|---|
committer | Michael Niedermayer <michaelni@gmx.at> | 2012-03-22 00:40:11 +0100 |
commit | 967facb6950549d0cc4e0ba79a056ebc6f93a049 (patch) | |
tree | 872266e5d486be0ab8cf9e378bf567c191fba71a /libavcodec/flacenc.c | |
parent | f1fdd208cc0a1fce7aaaf6b0fe72b013525f49e0 (diff) | |
parent | 6aba117f1273c7704312c6d892c9f552fa0661bb (diff) | |
download | ffmpeg-967facb6950549d0cc4e0ba79a056ebc6f93a049.tar.gz |
Merge remote-tracking branch 'qatar/master'
* qatar/master: (26 commits)
adxenc: use AVCodec.encode2()
adxenc: Use the AVFrame in ADXContext for coded_frame
indeo4: fix out-of-bounds function call.
configure: Restructure help output.
configure: Internal-only components should not be command-line selectable.
vorbisenc: use AVCodec.encode2()
libvorbis: use AVCodec.encode2()
libopencore-amrnbenc: use AVCodec.encode2()
ra144enc: use AVCodec.encode2()
nellymoserenc: use AVCodec.encode2()
roqaudioenc: use AVCodec.encode2()
libspeex: use AVCodec.encode2()
libvo_amrwbenc: use AVCodec.encode2()
libvo_aacenc: use AVCodec.encode2()
wmaenc: use AVCodec.encode2()
mpegaudioenc: use AVCodec.encode2()
libmp3lame: use AVCodec.encode2()
libgsmenc: use AVCodec.encode2()
libfaac: use AVCodec.encode2()
g726enc: use AVCodec.encode2()
...
Conflicts:
configure
libavcodec/Makefile
libavcodec/ac3enc.c
libavcodec/adxenc.c
libavcodec/libgsm.c
libavcodec/libvorbis.c
libavcodec/vorbisenc.c
libavcodec/wmaenc.c
tests/ref/acodec/g722
tests/ref/lavf/asf
tests/ref/lavf/ffm
tests/ref/lavf/mkv
tests/ref/lavf/mpg
tests/ref/lavf/rm
tests/ref/lavf/ts
tests/ref/seek/lavf_asf
tests/ref/seek/lavf_ffm
tests/ref/seek/lavf_mkv
tests/ref/seek/lavf_mpg
tests/ref/seek/lavf_rm
tests/ref/seek/lavf_ts
Merged-by: Michael Niedermayer <michaelni@gmx.at>
Diffstat (limited to 'libavcodec/flacenc.c')
-rw-r--r-- | libavcodec/flacenc.c | 52 |
1 files changed, 31 insertions, 21 deletions
diff --git a/libavcodec/flacenc.c b/libavcodec/flacenc.c index ecf883e652..e8d6e8cb21 100644 --- a/libavcodec/flacenc.c +++ b/libavcodec/flacenc.c @@ -25,6 +25,7 @@ #include "avcodec.h" #include "get_bits.h" #include "golomb.h" +#include "internal.h" #include "lpc.h" #include "flac.h" #include "flacdata.h" @@ -367,9 +368,11 @@ static av_cold int flac_encode_init(AVCodecContext *avctx) s->frame_count = 0; s->min_framesize = s->max_framesize; +#if FF_API_OLD_ENCODE_AUDIO avctx->coded_frame = avcodec_alloc_frame(); if (!avctx->coded_frame) return AVERROR(ENOMEM); +#endif if (channels == 3 && avctx->channel_layout != (AV_CH_LAYOUT_STEREO|AV_CH_FRONT_CENTER) || @@ -402,7 +405,7 @@ static av_cold int flac_encode_init(AVCodecContext *avctx) } -static void init_frame(FlacEncodeContext *s) +static void init_frame(FlacEncodeContext *s, int nb_samples) { int i, ch; FlacFrame *frame; @@ -410,7 +413,7 @@ static void init_frame(FlacEncodeContext *s) frame = &s->frame; for (i = 0; i < 16; i++) { - if (s->avctx->frame_size == ff_flac_blocksize_table[i]) { + if (nb_samples == ff_flac_blocksize_table[i]) { frame->blocksize = ff_flac_blocksize_table[i]; frame->bs_code[0] = i; frame->bs_code[1] = 0; @@ -418,7 +421,7 @@ static void init_frame(FlacEncodeContext *s) } } if (i == 16) { - frame->blocksize = s->avctx->frame_size; + frame->blocksize = nb_samples; if (frame->blocksize <= 256) { frame->bs_code[0] = 6; frame->bs_code[1] = frame->blocksize-1; @@ -1188,9 +1191,9 @@ static void write_frame_footer(FlacEncodeContext *s) } -static int write_frame(FlacEncodeContext *s, uint8_t *frame, int buf_size) +static int write_frame(FlacEncodeContext *s, AVPacket *avpkt) { - init_put_bits(&s->pb, frame, buf_size); + init_put_bits(&s->pb, avpkt->data, avpkt->size); write_frame_header(s); write_subframes(s); write_frame_footer(s); @@ -1212,30 +1215,31 @@ static void update_md5_sum(FlacEncodeContext *s, const int16_t *samples) } -static int flac_encode_frame(AVCodecContext *avctx, uint8_t *frame, - int buf_size, void *data) +static int flac_encode_frame(AVCodecContext *avctx, AVPacket *avpkt, + const AVFrame *frame, int *got_packet_ptr) { FlacEncodeContext *s; - const int16_t *samples = data; - int frame_bytes, out_bytes; + const int16_t *samples; + int frame_bytes, out_bytes, ret; s = avctx->priv_data; /* when the last block is reached, update the header in extradata */ - if (!data) { + if (!frame) { s->max_framesize = s->max_encoded_framesize; av_md5_final(s->md5ctx, s->md5sum); write_streaminfo(s, avctx->extradata); return 0; } + samples = (const int16_t *)frame->data[0]; /* change max_framesize for small final frame */ - if (avctx->frame_size < s->frame.blocksize) { - s->max_framesize = ff_flac_get_max_frame_size(avctx->frame_size, + if (frame->nb_samples < s->frame.blocksize) { + s->max_framesize = ff_flac_get_max_frame_size(frame->nb_samples, s->channels, 16); } - init_frame(s); + init_frame(s, frame->nb_samples); copy_samples(s, samples); @@ -1250,22 +1254,26 @@ static int flac_encode_frame(AVCodecContext *avctx, uint8_t *frame, frame_bytes = encode_frame(s); } - if (buf_size < frame_bytes) { - av_log(avctx, AV_LOG_ERROR, "output buffer too small\n"); - return 0; + if ((ret = ff_alloc_packet(avpkt, frame_bytes))) { + av_log(avctx, AV_LOG_ERROR, "Error getting output packet\n"); + return ret; } - out_bytes = write_frame(s, frame, buf_size); + + out_bytes = write_frame(s, avpkt); s->frame_count++; - avctx->coded_frame->pts = s->sample_count; - s->sample_count += avctx->frame_size; + s->sample_count += frame->nb_samples; update_md5_sum(s, samples); if (out_bytes > s->max_encoded_framesize) s->max_encoded_framesize = out_bytes; if (out_bytes < s->min_framesize) s->min_framesize = out_bytes; - return out_bytes; + avpkt->pts = frame->pts; + avpkt->duration = ff_samples_to_time_base(avctx, frame->nb_samples); + avpkt->size = out_bytes; + *got_packet_ptr = 1; + return 0; } @@ -1278,7 +1286,9 @@ static av_cold int flac_encode_close(AVCodecContext *avctx) } av_freep(&avctx->extradata); avctx->extradata_size = 0; +#if FF_API_OLD_ENCODE_AUDIO av_freep(&avctx->coded_frame); +#endif return 0; } @@ -1316,7 +1326,7 @@ AVCodec ff_flac_encoder = { .id = CODEC_ID_FLAC, .priv_data_size = sizeof(FlacEncodeContext), .init = flac_encode_init, - .encode = flac_encode_frame, + .encode2 = flac_encode_frame, .close = flac_encode_close, .capabilities = CODEC_CAP_SMALL_LAST_FRAME | CODEC_CAP_DELAY | CODEC_CAP_LOSSLESS, .sample_fmts = (const enum AVSampleFormat[]){AV_SAMPLE_FMT_S16,AV_SAMPLE_FMT_NONE}, |