3 * Copyright (c) 2001 Fabrice Bellard
4 * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
6 * This file is part of FFmpeg.
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
29 #include "libavutil/atomic.h"
30 #include "libavutil/attributes.h"
31 #include "libavutil/avassert.h"
32 #include "libavutil/avstring.h"
33 #include "libavutil/bprint.h"
34 #include "libavutil/channel_layout.h"
35 #include "libavutil/crc.h"
36 #include "libavutil/frame.h"
37 #include "libavutil/hwcontext.h"
38 #include "libavutil/internal.h"
39 #include "libavutil/mathematics.h"
40 #include "libavutil/mem_internal.h"
41 #include "libavutil/pixdesc.h"
42 #include "libavutil/imgutils.h"
43 #include "libavutil/samplefmt.h"
44 #include "libavutil/dict.h"
45 #include "libavutil/thread.h"
47 #include "libavutil/opt.h"
49 #include "mpegvideo.h"
51 #include "frame_thread_encoder.h"
54 #include "bytestream.h"
64 #include "libavutil/ffversion.h"
65 const char av_codec_ffversion[] = "FFmpeg version " FFMPEG_VERSION;
67 #if HAVE_PTHREADS || HAVE_W32THREADS || HAVE_OS2THREADS
68 static int default_lockmgr_cb(void **arg, enum AVLockOp op)
70 void * volatile * mutex = arg;
78 pthread_mutex_t *tmp = av_malloc(sizeof(pthread_mutex_t));
80 return AVERROR(ENOMEM);
81 if ((err = pthread_mutex_init(tmp, NULL))) {
85 if (avpriv_atomic_ptr_cas(mutex, NULL, tmp)) {
86 pthread_mutex_destroy(tmp);
91 if ((err = pthread_mutex_lock(*mutex)))
96 if ((err = pthread_mutex_unlock(*mutex)))
100 case AV_LOCK_DESTROY:
102 pthread_mutex_destroy(*mutex);
104 avpriv_atomic_ptr_cas(mutex, *mutex, NULL);
109 static int (*lockmgr_cb)(void **mutex, enum AVLockOp op) = default_lockmgr_cb;
111 static int (*lockmgr_cb)(void **mutex, enum AVLockOp op) = NULL;
115 volatile int ff_avcodec_locked;
116 static int volatile entangled_thread_counter = 0;
117 static void *codec_mutex;
118 static void *avformat_mutex;
120 void av_fast_padded_malloc(void *ptr, unsigned int *size, size_t min_size)
123 if (min_size > SIZE_MAX - AV_INPUT_BUFFER_PADDING_SIZE) {
128 if (!ff_fast_malloc(p, size, min_size + AV_INPUT_BUFFER_PADDING_SIZE, 1))
129 memset(*p + min_size, 0, AV_INPUT_BUFFER_PADDING_SIZE);
132 void av_fast_padded_mallocz(void *ptr, unsigned int *size, size_t min_size)
135 if (min_size > SIZE_MAX - AV_INPUT_BUFFER_PADDING_SIZE) {
140 if (!ff_fast_malloc(p, size, min_size + AV_INPUT_BUFFER_PADDING_SIZE, 1))
141 memset(*p, 0, min_size + AV_INPUT_BUFFER_PADDING_SIZE);
144 /* encoder management */
145 static AVCodec *first_avcodec = NULL;
146 static AVCodec **last_avcodec = &first_avcodec;
148 AVCodec *av_codec_next(const AVCodec *c)
153 return first_avcodec;
156 static av_cold void avcodec_init(void)
158 static int initialized = 0;
160 if (initialized != 0)
165 ff_me_cmp_init_static();
168 int av_codec_is_encoder(const AVCodec *codec)
170 return codec && (codec->encode_sub || codec->encode2 ||codec->send_frame);
173 int av_codec_is_decoder(const AVCodec *codec)
175 return codec && (codec->decode || codec->send_packet);
178 av_cold void avcodec_register(AVCodec *codec)
185 while(*p || avpriv_atomic_ptr_cas((void * volatile *)p, NULL, codec))
187 last_avcodec = &codec->next;
189 if (codec->init_static_data)
190 codec->init_static_data(codec);
194 unsigned avcodec_get_edge_width(void)
200 #if FF_API_SET_DIMENSIONS
201 void avcodec_set_dimensions(AVCodecContext *s, int width, int height)
203 int ret = ff_set_dimensions(s, width, height);
205 av_log(s, AV_LOG_WARNING, "Failed to set dimensions %d %d\n", width, height);
210 int ff_set_dimensions(AVCodecContext *s, int width, int height)
212 int ret = av_image_check_size(width, height, 0, s);
217 s->coded_width = width;
218 s->coded_height = height;
219 s->width = AV_CEIL_RSHIFT(width, s->lowres);
220 s->height = AV_CEIL_RSHIFT(height, s->lowres);
225 int ff_set_sar(AVCodecContext *avctx, AVRational sar)
227 int ret = av_image_check_sar(avctx->width, avctx->height, sar);
230 av_log(avctx, AV_LOG_WARNING, "ignoring invalid SAR: %d/%d\n",
232 avctx->sample_aspect_ratio = (AVRational){ 0, 1 };
235 avctx->sample_aspect_ratio = sar;
240 int ff_side_data_update_matrix_encoding(AVFrame *frame,
241 enum AVMatrixEncoding matrix_encoding)
243 AVFrameSideData *side_data;
244 enum AVMatrixEncoding *data;
246 side_data = av_frame_get_side_data(frame, AV_FRAME_DATA_MATRIXENCODING);
248 side_data = av_frame_new_side_data(frame, AV_FRAME_DATA_MATRIXENCODING,
249 sizeof(enum AVMatrixEncoding));
252 return AVERROR(ENOMEM);
254 data = (enum AVMatrixEncoding*)side_data->data;
255 *data = matrix_encoding;
260 void avcodec_align_dimensions2(AVCodecContext *s, int *width, int *height,
261 int linesize_align[AV_NUM_DATA_POINTERS])
266 AVPixFmtDescriptor const *desc = av_pix_fmt_desc_get(s->pix_fmt);
269 w_align = 1 << desc->log2_chroma_w;
270 h_align = 1 << desc->log2_chroma_h;
273 switch (s->pix_fmt) {
274 case AV_PIX_FMT_YUV420P:
275 case AV_PIX_FMT_YUYV422:
276 case AV_PIX_FMT_YVYU422:
277 case AV_PIX_FMT_UYVY422:
278 case AV_PIX_FMT_YUV422P:
279 case AV_PIX_FMT_YUV440P:
280 case AV_PIX_FMT_YUV444P:
281 case AV_PIX_FMT_GBRP:
282 case AV_PIX_FMT_GBRAP:
283 case AV_PIX_FMT_GRAY8:
284 case AV_PIX_FMT_GRAY16BE:
285 case AV_PIX_FMT_GRAY16LE:
286 case AV_PIX_FMT_YUVJ420P:
287 case AV_PIX_FMT_YUVJ422P:
288 case AV_PIX_FMT_YUVJ440P:
289 case AV_PIX_FMT_YUVJ444P:
290 case AV_PIX_FMT_YUVA420P:
291 case AV_PIX_FMT_YUVA422P:
292 case AV_PIX_FMT_YUVA444P:
293 case AV_PIX_FMT_YUV420P9LE:
294 case AV_PIX_FMT_YUV420P9BE:
295 case AV_PIX_FMT_YUV420P10LE:
296 case AV_PIX_FMT_YUV420P10BE:
297 case AV_PIX_FMT_YUV420P12LE:
298 case AV_PIX_FMT_YUV420P12BE:
299 case AV_PIX_FMT_YUV420P14LE:
300 case AV_PIX_FMT_YUV420P14BE:
301 case AV_PIX_FMT_YUV420P16LE:
302 case AV_PIX_FMT_YUV420P16BE:
303 case AV_PIX_FMT_YUVA420P9LE:
304 case AV_PIX_FMT_YUVA420P9BE:
305 case AV_PIX_FMT_YUVA420P10LE:
306 case AV_PIX_FMT_YUVA420P10BE:
307 case AV_PIX_FMT_YUVA420P16LE:
308 case AV_PIX_FMT_YUVA420P16BE:
309 case AV_PIX_FMT_YUV422P9LE:
310 case AV_PIX_FMT_YUV422P9BE:
311 case AV_PIX_FMT_YUV422P10LE:
312 case AV_PIX_FMT_YUV422P10BE:
313 case AV_PIX_FMT_YUV422P12LE:
314 case AV_PIX_FMT_YUV422P12BE:
315 case AV_PIX_FMT_YUV422P14LE:
316 case AV_PIX_FMT_YUV422P14BE:
317 case AV_PIX_FMT_YUV422P16LE:
318 case AV_PIX_FMT_YUV422P16BE:
319 case AV_PIX_FMT_YUVA422P9LE:
320 case AV_PIX_FMT_YUVA422P9BE:
321 case AV_PIX_FMT_YUVA422P10LE:
322 case AV_PIX_FMT_YUVA422P10BE:
323 case AV_PIX_FMT_YUVA422P16LE:
324 case AV_PIX_FMT_YUVA422P16BE:
325 case AV_PIX_FMT_YUV440P10LE:
326 case AV_PIX_FMT_YUV440P10BE:
327 case AV_PIX_FMT_YUV440P12LE:
328 case AV_PIX_FMT_YUV440P12BE:
329 case AV_PIX_FMT_YUV444P9LE:
330 case AV_PIX_FMT_YUV444P9BE:
331 case AV_PIX_FMT_YUV444P10LE:
332 case AV_PIX_FMT_YUV444P10BE:
333 case AV_PIX_FMT_YUV444P12LE:
334 case AV_PIX_FMT_YUV444P12BE:
335 case AV_PIX_FMT_YUV444P14LE:
336 case AV_PIX_FMT_YUV444P14BE:
337 case AV_PIX_FMT_YUV444P16LE:
338 case AV_PIX_FMT_YUV444P16BE:
339 case AV_PIX_FMT_YUVA444P9LE:
340 case AV_PIX_FMT_YUVA444P9BE:
341 case AV_PIX_FMT_YUVA444P10LE:
342 case AV_PIX_FMT_YUVA444P10BE:
343 case AV_PIX_FMT_YUVA444P16LE:
344 case AV_PIX_FMT_YUVA444P16BE:
345 case AV_PIX_FMT_GBRP9LE:
346 case AV_PIX_FMT_GBRP9BE:
347 case AV_PIX_FMT_GBRP10LE:
348 case AV_PIX_FMT_GBRP10BE:
349 case AV_PIX_FMT_GBRP12LE:
350 case AV_PIX_FMT_GBRP12BE:
351 case AV_PIX_FMT_GBRP14LE:
352 case AV_PIX_FMT_GBRP14BE:
353 case AV_PIX_FMT_GBRP16LE:
354 case AV_PIX_FMT_GBRP16BE:
355 case AV_PIX_FMT_GBRAP12LE:
356 case AV_PIX_FMT_GBRAP12BE:
357 case AV_PIX_FMT_GBRAP16LE:
358 case AV_PIX_FMT_GBRAP16BE:
359 w_align = 16; //FIXME assume 16 pixel per macroblock
360 h_align = 16 * 2; // interlaced needs 2 macroblocks height
362 case AV_PIX_FMT_YUV411P:
363 case AV_PIX_FMT_YUVJ411P:
364 case AV_PIX_FMT_UYYVYY411:
368 case AV_PIX_FMT_YUV410P:
369 if (s->codec_id == AV_CODEC_ID_SVQ1) {
374 case AV_PIX_FMT_RGB555:
375 if (s->codec_id == AV_CODEC_ID_RPZA) {
380 case AV_PIX_FMT_PAL8:
381 case AV_PIX_FMT_BGR8:
382 case AV_PIX_FMT_RGB8:
383 if (s->codec_id == AV_CODEC_ID_SMC ||
384 s->codec_id == AV_CODEC_ID_CINEPAK) {
388 if (s->codec_id == AV_CODEC_ID_JV) {
393 case AV_PIX_FMT_BGR24:
394 if ((s->codec_id == AV_CODEC_ID_MSZH) ||
395 (s->codec_id == AV_CODEC_ID_ZLIB)) {
400 case AV_PIX_FMT_RGB24:
401 if (s->codec_id == AV_CODEC_ID_CINEPAK) {
410 if (s->codec_id == AV_CODEC_ID_IFF_ILBM) {
411 w_align = FFMAX(w_align, 8);
414 *width = FFALIGN(*width, w_align);
415 *height = FFALIGN(*height, h_align);
416 if (s->codec_id == AV_CODEC_ID_H264 || s->lowres) {
417 // some of the optimized chroma MC reads one line too much
418 // which is also done in mpeg decoders with lowres > 0
421 // H.264 uses edge emulation for out of frame motion vectors, for this
422 // it requires a temporary area large enough to hold a 21x21 block,
423 // increasing witdth ensure that the temporary area is large enough,
424 // the next rounded up width is 32
425 *width = FFMAX(*width, 32);
428 for (i = 0; i < 4; i++)
429 linesize_align[i] = STRIDE_ALIGN;
432 void avcodec_align_dimensions(AVCodecContext *s, int *width, int *height)
434 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(s->pix_fmt);
435 int chroma_shift = desc->log2_chroma_w;
436 int linesize_align[AV_NUM_DATA_POINTERS];
439 avcodec_align_dimensions2(s, width, height, linesize_align);
440 align = FFMAX(linesize_align[0], linesize_align[3]);
441 linesize_align[1] <<= chroma_shift;
442 linesize_align[2] <<= chroma_shift;
443 align = FFMAX3(align, linesize_align[1], linesize_align[2]);
444 *width = FFALIGN(*width, align);
447 int avcodec_enum_to_chroma_pos(int *xpos, int *ypos, enum AVChromaLocation pos)
449 if (pos <= AVCHROMA_LOC_UNSPECIFIED || pos >= AVCHROMA_LOC_NB)
450 return AVERROR(EINVAL);
453 *xpos = (pos&1) * 128;
454 *ypos = ((pos>>1)^(pos<4)) * 128;
459 enum AVChromaLocation avcodec_chroma_pos_to_enum(int xpos, int ypos)
463 for (pos = AVCHROMA_LOC_UNSPECIFIED + 1; pos < AVCHROMA_LOC_NB; pos++) {
464 if (avcodec_enum_to_chroma_pos(&xout, &yout, pos) == 0 && xout == xpos && yout == ypos)
467 return AVCHROMA_LOC_UNSPECIFIED;
470 int avcodec_fill_audio_frame(AVFrame *frame, int nb_channels,
471 enum AVSampleFormat sample_fmt, const uint8_t *buf,
472 int buf_size, int align)
474 int ch, planar, needed_size, ret = 0;
476 needed_size = av_samples_get_buffer_size(NULL, nb_channels,
477 frame->nb_samples, sample_fmt,
479 if (buf_size < needed_size)
480 return AVERROR(EINVAL);
482 planar = av_sample_fmt_is_planar(sample_fmt);
483 if (planar && nb_channels > AV_NUM_DATA_POINTERS) {
484 if (!(frame->extended_data = av_mallocz_array(nb_channels,
485 sizeof(*frame->extended_data))))
486 return AVERROR(ENOMEM);
488 frame->extended_data = frame->data;
491 if ((ret = av_samples_fill_arrays(frame->extended_data, &frame->linesize[0],
492 (uint8_t *)(intptr_t)buf, nb_channels, frame->nb_samples,
493 sample_fmt, align)) < 0) {
494 if (frame->extended_data != frame->data)
495 av_freep(&frame->extended_data);
498 if (frame->extended_data != frame->data) {
499 for (ch = 0; ch < AV_NUM_DATA_POINTERS; ch++)
500 frame->data[ch] = frame->extended_data[ch];
506 static int update_frame_pool(AVCodecContext *avctx, AVFrame *frame)
508 FramePool *pool = avctx->internal->pool;
511 switch (avctx->codec_type) {
512 case AVMEDIA_TYPE_VIDEO: {
516 int w = frame->width;
517 int h = frame->height;
518 int tmpsize, unaligned;
520 if (pool->format == frame->format &&
521 pool->width == frame->width && pool->height == frame->height)
524 avcodec_align_dimensions2(avctx, &w, &h, pool->stride_align);
527 // NOTE: do not align linesizes individually, this breaks e.g. assumptions
528 // that linesize[0] == 2*linesize[1] in the MPEG-encoder for 4:2:2
529 ret = av_image_fill_linesizes(linesize, avctx->pix_fmt, w);
532 // increase alignment of w for next try (rhs gives the lowest bit set in w)
536 for (i = 0; i < 4; i++)
537 unaligned |= linesize[i] % pool->stride_align[i];
540 tmpsize = av_image_fill_pointers(data, avctx->pix_fmt, h,
545 for (i = 0; i < 3 && data[i + 1]; i++)
546 size[i] = data[i + 1] - data[i];
547 size[i] = tmpsize - (data[i] - data[0]);
549 for (i = 0; i < 4; i++) {
550 av_buffer_pool_uninit(&pool->pools[i]);
551 pool->linesize[i] = linesize[i];
553 pool->pools[i] = av_buffer_pool_init(size[i] + 16 + STRIDE_ALIGN - 1,
554 CONFIG_MEMORY_POISONING ?
557 if (!pool->pools[i]) {
558 ret = AVERROR(ENOMEM);
563 pool->format = frame->format;
564 pool->width = frame->width;
565 pool->height = frame->height;
569 case AVMEDIA_TYPE_AUDIO: {
570 int ch = av_frame_get_channels(frame); //av_get_channel_layout_nb_channels(frame->channel_layout);
571 int planar = av_sample_fmt_is_planar(frame->format);
572 int planes = planar ? ch : 1;
574 if (pool->format == frame->format && pool->planes == planes &&
575 pool->channels == ch && frame->nb_samples == pool->samples)
578 av_buffer_pool_uninit(&pool->pools[0]);
579 ret = av_samples_get_buffer_size(&pool->linesize[0], ch,
580 frame->nb_samples, frame->format, 0);
584 pool->pools[0] = av_buffer_pool_init(pool->linesize[0], NULL);
585 if (!pool->pools[0]) {
586 ret = AVERROR(ENOMEM);
590 pool->format = frame->format;
591 pool->planes = planes;
593 pool->samples = frame->nb_samples;
596 default: av_assert0(0);
600 for (i = 0; i < 4; i++)
601 av_buffer_pool_uninit(&pool->pools[i]);
603 pool->planes = pool->channels = pool->samples = 0;
604 pool->width = pool->height = 0;
608 static int audio_get_buffer(AVCodecContext *avctx, AVFrame *frame)
610 FramePool *pool = avctx->internal->pool;
611 int planes = pool->planes;
614 frame->linesize[0] = pool->linesize[0];
616 if (planes > AV_NUM_DATA_POINTERS) {
617 frame->extended_data = av_mallocz_array(planes, sizeof(*frame->extended_data));
618 frame->nb_extended_buf = planes - AV_NUM_DATA_POINTERS;
619 frame->extended_buf = av_mallocz_array(frame->nb_extended_buf,
620 sizeof(*frame->extended_buf));
621 if (!frame->extended_data || !frame->extended_buf) {
622 av_freep(&frame->extended_data);
623 av_freep(&frame->extended_buf);
624 return AVERROR(ENOMEM);
627 frame->extended_data = frame->data;
628 av_assert0(frame->nb_extended_buf == 0);
631 for (i = 0; i < FFMIN(planes, AV_NUM_DATA_POINTERS); i++) {
632 frame->buf[i] = av_buffer_pool_get(pool->pools[0]);
635 frame->extended_data[i] = frame->data[i] = frame->buf[i]->data;
637 for (i = 0; i < frame->nb_extended_buf; i++) {
638 frame->extended_buf[i] = av_buffer_pool_get(pool->pools[0]);
639 if (!frame->extended_buf[i])
641 frame->extended_data[i + AV_NUM_DATA_POINTERS] = frame->extended_buf[i]->data;
644 if (avctx->debug & FF_DEBUG_BUFFERS)
645 av_log(avctx, AV_LOG_DEBUG, "default_get_buffer called on frame %p", frame);
649 av_frame_unref(frame);
650 return AVERROR(ENOMEM);
653 static int video_get_buffer(AVCodecContext *s, AVFrame *pic)
655 FramePool *pool = s->internal->pool;
656 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pic->format);
659 if (pic->data[0] || pic->data[1] || pic->data[2] || pic->data[3]) {
660 av_log(s, AV_LOG_ERROR, "pic->data[*]!=NULL in avcodec_default_get_buffer\n");
665 av_log(s, AV_LOG_ERROR,
666 "Unable to get pixel format descriptor for format %s\n",
667 av_get_pix_fmt_name(pic->format));
668 return AVERROR(EINVAL);
671 memset(pic->data, 0, sizeof(pic->data));
672 pic->extended_data = pic->data;
674 for (i = 0; i < 4 && pool->pools[i]; i++) {
675 pic->linesize[i] = pool->linesize[i];
677 pic->buf[i] = av_buffer_pool_get(pool->pools[i]);
681 pic->data[i] = pic->buf[i]->data;
683 for (; i < AV_NUM_DATA_POINTERS; i++) {
685 pic->linesize[i] = 0;
687 if (desc->flags & AV_PIX_FMT_FLAG_PAL ||
688 desc->flags & AV_PIX_FMT_FLAG_PSEUDOPAL)
689 avpriv_set_systematic_pal2((uint32_t *)pic->data[1], pic->format);
691 if (s->debug & FF_DEBUG_BUFFERS)
692 av_log(s, AV_LOG_DEBUG, "default_get_buffer called on pic %p\n", pic);
697 return AVERROR(ENOMEM);
700 void ff_color_frame(AVFrame *frame, const int c[4])
702 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(frame->format);
705 av_assert0(desc->flags & AV_PIX_FMT_FLAG_PLANAR);
707 for (p = 0; p<desc->nb_components; p++) {
708 uint8_t *dst = frame->data[p];
709 int is_chroma = p == 1 || p == 2;
710 int bytes = is_chroma ? AV_CEIL_RSHIFT(frame->width, desc->log2_chroma_w) : frame->width;
711 int height = is_chroma ? AV_CEIL_RSHIFT(frame->height, desc->log2_chroma_h) : frame->height;
712 for (y = 0; y < height; y++) {
713 if (desc->comp[0].depth >= 9) {
714 for (x = 0; x<bytes; x++)
715 ((uint16_t*)dst)[x] = c[p];
717 memset(dst, c[p], bytes);
718 dst += frame->linesize[p];
723 int avcodec_default_get_buffer2(AVCodecContext *avctx, AVFrame *frame, int flags)
727 if ((ret = update_frame_pool(avctx, frame)) < 0)
730 switch (avctx->codec_type) {
731 case AVMEDIA_TYPE_VIDEO:
732 return video_get_buffer(avctx, frame);
733 case AVMEDIA_TYPE_AUDIO:
734 return audio_get_buffer(avctx, frame);
740 static int add_metadata_from_side_data(AVPacket *avpkt, AVFrame *frame)
743 const uint8_t *side_metadata;
745 AVDictionary **frame_md = avpriv_frame_get_metadatap(frame);
747 side_metadata = av_packet_get_side_data(avpkt,
748 AV_PKT_DATA_STRINGS_METADATA, &size);
749 return av_packet_unpack_dictionary(side_metadata, size, frame_md);
752 int ff_init_buffer_info(AVCodecContext *avctx, AVFrame *frame)
754 AVPacket *pkt = avctx->internal->pkt;
756 static const struct {
757 enum AVPacketSideDataType packet;
758 enum AVFrameSideDataType frame;
760 { AV_PKT_DATA_REPLAYGAIN , AV_FRAME_DATA_REPLAYGAIN },
761 { AV_PKT_DATA_DISPLAYMATRIX, AV_FRAME_DATA_DISPLAYMATRIX },
762 { AV_PKT_DATA_STEREO3D, AV_FRAME_DATA_STEREO3D },
763 { AV_PKT_DATA_AUDIO_SERVICE_TYPE, AV_FRAME_DATA_AUDIO_SERVICE_TYPE },
764 { AV_PKT_DATA_MASTERING_DISPLAY_METADATA, AV_FRAME_DATA_MASTERING_DISPLAY_METADATA },
768 frame->pkt_pts = pkt->pts;
769 av_frame_set_pkt_pos (frame, pkt->pos);
770 av_frame_set_pkt_duration(frame, pkt->duration);
771 av_frame_set_pkt_size (frame, pkt->size);
773 for (i = 0; i < FF_ARRAY_ELEMS(sd); i++) {
775 uint8_t *packet_sd = av_packet_get_side_data(pkt, sd[i].packet, &size);
777 AVFrameSideData *frame_sd = av_frame_new_side_data(frame,
781 return AVERROR(ENOMEM);
783 memcpy(frame_sd->data, packet_sd, size);
786 add_metadata_from_side_data(pkt, frame);
788 frame->pkt_pts = AV_NOPTS_VALUE;
789 av_frame_set_pkt_pos (frame, -1);
790 av_frame_set_pkt_duration(frame, 0);
791 av_frame_set_pkt_size (frame, -1);
793 frame->reordered_opaque = avctx->reordered_opaque;
795 if (frame->color_primaries == AVCOL_PRI_UNSPECIFIED)
796 frame->color_primaries = avctx->color_primaries;
797 if (frame->color_trc == AVCOL_TRC_UNSPECIFIED)
798 frame->color_trc = avctx->color_trc;
799 if (av_frame_get_colorspace(frame) == AVCOL_SPC_UNSPECIFIED)
800 av_frame_set_colorspace(frame, avctx->colorspace);
801 if (av_frame_get_color_range(frame) == AVCOL_RANGE_UNSPECIFIED)
802 av_frame_set_color_range(frame, avctx->color_range);
803 if (frame->chroma_location == AVCHROMA_LOC_UNSPECIFIED)
804 frame->chroma_location = avctx->chroma_sample_location;
806 switch (avctx->codec->type) {
807 case AVMEDIA_TYPE_VIDEO:
808 frame->format = avctx->pix_fmt;
809 if (!frame->sample_aspect_ratio.num)
810 frame->sample_aspect_ratio = avctx->sample_aspect_ratio;
812 if (frame->width && frame->height &&
813 av_image_check_sar(frame->width, frame->height,
814 frame->sample_aspect_ratio) < 0) {
815 av_log(avctx, AV_LOG_WARNING, "ignoring invalid SAR: %u/%u\n",
816 frame->sample_aspect_ratio.num,
817 frame->sample_aspect_ratio.den);
818 frame->sample_aspect_ratio = (AVRational){ 0, 1 };
822 case AVMEDIA_TYPE_AUDIO:
823 if (!frame->sample_rate)
824 frame->sample_rate = avctx->sample_rate;
825 if (frame->format < 0)
826 frame->format = avctx->sample_fmt;
827 if (!frame->channel_layout) {
828 if (avctx->channel_layout) {
829 if (av_get_channel_layout_nb_channels(avctx->channel_layout) !=
831 av_log(avctx, AV_LOG_ERROR, "Inconsistent channel "
833 return AVERROR(EINVAL);
836 frame->channel_layout = avctx->channel_layout;
838 if (avctx->channels > FF_SANE_NB_CHANNELS) {
839 av_log(avctx, AV_LOG_ERROR, "Too many channels: %d.\n",
841 return AVERROR(ENOSYS);
845 av_frame_set_channels(frame, avctx->channels);
851 int ff_decode_frame_props(AVCodecContext *avctx, AVFrame *frame)
853 return ff_init_buffer_info(avctx, frame);
856 static void validate_avframe_allocation(AVCodecContext *avctx, AVFrame *frame)
858 if (avctx->codec_type == AVMEDIA_TYPE_VIDEO) {
860 int num_planes = av_pix_fmt_count_planes(frame->format);
861 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(frame->format);
862 int flags = desc ? desc->flags : 0;
863 if (num_planes == 1 && (flags & AV_PIX_FMT_FLAG_PAL))
865 for (i = 0; i < num_planes; i++) {
866 av_assert0(frame->data[i]);
868 // For now do not enforce anything for palette of pseudopal formats
869 if (num_planes == 1 && (flags & AV_PIX_FMT_FLAG_PSEUDOPAL))
871 // For formats without data like hwaccel allow unused pointers to be non-NULL.
872 for (i = num_planes; num_planes > 0 && i < FF_ARRAY_ELEMS(frame->data); i++) {
874 av_log(avctx, AV_LOG_ERROR, "Buffer returned by get_buffer2() did not zero unused plane pointers\n");
875 frame->data[i] = NULL;
880 static int get_buffer_internal(AVCodecContext *avctx, AVFrame *frame, int flags)
882 const AVHWAccel *hwaccel = avctx->hwaccel;
883 int override_dimensions = 1;
886 if (avctx->codec_type == AVMEDIA_TYPE_VIDEO) {
887 if ((ret = av_image_check_size(avctx->width, avctx->height, 0, avctx)) < 0 || avctx->pix_fmt<0) {
888 av_log(avctx, AV_LOG_ERROR, "video_get_buffer: image parameters invalid\n");
889 return AVERROR(EINVAL);
892 if (frame->width <= 0 || frame->height <= 0) {
893 frame->width = FFMAX(avctx->width, AV_CEIL_RSHIFT(avctx->coded_width, avctx->lowres));
894 frame->height = FFMAX(avctx->height, AV_CEIL_RSHIFT(avctx->coded_height, avctx->lowres));
895 override_dimensions = 0;
898 if (frame->data[0] || frame->data[1] || frame->data[2] || frame->data[3]) {
899 av_log(avctx, AV_LOG_ERROR, "pic->data[*]!=NULL in get_buffer_internal\n");
900 return AVERROR(EINVAL);
903 ret = ff_decode_frame_props(avctx, frame);
908 if (hwaccel->alloc_frame) {
909 ret = hwaccel->alloc_frame(avctx, frame);
913 avctx->sw_pix_fmt = avctx->pix_fmt;
915 ret = avctx->get_buffer2(avctx, frame, flags);
917 validate_avframe_allocation(avctx, frame);
920 if (avctx->codec_type == AVMEDIA_TYPE_VIDEO && !override_dimensions) {
921 frame->width = avctx->width;
922 frame->height = avctx->height;
928 int ff_get_buffer(AVCodecContext *avctx, AVFrame *frame, int flags)
930 int ret = get_buffer_internal(avctx, frame, flags);
932 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
933 frame->width = frame->height = 0;
938 static int reget_buffer_internal(AVCodecContext *avctx, AVFrame *frame)
943 av_assert0(avctx->codec_type == AVMEDIA_TYPE_VIDEO);
945 if (frame->data[0] && (frame->width != avctx->width || frame->height != avctx->height || frame->format != avctx->pix_fmt)) {
946 av_log(avctx, AV_LOG_WARNING, "Picture changed from size:%dx%d fmt:%s to size:%dx%d fmt:%s in reget buffer()\n",
947 frame->width, frame->height, av_get_pix_fmt_name(frame->format), avctx->width, avctx->height, av_get_pix_fmt_name(avctx->pix_fmt));
948 av_frame_unref(frame);
951 ff_init_buffer_info(avctx, frame);
954 return ff_get_buffer(avctx, frame, AV_GET_BUFFER_FLAG_REF);
956 if (av_frame_is_writable(frame))
957 return ff_decode_frame_props(avctx, frame);
959 tmp = av_frame_alloc();
961 return AVERROR(ENOMEM);
963 av_frame_move_ref(tmp, frame);
965 ret = ff_get_buffer(avctx, frame, AV_GET_BUFFER_FLAG_REF);
971 av_frame_copy(frame, tmp);
977 int ff_reget_buffer(AVCodecContext *avctx, AVFrame *frame)
979 int ret = reget_buffer_internal(avctx, frame);
981 av_log(avctx, AV_LOG_ERROR, "reget_buffer() failed\n");
985 int avcodec_default_execute(AVCodecContext *c, int (*func)(AVCodecContext *c2, void *arg2), void *arg, int *ret, int count, int size)
989 for (i = 0; i < count; i++) {
990 int r = func(c, (char *)arg + i * size);
997 int avcodec_default_execute2(AVCodecContext *c, int (*func)(AVCodecContext *c2, void *arg2, int jobnr, int threadnr), void *arg, int *ret, int count)
1001 for (i = 0; i < count; i++) {
1002 int r = func(c, arg, i, 0);
1009 enum AVPixelFormat avpriv_find_pix_fmt(const PixelFormatTag *tags,
1010 unsigned int fourcc)
1012 while (tags->pix_fmt >= 0) {
1013 if (tags->fourcc == fourcc)
1014 return tags->pix_fmt;
1017 return AV_PIX_FMT_NONE;
1020 static int is_hwaccel_pix_fmt(enum AVPixelFormat pix_fmt)
1022 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
1023 return desc->flags & AV_PIX_FMT_FLAG_HWACCEL;
1026 enum AVPixelFormat avcodec_default_get_format(struct AVCodecContext *s, const enum AVPixelFormat *fmt)
1028 while (*fmt != AV_PIX_FMT_NONE && is_hwaccel_pix_fmt(*fmt))
1033 static AVHWAccel *find_hwaccel(enum AVCodecID codec_id,
1034 enum AVPixelFormat pix_fmt)
1036 AVHWAccel *hwaccel = NULL;
1038 while ((hwaccel = av_hwaccel_next(hwaccel)))
1039 if (hwaccel->id == codec_id
1040 && hwaccel->pix_fmt == pix_fmt)
1045 static int setup_hwaccel(AVCodecContext *avctx,
1046 const enum AVPixelFormat fmt,
1049 AVHWAccel *hwa = find_hwaccel(avctx->codec_id, fmt);
1052 if (avctx->active_thread_type & FF_THREAD_FRAME) {
1053 av_log(avctx, AV_LOG_WARNING,
1054 "Hardware accelerated decoding with frame threading is known to be unstable and its use is discouraged.\n");
1058 av_log(avctx, AV_LOG_ERROR,
1059 "Could not find an AVHWAccel for the pixel format: %s",
1061 return AVERROR(ENOENT);
1064 if (hwa->capabilities & HWACCEL_CODEC_CAP_EXPERIMENTAL &&
1065 avctx->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) {
1066 av_log(avctx, AV_LOG_WARNING, "Ignoring experimental hwaccel: %s\n",
1068 return AVERROR_PATCHWELCOME;
1071 if (hwa->priv_data_size) {
1072 avctx->internal->hwaccel_priv_data = av_mallocz(hwa->priv_data_size);
1073 if (!avctx->internal->hwaccel_priv_data)
1074 return AVERROR(ENOMEM);
1078 ret = hwa->init(avctx);
1080 av_freep(&avctx->internal->hwaccel_priv_data);
1085 avctx->hwaccel = hwa;
1090 int ff_get_format(AVCodecContext *avctx, const enum AVPixelFormat *fmt)
1092 const AVPixFmtDescriptor *desc;
1093 enum AVPixelFormat *choices;
1094 enum AVPixelFormat ret;
1097 while (fmt[n] != AV_PIX_FMT_NONE)
1101 avctx->sw_pix_fmt = fmt[n - 1];
1102 av_assert2(!is_hwaccel_pix_fmt(avctx->sw_pix_fmt));
1104 choices = av_malloc_array(n + 1, sizeof(*choices));
1106 return AV_PIX_FMT_NONE;
1108 memcpy(choices, fmt, (n + 1) * sizeof(*choices));
1111 if (avctx->hwaccel && avctx->hwaccel->uninit)
1112 avctx->hwaccel->uninit(avctx);
1113 av_freep(&avctx->internal->hwaccel_priv_data);
1114 avctx->hwaccel = NULL;
1116 ret = avctx->get_format(avctx, choices);
1118 desc = av_pix_fmt_desc_get(ret);
1120 ret = AV_PIX_FMT_NONE;
1124 if (!(desc->flags & AV_PIX_FMT_FLAG_HWACCEL))
1126 #if FF_API_CAP_VDPAU
1127 if (avctx->codec->capabilities&AV_CODEC_CAP_HWACCEL_VDPAU)
1131 if (!setup_hwaccel(avctx, ret, desc->name))
1134 /* Remove failed hwaccel from choices */
1135 for (n = 0; choices[n] != ret; n++)
1136 av_assert0(choices[n] != AV_PIX_FMT_NONE);
1139 choices[n] = choices[n + 1];
1140 while (choices[n++] != AV_PIX_FMT_NONE);
1147 MAKE_ACCESSORS(AVCodecContext, codec, AVRational, pkt_timebase)
1148 MAKE_ACCESSORS(AVCodecContext, codec, const AVCodecDescriptor *, codec_descriptor)
1149 MAKE_ACCESSORS(AVCodecContext, codec, int, lowres)
1150 MAKE_ACCESSORS(AVCodecContext, codec, int, seek_preroll)
1151 MAKE_ACCESSORS(AVCodecContext, codec, uint16_t*, chroma_intra_matrix)
1153 unsigned av_codec_get_codec_properties(const AVCodecContext *codec)
1155 return codec->properties;
1158 int av_codec_get_max_lowres(const AVCodec *codec)
1160 return codec->max_lowres;
1163 int avpriv_codec_get_cap_skip_frame_fill_param(const AVCodec *codec){
1164 return !!(codec->caps_internal & FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM);
1167 static void get_subtitle_defaults(AVSubtitle *sub)
1169 memset(sub, 0, sizeof(*sub));
1170 sub->pts = AV_NOPTS_VALUE;
1173 static int64_t get_bit_rate(AVCodecContext *ctx)
1176 int bits_per_sample;
1178 switch (ctx->codec_type) {
1179 case AVMEDIA_TYPE_VIDEO:
1180 case AVMEDIA_TYPE_DATA:
1181 case AVMEDIA_TYPE_SUBTITLE:
1182 case AVMEDIA_TYPE_ATTACHMENT:
1183 bit_rate = ctx->bit_rate;
1185 case AVMEDIA_TYPE_AUDIO:
1186 bits_per_sample = av_get_bits_per_sample(ctx->codec_id);
1187 bit_rate = bits_per_sample ? ctx->sample_rate * (int64_t)ctx->channels * bits_per_sample : ctx->bit_rate;
1196 int attribute_align_arg ff_codec_open2_recursive(AVCodecContext *avctx, const AVCodec *codec, AVDictionary **options)
1200 ff_unlock_avcodec(codec);
1202 ret = avcodec_open2(avctx, codec, options);
1204 ff_lock_avcodec(avctx, codec);
1208 int attribute_align_arg avcodec_open2(AVCodecContext *avctx, const AVCodec *codec, AVDictionary **options)
1211 AVDictionary *tmp = NULL;
1212 const AVPixFmtDescriptor *pixdesc;
1214 if (avcodec_is_open(avctx))
1217 if ((!codec && !avctx->codec)) {
1218 av_log(avctx, AV_LOG_ERROR, "No codec provided to avcodec_open2()\n");
1219 return AVERROR(EINVAL);
1221 if ((codec && avctx->codec && codec != avctx->codec)) {
1222 av_log(avctx, AV_LOG_ERROR, "This AVCodecContext was allocated for %s, "
1223 "but %s passed to avcodec_open2()\n", avctx->codec->name, codec->name);
1224 return AVERROR(EINVAL);
1227 codec = avctx->codec;
1229 if (avctx->extradata_size < 0 || avctx->extradata_size >= FF_MAX_EXTRADATA_SIZE)
1230 return AVERROR(EINVAL);
1233 av_dict_copy(&tmp, *options, 0);
1235 ret = ff_lock_avcodec(avctx, codec);
1239 avctx->internal = av_mallocz(sizeof(AVCodecInternal));
1240 if (!avctx->internal) {
1241 ret = AVERROR(ENOMEM);
1245 avctx->internal->pool = av_mallocz(sizeof(*avctx->internal->pool));
1246 if (!avctx->internal->pool) {
1247 ret = AVERROR(ENOMEM);
1251 avctx->internal->to_free = av_frame_alloc();
1252 if (!avctx->internal->to_free) {
1253 ret = AVERROR(ENOMEM);
1257 avctx->internal->buffer_frame = av_frame_alloc();
1258 if (!avctx->internal->buffer_frame) {
1259 ret = AVERROR(ENOMEM);
1263 avctx->internal->buffer_pkt = av_packet_alloc();
1264 if (!avctx->internal->buffer_pkt) {
1265 ret = AVERROR(ENOMEM);
1269 if (codec->priv_data_size > 0) {
1270 if (!avctx->priv_data) {
1271 avctx->priv_data = av_mallocz(codec->priv_data_size);
1272 if (!avctx->priv_data) {
1273 ret = AVERROR(ENOMEM);
1276 if (codec->priv_class) {
1277 *(const AVClass **)avctx->priv_data = codec->priv_class;
1278 av_opt_set_defaults(avctx->priv_data);
1281 if (codec->priv_class && (ret = av_opt_set_dict(avctx->priv_data, &tmp)) < 0)
1284 avctx->priv_data = NULL;
1286 if ((ret = av_opt_set_dict(avctx, &tmp)) < 0)
1289 if (avctx->codec_whitelist && av_match_list(codec->name, avctx->codec_whitelist, ',') <= 0) {
1290 av_log(avctx, AV_LOG_ERROR, "Codec (%s) not on whitelist \'%s\'\n", codec->name, avctx->codec_whitelist);
1291 ret = AVERROR(EINVAL);
1295 // only call ff_set_dimensions() for non H.264/VP6F/DXV codecs so as not to overwrite previously setup dimensions
1296 if (!(avctx->coded_width && avctx->coded_height && avctx->width && avctx->height &&
1297 (avctx->codec_id == AV_CODEC_ID_H264 || avctx->codec_id == AV_CODEC_ID_VP6F || avctx->codec_id == AV_CODEC_ID_DXV))) {
1298 if (avctx->coded_width && avctx->coded_height)
1299 ret = ff_set_dimensions(avctx, avctx->coded_width, avctx->coded_height);
1300 else if (avctx->width && avctx->height)
1301 ret = ff_set_dimensions(avctx, avctx->width, avctx->height);
1306 if ((avctx->coded_width || avctx->coded_height || avctx->width || avctx->height)
1307 && ( av_image_check_size(avctx->coded_width, avctx->coded_height, 0, avctx) < 0
1308 || av_image_check_size(avctx->width, avctx->height, 0, avctx) < 0)) {
1309 av_log(avctx, AV_LOG_WARNING, "Ignoring invalid width/height values\n");
1310 ff_set_dimensions(avctx, 0, 0);
1313 if (avctx->width > 0 && avctx->height > 0) {
1314 if (av_image_check_sar(avctx->width, avctx->height,
1315 avctx->sample_aspect_ratio) < 0) {
1316 av_log(avctx, AV_LOG_WARNING, "ignoring invalid SAR: %u/%u\n",
1317 avctx->sample_aspect_ratio.num,
1318 avctx->sample_aspect_ratio.den);
1319 avctx->sample_aspect_ratio = (AVRational){ 0, 1 };
1323 /* if the decoder init function was already called previously,
1324 * free the already allocated subtitle_header before overwriting it */
1325 if (av_codec_is_decoder(codec))
1326 av_freep(&avctx->subtitle_header);
1328 if (avctx->channels > FF_SANE_NB_CHANNELS) {
1329 ret = AVERROR(EINVAL);
1333 avctx->codec = codec;
1334 if ((avctx->codec_type == AVMEDIA_TYPE_UNKNOWN || avctx->codec_type == codec->type) &&
1335 avctx->codec_id == AV_CODEC_ID_NONE) {
1336 avctx->codec_type = codec->type;
1337 avctx->codec_id = codec->id;
1339 if (avctx->codec_id != codec->id || (avctx->codec_type != codec->type
1340 && avctx->codec_type != AVMEDIA_TYPE_ATTACHMENT)) {
1341 av_log(avctx, AV_LOG_ERROR, "Codec type or id mismatches\n");
1342 ret = AVERROR(EINVAL);
1345 avctx->frame_number = 0;
1346 avctx->codec_descriptor = avcodec_descriptor_get(avctx->codec_id);
1348 if ((avctx->codec->capabilities & AV_CODEC_CAP_EXPERIMENTAL) &&
1349 avctx->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) {
1350 const char *codec_string = av_codec_is_encoder(codec) ? "encoder" : "decoder";
1352 av_log(avctx, AV_LOG_ERROR,
1353 "The %s '%s' is experimental but experimental codecs are not enabled, "
1354 "add '-strict %d' if you want to use it.\n",
1355 codec_string, codec->name, FF_COMPLIANCE_EXPERIMENTAL);
1356 codec2 = av_codec_is_encoder(codec) ? avcodec_find_encoder(codec->id) : avcodec_find_decoder(codec->id);
1357 if (!(codec2->capabilities & AV_CODEC_CAP_EXPERIMENTAL))
1358 av_log(avctx, AV_LOG_ERROR, "Alternatively use the non experimental %s '%s'.\n",
1359 codec_string, codec2->name);
1360 ret = AVERROR_EXPERIMENTAL;
1364 if (avctx->codec_type == AVMEDIA_TYPE_AUDIO &&
1365 (!avctx->time_base.num || !avctx->time_base.den)) {
1366 avctx->time_base.num = 1;
1367 avctx->time_base.den = avctx->sample_rate;
1371 av_log(avctx, AV_LOG_WARNING, "Warning: not compiled with thread support, using thread emulation\n");
1373 if (CONFIG_FRAME_THREAD_ENCODER && av_codec_is_encoder(avctx->codec)) {
1374 ff_unlock_avcodec(codec); //we will instantiate a few encoders thus kick the counter to prevent false detection of a problem
1375 ret = ff_frame_thread_encoder_init(avctx, options ? *options : NULL);
1376 ff_lock_avcodec(avctx, codec);
1382 && !(avctx->internal->frame_thread_encoder && (avctx->active_thread_type&FF_THREAD_FRAME))) {
1383 ret = ff_thread_init(avctx);
1388 if (!HAVE_THREADS && !(codec->capabilities & AV_CODEC_CAP_AUTO_THREADS))
1389 avctx->thread_count = 1;
1391 if (avctx->codec->max_lowres < avctx->lowres || avctx->lowres < 0) {
1392 av_log(avctx, AV_LOG_ERROR, "The maximum value for lowres supported by the decoder is %d\n",
1393 avctx->codec->max_lowres);
1394 ret = AVERROR(EINVAL);
1399 if (avctx->debug_mv)
1400 av_log(avctx, AV_LOG_WARNING, "The 'vismv' option is deprecated, "
1401 "see the codecview filter instead.\n");
1404 if (av_codec_is_encoder(avctx->codec)) {
1406 #if FF_API_CODED_FRAME
1407 FF_DISABLE_DEPRECATION_WARNINGS
1408 avctx->coded_frame = av_frame_alloc();
1409 if (!avctx->coded_frame) {
1410 ret = AVERROR(ENOMEM);
1413 FF_ENABLE_DEPRECATION_WARNINGS
1415 if (avctx->codec->sample_fmts) {
1416 for (i = 0; avctx->codec->sample_fmts[i] != AV_SAMPLE_FMT_NONE; i++) {
1417 if (avctx->sample_fmt == avctx->codec->sample_fmts[i])
1419 if (avctx->channels == 1 &&
1420 av_get_planar_sample_fmt(avctx->sample_fmt) ==
1421 av_get_planar_sample_fmt(avctx->codec->sample_fmts[i])) {
1422 avctx->sample_fmt = avctx->codec->sample_fmts[i];
1426 if (avctx->codec->sample_fmts[i] == AV_SAMPLE_FMT_NONE) {
1428 snprintf(buf, sizeof(buf), "%d", avctx->sample_fmt);
1429 av_log(avctx, AV_LOG_ERROR, "Specified sample format %s is invalid or not supported\n",
1430 (char *)av_x_if_null(av_get_sample_fmt_name(avctx->sample_fmt), buf));
1431 ret = AVERROR(EINVAL);
1435 if (avctx->codec->pix_fmts) {
1436 for (i = 0; avctx->codec->pix_fmts[i] != AV_PIX_FMT_NONE; i++)
1437 if (avctx->pix_fmt == avctx->codec->pix_fmts[i])
1439 if (avctx->codec->pix_fmts[i] == AV_PIX_FMT_NONE
1440 && !((avctx->codec_id == AV_CODEC_ID_MJPEG || avctx->codec_id == AV_CODEC_ID_LJPEG)
1441 && avctx->strict_std_compliance <= FF_COMPLIANCE_UNOFFICIAL)) {
1443 snprintf(buf, sizeof(buf), "%d", avctx->pix_fmt);
1444 av_log(avctx, AV_LOG_ERROR, "Specified pixel format %s is invalid or not supported\n",
1445 (char *)av_x_if_null(av_get_pix_fmt_name(avctx->pix_fmt), buf));
1446 ret = AVERROR(EINVAL);
1449 if (avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ420P ||
1450 avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ411P ||
1451 avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ422P ||
1452 avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ440P ||
1453 avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ444P)
1454 avctx->color_range = AVCOL_RANGE_JPEG;
1456 if (avctx->codec->supported_samplerates) {
1457 for (i = 0; avctx->codec->supported_samplerates[i] != 0; i++)
1458 if (avctx->sample_rate == avctx->codec->supported_samplerates[i])
1460 if (avctx->codec->supported_samplerates[i] == 0) {
1461 av_log(avctx, AV_LOG_ERROR, "Specified sample rate %d is not supported\n",
1462 avctx->sample_rate);
1463 ret = AVERROR(EINVAL);
1467 if (avctx->sample_rate < 0) {
1468 av_log(avctx, AV_LOG_ERROR, "Specified sample rate %d is not supported\n",
1469 avctx->sample_rate);
1470 ret = AVERROR(EINVAL);
1473 if (avctx->codec->channel_layouts) {
1474 if (!avctx->channel_layout) {
1475 av_log(avctx, AV_LOG_WARNING, "Channel layout not specified\n");
1477 for (i = 0; avctx->codec->channel_layouts[i] != 0; i++)
1478 if (avctx->channel_layout == avctx->codec->channel_layouts[i])
1480 if (avctx->codec->channel_layouts[i] == 0) {
1482 av_get_channel_layout_string(buf, sizeof(buf), -1, avctx->channel_layout);
1483 av_log(avctx, AV_LOG_ERROR, "Specified channel layout '%s' is not supported\n", buf);
1484 ret = AVERROR(EINVAL);
1489 if (avctx->channel_layout && avctx->channels) {
1490 int channels = av_get_channel_layout_nb_channels(avctx->channel_layout);
1491 if (channels != avctx->channels) {
1493 av_get_channel_layout_string(buf, sizeof(buf), -1, avctx->channel_layout);
1494 av_log(avctx, AV_LOG_ERROR,
1495 "Channel layout '%s' with %d channels does not match number of specified channels %d\n",
1496 buf, channels, avctx->channels);
1497 ret = AVERROR(EINVAL);
1500 } else if (avctx->channel_layout) {
1501 avctx->channels = av_get_channel_layout_nb_channels(avctx->channel_layout);
1503 if (avctx->channels < 0) {
1504 av_log(avctx, AV_LOG_ERROR, "Specified number of channels %d is not supported\n",
1506 ret = AVERROR(EINVAL);
1509 if(avctx->codec_type == AVMEDIA_TYPE_VIDEO) {
1510 pixdesc = av_pix_fmt_desc_get(avctx->pix_fmt);
1511 if ( avctx->bits_per_raw_sample < 0
1512 || (avctx->bits_per_raw_sample > 8 && pixdesc->comp[0].depth <= 8)) {
1513 av_log(avctx, AV_LOG_WARNING, "Specified bit depth %d not possible with the specified pixel formats depth %d\n",
1514 avctx->bits_per_raw_sample, pixdesc->comp[0].depth);
1515 avctx->bits_per_raw_sample = pixdesc->comp[0].depth;
1517 if (avctx->width <= 0 || avctx->height <= 0) {
1518 av_log(avctx, AV_LOG_ERROR, "dimensions not set\n");
1519 ret = AVERROR(EINVAL);
1523 if ( (avctx->codec_type == AVMEDIA_TYPE_VIDEO || avctx->codec_type == AVMEDIA_TYPE_AUDIO)
1524 && avctx->bit_rate>0 && avctx->bit_rate<1000) {
1525 av_log(avctx, AV_LOG_WARNING, "Bitrate %"PRId64" is extremely low, maybe you mean %"PRId64"k\n", (int64_t)avctx->bit_rate, (int64_t)avctx->bit_rate);
1528 if (!avctx->rc_initial_buffer_occupancy)
1529 avctx->rc_initial_buffer_occupancy = avctx->rc_buffer_size * 3 / 4;
1531 if (avctx->ticks_per_frame && avctx->time_base.num &&
1532 avctx->ticks_per_frame > INT_MAX / avctx->time_base.num) {
1533 av_log(avctx, AV_LOG_ERROR,
1534 "ticks_per_frame %d too large for the timebase %d/%d.",
1535 avctx->ticks_per_frame,
1536 avctx->time_base.num,
1537 avctx->time_base.den);
1541 if (avctx->hw_frames_ctx) {
1542 AVHWFramesContext *frames_ctx = (AVHWFramesContext*)avctx->hw_frames_ctx->data;
1543 if (frames_ctx->format != avctx->pix_fmt) {
1544 av_log(avctx, AV_LOG_ERROR,
1545 "Mismatching AVCodecContext.pix_fmt and AVHWFramesContext.format\n");
1546 ret = AVERROR(EINVAL);
1552 avctx->pts_correction_num_faulty_pts =
1553 avctx->pts_correction_num_faulty_dts = 0;
1554 avctx->pts_correction_last_pts =
1555 avctx->pts_correction_last_dts = INT64_MIN;
1557 if ( !CONFIG_GRAY && avctx->flags & AV_CODEC_FLAG_GRAY
1558 && avctx->codec_descriptor->type == AVMEDIA_TYPE_VIDEO)
1559 av_log(avctx, AV_LOG_WARNING,
1560 "gray decoding requested but not enabled at configuration time\n");
1562 if ( avctx->codec->init && (!(avctx->active_thread_type&FF_THREAD_FRAME)
1563 || avctx->internal->frame_thread_encoder)) {
1564 ret = avctx->codec->init(avctx);
1572 #if FF_API_AUDIOENC_DELAY
1573 if (av_codec_is_encoder(avctx->codec))
1574 avctx->delay = avctx->initial_padding;
1577 if (av_codec_is_decoder(avctx->codec)) {
1578 if (!avctx->bit_rate)
1579 avctx->bit_rate = get_bit_rate(avctx);
1580 /* validate channel layout from the decoder */
1581 if (avctx->channel_layout) {
1582 int channels = av_get_channel_layout_nb_channels(avctx->channel_layout);
1583 if (!avctx->channels)
1584 avctx->channels = channels;
1585 else if (channels != avctx->channels) {
1587 av_get_channel_layout_string(buf, sizeof(buf), -1, avctx->channel_layout);
1588 av_log(avctx, AV_LOG_WARNING,
1589 "Channel layout '%s' with %d channels does not match specified number of channels %d: "
1590 "ignoring specified channel layout\n",
1591 buf, channels, avctx->channels);
1592 avctx->channel_layout = 0;
1595 if (avctx->channels && avctx->channels < 0 ||
1596 avctx->channels > FF_SANE_NB_CHANNELS) {
1597 ret = AVERROR(EINVAL);
1600 if (avctx->sub_charenc) {
1601 if (avctx->codec_type != AVMEDIA_TYPE_SUBTITLE) {
1602 av_log(avctx, AV_LOG_ERROR, "Character encoding is only "
1603 "supported with subtitles codecs\n");
1604 ret = AVERROR(EINVAL);
1606 } else if (avctx->codec_descriptor->props & AV_CODEC_PROP_BITMAP_SUB) {
1607 av_log(avctx, AV_LOG_WARNING, "Codec '%s' is bitmap-based, "
1608 "subtitles character encoding will be ignored\n",
1609 avctx->codec_descriptor->name);
1610 avctx->sub_charenc_mode = FF_SUB_CHARENC_MODE_DO_NOTHING;
1612 /* input character encoding is set for a text based subtitle
1613 * codec at this point */
1614 if (avctx->sub_charenc_mode == FF_SUB_CHARENC_MODE_AUTOMATIC)
1615 avctx->sub_charenc_mode = FF_SUB_CHARENC_MODE_PRE_DECODER;
1617 if (avctx->sub_charenc_mode == FF_SUB_CHARENC_MODE_PRE_DECODER) {
1619 iconv_t cd = iconv_open("UTF-8", avctx->sub_charenc);
1620 if (cd == (iconv_t)-1) {
1621 ret = AVERROR(errno);
1622 av_log(avctx, AV_LOG_ERROR, "Unable to open iconv context "
1623 "with input character encoding \"%s\"\n", avctx->sub_charenc);
1628 av_log(avctx, AV_LOG_ERROR, "Character encoding subtitles "
1629 "conversion needs a libavcodec built with iconv support "
1630 "for this codec\n");
1631 ret = AVERROR(ENOSYS);
1638 #if FF_API_AVCTX_TIMEBASE
1639 if (avctx->framerate.num > 0 && avctx->framerate.den > 0)
1640 avctx->time_base = av_inv_q(av_mul_q(avctx->framerate, (AVRational){avctx->ticks_per_frame, 1}));
1643 if (codec->priv_data_size > 0 && avctx->priv_data && codec->priv_class) {
1644 av_assert0(*(const AVClass **)avctx->priv_data == codec->priv_class);
1648 ff_unlock_avcodec(codec);
1650 av_dict_free(options);
1657 (avctx->codec->caps_internal & FF_CODEC_CAP_INIT_CLEANUP))
1658 avctx->codec->close(avctx);
1660 if (codec->priv_class && codec->priv_data_size)
1661 av_opt_free(avctx->priv_data);
1664 #if FF_API_CODED_FRAME
1665 FF_DISABLE_DEPRECATION_WARNINGS
1666 av_frame_free(&avctx->coded_frame);
1667 FF_ENABLE_DEPRECATION_WARNINGS
1671 av_freep(&avctx->priv_data);
1672 if (avctx->internal) {
1673 av_frame_free(&avctx->internal->to_free);
1674 av_freep(&avctx->internal->pool);
1676 av_freep(&avctx->internal);
1677 avctx->codec = NULL;
1681 int ff_alloc_packet2(AVCodecContext *avctx, AVPacket *avpkt, int64_t size, int64_t min_size)
1683 if (avpkt->size < 0) {
1684 av_log(avctx, AV_LOG_ERROR, "Invalid negative user packet size %d\n", avpkt->size);
1685 return AVERROR(EINVAL);
1687 if (size < 0 || size > INT_MAX - AV_INPUT_BUFFER_PADDING_SIZE) {
1688 av_log(avctx, AV_LOG_ERROR, "Invalid minimum required packet size %"PRId64" (max allowed is %d)\n",
1689 size, INT_MAX - AV_INPUT_BUFFER_PADDING_SIZE);
1690 return AVERROR(EINVAL);
1693 if (avctx && 2*min_size < size) { // FIXME The factor needs to be finetuned
1694 av_assert0(!avpkt->data || avpkt->data != avctx->internal->byte_buffer);
1695 if (!avpkt->data || avpkt->size < size) {
1696 av_fast_padded_malloc(&avctx->internal->byte_buffer, &avctx->internal->byte_buffer_size, size);
1697 avpkt->data = avctx->internal->byte_buffer;
1698 avpkt->size = avctx->internal->byte_buffer_size;
1703 AVBufferRef *buf = avpkt->buf;
1705 if (avpkt->size < size) {
1706 av_log(avctx, AV_LOG_ERROR, "User packet is too small (%d < %"PRId64")\n", avpkt->size, size);
1707 return AVERROR(EINVAL);
1710 av_init_packet(avpkt);
1715 int ret = av_new_packet(avpkt, size);
1717 av_log(avctx, AV_LOG_ERROR, "Failed to allocate packet of size %"PRId64"\n", size);
1722 int ff_alloc_packet(AVPacket *avpkt, int size)
1724 return ff_alloc_packet2(NULL, avpkt, size, 0);
1728 * Pad last frame with silence.
1730 static int pad_last_frame(AVCodecContext *s, AVFrame **dst, const AVFrame *src)
1732 AVFrame *frame = NULL;
1735 if (!(frame = av_frame_alloc()))
1736 return AVERROR(ENOMEM);
1738 frame->format = src->format;
1739 frame->channel_layout = src->channel_layout;
1740 av_frame_set_channels(frame, av_frame_get_channels(src));
1741 frame->nb_samples = s->frame_size;
1742 ret = av_frame_get_buffer(frame, 32);
1746 ret = av_frame_copy_props(frame, src);
1750 if ((ret = av_samples_copy(frame->extended_data, src->extended_data, 0, 0,
1751 src->nb_samples, s->channels, s->sample_fmt)) < 0)
1753 if ((ret = av_samples_set_silence(frame->extended_data, src->nb_samples,
1754 frame->nb_samples - src->nb_samples,
1755 s->channels, s->sample_fmt)) < 0)
1763 av_frame_free(&frame);
1767 int attribute_align_arg avcodec_encode_audio2(AVCodecContext *avctx,
1769 const AVFrame *frame,
1770 int *got_packet_ptr)
1772 AVFrame *extended_frame = NULL;
1773 AVFrame *padded_frame = NULL;
1775 AVPacket user_pkt = *avpkt;
1776 int needs_realloc = !user_pkt.data;
1778 *got_packet_ptr = 0;
1780 if (!avctx->codec->encode2) {
1781 av_log(avctx, AV_LOG_ERROR, "This encoder requires using the avcodec_send_frame() API.\n");
1782 return AVERROR(ENOSYS);
1785 if (!(avctx->codec->capabilities & AV_CODEC_CAP_DELAY) && !frame) {
1786 av_packet_unref(avpkt);
1787 av_init_packet(avpkt);
1791 /* ensure that extended_data is properly set */
1792 if (frame && !frame->extended_data) {
1793 if (av_sample_fmt_is_planar(avctx->sample_fmt) &&
1794 avctx->channels > AV_NUM_DATA_POINTERS) {
1795 av_log(avctx, AV_LOG_ERROR, "Encoding to a planar sample format, "
1796 "with more than %d channels, but extended_data is not set.\n",
1797 AV_NUM_DATA_POINTERS);
1798 return AVERROR(EINVAL);
1800 av_log(avctx, AV_LOG_WARNING, "extended_data is not set.\n");
1802 extended_frame = av_frame_alloc();
1803 if (!extended_frame)
1804 return AVERROR(ENOMEM);
1806 memcpy(extended_frame, frame, sizeof(AVFrame));
1807 extended_frame->extended_data = extended_frame->data;
1808 frame = extended_frame;
1811 /* extract audio service type metadata */
1813 AVFrameSideData *sd = av_frame_get_side_data(frame, AV_FRAME_DATA_AUDIO_SERVICE_TYPE);
1814 if (sd && sd->size >= sizeof(enum AVAudioServiceType))
1815 avctx->audio_service_type = *(enum AVAudioServiceType*)sd->data;
1818 /* check for valid frame size */
1820 if (avctx->codec->capabilities & AV_CODEC_CAP_SMALL_LAST_FRAME) {
1821 if (frame->nb_samples > avctx->frame_size) {
1822 av_log(avctx, AV_LOG_ERROR, "more samples than frame size (avcodec_encode_audio2)\n");
1823 ret = AVERROR(EINVAL);
1826 } else if (!(avctx->codec->capabilities & AV_CODEC_CAP_VARIABLE_FRAME_SIZE)) {
1827 if (frame->nb_samples < avctx->frame_size &&
1828 !avctx->internal->last_audio_frame) {
1829 ret = pad_last_frame(avctx, &padded_frame, frame);
1833 frame = padded_frame;
1834 avctx->internal->last_audio_frame = 1;
1837 if (frame->nb_samples != avctx->frame_size) {
1838 av_log(avctx, AV_LOG_ERROR, "nb_samples (%d) != frame_size (%d) (avcodec_encode_audio2)\n", frame->nb_samples, avctx->frame_size);
1839 ret = AVERROR(EINVAL);
1845 av_assert0(avctx->codec->encode2);
1847 ret = avctx->codec->encode2(avctx, avpkt, frame, got_packet_ptr);
1849 if (*got_packet_ptr) {
1850 if (!(avctx->codec->capabilities & AV_CODEC_CAP_DELAY)) {
1851 if (avpkt->pts == AV_NOPTS_VALUE)
1852 avpkt->pts = frame->pts;
1853 if (!avpkt->duration)
1854 avpkt->duration = ff_samples_to_time_base(avctx,
1857 avpkt->dts = avpkt->pts;
1862 if (avpkt->data && avpkt->data == avctx->internal->byte_buffer) {
1864 if (user_pkt.data) {
1865 if (user_pkt.size >= avpkt->size) {
1866 memcpy(user_pkt.data, avpkt->data, avpkt->size);
1868 av_log(avctx, AV_LOG_ERROR, "Provided packet is too small, needs to be %d\n", avpkt->size);
1869 avpkt->size = user_pkt.size;
1872 avpkt->buf = user_pkt.buf;
1873 avpkt->data = user_pkt.data;
1875 if (av_dup_packet(avpkt) < 0) {
1876 ret = AVERROR(ENOMEM);
1882 if (needs_realloc && avpkt->data) {
1883 ret = av_buffer_realloc(&avpkt->buf, avpkt->size + AV_INPUT_BUFFER_PADDING_SIZE);
1885 avpkt->data = avpkt->buf->data;
1888 avctx->frame_number++;
1891 if (ret < 0 || !*got_packet_ptr) {
1892 av_packet_unref(avpkt);
1893 av_init_packet(avpkt);
1897 /* NOTE: if we add any audio encoders which output non-keyframe packets,
1898 * this needs to be moved to the encoders, but for now we can do it
1899 * here to simplify things */
1900 avpkt->flags |= AV_PKT_FLAG_KEY;
1903 av_frame_free(&padded_frame);
1904 av_free(extended_frame);
1906 #if FF_API_AUDIOENC_DELAY
1907 avctx->delay = avctx->initial_padding;
1913 int attribute_align_arg avcodec_encode_video2(AVCodecContext *avctx,
1915 const AVFrame *frame,
1916 int *got_packet_ptr)
1919 AVPacket user_pkt = *avpkt;
1920 int needs_realloc = !user_pkt.data;
1922 *got_packet_ptr = 0;
1924 if (!avctx->codec->encode2) {
1925 av_log(avctx, AV_LOG_ERROR, "This encoder requires using the avcodec_send_frame() API.\n");
1926 return AVERROR(ENOSYS);
1929 if(CONFIG_FRAME_THREAD_ENCODER &&
1930 avctx->internal->frame_thread_encoder && (avctx->active_thread_type&FF_THREAD_FRAME))
1931 return ff_thread_video_encode_frame(avctx, avpkt, frame, got_packet_ptr);
1933 if ((avctx->flags&AV_CODEC_FLAG_PASS1) && avctx->stats_out)
1934 avctx->stats_out[0] = '\0';
1936 if (!(avctx->codec->capabilities & AV_CODEC_CAP_DELAY) && !frame) {
1937 av_packet_unref(avpkt);
1938 av_init_packet(avpkt);
1943 if (av_image_check_size(avctx->width, avctx->height, 0, avctx))
1944 return AVERROR(EINVAL);
1946 if (frame && frame->format == AV_PIX_FMT_NONE)
1947 av_log(avctx, AV_LOG_WARNING, "AVFrame.format is not set\n");
1948 if (frame && (frame->width == 0 || frame->height == 0))
1949 av_log(avctx, AV_LOG_WARNING, "AVFrame.width or height is not set\n");
1951 av_assert0(avctx->codec->encode2);
1953 ret = avctx->codec->encode2(avctx, avpkt, frame, got_packet_ptr);
1954 av_assert0(ret <= 0);
1956 if (avpkt->data && avpkt->data == avctx->internal->byte_buffer) {
1958 if (user_pkt.data) {
1959 if (user_pkt.size >= avpkt->size) {
1960 memcpy(user_pkt.data, avpkt->data, avpkt->size);
1962 av_log(avctx, AV_LOG_ERROR, "Provided packet is too small, needs to be %d\n", avpkt->size);
1963 avpkt->size = user_pkt.size;
1966 avpkt->buf = user_pkt.buf;
1967 avpkt->data = user_pkt.data;
1969 if (av_dup_packet(avpkt) < 0) {
1970 ret = AVERROR(ENOMEM);
1976 if (!*got_packet_ptr)
1978 else if (!(avctx->codec->capabilities & AV_CODEC_CAP_DELAY))
1979 avpkt->pts = avpkt->dts = frame->pts;
1981 if (needs_realloc && avpkt->data) {
1982 ret = av_buffer_realloc(&avpkt->buf, avpkt->size + AV_INPUT_BUFFER_PADDING_SIZE);
1984 avpkt->data = avpkt->buf->data;
1987 avctx->frame_number++;
1990 if (ret < 0 || !*got_packet_ptr)
1991 av_packet_unref(avpkt);
1997 int avcodec_encode_subtitle(AVCodecContext *avctx, uint8_t *buf, int buf_size,
1998 const AVSubtitle *sub)
2001 if (sub->start_display_time) {
2002 av_log(avctx, AV_LOG_ERROR, "start_display_time must be 0.\n");
2006 ret = avctx->codec->encode_sub(avctx, buf, buf_size, sub);
2007 avctx->frame_number++;
2012 * Attempt to guess proper monotonic timestamps for decoded video frames
2013 * which might have incorrect times. Input timestamps may wrap around, in
2014 * which case the output will as well.
2016 * @param pts the pts field of the decoded AVPacket, as passed through
2018 * @param dts the dts field of the decoded AVPacket
2019 * @return one of the input values, may be AV_NOPTS_VALUE
2021 static int64_t guess_correct_pts(AVCodecContext *ctx,
2022 int64_t reordered_pts, int64_t dts)
2024 int64_t pts = AV_NOPTS_VALUE;
2026 if (dts != AV_NOPTS_VALUE) {
2027 ctx->pts_correction_num_faulty_dts += dts <= ctx->pts_correction_last_dts;
2028 ctx->pts_correction_last_dts = dts;
2029 } else if (reordered_pts != AV_NOPTS_VALUE)
2030 ctx->pts_correction_last_dts = reordered_pts;
2032 if (reordered_pts != AV_NOPTS_VALUE) {
2033 ctx->pts_correction_num_faulty_pts += reordered_pts <= ctx->pts_correction_last_pts;
2034 ctx->pts_correction_last_pts = reordered_pts;
2035 } else if(dts != AV_NOPTS_VALUE)
2036 ctx->pts_correction_last_pts = dts;
2038 if ((ctx->pts_correction_num_faulty_pts<=ctx->pts_correction_num_faulty_dts || dts == AV_NOPTS_VALUE)
2039 && reordered_pts != AV_NOPTS_VALUE)
2040 pts = reordered_pts;
2047 static int apply_param_change(AVCodecContext *avctx, AVPacket *avpkt)
2050 const uint8_t *data;
2054 data = av_packet_get_side_data(avpkt, AV_PKT_DATA_PARAM_CHANGE, &size);
2058 if (!(avctx->codec->capabilities & AV_CODEC_CAP_PARAM_CHANGE)) {
2059 av_log(avctx, AV_LOG_ERROR, "This decoder does not support parameter "
2060 "changes, but PARAM_CHANGE side data was sent to it.\n");
2061 ret = AVERROR(EINVAL);
2068 flags = bytestream_get_le32(&data);
2071 if (flags & AV_SIDE_DATA_PARAM_CHANGE_CHANNEL_COUNT) {
2074 val = bytestream_get_le32(&data);
2075 if (val <= 0 || val > INT_MAX) {
2076 av_log(avctx, AV_LOG_ERROR, "Invalid channel count");
2077 ret = AVERROR_INVALIDDATA;
2080 avctx->channels = val;
2083 if (flags & AV_SIDE_DATA_PARAM_CHANGE_CHANNEL_LAYOUT) {
2086 avctx->channel_layout = bytestream_get_le64(&data);
2089 if (flags & AV_SIDE_DATA_PARAM_CHANGE_SAMPLE_RATE) {
2092 val = bytestream_get_le32(&data);
2093 if (val <= 0 || val > INT_MAX) {
2094 av_log(avctx, AV_LOG_ERROR, "Invalid sample rate");
2095 ret = AVERROR_INVALIDDATA;
2098 avctx->sample_rate = val;
2101 if (flags & AV_SIDE_DATA_PARAM_CHANGE_DIMENSIONS) {
2104 avctx->width = bytestream_get_le32(&data);
2105 avctx->height = bytestream_get_le32(&data);
2107 ret = ff_set_dimensions(avctx, avctx->width, avctx->height);
2114 av_log(avctx, AV_LOG_ERROR, "PARAM_CHANGE side data too small.\n");
2115 ret = AVERROR_INVALIDDATA;
2118 av_log(avctx, AV_LOG_ERROR, "Error applying parameter changes.\n");
2119 if (avctx->err_recognition & AV_EF_EXPLODE)
2125 static int unrefcount_frame(AVCodecInternal *avci, AVFrame *frame)
2129 /* move the original frame to our backup */
2130 av_frame_unref(avci->to_free);
2131 av_frame_move_ref(avci->to_free, frame);
2133 /* now copy everything except the AVBufferRefs back
2134 * note that we make a COPY of the side data, so calling av_frame_free() on
2135 * the caller's frame will work properly */
2136 ret = av_frame_copy_props(frame, avci->to_free);
2140 memcpy(frame->data, avci->to_free->data, sizeof(frame->data));
2141 memcpy(frame->linesize, avci->to_free->linesize, sizeof(frame->linesize));
2142 if (avci->to_free->extended_data != avci->to_free->data) {
2143 int planes = av_frame_get_channels(avci->to_free);
2144 int size = planes * sizeof(*frame->extended_data);
2147 av_frame_unref(frame);
2151 frame->extended_data = av_malloc(size);
2152 if (!frame->extended_data) {
2153 av_frame_unref(frame);
2154 return AVERROR(ENOMEM);
2156 memcpy(frame->extended_data, avci->to_free->extended_data,
2159 frame->extended_data = frame->data;
2161 frame->format = avci->to_free->format;
2162 frame->width = avci->to_free->width;
2163 frame->height = avci->to_free->height;
2164 frame->channel_layout = avci->to_free->channel_layout;
2165 frame->nb_samples = avci->to_free->nb_samples;
2166 av_frame_set_channels(frame, av_frame_get_channels(avci->to_free));
2171 int attribute_align_arg avcodec_decode_video2(AVCodecContext *avctx, AVFrame *picture,
2172 int *got_picture_ptr,
2173 const AVPacket *avpkt)
2175 AVCodecInternal *avci = avctx->internal;
2177 // copy to ensure we do not change avpkt
2178 AVPacket tmp = *avpkt;
2181 return AVERROR(EINVAL);
2182 if (avctx->codec->type != AVMEDIA_TYPE_VIDEO) {
2183 av_log(avctx, AV_LOG_ERROR, "Invalid media type for video\n");
2184 return AVERROR(EINVAL);
2187 if (!avctx->codec->decode) {
2188 av_log(avctx, AV_LOG_ERROR, "This decoder requires using the avcodec_send_packet() API.\n");
2189 return AVERROR(ENOSYS);
2192 *got_picture_ptr = 0;
2193 if ((avctx->coded_width || avctx->coded_height) && av_image_check_size(avctx->coded_width, avctx->coded_height, 0, avctx))
2194 return AVERROR(EINVAL);
2196 avctx->internal->pkt = avpkt;
2197 ret = apply_param_change(avctx, avpkt);
2201 av_frame_unref(picture);
2203 if ((avctx->codec->capabilities & AV_CODEC_CAP_DELAY) || avpkt->size ||
2204 (avctx->active_thread_type & FF_THREAD_FRAME)) {
2205 int did_split = av_packet_split_side_data(&tmp);
2206 ret = apply_param_change(avctx, &tmp);
2210 avctx->internal->pkt = &tmp;
2211 if (HAVE_THREADS && avctx->active_thread_type & FF_THREAD_FRAME)
2212 ret = ff_thread_decode_frame(avctx, picture, got_picture_ptr,
2215 ret = avctx->codec->decode(avctx, picture, got_picture_ptr,
2217 if (!(avctx->codec->caps_internal & FF_CODEC_CAP_SETS_PKT_DTS))
2218 picture->pkt_dts = avpkt->dts;
2220 if(!avctx->has_b_frames){
2221 av_frame_set_pkt_pos(picture, avpkt->pos);
2223 //FIXME these should be under if(!avctx->has_b_frames)
2224 /* get_buffer is supposed to set frame parameters */
2225 if (!(avctx->codec->capabilities & AV_CODEC_CAP_DR1)) {
2226 if (!picture->sample_aspect_ratio.num) picture->sample_aspect_ratio = avctx->sample_aspect_ratio;
2227 if (!picture->width) picture->width = avctx->width;
2228 if (!picture->height) picture->height = avctx->height;
2229 if (picture->format == AV_PIX_FMT_NONE) picture->format = avctx->pix_fmt;
2234 emms_c(); //needed to avoid an emms_c() call before every return;
2236 avctx->internal->pkt = NULL;
2238 av_packet_free_side_data(&tmp);
2243 if (*got_picture_ptr) {
2244 if (!avctx->refcounted_frames) {
2245 int err = unrefcount_frame(avci, picture);
2250 avctx->frame_number++;
2251 av_frame_set_best_effort_timestamp(picture,
2252 guess_correct_pts(avctx,
2256 av_frame_unref(picture);
2260 /* many decoders assign whole AVFrames, thus overwriting extended_data;
2261 * make sure it's set correctly */
2262 av_assert0(!picture->extended_data || picture->extended_data == picture->data);
2264 #if FF_API_AVCTX_TIMEBASE
2265 if (avctx->framerate.num > 0 && avctx->framerate.den > 0)
2266 avctx->time_base = av_inv_q(av_mul_q(avctx->framerate, (AVRational){avctx->ticks_per_frame, 1}));
2272 int attribute_align_arg avcodec_decode_audio4(AVCodecContext *avctx,
2275 const AVPacket *avpkt)
2277 AVCodecInternal *avci = avctx->internal;
2282 if (!avctx->codec->decode) {
2283 av_log(avctx, AV_LOG_ERROR, "This decoder requires using the avcodec_send_packet() API.\n");
2284 return AVERROR(ENOSYS);
2287 if (!avpkt->data && avpkt->size) {
2288 av_log(avctx, AV_LOG_ERROR, "invalid packet: NULL data, size != 0\n");
2289 return AVERROR(EINVAL);
2292 return AVERROR(EINVAL);
2293 if (avctx->codec->type != AVMEDIA_TYPE_AUDIO) {
2294 av_log(avctx, AV_LOG_ERROR, "Invalid media type for audio\n");
2295 return AVERROR(EINVAL);
2298 av_frame_unref(frame);
2300 if ((avctx->codec->capabilities & AV_CODEC_CAP_DELAY) || avpkt->size || (avctx->active_thread_type & FF_THREAD_FRAME)) {
2303 uint32_t discard_padding = 0;
2304 uint8_t skip_reason = 0;
2305 uint8_t discard_reason = 0;
2306 // copy to ensure we do not change avpkt
2307 AVPacket tmp = *avpkt;
2308 int did_split = av_packet_split_side_data(&tmp);
2309 ret = apply_param_change(avctx, &tmp);
2313 avctx->internal->pkt = &tmp;
2314 if (HAVE_THREADS && avctx->active_thread_type & FF_THREAD_FRAME)
2315 ret = ff_thread_decode_frame(avctx, frame, got_frame_ptr, &tmp);
2317 ret = avctx->codec->decode(avctx, frame, got_frame_ptr, &tmp);
2318 av_assert0(ret <= tmp.size);
2319 frame->pkt_dts = avpkt->dts;
2321 if (ret >= 0 && *got_frame_ptr) {
2322 avctx->frame_number++;
2323 av_frame_set_best_effort_timestamp(frame,
2324 guess_correct_pts(avctx,
2327 if (frame->format == AV_SAMPLE_FMT_NONE)
2328 frame->format = avctx->sample_fmt;
2329 if (!frame->channel_layout)
2330 frame->channel_layout = avctx->channel_layout;
2331 if (!av_frame_get_channels(frame))
2332 av_frame_set_channels(frame, avctx->channels);
2333 if (!frame->sample_rate)
2334 frame->sample_rate = avctx->sample_rate;
2337 side= av_packet_get_side_data(avctx->internal->pkt, AV_PKT_DATA_SKIP_SAMPLES, &side_size);
2338 if(side && side_size>=10) {
2339 avctx->internal->skip_samples = AV_RL32(side);
2340 discard_padding = AV_RL32(side + 4);
2341 av_log(avctx, AV_LOG_DEBUG, "skip %d / discard %d samples due to side data\n",
2342 avctx->internal->skip_samples, (int)discard_padding);
2343 skip_reason = AV_RL8(side + 8);
2344 discard_reason = AV_RL8(side + 9);
2346 if (avctx->internal->skip_samples && *got_frame_ptr &&
2347 !(avctx->flags2 & AV_CODEC_FLAG2_SKIP_MANUAL)) {
2348 if(frame->nb_samples <= avctx->internal->skip_samples){
2350 avctx->internal->skip_samples -= frame->nb_samples;
2351 av_log(avctx, AV_LOG_DEBUG, "skip whole frame, skip left: %d\n",
2352 avctx->internal->skip_samples);
2354 av_samples_copy(frame->extended_data, frame->extended_data, 0, avctx->internal->skip_samples,
2355 frame->nb_samples - avctx->internal->skip_samples, avctx->channels, frame->format);
2356 if(avctx->pkt_timebase.num && avctx->sample_rate) {
2357 int64_t diff_ts = av_rescale_q(avctx->internal->skip_samples,
2358 (AVRational){1, avctx->sample_rate},
2359 avctx->pkt_timebase);
2360 if(frame->pkt_pts!=AV_NOPTS_VALUE)
2361 frame->pkt_pts += diff_ts;
2362 if(frame->pkt_dts!=AV_NOPTS_VALUE)
2363 frame->pkt_dts += diff_ts;
2364 if (av_frame_get_pkt_duration(frame) >= diff_ts)
2365 av_frame_set_pkt_duration(frame, av_frame_get_pkt_duration(frame) - diff_ts);
2367 av_log(avctx, AV_LOG_WARNING, "Could not update timestamps for skipped samples.\n");
2369 av_log(avctx, AV_LOG_DEBUG, "skip %d/%d samples\n",
2370 avctx->internal->skip_samples, frame->nb_samples);
2371 frame->nb_samples -= avctx->internal->skip_samples;
2372 avctx->internal->skip_samples = 0;
2376 if (discard_padding > 0 && discard_padding <= frame->nb_samples && *got_frame_ptr &&
2377 !(avctx->flags2 & AV_CODEC_FLAG2_SKIP_MANUAL)) {
2378 if (discard_padding == frame->nb_samples) {
2381 if(avctx->pkt_timebase.num && avctx->sample_rate) {
2382 int64_t diff_ts = av_rescale_q(frame->nb_samples - discard_padding,
2383 (AVRational){1, avctx->sample_rate},
2384 avctx->pkt_timebase);
2385 av_frame_set_pkt_duration(frame, diff_ts);
2387 av_log(avctx, AV_LOG_WARNING, "Could not update timestamps for discarded samples.\n");
2389 av_log(avctx, AV_LOG_DEBUG, "discard %d/%d samples\n",
2390 (int)discard_padding, frame->nb_samples);
2391 frame->nb_samples -= discard_padding;
2395 if ((avctx->flags2 & AV_CODEC_FLAG2_SKIP_MANUAL) && *got_frame_ptr) {
2396 AVFrameSideData *fside = av_frame_new_side_data(frame, AV_FRAME_DATA_SKIP_SAMPLES, 10);
2398 AV_WL32(fside->data, avctx->internal->skip_samples);
2399 AV_WL32(fside->data + 4, discard_padding);
2400 AV_WL8(fside->data + 8, skip_reason);
2401 AV_WL8(fside->data + 9, discard_reason);
2402 avctx->internal->skip_samples = 0;
2406 avctx->internal->pkt = NULL;
2408 av_packet_free_side_data(&tmp);
2413 if (ret >= 0 && *got_frame_ptr) {
2414 if (!avctx->refcounted_frames) {
2415 int err = unrefcount_frame(avci, frame);
2420 av_frame_unref(frame);
2426 #define UTF8_MAX_BYTES 4 /* 5 and 6 bytes sequences should not be used */
2427 static int recode_subtitle(AVCodecContext *avctx,
2428 AVPacket *outpkt, const AVPacket *inpkt)
2431 iconv_t cd = (iconv_t)-1;
2438 if (avctx->sub_charenc_mode != FF_SUB_CHARENC_MODE_PRE_DECODER || inpkt->size == 0)
2442 cd = iconv_open("UTF-8", avctx->sub_charenc);
2443 av_assert0(cd != (iconv_t)-1);
2448 if (inl >= INT_MAX / UTF8_MAX_BYTES - AV_INPUT_BUFFER_PADDING_SIZE) {
2449 av_log(avctx, AV_LOG_ERROR, "Subtitles packet is too big for recoding\n");
2450 ret = AVERROR(ENOMEM);
2454 ret = av_new_packet(&tmp, inl * UTF8_MAX_BYTES);
2457 outpkt->buf = tmp.buf;
2458 outpkt->data = tmp.data;
2459 outpkt->size = tmp.size;
2460 outb = outpkt->data;
2461 outl = outpkt->size;
2463 if (iconv(cd, &inb, &inl, &outb, &outl) == (size_t)-1 ||
2464 iconv(cd, NULL, NULL, &outb, &outl) == (size_t)-1 ||
2465 outl >= outpkt->size || inl != 0) {
2466 ret = FFMIN(AVERROR(errno), -1);
2467 av_log(avctx, AV_LOG_ERROR, "Unable to recode subtitle event \"%s\" "
2468 "from %s to UTF-8\n", inpkt->data, avctx->sub_charenc);
2469 av_packet_unref(&tmp);
2472 outpkt->size -= outl;
2473 memset(outpkt->data + outpkt->size, 0, outl);
2476 if (cd != (iconv_t)-1)
2480 av_log(avctx, AV_LOG_ERROR, "requesting subtitles recoding without iconv");
2481 return AVERROR(EINVAL);
2485 static int utf8_check(const uint8_t *str)
2487 const uint8_t *byte;
2488 uint32_t codepoint, min;
2492 GET_UTF8(codepoint, *(byte++), return 0;);
2493 min = byte - str == 1 ? 0 : byte - str == 2 ? 0x80 :
2494 1 << (5 * (byte - str) - 4);
2495 if (codepoint < min || codepoint >= 0x110000 ||
2496 codepoint == 0xFFFE /* BOM */ ||
2497 codepoint >= 0xD800 && codepoint <= 0xDFFF /* surrogates */)
2504 #if FF_API_ASS_TIMING
2505 static void insert_ts(AVBPrint *buf, int ts)
2508 av_bprintf(buf, "9:59:59.99,");
2512 h = ts/360000; ts -= 360000*h;
2513 m = ts/ 6000; ts -= 6000*m;
2514 s = ts/ 100; ts -= 100*s;
2515 av_bprintf(buf, "%d:%02d:%02d.%02d,", h, m, s, ts);
2519 static int convert_sub_to_old_ass_form(AVSubtitle *sub, const AVPacket *pkt, AVRational tb)
2524 av_bprint_init(&buf, 0, AV_BPRINT_SIZE_UNLIMITED);
2526 for (i = 0; i < sub->num_rects; i++) {
2529 AVSubtitleRect *rect = sub->rects[i];
2530 int ts_start, ts_duration = -1;
2533 if (rect->type != SUBTITLE_ASS || !strncmp(rect->ass, "Dialogue: ", 10))
2536 av_bprint_clear(&buf);
2538 /* skip ReadOrder */
2539 dialog = strchr(rect->ass, ',');
2544 /* extract Layer or Marked */
2545 layer = strtol(dialog, (char**)&dialog, 10);
2550 /* rescale timing to ASS time base (ms) */
2551 ts_start = av_rescale_q(pkt->pts, tb, av_make_q(1, 100));
2552 if (pkt->duration != -1)
2553 ts_duration = av_rescale_q(pkt->duration, tb, av_make_q(1, 100));
2554 sub->end_display_time = FFMAX(sub->end_display_time, 10 * ts_duration);
2556 /* construct ASS (standalone file form with timestamps) string */
2557 av_bprintf(&buf, "Dialogue: %ld,", layer);
2558 insert_ts(&buf, ts_start);
2559 insert_ts(&buf, ts_duration == -1 ? -1 : ts_start + ts_duration);
2560 av_bprintf(&buf, "%s\r\n", dialog);
2562 final_dialog = av_strdup(buf.str);
2563 if (!av_bprint_is_complete(&buf) || !final_dialog) {
2564 av_freep(&final_dialog);
2565 av_bprint_finalize(&buf, NULL);
2566 return AVERROR(ENOMEM);
2568 av_freep(&rect->ass);
2569 rect->ass = final_dialog;
2572 av_bprint_finalize(&buf, NULL);
2577 int avcodec_decode_subtitle2(AVCodecContext *avctx, AVSubtitle *sub,
2583 if (!avpkt->data && avpkt->size) {
2584 av_log(avctx, AV_LOG_ERROR, "invalid packet: NULL data, size != 0\n");
2585 return AVERROR(EINVAL);
2588 return AVERROR(EINVAL);
2589 if (avctx->codec->type != AVMEDIA_TYPE_SUBTITLE) {
2590 av_log(avctx, AV_LOG_ERROR, "Invalid media type for subtitles\n");
2591 return AVERROR(EINVAL);
2595 get_subtitle_defaults(sub);
2597 if ((avctx->codec->capabilities & AV_CODEC_CAP_DELAY) || avpkt->size) {
2598 AVPacket pkt_recoded;
2599 AVPacket tmp = *avpkt;
2600 int did_split = av_packet_split_side_data(&tmp);
2601 //apply_param_change(avctx, &tmp);
2604 /* FFMIN() prevents overflow in case the packet wasn't allocated with
2606 * If the side data is smaller than the buffer padding size, the
2607 * remaining bytes should have already been filled with zeros by the
2608 * original packet allocation anyway. */
2609 memset(tmp.data + tmp.size, 0,
2610 FFMIN(avpkt->size - tmp.size, AV_INPUT_BUFFER_PADDING_SIZE));
2614 ret = recode_subtitle(avctx, &pkt_recoded, &tmp);
2618 avctx->internal->pkt = &pkt_recoded;
2620 if (avctx->pkt_timebase.num && avpkt->pts != AV_NOPTS_VALUE)
2621 sub->pts = av_rescale_q(avpkt->pts,
2622 avctx->pkt_timebase, AV_TIME_BASE_Q);
2623 ret = avctx->codec->decode(avctx, sub, got_sub_ptr, &pkt_recoded);
2624 av_assert1((ret >= 0) >= !!*got_sub_ptr &&
2625 !!*got_sub_ptr >= !!sub->num_rects);
2627 #if FF_API_ASS_TIMING
2628 if (avctx->sub_text_format == FF_SUB_TEXT_FMT_ASS_WITH_TIMINGS
2629 && *got_sub_ptr && sub->num_rects) {
2630 const AVRational tb = avctx->pkt_timebase.num ? avctx->pkt_timebase
2632 ret = convert_sub_to_old_ass_form(sub, avpkt, tb);
2636 if (sub->num_rects && !sub->end_display_time && avpkt->duration &&
2637 avctx->pkt_timebase.num) {
2638 AVRational ms = { 1, 1000 };
2639 sub->end_display_time = av_rescale_q(avpkt->duration,
2640 avctx->pkt_timebase, ms);
2643 for (i = 0; i < sub->num_rects; i++) {
2644 if (sub->rects[i]->ass && !utf8_check(sub->rects[i]->ass)) {
2645 av_log(avctx, AV_LOG_ERROR,
2646 "Invalid UTF-8 in decoded subtitles text; "
2647 "maybe missing -sub_charenc option\n");
2648 avsubtitle_free(sub);
2649 return AVERROR_INVALIDDATA;
2653 if (tmp.data != pkt_recoded.data) { // did we recode?
2654 /* prevent from destroying side data from original packet */
2655 pkt_recoded.side_data = NULL;
2656 pkt_recoded.side_data_elems = 0;
2658 av_packet_unref(&pkt_recoded);
2660 if (avctx->codec_descriptor->props & AV_CODEC_PROP_BITMAP_SUB)
2662 else if (avctx->codec_descriptor->props & AV_CODEC_PROP_TEXT_SUB)
2664 avctx->internal->pkt = NULL;
2668 av_packet_free_side_data(&tmp);
2674 avctx->frame_number++;
2680 void avsubtitle_free(AVSubtitle *sub)
2684 for (i = 0; i < sub->num_rects; i++) {
2685 av_freep(&sub->rects[i]->data[0]);
2686 av_freep(&sub->rects[i]->data[1]);
2687 av_freep(&sub->rects[i]->data[2]);
2688 av_freep(&sub->rects[i]->data[3]);
2689 av_freep(&sub->rects[i]->text);
2690 av_freep(&sub->rects[i]->ass);
2691 av_freep(&sub->rects[i]);
2694 av_freep(&sub->rects);
2696 memset(sub, 0, sizeof(AVSubtitle));
2699 static int do_decode(AVCodecContext *avctx, AVPacket *pkt)
2704 av_assert0(!avctx->internal->buffer_frame->buf[0]);
2707 pkt = avctx->internal->buffer_pkt;
2709 // This is the lesser evil. The field is for compatibility with legacy users
2710 // of the legacy API, and users using the new API should not be forced to
2711 // even know about this field.
2712 avctx->refcounted_frames = 1;
2714 // Some codecs (at least wma lossless) will crash when feeding drain packets
2715 // after EOF was signaled.
2716 if (avctx->internal->draining_done)
2719 if (avctx->codec_type == AVMEDIA_TYPE_VIDEO) {
2720 ret = avcodec_decode_video2(avctx, avctx->internal->buffer_frame,
2724 } else if (avctx->codec_type == AVMEDIA_TYPE_AUDIO) {
2725 ret = avcodec_decode_audio4(avctx, avctx->internal->buffer_frame,
2728 ret = AVERROR(EINVAL);
2731 if (ret == AVERROR(EAGAIN))
2737 if (avctx->internal->draining && !got_frame)
2738 avctx->internal->draining_done = 1;
2740 if (ret >= pkt->size) {
2741 av_packet_unref(avctx->internal->buffer_pkt);
2745 if (pkt != avctx->internal->buffer_pkt) {
2746 av_packet_unref(avctx->internal->buffer_pkt);
2747 if ((ret = av_packet_ref(avctx->internal->buffer_pkt, pkt)) < 0)
2751 avctx->internal->buffer_pkt->data += consumed;
2752 avctx->internal->buffer_pkt->size -= consumed;
2753 avctx->internal->buffer_pkt->pts = AV_NOPTS_VALUE;
2754 avctx->internal->buffer_pkt->dts = AV_NOPTS_VALUE;
2758 av_assert0(avctx->internal->buffer_frame->buf[0]);
2763 int attribute_align_arg avcodec_send_packet(AVCodecContext *avctx, const AVPacket *avpkt)
2767 if (!avcodec_is_open(avctx) || !av_codec_is_decoder(avctx->codec))
2768 return AVERROR(EINVAL);
2770 if (avctx->internal->draining)
2773 if (!avpkt || !avpkt->size) {
2774 avctx->internal->draining = 1;
2777 if (!(avctx->codec->capabilities & AV_CODEC_CAP_DELAY))
2781 if (avctx->codec->send_packet) {
2783 ret = apply_param_change(avctx, (AVPacket *)avpkt);
2787 return avctx->codec->send_packet(avctx, avpkt);
2790 // Emulation via old API. Assume avpkt is likely not refcounted, while
2791 // decoder output is always refcounted, and avoid copying.
2793 if (avctx->internal->buffer_pkt->size || avctx->internal->buffer_frame->buf[0])
2794 return AVERROR(EAGAIN);
2796 // The goal is decoding the first frame of the packet without using memcpy,
2797 // because the common case is having only 1 frame per packet (especially
2798 // with video, but audio too). In other cases, it can't be avoided, unless
2799 // the user is feeding refcounted packets.
2800 return do_decode(avctx, (AVPacket *)avpkt);
2803 int attribute_align_arg avcodec_receive_frame(AVCodecContext *avctx, AVFrame *frame)
2807 av_frame_unref(frame);
2809 if (!avcodec_is_open(avctx) || !av_codec_is_decoder(avctx->codec))
2810 return AVERROR(EINVAL);
2812 if (avctx->codec->receive_frame) {
2813 if (avctx->internal->draining && !(avctx->codec->capabilities & AV_CODEC_CAP_DELAY))
2815 return avctx->codec->receive_frame(avctx, frame);
2818 // Emulation via old API.
2820 if (!avctx->internal->buffer_frame->buf[0]) {
2821 if (!avctx->internal->buffer_pkt->size && !avctx->internal->draining)
2822 return AVERROR(EAGAIN);
2825 if ((ret = do_decode(avctx, avctx->internal->buffer_pkt)) < 0) {
2826 av_packet_unref(avctx->internal->buffer_pkt);
2829 // Some audio decoders may consume partial data without returning
2830 // a frame (fate-wmapro-2ch). There is no way to make the caller
2831 // call avcodec_receive_frame() again without returning a frame,
2832 // so try to decode more in these cases.
2833 if (avctx->internal->buffer_frame->buf[0] ||
2834 !avctx->internal->buffer_pkt->size)
2839 if (!avctx->internal->buffer_frame->buf[0])
2840 return avctx->internal->draining ? AVERROR_EOF : AVERROR(EAGAIN);
2842 av_frame_move_ref(frame, avctx->internal->buffer_frame);
2846 static int do_encode(AVCodecContext *avctx, const AVFrame *frame, int *got_packet)
2851 av_packet_unref(avctx->internal->buffer_pkt);
2852 avctx->internal->buffer_pkt_valid = 0;
2854 if (avctx->codec_type == AVMEDIA_TYPE_VIDEO) {
2855 ret = avcodec_encode_video2(avctx, avctx->internal->buffer_pkt,
2857 } else if (avctx->codec_type == AVMEDIA_TYPE_AUDIO) {
2858 ret = avcodec_encode_audio2(avctx, avctx->internal->buffer_pkt,
2861 ret = AVERROR(EINVAL);
2864 if (ret >= 0 && *got_packet) {
2865 // Encoders must always return ref-counted buffers.
2866 // Side-data only packets have no data and can be not ref-counted.
2867 av_assert0(!avctx->internal->buffer_pkt->data || avctx->internal->buffer_pkt->buf);
2868 avctx->internal->buffer_pkt_valid = 1;
2871 av_packet_unref(avctx->internal->buffer_pkt);
2877 int attribute_align_arg avcodec_send_frame(AVCodecContext *avctx, const AVFrame *frame)
2879 if (!avcodec_is_open(avctx) || !av_codec_is_encoder(avctx->codec))
2880 return AVERROR(EINVAL);
2882 if (avctx->internal->draining)
2886 avctx->internal->draining = 1;
2888 if (!(avctx->codec->capabilities & AV_CODEC_CAP_DELAY))
2892 if (avctx->codec->send_frame)
2893 return avctx->codec->send_frame(avctx, frame);
2895 // Emulation via old API. Do it here instead of avcodec_receive_packet, because:
2896 // 1. if the AVFrame is not refcounted, the copying will be much more
2897 // expensive than copying the packet data
2898 // 2. assume few users use non-refcounted AVPackets, so usually no copy is
2901 if (avctx->internal->buffer_pkt_valid)
2902 return AVERROR(EAGAIN);
2904 return do_encode(avctx, frame, &(int){0});
2907 int attribute_align_arg avcodec_receive_packet(AVCodecContext *avctx, AVPacket *avpkt)
2909 av_packet_unref(avpkt);
2911 if (!avcodec_is_open(avctx) || !av_codec_is_encoder(avctx->codec))
2912 return AVERROR(EINVAL);
2914 if (avctx->codec->receive_packet) {
2915 if (avctx->internal->draining && !(avctx->codec->capabilities & AV_CODEC_CAP_DELAY))
2917 return avctx->codec->receive_packet(avctx, avpkt);
2920 // Emulation via old API.
2922 if (!avctx->internal->buffer_pkt_valid) {
2925 if (!avctx->internal->draining)
2926 return AVERROR(EAGAIN);
2927 ret = do_encode(avctx, NULL, &got_packet);
2930 if (ret >= 0 && !got_packet)
2934 av_packet_move_ref(avpkt, avctx->internal->buffer_pkt);
2935 avctx->internal->buffer_pkt_valid = 0;
2939 av_cold int avcodec_close(AVCodecContext *avctx)
2946 if (avcodec_is_open(avctx)) {
2947 FramePool *pool = avctx->internal->pool;
2948 if (CONFIG_FRAME_THREAD_ENCODER &&
2949 avctx->internal->frame_thread_encoder && avctx->thread_count > 1) {
2950 ff_frame_thread_encoder_free(avctx);
2952 if (HAVE_THREADS && avctx->internal->thread_ctx)
2953 ff_thread_free(avctx);
2954 if (avctx->codec && avctx->codec->close)
2955 avctx->codec->close(avctx);
2956 avctx->internal->byte_buffer_size = 0;
2957 av_freep(&avctx->internal->byte_buffer);
2958 av_frame_free(&avctx->internal->to_free);
2959 av_frame_free(&avctx->internal->buffer_frame);
2960 av_packet_free(&avctx->internal->buffer_pkt);
2961 for (i = 0; i < FF_ARRAY_ELEMS(pool->pools); i++)
2962 av_buffer_pool_uninit(&pool->pools[i]);
2963 av_freep(&avctx->internal->pool);
2965 if (avctx->hwaccel && avctx->hwaccel->uninit)
2966 avctx->hwaccel->uninit(avctx);
2967 av_freep(&avctx->internal->hwaccel_priv_data);
2969 av_freep(&avctx->internal);
2972 for (i = 0; i < avctx->nb_coded_side_data; i++)
2973 av_freep(&avctx->coded_side_data[i].data);
2974 av_freep(&avctx->coded_side_data);
2975 avctx->nb_coded_side_data = 0;
2977 av_buffer_unref(&avctx->hw_frames_ctx);
2979 if (avctx->priv_data && avctx->codec && avctx->codec->priv_class)
2980 av_opt_free(avctx->priv_data);
2982 av_freep(&avctx->priv_data);
2983 if (av_codec_is_encoder(avctx->codec)) {
2984 av_freep(&avctx->extradata);
2985 #if FF_API_CODED_FRAME
2986 FF_DISABLE_DEPRECATION_WARNINGS
2987 av_frame_free(&avctx->coded_frame);
2988 FF_ENABLE_DEPRECATION_WARNINGS
2991 avctx->codec = NULL;
2992 avctx->active_thread_type = 0;
2997 static enum AVCodecID remap_deprecated_codec_id(enum AVCodecID id)
3000 //This is for future deprecatec codec ids, its empty since
3001 //last major bump but will fill up again over time, please don't remove it
3002 default : return id;
3006 static AVCodec *find_encdec(enum AVCodecID id, int encoder)
3008 AVCodec *p, *experimental = NULL;
3010 id= remap_deprecated_codec_id(id);
3012 if ((encoder ? av_codec_is_encoder(p) : av_codec_is_decoder(p)) &&
3014 if (p->capabilities & AV_CODEC_CAP_EXPERIMENTAL && !experimental) {
3021 return experimental;
3024 AVCodec *avcodec_find_encoder(enum AVCodecID id)
3026 return find_encdec(id, 1);
3029 AVCodec *avcodec_find_encoder_by_name(const char *name)
3036 if (av_codec_is_encoder(p) && strcmp(name, p->name) == 0)
3043 AVCodec *avcodec_find_decoder(enum AVCodecID id)
3045 return find_encdec(id, 0);
3048 AVCodec *avcodec_find_decoder_by_name(const char *name)
3055 if (av_codec_is_decoder(p) && strcmp(name, p->name) == 0)
3062 const char *avcodec_get_name(enum AVCodecID id)
3064 const AVCodecDescriptor *cd;
3067 if (id == AV_CODEC_ID_NONE)
3069 cd = avcodec_descriptor_get(id);
3072 av_log(NULL, AV_LOG_WARNING, "Codec 0x%x is not in the full list.\n", id);
3073 codec = avcodec_find_decoder(id);
3076 codec = avcodec_find_encoder(id);
3079 return "unknown_codec";
3082 size_t av_get_codec_tag_string(char *buf, size_t buf_size, unsigned int codec_tag)
3084 int i, len, ret = 0;
3086 #define TAG_PRINT(x) \
3087 (((x) >= '0' && (x) <= '9') || \
3088 ((x) >= 'a' && (x) <= 'z') || ((x) >= 'A' && (x) <= 'Z') || \
3089 ((x) == '.' || (x) == ' ' || (x) == '-' || (x) == '_'))
3091 for (i = 0; i < 4; i++) {
3092 len = snprintf(buf, buf_size,
3093 TAG_PRINT(codec_tag & 0xFF) ? "%c" : "[%d]", codec_tag & 0xFF);
3095 buf_size = buf_size > len ? buf_size - len : 0;
3102 void avcodec_string(char *buf, int buf_size, AVCodecContext *enc, int encode)
3104 const char *codec_type;
3105 const char *codec_name;
3106 const char *profile = NULL;
3109 AVRational display_aspect_ratio;
3110 const char *separator = enc->dump_separator ? (const char *)enc->dump_separator : ", ";
3112 if (!buf || buf_size <= 0)
3114 codec_type = av_get_media_type_string(enc->codec_type);
3115 codec_name = avcodec_get_name(enc->codec_id);
3116 profile = avcodec_profile_name(enc->codec_id, enc->profile);
3118 snprintf(buf, buf_size, "%s: %s", codec_type ? codec_type : "unknown",
3120 buf[0] ^= 'a' ^ 'A'; /* first letter in uppercase */
3122 if (enc->codec && strcmp(enc->codec->name, codec_name))
3123 snprintf(buf + strlen(buf), buf_size - strlen(buf), " (%s)", enc->codec->name);
3126 snprintf(buf + strlen(buf), buf_size - strlen(buf), " (%s)", profile);
3127 if ( enc->codec_type == AVMEDIA_TYPE_VIDEO
3128 && av_log_get_level() >= AV_LOG_VERBOSE
3130 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3131 ", %d reference frame%s",
3132 enc->refs, enc->refs > 1 ? "s" : "");
3134 if (enc->codec_tag) {
3136 av_get_codec_tag_string(tag_buf, sizeof(tag_buf), enc->codec_tag);
3137 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3138 " (%s / 0x%04X)", tag_buf, enc->codec_tag);
3141 switch (enc->codec_type) {
3142 case AVMEDIA_TYPE_VIDEO:
3144 char detail[256] = "(";
3146 av_strlcat(buf, separator, buf_size);
3148 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3149 "%s", enc->pix_fmt == AV_PIX_FMT_NONE ? "none" :
3150 av_get_pix_fmt_name(enc->pix_fmt));
3151 if (enc->bits_per_raw_sample && enc->pix_fmt != AV_PIX_FMT_NONE &&
3152 enc->bits_per_raw_sample < av_pix_fmt_desc_get(enc->pix_fmt)->comp[0].depth)
3153 av_strlcatf(detail, sizeof(detail), "%d bpc, ", enc->bits_per_raw_sample);
3154 if (enc->color_range != AVCOL_RANGE_UNSPECIFIED)
3155 av_strlcatf(detail, sizeof(detail), "%s, ",
3156 av_color_range_name(enc->color_range));
3158 if (enc->colorspace != AVCOL_SPC_UNSPECIFIED ||
3159 enc->color_primaries != AVCOL_PRI_UNSPECIFIED ||
3160 enc->color_trc != AVCOL_TRC_UNSPECIFIED) {
3161 if (enc->colorspace != (int)enc->color_primaries ||
3162 enc->colorspace != (int)enc->color_trc) {
3164 av_strlcatf(detail, sizeof(detail), "%s/%s/%s, ",
3165 av_color_space_name(enc->colorspace),
3166 av_color_primaries_name(enc->color_primaries),
3167 av_color_transfer_name(enc->color_trc));
3169 av_strlcatf(detail, sizeof(detail), "%s, ",
3170 av_get_colorspace_name(enc->colorspace));
3173 if (av_log_get_level() >= AV_LOG_DEBUG &&
3174 enc->chroma_sample_location != AVCHROMA_LOC_UNSPECIFIED)
3175 av_strlcatf(detail, sizeof(detail), "%s, ",
3176 av_chroma_location_name(enc->chroma_sample_location));
3178 if (strlen(detail) > 1) {
3179 detail[strlen(detail) - 2] = 0;
3180 av_strlcatf(buf, buf_size, "%s)", detail);
3185 av_strlcat(buf, new_line ? separator : ", ", buf_size);
3187 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3189 enc->width, enc->height);
3191 if (av_log_get_level() >= AV_LOG_VERBOSE &&
3192 (enc->width != enc->coded_width ||
3193 enc->height != enc->coded_height))
3194 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3195 " (%dx%d)", enc->coded_width, enc->coded_height);
3197 if (enc->sample_aspect_ratio.num) {
3198 av_reduce(&display_aspect_ratio.num, &display_aspect_ratio.den,
3199 enc->width * (int64_t)enc->sample_aspect_ratio.num,
3200 enc->height * (int64_t)enc->sample_aspect_ratio.den,
3202 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3203 " [SAR %d:%d DAR %d:%d]",
3204 enc->sample_aspect_ratio.num, enc->sample_aspect_ratio.den,
3205 display_aspect_ratio.num, display_aspect_ratio.den);
3207 if (av_log_get_level() >= AV_LOG_DEBUG) {
3208 int g = av_gcd(enc->time_base.num, enc->time_base.den);
3209 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3211 enc->time_base.num / g, enc->time_base.den / g);
3215 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3216 ", q=%d-%d", enc->qmin, enc->qmax);
3218 if (enc->properties & FF_CODEC_PROPERTY_CLOSED_CAPTIONS)
3219 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3220 ", Closed Captions");
3221 if (enc->properties & FF_CODEC_PROPERTY_LOSSLESS)
3222 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3226 case AVMEDIA_TYPE_AUDIO:
3227 av_strlcat(buf, separator, buf_size);
3229 if (enc->sample_rate) {
3230 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3231 "%d Hz, ", enc->sample_rate);
3233 av_get_channel_layout_string(buf + strlen(buf), buf_size - strlen(buf), enc->channels, enc->channel_layout);
3234 if (enc->sample_fmt != AV_SAMPLE_FMT_NONE) {
3235 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3236 ", %s", av_get_sample_fmt_name(enc->sample_fmt));
3238 if ( enc->bits_per_raw_sample > 0
3239 && enc->bits_per_raw_sample != av_get_bytes_per_sample(enc->sample_fmt) * 8)
3240 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3241 " (%d bit)", enc->bits_per_raw_sample);
3243 case AVMEDIA_TYPE_DATA:
3244 if (av_log_get_level() >= AV_LOG_DEBUG) {
3245 int g = av_gcd(enc->time_base.num, enc->time_base.den);
3247 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3249 enc->time_base.num / g, enc->time_base.den / g);
3252 case AVMEDIA_TYPE_SUBTITLE:
3254 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3255 ", %dx%d", enc->width, enc->height);
3261 if (enc->flags & AV_CODEC_FLAG_PASS1)
3262 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3264 if (enc->flags & AV_CODEC_FLAG_PASS2)
3265 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3268 bitrate = get_bit_rate(enc);
3270 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3271 ", %"PRId64" kb/s", bitrate / 1000);
3272 } else if (enc->rc_max_rate > 0) {
3273 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3274 ", max. %"PRId64" kb/s", (int64_t)enc->rc_max_rate / 1000);
3278 const char *av_get_profile_name(const AVCodec *codec, int profile)
3281 if (profile == FF_PROFILE_UNKNOWN || !codec->profiles)
3284 for (p = codec->profiles; p->profile != FF_PROFILE_UNKNOWN; p++)
3285 if (p->profile == profile)
3291 const char *avcodec_profile_name(enum AVCodecID codec_id, int profile)
3293 const AVCodecDescriptor *desc = avcodec_descriptor_get(codec_id);
3296 if (profile == FF_PROFILE_UNKNOWN || !desc || !desc->profiles)
3299 for (p = desc->profiles; p->profile != FF_PROFILE_UNKNOWN; p++)
3300 if (p->profile == profile)
3306 unsigned avcodec_version(void)
3308 // av_assert0(AV_CODEC_ID_V410==164);
3309 av_assert0(AV_CODEC_ID_PCM_S8_PLANAR==65563);
3310 av_assert0(AV_CODEC_ID_ADPCM_G722==69660);
3311 // av_assert0(AV_CODEC_ID_BMV_AUDIO==86071);
3312 av_assert0(AV_CODEC_ID_SRT==94216);
3313 av_assert0(LIBAVCODEC_VERSION_MICRO >= 100);
3315 return LIBAVCODEC_VERSION_INT;
3318 const char *avcodec_configuration(void)
3320 return FFMPEG_CONFIGURATION;
3323 const char *avcodec_license(void)
3325 #define LICENSE_PREFIX "libavcodec license: "
3326 return LICENSE_PREFIX FFMPEG_LICENSE + sizeof(LICENSE_PREFIX) - 1;
3329 void avcodec_flush_buffers(AVCodecContext *avctx)
3331 avctx->internal->draining = 0;
3332 avctx->internal->draining_done = 0;
3333 av_frame_unref(avctx->internal->buffer_frame);
3334 av_packet_unref(avctx->internal->buffer_pkt);
3335 avctx->internal->buffer_pkt_valid = 0;
3337 if (HAVE_THREADS && avctx->active_thread_type & FF_THREAD_FRAME)
3338 ff_thread_flush(avctx);
3339 else if (avctx->codec->flush)
3340 avctx->codec->flush(avctx);
3342 avctx->pts_correction_last_pts =
3343 avctx->pts_correction_last_dts = INT64_MIN;
3345 if (!avctx->refcounted_frames)
3346 av_frame_unref(avctx->internal->to_free);
3349 int av_get_exact_bits_per_sample(enum AVCodecID codec_id)
3352 case AV_CODEC_ID_8SVX_EXP:
3353 case AV_CODEC_ID_8SVX_FIB:
3354 case AV_CODEC_ID_ADPCM_CT:
3355 case AV_CODEC_ID_ADPCM_IMA_APC:
3356 case AV_CODEC_ID_ADPCM_IMA_EA_SEAD:
3357 case AV_CODEC_ID_ADPCM_IMA_OKI:
3358 case AV_CODEC_ID_ADPCM_IMA_WS:
3359 case AV_CODEC_ID_ADPCM_G722:
3360 case AV_CODEC_ID_ADPCM_YAMAHA:
3361 case AV_CODEC_ID_ADPCM_AICA:
3363 case AV_CODEC_ID_DSD_LSBF:
3364 case AV_CODEC_ID_DSD_MSBF:
3365 case AV_CODEC_ID_DSD_LSBF_PLANAR:
3366 case AV_CODEC_ID_DSD_MSBF_PLANAR:
3367 case AV_CODEC_ID_PCM_ALAW:
3368 case AV_CODEC_ID_PCM_MULAW:
3369 case AV_CODEC_ID_PCM_S8:
3370 case AV_CODEC_ID_PCM_S8_PLANAR:
3371 case AV_CODEC_ID_PCM_U8:
3372 case AV_CODEC_ID_PCM_ZORK:
3373 case AV_CODEC_ID_SDX2_DPCM:
3375 case AV_CODEC_ID_PCM_S16BE:
3376 case AV_CODEC_ID_PCM_S16BE_PLANAR:
3377 case AV_CODEC_ID_PCM_S16LE:
3378 case AV_CODEC_ID_PCM_S16LE_PLANAR:
3379 case AV_CODEC_ID_PCM_U16BE:
3380 case AV_CODEC_ID_PCM_U16LE:
3382 case AV_CODEC_ID_PCM_S24DAUD:
3383 case AV_CODEC_ID_PCM_S24BE:
3384 case AV_CODEC_ID_PCM_S24LE:
3385 case AV_CODEC_ID_PCM_S24LE_PLANAR:
3386 case AV_CODEC_ID_PCM_U24BE:
3387 case AV_CODEC_ID_PCM_U24LE:
3389 case AV_CODEC_ID_PCM_S32BE:
3390 case AV_CODEC_ID_PCM_S32LE:
3391 case AV_CODEC_ID_PCM_S32LE_PLANAR:
3392 case AV_CODEC_ID_PCM_U32BE:
3393 case AV_CODEC_ID_PCM_U32LE:
3394 case AV_CODEC_ID_PCM_F32BE:
3395 case AV_CODEC_ID_PCM_F32LE:
3397 case AV_CODEC_ID_PCM_F64BE:
3398 case AV_CODEC_ID_PCM_F64LE:
3405 enum AVCodecID av_get_pcm_codec(enum AVSampleFormat fmt, int be)
3407 static const enum AVCodecID map[AV_SAMPLE_FMT_NB][2] = {
3408 [AV_SAMPLE_FMT_U8 ] = { AV_CODEC_ID_PCM_U8, AV_CODEC_ID_PCM_U8 },
3409 [AV_SAMPLE_FMT_S16 ] = { AV_CODEC_ID_PCM_S16LE, AV_CODEC_ID_PCM_S16BE },
3410 [AV_SAMPLE_FMT_S32 ] = { AV_CODEC_ID_PCM_S32LE, AV_CODEC_ID_PCM_S32BE },
3411 [AV_SAMPLE_FMT_FLT ] = { AV_CODEC_ID_PCM_F32LE, AV_CODEC_ID_PCM_F32BE },
3412 [AV_SAMPLE_FMT_DBL ] = { AV_CODEC_ID_PCM_F64LE, AV_CODEC_ID_PCM_F64BE },
3413 [AV_SAMPLE_FMT_U8P ] = { AV_CODEC_ID_PCM_U8, AV_CODEC_ID_PCM_U8 },
3414 [AV_SAMPLE_FMT_S16P] = { AV_CODEC_ID_PCM_S16LE, AV_CODEC_ID_PCM_S16BE },
3415 [AV_SAMPLE_FMT_S32P] = { AV_CODEC_ID_PCM_S32LE, AV_CODEC_ID_PCM_S32BE },
3416 [AV_SAMPLE_FMT_FLTP] = { AV_CODEC_ID_PCM_F32LE, AV_CODEC_ID_PCM_F32BE },
3417 [AV_SAMPLE_FMT_DBLP] = { AV_CODEC_ID_PCM_F64LE, AV_CODEC_ID_PCM_F64BE },
3419 if (fmt < 0 || fmt >= AV_SAMPLE_FMT_NB)
3420 return AV_CODEC_ID_NONE;
3421 if (be < 0 || be > 1)
3423 return map[fmt][be];
3426 int av_get_bits_per_sample(enum AVCodecID codec_id)
3429 case AV_CODEC_ID_ADPCM_SBPRO_2:
3431 case AV_CODEC_ID_ADPCM_SBPRO_3:
3433 case AV_CODEC_ID_ADPCM_SBPRO_4:
3434 case AV_CODEC_ID_ADPCM_IMA_WAV:
3435 case AV_CODEC_ID_ADPCM_IMA_QT:
3436 case AV_CODEC_ID_ADPCM_SWF:
3437 case AV_CODEC_ID_ADPCM_MS:
3440 return av_get_exact_bits_per_sample(codec_id);
3444 static int get_audio_frame_duration(enum AVCodecID id, int sr, int ch, int ba,
3445 uint32_t tag, int bits_per_coded_sample, int64_t bitrate,
3446 uint8_t * extradata, int frame_size, int frame_bytes)
3448 int bps = av_get_exact_bits_per_sample(id);
3449 int framecount = (ba > 0 && frame_bytes / ba > 0) ? frame_bytes / ba : 1;
3451 /* codecs with an exact constant bits per sample */
3452 if (bps > 0 && ch > 0 && frame_bytes > 0 && ch < 32768 && bps < 32768)
3453 return (frame_bytes * 8LL) / (bps * ch);
3454 bps = bits_per_coded_sample;
3456 /* codecs with a fixed packet duration */
3458 case AV_CODEC_ID_ADPCM_ADX: return 32;
3459 case AV_CODEC_ID_ADPCM_IMA_QT: return 64;
3460 case AV_CODEC_ID_ADPCM_EA_XAS: return 128;
3461 case AV_CODEC_ID_AMR_NB:
3462 case AV_CODEC_ID_EVRC:
3463 case AV_CODEC_ID_GSM:
3464 case AV_CODEC_ID_QCELP:
3465 case AV_CODEC_ID_RA_288: return 160;
3466 case AV_CODEC_ID_AMR_WB:
3467 case AV_CODEC_ID_GSM_MS: return 320;
3468 case AV_CODEC_ID_MP1: return 384;
3469 case AV_CODEC_ID_ATRAC1: return 512;
3470 case AV_CODEC_ID_ATRAC3: return 1024 * framecount;
3471 case AV_CODEC_ID_ATRAC3P: return 2048;
3472 case AV_CODEC_ID_MP2:
3473 case AV_CODEC_ID_MUSEPACK7: return 1152;
3474 case AV_CODEC_ID_AC3: return 1536;
3478 /* calc from sample rate */
3479 if (id == AV_CODEC_ID_TTA)
3480 return 256 * sr / 245;
3483 /* calc from sample rate and channels */
3484 if (id == AV_CODEC_ID_BINKAUDIO_DCT)
3485 return (480 << (sr / 22050)) / ch;
3490 /* calc from block_align */
3491 if (id == AV_CODEC_ID_SIPR) {
3493 case 20: return 160;
3494 case 19: return 144;
3495 case 29: return 288;
3496 case 37: return 480;
3498 } else if (id == AV_CODEC_ID_ILBC) {
3500 case 38: return 160;
3501 case 50: return 240;
3506 if (frame_bytes > 0) {
3507 /* calc from frame_bytes only */
3508 if (id == AV_CODEC_ID_TRUESPEECH)
3509 return 240 * (frame_bytes / 32);
3510 if (id == AV_CODEC_ID_NELLYMOSER)
3511 return 256 * (frame_bytes / 64);
3512 if (id == AV_CODEC_ID_RA_144)
3513 return 160 * (frame_bytes / 20);
3514 if (id == AV_CODEC_ID_G723_1)
3515 return 240 * (frame_bytes / 24);
3518 /* calc from frame_bytes and bits_per_coded_sample */
3519 if (id == AV_CODEC_ID_ADPCM_G726)
3520 return frame_bytes * 8 / bps;
3523 if (ch > 0 && ch < INT_MAX/16) {
3524 /* calc from frame_bytes and channels */
3526 case AV_CODEC_ID_ADPCM_AFC:
3527 return frame_bytes / (9 * ch) * 16;
3528 case AV_CODEC_ID_ADPCM_PSX:
3529 case AV_CODEC_ID_ADPCM_DTK:
3530 return frame_bytes / (16 * ch) * 28;
3531 case AV_CODEC_ID_ADPCM_4XM:
3532 case AV_CODEC_ID_ADPCM_IMA_DAT4:
3533 case AV_CODEC_ID_ADPCM_IMA_ISS:
3534 return (frame_bytes - 4 * ch) * 2 / ch;
3535 case AV_CODEC_ID_ADPCM_IMA_SMJPEG:
3536 return (frame_bytes - 4) * 2 / ch;
3537 case AV_CODEC_ID_ADPCM_IMA_AMV:
3538 return (frame_bytes - 8) * 2 / ch;
3539 case AV_CODEC_ID_ADPCM_THP:
3540 case AV_CODEC_ID_ADPCM_THP_LE:
3542 return frame_bytes * 14 / (8 * ch);
3544 case AV_CODEC_ID_ADPCM_XA:
3545 return (frame_bytes / 128) * 224 / ch;
3546 case AV_CODEC_ID_INTERPLAY_DPCM:
3547 return (frame_bytes - 6 - ch) / ch;
3548 case AV_CODEC_ID_ROQ_DPCM:
3549 return (frame_bytes - 8) / ch;
3550 case AV_CODEC_ID_XAN_DPCM:
3551 return (frame_bytes - 2 * ch) / ch;
3552 case AV_CODEC_ID_MACE3:
3553 return 3 * frame_bytes / ch;
3554 case AV_CODEC_ID_MACE6:
3555 return 6 * frame_bytes / ch;
3556 case AV_CODEC_ID_PCM_LXF:
3557 return 2 * (frame_bytes / (5 * ch));
3558 case AV_CODEC_ID_IAC:
3559 case AV_CODEC_ID_IMC:
3560 return 4 * frame_bytes / ch;
3564 /* calc from frame_bytes, channels, and codec_tag */
3565 if (id == AV_CODEC_ID_SOL_DPCM) {
3567 return frame_bytes / ch;
3569 return frame_bytes * 2 / ch;
3574 /* calc from frame_bytes, channels, and block_align */
3575 int blocks = frame_bytes / ba;
3577 case AV_CODEC_ID_ADPCM_IMA_WAV:
3578 if (bps < 2 || bps > 5)
3580 return blocks * (1 + (ba - 4 * ch) / (bps * ch) * 8);
3581 case AV_CODEC_ID_ADPCM_IMA_DK3:
3582 return blocks * (((ba - 16) * 2 / 3 * 4) / ch);
3583 case AV_CODEC_ID_ADPCM_IMA_DK4:
3584 return blocks * (1 + (ba - 4 * ch) * 2 / ch);
3585 case AV_CODEC_ID_ADPCM_IMA_RAD:
3586 return blocks * ((ba - 4 * ch) * 2 / ch);
3587 case AV_CODEC_ID_ADPCM_MS:
3588 return blocks * (2 + (ba - 7 * ch) * 2 / ch);
3593 /* calc from frame_bytes, channels, and bits_per_coded_sample */
3595 case AV_CODEC_ID_PCM_DVD:
3598 return 2 * (frame_bytes / ((bps * 2 / 8) * ch));
3599 case AV_CODEC_ID_PCM_BLURAY:
3602 return frame_bytes / ((FFALIGN(ch, 2) * bps) / 8);
3603 case AV_CODEC_ID_S302M:
3604 return 2 * (frame_bytes / ((bps + 4) / 4)) / ch;
3610 /* Fall back on using frame_size */
3611 if (frame_size > 1 && frame_bytes)
3614 //For WMA we currently have no other means to calculate duration thus we
3615 //do it here by assuming CBR, which is true for all known cases.
3616 if (bitrate > 0 && frame_bytes > 0 && sr > 0 && ba > 1) {
3617 if (id == AV_CODEC_ID_WMAV1 || id == AV_CODEC_ID_WMAV2)
3618 return (frame_bytes * 8LL * sr) / bitrate;
3624 int av_get_audio_frame_duration(AVCodecContext *avctx, int frame_bytes)
3626 return get_audio_frame_duration(avctx->codec_id, avctx->sample_rate,
3627 avctx->channels, avctx->block_align,
3628 avctx->codec_tag, avctx->bits_per_coded_sample,
3629 avctx->bit_rate, avctx->extradata, avctx->frame_size,
3633 int av_get_audio_frame_duration2(AVCodecParameters *par, int frame_bytes)
3635 return get_audio_frame_duration(par->codec_id, par->sample_rate,
3636 par->channels, par->block_align,
3637 par->codec_tag, par->bits_per_coded_sample,
3638 par->bit_rate, par->extradata, par->frame_size,
3643 int ff_thread_init(AVCodecContext *s)
3650 unsigned int av_xiphlacing(unsigned char *s, unsigned int v)
3664 int ff_match_2uint16(const uint16_t(*tab)[2], int size, int a, int b)
3667 for (i = 0; i < size && !(tab[i][0] == a && tab[i][1] == b); i++) ;
3671 #if FF_API_MISSING_SAMPLE
3672 FF_DISABLE_DEPRECATION_WARNINGS
3673 void av_log_missing_feature(void *avc, const char *feature, int want_sample)
3675 av_log(avc, AV_LOG_WARNING, "%s is not implemented. Update your FFmpeg "
3676 "version to the newest one from Git. If the problem still "
3677 "occurs, it means that your file has a feature which has not "
3678 "been implemented.\n", feature);
3680 av_log_ask_for_sample(avc, NULL);
3683 void av_log_ask_for_sample(void *avc, const char *msg, ...)
3685 va_list argument_list;
3687 va_start(argument_list, msg);
3690 av_vlog(avc, AV_LOG_WARNING, msg, argument_list);
3691 av_log(avc, AV_LOG_WARNING, "If you want to help, upload a sample "
3692 "of this file to ftp://upload.ffmpeg.org/incoming/ "
3693 "and contact the ffmpeg-devel mailing list. (ffmpeg-devel@ffmpeg.org)\n");
3695 va_end(argument_list);
3697 FF_ENABLE_DEPRECATION_WARNINGS
3698 #endif /* FF_API_MISSING_SAMPLE */
3700 static AVHWAccel *first_hwaccel = NULL;
3701 static AVHWAccel **last_hwaccel = &first_hwaccel;
3703 void av_register_hwaccel(AVHWAccel *hwaccel)
3705 AVHWAccel **p = last_hwaccel;
3706 hwaccel->next = NULL;
3707 while(*p || avpriv_atomic_ptr_cas((void * volatile *)p, NULL, hwaccel))
3709 last_hwaccel = &hwaccel->next;
3712 AVHWAccel *av_hwaccel_next(const AVHWAccel *hwaccel)
3714 return hwaccel ? hwaccel->next : first_hwaccel;
3717 int av_lockmgr_register(int (*cb)(void **mutex, enum AVLockOp op))
3720 // There is no good way to rollback a failure to destroy the
3721 // mutex, so we ignore failures.
3722 lockmgr_cb(&codec_mutex, AV_LOCK_DESTROY);
3723 lockmgr_cb(&avformat_mutex, AV_LOCK_DESTROY);
3726 avformat_mutex = NULL;
3730 void *new_codec_mutex = NULL;
3731 void *new_avformat_mutex = NULL;
3733 if (err = cb(&new_codec_mutex, AV_LOCK_CREATE)) {
3734 return err > 0 ? AVERROR_UNKNOWN : err;
3736 if (err = cb(&new_avformat_mutex, AV_LOCK_CREATE)) {
3737 // Ignore failures to destroy the newly created mutex.
3738 cb(&new_codec_mutex, AV_LOCK_DESTROY);
3739 return err > 0 ? AVERROR_UNKNOWN : err;
3742 codec_mutex = new_codec_mutex;
3743 avformat_mutex = new_avformat_mutex;
3749 int ff_lock_avcodec(AVCodecContext *log_ctx, const AVCodec *codec)
3751 if (codec->caps_internal & FF_CODEC_CAP_INIT_THREADSAFE || !codec->init)
3755 if ((*lockmgr_cb)(&codec_mutex, AV_LOCK_OBTAIN))
3759 if (avpriv_atomic_int_add_and_fetch(&entangled_thread_counter, 1) != 1) {
3760 av_log(log_ctx, AV_LOG_ERROR,
3761 "Insufficient thread locking. At least %d threads are "
3762 "calling avcodec_open2() at the same time right now.\n",
3763 entangled_thread_counter);
3765 av_log(log_ctx, AV_LOG_ERROR, "No lock manager is set, please see av_lockmgr_register()\n");
3766 ff_avcodec_locked = 1;
3767 ff_unlock_avcodec(codec);
3768 return AVERROR(EINVAL);
3770 av_assert0(!ff_avcodec_locked);
3771 ff_avcodec_locked = 1;
3775 int ff_unlock_avcodec(const AVCodec *codec)
3777 if (codec->caps_internal & FF_CODEC_CAP_INIT_THREADSAFE || !codec->init)
3780 av_assert0(ff_avcodec_locked);
3781 ff_avcodec_locked = 0;
3782 avpriv_atomic_int_add_and_fetch(&entangled_thread_counter, -1);
3784 if ((*lockmgr_cb)(&codec_mutex, AV_LOCK_RELEASE))
3791 int avpriv_lock_avformat(void)
3794 if ((*lockmgr_cb)(&avformat_mutex, AV_LOCK_OBTAIN))
3800 int avpriv_unlock_avformat(void)
3803 if ((*lockmgr_cb)(&avformat_mutex, AV_LOCK_RELEASE))
3809 unsigned int avpriv_toupper4(unsigned int x)
3811 return av_toupper(x & 0xFF) +
3812 (av_toupper((x >> 8) & 0xFF) << 8) +
3813 (av_toupper((x >> 16) & 0xFF) << 16) +
3814 ((unsigned)av_toupper((x >> 24) & 0xFF) << 24);
3817 int ff_thread_ref_frame(ThreadFrame *dst, ThreadFrame *src)
3821 dst->owner = src->owner;
3823 ret = av_frame_ref(dst->f, src->f);
3827 av_assert0(!dst->progress);
3829 if (src->progress &&
3830 !(dst->progress = av_buffer_ref(src->progress))) {
3831 ff_thread_release_buffer(dst->owner, dst);
3832 return AVERROR(ENOMEM);
3840 enum AVPixelFormat ff_thread_get_format(AVCodecContext *avctx, const enum AVPixelFormat *fmt)
3842 return ff_get_format(avctx, fmt);
3845 int ff_thread_get_buffer(AVCodecContext *avctx, ThreadFrame *f, int flags)
3848 return ff_get_buffer(avctx, f->f, flags);
3851 void ff_thread_release_buffer(AVCodecContext *avctx, ThreadFrame *f)
3854 av_frame_unref(f->f);
3857 void ff_thread_finish_setup(AVCodecContext *avctx)
3861 void ff_thread_report_progress(ThreadFrame *f, int progress, int field)
3865 void ff_thread_await_progress(ThreadFrame *f, int progress, int field)
3869 int ff_thread_can_start_frame(AVCodecContext *avctx)
3874 int ff_alloc_entries(AVCodecContext *avctx, int count)
3879 void ff_reset_entries(AVCodecContext *avctx)
3883 void ff_thread_await_progress2(AVCodecContext *avctx, int field, int thread, int shift)
3887 void ff_thread_report_progress2(AVCodecContext *avctx, int field, int thread, int n)
3893 int avcodec_is_open(AVCodecContext *s)
3895 return !!s->internal;
3898 int avpriv_bprint_to_extradata(AVCodecContext *avctx, struct AVBPrint *buf)
3903 ret = av_bprint_finalize(buf, &str);
3906 if (!av_bprint_is_complete(buf)) {
3908 return AVERROR(ENOMEM);
3911 avctx->extradata = str;
3912 /* Note: the string is NUL terminated (so extradata can be read as a
3913 * string), but the ending character is not accounted in the size (in
3914 * binary formats you are likely not supposed to mux that character). When
3915 * extradata is copied, it is also padded with AV_INPUT_BUFFER_PADDING_SIZE
3917 avctx->extradata_size = buf->len;
3921 const uint8_t *avpriv_find_start_code(const uint8_t *av_restrict p,
3923 uint32_t *av_restrict state)
3927 av_assert0(p <= end);
3931 for (i = 0; i < 3; i++) {
3932 uint32_t tmp = *state << 8;
3933 *state = tmp + *(p++);
3934 if (tmp == 0x100 || p == end)
3939 if (p[-1] > 1 ) p += 3;
3940 else if (p[-2] ) p += 2;
3941 else if (p[-3]|(p[-1]-1)) p++;
3948 p = FFMIN(p, end) - 4;
3949 *state = AV_RB32(p);
3954 AVCPBProperties *av_cpb_properties_alloc(size_t *size)
3956 AVCPBProperties *props = av_mallocz(sizeof(AVCPBProperties));
3961 *size = sizeof(*props);
3963 props->vbv_delay = UINT64_MAX;
3968 AVCPBProperties *ff_add_cpb_side_data(AVCodecContext *avctx)
3970 AVPacketSideData *tmp;
3971 AVCPBProperties *props;
3974 props = av_cpb_properties_alloc(&size);
3978 tmp = av_realloc_array(avctx->coded_side_data, avctx->nb_coded_side_data + 1, sizeof(*tmp));
3984 avctx->coded_side_data = tmp;
3985 avctx->nb_coded_side_data++;
3987 avctx->coded_side_data[avctx->nb_coded_side_data - 1].type = AV_PKT_DATA_CPB_PROPERTIES;
3988 avctx->coded_side_data[avctx->nb_coded_side_data - 1].data = (uint8_t*)props;
3989 avctx->coded_side_data[avctx->nb_coded_side_data - 1].size = size;
3994 static void codec_parameters_reset(AVCodecParameters *par)
3996 av_freep(&par->extradata);
3998 memset(par, 0, sizeof(*par));
4000 par->codec_type = AVMEDIA_TYPE_UNKNOWN;
4001 par->codec_id = AV_CODEC_ID_NONE;
4003 par->field_order = AV_FIELD_UNKNOWN;
4004 par->color_range = AVCOL_RANGE_UNSPECIFIED;
4005 par->color_primaries = AVCOL_PRI_UNSPECIFIED;
4006 par->color_trc = AVCOL_TRC_UNSPECIFIED;
4007 par->color_space = AVCOL_SPC_UNSPECIFIED;
4008 par->chroma_location = AVCHROMA_LOC_UNSPECIFIED;
4009 par->sample_aspect_ratio = (AVRational){ 0, 1 };
4010 par->profile = FF_PROFILE_UNKNOWN;
4011 par->level = FF_LEVEL_UNKNOWN;
4014 AVCodecParameters *avcodec_parameters_alloc(void)
4016 AVCodecParameters *par = av_mallocz(sizeof(*par));
4020 codec_parameters_reset(par);
4024 void avcodec_parameters_free(AVCodecParameters **ppar)
4026 AVCodecParameters *par = *ppar;
4030 codec_parameters_reset(par);
4035 int avcodec_parameters_copy(AVCodecParameters *dst, const AVCodecParameters *src)
4037 codec_parameters_reset(dst);
4038 memcpy(dst, src, sizeof(*dst));
4040 dst->extradata = NULL;
4041 dst->extradata_size = 0;
4042 if (src->extradata) {
4043 dst->extradata = av_mallocz(src->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE);
4044 if (!dst->extradata)
4045 return AVERROR(ENOMEM);
4046 memcpy(dst->extradata, src->extradata, src->extradata_size);
4047 dst->extradata_size = src->extradata_size;
4053 int avcodec_parameters_from_context(AVCodecParameters *par,
4054 const AVCodecContext *codec)
4056 codec_parameters_reset(par);
4058 par->codec_type = codec->codec_type;
4059 par->codec_id = codec->codec_id;
4060 par->codec_tag = codec->codec_tag;
4062 par->bit_rate = codec->bit_rate;
4063 par->bits_per_coded_sample = codec->bits_per_coded_sample;
4064 par->bits_per_raw_sample = codec->bits_per_raw_sample;
4065 par->profile = codec->profile;
4066 par->level = codec->level;
4068 switch (par->codec_type) {
4069 case AVMEDIA_TYPE_VIDEO:
4070 par->format = codec->pix_fmt;
4071 par->width = codec->width;
4072 par->height = codec->height;
4073 par->field_order = codec->field_order;
4074 par->color_range = codec->color_range;
4075 par->color_primaries = codec->color_primaries;
4076 par->color_trc = codec->color_trc;
4077 par->color_space = codec->colorspace;
4078 par->chroma_location = codec->chroma_sample_location;
4079 par->sample_aspect_ratio = codec->sample_aspect_ratio;
4080 par->video_delay = codec->has_b_frames;
4082 case AVMEDIA_TYPE_AUDIO:
4083 par->format = codec->sample_fmt;
4084 par->channel_layout = codec->channel_layout;
4085 par->channels = codec->channels;
4086 par->sample_rate = codec->sample_rate;
4087 par->block_align = codec->block_align;
4088 par->frame_size = codec->frame_size;
4089 par->initial_padding = codec->initial_padding;
4090 par->seek_preroll = codec->seek_preroll;
4092 case AVMEDIA_TYPE_SUBTITLE:
4093 par->width = codec->width;
4094 par->height = codec->height;
4098 if (codec->extradata) {
4099 par->extradata = av_mallocz(codec->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE);
4100 if (!par->extradata)
4101 return AVERROR(ENOMEM);
4102 memcpy(par->extradata, codec->extradata, codec->extradata_size);
4103 par->extradata_size = codec->extradata_size;
4109 int avcodec_parameters_to_context(AVCodecContext *codec,
4110 const AVCodecParameters *par)
4112 codec->codec_type = par->codec_type;
4113 codec->codec_id = par->codec_id;
4114 codec->codec_tag = par->codec_tag;
4116 codec->bit_rate = par->bit_rate;
4117 codec->bits_per_coded_sample = par->bits_per_coded_sample;
4118 codec->bits_per_raw_sample = par->bits_per_raw_sample;
4119 codec->profile = par->profile;
4120 codec->level = par->level;
4122 switch (par->codec_type) {
4123 case AVMEDIA_TYPE_VIDEO:
4124 codec->pix_fmt = par->format;
4125 codec->width = par->width;
4126 codec->height = par->height;
4127 codec->field_order = par->field_order;
4128 codec->color_range = par->color_range;
4129 codec->color_primaries = par->color_primaries;
4130 codec->color_trc = par->color_trc;
4131 codec->colorspace = par->color_space;
4132 codec->chroma_sample_location = par->chroma_location;
4133 codec->sample_aspect_ratio = par->sample_aspect_ratio;
4134 codec->has_b_frames = par->video_delay;
4136 case AVMEDIA_TYPE_AUDIO:
4137 codec->sample_fmt = par->format;
4138 codec->channel_layout = par->channel_layout;
4139 codec->channels = par->channels;
4140 codec->sample_rate = par->sample_rate;
4141 codec->block_align = par->block_align;
4142 codec->frame_size = par->frame_size;
4143 codec->initial_padding = par->initial_padding;
4144 codec->seek_preroll = par->seek_preroll;
4146 case AVMEDIA_TYPE_SUBTITLE:
4147 codec->width = par->width;
4148 codec->height = par->height;
4152 if (par->extradata) {
4153 av_freep(&codec->extradata);
4154 codec->extradata = av_mallocz(par->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE);
4155 if (!codec->extradata)
4156 return AVERROR(ENOMEM);
4157 memcpy(codec->extradata, par->extradata, par->extradata_size);
4158 codec->extradata_size = par->extradata_size;
4166 AVCodec *codec = NULL;
4168 avcodec_register_all();
4170 while (codec = av_codec_next(codec)) {
4171 if (av_codec_is_encoder(codec)) {
4172 if (codec->type == AVMEDIA_TYPE_AUDIO) {
4173 if (!codec->sample_fmts) {
4174 av_log(NULL, AV_LOG_FATAL, "Encoder %s is missing the sample_fmts field\n", codec->name);