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/internal.h"
38 #include "libavutil/mathematics.h"
39 #include "libavutil/mem_internal.h"
40 #include "libavutil/pixdesc.h"
41 #include "libavutil/imgutils.h"
42 #include "libavutil/samplefmt.h"
43 #include "libavutil/dict.h"
45 #include "libavutil/opt.h"
47 #include "mpegvideo.h"
49 #include "frame_thread_encoder.h"
52 #include "bytestream.h"
65 #include "compat/w32pthreads.h"
67 #include "compat/os2threads.h"
70 #include "libavutil/ffversion.h"
71 const char av_codec_ffversion[] = "FFmpeg version " FFMPEG_VERSION;
73 #if HAVE_PTHREADS || HAVE_W32THREADS || HAVE_OS2THREADS
74 static int default_lockmgr_cb(void **arg, enum AVLockOp op)
76 void * volatile * mutex = arg;
84 pthread_mutex_t *tmp = av_malloc(sizeof(pthread_mutex_t));
86 return AVERROR(ENOMEM);
87 if ((err = pthread_mutex_init(tmp, NULL))) {
91 if (avpriv_atomic_ptr_cas(mutex, NULL, tmp)) {
92 pthread_mutex_destroy(tmp);
97 if ((err = pthread_mutex_lock(*mutex)))
101 case AV_LOCK_RELEASE:
102 if ((err = pthread_mutex_unlock(*mutex)))
106 case AV_LOCK_DESTROY:
108 pthread_mutex_destroy(*mutex);
110 avpriv_atomic_ptr_cas(mutex, *mutex, NULL);
115 static int (*lockmgr_cb)(void **mutex, enum AVLockOp op) = default_lockmgr_cb;
117 static int (*lockmgr_cb)(void **mutex, enum AVLockOp op) = NULL;
121 volatile int ff_avcodec_locked;
122 static int volatile entangled_thread_counter = 0;
123 static void *codec_mutex;
124 static void *avformat_mutex;
126 void av_fast_padded_malloc(void *ptr, unsigned int *size, size_t min_size)
129 if (min_size > SIZE_MAX - AV_INPUT_BUFFER_PADDING_SIZE) {
134 if (!ff_fast_malloc(p, size, min_size + AV_INPUT_BUFFER_PADDING_SIZE, 1))
135 memset(*p + min_size, 0, AV_INPUT_BUFFER_PADDING_SIZE);
138 void av_fast_padded_mallocz(void *ptr, unsigned int *size, size_t min_size)
141 if (min_size > SIZE_MAX - AV_INPUT_BUFFER_PADDING_SIZE) {
146 if (!ff_fast_malloc(p, size, min_size + AV_INPUT_BUFFER_PADDING_SIZE, 1))
147 memset(*p, 0, min_size + AV_INPUT_BUFFER_PADDING_SIZE);
150 /* encoder management */
151 static AVCodec *first_avcodec = NULL;
152 static AVCodec **last_avcodec = &first_avcodec;
154 AVCodec *av_codec_next(const AVCodec *c)
159 return first_avcodec;
162 static av_cold void avcodec_init(void)
164 static int initialized = 0;
166 if (initialized != 0)
171 ff_me_cmp_init_static();
174 int av_codec_is_encoder(const AVCodec *codec)
176 return codec && (codec->encode_sub || codec->encode2);
179 int av_codec_is_decoder(const AVCodec *codec)
181 return codec && codec->decode;
184 av_cold void avcodec_register(AVCodec *codec)
191 while(*p || avpriv_atomic_ptr_cas((void * volatile *)p, NULL, codec))
193 last_avcodec = &codec->next;
195 if (codec->init_static_data)
196 codec->init_static_data(codec);
200 unsigned avcodec_get_edge_width(void)
206 #if FF_API_SET_DIMENSIONS
207 void avcodec_set_dimensions(AVCodecContext *s, int width, int height)
209 int ret = ff_set_dimensions(s, width, height);
211 av_log(s, AV_LOG_WARNING, "Failed to set dimensions %d %d\n", width, height);
216 int ff_set_dimensions(AVCodecContext *s, int width, int height)
218 int ret = av_image_check_size(width, height, 0, s);
223 s->coded_width = width;
224 s->coded_height = height;
225 s->width = FF_CEIL_RSHIFT(width, s->lowres);
226 s->height = FF_CEIL_RSHIFT(height, s->lowres);
231 int ff_set_sar(AVCodecContext *avctx, AVRational sar)
233 int ret = av_image_check_sar(avctx->width, avctx->height, sar);
236 av_log(avctx, AV_LOG_WARNING, "ignoring invalid SAR: %d/%d\n",
238 avctx->sample_aspect_ratio = (AVRational){ 0, 1 };
241 avctx->sample_aspect_ratio = sar;
246 int ff_side_data_update_matrix_encoding(AVFrame *frame,
247 enum AVMatrixEncoding matrix_encoding)
249 AVFrameSideData *side_data;
250 enum AVMatrixEncoding *data;
252 side_data = av_frame_get_side_data(frame, AV_FRAME_DATA_MATRIXENCODING);
254 side_data = av_frame_new_side_data(frame, AV_FRAME_DATA_MATRIXENCODING,
255 sizeof(enum AVMatrixEncoding));
258 return AVERROR(ENOMEM);
260 data = (enum AVMatrixEncoding*)side_data->data;
261 *data = matrix_encoding;
266 void avcodec_align_dimensions2(AVCodecContext *s, int *width, int *height,
267 int linesize_align[AV_NUM_DATA_POINTERS])
272 AVPixFmtDescriptor const *desc = av_pix_fmt_desc_get(s->pix_fmt);
275 w_align = 1 << desc->log2_chroma_w;
276 h_align = 1 << desc->log2_chroma_h;
279 switch (s->pix_fmt) {
280 case AV_PIX_FMT_YUV420P:
281 case AV_PIX_FMT_YUYV422:
282 case AV_PIX_FMT_YVYU422:
283 case AV_PIX_FMT_UYVY422:
284 case AV_PIX_FMT_YUV422P:
285 case AV_PIX_FMT_YUV440P:
286 case AV_PIX_FMT_YUV444P:
287 case AV_PIX_FMT_GBRP:
288 case AV_PIX_FMT_GBRAP:
289 case AV_PIX_FMT_GRAY8:
290 case AV_PIX_FMT_GRAY16BE:
291 case AV_PIX_FMT_GRAY16LE:
292 case AV_PIX_FMT_YUVJ420P:
293 case AV_PIX_FMT_YUVJ422P:
294 case AV_PIX_FMT_YUVJ440P:
295 case AV_PIX_FMT_YUVJ444P:
296 case AV_PIX_FMT_YUVA420P:
297 case AV_PIX_FMT_YUVA422P:
298 case AV_PIX_FMT_YUVA444P:
299 case AV_PIX_FMT_YUV420P9LE:
300 case AV_PIX_FMT_YUV420P9BE:
301 case AV_PIX_FMT_YUV420P10LE:
302 case AV_PIX_FMT_YUV420P10BE:
303 case AV_PIX_FMT_YUV420P12LE:
304 case AV_PIX_FMT_YUV420P12BE:
305 case AV_PIX_FMT_YUV420P14LE:
306 case AV_PIX_FMT_YUV420P14BE:
307 case AV_PIX_FMT_YUV420P16LE:
308 case AV_PIX_FMT_YUV420P16BE:
309 case AV_PIX_FMT_YUVA420P9LE:
310 case AV_PIX_FMT_YUVA420P9BE:
311 case AV_PIX_FMT_YUVA420P10LE:
312 case AV_PIX_FMT_YUVA420P10BE:
313 case AV_PIX_FMT_YUVA420P16LE:
314 case AV_PIX_FMT_YUVA420P16BE:
315 case AV_PIX_FMT_YUV422P9LE:
316 case AV_PIX_FMT_YUV422P9BE:
317 case AV_PIX_FMT_YUV422P10LE:
318 case AV_PIX_FMT_YUV422P10BE:
319 case AV_PIX_FMT_YUV422P12LE:
320 case AV_PIX_FMT_YUV422P12BE:
321 case AV_PIX_FMT_YUV422P14LE:
322 case AV_PIX_FMT_YUV422P14BE:
323 case AV_PIX_FMT_YUV422P16LE:
324 case AV_PIX_FMT_YUV422P16BE:
325 case AV_PIX_FMT_YUVA422P9LE:
326 case AV_PIX_FMT_YUVA422P9BE:
327 case AV_PIX_FMT_YUVA422P10LE:
328 case AV_PIX_FMT_YUVA422P10BE:
329 case AV_PIX_FMT_YUVA422P16LE:
330 case AV_PIX_FMT_YUVA422P16BE:
331 case AV_PIX_FMT_YUV440P10LE:
332 case AV_PIX_FMT_YUV440P10BE:
333 case AV_PIX_FMT_YUV440P12LE:
334 case AV_PIX_FMT_YUV440P12BE:
335 case AV_PIX_FMT_YUV444P9LE:
336 case AV_PIX_FMT_YUV444P9BE:
337 case AV_PIX_FMT_YUV444P10LE:
338 case AV_PIX_FMT_YUV444P10BE:
339 case AV_PIX_FMT_YUV444P12LE:
340 case AV_PIX_FMT_YUV444P12BE:
341 case AV_PIX_FMT_YUV444P14LE:
342 case AV_PIX_FMT_YUV444P14BE:
343 case AV_PIX_FMT_YUV444P16LE:
344 case AV_PIX_FMT_YUV444P16BE:
345 case AV_PIX_FMT_YUVA444P9LE:
346 case AV_PIX_FMT_YUVA444P9BE:
347 case AV_PIX_FMT_YUVA444P10LE:
348 case AV_PIX_FMT_YUVA444P10BE:
349 case AV_PIX_FMT_YUVA444P16LE:
350 case AV_PIX_FMT_YUVA444P16BE:
351 case AV_PIX_FMT_GBRP9LE:
352 case AV_PIX_FMT_GBRP9BE:
353 case AV_PIX_FMT_GBRP10LE:
354 case AV_PIX_FMT_GBRP10BE:
355 case AV_PIX_FMT_GBRP12LE:
356 case AV_PIX_FMT_GBRP12BE:
357 case AV_PIX_FMT_GBRP14LE:
358 case AV_PIX_FMT_GBRP14BE:
359 case AV_PIX_FMT_GBRP16LE:
360 case AV_PIX_FMT_GBRP16BE:
361 w_align = 16; //FIXME assume 16 pixel per macroblock
362 h_align = 16 * 2; // interlaced needs 2 macroblocks height
364 case AV_PIX_FMT_YUV411P:
365 case AV_PIX_FMT_YUVJ411P:
366 case AV_PIX_FMT_UYYVYY411:
370 case AV_PIX_FMT_YUV410P:
371 if (s->codec_id == AV_CODEC_ID_SVQ1) {
376 case AV_PIX_FMT_RGB555:
377 if (s->codec_id == AV_CODEC_ID_RPZA) {
382 case AV_PIX_FMT_PAL8:
383 case AV_PIX_FMT_BGR8:
384 case AV_PIX_FMT_RGB8:
385 if (s->codec_id == AV_CODEC_ID_SMC ||
386 s->codec_id == AV_CODEC_ID_CINEPAK) {
390 if (s->codec_id == AV_CODEC_ID_JV) {
395 case AV_PIX_FMT_BGR24:
396 if ((s->codec_id == AV_CODEC_ID_MSZH) ||
397 (s->codec_id == AV_CODEC_ID_ZLIB)) {
402 case AV_PIX_FMT_RGB24:
403 if (s->codec_id == AV_CODEC_ID_CINEPAK) {
412 if (s->codec_id == AV_CODEC_ID_IFF_ILBM) {
413 w_align = FFMAX(w_align, 8);
416 *width = FFALIGN(*width, w_align);
417 *height = FFALIGN(*height, h_align);
418 if (s->codec_id == AV_CODEC_ID_H264 || s->lowres) {
419 // some of the optimized chroma MC reads one line too much
420 // which is also done in mpeg decoders with lowres > 0
423 // H.264 uses edge emulation for out of frame motion vectors, for this
424 // it requires a temporary area large enough to hold a 21x21 block,
425 // increasing witdth ensure that the temporary area is large enough,
426 // the next rounded up width is 32
427 *width = FFMAX(*width, 32);
430 for (i = 0; i < 4; i++)
431 linesize_align[i] = STRIDE_ALIGN;
434 void avcodec_align_dimensions(AVCodecContext *s, int *width, int *height)
436 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(s->pix_fmt);
437 int chroma_shift = desc->log2_chroma_w;
438 int linesize_align[AV_NUM_DATA_POINTERS];
441 avcodec_align_dimensions2(s, width, height, linesize_align);
442 align = FFMAX(linesize_align[0], linesize_align[3]);
443 linesize_align[1] <<= chroma_shift;
444 linesize_align[2] <<= chroma_shift;
445 align = FFMAX3(align, linesize_align[1], linesize_align[2]);
446 *width = FFALIGN(*width, align);
449 int avcodec_enum_to_chroma_pos(int *xpos, int *ypos, enum AVChromaLocation pos)
451 if (pos <= AVCHROMA_LOC_UNSPECIFIED || pos >= AVCHROMA_LOC_NB)
452 return AVERROR(EINVAL);
455 *xpos = (pos&1) * 128;
456 *ypos = ((pos>>1)^(pos<4)) * 128;
461 enum AVChromaLocation avcodec_chroma_pos_to_enum(int xpos, int ypos)
465 for (pos = AVCHROMA_LOC_UNSPECIFIED + 1; pos < AVCHROMA_LOC_NB; pos++) {
466 if (avcodec_enum_to_chroma_pos(&xout, &yout, pos) == 0 && xout == xpos && yout == ypos)
469 return AVCHROMA_LOC_UNSPECIFIED;
472 int avcodec_fill_audio_frame(AVFrame *frame, int nb_channels,
473 enum AVSampleFormat sample_fmt, const uint8_t *buf,
474 int buf_size, int align)
476 int ch, planar, needed_size, ret = 0;
478 needed_size = av_samples_get_buffer_size(NULL, nb_channels,
479 frame->nb_samples, sample_fmt,
481 if (buf_size < needed_size)
482 return AVERROR(EINVAL);
484 planar = av_sample_fmt_is_planar(sample_fmt);
485 if (planar && nb_channels > AV_NUM_DATA_POINTERS) {
486 if (!(frame->extended_data = av_mallocz_array(nb_channels,
487 sizeof(*frame->extended_data))))
488 return AVERROR(ENOMEM);
490 frame->extended_data = frame->data;
493 if ((ret = av_samples_fill_arrays(frame->extended_data, &frame->linesize[0],
494 (uint8_t *)(intptr_t)buf, nb_channels, frame->nb_samples,
495 sample_fmt, align)) < 0) {
496 if (frame->extended_data != frame->data)
497 av_freep(&frame->extended_data);
500 if (frame->extended_data != frame->data) {
501 for (ch = 0; ch < AV_NUM_DATA_POINTERS; ch++)
502 frame->data[ch] = frame->extended_data[ch];
508 static int update_frame_pool(AVCodecContext *avctx, AVFrame *frame)
510 FramePool *pool = avctx->internal->pool;
513 switch (avctx->codec_type) {
514 case AVMEDIA_TYPE_VIDEO: {
518 int w = frame->width;
519 int h = frame->height;
520 int tmpsize, unaligned;
522 if (pool->format == frame->format &&
523 pool->width == frame->width && pool->height == frame->height)
526 avcodec_align_dimensions2(avctx, &w, &h, pool->stride_align);
529 // NOTE: do not align linesizes individually, this breaks e.g. assumptions
530 // that linesize[0] == 2*linesize[1] in the MPEG-encoder for 4:2:2
531 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;
659 av_log(s, AV_LOG_ERROR, "pic->data[0]!=NULL in avcodec_default_get_buffer\n");
663 memset(pic->data, 0, sizeof(pic->data));
664 pic->extended_data = pic->data;
666 for (i = 0; i < 4 && pool->pools[i]; i++) {
667 pic->linesize[i] = pool->linesize[i];
669 pic->buf[i] = av_buffer_pool_get(pool->pools[i]);
673 pic->data[i] = pic->buf[i]->data;
675 for (; i < AV_NUM_DATA_POINTERS; i++) {
677 pic->linesize[i] = 0;
679 if (pic->data[1] && !pic->data[2])
680 avpriv_set_systematic_pal2((uint32_t *)pic->data[1], s->pix_fmt);
682 if (s->debug & FF_DEBUG_BUFFERS)
683 av_log(s, AV_LOG_DEBUG, "default_get_buffer called on pic %p\n", pic);
688 return AVERROR(ENOMEM);
691 void ff_color_frame(AVFrame *frame, const int c[4])
693 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(frame->format);
696 av_assert0(desc->flags & AV_PIX_FMT_FLAG_PLANAR);
698 for (p = 0; p<desc->nb_components; p++) {
699 uint8_t *dst = frame->data[p];
700 int is_chroma = p == 1 || p == 2;
701 int bytes = is_chroma ? FF_CEIL_RSHIFT(frame->width, desc->log2_chroma_w) : frame->width;
702 int height = is_chroma ? FF_CEIL_RSHIFT(frame->height, desc->log2_chroma_h) : frame->height;
703 for (y = 0; y < height; y++) {
704 if (desc->comp[0].depth >= 9) {
705 for (x = 0; x<bytes; x++)
706 ((uint16_t*)dst)[x] = c[p];
708 memset(dst, c[p], bytes);
709 dst += frame->linesize[p];
714 int avcodec_default_get_buffer2(AVCodecContext *avctx, AVFrame *frame, int flags)
718 if ((ret = update_frame_pool(avctx, frame)) < 0)
721 switch (avctx->codec_type) {
722 case AVMEDIA_TYPE_VIDEO:
723 return video_get_buffer(avctx, frame);
724 case AVMEDIA_TYPE_AUDIO:
725 return audio_get_buffer(avctx, frame);
731 static int add_metadata_from_side_data(AVPacket *avpkt, AVFrame *frame)
734 const uint8_t *side_metadata;
736 AVDictionary **frame_md = avpriv_frame_get_metadatap(frame);
738 side_metadata = av_packet_get_side_data(avpkt,
739 AV_PKT_DATA_STRINGS_METADATA, &size);
740 return av_packet_unpack_dictionary(side_metadata, size, frame_md);
743 int ff_init_buffer_info(AVCodecContext *avctx, AVFrame *frame)
745 AVPacket *pkt = avctx->internal->pkt;
747 static const struct {
748 enum AVPacketSideDataType packet;
749 enum AVFrameSideDataType frame;
751 { AV_PKT_DATA_REPLAYGAIN , AV_FRAME_DATA_REPLAYGAIN },
752 { AV_PKT_DATA_DISPLAYMATRIX, AV_FRAME_DATA_DISPLAYMATRIX },
753 { AV_PKT_DATA_STEREO3D, AV_FRAME_DATA_STEREO3D },
754 { AV_PKT_DATA_AUDIO_SERVICE_TYPE, AV_FRAME_DATA_AUDIO_SERVICE_TYPE },
758 frame->pkt_pts = pkt->pts;
759 av_frame_set_pkt_pos (frame, pkt->pos);
760 av_frame_set_pkt_duration(frame, pkt->duration);
761 av_frame_set_pkt_size (frame, pkt->size);
763 for (i = 0; i < FF_ARRAY_ELEMS(sd); i++) {
765 uint8_t *packet_sd = av_packet_get_side_data(pkt, sd[i].packet, &size);
767 AVFrameSideData *frame_sd = av_frame_new_side_data(frame,
771 return AVERROR(ENOMEM);
773 memcpy(frame_sd->data, packet_sd, size);
776 add_metadata_from_side_data(pkt, frame);
778 frame->pkt_pts = AV_NOPTS_VALUE;
779 av_frame_set_pkt_pos (frame, -1);
780 av_frame_set_pkt_duration(frame, 0);
781 av_frame_set_pkt_size (frame, -1);
783 frame->reordered_opaque = avctx->reordered_opaque;
785 if (frame->color_primaries == AVCOL_PRI_UNSPECIFIED)
786 frame->color_primaries = avctx->color_primaries;
787 if (frame->color_trc == AVCOL_TRC_UNSPECIFIED)
788 frame->color_trc = avctx->color_trc;
789 if (av_frame_get_colorspace(frame) == AVCOL_SPC_UNSPECIFIED)
790 av_frame_set_colorspace(frame, avctx->colorspace);
791 if (av_frame_get_color_range(frame) == AVCOL_RANGE_UNSPECIFIED)
792 av_frame_set_color_range(frame, avctx->color_range);
793 if (frame->chroma_location == AVCHROMA_LOC_UNSPECIFIED)
794 frame->chroma_location = avctx->chroma_sample_location;
796 switch (avctx->codec->type) {
797 case AVMEDIA_TYPE_VIDEO:
798 frame->format = avctx->pix_fmt;
799 if (!frame->sample_aspect_ratio.num)
800 frame->sample_aspect_ratio = avctx->sample_aspect_ratio;
802 if (frame->width && frame->height &&
803 av_image_check_sar(frame->width, frame->height,
804 frame->sample_aspect_ratio) < 0) {
805 av_log(avctx, AV_LOG_WARNING, "ignoring invalid SAR: %u/%u\n",
806 frame->sample_aspect_ratio.num,
807 frame->sample_aspect_ratio.den);
808 frame->sample_aspect_ratio = (AVRational){ 0, 1 };
812 case AVMEDIA_TYPE_AUDIO:
813 if (!frame->sample_rate)
814 frame->sample_rate = avctx->sample_rate;
815 if (frame->format < 0)
816 frame->format = avctx->sample_fmt;
817 if (!frame->channel_layout) {
818 if (avctx->channel_layout) {
819 if (av_get_channel_layout_nb_channels(avctx->channel_layout) !=
821 av_log(avctx, AV_LOG_ERROR, "Inconsistent channel "
823 return AVERROR(EINVAL);
826 frame->channel_layout = avctx->channel_layout;
828 if (avctx->channels > FF_SANE_NB_CHANNELS) {
829 av_log(avctx, AV_LOG_ERROR, "Too many channels: %d.\n",
831 return AVERROR(ENOSYS);
835 av_frame_set_channels(frame, avctx->channels);
841 int ff_decode_frame_props(AVCodecContext *avctx, AVFrame *frame)
843 return ff_init_buffer_info(avctx, frame);
846 static int get_buffer_internal(AVCodecContext *avctx, AVFrame *frame, int flags)
848 const AVHWAccel *hwaccel = avctx->hwaccel;
849 int override_dimensions = 1;
852 if (avctx->codec_type == AVMEDIA_TYPE_VIDEO) {
853 if ((ret = av_image_check_size(avctx->width, avctx->height, 0, avctx)) < 0 || avctx->pix_fmt<0) {
854 av_log(avctx, AV_LOG_ERROR, "video_get_buffer: image parameters invalid\n");
855 return AVERROR(EINVAL);
858 if (avctx->codec_type == AVMEDIA_TYPE_VIDEO) {
859 if (frame->width <= 0 || frame->height <= 0) {
860 frame->width = FFMAX(avctx->width, FF_CEIL_RSHIFT(avctx->coded_width, avctx->lowres));
861 frame->height = FFMAX(avctx->height, FF_CEIL_RSHIFT(avctx->coded_height, avctx->lowres));
862 override_dimensions = 0;
865 ret = ff_decode_frame_props(avctx, frame);
870 if (hwaccel->alloc_frame) {
871 ret = hwaccel->alloc_frame(avctx, frame);
875 avctx->sw_pix_fmt = avctx->pix_fmt;
877 ret = avctx->get_buffer2(avctx, frame, flags);
880 if (avctx->codec_type == AVMEDIA_TYPE_VIDEO && !override_dimensions) {
881 frame->width = avctx->width;
882 frame->height = avctx->height;
888 int ff_get_buffer(AVCodecContext *avctx, AVFrame *frame, int flags)
890 int ret = get_buffer_internal(avctx, frame, flags);
892 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
893 frame->width = frame->height = 0;
898 static int reget_buffer_internal(AVCodecContext *avctx, AVFrame *frame)
903 av_assert0(avctx->codec_type == AVMEDIA_TYPE_VIDEO);
905 if (frame->data[0] && (frame->width != avctx->width || frame->height != avctx->height || frame->format != avctx->pix_fmt)) {
906 av_log(avctx, AV_LOG_WARNING, "Picture changed from size:%dx%d fmt:%s to size:%dx%d fmt:%s in reget buffer()\n",
907 frame->width, frame->height, av_get_pix_fmt_name(frame->format), avctx->width, avctx->height, av_get_pix_fmt_name(avctx->pix_fmt));
908 av_frame_unref(frame);
911 ff_init_buffer_info(avctx, frame);
914 return ff_get_buffer(avctx, frame, AV_GET_BUFFER_FLAG_REF);
916 if (av_frame_is_writable(frame))
917 return ff_decode_frame_props(avctx, frame);
919 tmp = av_frame_alloc();
921 return AVERROR(ENOMEM);
923 av_frame_move_ref(tmp, frame);
925 ret = ff_get_buffer(avctx, frame, AV_GET_BUFFER_FLAG_REF);
931 av_frame_copy(frame, tmp);
937 int ff_reget_buffer(AVCodecContext *avctx, AVFrame *frame)
939 int ret = reget_buffer_internal(avctx, frame);
941 av_log(avctx, AV_LOG_ERROR, "reget_buffer() failed\n");
945 int avcodec_default_execute(AVCodecContext *c, int (*func)(AVCodecContext *c2, void *arg2), void *arg, int *ret, int count, int size)
949 for (i = 0; i < count; i++) {
950 int r = func(c, (char *)arg + i * size);
957 int avcodec_default_execute2(AVCodecContext *c, int (*func)(AVCodecContext *c2, void *arg2, int jobnr, int threadnr), void *arg, int *ret, int count)
961 for (i = 0; i < count; i++) {
962 int r = func(c, arg, i, 0);
969 enum AVPixelFormat avpriv_find_pix_fmt(const PixelFormatTag *tags,
972 while (tags->pix_fmt >= 0) {
973 if (tags->fourcc == fourcc)
974 return tags->pix_fmt;
977 return AV_PIX_FMT_NONE;
980 static int is_hwaccel_pix_fmt(enum AVPixelFormat pix_fmt)
982 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
983 return desc->flags & AV_PIX_FMT_FLAG_HWACCEL;
986 enum AVPixelFormat avcodec_default_get_format(struct AVCodecContext *s, const enum AVPixelFormat *fmt)
988 while (*fmt != AV_PIX_FMT_NONE && is_hwaccel_pix_fmt(*fmt))
993 static AVHWAccel *find_hwaccel(enum AVCodecID codec_id,
994 enum AVPixelFormat pix_fmt)
996 AVHWAccel *hwaccel = NULL;
998 while ((hwaccel = av_hwaccel_next(hwaccel)))
999 if (hwaccel->id == codec_id
1000 && hwaccel->pix_fmt == pix_fmt)
1005 static int setup_hwaccel(AVCodecContext *avctx,
1006 const enum AVPixelFormat fmt,
1009 AVHWAccel *hwa = find_hwaccel(avctx->codec_id, fmt);
1012 if (avctx->active_thread_type & FF_THREAD_FRAME) {
1013 av_log(avctx, AV_LOG_ERROR,
1014 "Hardware accelerated decoding with frame threading is not supported.\n");
1015 return AVERROR(EINVAL);
1019 av_log(avctx, AV_LOG_ERROR,
1020 "Could not find an AVHWAccel for the pixel format: %s",
1022 return AVERROR(ENOENT);
1025 if (hwa->capabilities & HWACCEL_CODEC_CAP_EXPERIMENTAL &&
1026 avctx->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) {
1027 av_log(avctx, AV_LOG_WARNING, "Ignoring experimental hwaccel: %s\n",
1029 return AVERROR_PATCHWELCOME;
1032 if (hwa->priv_data_size) {
1033 avctx->internal->hwaccel_priv_data = av_mallocz(hwa->priv_data_size);
1034 if (!avctx->internal->hwaccel_priv_data)
1035 return AVERROR(ENOMEM);
1039 ret = hwa->init(avctx);
1041 av_freep(&avctx->internal->hwaccel_priv_data);
1046 avctx->hwaccel = hwa;
1051 int ff_get_format(AVCodecContext *avctx, const enum AVPixelFormat *fmt)
1053 const AVPixFmtDescriptor *desc;
1054 enum AVPixelFormat *choices;
1055 enum AVPixelFormat ret;
1058 while (fmt[n] != AV_PIX_FMT_NONE)
1062 avctx->sw_pix_fmt = fmt[n - 1];
1063 av_assert2(!is_hwaccel_pix_fmt(avctx->sw_pix_fmt));
1065 choices = av_malloc_array(n + 1, sizeof(*choices));
1067 return AV_PIX_FMT_NONE;
1069 memcpy(choices, fmt, (n + 1) * sizeof(*choices));
1072 if (avctx->hwaccel && avctx->hwaccel->uninit)
1073 avctx->hwaccel->uninit(avctx);
1074 av_freep(&avctx->internal->hwaccel_priv_data);
1075 avctx->hwaccel = NULL;
1077 ret = avctx->get_format(avctx, choices);
1079 desc = av_pix_fmt_desc_get(ret);
1081 ret = AV_PIX_FMT_NONE;
1085 if (!(desc->flags & AV_PIX_FMT_FLAG_HWACCEL))
1087 #if FF_API_CAP_VDPAU
1088 if (avctx->codec->capabilities&AV_CODEC_CAP_HWACCEL_VDPAU)
1092 if (!setup_hwaccel(avctx, ret, desc->name))
1095 /* Remove failed hwaccel from choices */
1096 for (n = 0; choices[n] != ret; n++)
1097 av_assert0(choices[n] != AV_PIX_FMT_NONE);
1100 choices[n] = choices[n + 1];
1101 while (choices[n++] != AV_PIX_FMT_NONE);
1108 MAKE_ACCESSORS(AVCodecContext, codec, AVRational, pkt_timebase)
1109 MAKE_ACCESSORS(AVCodecContext, codec, const AVCodecDescriptor *, codec_descriptor)
1110 MAKE_ACCESSORS(AVCodecContext, codec, int, lowres)
1111 MAKE_ACCESSORS(AVCodecContext, codec, int, seek_preroll)
1112 MAKE_ACCESSORS(AVCodecContext, codec, uint16_t*, chroma_intra_matrix)
1114 unsigned av_codec_get_codec_properties(const AVCodecContext *codec)
1116 return codec->properties;
1119 int av_codec_get_max_lowres(const AVCodec *codec)
1121 return codec->max_lowres;
1124 static void get_subtitle_defaults(AVSubtitle *sub)
1126 memset(sub, 0, sizeof(*sub));
1127 sub->pts = AV_NOPTS_VALUE;
1130 static int64_t get_bit_rate(AVCodecContext *ctx)
1133 int bits_per_sample;
1135 switch (ctx->codec_type) {
1136 case AVMEDIA_TYPE_VIDEO:
1137 case AVMEDIA_TYPE_DATA:
1138 case AVMEDIA_TYPE_SUBTITLE:
1139 case AVMEDIA_TYPE_ATTACHMENT:
1140 bit_rate = ctx->bit_rate;
1142 case AVMEDIA_TYPE_AUDIO:
1143 bits_per_sample = av_get_bits_per_sample(ctx->codec_id);
1144 bit_rate = bits_per_sample ? ctx->sample_rate * (int64_t)ctx->channels * bits_per_sample : ctx->bit_rate;
1153 int attribute_align_arg ff_codec_open2_recursive(AVCodecContext *avctx, const AVCodec *codec, AVDictionary **options)
1157 ff_unlock_avcodec(codec);
1159 ret = avcodec_open2(avctx, codec, options);
1161 ff_lock_avcodec(avctx, codec);
1165 int attribute_align_arg avcodec_open2(AVCodecContext *avctx, const AVCodec *codec, AVDictionary **options)
1168 AVDictionary *tmp = NULL;
1170 if (avcodec_is_open(avctx))
1173 if ((!codec && !avctx->codec)) {
1174 av_log(avctx, AV_LOG_ERROR, "No codec provided to avcodec_open2()\n");
1175 return AVERROR(EINVAL);
1177 if ((codec && avctx->codec && codec != avctx->codec)) {
1178 av_log(avctx, AV_LOG_ERROR, "This AVCodecContext was allocated for %s, "
1179 "but %s passed to avcodec_open2()\n", avctx->codec->name, codec->name);
1180 return AVERROR(EINVAL);
1183 codec = avctx->codec;
1185 if (avctx->extradata_size < 0 || avctx->extradata_size >= FF_MAX_EXTRADATA_SIZE)
1186 return AVERROR(EINVAL);
1189 av_dict_copy(&tmp, *options, 0);
1191 ret = ff_lock_avcodec(avctx, codec);
1195 avctx->internal = av_mallocz(sizeof(AVCodecInternal));
1196 if (!avctx->internal) {
1197 ret = AVERROR(ENOMEM);
1201 avctx->internal->pool = av_mallocz(sizeof(*avctx->internal->pool));
1202 if (!avctx->internal->pool) {
1203 ret = AVERROR(ENOMEM);
1207 avctx->internal->to_free = av_frame_alloc();
1208 if (!avctx->internal->to_free) {
1209 ret = AVERROR(ENOMEM);
1213 if (codec->priv_data_size > 0) {
1214 if (!avctx->priv_data) {
1215 avctx->priv_data = av_mallocz(codec->priv_data_size);
1216 if (!avctx->priv_data) {
1217 ret = AVERROR(ENOMEM);
1220 if (codec->priv_class) {
1221 *(const AVClass **)avctx->priv_data = codec->priv_class;
1222 av_opt_set_defaults(avctx->priv_data);
1225 if (codec->priv_class && (ret = av_opt_set_dict(avctx->priv_data, &tmp)) < 0)
1228 avctx->priv_data = NULL;
1230 if ((ret = av_opt_set_dict(avctx, &tmp)) < 0)
1233 if (avctx->codec_whitelist && av_match_list(codec->name, avctx->codec_whitelist, ',') <= 0) {
1234 av_log(avctx, AV_LOG_ERROR, "Codec (%s) not on whitelist\n", codec->name);
1235 ret = AVERROR(EINVAL);
1239 // only call ff_set_dimensions() for non H.264/VP6F/DXV codecs so as not to overwrite previously setup dimensions
1240 if (!(avctx->coded_width && avctx->coded_height && avctx->width && avctx->height &&
1241 (avctx->codec_id == AV_CODEC_ID_H264 || avctx->codec_id == AV_CODEC_ID_VP6F || avctx->codec_id == AV_CODEC_ID_DXV))) {
1242 if (avctx->coded_width && avctx->coded_height)
1243 ret = ff_set_dimensions(avctx, avctx->coded_width, avctx->coded_height);
1244 else if (avctx->width && avctx->height)
1245 ret = ff_set_dimensions(avctx, avctx->width, avctx->height);
1250 if ((avctx->coded_width || avctx->coded_height || avctx->width || avctx->height)
1251 && ( av_image_check_size(avctx->coded_width, avctx->coded_height, 0, avctx) < 0
1252 || av_image_check_size(avctx->width, avctx->height, 0, avctx) < 0)) {
1253 av_log(avctx, AV_LOG_WARNING, "Ignoring invalid width/height values\n");
1254 ff_set_dimensions(avctx, 0, 0);
1257 if (avctx->width > 0 && avctx->height > 0) {
1258 if (av_image_check_sar(avctx->width, avctx->height,
1259 avctx->sample_aspect_ratio) < 0) {
1260 av_log(avctx, AV_LOG_WARNING, "ignoring invalid SAR: %u/%u\n",
1261 avctx->sample_aspect_ratio.num,
1262 avctx->sample_aspect_ratio.den);
1263 avctx->sample_aspect_ratio = (AVRational){ 0, 1 };
1267 /* if the decoder init function was already called previously,
1268 * free the already allocated subtitle_header before overwriting it */
1269 if (av_codec_is_decoder(codec))
1270 av_freep(&avctx->subtitle_header);
1272 if (avctx->channels > FF_SANE_NB_CHANNELS) {
1273 ret = AVERROR(EINVAL);
1277 avctx->codec = codec;
1278 if ((avctx->codec_type == AVMEDIA_TYPE_UNKNOWN || avctx->codec_type == codec->type) &&
1279 avctx->codec_id == AV_CODEC_ID_NONE) {
1280 avctx->codec_type = codec->type;
1281 avctx->codec_id = codec->id;
1283 if (avctx->codec_id != codec->id || (avctx->codec_type != codec->type
1284 && avctx->codec_type != AVMEDIA_TYPE_ATTACHMENT)) {
1285 av_log(avctx, AV_LOG_ERROR, "Codec type or id mismatches\n");
1286 ret = AVERROR(EINVAL);
1289 avctx->frame_number = 0;
1290 avctx->codec_descriptor = avcodec_descriptor_get(avctx->codec_id);
1292 if ((avctx->codec->capabilities & AV_CODEC_CAP_EXPERIMENTAL) &&
1293 avctx->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) {
1294 const char *codec_string = av_codec_is_encoder(codec) ? "encoder" : "decoder";
1296 av_log(avctx, AV_LOG_ERROR,
1297 "The %s '%s' is experimental but experimental codecs are not enabled, "
1298 "add '-strict %d' if you want to use it.\n",
1299 codec_string, codec->name, FF_COMPLIANCE_EXPERIMENTAL);
1300 codec2 = av_codec_is_encoder(codec) ? avcodec_find_encoder(codec->id) : avcodec_find_decoder(codec->id);
1301 if (!(codec2->capabilities & AV_CODEC_CAP_EXPERIMENTAL))
1302 av_log(avctx, AV_LOG_ERROR, "Alternatively use the non experimental %s '%s'.\n",
1303 codec_string, codec2->name);
1304 ret = AVERROR_EXPERIMENTAL;
1308 if (avctx->codec_type == AVMEDIA_TYPE_AUDIO &&
1309 (!avctx->time_base.num || !avctx->time_base.den)) {
1310 avctx->time_base.num = 1;
1311 avctx->time_base.den = avctx->sample_rate;
1315 av_log(avctx, AV_LOG_WARNING, "Warning: not compiled with thread support, using thread emulation\n");
1317 if (CONFIG_FRAME_THREAD_ENCODER) {
1318 ff_unlock_avcodec(codec); //we will instanciate a few encoders thus kick the counter to prevent false detection of a problem
1319 ret = ff_frame_thread_encoder_init(avctx, options ? *options : NULL);
1320 ff_lock_avcodec(avctx, codec);
1326 && !(avctx->internal->frame_thread_encoder && (avctx->active_thread_type&FF_THREAD_FRAME))) {
1327 ret = ff_thread_init(avctx);
1332 if (!HAVE_THREADS && !(codec->capabilities & AV_CODEC_CAP_AUTO_THREADS))
1333 avctx->thread_count = 1;
1335 if (avctx->codec->max_lowres < avctx->lowres || avctx->lowres < 0) {
1336 av_log(avctx, AV_LOG_ERROR, "The maximum value for lowres supported by the decoder is %d\n",
1337 avctx->codec->max_lowres);
1338 ret = AVERROR(EINVAL);
1343 if (avctx->debug_mv)
1344 av_log(avctx, AV_LOG_WARNING, "The 'vismv' option is deprecated, "
1345 "see the codecview filter instead.\n");
1348 if (av_codec_is_encoder(avctx->codec)) {
1350 #if FF_API_CODED_FRAME
1351 FF_DISABLE_DEPRECATION_WARNINGS
1352 avctx->coded_frame = av_frame_alloc();
1353 if (!avctx->coded_frame) {
1354 ret = AVERROR(ENOMEM);
1357 FF_ENABLE_DEPRECATION_WARNINGS
1359 if (avctx->codec->sample_fmts) {
1360 for (i = 0; avctx->codec->sample_fmts[i] != AV_SAMPLE_FMT_NONE; i++) {
1361 if (avctx->sample_fmt == avctx->codec->sample_fmts[i])
1363 if (avctx->channels == 1 &&
1364 av_get_planar_sample_fmt(avctx->sample_fmt) ==
1365 av_get_planar_sample_fmt(avctx->codec->sample_fmts[i])) {
1366 avctx->sample_fmt = avctx->codec->sample_fmts[i];
1370 if (avctx->codec->sample_fmts[i] == AV_SAMPLE_FMT_NONE) {
1372 snprintf(buf, sizeof(buf), "%d", avctx->sample_fmt);
1373 av_log(avctx, AV_LOG_ERROR, "Specified sample format %s is invalid or not supported\n",
1374 (char *)av_x_if_null(av_get_sample_fmt_name(avctx->sample_fmt), buf));
1375 ret = AVERROR(EINVAL);
1379 if (avctx->codec->pix_fmts) {
1380 for (i = 0; avctx->codec->pix_fmts[i] != AV_PIX_FMT_NONE; i++)
1381 if (avctx->pix_fmt == avctx->codec->pix_fmts[i])
1383 if (avctx->codec->pix_fmts[i] == AV_PIX_FMT_NONE
1384 && !((avctx->codec_id == AV_CODEC_ID_MJPEG || avctx->codec_id == AV_CODEC_ID_LJPEG)
1385 && avctx->strict_std_compliance <= FF_COMPLIANCE_UNOFFICIAL)) {
1387 snprintf(buf, sizeof(buf), "%d", avctx->pix_fmt);
1388 av_log(avctx, AV_LOG_ERROR, "Specified pixel format %s is invalid or not supported\n",
1389 (char *)av_x_if_null(av_get_pix_fmt_name(avctx->pix_fmt), buf));
1390 ret = AVERROR(EINVAL);
1393 if (avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ420P ||
1394 avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ411P ||
1395 avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ422P ||
1396 avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ440P ||
1397 avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ444P)
1398 avctx->color_range = AVCOL_RANGE_JPEG;
1400 if (avctx->codec->supported_samplerates) {
1401 for (i = 0; avctx->codec->supported_samplerates[i] != 0; i++)
1402 if (avctx->sample_rate == avctx->codec->supported_samplerates[i])
1404 if (avctx->codec->supported_samplerates[i] == 0) {
1405 av_log(avctx, AV_LOG_ERROR, "Specified sample rate %d is not supported\n",
1406 avctx->sample_rate);
1407 ret = AVERROR(EINVAL);
1411 if (avctx->sample_rate < 0) {
1412 av_log(avctx, AV_LOG_ERROR, "Specified sample rate %d is not supported\n",
1413 avctx->sample_rate);
1414 ret = AVERROR(EINVAL);
1417 if (avctx->codec->channel_layouts) {
1418 if (!avctx->channel_layout) {
1419 av_log(avctx, AV_LOG_WARNING, "Channel layout not specified\n");
1421 for (i = 0; avctx->codec->channel_layouts[i] != 0; i++)
1422 if (avctx->channel_layout == avctx->codec->channel_layouts[i])
1424 if (avctx->codec->channel_layouts[i] == 0) {
1426 av_get_channel_layout_string(buf, sizeof(buf), -1, avctx->channel_layout);
1427 av_log(avctx, AV_LOG_ERROR, "Specified channel layout '%s' is not supported\n", buf);
1428 ret = AVERROR(EINVAL);
1433 if (avctx->channel_layout && avctx->channels) {
1434 int channels = av_get_channel_layout_nb_channels(avctx->channel_layout);
1435 if (channels != avctx->channels) {
1437 av_get_channel_layout_string(buf, sizeof(buf), -1, avctx->channel_layout);
1438 av_log(avctx, AV_LOG_ERROR,
1439 "Channel layout '%s' with %d channels does not match number of specified channels %d\n",
1440 buf, channels, avctx->channels);
1441 ret = AVERROR(EINVAL);
1444 } else if (avctx->channel_layout) {
1445 avctx->channels = av_get_channel_layout_nb_channels(avctx->channel_layout);
1447 if (avctx->channels < 0) {
1448 av_log(avctx, AV_LOG_ERROR, "Specified number of channels %d is not supported\n",
1450 ret = AVERROR(EINVAL);
1453 if(avctx->codec_type == AVMEDIA_TYPE_VIDEO) {
1454 if (avctx->width <= 0 || avctx->height <= 0) {
1455 av_log(avctx, AV_LOG_ERROR, "dimensions not set\n");
1456 ret = AVERROR(EINVAL);
1460 if ( (avctx->codec_type == AVMEDIA_TYPE_VIDEO || avctx->codec_type == AVMEDIA_TYPE_AUDIO)
1461 && avctx->bit_rate>0 && avctx->bit_rate<1000) {
1462 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);
1465 if (!avctx->rc_initial_buffer_occupancy)
1466 avctx->rc_initial_buffer_occupancy = avctx->rc_buffer_size * 3 / 4;
1469 avctx->pts_correction_num_faulty_pts =
1470 avctx->pts_correction_num_faulty_dts = 0;
1471 avctx->pts_correction_last_pts =
1472 avctx->pts_correction_last_dts = INT64_MIN;
1474 if ( !CONFIG_GRAY && avctx->flags & AV_CODEC_FLAG_GRAY
1475 && avctx->codec_descriptor->type == AVMEDIA_TYPE_VIDEO)
1476 av_log(avctx, AV_LOG_WARNING,
1477 "gray decoding requested but not enabled at configuration time\n");
1479 if ( avctx->codec->init && (!(avctx->active_thread_type&FF_THREAD_FRAME)
1480 || avctx->internal->frame_thread_encoder)) {
1481 ret = avctx->codec->init(avctx);
1489 #if FF_API_AUDIOENC_DELAY
1490 if (av_codec_is_encoder(avctx->codec))
1491 avctx->delay = avctx->initial_padding;
1494 if (av_codec_is_decoder(avctx->codec)) {
1495 if (!avctx->bit_rate)
1496 avctx->bit_rate = get_bit_rate(avctx);
1497 /* validate channel layout from the decoder */
1498 if (avctx->channel_layout) {
1499 int channels = av_get_channel_layout_nb_channels(avctx->channel_layout);
1500 if (!avctx->channels)
1501 avctx->channels = channels;
1502 else if (channels != avctx->channels) {
1504 av_get_channel_layout_string(buf, sizeof(buf), -1, avctx->channel_layout);
1505 av_log(avctx, AV_LOG_WARNING,
1506 "Channel layout '%s' with %d channels does not match specified number of channels %d: "
1507 "ignoring specified channel layout\n",
1508 buf, channels, avctx->channels);
1509 avctx->channel_layout = 0;
1512 if (avctx->channels && avctx->channels < 0 ||
1513 avctx->channels > FF_SANE_NB_CHANNELS) {
1514 ret = AVERROR(EINVAL);
1517 if (avctx->sub_charenc) {
1518 if (avctx->codec_type != AVMEDIA_TYPE_SUBTITLE) {
1519 av_log(avctx, AV_LOG_ERROR, "Character encoding is only "
1520 "supported with subtitles codecs\n");
1521 ret = AVERROR(EINVAL);
1523 } else if (avctx->codec_descriptor->props & AV_CODEC_PROP_BITMAP_SUB) {
1524 av_log(avctx, AV_LOG_WARNING, "Codec '%s' is bitmap-based, "
1525 "subtitles character encoding will be ignored\n",
1526 avctx->codec_descriptor->name);
1527 avctx->sub_charenc_mode = FF_SUB_CHARENC_MODE_DO_NOTHING;
1529 /* input character encoding is set for a text based subtitle
1530 * codec at this point */
1531 if (avctx->sub_charenc_mode == FF_SUB_CHARENC_MODE_AUTOMATIC)
1532 avctx->sub_charenc_mode = FF_SUB_CHARENC_MODE_PRE_DECODER;
1534 if (avctx->sub_charenc_mode == FF_SUB_CHARENC_MODE_PRE_DECODER) {
1536 iconv_t cd = iconv_open("UTF-8", avctx->sub_charenc);
1537 if (cd == (iconv_t)-1) {
1538 ret = AVERROR(errno);
1539 av_log(avctx, AV_LOG_ERROR, "Unable to open iconv context "
1540 "with input character encoding \"%s\"\n", avctx->sub_charenc);
1545 av_log(avctx, AV_LOG_ERROR, "Character encoding subtitles "
1546 "conversion needs a libavcodec built with iconv support "
1547 "for this codec\n");
1548 ret = AVERROR(ENOSYS);
1555 #if FF_API_AVCTX_TIMEBASE
1556 if (avctx->framerate.num > 0 && avctx->framerate.den > 0)
1557 avctx->time_base = av_inv_q(av_mul_q(avctx->framerate, (AVRational){avctx->ticks_per_frame, 1}));
1560 if (codec->priv_data_size > 0 && avctx->priv_data && codec->priv_class) {
1561 av_assert0(*(const AVClass **)avctx->priv_data == codec->priv_class);
1565 ff_unlock_avcodec(codec);
1567 av_dict_free(options);
1574 (avctx->codec->caps_internal & FF_CODEC_CAP_INIT_CLEANUP))
1575 avctx->codec->close(avctx);
1577 if (codec->priv_class && codec->priv_data_size)
1578 av_opt_free(avctx->priv_data);
1581 #if FF_API_CODED_FRAME
1582 FF_DISABLE_DEPRECATION_WARNINGS
1583 av_frame_free(&avctx->coded_frame);
1584 FF_ENABLE_DEPRECATION_WARNINGS
1588 av_freep(&avctx->priv_data);
1589 if (avctx->internal) {
1590 av_frame_free(&avctx->internal->to_free);
1591 av_freep(&avctx->internal->pool);
1593 av_freep(&avctx->internal);
1594 avctx->codec = NULL;
1598 int ff_alloc_packet2(AVCodecContext *avctx, AVPacket *avpkt, int64_t size, int64_t min_size)
1600 if (avpkt->size < 0) {
1601 av_log(avctx, AV_LOG_ERROR, "Invalid negative user packet size %d\n", avpkt->size);
1602 return AVERROR(EINVAL);
1604 if (size < 0 || size > INT_MAX - AV_INPUT_BUFFER_PADDING_SIZE) {
1605 av_log(avctx, AV_LOG_ERROR, "Invalid minimum required packet size %"PRId64" (max allowed is %d)\n",
1606 size, INT_MAX - AV_INPUT_BUFFER_PADDING_SIZE);
1607 return AVERROR(EINVAL);
1610 if (avctx && 2*min_size < size) { // FIXME The factor needs to be finetuned
1611 av_assert0(!avpkt->data || avpkt->data != avctx->internal->byte_buffer);
1612 if (!avpkt->data || avpkt->size < size) {
1613 av_fast_padded_malloc(&avctx->internal->byte_buffer, &avctx->internal->byte_buffer_size, size);
1614 avpkt->data = avctx->internal->byte_buffer;
1615 avpkt->size = avctx->internal->byte_buffer_size;
1620 AVBufferRef *buf = avpkt->buf;
1622 if (avpkt->size < size) {
1623 av_log(avctx, AV_LOG_ERROR, "User packet is too small (%d < %"PRId64")\n", avpkt->size, size);
1624 return AVERROR(EINVAL);
1627 av_init_packet(avpkt);
1632 int ret = av_new_packet(avpkt, size);
1634 av_log(avctx, AV_LOG_ERROR, "Failed to allocate packet of size %"PRId64"\n", size);
1639 int ff_alloc_packet(AVPacket *avpkt, int size)
1641 return ff_alloc_packet2(NULL, avpkt, size, 0);
1645 * Pad last frame with silence.
1647 static int pad_last_frame(AVCodecContext *s, AVFrame **dst, const AVFrame *src)
1649 AVFrame *frame = NULL;
1652 if (!(frame = av_frame_alloc()))
1653 return AVERROR(ENOMEM);
1655 frame->format = src->format;
1656 frame->channel_layout = src->channel_layout;
1657 av_frame_set_channels(frame, av_frame_get_channels(src));
1658 frame->nb_samples = s->frame_size;
1659 ret = av_frame_get_buffer(frame, 32);
1663 ret = av_frame_copy_props(frame, src);
1667 if ((ret = av_samples_copy(frame->extended_data, src->extended_data, 0, 0,
1668 src->nb_samples, s->channels, s->sample_fmt)) < 0)
1670 if ((ret = av_samples_set_silence(frame->extended_data, src->nb_samples,
1671 frame->nb_samples - src->nb_samples,
1672 s->channels, s->sample_fmt)) < 0)
1680 av_frame_free(&frame);
1684 int attribute_align_arg avcodec_encode_audio2(AVCodecContext *avctx,
1686 const AVFrame *frame,
1687 int *got_packet_ptr)
1689 AVFrame *extended_frame = NULL;
1690 AVFrame *padded_frame = NULL;
1692 AVPacket user_pkt = *avpkt;
1693 int needs_realloc = !user_pkt.data;
1695 *got_packet_ptr = 0;
1697 if (!(avctx->codec->capabilities & AV_CODEC_CAP_DELAY) && !frame) {
1698 av_packet_unref(avpkt);
1699 av_init_packet(avpkt);
1703 /* ensure that extended_data is properly set */
1704 if (frame && !frame->extended_data) {
1705 if (av_sample_fmt_is_planar(avctx->sample_fmt) &&
1706 avctx->channels > AV_NUM_DATA_POINTERS) {
1707 av_log(avctx, AV_LOG_ERROR, "Encoding to a planar sample format, "
1708 "with more than %d channels, but extended_data is not set.\n",
1709 AV_NUM_DATA_POINTERS);
1710 return AVERROR(EINVAL);
1712 av_log(avctx, AV_LOG_WARNING, "extended_data is not set.\n");
1714 extended_frame = av_frame_alloc();
1715 if (!extended_frame)
1716 return AVERROR(ENOMEM);
1718 memcpy(extended_frame, frame, sizeof(AVFrame));
1719 extended_frame->extended_data = extended_frame->data;
1720 frame = extended_frame;
1723 /* extract audio service type metadata */
1725 AVFrameSideData *sd = av_frame_get_side_data(frame, AV_FRAME_DATA_AUDIO_SERVICE_TYPE);
1726 if (sd && sd->size >= sizeof(enum AVAudioServiceType))
1727 avctx->audio_service_type = *(enum AVAudioServiceType*)sd->data;
1730 /* check for valid frame size */
1732 if (avctx->codec->capabilities & AV_CODEC_CAP_SMALL_LAST_FRAME) {
1733 if (frame->nb_samples > avctx->frame_size) {
1734 av_log(avctx, AV_LOG_ERROR, "more samples than frame size (avcodec_encode_audio2)\n");
1735 ret = AVERROR(EINVAL);
1738 } else if (!(avctx->codec->capabilities & AV_CODEC_CAP_VARIABLE_FRAME_SIZE)) {
1739 if (frame->nb_samples < avctx->frame_size &&
1740 !avctx->internal->last_audio_frame) {
1741 ret = pad_last_frame(avctx, &padded_frame, frame);
1745 frame = padded_frame;
1746 avctx->internal->last_audio_frame = 1;
1749 if (frame->nb_samples != avctx->frame_size) {
1750 av_log(avctx, AV_LOG_ERROR, "nb_samples (%d) != frame_size (%d) (avcodec_encode_audio2)\n", frame->nb_samples, avctx->frame_size);
1751 ret = AVERROR(EINVAL);
1757 av_assert0(avctx->codec->encode2);
1759 ret = avctx->codec->encode2(avctx, avpkt, frame, got_packet_ptr);
1761 if (*got_packet_ptr) {
1762 if (!(avctx->codec->capabilities & AV_CODEC_CAP_DELAY)) {
1763 if (avpkt->pts == AV_NOPTS_VALUE)
1764 avpkt->pts = frame->pts;
1765 if (!avpkt->duration)
1766 avpkt->duration = ff_samples_to_time_base(avctx,
1769 avpkt->dts = avpkt->pts;
1774 if (avpkt->data && avpkt->data == avctx->internal->byte_buffer) {
1776 if (user_pkt.data) {
1777 if (user_pkt.size >= avpkt->size) {
1778 memcpy(user_pkt.data, avpkt->data, avpkt->size);
1780 av_log(avctx, AV_LOG_ERROR, "Provided packet is too small, needs to be %d\n", avpkt->size);
1781 avpkt->size = user_pkt.size;
1784 avpkt->buf = user_pkt.buf;
1785 avpkt->data = user_pkt.data;
1787 if (av_dup_packet(avpkt) < 0) {
1788 ret = AVERROR(ENOMEM);
1794 if (needs_realloc && avpkt->data) {
1795 ret = av_buffer_realloc(&avpkt->buf, avpkt->size + AV_INPUT_BUFFER_PADDING_SIZE);
1797 avpkt->data = avpkt->buf->data;
1800 avctx->frame_number++;
1803 if (ret < 0 || !*got_packet_ptr) {
1804 av_packet_unref(avpkt);
1805 av_init_packet(avpkt);
1809 /* NOTE: if we add any audio encoders which output non-keyframe packets,
1810 * this needs to be moved to the encoders, but for now we can do it
1811 * here to simplify things */
1812 avpkt->flags |= AV_PKT_FLAG_KEY;
1815 av_frame_free(&padded_frame);
1816 av_free(extended_frame);
1818 #if FF_API_AUDIOENC_DELAY
1819 avctx->delay = avctx->initial_padding;
1825 int attribute_align_arg avcodec_encode_video2(AVCodecContext *avctx,
1827 const AVFrame *frame,
1828 int *got_packet_ptr)
1831 AVPacket user_pkt = *avpkt;
1832 int needs_realloc = !user_pkt.data;
1834 *got_packet_ptr = 0;
1836 if(CONFIG_FRAME_THREAD_ENCODER &&
1837 avctx->internal->frame_thread_encoder && (avctx->active_thread_type&FF_THREAD_FRAME))
1838 return ff_thread_video_encode_frame(avctx, avpkt, frame, got_packet_ptr);
1840 if ((avctx->flags&AV_CODEC_FLAG_PASS1) && avctx->stats_out)
1841 avctx->stats_out[0] = '\0';
1843 if (!(avctx->codec->capabilities & AV_CODEC_CAP_DELAY) && !frame) {
1844 av_packet_unref(avpkt);
1845 av_init_packet(avpkt);
1850 if (av_image_check_size(avctx->width, avctx->height, 0, avctx))
1851 return AVERROR(EINVAL);
1853 if (frame && frame->format == AV_PIX_FMT_NONE)
1854 av_log(avctx, AV_LOG_WARNING, "AVFrame.format is not set\n");
1855 if (frame && (frame->width == 0 || frame->height == 0))
1856 av_log(avctx, AV_LOG_WARNING, "AVFrame.width or height is not set\n");
1858 av_assert0(avctx->codec->encode2);
1860 ret = avctx->codec->encode2(avctx, avpkt, frame, got_packet_ptr);
1861 av_assert0(ret <= 0);
1863 if (avpkt->data && avpkt->data == avctx->internal->byte_buffer) {
1865 if (user_pkt.data) {
1866 if (user_pkt.size >= avpkt->size) {
1867 memcpy(user_pkt.data, avpkt->data, avpkt->size);
1869 av_log(avctx, AV_LOG_ERROR, "Provided packet is too small, needs to be %d\n", avpkt->size);
1870 avpkt->size = user_pkt.size;
1873 avpkt->buf = user_pkt.buf;
1874 avpkt->data = user_pkt.data;
1876 if (av_dup_packet(avpkt) < 0) {
1877 ret = AVERROR(ENOMEM);
1883 if (!*got_packet_ptr)
1885 else if (!(avctx->codec->capabilities & AV_CODEC_CAP_DELAY))
1886 avpkt->pts = avpkt->dts = frame->pts;
1888 if (needs_realloc && avpkt->data) {
1889 ret = av_buffer_realloc(&avpkt->buf, avpkt->size + AV_INPUT_BUFFER_PADDING_SIZE);
1891 avpkt->data = avpkt->buf->data;
1894 avctx->frame_number++;
1897 if (ret < 0 || !*got_packet_ptr)
1898 av_packet_unref(avpkt);
1904 int avcodec_encode_subtitle(AVCodecContext *avctx, uint8_t *buf, int buf_size,
1905 const AVSubtitle *sub)
1908 if (sub->start_display_time) {
1909 av_log(avctx, AV_LOG_ERROR, "start_display_time must be 0.\n");
1913 ret = avctx->codec->encode_sub(avctx, buf, buf_size, sub);
1914 avctx->frame_number++;
1919 * Attempt to guess proper monotonic timestamps for decoded video frames
1920 * which might have incorrect times. Input timestamps may wrap around, in
1921 * which case the output will as well.
1923 * @param pts the pts field of the decoded AVPacket, as passed through
1925 * @param dts the dts field of the decoded AVPacket
1926 * @return one of the input values, may be AV_NOPTS_VALUE
1928 static int64_t guess_correct_pts(AVCodecContext *ctx,
1929 int64_t reordered_pts, int64_t dts)
1931 int64_t pts = AV_NOPTS_VALUE;
1933 if (dts != AV_NOPTS_VALUE) {
1934 ctx->pts_correction_num_faulty_dts += dts <= ctx->pts_correction_last_dts;
1935 ctx->pts_correction_last_dts = dts;
1936 } else if (reordered_pts != AV_NOPTS_VALUE)
1937 ctx->pts_correction_last_dts = reordered_pts;
1939 if (reordered_pts != AV_NOPTS_VALUE) {
1940 ctx->pts_correction_num_faulty_pts += reordered_pts <= ctx->pts_correction_last_pts;
1941 ctx->pts_correction_last_pts = reordered_pts;
1942 } else if(dts != AV_NOPTS_VALUE)
1943 ctx->pts_correction_last_pts = dts;
1945 if ((ctx->pts_correction_num_faulty_pts<=ctx->pts_correction_num_faulty_dts || dts == AV_NOPTS_VALUE)
1946 && reordered_pts != AV_NOPTS_VALUE)
1947 pts = reordered_pts;
1954 static int apply_param_change(AVCodecContext *avctx, AVPacket *avpkt)
1957 const uint8_t *data;
1961 data = av_packet_get_side_data(avpkt, AV_PKT_DATA_PARAM_CHANGE, &size);
1965 if (!(avctx->codec->capabilities & AV_CODEC_CAP_PARAM_CHANGE)) {
1966 av_log(avctx, AV_LOG_ERROR, "This decoder does not support parameter "
1967 "changes, but PARAM_CHANGE side data was sent to it.\n");
1968 return AVERROR(EINVAL);
1974 flags = bytestream_get_le32(&data);
1977 if (flags & AV_SIDE_DATA_PARAM_CHANGE_CHANNEL_COUNT) {
1980 val = bytestream_get_le32(&data);
1981 if (val <= 0 || val > INT_MAX) {
1982 av_log(avctx, AV_LOG_ERROR, "Invalid channel count");
1983 return AVERROR_INVALIDDATA;
1985 avctx->channels = val;
1988 if (flags & AV_SIDE_DATA_PARAM_CHANGE_CHANNEL_LAYOUT) {
1991 avctx->channel_layout = bytestream_get_le64(&data);
1994 if (flags & AV_SIDE_DATA_PARAM_CHANGE_SAMPLE_RATE) {
1997 val = bytestream_get_le32(&data);
1998 if (val <= 0 || val > INT_MAX) {
1999 av_log(avctx, AV_LOG_ERROR, "Invalid sample rate");
2000 return AVERROR_INVALIDDATA;
2002 avctx->sample_rate = val;
2005 if (flags & AV_SIDE_DATA_PARAM_CHANGE_DIMENSIONS) {
2008 avctx->width = bytestream_get_le32(&data);
2009 avctx->height = bytestream_get_le32(&data);
2011 ret = ff_set_dimensions(avctx, avctx->width, avctx->height);
2018 av_log(avctx, AV_LOG_ERROR, "PARAM_CHANGE side data too small.\n");
2019 return AVERROR_INVALIDDATA;
2022 static int unrefcount_frame(AVCodecInternal *avci, AVFrame *frame)
2026 /* move the original frame to our backup */
2027 av_frame_unref(avci->to_free);
2028 av_frame_move_ref(avci->to_free, frame);
2030 /* now copy everything except the AVBufferRefs back
2031 * note that we make a COPY of the side data, so calling av_frame_free() on
2032 * the caller's frame will work properly */
2033 ret = av_frame_copy_props(frame, avci->to_free);
2037 memcpy(frame->data, avci->to_free->data, sizeof(frame->data));
2038 memcpy(frame->linesize, avci->to_free->linesize, sizeof(frame->linesize));
2039 if (avci->to_free->extended_data != avci->to_free->data) {
2040 int planes = av_frame_get_channels(avci->to_free);
2041 int size = planes * sizeof(*frame->extended_data);
2044 av_frame_unref(frame);
2048 frame->extended_data = av_malloc(size);
2049 if (!frame->extended_data) {
2050 av_frame_unref(frame);
2051 return AVERROR(ENOMEM);
2053 memcpy(frame->extended_data, avci->to_free->extended_data,
2056 frame->extended_data = frame->data;
2058 frame->format = avci->to_free->format;
2059 frame->width = avci->to_free->width;
2060 frame->height = avci->to_free->height;
2061 frame->channel_layout = avci->to_free->channel_layout;
2062 frame->nb_samples = avci->to_free->nb_samples;
2063 av_frame_set_channels(frame, av_frame_get_channels(avci->to_free));
2068 int attribute_align_arg avcodec_decode_video2(AVCodecContext *avctx, AVFrame *picture,
2069 int *got_picture_ptr,
2070 const AVPacket *avpkt)
2072 AVCodecInternal *avci = avctx->internal;
2074 // copy to ensure we do not change avpkt
2075 AVPacket tmp = *avpkt;
2078 return AVERROR(EINVAL);
2079 if (avctx->codec->type != AVMEDIA_TYPE_VIDEO) {
2080 av_log(avctx, AV_LOG_ERROR, "Invalid media type for video\n");
2081 return AVERROR(EINVAL);
2084 *got_picture_ptr = 0;
2085 if ((avctx->coded_width || avctx->coded_height) && av_image_check_size(avctx->coded_width, avctx->coded_height, 0, avctx))
2086 return AVERROR(EINVAL);
2088 av_frame_unref(picture);
2090 if ((avctx->codec->capabilities & AV_CODEC_CAP_DELAY) || avpkt->size ||
2091 (avctx->active_thread_type & FF_THREAD_FRAME)) {
2092 int did_split = av_packet_split_side_data(&tmp);
2093 ret = apply_param_change(avctx, &tmp);
2095 av_log(avctx, AV_LOG_ERROR, "Error applying parameter changes.\n");
2096 if (avctx->err_recognition & AV_EF_EXPLODE)
2100 avctx->internal->pkt = &tmp;
2101 if (HAVE_THREADS && avctx->active_thread_type & FF_THREAD_FRAME)
2102 ret = ff_thread_decode_frame(avctx, picture, got_picture_ptr,
2105 ret = avctx->codec->decode(avctx, picture, got_picture_ptr,
2107 if (!(avctx->codec->caps_internal & FF_CODEC_CAP_SETS_PKT_DTS))
2108 picture->pkt_dts = avpkt->dts;
2110 if(!avctx->has_b_frames){
2111 av_frame_set_pkt_pos(picture, avpkt->pos);
2113 //FIXME these should be under if(!avctx->has_b_frames)
2114 /* get_buffer is supposed to set frame parameters */
2115 if (!(avctx->codec->capabilities & AV_CODEC_CAP_DR1)) {
2116 if (!picture->sample_aspect_ratio.num) picture->sample_aspect_ratio = avctx->sample_aspect_ratio;
2117 if (!picture->width) picture->width = avctx->width;
2118 if (!picture->height) picture->height = avctx->height;
2119 if (picture->format == AV_PIX_FMT_NONE) picture->format = avctx->pix_fmt;
2124 emms_c(); //needed to avoid an emms_c() call before every return;
2126 avctx->internal->pkt = NULL;
2128 av_packet_free_side_data(&tmp);
2133 if (*got_picture_ptr) {
2134 if (!avctx->refcounted_frames) {
2135 int err = unrefcount_frame(avci, picture);
2140 avctx->frame_number++;
2141 av_frame_set_best_effort_timestamp(picture,
2142 guess_correct_pts(avctx,
2146 av_frame_unref(picture);
2150 /* many decoders assign whole AVFrames, thus overwriting extended_data;
2151 * make sure it's set correctly */
2152 av_assert0(!picture->extended_data || picture->extended_data == picture->data);
2154 #if FF_API_AVCTX_TIMEBASE
2155 if (avctx->framerate.num > 0 && avctx->framerate.den > 0)
2156 avctx->time_base = av_inv_q(av_mul_q(avctx->framerate, (AVRational){avctx->ticks_per_frame, 1}));
2162 int attribute_align_arg avcodec_decode_audio4(AVCodecContext *avctx,
2165 const AVPacket *avpkt)
2167 AVCodecInternal *avci = avctx->internal;
2172 if (!avpkt->data && avpkt->size) {
2173 av_log(avctx, AV_LOG_ERROR, "invalid packet: NULL data, size != 0\n");
2174 return AVERROR(EINVAL);
2177 return AVERROR(EINVAL);
2178 if (avctx->codec->type != AVMEDIA_TYPE_AUDIO) {
2179 av_log(avctx, AV_LOG_ERROR, "Invalid media type for audio\n");
2180 return AVERROR(EINVAL);
2183 av_frame_unref(frame);
2185 if ((avctx->codec->capabilities & AV_CODEC_CAP_DELAY) || avpkt->size || (avctx->active_thread_type & FF_THREAD_FRAME)) {
2188 uint32_t discard_padding = 0;
2189 uint8_t skip_reason = 0;
2190 uint8_t discard_reason = 0;
2191 // copy to ensure we do not change avpkt
2192 AVPacket tmp = *avpkt;
2193 int did_split = av_packet_split_side_data(&tmp);
2194 ret = apply_param_change(avctx, &tmp);
2196 av_log(avctx, AV_LOG_ERROR, "Error applying parameter changes.\n");
2197 if (avctx->err_recognition & AV_EF_EXPLODE)
2201 avctx->internal->pkt = &tmp;
2202 if (HAVE_THREADS && avctx->active_thread_type & FF_THREAD_FRAME)
2203 ret = ff_thread_decode_frame(avctx, frame, got_frame_ptr, &tmp);
2205 ret = avctx->codec->decode(avctx, frame, got_frame_ptr, &tmp);
2206 av_assert0(ret <= tmp.size);
2207 frame->pkt_dts = avpkt->dts;
2209 if (ret >= 0 && *got_frame_ptr) {
2210 avctx->frame_number++;
2211 av_frame_set_best_effort_timestamp(frame,
2212 guess_correct_pts(avctx,
2215 if (frame->format == AV_SAMPLE_FMT_NONE)
2216 frame->format = avctx->sample_fmt;
2217 if (!frame->channel_layout)
2218 frame->channel_layout = avctx->channel_layout;
2219 if (!av_frame_get_channels(frame))
2220 av_frame_set_channels(frame, avctx->channels);
2221 if (!frame->sample_rate)
2222 frame->sample_rate = avctx->sample_rate;
2225 side= av_packet_get_side_data(avctx->internal->pkt, AV_PKT_DATA_SKIP_SAMPLES, &side_size);
2226 if(side && side_size>=10) {
2227 avctx->internal->skip_samples = AV_RL32(side);
2228 discard_padding = AV_RL32(side + 4);
2229 av_log(avctx, AV_LOG_DEBUG, "skip %d / discard %d samples due to side data\n",
2230 avctx->internal->skip_samples, (int)discard_padding);
2231 skip_reason = AV_RL8(side + 8);
2232 discard_reason = AV_RL8(side + 9);
2234 if (avctx->internal->skip_samples && *got_frame_ptr &&
2235 !(avctx->flags2 & AV_CODEC_FLAG2_SKIP_MANUAL)) {
2236 if(frame->nb_samples <= avctx->internal->skip_samples){
2238 avctx->internal->skip_samples -= frame->nb_samples;
2239 av_log(avctx, AV_LOG_DEBUG, "skip whole frame, skip left: %d\n",
2240 avctx->internal->skip_samples);
2242 av_samples_copy(frame->extended_data, frame->extended_data, 0, avctx->internal->skip_samples,
2243 frame->nb_samples - avctx->internal->skip_samples, avctx->channels, frame->format);
2244 if(avctx->pkt_timebase.num && avctx->sample_rate) {
2245 int64_t diff_ts = av_rescale_q(avctx->internal->skip_samples,
2246 (AVRational){1, avctx->sample_rate},
2247 avctx->pkt_timebase);
2248 if(frame->pkt_pts!=AV_NOPTS_VALUE)
2249 frame->pkt_pts += diff_ts;
2250 if(frame->pkt_dts!=AV_NOPTS_VALUE)
2251 frame->pkt_dts += diff_ts;
2252 if (av_frame_get_pkt_duration(frame) >= diff_ts)
2253 av_frame_set_pkt_duration(frame, av_frame_get_pkt_duration(frame) - diff_ts);
2255 av_log(avctx, AV_LOG_WARNING, "Could not update timestamps for skipped samples.\n");
2257 av_log(avctx, AV_LOG_DEBUG, "skip %d/%d samples\n",
2258 avctx->internal->skip_samples, frame->nb_samples);
2259 frame->nb_samples -= avctx->internal->skip_samples;
2260 avctx->internal->skip_samples = 0;
2264 if (discard_padding > 0 && discard_padding <= frame->nb_samples && *got_frame_ptr &&
2265 !(avctx->flags2 & AV_CODEC_FLAG2_SKIP_MANUAL)) {
2266 if (discard_padding == frame->nb_samples) {
2269 if(avctx->pkt_timebase.num && avctx->sample_rate) {
2270 int64_t diff_ts = av_rescale_q(frame->nb_samples - discard_padding,
2271 (AVRational){1, avctx->sample_rate},
2272 avctx->pkt_timebase);
2273 if (av_frame_get_pkt_duration(frame) >= diff_ts)
2274 av_frame_set_pkt_duration(frame, av_frame_get_pkt_duration(frame) - diff_ts);
2276 av_log(avctx, AV_LOG_WARNING, "Could not update timestamps for discarded samples.\n");
2278 av_log(avctx, AV_LOG_DEBUG, "discard %d/%d samples\n",
2279 (int)discard_padding, frame->nb_samples);
2280 frame->nb_samples -= discard_padding;
2284 if ((avctx->flags2 & AV_CODEC_FLAG2_SKIP_MANUAL) && *got_frame_ptr) {
2285 AVFrameSideData *fside = av_frame_new_side_data(frame, AV_FRAME_DATA_SKIP_SAMPLES, 10);
2287 AV_WL32(fside->data, avctx->internal->skip_samples);
2288 AV_WL32(fside->data + 4, discard_padding);
2289 AV_WL8(fside->data + 8, skip_reason);
2290 AV_WL8(fside->data + 9, discard_reason);
2291 avctx->internal->skip_samples = 0;
2295 avctx->internal->pkt = NULL;
2297 av_packet_free_side_data(&tmp);
2302 if (ret >= 0 && *got_frame_ptr) {
2303 if (!avctx->refcounted_frames) {
2304 int err = unrefcount_frame(avci, frame);
2309 av_frame_unref(frame);
2315 #define UTF8_MAX_BYTES 4 /* 5 and 6 bytes sequences should not be used */
2316 static int recode_subtitle(AVCodecContext *avctx,
2317 AVPacket *outpkt, const AVPacket *inpkt)
2320 iconv_t cd = (iconv_t)-1;
2327 if (avctx->sub_charenc_mode != FF_SUB_CHARENC_MODE_PRE_DECODER || inpkt->size == 0)
2331 cd = iconv_open("UTF-8", avctx->sub_charenc);
2332 av_assert0(cd != (iconv_t)-1);
2337 if (inl >= INT_MAX / UTF8_MAX_BYTES - AV_INPUT_BUFFER_PADDING_SIZE) {
2338 av_log(avctx, AV_LOG_ERROR, "Subtitles packet is too big for recoding\n");
2339 ret = AVERROR(ENOMEM);
2343 ret = av_new_packet(&tmp, inl * UTF8_MAX_BYTES);
2346 outpkt->buf = tmp.buf;
2347 outpkt->data = tmp.data;
2348 outpkt->size = tmp.size;
2349 outb = outpkt->data;
2350 outl = outpkt->size;
2352 if (iconv(cd, &inb, &inl, &outb, &outl) == (size_t)-1 ||
2353 iconv(cd, NULL, NULL, &outb, &outl) == (size_t)-1 ||
2354 outl >= outpkt->size || inl != 0) {
2355 ret = FFMIN(AVERROR(errno), -1);
2356 av_log(avctx, AV_LOG_ERROR, "Unable to recode subtitle event \"%s\" "
2357 "from %s to UTF-8\n", inpkt->data, avctx->sub_charenc);
2358 av_packet_unref(&tmp);
2361 outpkt->size -= outl;
2362 memset(outpkt->data + outpkt->size, 0, outl);
2365 if (cd != (iconv_t)-1)
2369 av_log(avctx, AV_LOG_ERROR, "requesting subtitles recoding without iconv");
2370 return AVERROR(EINVAL);
2374 static int utf8_check(const uint8_t *str)
2376 const uint8_t *byte;
2377 uint32_t codepoint, min;
2381 GET_UTF8(codepoint, *(byte++), return 0;);
2382 min = byte - str == 1 ? 0 : byte - str == 2 ? 0x80 :
2383 1 << (5 * (byte - str) - 4);
2384 if (codepoint < min || codepoint >= 0x110000 ||
2385 codepoint == 0xFFFE /* BOM */ ||
2386 codepoint >= 0xD800 && codepoint <= 0xDFFF /* surrogates */)
2393 int avcodec_decode_subtitle2(AVCodecContext *avctx, AVSubtitle *sub,
2399 if (!avpkt->data && avpkt->size) {
2400 av_log(avctx, AV_LOG_ERROR, "invalid packet: NULL data, size != 0\n");
2401 return AVERROR(EINVAL);
2404 return AVERROR(EINVAL);
2405 if (avctx->codec->type != AVMEDIA_TYPE_SUBTITLE) {
2406 av_log(avctx, AV_LOG_ERROR, "Invalid media type for subtitles\n");
2407 return AVERROR(EINVAL);
2411 get_subtitle_defaults(sub);
2413 if ((avctx->codec->capabilities & AV_CODEC_CAP_DELAY) || avpkt->size) {
2414 AVPacket pkt_recoded;
2415 AVPacket tmp = *avpkt;
2416 int did_split = av_packet_split_side_data(&tmp);
2417 //apply_param_change(avctx, &tmp);
2420 /* FFMIN() prevents overflow in case the packet wasn't allocated with
2422 * If the side data is smaller than the buffer padding size, the
2423 * remaining bytes should have already been filled with zeros by the
2424 * original packet allocation anyway. */
2425 memset(tmp.data + tmp.size, 0,
2426 FFMIN(avpkt->size - tmp.size, AV_INPUT_BUFFER_PADDING_SIZE));
2430 ret = recode_subtitle(avctx, &pkt_recoded, &tmp);
2434 avctx->internal->pkt = &pkt_recoded;
2436 if (avctx->pkt_timebase.den && avpkt->pts != AV_NOPTS_VALUE)
2437 sub->pts = av_rescale_q(avpkt->pts,
2438 avctx->pkt_timebase, AV_TIME_BASE_Q);
2439 ret = avctx->codec->decode(avctx, sub, got_sub_ptr, &pkt_recoded);
2440 av_assert1((ret >= 0) >= !!*got_sub_ptr &&
2441 !!*got_sub_ptr >= !!sub->num_rects);
2443 if (sub->num_rects && !sub->end_display_time && avpkt->duration &&
2444 avctx->pkt_timebase.num) {
2445 AVRational ms = { 1, 1000 };
2446 sub->end_display_time = av_rescale_q(avpkt->duration,
2447 avctx->pkt_timebase, ms);
2450 for (i = 0; i < sub->num_rects; i++) {
2451 if (sub->rects[i]->ass && !utf8_check(sub->rects[i]->ass)) {
2452 av_log(avctx, AV_LOG_ERROR,
2453 "Invalid UTF-8 in decoded subtitles text; "
2454 "maybe missing -sub_charenc option\n");
2455 avsubtitle_free(sub);
2456 return AVERROR_INVALIDDATA;
2460 if (tmp.data != pkt_recoded.data) { // did we recode?
2461 /* prevent from destroying side data from original packet */
2462 pkt_recoded.side_data = NULL;
2463 pkt_recoded.side_data_elems = 0;
2465 av_packet_unref(&pkt_recoded);
2467 if (avctx->codec_descriptor->props & AV_CODEC_PROP_BITMAP_SUB)
2469 else if (avctx->codec_descriptor->props & AV_CODEC_PROP_TEXT_SUB)
2471 avctx->internal->pkt = NULL;
2475 av_packet_free_side_data(&tmp);
2481 avctx->frame_number++;
2487 void avsubtitle_free(AVSubtitle *sub)
2491 for (i = 0; i < sub->num_rects; i++) {
2492 av_freep(&sub->rects[i]->data[0]);
2493 av_freep(&sub->rects[i]->data[1]);
2494 av_freep(&sub->rects[i]->data[2]);
2495 av_freep(&sub->rects[i]->data[3]);
2496 av_freep(&sub->rects[i]->text);
2497 av_freep(&sub->rects[i]->ass);
2498 av_freep(&sub->rects[i]);
2501 av_freep(&sub->rects);
2503 memset(sub, 0, sizeof(AVSubtitle));
2506 av_cold int avcodec_close(AVCodecContext *avctx)
2511 if (avcodec_is_open(avctx)) {
2512 FramePool *pool = avctx->internal->pool;
2514 if (CONFIG_FRAME_THREAD_ENCODER &&
2515 avctx->internal->frame_thread_encoder && avctx->thread_count > 1) {
2516 ff_frame_thread_encoder_free(avctx);
2518 if (HAVE_THREADS && avctx->internal->thread_ctx)
2519 ff_thread_free(avctx);
2520 if (avctx->codec && avctx->codec->close)
2521 avctx->codec->close(avctx);
2522 avctx->internal->byte_buffer_size = 0;
2523 av_freep(&avctx->internal->byte_buffer);
2524 av_frame_free(&avctx->internal->to_free);
2525 for (i = 0; i < FF_ARRAY_ELEMS(pool->pools); i++)
2526 av_buffer_pool_uninit(&pool->pools[i]);
2527 av_freep(&avctx->internal->pool);
2529 if (avctx->hwaccel && avctx->hwaccel->uninit)
2530 avctx->hwaccel->uninit(avctx);
2531 av_freep(&avctx->internal->hwaccel_priv_data);
2533 av_freep(&avctx->internal);
2536 if (avctx->priv_data && avctx->codec && avctx->codec->priv_class)
2537 av_opt_free(avctx->priv_data);
2539 av_freep(&avctx->priv_data);
2540 if (av_codec_is_encoder(avctx->codec)) {
2541 av_freep(&avctx->extradata);
2542 #if FF_API_CODED_FRAME
2543 FF_DISABLE_DEPRECATION_WARNINGS
2544 av_frame_free(&avctx->coded_frame);
2545 FF_ENABLE_DEPRECATION_WARNINGS
2548 avctx->codec = NULL;
2549 avctx->active_thread_type = 0;
2554 static enum AVCodecID remap_deprecated_codec_id(enum AVCodecID id)
2557 //This is for future deprecatec codec ids, its empty since
2558 //last major bump but will fill up again over time, please don't remove it
2559 default : return id;
2563 static AVCodec *find_encdec(enum AVCodecID id, int encoder)
2565 AVCodec *p, *experimental = NULL;
2567 id= remap_deprecated_codec_id(id);
2569 if ((encoder ? av_codec_is_encoder(p) : av_codec_is_decoder(p)) &&
2571 if (p->capabilities & AV_CODEC_CAP_EXPERIMENTAL && !experimental) {
2578 return experimental;
2581 AVCodec *avcodec_find_encoder(enum AVCodecID id)
2583 return find_encdec(id, 1);
2586 AVCodec *avcodec_find_encoder_by_name(const char *name)
2593 if (av_codec_is_encoder(p) && strcmp(name, p->name) == 0)
2600 AVCodec *avcodec_find_decoder(enum AVCodecID id)
2602 return find_encdec(id, 0);
2605 AVCodec *avcodec_find_decoder_by_name(const char *name)
2612 if (av_codec_is_decoder(p) && strcmp(name, p->name) == 0)
2619 const char *avcodec_get_name(enum AVCodecID id)
2621 const AVCodecDescriptor *cd;
2624 if (id == AV_CODEC_ID_NONE)
2626 cd = avcodec_descriptor_get(id);
2629 av_log(NULL, AV_LOG_WARNING, "Codec 0x%x is not in the full list.\n", id);
2630 codec = avcodec_find_decoder(id);
2633 codec = avcodec_find_encoder(id);
2636 return "unknown_codec";
2639 size_t av_get_codec_tag_string(char *buf, size_t buf_size, unsigned int codec_tag)
2641 int i, len, ret = 0;
2643 #define TAG_PRINT(x) \
2644 (((x) >= '0' && (x) <= '9') || \
2645 ((x) >= 'a' && (x) <= 'z') || ((x) >= 'A' && (x) <= 'Z') || \
2646 ((x) == '.' || (x) == ' ' || (x) == '-' || (x) == '_'))
2648 for (i = 0; i < 4; i++) {
2649 len = snprintf(buf, buf_size,
2650 TAG_PRINT(codec_tag & 0xFF) ? "%c" : "[%d]", codec_tag & 0xFF);
2652 buf_size = buf_size > len ? buf_size - len : 0;
2659 void avcodec_string(char *buf, int buf_size, AVCodecContext *enc, int encode)
2661 const char *codec_type;
2662 const char *codec_name;
2663 const char *profile = NULL;
2667 AVRational display_aspect_ratio;
2668 const char *separator = enc->dump_separator ? (const char *)enc->dump_separator : ", ";
2670 if (!buf || buf_size <= 0)
2672 codec_type = av_get_media_type_string(enc->codec_type);
2673 codec_name = avcodec_get_name(enc->codec_id);
2674 if (enc->profile != FF_PROFILE_UNKNOWN) {
2678 p = encode ? avcodec_find_encoder(enc->codec_id) :
2679 avcodec_find_decoder(enc->codec_id);
2681 profile = av_get_profile_name(p, enc->profile);
2684 snprintf(buf, buf_size, "%s: %s", codec_type ? codec_type : "unknown",
2686 buf[0] ^= 'a' ^ 'A'; /* first letter in uppercase */
2688 if (enc->codec && strcmp(enc->codec->name, codec_name))
2689 snprintf(buf + strlen(buf), buf_size - strlen(buf), " (%s)", enc->codec->name);
2692 snprintf(buf + strlen(buf), buf_size - strlen(buf), " (%s)", profile);
2693 if ( enc->codec_type == AVMEDIA_TYPE_VIDEO
2694 && av_log_get_level() >= AV_LOG_VERBOSE
2696 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2697 ", %d reference frame%s",
2698 enc->refs, enc->refs > 1 ? "s" : "");
2700 if (enc->codec_tag) {
2702 av_get_codec_tag_string(tag_buf, sizeof(tag_buf), enc->codec_tag);
2703 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2704 " (%s / 0x%04X)", tag_buf, enc->codec_tag);
2707 switch (enc->codec_type) {
2708 case AVMEDIA_TYPE_VIDEO:
2710 char detail[256] = "(";
2712 av_strlcat(buf, separator, buf_size);
2714 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2715 "%s", enc->pix_fmt == AV_PIX_FMT_NONE ? "none" :
2716 av_get_pix_fmt_name(enc->pix_fmt));
2717 if (enc->bits_per_raw_sample && enc->pix_fmt != AV_PIX_FMT_NONE &&
2718 enc->bits_per_raw_sample < av_pix_fmt_desc_get(enc->pix_fmt)->comp[0].depth)
2719 av_strlcatf(detail, sizeof(detail), "%d bpc, ", enc->bits_per_raw_sample);
2720 if (enc->color_range != AVCOL_RANGE_UNSPECIFIED)
2721 av_strlcatf(detail, sizeof(detail), "%s, ",
2722 av_color_range_name(enc->color_range));
2724 if (enc->colorspace != AVCOL_SPC_UNSPECIFIED ||
2725 enc->color_primaries != AVCOL_PRI_UNSPECIFIED ||
2726 enc->color_trc != AVCOL_TRC_UNSPECIFIED) {
2727 if (enc->colorspace != (int)enc->color_primaries ||
2728 enc->colorspace != (int)enc->color_trc) {
2730 av_strlcatf(detail, sizeof(detail), "%s/%s/%s, ",
2731 av_color_space_name(enc->colorspace),
2732 av_color_primaries_name(enc->color_primaries),
2733 av_color_transfer_name(enc->color_trc));
2735 av_strlcatf(detail, sizeof(detail), "%s, ",
2736 av_get_colorspace_name(enc->colorspace));
2739 if (av_log_get_level() >= AV_LOG_DEBUG &&
2740 enc->chroma_sample_location != AVCHROMA_LOC_UNSPECIFIED)
2741 av_strlcatf(detail, sizeof(detail), "%s, ",
2742 av_chroma_location_name(enc->chroma_sample_location));
2744 if (strlen(detail) > 1) {
2745 detail[strlen(detail) - 2] = 0;
2746 av_strlcatf(buf, buf_size, "%s)", detail);
2751 av_strlcat(buf, new_line ? separator : ", ", buf_size);
2753 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2755 enc->width, enc->height);
2757 if (av_log_get_level() >= AV_LOG_VERBOSE &&
2758 (enc->width != enc->coded_width ||
2759 enc->height != enc->coded_height))
2760 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2761 " (%dx%d)", enc->coded_width, enc->coded_height);
2763 if (enc->sample_aspect_ratio.num) {
2764 av_reduce(&display_aspect_ratio.num, &display_aspect_ratio.den,
2765 enc->width * (int64_t)enc->sample_aspect_ratio.num,
2766 enc->height * (int64_t)enc->sample_aspect_ratio.den,
2768 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2769 " [SAR %d:%d DAR %d:%d]",
2770 enc->sample_aspect_ratio.num, enc->sample_aspect_ratio.den,
2771 display_aspect_ratio.num, display_aspect_ratio.den);
2773 if (av_log_get_level() >= AV_LOG_DEBUG) {
2774 int g = av_gcd(enc->time_base.num, enc->time_base.den);
2775 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2777 enc->time_base.num / g, enc->time_base.den / g);
2781 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2782 ", q=%d-%d", enc->qmin, enc->qmax);
2784 if (enc->properties & FF_CODEC_PROPERTY_CLOSED_CAPTIONS)
2785 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2786 ", Closed Captions");
2787 if (enc->properties & FF_CODEC_PROPERTY_LOSSLESS)
2788 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2792 case AVMEDIA_TYPE_AUDIO:
2793 av_strlcat(buf, separator, buf_size);
2795 if (enc->sample_rate) {
2796 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2797 "%d Hz, ", enc->sample_rate);
2799 av_get_channel_layout_string(buf + strlen(buf), buf_size - strlen(buf), enc->channels, enc->channel_layout);
2800 if (enc->sample_fmt != AV_SAMPLE_FMT_NONE) {
2801 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2802 ", %s", av_get_sample_fmt_name(enc->sample_fmt));
2804 if ( enc->bits_per_raw_sample > 0
2805 && enc->bits_per_raw_sample != av_get_bytes_per_sample(enc->sample_fmt) * 8)
2806 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2807 " (%d bit)", enc->bits_per_raw_sample);
2809 case AVMEDIA_TYPE_DATA:
2810 if (av_log_get_level() >= AV_LOG_DEBUG) {
2811 int g = av_gcd(enc->time_base.num, enc->time_base.den);
2813 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2815 enc->time_base.num / g, enc->time_base.den / g);
2818 case AVMEDIA_TYPE_SUBTITLE:
2820 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2821 ", %dx%d", enc->width, enc->height);
2827 if (enc->flags & AV_CODEC_FLAG_PASS1)
2828 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2830 if (enc->flags & AV_CODEC_FLAG_PASS2)
2831 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2834 bitrate = get_bit_rate(enc);
2836 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2837 ", %"PRId64" kb/s", bitrate / 1000);
2838 } else if (enc->rc_max_rate > 0) {
2839 snprintf(buf + strlen(buf), buf_size - strlen(buf),
2840 ", max. %"PRId64" kb/s", (int64_t)enc->rc_max_rate / 1000);
2844 const char *av_get_profile_name(const AVCodec *codec, int profile)
2847 if (profile == FF_PROFILE_UNKNOWN || !codec->profiles)
2850 for (p = codec->profiles; p->profile != FF_PROFILE_UNKNOWN; p++)
2851 if (p->profile == profile)
2857 unsigned avcodec_version(void)
2859 // av_assert0(AV_CODEC_ID_V410==164);
2860 av_assert0(AV_CODEC_ID_PCM_S8_PLANAR==65563);
2861 av_assert0(AV_CODEC_ID_ADPCM_G722==69660);
2862 // av_assert0(AV_CODEC_ID_BMV_AUDIO==86071);
2863 av_assert0(AV_CODEC_ID_SRT==94216);
2864 av_assert0(LIBAVCODEC_VERSION_MICRO >= 100);
2866 return LIBAVCODEC_VERSION_INT;
2869 const char *avcodec_configuration(void)
2871 return FFMPEG_CONFIGURATION;
2874 const char *avcodec_license(void)
2876 #define LICENSE_PREFIX "libavcodec license: "
2877 return LICENSE_PREFIX FFMPEG_LICENSE + sizeof(LICENSE_PREFIX) - 1;
2880 void avcodec_flush_buffers(AVCodecContext *avctx)
2882 if (HAVE_THREADS && avctx->active_thread_type & FF_THREAD_FRAME)
2883 ff_thread_flush(avctx);
2884 else if (avctx->codec->flush)
2885 avctx->codec->flush(avctx);
2887 avctx->pts_correction_last_pts =
2888 avctx->pts_correction_last_dts = INT64_MIN;
2890 if (!avctx->refcounted_frames)
2891 av_frame_unref(avctx->internal->to_free);
2894 int av_get_exact_bits_per_sample(enum AVCodecID codec_id)
2897 case AV_CODEC_ID_8SVX_EXP:
2898 case AV_CODEC_ID_8SVX_FIB:
2899 case AV_CODEC_ID_ADPCM_CT:
2900 case AV_CODEC_ID_ADPCM_IMA_APC:
2901 case AV_CODEC_ID_ADPCM_IMA_EA_SEAD:
2902 case AV_CODEC_ID_ADPCM_IMA_OKI:
2903 case AV_CODEC_ID_ADPCM_IMA_WS:
2904 case AV_CODEC_ID_ADPCM_G722:
2905 case AV_CODEC_ID_ADPCM_YAMAHA:
2906 case AV_CODEC_ID_ADPCM_AICA:
2908 case AV_CODEC_ID_DSD_LSBF:
2909 case AV_CODEC_ID_DSD_MSBF:
2910 case AV_CODEC_ID_DSD_LSBF_PLANAR:
2911 case AV_CODEC_ID_DSD_MSBF_PLANAR:
2912 case AV_CODEC_ID_PCM_ALAW:
2913 case AV_CODEC_ID_PCM_MULAW:
2914 case AV_CODEC_ID_PCM_S8:
2915 case AV_CODEC_ID_PCM_S8_PLANAR:
2916 case AV_CODEC_ID_PCM_U8:
2917 case AV_CODEC_ID_PCM_ZORK:
2918 case AV_CODEC_ID_SDX2_DPCM:
2920 case AV_CODEC_ID_PCM_S16BE:
2921 case AV_CODEC_ID_PCM_S16BE_PLANAR:
2922 case AV_CODEC_ID_PCM_S16LE:
2923 case AV_CODEC_ID_PCM_S16LE_PLANAR:
2924 case AV_CODEC_ID_PCM_U16BE:
2925 case AV_CODEC_ID_PCM_U16LE:
2927 case AV_CODEC_ID_PCM_S24DAUD:
2928 case AV_CODEC_ID_PCM_S24BE:
2929 case AV_CODEC_ID_PCM_S24LE:
2930 case AV_CODEC_ID_PCM_S24LE_PLANAR:
2931 case AV_CODEC_ID_PCM_U24BE:
2932 case AV_CODEC_ID_PCM_U24LE:
2934 case AV_CODEC_ID_PCM_S32BE:
2935 case AV_CODEC_ID_PCM_S32LE:
2936 case AV_CODEC_ID_PCM_S32LE_PLANAR:
2937 case AV_CODEC_ID_PCM_U32BE:
2938 case AV_CODEC_ID_PCM_U32LE:
2939 case AV_CODEC_ID_PCM_F32BE:
2940 case AV_CODEC_ID_PCM_F32LE:
2942 case AV_CODEC_ID_PCM_F64BE:
2943 case AV_CODEC_ID_PCM_F64LE:
2950 enum AVCodecID av_get_pcm_codec(enum AVSampleFormat fmt, int be)
2952 static const enum AVCodecID map[AV_SAMPLE_FMT_NB][2] = {
2953 [AV_SAMPLE_FMT_U8 ] = { AV_CODEC_ID_PCM_U8, AV_CODEC_ID_PCM_U8 },
2954 [AV_SAMPLE_FMT_S16 ] = { AV_CODEC_ID_PCM_S16LE, AV_CODEC_ID_PCM_S16BE },
2955 [AV_SAMPLE_FMT_S32 ] = { AV_CODEC_ID_PCM_S32LE, AV_CODEC_ID_PCM_S32BE },
2956 [AV_SAMPLE_FMT_FLT ] = { AV_CODEC_ID_PCM_F32LE, AV_CODEC_ID_PCM_F32BE },
2957 [AV_SAMPLE_FMT_DBL ] = { AV_CODEC_ID_PCM_F64LE, AV_CODEC_ID_PCM_F64BE },
2958 [AV_SAMPLE_FMT_U8P ] = { AV_CODEC_ID_PCM_U8, AV_CODEC_ID_PCM_U8 },
2959 [AV_SAMPLE_FMT_S16P] = { AV_CODEC_ID_PCM_S16LE, AV_CODEC_ID_PCM_S16BE },
2960 [AV_SAMPLE_FMT_S32P] = { AV_CODEC_ID_PCM_S32LE, AV_CODEC_ID_PCM_S32BE },
2961 [AV_SAMPLE_FMT_FLTP] = { AV_CODEC_ID_PCM_F32LE, AV_CODEC_ID_PCM_F32BE },
2962 [AV_SAMPLE_FMT_DBLP] = { AV_CODEC_ID_PCM_F64LE, AV_CODEC_ID_PCM_F64BE },
2964 if (fmt < 0 || fmt >= AV_SAMPLE_FMT_NB)
2965 return AV_CODEC_ID_NONE;
2966 if (be < 0 || be > 1)
2968 return map[fmt][be];
2971 int av_get_bits_per_sample(enum AVCodecID codec_id)
2974 case AV_CODEC_ID_ADPCM_SBPRO_2:
2976 case AV_CODEC_ID_ADPCM_SBPRO_3:
2978 case AV_CODEC_ID_ADPCM_SBPRO_4:
2979 case AV_CODEC_ID_ADPCM_IMA_WAV:
2980 case AV_CODEC_ID_ADPCM_IMA_QT:
2981 case AV_CODEC_ID_ADPCM_SWF:
2982 case AV_CODEC_ID_ADPCM_MS:
2985 return av_get_exact_bits_per_sample(codec_id);
2989 int av_get_audio_frame_duration(AVCodecContext *avctx, int frame_bytes)
2991 int id, sr, ch, ba, tag, bps;
2993 id = avctx->codec_id;
2994 sr = avctx->sample_rate;
2995 ch = avctx->channels;
2996 ba = avctx->block_align;
2997 tag = avctx->codec_tag;
2998 bps = av_get_exact_bits_per_sample(avctx->codec_id);
3000 /* codecs with an exact constant bits per sample */
3001 if (bps > 0 && ch > 0 && frame_bytes > 0 && ch < 32768 && bps < 32768)
3002 return (frame_bytes * 8LL) / (bps * ch);
3003 bps = avctx->bits_per_coded_sample;
3005 /* codecs with a fixed packet duration */
3007 case AV_CODEC_ID_ADPCM_ADX: return 32;
3008 case AV_CODEC_ID_ADPCM_IMA_QT: return 64;
3009 case AV_CODEC_ID_ADPCM_EA_XAS: return 128;
3010 case AV_CODEC_ID_AMR_NB:
3011 case AV_CODEC_ID_EVRC:
3012 case AV_CODEC_ID_GSM:
3013 case AV_CODEC_ID_QCELP:
3014 case AV_CODEC_ID_RA_288: return 160;
3015 case AV_CODEC_ID_AMR_WB:
3016 case AV_CODEC_ID_GSM_MS: return 320;
3017 case AV_CODEC_ID_MP1: return 384;
3018 case AV_CODEC_ID_ATRAC1: return 512;
3019 case AV_CODEC_ID_ATRAC3: return 1024;
3020 case AV_CODEC_ID_ATRAC3P: return 2048;
3021 case AV_CODEC_ID_MP2:
3022 case AV_CODEC_ID_MUSEPACK7: return 1152;
3023 case AV_CODEC_ID_AC3: return 1536;
3027 /* calc from sample rate */
3028 if (id == AV_CODEC_ID_TTA)
3029 return 256 * sr / 245;
3032 /* calc from sample rate and channels */
3033 if (id == AV_CODEC_ID_BINKAUDIO_DCT)
3034 return (480 << (sr / 22050)) / ch;
3039 /* calc from block_align */
3040 if (id == AV_CODEC_ID_SIPR) {
3042 case 20: return 160;
3043 case 19: return 144;
3044 case 29: return 288;
3045 case 37: return 480;
3047 } else if (id == AV_CODEC_ID_ILBC) {
3049 case 38: return 160;
3050 case 50: return 240;
3055 if (frame_bytes > 0) {
3056 /* calc from frame_bytes only */
3057 if (id == AV_CODEC_ID_TRUESPEECH)
3058 return 240 * (frame_bytes / 32);
3059 if (id == AV_CODEC_ID_NELLYMOSER)
3060 return 256 * (frame_bytes / 64);
3061 if (id == AV_CODEC_ID_RA_144)
3062 return 160 * (frame_bytes / 20);
3063 if (id == AV_CODEC_ID_G723_1)
3064 return 240 * (frame_bytes / 24);
3067 /* calc from frame_bytes and bits_per_coded_sample */
3068 if (id == AV_CODEC_ID_ADPCM_G726)
3069 return frame_bytes * 8 / bps;
3072 if (ch > 0 && ch < INT_MAX/16) {
3073 /* calc from frame_bytes and channels */
3075 case AV_CODEC_ID_ADPCM_AFC:
3076 return frame_bytes / (9 * ch) * 16;
3077 case AV_CODEC_ID_ADPCM_PSX:
3078 case AV_CODEC_ID_ADPCM_DTK:
3079 return frame_bytes / (16 * ch) * 28;
3080 case AV_CODEC_ID_ADPCM_4XM:
3081 case AV_CODEC_ID_ADPCM_IMA_ISS:
3082 return (frame_bytes - 4 * ch) * 2 / ch;
3083 case AV_CODEC_ID_ADPCM_IMA_SMJPEG:
3084 return (frame_bytes - 4) * 2 / ch;
3085 case AV_CODEC_ID_ADPCM_IMA_AMV:
3086 return (frame_bytes - 8) * 2 / ch;
3087 case AV_CODEC_ID_ADPCM_THP:
3088 case AV_CODEC_ID_ADPCM_THP_LE:
3089 if (avctx->extradata)
3090 return frame_bytes * 14 / (8 * ch);
3092 case AV_CODEC_ID_ADPCM_XA:
3093 return (frame_bytes / 128) * 224 / ch;
3094 case AV_CODEC_ID_INTERPLAY_DPCM:
3095 return (frame_bytes - 6 - ch) / ch;
3096 case AV_CODEC_ID_ROQ_DPCM:
3097 return (frame_bytes - 8) / ch;
3098 case AV_CODEC_ID_XAN_DPCM:
3099 return (frame_bytes - 2 * ch) / ch;
3100 case AV_CODEC_ID_MACE3:
3101 return 3 * frame_bytes / ch;
3102 case AV_CODEC_ID_MACE6:
3103 return 6 * frame_bytes / ch;
3104 case AV_CODEC_ID_PCM_LXF:
3105 return 2 * (frame_bytes / (5 * ch));
3106 case AV_CODEC_ID_IAC:
3107 case AV_CODEC_ID_IMC:
3108 return 4 * frame_bytes / ch;
3112 /* calc from frame_bytes, channels, and codec_tag */
3113 if (id == AV_CODEC_ID_SOL_DPCM) {
3115 return frame_bytes / ch;
3117 return frame_bytes * 2 / ch;
3122 /* calc from frame_bytes, channels, and block_align */
3123 int blocks = frame_bytes / ba;
3124 switch (avctx->codec_id) {
3125 case AV_CODEC_ID_ADPCM_IMA_WAV:
3126 if (bps < 2 || bps > 5)
3128 return blocks * (1 + (ba - 4 * ch) / (bps * ch) * 8);
3129 case AV_CODEC_ID_ADPCM_IMA_DK3:
3130 return blocks * (((ba - 16) * 2 / 3 * 4) / ch);
3131 case AV_CODEC_ID_ADPCM_IMA_DK4:
3132 return blocks * (1 + (ba - 4 * ch) * 2 / ch);
3133 case AV_CODEC_ID_ADPCM_IMA_RAD:
3134 return blocks * ((ba - 4 * ch) * 2 / ch);
3135 case AV_CODEC_ID_ADPCM_MS:
3136 return blocks * (2 + (ba - 7 * ch) * 2 / ch);
3141 /* calc from frame_bytes, channels, and bits_per_coded_sample */
3142 switch (avctx->codec_id) {
3143 case AV_CODEC_ID_PCM_DVD:
3146 return 2 * (frame_bytes / ((bps * 2 / 8) * ch));
3147 case AV_CODEC_ID_PCM_BLURAY:
3150 return frame_bytes / ((FFALIGN(ch, 2) * bps) / 8);
3151 case AV_CODEC_ID_S302M:
3152 return 2 * (frame_bytes / ((bps + 4) / 4)) / ch;
3158 /* Fall back on using frame_size */
3159 if (avctx->frame_size > 1 && frame_bytes)
3160 return avctx->frame_size;
3162 //For WMA we currently have no other means to calculate duration thus we
3163 //do it here by assuming CBR, which is true for all known cases.
3164 if (avctx->bit_rate>0 && frame_bytes>0 && avctx->sample_rate>0 && avctx->block_align>1) {
3165 if (avctx->codec_id == AV_CODEC_ID_WMAV1 || avctx->codec_id == AV_CODEC_ID_WMAV2)
3166 return (frame_bytes * 8LL * avctx->sample_rate) / avctx->bit_rate;
3173 int ff_thread_init(AVCodecContext *s)
3180 unsigned int av_xiphlacing(unsigned char *s, unsigned int v)
3194 int ff_match_2uint16(const uint16_t(*tab)[2], int size, int a, int b)
3197 for (i = 0; i < size && !(tab[i][0] == a && tab[i][1] == b); i++) ;
3201 #if FF_API_MISSING_SAMPLE
3202 FF_DISABLE_DEPRECATION_WARNINGS
3203 void av_log_missing_feature(void *avc, const char *feature, int want_sample)
3205 av_log(avc, AV_LOG_WARNING, "%s is not implemented. Update your FFmpeg "
3206 "version to the newest one from Git. If the problem still "
3207 "occurs, it means that your file has a feature which has not "
3208 "been implemented.\n", feature);
3210 av_log_ask_for_sample(avc, NULL);
3213 void av_log_ask_for_sample(void *avc, const char *msg, ...)
3215 va_list argument_list;
3217 va_start(argument_list, msg);
3220 av_vlog(avc, AV_LOG_WARNING, msg, argument_list);
3221 av_log(avc, AV_LOG_WARNING, "If you want to help, upload a sample "
3222 "of this file to ftp://upload.ffmpeg.org/incoming/ "
3223 "and contact the ffmpeg-devel mailing list. (ffmpeg-devel@ffmpeg.org)\n");
3225 va_end(argument_list);
3227 FF_ENABLE_DEPRECATION_WARNINGS
3228 #endif /* FF_API_MISSING_SAMPLE */
3230 static AVHWAccel *first_hwaccel = NULL;
3231 static AVHWAccel **last_hwaccel = &first_hwaccel;
3233 void av_register_hwaccel(AVHWAccel *hwaccel)
3235 AVHWAccel **p = last_hwaccel;
3236 hwaccel->next = NULL;
3237 while(*p || avpriv_atomic_ptr_cas((void * volatile *)p, NULL, hwaccel))
3239 last_hwaccel = &hwaccel->next;
3242 AVHWAccel *av_hwaccel_next(const AVHWAccel *hwaccel)
3244 return hwaccel ? hwaccel->next : first_hwaccel;
3247 int av_lockmgr_register(int (*cb)(void **mutex, enum AVLockOp op))
3250 // There is no good way to rollback a failure to destroy the
3251 // mutex, so we ignore failures.
3252 lockmgr_cb(&codec_mutex, AV_LOCK_DESTROY);
3253 lockmgr_cb(&avformat_mutex, AV_LOCK_DESTROY);
3256 avformat_mutex = NULL;
3260 void *new_codec_mutex = NULL;
3261 void *new_avformat_mutex = NULL;
3263 if (err = cb(&new_codec_mutex, AV_LOCK_CREATE)) {
3264 return err > 0 ? AVERROR_UNKNOWN : err;
3266 if (err = cb(&new_avformat_mutex, AV_LOCK_CREATE)) {
3267 // Ignore failures to destroy the newly created mutex.
3268 cb(&new_codec_mutex, AV_LOCK_DESTROY);
3269 return err > 0 ? AVERROR_UNKNOWN : err;
3272 codec_mutex = new_codec_mutex;
3273 avformat_mutex = new_avformat_mutex;
3279 int ff_lock_avcodec(AVCodecContext *log_ctx, const AVCodec *codec)
3281 if (codec->caps_internal & FF_CODEC_CAP_INIT_THREADSAFE || !codec->init)
3285 if ((*lockmgr_cb)(&codec_mutex, AV_LOCK_OBTAIN))
3289 if (avpriv_atomic_int_add_and_fetch(&entangled_thread_counter, 1) != 1) {
3290 av_log(log_ctx, AV_LOG_ERROR,
3291 "Insufficient thread locking. At least %d threads are "
3292 "calling avcodec_open2() at the same time right now.\n",
3293 entangled_thread_counter);
3295 av_log(log_ctx, AV_LOG_ERROR, "No lock manager is set, please see av_lockmgr_register()\n");
3296 ff_avcodec_locked = 1;
3297 ff_unlock_avcodec(codec);
3298 return AVERROR(EINVAL);
3300 av_assert0(!ff_avcodec_locked);
3301 ff_avcodec_locked = 1;
3305 int ff_unlock_avcodec(const AVCodec *codec)
3307 if (codec->caps_internal & FF_CODEC_CAP_INIT_THREADSAFE || !codec->init)
3310 av_assert0(ff_avcodec_locked);
3311 ff_avcodec_locked = 0;
3312 avpriv_atomic_int_add_and_fetch(&entangled_thread_counter, -1);
3314 if ((*lockmgr_cb)(&codec_mutex, AV_LOCK_RELEASE))
3321 int avpriv_lock_avformat(void)
3324 if ((*lockmgr_cb)(&avformat_mutex, AV_LOCK_OBTAIN))
3330 int avpriv_unlock_avformat(void)
3333 if ((*lockmgr_cb)(&avformat_mutex, AV_LOCK_RELEASE))
3339 unsigned int avpriv_toupper4(unsigned int x)
3341 return av_toupper(x & 0xFF) +
3342 (av_toupper((x >> 8) & 0xFF) << 8) +
3343 (av_toupper((x >> 16) & 0xFF) << 16) +
3344 ((unsigned)av_toupper((x >> 24) & 0xFF) << 24);
3347 int ff_thread_ref_frame(ThreadFrame *dst, ThreadFrame *src)
3351 dst->owner = src->owner;
3353 ret = av_frame_ref(dst->f, src->f);
3357 av_assert0(!dst->progress);
3359 if (src->progress &&
3360 !(dst->progress = av_buffer_ref(src->progress))) {
3361 ff_thread_release_buffer(dst->owner, dst);
3362 return AVERROR(ENOMEM);
3370 enum AVPixelFormat ff_thread_get_format(AVCodecContext *avctx, const enum AVPixelFormat *fmt)
3372 return ff_get_format(avctx, fmt);
3375 int ff_thread_get_buffer(AVCodecContext *avctx, ThreadFrame *f, int flags)
3378 return ff_get_buffer(avctx, f->f, flags);
3381 void ff_thread_release_buffer(AVCodecContext *avctx, ThreadFrame *f)
3384 av_frame_unref(f->f);
3387 void ff_thread_finish_setup(AVCodecContext *avctx)
3391 void ff_thread_report_progress(ThreadFrame *f, int progress, int field)
3395 void ff_thread_await_progress(ThreadFrame *f, int progress, int field)
3399 int ff_thread_can_start_frame(AVCodecContext *avctx)
3404 int ff_alloc_entries(AVCodecContext *avctx, int count)
3409 void ff_reset_entries(AVCodecContext *avctx)
3413 void ff_thread_await_progress2(AVCodecContext *avctx, int field, int thread, int shift)
3417 void ff_thread_report_progress2(AVCodecContext *avctx, int field, int thread, int n)
3423 int avcodec_is_open(AVCodecContext *s)
3425 return !!s->internal;
3428 int avpriv_bprint_to_extradata(AVCodecContext *avctx, struct AVBPrint *buf)
3433 ret = av_bprint_finalize(buf, &str);
3436 if (!av_bprint_is_complete(buf)) {
3438 return AVERROR(ENOMEM);
3441 avctx->extradata = str;
3442 /* Note: the string is NUL terminated (so extradata can be read as a
3443 * string), but the ending character is not accounted in the size (in
3444 * binary formats you are likely not supposed to mux that character). When
3445 * extradata is copied, it is also padded with AV_INPUT_BUFFER_PADDING_SIZE
3447 avctx->extradata_size = buf->len;
3451 const uint8_t *avpriv_find_start_code(const uint8_t *av_restrict p,
3453 uint32_t *av_restrict state)
3457 av_assert0(p <= end);
3461 for (i = 0; i < 3; i++) {
3462 uint32_t tmp = *state << 8;
3463 *state = tmp + *(p++);
3464 if (tmp == 0x100 || p == end)
3469 if (p[-1] > 1 ) p += 3;
3470 else if (p[-2] ) p += 2;
3471 else if (p[-3]|(p[-1]-1)) p++;
3478 p = FFMIN(p, end) - 4;
3479 *state = AV_RB32(p);