]> git.sesse.net Git - ffmpeg/blob - libavcodec/utils.c
Merge commit '728685f37ab333ca35980bd01766c78d197f784a'
[ffmpeg] / libavcodec / utils.c
1 /*
2  * utils for libavcodec
3  * Copyright (c) 2001 Fabrice Bellard
4  * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
5  *
6  * This file is part of FFmpeg.
7  *
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.
12  *
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.
17  *
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
21  */
22
23 /**
24  * @file
25  * utils.
26  */
27
28 #include "config.h"
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/pixdesc.h"
40 #include "libavutil/imgutils.h"
41 #include "libavutil/samplefmt.h"
42 #include "libavutil/dict.h"
43 #include "avcodec.h"
44 #include "libavutil/opt.h"
45 #include "me_cmp.h"
46 #include "mpegvideo.h"
47 #include "thread.h"
48 #include "frame_thread_encoder.h"
49 #include "internal.h"
50 #include "raw.h"
51 #include "bytestream.h"
52 #include "version.h"
53 #include <stdlib.h>
54 #include <stdarg.h>
55 #include <limits.h>
56 #include <float.h>
57 #if CONFIG_ICONV
58 # include <iconv.h>
59 #endif
60
61 #if HAVE_PTHREADS
62 #include <pthread.h>
63 #elif HAVE_W32THREADS
64 #include "compat/w32pthreads.h"
65 #elif HAVE_OS2THREADS
66 #include "compat/os2threads.h"
67 #endif
68
69 #include "libavutil/ffversion.h"
70 const char av_codec_ffversion[] = "FFmpeg version " FFMPEG_VERSION;
71
72 #if HAVE_PTHREADS || HAVE_W32THREADS || HAVE_OS2THREADS
73 static int default_lockmgr_cb(void **arg, enum AVLockOp op)
74 {
75     void * volatile * mutex = arg;
76     int err;
77
78     switch (op) {
79     case AV_LOCK_CREATE:
80         return 0;
81     case AV_LOCK_OBTAIN:
82         if (!*mutex) {
83             pthread_mutex_t *tmp = av_malloc(sizeof(pthread_mutex_t));
84             if (!tmp)
85                 return AVERROR(ENOMEM);
86             if ((err = pthread_mutex_init(tmp, NULL))) {
87                 av_free(tmp);
88                 return AVERROR(err);
89             }
90             if (avpriv_atomic_ptr_cas(mutex, NULL, tmp)) {
91                 pthread_mutex_destroy(tmp);
92                 av_free(tmp);
93             }
94         }
95
96         if ((err = pthread_mutex_lock(*mutex)))
97             return AVERROR(err);
98
99         return 0;
100     case AV_LOCK_RELEASE:
101         if ((err = pthread_mutex_unlock(*mutex)))
102             return AVERROR(err);
103
104         return 0;
105     case AV_LOCK_DESTROY:
106         if (*mutex)
107             pthread_mutex_destroy(*mutex);
108         av_free(*mutex);
109         avpriv_atomic_ptr_cas(mutex, *mutex, NULL);
110         return 0;
111     }
112     return 1;
113 }
114 static int (*lockmgr_cb)(void **mutex, enum AVLockOp op) = default_lockmgr_cb;
115 #else
116 static int (*lockmgr_cb)(void **mutex, enum AVLockOp op) = NULL;
117 #endif
118
119
120 volatile int ff_avcodec_locked;
121 static int volatile entangled_thread_counter = 0;
122 static void *codec_mutex;
123 static void *avformat_mutex;
124
125 static inline int ff_fast_malloc(void *ptr, unsigned int *size, size_t min_size, int zero_realloc)
126 {
127     void **p = ptr;
128     if (min_size <= *size && *p)
129         return 0;
130     min_size = FFMAX(17 * min_size / 16 + 32, min_size);
131     av_free(*p);
132     *p = zero_realloc ? av_mallocz(min_size) : av_malloc(min_size);
133     if (!*p)
134         min_size = 0;
135     *size = min_size;
136     return 1;
137 }
138
139 void av_fast_padded_malloc(void *ptr, unsigned int *size, size_t min_size)
140 {
141     uint8_t **p = ptr;
142     if (min_size > SIZE_MAX - FF_INPUT_BUFFER_PADDING_SIZE) {
143         av_freep(p);
144         *size = 0;
145         return;
146     }
147     if (!ff_fast_malloc(p, size, min_size + FF_INPUT_BUFFER_PADDING_SIZE, 1))
148         memset(*p + min_size, 0, FF_INPUT_BUFFER_PADDING_SIZE);
149 }
150
151 void av_fast_padded_mallocz(void *ptr, unsigned int *size, size_t min_size)
152 {
153     uint8_t **p = ptr;
154     if (min_size > SIZE_MAX - FF_INPUT_BUFFER_PADDING_SIZE) {
155         av_freep(p);
156         *size = 0;
157         return;
158     }
159     if (!ff_fast_malloc(p, size, min_size + FF_INPUT_BUFFER_PADDING_SIZE, 1))
160         memset(*p, 0, min_size + FF_INPUT_BUFFER_PADDING_SIZE);
161 }
162
163 /* encoder management */
164 static AVCodec *first_avcodec = NULL;
165 static AVCodec **last_avcodec = &first_avcodec;
166
167 AVCodec *av_codec_next(const AVCodec *c)
168 {
169     if (c)
170         return c->next;
171     else
172         return first_avcodec;
173 }
174
175 static av_cold void avcodec_init(void)
176 {
177     static int initialized = 0;
178
179     if (initialized != 0)
180         return;
181     initialized = 1;
182
183     if (CONFIG_ME_CMP)
184         ff_me_cmp_init_static();
185 }
186
187 int av_codec_is_encoder(const AVCodec *codec)
188 {
189     return codec && (codec->encode_sub || codec->encode2);
190 }
191
192 int av_codec_is_decoder(const AVCodec *codec)
193 {
194     return codec && codec->decode;
195 }
196
197 av_cold void avcodec_register(AVCodec *codec)
198 {
199     AVCodec **p;
200     avcodec_init();
201     p = last_avcodec;
202     codec->next = NULL;
203
204     while(*p || avpriv_atomic_ptr_cas((void * volatile *)p, NULL, codec))
205         p = &(*p)->next;
206     last_avcodec = &codec->next;
207
208     if (codec->init_static_data)
209         codec->init_static_data(codec);
210 }
211
212 #if FF_API_EMU_EDGE
213 unsigned avcodec_get_edge_width(void)
214 {
215     return EDGE_WIDTH;
216 }
217 #endif
218
219 #if FF_API_SET_DIMENSIONS
220 void avcodec_set_dimensions(AVCodecContext *s, int width, int height)
221 {
222     int ret = ff_set_dimensions(s, width, height);
223     if (ret < 0) {
224         av_log(s, AV_LOG_WARNING, "Failed to set dimensions %d %d\n", width, height);
225     }
226 }
227 #endif
228
229 int ff_set_dimensions(AVCodecContext *s, int width, int height)
230 {
231     int ret = av_image_check_size(width, height, 0, s);
232
233     if (ret < 0)
234         width = height = 0;
235
236     s->coded_width  = width;
237     s->coded_height = height;
238     s->width        = FF_CEIL_RSHIFT(width,  s->lowres);
239     s->height       = FF_CEIL_RSHIFT(height, s->lowres);
240
241     return ret;
242 }
243
244 int ff_set_sar(AVCodecContext *avctx, AVRational sar)
245 {
246     int ret = av_image_check_sar(avctx->width, avctx->height, sar);
247
248     if (ret < 0) {
249         av_log(avctx, AV_LOG_WARNING, "ignoring invalid SAR: %u/%u\n",
250                sar.num, sar.den);
251         avctx->sample_aspect_ratio = (AVRational){ 0, 1 };
252         return ret;
253     } else {
254         avctx->sample_aspect_ratio = sar;
255     }
256     return 0;
257 }
258
259 int ff_side_data_update_matrix_encoding(AVFrame *frame,
260                                         enum AVMatrixEncoding matrix_encoding)
261 {
262     AVFrameSideData *side_data;
263     enum AVMatrixEncoding *data;
264
265     side_data = av_frame_get_side_data(frame, AV_FRAME_DATA_MATRIXENCODING);
266     if (!side_data)
267         side_data = av_frame_new_side_data(frame, AV_FRAME_DATA_MATRIXENCODING,
268                                            sizeof(enum AVMatrixEncoding));
269
270     if (!side_data)
271         return AVERROR(ENOMEM);
272
273     data  = (enum AVMatrixEncoding*)side_data->data;
274     *data = matrix_encoding;
275
276     return 0;
277 }
278
279 void avcodec_align_dimensions2(AVCodecContext *s, int *width, int *height,
280                                int linesize_align[AV_NUM_DATA_POINTERS])
281 {
282     int i;
283     int w_align = 1;
284     int h_align = 1;
285     AVPixFmtDescriptor const *desc = av_pix_fmt_desc_get(s->pix_fmt);
286
287     if (desc) {
288         w_align = 1 << desc->log2_chroma_w;
289         h_align = 1 << desc->log2_chroma_h;
290     }
291
292     switch (s->pix_fmt) {
293     case AV_PIX_FMT_YUV420P:
294     case AV_PIX_FMT_YUYV422:
295     case AV_PIX_FMT_YVYU422:
296     case AV_PIX_FMT_UYVY422:
297     case AV_PIX_FMT_YUV422P:
298     case AV_PIX_FMT_YUV440P:
299     case AV_PIX_FMT_YUV444P:
300     case AV_PIX_FMT_GBRP:
301     case AV_PIX_FMT_GBRAP:
302     case AV_PIX_FMT_GRAY8:
303     case AV_PIX_FMT_GRAY16BE:
304     case AV_PIX_FMT_GRAY16LE:
305     case AV_PIX_FMT_YUVJ420P:
306     case AV_PIX_FMT_YUVJ422P:
307     case AV_PIX_FMT_YUVJ440P:
308     case AV_PIX_FMT_YUVJ444P:
309     case AV_PIX_FMT_YUVA420P:
310     case AV_PIX_FMT_YUVA422P:
311     case AV_PIX_FMT_YUVA444P:
312     case AV_PIX_FMT_YUV420P9LE:
313     case AV_PIX_FMT_YUV420P9BE:
314     case AV_PIX_FMT_YUV420P10LE:
315     case AV_PIX_FMT_YUV420P10BE:
316     case AV_PIX_FMT_YUV420P12LE:
317     case AV_PIX_FMT_YUV420P12BE:
318     case AV_PIX_FMT_YUV420P14LE:
319     case AV_PIX_FMT_YUV420P14BE:
320     case AV_PIX_FMT_YUV420P16LE:
321     case AV_PIX_FMT_YUV420P16BE:
322     case AV_PIX_FMT_YUVA420P9LE:
323     case AV_PIX_FMT_YUVA420P9BE:
324     case AV_PIX_FMT_YUVA420P10LE:
325     case AV_PIX_FMT_YUVA420P10BE:
326     case AV_PIX_FMT_YUVA420P16LE:
327     case AV_PIX_FMT_YUVA420P16BE:
328     case AV_PIX_FMT_YUV422P9LE:
329     case AV_PIX_FMT_YUV422P9BE:
330     case AV_PIX_FMT_YUV422P10LE:
331     case AV_PIX_FMT_YUV422P10BE:
332     case AV_PIX_FMT_YUV422P12LE:
333     case AV_PIX_FMT_YUV422P12BE:
334     case AV_PIX_FMT_YUV422P14LE:
335     case AV_PIX_FMT_YUV422P14BE:
336     case AV_PIX_FMT_YUV422P16LE:
337     case AV_PIX_FMT_YUV422P16BE:
338     case AV_PIX_FMT_YUVA422P9LE:
339     case AV_PIX_FMT_YUVA422P9BE:
340     case AV_PIX_FMT_YUVA422P10LE:
341     case AV_PIX_FMT_YUVA422P10BE:
342     case AV_PIX_FMT_YUVA422P16LE:
343     case AV_PIX_FMT_YUVA422P16BE:
344     case AV_PIX_FMT_YUV444P9LE:
345     case AV_PIX_FMT_YUV444P9BE:
346     case AV_PIX_FMT_YUV444P10LE:
347     case AV_PIX_FMT_YUV444P10BE:
348     case AV_PIX_FMT_YUV444P12LE:
349     case AV_PIX_FMT_YUV444P12BE:
350     case AV_PIX_FMT_YUV444P14LE:
351     case AV_PIX_FMT_YUV444P14BE:
352     case AV_PIX_FMT_YUV444P16LE:
353     case AV_PIX_FMT_YUV444P16BE:
354     case AV_PIX_FMT_YUVA444P9LE:
355     case AV_PIX_FMT_YUVA444P9BE:
356     case AV_PIX_FMT_YUVA444P10LE:
357     case AV_PIX_FMT_YUVA444P10BE:
358     case AV_PIX_FMT_YUVA444P16LE:
359     case AV_PIX_FMT_YUVA444P16BE:
360     case AV_PIX_FMT_GBRP9LE:
361     case AV_PIX_FMT_GBRP9BE:
362     case AV_PIX_FMT_GBRP10LE:
363     case AV_PIX_FMT_GBRP10BE:
364     case AV_PIX_FMT_GBRP12LE:
365     case AV_PIX_FMT_GBRP12BE:
366     case AV_PIX_FMT_GBRP14LE:
367     case AV_PIX_FMT_GBRP14BE:
368     case AV_PIX_FMT_GBRP16LE:
369     case AV_PIX_FMT_GBRP16BE:
370         w_align = 16; //FIXME assume 16 pixel per macroblock
371         h_align = 16 * 2; // interlaced needs 2 macroblocks height
372         break;
373     case AV_PIX_FMT_YUV411P:
374     case AV_PIX_FMT_YUVJ411P:
375     case AV_PIX_FMT_UYYVYY411:
376         w_align = 32;
377         h_align = 8;
378         break;
379     case AV_PIX_FMT_YUV410P:
380         if (s->codec_id == AV_CODEC_ID_SVQ1) {
381             w_align = 64;
382             h_align = 64;
383         }
384         break;
385     case AV_PIX_FMT_RGB555:
386         if (s->codec_id == AV_CODEC_ID_RPZA) {
387             w_align = 4;
388             h_align = 4;
389         }
390         break;
391     case AV_PIX_FMT_PAL8:
392     case AV_PIX_FMT_BGR8:
393     case AV_PIX_FMT_RGB8:
394         if (s->codec_id == AV_CODEC_ID_SMC ||
395             s->codec_id == AV_CODEC_ID_CINEPAK) {
396             w_align = 4;
397             h_align = 4;
398         }
399         if (s->codec_id == AV_CODEC_ID_JV) {
400             w_align = 8;
401             h_align = 8;
402         }
403         break;
404     case AV_PIX_FMT_BGR24:
405         if ((s->codec_id == AV_CODEC_ID_MSZH) ||
406             (s->codec_id == AV_CODEC_ID_ZLIB)) {
407             w_align = 4;
408             h_align = 4;
409         }
410         break;
411     case AV_PIX_FMT_RGB24:
412         if (s->codec_id == AV_CODEC_ID_CINEPAK) {
413             w_align = 4;
414             h_align = 4;
415         }
416         break;
417     default:
418         break;
419     }
420
421     if (s->codec_id == AV_CODEC_ID_IFF_ILBM || s->codec_id == AV_CODEC_ID_IFF_BYTERUN1) {
422         w_align = FFMAX(w_align, 8);
423     }
424
425     *width  = FFALIGN(*width, w_align);
426     *height = FFALIGN(*height, h_align);
427     if (s->codec_id == AV_CODEC_ID_H264 || s->lowres)
428         // some of the optimized chroma MC reads one line too much
429         // which is also done in mpeg decoders with lowres > 0
430         *height += 2;
431
432     for (i = 0; i < 4; i++)
433         linesize_align[i] = STRIDE_ALIGN;
434 }
435
436 void avcodec_align_dimensions(AVCodecContext *s, int *width, int *height)
437 {
438     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(s->pix_fmt);
439     int chroma_shift = desc->log2_chroma_w;
440     int linesize_align[AV_NUM_DATA_POINTERS];
441     int align;
442
443     avcodec_align_dimensions2(s, width, height, linesize_align);
444     align               = FFMAX(linesize_align[0], linesize_align[3]);
445     linesize_align[1] <<= chroma_shift;
446     linesize_align[2] <<= chroma_shift;
447     align               = FFMAX3(align, linesize_align[1], linesize_align[2]);
448     *width              = FFALIGN(*width, align);
449 }
450
451 int avcodec_enum_to_chroma_pos(int *xpos, int *ypos, enum AVChromaLocation pos)
452 {
453     if (pos <= AVCHROMA_LOC_UNSPECIFIED || pos >= AVCHROMA_LOC_NB)
454         return AVERROR(EINVAL);
455     pos--;
456
457     *xpos = (pos&1) * 128;
458     *ypos = ((pos>>1)^(pos<4)) * 128;
459
460     return 0;
461 }
462
463 enum AVChromaLocation avcodec_chroma_pos_to_enum(int xpos, int ypos)
464 {
465     int pos, xout, yout;
466
467     for (pos = AVCHROMA_LOC_UNSPECIFIED + 1; pos < AVCHROMA_LOC_NB; pos++) {
468         if (avcodec_enum_to_chroma_pos(&xout, &yout, pos) == 0 && xout == xpos && yout == ypos)
469             return pos;
470     }
471     return AVCHROMA_LOC_UNSPECIFIED;
472 }
473
474 int avcodec_fill_audio_frame(AVFrame *frame, int nb_channels,
475                              enum AVSampleFormat sample_fmt, const uint8_t *buf,
476                              int buf_size, int align)
477 {
478     int ch, planar, needed_size, ret = 0;
479
480     needed_size = av_samples_get_buffer_size(NULL, nb_channels,
481                                              frame->nb_samples, sample_fmt,
482                                              align);
483     if (buf_size < needed_size)
484         return AVERROR(EINVAL);
485
486     planar = av_sample_fmt_is_planar(sample_fmt);
487     if (planar && nb_channels > AV_NUM_DATA_POINTERS) {
488         if (!(frame->extended_data = av_mallocz_array(nb_channels,
489                                                 sizeof(*frame->extended_data))))
490             return AVERROR(ENOMEM);
491     } else {
492         frame->extended_data = frame->data;
493     }
494
495     if ((ret = av_samples_fill_arrays(frame->extended_data, &frame->linesize[0],
496                                       (uint8_t *)(intptr_t)buf, nb_channels, frame->nb_samples,
497                                       sample_fmt, align)) < 0) {
498         if (frame->extended_data != frame->data)
499             av_freep(&frame->extended_data);
500         return ret;
501     }
502     if (frame->extended_data != frame->data) {
503         for (ch = 0; ch < AV_NUM_DATA_POINTERS; ch++)
504             frame->data[ch] = frame->extended_data[ch];
505     }
506
507     return ret;
508 }
509
510 static int update_frame_pool(AVCodecContext *avctx, AVFrame *frame)
511 {
512     FramePool *pool = avctx->internal->pool;
513     int i, ret;
514
515     switch (avctx->codec_type) {
516     case AVMEDIA_TYPE_VIDEO: {
517         AVPicture picture;
518         int size[4] = { 0 };
519         int w = frame->width;
520         int h = frame->height;
521         int tmpsize, unaligned;
522
523         if (pool->format == frame->format &&
524             pool->width == frame->width && pool->height == frame->height)
525             return 0;
526
527         avcodec_align_dimensions2(avctx, &w, &h, pool->stride_align);
528
529         do {
530             // NOTE: do not align linesizes individually, this breaks e.g. assumptions
531             // that linesize[0] == 2*linesize[1] in the MPEG-encoder for 4:2:2
532             av_image_fill_linesizes(picture.linesize, avctx->pix_fmt, w);
533             // increase alignment of w for next try (rhs gives the lowest bit set in w)
534             w += w & ~(w - 1);
535
536             unaligned = 0;
537             for (i = 0; i < 4; i++)
538                 unaligned |= picture.linesize[i] % pool->stride_align[i];
539         } while (unaligned);
540
541         tmpsize = av_image_fill_pointers(picture.data, avctx->pix_fmt, h,
542                                          NULL, picture.linesize);
543         if (tmpsize < 0)
544             return -1;
545
546         for (i = 0; i < 3 && picture.data[i + 1]; i++)
547             size[i] = picture.data[i + 1] - picture.data[i];
548         size[i] = tmpsize - (picture.data[i] - picture.data[0]);
549
550         for (i = 0; i < 4; i++) {
551             av_buffer_pool_uninit(&pool->pools[i]);
552             pool->linesize[i] = picture.linesize[i];
553             if (size[i]) {
554                 pool->pools[i] = av_buffer_pool_init(size[i] + 16 + STRIDE_ALIGN - 1,
555                                                      CONFIG_MEMORY_POISONING ?
556                                                         NULL :
557                                                         av_buffer_allocz);
558                 if (!pool->pools[i]) {
559                     ret = AVERROR(ENOMEM);
560                     goto fail;
561                 }
562             }
563         }
564         pool->format = frame->format;
565         pool->width  = frame->width;
566         pool->height = frame->height;
567
568         break;
569         }
570     case AVMEDIA_TYPE_AUDIO: {
571         int ch     = av_frame_get_channels(frame); //av_get_channel_layout_nb_channels(frame->channel_layout);
572         int planar = av_sample_fmt_is_planar(frame->format);
573         int planes = planar ? ch : 1;
574
575         if (pool->format == frame->format && pool->planes == planes &&
576             pool->channels == ch && frame->nb_samples == pool->samples)
577             return 0;
578
579         av_buffer_pool_uninit(&pool->pools[0]);
580         ret = av_samples_get_buffer_size(&pool->linesize[0], ch,
581                                          frame->nb_samples, frame->format, 0);
582         if (ret < 0)
583             goto fail;
584
585         pool->pools[0] = av_buffer_pool_init(pool->linesize[0], NULL);
586         if (!pool->pools[0]) {
587             ret = AVERROR(ENOMEM);
588             goto fail;
589         }
590
591         pool->format     = frame->format;
592         pool->planes     = planes;
593         pool->channels   = ch;
594         pool->samples = frame->nb_samples;
595         break;
596         }
597     default: av_assert0(0);
598     }
599     return 0;
600 fail:
601     for (i = 0; i < 4; i++)
602         av_buffer_pool_uninit(&pool->pools[i]);
603     pool->format = -1;
604     pool->planes = pool->channels = pool->samples = 0;
605     pool->width  = pool->height = 0;
606     return ret;
607 }
608
609 static int audio_get_buffer(AVCodecContext *avctx, AVFrame *frame)
610 {
611     FramePool *pool = avctx->internal->pool;
612     int planes = pool->planes;
613     int i;
614
615     frame->linesize[0] = pool->linesize[0];
616
617     if (planes > AV_NUM_DATA_POINTERS) {
618         frame->extended_data = av_mallocz_array(planes, sizeof(*frame->extended_data));
619         frame->nb_extended_buf = planes - AV_NUM_DATA_POINTERS;
620         frame->extended_buf  = av_mallocz_array(frame->nb_extended_buf,
621                                           sizeof(*frame->extended_buf));
622         if (!frame->extended_data || !frame->extended_buf) {
623             av_freep(&frame->extended_data);
624             av_freep(&frame->extended_buf);
625             return AVERROR(ENOMEM);
626         }
627     } else {
628         frame->extended_data = frame->data;
629         av_assert0(frame->nb_extended_buf == 0);
630     }
631
632     for (i = 0; i < FFMIN(planes, AV_NUM_DATA_POINTERS); i++) {
633         frame->buf[i] = av_buffer_pool_get(pool->pools[0]);
634         if (!frame->buf[i])
635             goto fail;
636         frame->extended_data[i] = frame->data[i] = frame->buf[i]->data;
637     }
638     for (i = 0; i < frame->nb_extended_buf; i++) {
639         frame->extended_buf[i] = av_buffer_pool_get(pool->pools[0]);
640         if (!frame->extended_buf[i])
641             goto fail;
642         frame->extended_data[i + AV_NUM_DATA_POINTERS] = frame->extended_buf[i]->data;
643     }
644
645     if (avctx->debug & FF_DEBUG_BUFFERS)
646         av_log(avctx, AV_LOG_DEBUG, "default_get_buffer called on frame %p", frame);
647
648     return 0;
649 fail:
650     av_frame_unref(frame);
651     return AVERROR(ENOMEM);
652 }
653
654 static int video_get_buffer(AVCodecContext *s, AVFrame *pic)
655 {
656     FramePool *pool = s->internal->pool;
657     int i;
658
659     if (pic->data[0]) {
660         av_log(s, AV_LOG_ERROR, "pic->data[0]!=NULL in avcodec_default_get_buffer\n");
661         return -1;
662     }
663
664     memset(pic->data, 0, sizeof(pic->data));
665     pic->extended_data = pic->data;
666
667     for (i = 0; i < 4 && pool->pools[i]; i++) {
668         pic->linesize[i] = pool->linesize[i];
669
670         pic->buf[i] = av_buffer_pool_get(pool->pools[i]);
671         if (!pic->buf[i])
672             goto fail;
673
674         pic->data[i] = pic->buf[i]->data;
675     }
676     for (; i < AV_NUM_DATA_POINTERS; i++) {
677         pic->data[i] = NULL;
678         pic->linesize[i] = 0;
679     }
680     if (pic->data[1] && !pic->data[2])
681         avpriv_set_systematic_pal2((uint32_t *)pic->data[1], s->pix_fmt);
682
683     if (s->debug & FF_DEBUG_BUFFERS)
684         av_log(s, AV_LOG_DEBUG, "default_get_buffer called on pic %p\n", pic);
685
686     return 0;
687 fail:
688     av_frame_unref(pic);
689     return AVERROR(ENOMEM);
690 }
691
692 void avpriv_color_frame(AVFrame *frame, const int c[4])
693 {
694     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(frame->format);
695     int p, y, x;
696
697     av_assert0(desc->flags & AV_PIX_FMT_FLAG_PLANAR);
698
699     for (p = 0; p<desc->nb_components; p++) {
700         uint8_t *dst = frame->data[p];
701         int is_chroma = p == 1 || p == 2;
702         int bytes  = is_chroma ? FF_CEIL_RSHIFT(frame->width,  desc->log2_chroma_w) : frame->width;
703         int height = is_chroma ? FF_CEIL_RSHIFT(frame->height, desc->log2_chroma_h) : frame->height;
704         for (y = 0; y < height; y++) {
705             if (desc->comp[0].depth_minus1 >= 8) {
706                 for (x = 0; x<bytes; x++)
707                     ((uint16_t*)dst)[x] = c[p];
708             }else
709                 memset(dst, c[p], bytes);
710             dst += frame->linesize[p];
711         }
712     }
713 }
714
715 int avcodec_default_get_buffer2(AVCodecContext *avctx, AVFrame *frame, int flags)
716 {
717     int ret;
718
719     if ((ret = update_frame_pool(avctx, frame)) < 0)
720         return ret;
721
722 #if FF_API_GET_BUFFER
723 FF_DISABLE_DEPRECATION_WARNINGS
724     frame->type = FF_BUFFER_TYPE_INTERNAL;
725 FF_ENABLE_DEPRECATION_WARNINGS
726 #endif
727
728     switch (avctx->codec_type) {
729     case AVMEDIA_TYPE_VIDEO:
730         return video_get_buffer(avctx, frame);
731     case AVMEDIA_TYPE_AUDIO:
732         return audio_get_buffer(avctx, frame);
733     default:
734         return -1;
735     }
736 }
737
738 int ff_init_buffer_info(AVCodecContext *avctx, AVFrame *frame)
739 {
740     AVPacket *pkt = avctx->internal->pkt;
741     int i;
742     static const struct {
743         enum AVPacketSideDataType packet;
744         enum AVFrameSideDataType frame;
745     } sd[] = {
746         { AV_PKT_DATA_REPLAYGAIN ,   AV_FRAME_DATA_REPLAYGAIN },
747         { AV_PKT_DATA_DISPLAYMATRIX, AV_FRAME_DATA_DISPLAYMATRIX },
748         { AV_PKT_DATA_STEREO3D,      AV_FRAME_DATA_STEREO3D },
749         { AV_PKT_DATA_AUDIO_SERVICE_TYPE, AV_FRAME_DATA_AUDIO_SERVICE_TYPE },
750     };
751
752     if (pkt) {
753         frame->pkt_pts = pkt->pts;
754         av_frame_set_pkt_pos     (frame, pkt->pos);
755         av_frame_set_pkt_duration(frame, pkt->duration);
756         av_frame_set_pkt_size    (frame, pkt->size);
757
758         for (i = 0; i < FF_ARRAY_ELEMS(sd); i++) {
759             int size;
760             uint8_t *packet_sd = av_packet_get_side_data(pkt, sd[i].packet, &size);
761             if (packet_sd) {
762                 AVFrameSideData *frame_sd = av_frame_new_side_data(frame,
763                                                                    sd[i].frame,
764                                                                    size);
765                 if (!frame_sd)
766                     return AVERROR(ENOMEM);
767
768                 memcpy(frame_sd->data, packet_sd, size);
769             }
770         }
771     } else {
772         frame->pkt_pts = AV_NOPTS_VALUE;
773         av_frame_set_pkt_pos     (frame, -1);
774         av_frame_set_pkt_duration(frame, 0);
775         av_frame_set_pkt_size    (frame, -1);
776     }
777     frame->reordered_opaque = avctx->reordered_opaque;
778
779     if (frame->color_primaries == AVCOL_PRI_UNSPECIFIED)
780         frame->color_primaries = avctx->color_primaries;
781     if (frame->color_trc == AVCOL_TRC_UNSPECIFIED)
782         frame->color_trc = avctx->color_trc;
783     if (av_frame_get_colorspace(frame) == AVCOL_SPC_UNSPECIFIED)
784         av_frame_set_colorspace(frame, avctx->colorspace);
785     if (av_frame_get_color_range(frame) == AVCOL_RANGE_UNSPECIFIED)
786         av_frame_set_color_range(frame, avctx->color_range);
787     if (frame->chroma_location == AVCHROMA_LOC_UNSPECIFIED)
788         frame->chroma_location = avctx->chroma_sample_location;
789
790     switch (avctx->codec->type) {
791     case AVMEDIA_TYPE_VIDEO:
792         frame->format              = avctx->pix_fmt;
793         if (!frame->sample_aspect_ratio.num)
794             frame->sample_aspect_ratio = avctx->sample_aspect_ratio;
795
796         if (frame->width && frame->height &&
797             av_image_check_sar(frame->width, frame->height,
798                                frame->sample_aspect_ratio) < 0) {
799             av_log(avctx, AV_LOG_WARNING, "ignoring invalid SAR: %u/%u\n",
800                    frame->sample_aspect_ratio.num,
801                    frame->sample_aspect_ratio.den);
802             frame->sample_aspect_ratio = (AVRational){ 0, 1 };
803         }
804
805         break;
806     case AVMEDIA_TYPE_AUDIO:
807         if (!frame->sample_rate)
808             frame->sample_rate    = avctx->sample_rate;
809         if (frame->format < 0)
810             frame->format         = avctx->sample_fmt;
811         if (!frame->channel_layout) {
812             if (avctx->channel_layout) {
813                  if (av_get_channel_layout_nb_channels(avctx->channel_layout) !=
814                      avctx->channels) {
815                      av_log(avctx, AV_LOG_ERROR, "Inconsistent channel "
816                             "configuration.\n");
817                      return AVERROR(EINVAL);
818                  }
819
820                 frame->channel_layout = avctx->channel_layout;
821             } else {
822                 if (avctx->channels > FF_SANE_NB_CHANNELS) {
823                     av_log(avctx, AV_LOG_ERROR, "Too many channels: %d.\n",
824                            avctx->channels);
825                     return AVERROR(ENOSYS);
826                 }
827             }
828         }
829         av_frame_set_channels(frame, avctx->channels);
830         break;
831     }
832     return 0;
833 }
834
835 #if FF_API_GET_BUFFER
836 FF_DISABLE_DEPRECATION_WARNINGS
837 int avcodec_default_get_buffer(AVCodecContext *avctx, AVFrame *frame)
838 {
839     return avcodec_default_get_buffer2(avctx, frame, 0);
840 }
841
842 typedef struct CompatReleaseBufPriv {
843     AVCodecContext avctx;
844     AVFrame frame;
845     uint8_t avframe_padding[1024]; // hack to allow linking to a avutil with larger AVFrame
846 } CompatReleaseBufPriv;
847
848 static void compat_free_buffer(void *opaque, uint8_t *data)
849 {
850     CompatReleaseBufPriv *priv = opaque;
851     if (priv->avctx.release_buffer)
852         priv->avctx.release_buffer(&priv->avctx, &priv->frame);
853     av_freep(&priv);
854 }
855
856 static void compat_release_buffer(void *opaque, uint8_t *data)
857 {
858     AVBufferRef *buf = opaque;
859     av_buffer_unref(&buf);
860 }
861 FF_ENABLE_DEPRECATION_WARNINGS
862 #endif
863
864 int ff_decode_frame_props(AVCodecContext *avctx, AVFrame *frame)
865 {
866     return ff_init_buffer_info(avctx, frame);
867 }
868
869 static int get_buffer_internal(AVCodecContext *avctx, AVFrame *frame, int flags)
870 {
871     const AVHWAccel *hwaccel = avctx->hwaccel;
872     int override_dimensions = 1;
873     int ret;
874
875     if (avctx->codec_type == AVMEDIA_TYPE_VIDEO) {
876         if ((ret = av_image_check_size(avctx->width, avctx->height, 0, avctx)) < 0 || avctx->pix_fmt<0) {
877             av_log(avctx, AV_LOG_ERROR, "video_get_buffer: image parameters invalid\n");
878             return AVERROR(EINVAL);
879         }
880     }
881     if (avctx->codec_type == AVMEDIA_TYPE_VIDEO) {
882         if (frame->width <= 0 || frame->height <= 0) {
883             frame->width  = FFMAX(avctx->width,  FF_CEIL_RSHIFT(avctx->coded_width,  avctx->lowres));
884             frame->height = FFMAX(avctx->height, FF_CEIL_RSHIFT(avctx->coded_height, avctx->lowres));
885             override_dimensions = 0;
886         }
887     }
888     ret = ff_decode_frame_props(avctx, frame);
889     if (ret < 0)
890         return ret;
891     if ((ret = ff_init_buffer_info(avctx, frame)) < 0)
892         return ret;
893
894     if (hwaccel) {
895         if (hwaccel->alloc_frame) {
896             ret = hwaccel->alloc_frame(avctx, frame);
897             goto end;
898         }
899     } else
900         avctx->sw_pix_fmt = avctx->pix_fmt;
901
902 #if FF_API_GET_BUFFER
903 FF_DISABLE_DEPRECATION_WARNINGS
904     /*
905      * Wrap an old get_buffer()-allocated buffer in a bunch of AVBuffers.
906      * We wrap each plane in its own AVBuffer. Each of those has a reference to
907      * a dummy AVBuffer as its private data, unreffing it on free.
908      * When all the planes are freed, the dummy buffer's free callback calls
909      * release_buffer().
910      */
911     if (avctx->get_buffer) {
912         CompatReleaseBufPriv *priv = NULL;
913         AVBufferRef *dummy_buf = NULL;
914         int planes, i, ret;
915
916         if (flags & AV_GET_BUFFER_FLAG_REF)
917             frame->reference    = 1;
918
919         ret = avctx->get_buffer(avctx, frame);
920         if (ret < 0)
921             return ret;
922
923         /* return if the buffers are already set up
924          * this would happen e.g. when a custom get_buffer() calls
925          * avcodec_default_get_buffer
926          */
927         if (frame->buf[0])
928             goto end0;
929
930         priv = av_mallocz(sizeof(*priv));
931         if (!priv) {
932             ret = AVERROR(ENOMEM);
933             goto fail;
934         }
935         priv->avctx = *avctx;
936         priv->frame = *frame;
937
938         dummy_buf = av_buffer_create(NULL, 0, compat_free_buffer, priv, 0);
939         if (!dummy_buf) {
940             ret = AVERROR(ENOMEM);
941             goto fail;
942         }
943
944 #define WRAP_PLANE(ref_out, data, data_size)                            \
945 do {                                                                    \
946     AVBufferRef *dummy_ref = av_buffer_ref(dummy_buf);                  \
947     if (!dummy_ref) {                                                   \
948         ret = AVERROR(ENOMEM);                                          \
949         goto fail;                                                      \
950     }                                                                   \
951     ref_out = av_buffer_create(data, data_size, compat_release_buffer,  \
952                                dummy_ref, 0);                           \
953     if (!ref_out) {                                                     \
954         av_frame_unref(frame);                                          \
955         ret = AVERROR(ENOMEM);                                          \
956         goto fail;                                                      \
957     }                                                                   \
958 } while (0)
959
960         if (avctx->codec_type == AVMEDIA_TYPE_VIDEO) {
961             const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(frame->format);
962
963             planes = av_pix_fmt_count_planes(frame->format);
964             /* workaround for AVHWAccel plane count of 0, buf[0] is used as
965                check for allocated buffers: make libavcodec happy */
966             if (desc && desc->flags & AV_PIX_FMT_FLAG_HWACCEL)
967                 planes = 1;
968             if (!desc || planes <= 0) {
969                 ret = AVERROR(EINVAL);
970                 goto fail;
971             }
972
973             for (i = 0; i < planes; i++) {
974                 int v_shift    = (i == 1 || i == 2) ? desc->log2_chroma_h : 0;
975                 int plane_size = (frame->height >> v_shift) * frame->linesize[i];
976
977                 WRAP_PLANE(frame->buf[i], frame->data[i], plane_size);
978             }
979         } else {
980             int planar = av_sample_fmt_is_planar(frame->format);
981             planes = planar ? avctx->channels : 1;
982
983             if (planes > FF_ARRAY_ELEMS(frame->buf)) {
984                 frame->nb_extended_buf = planes - FF_ARRAY_ELEMS(frame->buf);
985                 frame->extended_buf = av_malloc_array(sizeof(*frame->extended_buf),
986                                                 frame->nb_extended_buf);
987                 if (!frame->extended_buf) {
988                     ret = AVERROR(ENOMEM);
989                     goto fail;
990                 }
991             }
992
993             for (i = 0; i < FFMIN(planes, FF_ARRAY_ELEMS(frame->buf)); i++)
994                 WRAP_PLANE(frame->buf[i], frame->extended_data[i], frame->linesize[0]);
995
996             for (i = 0; i < frame->nb_extended_buf; i++)
997                 WRAP_PLANE(frame->extended_buf[i],
998                            frame->extended_data[i + FF_ARRAY_ELEMS(frame->buf)],
999                            frame->linesize[0]);
1000         }
1001
1002         av_buffer_unref(&dummy_buf);
1003
1004 end0:
1005         frame->width  = avctx->width;
1006         frame->height = avctx->height;
1007
1008         return 0;
1009
1010 fail:
1011         avctx->release_buffer(avctx, frame);
1012         av_freep(&priv);
1013         av_buffer_unref(&dummy_buf);
1014         return ret;
1015     }
1016 FF_ENABLE_DEPRECATION_WARNINGS
1017 #endif
1018
1019     ret = avctx->get_buffer2(avctx, frame, flags);
1020
1021 end:
1022     if (avctx->codec_type == AVMEDIA_TYPE_VIDEO && !override_dimensions) {
1023         frame->width  = avctx->width;
1024         frame->height = avctx->height;
1025     }
1026
1027     return ret;
1028 }
1029
1030 int ff_get_buffer(AVCodecContext *avctx, AVFrame *frame, int flags)
1031 {
1032     int ret = get_buffer_internal(avctx, frame, flags);
1033     if (ret < 0)
1034         av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
1035     return ret;
1036 }
1037
1038 static int reget_buffer_internal(AVCodecContext *avctx, AVFrame *frame)
1039 {
1040     AVFrame *tmp;
1041     int ret;
1042
1043     av_assert0(avctx->codec_type == AVMEDIA_TYPE_VIDEO);
1044
1045     if (frame->data[0] && (frame->width != avctx->width || frame->height != avctx->height || frame->format != avctx->pix_fmt)) {
1046         av_log(avctx, AV_LOG_WARNING, "Picture changed from size:%dx%d fmt:%s to size:%dx%d fmt:%s in reget buffer()\n",
1047                frame->width, frame->height, av_get_pix_fmt_name(frame->format), avctx->width, avctx->height, av_get_pix_fmt_name(avctx->pix_fmt));
1048         av_frame_unref(frame);
1049     }
1050
1051     ff_init_buffer_info(avctx, frame);
1052
1053     if (!frame->data[0])
1054         return ff_get_buffer(avctx, frame, AV_GET_BUFFER_FLAG_REF);
1055
1056     if (av_frame_is_writable(frame))
1057         return ff_decode_frame_props(avctx, frame);
1058
1059     tmp = av_frame_alloc();
1060     if (!tmp)
1061         return AVERROR(ENOMEM);
1062
1063     av_frame_move_ref(tmp, frame);
1064
1065     ret = ff_get_buffer(avctx, frame, AV_GET_BUFFER_FLAG_REF);
1066     if (ret < 0) {
1067         av_frame_free(&tmp);
1068         return ret;
1069     }
1070
1071     av_frame_copy(frame, tmp);
1072     av_frame_free(&tmp);
1073
1074     return 0;
1075 }
1076
1077 int ff_reget_buffer(AVCodecContext *avctx, AVFrame *frame)
1078 {
1079     int ret = reget_buffer_internal(avctx, frame);
1080     if (ret < 0)
1081         av_log(avctx, AV_LOG_ERROR, "reget_buffer() failed\n");
1082     return ret;
1083 }
1084
1085 #if FF_API_GET_BUFFER
1086 void avcodec_default_release_buffer(AVCodecContext *s, AVFrame *pic)
1087 {
1088     av_assert0(s->codec_type == AVMEDIA_TYPE_VIDEO);
1089
1090     av_frame_unref(pic);
1091 }
1092
1093 int avcodec_default_reget_buffer(AVCodecContext *s, AVFrame *pic)
1094 {
1095     av_assert0(0);
1096     return AVERROR_BUG;
1097 }
1098 #endif
1099
1100 int avcodec_default_execute(AVCodecContext *c, int (*func)(AVCodecContext *c2, void *arg2), void *arg, int *ret, int count, int size)
1101 {
1102     int i;
1103
1104     for (i = 0; i < count; i++) {
1105         int r = func(c, (char *)arg + i * size);
1106         if (ret)
1107             ret[i] = r;
1108     }
1109     return 0;
1110 }
1111
1112 int avcodec_default_execute2(AVCodecContext *c, int (*func)(AVCodecContext *c2, void *arg2, int jobnr, int threadnr), void *arg, int *ret, int count)
1113 {
1114     int i;
1115
1116     for (i = 0; i < count; i++) {
1117         int r = func(c, arg, i, 0);
1118         if (ret)
1119             ret[i] = r;
1120     }
1121     return 0;
1122 }
1123
1124 enum AVPixelFormat avpriv_find_pix_fmt(const PixelFormatTag *tags,
1125                                        unsigned int fourcc)
1126 {
1127     while (tags->pix_fmt >= 0) {
1128         if (tags->fourcc == fourcc)
1129             return tags->pix_fmt;
1130         tags++;
1131     }
1132     return AV_PIX_FMT_NONE;
1133 }
1134
1135 static int is_hwaccel_pix_fmt(enum AVPixelFormat pix_fmt)
1136 {
1137     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
1138     return desc->flags & AV_PIX_FMT_FLAG_HWACCEL;
1139 }
1140
1141 enum AVPixelFormat avcodec_default_get_format(struct AVCodecContext *s, const enum AVPixelFormat *fmt)
1142 {
1143     while (*fmt != AV_PIX_FMT_NONE && is_hwaccel_pix_fmt(*fmt))
1144         ++fmt;
1145     return fmt[0];
1146 }
1147
1148 static AVHWAccel *find_hwaccel(enum AVCodecID codec_id,
1149                                enum AVPixelFormat pix_fmt)
1150 {
1151     AVHWAccel *hwaccel = NULL;
1152
1153     while ((hwaccel = av_hwaccel_next(hwaccel)))
1154         if (hwaccel->id == codec_id
1155             && hwaccel->pix_fmt == pix_fmt)
1156             return hwaccel;
1157     return NULL;
1158 }
1159
1160 static int setup_hwaccel(AVCodecContext *avctx,
1161                          const enum AVPixelFormat fmt,
1162                          const char *name)
1163 {
1164     AVHWAccel *hwa = find_hwaccel(avctx->codec_id, fmt);
1165     int ret        = 0;
1166
1167     if (!hwa) {
1168         av_log(avctx, AV_LOG_ERROR,
1169                "Could not find an AVHWAccel for the pixel format: %s",
1170                name);
1171         return AVERROR(ENOENT);
1172     }
1173
1174     if (hwa->priv_data_size) {
1175         avctx->internal->hwaccel_priv_data = av_mallocz(hwa->priv_data_size);
1176         if (!avctx->internal->hwaccel_priv_data)
1177             return AVERROR(ENOMEM);
1178     }
1179
1180     if (hwa->init) {
1181         ret = hwa->init(avctx);
1182         if (ret < 0) {
1183             av_freep(&avctx->internal->hwaccel_priv_data);
1184             return ret;
1185         }
1186     }
1187
1188     avctx->hwaccel = hwa;
1189
1190     return 0;
1191 }
1192
1193 int ff_get_format(AVCodecContext *avctx, const enum AVPixelFormat *fmt)
1194 {
1195     const AVPixFmtDescriptor *desc;
1196     enum AVPixelFormat *choices;
1197     enum AVPixelFormat ret;
1198     unsigned n = 0;
1199
1200     while (fmt[n] != AV_PIX_FMT_NONE)
1201         ++n;
1202
1203     av_assert0(n >= 1);
1204     avctx->sw_pix_fmt = fmt[n - 1];
1205     av_assert2(!is_hwaccel_pix_fmt(avctx->sw_pix_fmt));
1206
1207     choices = av_malloc_array(n + 1, sizeof(*choices));
1208     if (!choices)
1209         return AV_PIX_FMT_NONE;
1210
1211     memcpy(choices, fmt, (n + 1) * sizeof(*choices));
1212
1213     for (;;) {
1214         if (avctx->hwaccel && avctx->hwaccel->uninit)
1215             avctx->hwaccel->uninit(avctx);
1216         av_freep(&avctx->internal->hwaccel_priv_data);
1217         avctx->hwaccel = NULL;
1218
1219         ret = avctx->get_format(avctx, choices);
1220
1221         desc = av_pix_fmt_desc_get(ret);
1222         if (!desc) {
1223             ret = AV_PIX_FMT_NONE;
1224             break;
1225         }
1226
1227         if (!(desc->flags & AV_PIX_FMT_FLAG_HWACCEL))
1228             break;
1229         if (avctx->codec->capabilities&CODEC_CAP_HWACCEL_VDPAU)
1230             break;
1231
1232         if (!setup_hwaccel(avctx, ret, desc->name))
1233             break;
1234
1235         /* Remove failed hwaccel from choices */
1236         for (n = 0; choices[n] != ret; n++)
1237             av_assert0(choices[n] != AV_PIX_FMT_NONE);
1238
1239         do
1240             choices[n] = choices[n + 1];
1241         while (choices[n++] != AV_PIX_FMT_NONE);
1242     }
1243
1244     av_freep(&choices);
1245     return ret;
1246 }
1247
1248 #if FF_API_AVFRAME_LAVC
1249 void avcodec_get_frame_defaults(AVFrame *frame)
1250 {
1251 #if LIBAVCODEC_VERSION_MAJOR >= 55
1252      // extended_data should explicitly be freed when needed, this code is unsafe currently
1253      // also this is not compatible to the <55 ABI/API
1254     if (frame->extended_data != frame->data && 0)
1255         av_freep(&frame->extended_data);
1256 #endif
1257
1258     memset(frame, 0, sizeof(AVFrame));
1259     av_frame_unref(frame);
1260 }
1261
1262 AVFrame *avcodec_alloc_frame(void)
1263 {
1264     return av_frame_alloc();
1265 }
1266
1267 void avcodec_free_frame(AVFrame **frame)
1268 {
1269     av_frame_free(frame);
1270 }
1271 #endif
1272
1273 MAKE_ACCESSORS(AVCodecContext, codec, AVRational, pkt_timebase)
1274 MAKE_ACCESSORS(AVCodecContext, codec, const AVCodecDescriptor *, codec_descriptor)
1275 MAKE_ACCESSORS(AVCodecContext, codec, int, lowres)
1276 MAKE_ACCESSORS(AVCodecContext, codec, int, seek_preroll)
1277 MAKE_ACCESSORS(AVCodecContext, codec, uint16_t*, chroma_intra_matrix)
1278
1279 int av_codec_get_max_lowres(const AVCodec *codec)
1280 {
1281     return codec->max_lowres;
1282 }
1283
1284 static void get_subtitle_defaults(AVSubtitle *sub)
1285 {
1286     memset(sub, 0, sizeof(*sub));
1287     sub->pts = AV_NOPTS_VALUE;
1288 }
1289
1290 static int get_bit_rate(AVCodecContext *ctx)
1291 {
1292     int bit_rate;
1293     int bits_per_sample;
1294
1295     switch (ctx->codec_type) {
1296     case AVMEDIA_TYPE_VIDEO:
1297     case AVMEDIA_TYPE_DATA:
1298     case AVMEDIA_TYPE_SUBTITLE:
1299     case AVMEDIA_TYPE_ATTACHMENT:
1300         bit_rate = ctx->bit_rate;
1301         break;
1302     case AVMEDIA_TYPE_AUDIO:
1303         bits_per_sample = av_get_bits_per_sample(ctx->codec_id);
1304         bit_rate = bits_per_sample ? ctx->sample_rate * ctx->channels * bits_per_sample : ctx->bit_rate;
1305         break;
1306     default:
1307         bit_rate = 0;
1308         break;
1309     }
1310     return bit_rate;
1311 }
1312
1313 int attribute_align_arg ff_codec_open2_recursive(AVCodecContext *avctx, const AVCodec *codec, AVDictionary **options)
1314 {
1315     int ret = 0;
1316
1317     ff_unlock_avcodec();
1318
1319     ret = avcodec_open2(avctx, codec, options);
1320
1321     ff_lock_avcodec(avctx);
1322     return ret;
1323 }
1324
1325 int attribute_align_arg avcodec_open2(AVCodecContext *avctx, const AVCodec *codec, AVDictionary **options)
1326 {
1327     int ret = 0;
1328     AVDictionary *tmp = NULL;
1329
1330     if (avcodec_is_open(avctx))
1331         return 0;
1332
1333     if ((!codec && !avctx->codec)) {
1334         av_log(avctx, AV_LOG_ERROR, "No codec provided to avcodec_open2()\n");
1335         return AVERROR(EINVAL);
1336     }
1337     if ((codec && avctx->codec && codec != avctx->codec)) {
1338         av_log(avctx, AV_LOG_ERROR, "This AVCodecContext was allocated for %s, "
1339                                     "but %s passed to avcodec_open2()\n", avctx->codec->name, codec->name);
1340         return AVERROR(EINVAL);
1341     }
1342     if (!codec)
1343         codec = avctx->codec;
1344
1345     if (avctx->extradata_size < 0 || avctx->extradata_size >= FF_MAX_EXTRADATA_SIZE)
1346         return AVERROR(EINVAL);
1347
1348     if (options)
1349         av_dict_copy(&tmp, *options, 0);
1350
1351     ret = ff_lock_avcodec(avctx);
1352     if (ret < 0)
1353         return ret;
1354
1355     avctx->internal = av_mallocz(sizeof(AVCodecInternal));
1356     if (!avctx->internal) {
1357         ret = AVERROR(ENOMEM);
1358         goto end;
1359     }
1360
1361     avctx->internal->pool = av_mallocz(sizeof(*avctx->internal->pool));
1362     if (!avctx->internal->pool) {
1363         ret = AVERROR(ENOMEM);
1364         goto free_and_end;
1365     }
1366
1367     avctx->internal->to_free = av_frame_alloc();
1368     if (!avctx->internal->to_free) {
1369         ret = AVERROR(ENOMEM);
1370         goto free_and_end;
1371     }
1372
1373     if (codec->priv_data_size > 0) {
1374         if (!avctx->priv_data) {
1375             avctx->priv_data = av_mallocz(codec->priv_data_size);
1376             if (!avctx->priv_data) {
1377                 ret = AVERROR(ENOMEM);
1378                 goto end;
1379             }
1380             if (codec->priv_class) {
1381                 *(const AVClass **)avctx->priv_data = codec->priv_class;
1382                 av_opt_set_defaults(avctx->priv_data);
1383             }
1384         }
1385         if (codec->priv_class && (ret = av_opt_set_dict(avctx->priv_data, &tmp)) < 0)
1386             goto free_and_end;
1387     } else {
1388         avctx->priv_data = NULL;
1389     }
1390     if ((ret = av_opt_set_dict(avctx, &tmp)) < 0)
1391         goto free_and_end;
1392
1393     if (avctx->codec_whitelist && av_match_list(codec->name, avctx->codec_whitelist, ',') <= 0) {
1394         av_log(avctx, AV_LOG_ERROR, "Codec (%s) not on whitelist\n", codec->name);
1395         ret = AVERROR(EINVAL);
1396         goto free_and_end;
1397     }
1398
1399     // only call ff_set_dimensions() for non H.264/VP6F codecs so as not to overwrite previously setup dimensions
1400     if (!(avctx->coded_width && avctx->coded_height && avctx->width && avctx->height &&
1401           (avctx->codec_id == AV_CODEC_ID_H264 || avctx->codec_id == AV_CODEC_ID_VP6F))) {
1402     if (avctx->coded_width && avctx->coded_height)
1403         ret = ff_set_dimensions(avctx, avctx->coded_width, avctx->coded_height);
1404     else if (avctx->width && avctx->height)
1405         ret = ff_set_dimensions(avctx, avctx->width, avctx->height);
1406     if (ret < 0)
1407         goto free_and_end;
1408     }
1409
1410     if ((avctx->coded_width || avctx->coded_height || avctx->width || avctx->height)
1411         && (  av_image_check_size(avctx->coded_width, avctx->coded_height, 0, avctx) < 0
1412            || av_image_check_size(avctx->width,       avctx->height,       0, avctx) < 0)) {
1413         av_log(avctx, AV_LOG_WARNING, "Ignoring invalid width/height values\n");
1414         ff_set_dimensions(avctx, 0, 0);
1415     }
1416
1417     if (avctx->width > 0 && avctx->height > 0) {
1418         if (av_image_check_sar(avctx->width, avctx->height,
1419                                avctx->sample_aspect_ratio) < 0) {
1420             av_log(avctx, AV_LOG_WARNING, "ignoring invalid SAR: %u/%u\n",
1421                    avctx->sample_aspect_ratio.num,
1422                    avctx->sample_aspect_ratio.den);
1423             avctx->sample_aspect_ratio = (AVRational){ 0, 1 };
1424         }
1425     }
1426
1427     /* if the decoder init function was already called previously,
1428      * free the already allocated subtitle_header before overwriting it */
1429     if (av_codec_is_decoder(codec))
1430         av_freep(&avctx->subtitle_header);
1431
1432     if (avctx->channels > FF_SANE_NB_CHANNELS) {
1433         ret = AVERROR(EINVAL);
1434         goto free_and_end;
1435     }
1436
1437     avctx->codec = codec;
1438     if ((avctx->codec_type == AVMEDIA_TYPE_UNKNOWN || avctx->codec_type == codec->type) &&
1439         avctx->codec_id == AV_CODEC_ID_NONE) {
1440         avctx->codec_type = codec->type;
1441         avctx->codec_id   = codec->id;
1442     }
1443     if (avctx->codec_id != codec->id || (avctx->codec_type != codec->type
1444                                          && avctx->codec_type != AVMEDIA_TYPE_ATTACHMENT)) {
1445         av_log(avctx, AV_LOG_ERROR, "Codec type or id mismatches\n");
1446         ret = AVERROR(EINVAL);
1447         goto free_and_end;
1448     }
1449     avctx->frame_number = 0;
1450     avctx->codec_descriptor = avcodec_descriptor_get(avctx->codec_id);
1451
1452     if (avctx->codec->capabilities & CODEC_CAP_EXPERIMENTAL &&
1453         avctx->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) {
1454         const char *codec_string = av_codec_is_encoder(codec) ? "encoder" : "decoder";
1455         AVCodec *codec2;
1456         av_log(avctx, AV_LOG_ERROR,
1457                "The %s '%s' is experimental but experimental codecs are not enabled, "
1458                "add '-strict %d' if you want to use it.\n",
1459                codec_string, codec->name, FF_COMPLIANCE_EXPERIMENTAL);
1460         codec2 = av_codec_is_encoder(codec) ? avcodec_find_encoder(codec->id) : avcodec_find_decoder(codec->id);
1461         if (!(codec2->capabilities & CODEC_CAP_EXPERIMENTAL))
1462             av_log(avctx, AV_LOG_ERROR, "Alternatively use the non experimental %s '%s'.\n",
1463                 codec_string, codec2->name);
1464         ret = AVERROR_EXPERIMENTAL;
1465         goto free_and_end;
1466     }
1467
1468     if (avctx->codec_type == AVMEDIA_TYPE_AUDIO &&
1469         (!avctx->time_base.num || !avctx->time_base.den)) {
1470         avctx->time_base.num = 1;
1471         avctx->time_base.den = avctx->sample_rate;
1472     }
1473
1474     if (!HAVE_THREADS)
1475         av_log(avctx, AV_LOG_WARNING, "Warning: not compiled with thread support, using thread emulation\n");
1476
1477     if (CONFIG_FRAME_THREAD_ENCODER) {
1478         ff_unlock_avcodec(); //we will instanciate a few encoders thus kick the counter to prevent false detection of a problem
1479         ret = ff_frame_thread_encoder_init(avctx, options ? *options : NULL);
1480         ff_lock_avcodec(avctx);
1481         if (ret < 0)
1482             goto free_and_end;
1483     }
1484
1485     if (HAVE_THREADS
1486         && !(avctx->internal->frame_thread_encoder && (avctx->active_thread_type&FF_THREAD_FRAME))) {
1487         ret = ff_thread_init(avctx);
1488         if (ret < 0) {
1489             goto free_and_end;
1490         }
1491     }
1492     if (!HAVE_THREADS && !(codec->capabilities & CODEC_CAP_AUTO_THREADS))
1493         avctx->thread_count = 1;
1494
1495     if (avctx->codec->max_lowres < avctx->lowres || avctx->lowres < 0) {
1496         av_log(avctx, AV_LOG_ERROR, "The maximum value for lowres supported by the decoder is %d\n",
1497                avctx->codec->max_lowres);
1498         ret = AVERROR(EINVAL);
1499         goto free_and_end;
1500     }
1501
1502 #if FF_API_VISMV
1503     if (avctx->debug_mv)
1504         av_log(avctx, AV_LOG_WARNING, "The 'vismv' option is deprecated, "
1505                "see the codecview filter instead.\n");
1506 #endif
1507
1508     if (av_codec_is_encoder(avctx->codec)) {
1509         int i;
1510         if (avctx->codec->sample_fmts) {
1511             for (i = 0; avctx->codec->sample_fmts[i] != AV_SAMPLE_FMT_NONE; i++) {
1512                 if (avctx->sample_fmt == avctx->codec->sample_fmts[i])
1513                     break;
1514                 if (avctx->channels == 1 &&
1515                     av_get_planar_sample_fmt(avctx->sample_fmt) ==
1516                     av_get_planar_sample_fmt(avctx->codec->sample_fmts[i])) {
1517                     avctx->sample_fmt = avctx->codec->sample_fmts[i];
1518                     break;
1519                 }
1520             }
1521             if (avctx->codec->sample_fmts[i] == AV_SAMPLE_FMT_NONE) {
1522                 char buf[128];
1523                 snprintf(buf, sizeof(buf), "%d", avctx->sample_fmt);
1524                 av_log(avctx, AV_LOG_ERROR, "Specified sample format %s is invalid or not supported\n",
1525                        (char *)av_x_if_null(av_get_sample_fmt_name(avctx->sample_fmt), buf));
1526                 ret = AVERROR(EINVAL);
1527                 goto free_and_end;
1528             }
1529         }
1530         if (avctx->codec->pix_fmts) {
1531             for (i = 0; avctx->codec->pix_fmts[i] != AV_PIX_FMT_NONE; i++)
1532                 if (avctx->pix_fmt == avctx->codec->pix_fmts[i])
1533                     break;
1534             if (avctx->codec->pix_fmts[i] == AV_PIX_FMT_NONE
1535                 && !((avctx->codec_id == AV_CODEC_ID_MJPEG || avctx->codec_id == AV_CODEC_ID_LJPEG)
1536                      && avctx->strict_std_compliance <= FF_COMPLIANCE_UNOFFICIAL)) {
1537                 char buf[128];
1538                 snprintf(buf, sizeof(buf), "%d", avctx->pix_fmt);
1539                 av_log(avctx, AV_LOG_ERROR, "Specified pixel format %s is invalid or not supported\n",
1540                        (char *)av_x_if_null(av_get_pix_fmt_name(avctx->pix_fmt), buf));
1541                 ret = AVERROR(EINVAL);
1542                 goto free_and_end;
1543             }
1544             if (avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ420P ||
1545                 avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ411P ||
1546                 avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ422P ||
1547                 avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ440P ||
1548                 avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ444P)
1549                 avctx->color_range = AVCOL_RANGE_JPEG;
1550         }
1551         if (avctx->codec->supported_samplerates) {
1552             for (i = 0; avctx->codec->supported_samplerates[i] != 0; i++)
1553                 if (avctx->sample_rate == avctx->codec->supported_samplerates[i])
1554                     break;
1555             if (avctx->codec->supported_samplerates[i] == 0) {
1556                 av_log(avctx, AV_LOG_ERROR, "Specified sample rate %d is not supported\n",
1557                        avctx->sample_rate);
1558                 ret = AVERROR(EINVAL);
1559                 goto free_and_end;
1560             }
1561         }
1562         if (avctx->codec->channel_layouts) {
1563             if (!avctx->channel_layout) {
1564                 av_log(avctx, AV_LOG_WARNING, "Channel layout not specified\n");
1565             } else {
1566                 for (i = 0; avctx->codec->channel_layouts[i] != 0; i++)
1567                     if (avctx->channel_layout == avctx->codec->channel_layouts[i])
1568                         break;
1569                 if (avctx->codec->channel_layouts[i] == 0) {
1570                     char buf[512];
1571                     av_get_channel_layout_string(buf, sizeof(buf), -1, avctx->channel_layout);
1572                     av_log(avctx, AV_LOG_ERROR, "Specified channel layout '%s' is not supported\n", buf);
1573                     ret = AVERROR(EINVAL);
1574                     goto free_and_end;
1575                 }
1576             }
1577         }
1578         if (avctx->channel_layout && avctx->channels) {
1579             int channels = av_get_channel_layout_nb_channels(avctx->channel_layout);
1580             if (channels != avctx->channels) {
1581                 char buf[512];
1582                 av_get_channel_layout_string(buf, sizeof(buf), -1, avctx->channel_layout);
1583                 av_log(avctx, AV_LOG_ERROR,
1584                        "Channel layout '%s' with %d channels does not match number of specified channels %d\n",
1585                        buf, channels, avctx->channels);
1586                 ret = AVERROR(EINVAL);
1587                 goto free_and_end;
1588             }
1589         } else if (avctx->channel_layout) {
1590             avctx->channels = av_get_channel_layout_nb_channels(avctx->channel_layout);
1591         }
1592         if(avctx->codec_type == AVMEDIA_TYPE_VIDEO) {
1593             if (avctx->width <= 0 || avctx->height <= 0) {
1594                 av_log(avctx, AV_LOG_ERROR, "dimensions not set\n");
1595                 ret = AVERROR(EINVAL);
1596                 goto free_and_end;
1597             }
1598         }
1599         if (   (avctx->codec_type == AVMEDIA_TYPE_VIDEO || avctx->codec_type == AVMEDIA_TYPE_AUDIO)
1600             && avctx->bit_rate>0 && avctx->bit_rate<1000) {
1601             av_log(avctx, AV_LOG_WARNING, "Bitrate %d is extremely low, maybe you mean %dk\n", avctx->bit_rate, avctx->bit_rate);
1602         }
1603
1604         if (!avctx->rc_initial_buffer_occupancy)
1605             avctx->rc_initial_buffer_occupancy = avctx->rc_buffer_size * 3 / 4;
1606     }
1607
1608     avctx->pts_correction_num_faulty_pts =
1609     avctx->pts_correction_num_faulty_dts = 0;
1610     avctx->pts_correction_last_pts =
1611     avctx->pts_correction_last_dts = INT64_MIN;
1612
1613     if (   avctx->codec->init && (!(avctx->active_thread_type&FF_THREAD_FRAME)
1614         || avctx->internal->frame_thread_encoder)) {
1615         ret = avctx->codec->init(avctx);
1616         if (ret < 0) {
1617             goto free_and_end;
1618         }
1619     }
1620
1621     ret=0;
1622
1623 #if FF_API_AUDIOENC_DELAY
1624     if (av_codec_is_encoder(avctx->codec))
1625         avctx->delay = avctx->initial_padding;
1626 #endif
1627
1628     if (av_codec_is_decoder(avctx->codec)) {
1629         if (!avctx->bit_rate)
1630             avctx->bit_rate = get_bit_rate(avctx);
1631         /* validate channel layout from the decoder */
1632         if (avctx->channel_layout) {
1633             int channels = av_get_channel_layout_nb_channels(avctx->channel_layout);
1634             if (!avctx->channels)
1635                 avctx->channels = channels;
1636             else if (channels != avctx->channels) {
1637                 char buf[512];
1638                 av_get_channel_layout_string(buf, sizeof(buf), -1, avctx->channel_layout);
1639                 av_log(avctx, AV_LOG_WARNING,
1640                        "Channel layout '%s' with %d channels does not match specified number of channels %d: "
1641                        "ignoring specified channel layout\n",
1642                        buf, channels, avctx->channels);
1643                 avctx->channel_layout = 0;
1644             }
1645         }
1646         if (avctx->channels && avctx->channels < 0 ||
1647             avctx->channels > FF_SANE_NB_CHANNELS) {
1648             ret = AVERROR(EINVAL);
1649             goto free_and_end;
1650         }
1651         if (avctx->sub_charenc) {
1652             if (avctx->codec_type != AVMEDIA_TYPE_SUBTITLE) {
1653                 av_log(avctx, AV_LOG_ERROR, "Character encoding is only "
1654                        "supported with subtitles codecs\n");
1655                 ret = AVERROR(EINVAL);
1656                 goto free_and_end;
1657             } else if (avctx->codec_descriptor->props & AV_CODEC_PROP_BITMAP_SUB) {
1658                 av_log(avctx, AV_LOG_WARNING, "Codec '%s' is bitmap-based, "
1659                        "subtitles character encoding will be ignored\n",
1660                        avctx->codec_descriptor->name);
1661                 avctx->sub_charenc_mode = FF_SUB_CHARENC_MODE_DO_NOTHING;
1662             } else {
1663                 /* input character encoding is set for a text based subtitle
1664                  * codec at this point */
1665                 if (avctx->sub_charenc_mode == FF_SUB_CHARENC_MODE_AUTOMATIC)
1666                     avctx->sub_charenc_mode = FF_SUB_CHARENC_MODE_PRE_DECODER;
1667
1668                 if (avctx->sub_charenc_mode == FF_SUB_CHARENC_MODE_PRE_DECODER) {
1669 #if CONFIG_ICONV
1670                     iconv_t cd = iconv_open("UTF-8", avctx->sub_charenc);
1671                     if (cd == (iconv_t)-1) {
1672                         ret = AVERROR(errno);
1673                         av_log(avctx, AV_LOG_ERROR, "Unable to open iconv context "
1674                                "with input character encoding \"%s\"\n", avctx->sub_charenc);
1675                         goto free_and_end;
1676                     }
1677                     iconv_close(cd);
1678 #else
1679                     av_log(avctx, AV_LOG_ERROR, "Character encoding subtitles "
1680                            "conversion needs a libavcodec built with iconv support "
1681                            "for this codec\n");
1682                     ret = AVERROR(ENOSYS);
1683                     goto free_and_end;
1684 #endif
1685                 }
1686             }
1687         }
1688
1689 #if FF_API_AVCTX_TIMEBASE
1690         if (avctx->framerate.num > 0 && avctx->framerate.den > 0)
1691             avctx->time_base = av_inv_q(av_mul_q(avctx->framerate, (AVRational){avctx->ticks_per_frame, 1}));
1692 #endif
1693     }
1694     if (codec->priv_data_size > 0 && avctx->priv_data && codec->priv_class) {
1695         av_assert0(*(const AVClass **)avctx->priv_data == codec->priv_class);
1696     }
1697
1698 end:
1699     ff_unlock_avcodec();
1700     if (options) {
1701         av_dict_free(options);
1702         *options = tmp;
1703     }
1704
1705     return ret;
1706 free_and_end:
1707     av_dict_free(&tmp);
1708     if (codec->priv_class && codec->priv_data_size)
1709         av_opt_free(avctx->priv_data);
1710     av_freep(&avctx->priv_data);
1711     if (avctx->internal) {
1712         av_frame_free(&avctx->internal->to_free);
1713         av_freep(&avctx->internal->pool);
1714     }
1715     av_freep(&avctx->internal);
1716     avctx->codec = NULL;
1717     goto end;
1718 }
1719
1720 int ff_alloc_packet2(AVCodecContext *avctx, AVPacket *avpkt, int64_t size)
1721 {
1722     if (avpkt->size < 0) {
1723         av_log(avctx, AV_LOG_ERROR, "Invalid negative user packet size %d\n", avpkt->size);
1724         return AVERROR(EINVAL);
1725     }
1726     if (size < 0 || size > INT_MAX - FF_INPUT_BUFFER_PADDING_SIZE) {
1727         av_log(avctx, AV_LOG_ERROR, "Invalid minimum required packet size %"PRId64" (max allowed is %d)\n",
1728                size, INT_MAX - FF_INPUT_BUFFER_PADDING_SIZE);
1729         return AVERROR(EINVAL);
1730     }
1731
1732     if (avctx) {
1733         av_assert0(!avpkt->data || avpkt->data != avctx->internal->byte_buffer);
1734         if (!avpkt->data || avpkt->size < size) {
1735             av_fast_padded_malloc(&avctx->internal->byte_buffer, &avctx->internal->byte_buffer_size, size);
1736             avpkt->data = avctx->internal->byte_buffer;
1737             avpkt->size = avctx->internal->byte_buffer_size;
1738 #if FF_API_DESTRUCT_PACKET
1739 FF_DISABLE_DEPRECATION_WARNINGS
1740             avpkt->destruct = NULL;
1741 FF_ENABLE_DEPRECATION_WARNINGS
1742 #endif
1743         }
1744     }
1745
1746     if (avpkt->data) {
1747         AVBufferRef *buf = avpkt->buf;
1748 #if FF_API_DESTRUCT_PACKET
1749 FF_DISABLE_DEPRECATION_WARNINGS
1750         void *destruct = avpkt->destruct;
1751 FF_ENABLE_DEPRECATION_WARNINGS
1752 #endif
1753
1754         if (avpkt->size < size) {
1755             av_log(avctx, AV_LOG_ERROR, "User packet is too small (%d < %"PRId64")\n", avpkt->size, size);
1756             return AVERROR(EINVAL);
1757         }
1758
1759         av_init_packet(avpkt);
1760 #if FF_API_DESTRUCT_PACKET
1761 FF_DISABLE_DEPRECATION_WARNINGS
1762         avpkt->destruct = destruct;
1763 FF_ENABLE_DEPRECATION_WARNINGS
1764 #endif
1765         avpkt->buf      = buf;
1766         avpkt->size     = size;
1767         return 0;
1768     } else {
1769         int ret = av_new_packet(avpkt, size);
1770         if (ret < 0)
1771             av_log(avctx, AV_LOG_ERROR, "Failed to allocate packet of size %"PRId64"\n", size);
1772         return ret;
1773     }
1774 }
1775
1776 int ff_alloc_packet(AVPacket *avpkt, int size)
1777 {
1778     return ff_alloc_packet2(NULL, avpkt, size);
1779 }
1780
1781 /**
1782  * Pad last frame with silence.
1783  */
1784 static int pad_last_frame(AVCodecContext *s, AVFrame **dst, const AVFrame *src)
1785 {
1786     AVFrame *frame = NULL;
1787     int ret;
1788
1789     if (!(frame = av_frame_alloc()))
1790         return AVERROR(ENOMEM);
1791
1792     frame->format         = src->format;
1793     frame->channel_layout = src->channel_layout;
1794     av_frame_set_channels(frame, av_frame_get_channels(src));
1795     frame->nb_samples     = s->frame_size;
1796     ret = av_frame_get_buffer(frame, 32);
1797     if (ret < 0)
1798         goto fail;
1799
1800     ret = av_frame_copy_props(frame, src);
1801     if (ret < 0)
1802         goto fail;
1803
1804     if ((ret = av_samples_copy(frame->extended_data, src->extended_data, 0, 0,
1805                                src->nb_samples, s->channels, s->sample_fmt)) < 0)
1806         goto fail;
1807     if ((ret = av_samples_set_silence(frame->extended_data, src->nb_samples,
1808                                       frame->nb_samples - src->nb_samples,
1809                                       s->channels, s->sample_fmt)) < 0)
1810         goto fail;
1811
1812     *dst = frame;
1813
1814     return 0;
1815
1816 fail:
1817     av_frame_free(&frame);
1818     return ret;
1819 }
1820
1821 int attribute_align_arg avcodec_encode_audio2(AVCodecContext *avctx,
1822                                               AVPacket *avpkt,
1823                                               const AVFrame *frame,
1824                                               int *got_packet_ptr)
1825 {
1826     AVFrame *extended_frame = NULL;
1827     AVFrame *padded_frame = NULL;
1828     int ret;
1829     AVPacket user_pkt = *avpkt;
1830     int needs_realloc = !user_pkt.data;
1831
1832     *got_packet_ptr = 0;
1833
1834     if (!(avctx->codec->capabilities & CODEC_CAP_DELAY) && !frame) {
1835         av_free_packet(avpkt);
1836         av_init_packet(avpkt);
1837         return 0;
1838     }
1839
1840     /* ensure that extended_data is properly set */
1841     if (frame && !frame->extended_data) {
1842         if (av_sample_fmt_is_planar(avctx->sample_fmt) &&
1843             avctx->channels > AV_NUM_DATA_POINTERS) {
1844             av_log(avctx, AV_LOG_ERROR, "Encoding to a planar sample format, "
1845                                         "with more than %d channels, but extended_data is not set.\n",
1846                    AV_NUM_DATA_POINTERS);
1847             return AVERROR(EINVAL);
1848         }
1849         av_log(avctx, AV_LOG_WARNING, "extended_data is not set.\n");
1850
1851         extended_frame = av_frame_alloc();
1852         if (!extended_frame)
1853             return AVERROR(ENOMEM);
1854
1855         memcpy(extended_frame, frame, sizeof(AVFrame));
1856         extended_frame->extended_data = extended_frame->data;
1857         frame = extended_frame;
1858     }
1859
1860     /* check for valid frame size */
1861     if (frame) {
1862         if (avctx->codec->capabilities & CODEC_CAP_SMALL_LAST_FRAME) {
1863             if (frame->nb_samples > avctx->frame_size) {
1864                 av_log(avctx, AV_LOG_ERROR, "more samples than frame size (avcodec_encode_audio2)\n");
1865                 ret = AVERROR(EINVAL);
1866                 goto end;
1867             }
1868         } else if (!(avctx->codec->capabilities & CODEC_CAP_VARIABLE_FRAME_SIZE)) {
1869             if (frame->nb_samples < avctx->frame_size &&
1870                 !avctx->internal->last_audio_frame) {
1871                 ret = pad_last_frame(avctx, &padded_frame, frame);
1872                 if (ret < 0)
1873                     goto end;
1874
1875                 frame = padded_frame;
1876                 avctx->internal->last_audio_frame = 1;
1877             }
1878
1879             if (frame->nb_samples != avctx->frame_size) {
1880                 av_log(avctx, AV_LOG_ERROR, "nb_samples (%d) != frame_size (%d) (avcodec_encode_audio2)\n", frame->nb_samples, avctx->frame_size);
1881                 ret = AVERROR(EINVAL);
1882                 goto end;
1883             }
1884         }
1885     }
1886
1887     ret = avctx->codec->encode2(avctx, avpkt, frame, got_packet_ptr);
1888     if (!ret) {
1889         if (*got_packet_ptr) {
1890             if (!(avctx->codec->capabilities & CODEC_CAP_DELAY)) {
1891                 if (avpkt->pts == AV_NOPTS_VALUE)
1892                     avpkt->pts = frame->pts;
1893                 if (!avpkt->duration)
1894                     avpkt->duration = ff_samples_to_time_base(avctx,
1895                                                               frame->nb_samples);
1896             }
1897             avpkt->dts = avpkt->pts;
1898         } else {
1899             avpkt->size = 0;
1900         }
1901     }
1902     if (avpkt->data && avpkt->data == avctx->internal->byte_buffer) {
1903         needs_realloc = 0;
1904         if (user_pkt.data) {
1905             if (user_pkt.size >= avpkt->size) {
1906                 memcpy(user_pkt.data, avpkt->data, avpkt->size);
1907             } else {
1908                 av_log(avctx, AV_LOG_ERROR, "Provided packet is too small, needs to be %d\n", avpkt->size);
1909                 avpkt->size = user_pkt.size;
1910                 ret = -1;
1911             }
1912             avpkt->buf      = user_pkt.buf;
1913             avpkt->data     = user_pkt.data;
1914 #if FF_API_DESTRUCT_PACKET
1915 FF_DISABLE_DEPRECATION_WARNINGS
1916             avpkt->destruct = user_pkt.destruct;
1917 FF_ENABLE_DEPRECATION_WARNINGS
1918 #endif
1919         } else {
1920             if (av_dup_packet(avpkt) < 0) {
1921                 ret = AVERROR(ENOMEM);
1922             }
1923         }
1924     }
1925
1926     if (!ret) {
1927         if (needs_realloc && avpkt->data) {
1928             ret = av_buffer_realloc(&avpkt->buf, avpkt->size + FF_INPUT_BUFFER_PADDING_SIZE);
1929             if (ret >= 0)
1930                 avpkt->data = avpkt->buf->data;
1931         }
1932
1933         avctx->frame_number++;
1934     }
1935
1936     if (ret < 0 || !*got_packet_ptr) {
1937         av_free_packet(avpkt);
1938         av_init_packet(avpkt);
1939         goto end;
1940     }
1941
1942     /* NOTE: if we add any audio encoders which output non-keyframe packets,
1943      *       this needs to be moved to the encoders, but for now we can do it
1944      *       here to simplify things */
1945     avpkt->flags |= AV_PKT_FLAG_KEY;
1946
1947 end:
1948     av_frame_free(&padded_frame);
1949     av_free(extended_frame);
1950
1951 #if FF_API_AUDIOENC_DELAY
1952     avctx->delay = avctx->initial_padding;
1953 #endif
1954
1955     return ret;
1956 }
1957
1958 #if FF_API_OLD_ENCODE_AUDIO
1959 int attribute_align_arg avcodec_encode_audio(AVCodecContext *avctx,
1960                                              uint8_t *buf, int buf_size,
1961                                              const short *samples)
1962 {
1963     AVPacket pkt;
1964     AVFrame *frame;
1965     int ret, samples_size, got_packet;
1966
1967     av_init_packet(&pkt);
1968     pkt.data = buf;
1969     pkt.size = buf_size;
1970
1971     if (samples) {
1972         frame = av_frame_alloc();
1973         if (!frame)
1974             return AVERROR(ENOMEM);
1975
1976         if (avctx->frame_size) {
1977             frame->nb_samples = avctx->frame_size;
1978         } else {
1979             /* if frame_size is not set, the number of samples must be
1980              * calculated from the buffer size */
1981             int64_t nb_samples;
1982             if (!av_get_bits_per_sample(avctx->codec_id)) {
1983                 av_log(avctx, AV_LOG_ERROR, "avcodec_encode_audio() does not "
1984                                             "support this codec\n");
1985                 av_frame_free(&frame);
1986                 return AVERROR(EINVAL);
1987             }
1988             nb_samples = (int64_t)buf_size * 8 /
1989                          (av_get_bits_per_sample(avctx->codec_id) *
1990                           avctx->channels);
1991             if (nb_samples >= INT_MAX) {
1992                 av_frame_free(&frame);
1993                 return AVERROR(EINVAL);
1994             }
1995             frame->nb_samples = nb_samples;
1996         }
1997
1998         /* it is assumed that the samples buffer is large enough based on the
1999          * relevant parameters */
2000         samples_size = av_samples_get_buffer_size(NULL, avctx->channels,
2001                                                   frame->nb_samples,
2002                                                   avctx->sample_fmt, 1);
2003         if ((ret = avcodec_fill_audio_frame(frame, avctx->channels,
2004                                             avctx->sample_fmt,
2005                                             (const uint8_t *)samples,
2006                                             samples_size, 1)) < 0) {
2007             av_frame_free(&frame);
2008             return ret;
2009         }
2010
2011         /* fabricate frame pts from sample count.
2012          * this is needed because the avcodec_encode_audio() API does not have
2013          * a way for the user to provide pts */
2014         if (avctx->sample_rate && avctx->time_base.num)
2015             frame->pts = ff_samples_to_time_base(avctx,
2016                                                  avctx->internal->sample_count);
2017         else
2018             frame->pts = AV_NOPTS_VALUE;
2019         avctx->internal->sample_count += frame->nb_samples;
2020     } else {
2021         frame = NULL;
2022     }
2023
2024     got_packet = 0;
2025     ret = avcodec_encode_audio2(avctx, &pkt, frame, &got_packet);
2026     if (!ret && got_packet && avctx->coded_frame) {
2027         avctx->coded_frame->pts       = pkt.pts;
2028         avctx->coded_frame->key_frame = !!(pkt.flags & AV_PKT_FLAG_KEY);
2029     }
2030     /* free any side data since we cannot return it */
2031     av_packet_free_side_data(&pkt);
2032
2033     if (frame && frame->extended_data != frame->data)
2034         av_freep(&frame->extended_data);
2035
2036     av_frame_free(&frame);
2037     return ret ? ret : pkt.size;
2038 }
2039
2040 #endif
2041
2042 #if FF_API_OLD_ENCODE_VIDEO
2043 int attribute_align_arg avcodec_encode_video(AVCodecContext *avctx, uint8_t *buf, int buf_size,
2044                                              const AVFrame *pict)
2045 {
2046     AVPacket pkt;
2047     int ret, got_packet = 0;
2048
2049     if (buf_size < FF_MIN_BUFFER_SIZE) {
2050         av_log(avctx, AV_LOG_ERROR, "buffer smaller than minimum size\n");
2051         return -1;
2052     }
2053
2054     av_init_packet(&pkt);
2055     pkt.data = buf;
2056     pkt.size = buf_size;
2057
2058     ret = avcodec_encode_video2(avctx, &pkt, pict, &got_packet);
2059     if (!ret && got_packet && avctx->coded_frame) {
2060         avctx->coded_frame->pts       = pkt.pts;
2061         avctx->coded_frame->key_frame = !!(pkt.flags & AV_PKT_FLAG_KEY);
2062     }
2063
2064     /* free any side data since we cannot return it */
2065     if (pkt.side_data_elems > 0) {
2066         int i;
2067         for (i = 0; i < pkt.side_data_elems; i++)
2068             av_free(pkt.side_data[i].data);
2069         av_freep(&pkt.side_data);
2070         pkt.side_data_elems = 0;
2071     }
2072
2073     return ret ? ret : pkt.size;
2074 }
2075
2076 #endif
2077
2078 int attribute_align_arg avcodec_encode_video2(AVCodecContext *avctx,
2079                                               AVPacket *avpkt,
2080                                               const AVFrame *frame,
2081                                               int *got_packet_ptr)
2082 {
2083     int ret;
2084     AVPacket user_pkt = *avpkt;
2085     int needs_realloc = !user_pkt.data;
2086
2087     *got_packet_ptr = 0;
2088
2089     if(CONFIG_FRAME_THREAD_ENCODER &&
2090        avctx->internal->frame_thread_encoder && (avctx->active_thread_type&FF_THREAD_FRAME))
2091         return ff_thread_video_encode_frame(avctx, avpkt, frame, got_packet_ptr);
2092
2093     if ((avctx->flags&CODEC_FLAG_PASS1) && avctx->stats_out)
2094         avctx->stats_out[0] = '\0';
2095
2096     if (!(avctx->codec->capabilities & CODEC_CAP_DELAY) && !frame) {
2097         av_free_packet(avpkt);
2098         av_init_packet(avpkt);
2099         avpkt->size = 0;
2100         return 0;
2101     }
2102
2103     if (av_image_check_size(avctx->width, avctx->height, 0, avctx))
2104         return AVERROR(EINVAL);
2105
2106     if (frame && frame->format == AV_PIX_FMT_NONE)
2107         av_log(avctx, AV_LOG_WARNING, "AVFrame.format is not set\n");
2108     if (frame && (frame->width == 0 || frame->height == 0))
2109         av_log(avctx, AV_LOG_WARNING, "AVFrame.width or height is not set\n");
2110
2111     av_assert0(avctx->codec->encode2);
2112
2113     ret = avctx->codec->encode2(avctx, avpkt, frame, got_packet_ptr);
2114     av_assert0(ret <= 0);
2115
2116     if (avpkt->data && avpkt->data == avctx->internal->byte_buffer) {
2117         needs_realloc = 0;
2118         if (user_pkt.data) {
2119             if (user_pkt.size >= avpkt->size) {
2120                 memcpy(user_pkt.data, avpkt->data, avpkt->size);
2121             } else {
2122                 av_log(avctx, AV_LOG_ERROR, "Provided packet is too small, needs to be %d\n", avpkt->size);
2123                 avpkt->size = user_pkt.size;
2124                 ret = -1;
2125             }
2126             avpkt->buf      = user_pkt.buf;
2127             avpkt->data     = user_pkt.data;
2128 #if FF_API_DESTRUCT_PACKET
2129 FF_DISABLE_DEPRECATION_WARNINGS
2130             avpkt->destruct = user_pkt.destruct;
2131 FF_ENABLE_DEPRECATION_WARNINGS
2132 #endif
2133         } else {
2134             if (av_dup_packet(avpkt) < 0) {
2135                 ret = AVERROR(ENOMEM);
2136             }
2137         }
2138     }
2139
2140     if (!ret) {
2141         if (!*got_packet_ptr)
2142             avpkt->size = 0;
2143         else if (!(avctx->codec->capabilities & CODEC_CAP_DELAY))
2144             avpkt->pts = avpkt->dts = frame->pts;
2145
2146         if (needs_realloc && avpkt->data) {
2147             ret = av_buffer_realloc(&avpkt->buf, avpkt->size + FF_INPUT_BUFFER_PADDING_SIZE);
2148             if (ret >= 0)
2149                 avpkt->data = avpkt->buf->data;
2150         }
2151
2152         avctx->frame_number++;
2153     }
2154
2155     if (ret < 0 || !*got_packet_ptr)
2156         av_free_packet(avpkt);
2157     else
2158         av_packet_merge_side_data(avpkt);
2159
2160     emms_c();
2161     return ret;
2162 }
2163
2164 int avcodec_encode_subtitle(AVCodecContext *avctx, uint8_t *buf, int buf_size,
2165                             const AVSubtitle *sub)
2166 {
2167     int ret;
2168     if (sub->start_display_time) {
2169         av_log(avctx, AV_LOG_ERROR, "start_display_time must be 0.\n");
2170         return -1;
2171     }
2172
2173     ret = avctx->codec->encode_sub(avctx, buf, buf_size, sub);
2174     avctx->frame_number++;
2175     return ret;
2176 }
2177
2178 /**
2179  * Attempt to guess proper monotonic timestamps for decoded video frames
2180  * which might have incorrect times. Input timestamps may wrap around, in
2181  * which case the output will as well.
2182  *
2183  * @param pts the pts field of the decoded AVPacket, as passed through
2184  * AVFrame.pkt_pts
2185  * @param dts the dts field of the decoded AVPacket
2186  * @return one of the input values, may be AV_NOPTS_VALUE
2187  */
2188 static int64_t guess_correct_pts(AVCodecContext *ctx,
2189                                  int64_t reordered_pts, int64_t dts)
2190 {
2191     int64_t pts = AV_NOPTS_VALUE;
2192
2193     if (dts != AV_NOPTS_VALUE) {
2194         ctx->pts_correction_num_faulty_dts += dts <= ctx->pts_correction_last_dts;
2195         ctx->pts_correction_last_dts = dts;
2196     } else if (reordered_pts != AV_NOPTS_VALUE)
2197         ctx->pts_correction_last_dts = reordered_pts;
2198
2199     if (reordered_pts != AV_NOPTS_VALUE) {
2200         ctx->pts_correction_num_faulty_pts += reordered_pts <= ctx->pts_correction_last_pts;
2201         ctx->pts_correction_last_pts = reordered_pts;
2202     } else if(dts != AV_NOPTS_VALUE)
2203         ctx->pts_correction_last_pts = dts;
2204
2205     if ((ctx->pts_correction_num_faulty_pts<=ctx->pts_correction_num_faulty_dts || dts == AV_NOPTS_VALUE)
2206        && reordered_pts != AV_NOPTS_VALUE)
2207         pts = reordered_pts;
2208     else
2209         pts = dts;
2210
2211     return pts;
2212 }
2213
2214 static int apply_param_change(AVCodecContext *avctx, AVPacket *avpkt)
2215 {
2216     int size = 0, ret;
2217     const uint8_t *data;
2218     uint32_t flags;
2219
2220     data = av_packet_get_side_data(avpkt, AV_PKT_DATA_PARAM_CHANGE, &size);
2221     if (!data)
2222         return 0;
2223
2224     if (!(avctx->codec->capabilities & CODEC_CAP_PARAM_CHANGE)) {
2225         av_log(avctx, AV_LOG_ERROR, "This decoder does not support parameter "
2226                "changes, but PARAM_CHANGE side data was sent to it.\n");
2227         return AVERROR(EINVAL);
2228     }
2229
2230     if (size < 4)
2231         goto fail;
2232
2233     flags = bytestream_get_le32(&data);
2234     size -= 4;
2235
2236     if (flags & AV_SIDE_DATA_PARAM_CHANGE_CHANNEL_COUNT) {
2237         if (size < 4)
2238             goto fail;
2239         avctx->channels = bytestream_get_le32(&data);
2240         size -= 4;
2241     }
2242     if (flags & AV_SIDE_DATA_PARAM_CHANGE_CHANNEL_LAYOUT) {
2243         if (size < 8)
2244             goto fail;
2245         avctx->channel_layout = bytestream_get_le64(&data);
2246         size -= 8;
2247     }
2248     if (flags & AV_SIDE_DATA_PARAM_CHANGE_SAMPLE_RATE) {
2249         if (size < 4)
2250             goto fail;
2251         avctx->sample_rate = bytestream_get_le32(&data);
2252         size -= 4;
2253     }
2254     if (flags & AV_SIDE_DATA_PARAM_CHANGE_DIMENSIONS) {
2255         if (size < 8)
2256             goto fail;
2257         avctx->width  = bytestream_get_le32(&data);
2258         avctx->height = bytestream_get_le32(&data);
2259         size -= 8;
2260         ret = ff_set_dimensions(avctx, avctx->width, avctx->height);
2261         if (ret < 0)
2262             return ret;
2263     }
2264
2265     return 0;
2266 fail:
2267     av_log(avctx, AV_LOG_ERROR, "PARAM_CHANGE side data too small.\n");
2268     return AVERROR_INVALIDDATA;
2269 }
2270
2271 static int add_metadata_from_side_data(AVCodecContext *avctx, AVFrame *frame)
2272 {
2273     int size;
2274     const uint8_t *side_metadata;
2275
2276     AVDictionary **frame_md = avpriv_frame_get_metadatap(frame);
2277
2278     side_metadata = av_packet_get_side_data(avctx->internal->pkt,
2279                                             AV_PKT_DATA_STRINGS_METADATA, &size);
2280     return av_packet_unpack_dictionary(side_metadata, size, frame_md);
2281 }
2282
2283 static int unrefcount_frame(AVCodecInternal *avci, AVFrame *frame)
2284 {
2285     int ret;
2286
2287     /* move the original frame to our backup */
2288     av_frame_unref(avci->to_free);
2289     av_frame_move_ref(avci->to_free, frame);
2290
2291     /* now copy everything except the AVBufferRefs back
2292      * note that we make a COPY of the side data, so calling av_frame_free() on
2293      * the caller's frame will work properly */
2294     ret = av_frame_copy_props(frame, avci->to_free);
2295     if (ret < 0)
2296         return ret;
2297
2298     memcpy(frame->data,     avci->to_free->data,     sizeof(frame->data));
2299     memcpy(frame->linesize, avci->to_free->linesize, sizeof(frame->linesize));
2300     if (avci->to_free->extended_data != avci->to_free->data) {
2301         int planes = av_frame_get_channels(avci->to_free);
2302         int size   = planes * sizeof(*frame->extended_data);
2303
2304         if (!size) {
2305             av_frame_unref(frame);
2306             return AVERROR_BUG;
2307         }
2308
2309         frame->extended_data = av_malloc(size);
2310         if (!frame->extended_data) {
2311             av_frame_unref(frame);
2312             return AVERROR(ENOMEM);
2313         }
2314         memcpy(frame->extended_data, avci->to_free->extended_data,
2315                size);
2316     } else
2317         frame->extended_data = frame->data;
2318
2319     frame->format         = avci->to_free->format;
2320     frame->width          = avci->to_free->width;
2321     frame->height         = avci->to_free->height;
2322     frame->channel_layout = avci->to_free->channel_layout;
2323     frame->nb_samples     = avci->to_free->nb_samples;
2324     av_frame_set_channels(frame, av_frame_get_channels(avci->to_free));
2325
2326     return 0;
2327 }
2328
2329 int attribute_align_arg avcodec_decode_video2(AVCodecContext *avctx, AVFrame *picture,
2330                                               int *got_picture_ptr,
2331                                               const AVPacket *avpkt)
2332 {
2333     AVCodecInternal *avci = avctx->internal;
2334     int ret;
2335     // copy to ensure we do not change avpkt
2336     AVPacket tmp = *avpkt;
2337
2338     if (!avctx->codec)
2339         return AVERROR(EINVAL);
2340     if (avctx->codec->type != AVMEDIA_TYPE_VIDEO) {
2341         av_log(avctx, AV_LOG_ERROR, "Invalid media type for video\n");
2342         return AVERROR(EINVAL);
2343     }
2344
2345     *got_picture_ptr = 0;
2346     if ((avctx->coded_width || avctx->coded_height) && av_image_check_size(avctx->coded_width, avctx->coded_height, 0, avctx))
2347         return AVERROR(EINVAL);
2348
2349     av_frame_unref(picture);
2350
2351     if ((avctx->codec->capabilities & CODEC_CAP_DELAY) || avpkt->size || (avctx->active_thread_type & FF_THREAD_FRAME)) {
2352         int did_split = av_packet_split_side_data(&tmp);
2353         ret = apply_param_change(avctx, &tmp);
2354         if (ret < 0) {
2355             av_log(avctx, AV_LOG_ERROR, "Error applying parameter changes.\n");
2356             if (avctx->err_recognition & AV_EF_EXPLODE)
2357                 goto fail;
2358         }
2359
2360         avctx->internal->pkt = &tmp;
2361         if (HAVE_THREADS && avctx->active_thread_type & FF_THREAD_FRAME)
2362             ret = ff_thread_decode_frame(avctx, picture, got_picture_ptr,
2363                                          &tmp);
2364         else {
2365             ret = avctx->codec->decode(avctx, picture, got_picture_ptr,
2366                                        &tmp);
2367             picture->pkt_dts = avpkt->dts;
2368
2369             if(!avctx->has_b_frames){
2370                 av_frame_set_pkt_pos(picture, avpkt->pos);
2371             }
2372             //FIXME these should be under if(!avctx->has_b_frames)
2373             /* get_buffer is supposed to set frame parameters */
2374             if (!(avctx->codec->capabilities & CODEC_CAP_DR1)) {
2375                 if (!picture->sample_aspect_ratio.num)    picture->sample_aspect_ratio = avctx->sample_aspect_ratio;
2376                 if (!picture->width)                      picture->width               = avctx->width;
2377                 if (!picture->height)                     picture->height              = avctx->height;
2378                 if (picture->format == AV_PIX_FMT_NONE)   picture->format              = avctx->pix_fmt;
2379             }
2380         }
2381         add_metadata_from_side_data(avctx, picture);
2382
2383 fail:
2384         emms_c(); //needed to avoid an emms_c() call before every return;
2385
2386         avctx->internal->pkt = NULL;
2387         if (did_split) {
2388             av_packet_free_side_data(&tmp);
2389             if(ret == tmp.size)
2390                 ret = avpkt->size;
2391         }
2392
2393         if (*got_picture_ptr) {
2394             if (!avctx->refcounted_frames) {
2395                 int err = unrefcount_frame(avci, picture);
2396                 if (err < 0)
2397                     return err;
2398             }
2399
2400             avctx->frame_number++;
2401             av_frame_set_best_effort_timestamp(picture,
2402                                                guess_correct_pts(avctx,
2403                                                                  picture->pkt_pts,
2404                                                                  picture->pkt_dts));
2405         } else
2406             av_frame_unref(picture);
2407     } else
2408         ret = 0;
2409
2410     /* many decoders assign whole AVFrames, thus overwriting extended_data;
2411      * make sure it's set correctly */
2412     av_assert0(!picture->extended_data || picture->extended_data == picture->data);
2413
2414 #if FF_API_AVCTX_TIMEBASE
2415     if (avctx->framerate.num > 0 && avctx->framerate.den > 0)
2416         avctx->time_base = av_inv_q(av_mul_q(avctx->framerate, (AVRational){avctx->ticks_per_frame, 1}));
2417 #endif
2418
2419     return ret;
2420 }
2421
2422 #if FF_API_OLD_DECODE_AUDIO
2423 int attribute_align_arg avcodec_decode_audio3(AVCodecContext *avctx, int16_t *samples,
2424                                               int *frame_size_ptr,
2425                                               AVPacket *avpkt)
2426 {
2427     AVFrame *frame = av_frame_alloc();
2428     int ret, got_frame = 0;
2429
2430     if (!frame)
2431         return AVERROR(ENOMEM);
2432     if (avctx->get_buffer != avcodec_default_get_buffer) {
2433         av_log(avctx, AV_LOG_ERROR, "Custom get_buffer() for use with"
2434                                     "avcodec_decode_audio3() detected. Overriding with avcodec_default_get_buffer\n");
2435         av_log(avctx, AV_LOG_ERROR, "Please port your application to "
2436                                     "avcodec_decode_audio4()\n");
2437         avctx->get_buffer = avcodec_default_get_buffer;
2438         avctx->release_buffer = avcodec_default_release_buffer;
2439     }
2440
2441     ret = avcodec_decode_audio4(avctx, frame, &got_frame, avpkt);
2442
2443     if (ret >= 0 && got_frame) {
2444         int ch, plane_size;
2445         int planar    = av_sample_fmt_is_planar(avctx->sample_fmt);
2446         int data_size = av_samples_get_buffer_size(&plane_size, avctx->channels,
2447                                                    frame->nb_samples,
2448                                                    avctx->sample_fmt, 1);
2449         if (*frame_size_ptr < data_size) {
2450             av_log(avctx, AV_LOG_ERROR, "output buffer size is too small for "
2451                                         "the current frame (%d < %d)\n", *frame_size_ptr, data_size);
2452             av_frame_free(&frame);
2453             return AVERROR(EINVAL);
2454         }
2455
2456         memcpy(samples, frame->extended_data[0], plane_size);
2457
2458         if (planar && avctx->channels > 1) {
2459             uint8_t *out = ((uint8_t *)samples) + plane_size;
2460             for (ch = 1; ch < avctx->channels; ch++) {
2461                 memcpy(out, frame->extended_data[ch], plane_size);
2462                 out += plane_size;
2463             }
2464         }
2465         *frame_size_ptr = data_size;
2466     } else {
2467         *frame_size_ptr = 0;
2468     }
2469     av_frame_free(&frame);
2470     return ret;
2471 }
2472
2473 #endif
2474
2475 int attribute_align_arg avcodec_decode_audio4(AVCodecContext *avctx,
2476                                               AVFrame *frame,
2477                                               int *got_frame_ptr,
2478                                               const AVPacket *avpkt)
2479 {
2480     AVCodecInternal *avci = avctx->internal;
2481     int ret = 0;
2482
2483     *got_frame_ptr = 0;
2484
2485     if (!avpkt->data && avpkt->size) {
2486         av_log(avctx, AV_LOG_ERROR, "invalid packet: NULL data, size != 0\n");
2487         return AVERROR(EINVAL);
2488     }
2489     if (!avctx->codec)
2490         return AVERROR(EINVAL);
2491     if (avctx->codec->type != AVMEDIA_TYPE_AUDIO) {
2492         av_log(avctx, AV_LOG_ERROR, "Invalid media type for audio\n");
2493         return AVERROR(EINVAL);
2494     }
2495
2496     av_frame_unref(frame);
2497
2498     if ((avctx->codec->capabilities & CODEC_CAP_DELAY) || avpkt->size || (avctx->active_thread_type & FF_THREAD_FRAME)) {
2499         uint8_t *side;
2500         int side_size;
2501         uint32_t discard_padding = 0;
2502         uint8_t skip_reason = 0;
2503         uint8_t discard_reason = 0;
2504         // copy to ensure we do not change avpkt
2505         AVPacket tmp = *avpkt;
2506         int did_split = av_packet_split_side_data(&tmp);
2507         ret = apply_param_change(avctx, &tmp);
2508         if (ret < 0) {
2509             av_log(avctx, AV_LOG_ERROR, "Error applying parameter changes.\n");
2510             if (avctx->err_recognition & AV_EF_EXPLODE)
2511                 goto fail;
2512         }
2513
2514         avctx->internal->pkt = &tmp;
2515         if (HAVE_THREADS && avctx->active_thread_type & FF_THREAD_FRAME)
2516             ret = ff_thread_decode_frame(avctx, frame, got_frame_ptr, &tmp);
2517         else {
2518             ret = avctx->codec->decode(avctx, frame, got_frame_ptr, &tmp);
2519             frame->pkt_dts = avpkt->dts;
2520         }
2521         if (ret >= 0 && *got_frame_ptr) {
2522             add_metadata_from_side_data(avctx, frame);
2523             avctx->frame_number++;
2524             av_frame_set_best_effort_timestamp(frame,
2525                                                guess_correct_pts(avctx,
2526                                                                  frame->pkt_pts,
2527                                                                  frame->pkt_dts));
2528             if (frame->format == AV_SAMPLE_FMT_NONE)
2529                 frame->format = avctx->sample_fmt;
2530             if (!frame->channel_layout)
2531                 frame->channel_layout = avctx->channel_layout;
2532             if (!av_frame_get_channels(frame))
2533                 av_frame_set_channels(frame, avctx->channels);
2534             if (!frame->sample_rate)
2535                 frame->sample_rate = avctx->sample_rate;
2536         }
2537
2538         side= av_packet_get_side_data(avctx->internal->pkt, AV_PKT_DATA_SKIP_SAMPLES, &side_size);
2539         if(side && side_size>=10) {
2540             avctx->internal->skip_samples = AV_RL32(side);
2541             av_log(avctx, AV_LOG_DEBUG, "skip %d samples due to side data\n",
2542                    avctx->internal->skip_samples);
2543             discard_padding = AV_RL32(side + 4);
2544             skip_reason = AV_RL8(side + 8);
2545             discard_reason = AV_RL8(side + 9);
2546         }
2547         if (avctx->internal->skip_samples && *got_frame_ptr &&
2548             !(avctx->flags2 & CODEC_FLAG2_SKIP_MANUAL)) {
2549             if(frame->nb_samples <= avctx->internal->skip_samples){
2550                 *got_frame_ptr = 0;
2551                 avctx->internal->skip_samples -= frame->nb_samples;
2552                 av_log(avctx, AV_LOG_DEBUG, "skip whole frame, skip left: %d\n",
2553                        avctx->internal->skip_samples);
2554             } else {
2555                 av_samples_copy(frame->extended_data, frame->extended_data, 0, avctx->internal->skip_samples,
2556                                 frame->nb_samples - avctx->internal->skip_samples, avctx->channels, frame->format);
2557                 if(avctx->pkt_timebase.num && avctx->sample_rate) {
2558                     int64_t diff_ts = av_rescale_q(avctx->internal->skip_samples,
2559                                                    (AVRational){1, avctx->sample_rate},
2560                                                    avctx->pkt_timebase);
2561                     if(frame->pkt_pts!=AV_NOPTS_VALUE)
2562                         frame->pkt_pts += diff_ts;
2563                     if(frame->pkt_dts!=AV_NOPTS_VALUE)
2564                         frame->pkt_dts += diff_ts;
2565                     if (av_frame_get_pkt_duration(frame) >= diff_ts)
2566                         av_frame_set_pkt_duration(frame, av_frame_get_pkt_duration(frame) - diff_ts);
2567                 } else {
2568                     av_log(avctx, AV_LOG_WARNING, "Could not update timestamps for skipped samples.\n");
2569                 }
2570                 av_log(avctx, AV_LOG_DEBUG, "skip %d/%d samples\n",
2571                        avctx->internal->skip_samples, frame->nb_samples);
2572                 frame->nb_samples -= avctx->internal->skip_samples;
2573                 avctx->internal->skip_samples = 0;
2574             }
2575         }
2576
2577         if (discard_padding > 0 && discard_padding <= frame->nb_samples && *got_frame_ptr &&
2578             !(avctx->flags2 & CODEC_FLAG2_SKIP_MANUAL)) {
2579             if (discard_padding == frame->nb_samples) {
2580                 *got_frame_ptr = 0;
2581             } else {
2582                 if(avctx->pkt_timebase.num && avctx->sample_rate) {
2583                     int64_t diff_ts = av_rescale_q(frame->nb_samples - discard_padding,
2584                                                    (AVRational){1, avctx->sample_rate},
2585                                                    avctx->pkt_timebase);
2586                     if (av_frame_get_pkt_duration(frame) >= diff_ts)
2587                         av_frame_set_pkt_duration(frame, av_frame_get_pkt_duration(frame) - diff_ts);
2588                 } else {
2589                     av_log(avctx, AV_LOG_WARNING, "Could not update timestamps for discarded samples.\n");
2590                 }
2591                 av_log(avctx, AV_LOG_DEBUG, "discard %d/%d samples\n",
2592                        discard_padding, frame->nb_samples);
2593                 frame->nb_samples -= discard_padding;
2594             }
2595         }
2596
2597         if ((avctx->flags2 & CODEC_FLAG2_SKIP_MANUAL) && *got_frame_ptr) {
2598             AVFrameSideData *fside = av_frame_new_side_data(frame, AV_FRAME_DATA_SKIP_SAMPLES, 10);
2599             if (fside) {
2600                 AV_WL32(fside->data, avctx->internal->skip_samples);
2601                 AV_WL32(fside->data + 4, discard_padding);
2602                 AV_WL8(fside->data + 8, skip_reason);
2603                 AV_WL8(fside->data + 9, discard_reason);
2604                 avctx->internal->skip_samples = 0;
2605             }
2606         }
2607 fail:
2608         avctx->internal->pkt = NULL;
2609         if (did_split) {
2610             av_packet_free_side_data(&tmp);
2611             if(ret == tmp.size)
2612                 ret = avpkt->size;
2613         }
2614
2615         if (ret >= 0 && *got_frame_ptr) {
2616             if (!avctx->refcounted_frames) {
2617                 int err = unrefcount_frame(avci, frame);
2618                 if (err < 0)
2619                     return err;
2620             }
2621         } else
2622             av_frame_unref(frame);
2623     }
2624
2625     return ret;
2626 }
2627
2628 #define UTF8_MAX_BYTES 4 /* 5 and 6 bytes sequences should not be used */
2629 static int recode_subtitle(AVCodecContext *avctx,
2630                            AVPacket *outpkt, const AVPacket *inpkt)
2631 {
2632 #if CONFIG_ICONV
2633     iconv_t cd = (iconv_t)-1;
2634     int ret = 0;
2635     char *inb, *outb;
2636     size_t inl, outl;
2637     AVPacket tmp;
2638 #endif
2639
2640     if (avctx->sub_charenc_mode != FF_SUB_CHARENC_MODE_PRE_DECODER || inpkt->size == 0)
2641         return 0;
2642
2643 #if CONFIG_ICONV
2644     cd = iconv_open("UTF-8", avctx->sub_charenc);
2645     av_assert0(cd != (iconv_t)-1);
2646
2647     inb = inpkt->data;
2648     inl = inpkt->size;
2649
2650     if (inl >= INT_MAX / UTF8_MAX_BYTES - FF_INPUT_BUFFER_PADDING_SIZE) {
2651         av_log(avctx, AV_LOG_ERROR, "Subtitles packet is too big for recoding\n");
2652         ret = AVERROR(ENOMEM);
2653         goto end;
2654     }
2655
2656     ret = av_new_packet(&tmp, inl * UTF8_MAX_BYTES);
2657     if (ret < 0)
2658         goto end;
2659     outpkt->buf  = tmp.buf;
2660     outpkt->data = tmp.data;
2661     outpkt->size = tmp.size;
2662     outb = outpkt->data;
2663     outl = outpkt->size;
2664
2665     if (iconv(cd, &inb, &inl, &outb, &outl) == (size_t)-1 ||
2666         iconv(cd, NULL, NULL, &outb, &outl) == (size_t)-1 ||
2667         outl >= outpkt->size || inl != 0) {
2668         ret = FFMIN(AVERROR(errno), -1);
2669         av_log(avctx, AV_LOG_ERROR, "Unable to recode subtitle event \"%s\" "
2670                "from %s to UTF-8\n", inpkt->data, avctx->sub_charenc);
2671         av_free_packet(&tmp);
2672         goto end;
2673     }
2674     outpkt->size -= outl;
2675     memset(outpkt->data + outpkt->size, 0, outl);
2676
2677 end:
2678     if (cd != (iconv_t)-1)
2679         iconv_close(cd);
2680     return ret;
2681 #else
2682     av_log(avctx, AV_LOG_ERROR, "requesting subtitles recoding without iconv");
2683     return AVERROR(EINVAL);
2684 #endif
2685 }
2686
2687 static int utf8_check(const uint8_t *str)
2688 {
2689     const uint8_t *byte;
2690     uint32_t codepoint, min;
2691
2692     while (*str) {
2693         byte = str;
2694         GET_UTF8(codepoint, *(byte++), return 0;);
2695         min = byte - str == 1 ? 0 : byte - str == 2 ? 0x80 :
2696               1 << (5 * (byte - str) - 4);
2697         if (codepoint < min || codepoint >= 0x110000 ||
2698             codepoint == 0xFFFE /* BOM */ ||
2699             codepoint >= 0xD800 && codepoint <= 0xDFFF /* surrogates */)
2700             return 0;
2701         str = byte;
2702     }
2703     return 1;
2704 }
2705
2706 int avcodec_decode_subtitle2(AVCodecContext *avctx, AVSubtitle *sub,
2707                              int *got_sub_ptr,
2708                              AVPacket *avpkt)
2709 {
2710     int i, ret = 0;
2711
2712     if (!avpkt->data && avpkt->size) {
2713         av_log(avctx, AV_LOG_ERROR, "invalid packet: NULL data, size != 0\n");
2714         return AVERROR(EINVAL);
2715     }
2716     if (!avctx->codec)
2717         return AVERROR(EINVAL);
2718     if (avctx->codec->type != AVMEDIA_TYPE_SUBTITLE) {
2719         av_log(avctx, AV_LOG_ERROR, "Invalid media type for subtitles\n");
2720         return AVERROR(EINVAL);
2721     }
2722
2723     *got_sub_ptr = 0;
2724     get_subtitle_defaults(sub);
2725
2726     if ((avctx->codec->capabilities & CODEC_CAP_DELAY) || avpkt->size) {
2727         AVPacket pkt_recoded;
2728         AVPacket tmp = *avpkt;
2729         int did_split = av_packet_split_side_data(&tmp);
2730         //apply_param_change(avctx, &tmp);
2731
2732         if (did_split) {
2733             /* FFMIN() prevents overflow in case the packet wasn't allocated with
2734              * proper padding.
2735              * If the side data is smaller than the buffer padding size, the
2736              * remaining bytes should have already been filled with zeros by the
2737              * original packet allocation anyway. */
2738             memset(tmp.data + tmp.size, 0,
2739                    FFMIN(avpkt->size - tmp.size, FF_INPUT_BUFFER_PADDING_SIZE));
2740         }
2741
2742         pkt_recoded = tmp;
2743         ret = recode_subtitle(avctx, &pkt_recoded, &tmp);
2744         if (ret < 0) {
2745             *got_sub_ptr = 0;
2746         } else {
2747             avctx->internal->pkt = &pkt_recoded;
2748
2749             if (avctx->pkt_timebase.den && avpkt->pts != AV_NOPTS_VALUE)
2750                 sub->pts = av_rescale_q(avpkt->pts,
2751                                         avctx->pkt_timebase, AV_TIME_BASE_Q);
2752             ret = avctx->codec->decode(avctx, sub, got_sub_ptr, &pkt_recoded);
2753             av_assert1((ret >= 0) >= !!*got_sub_ptr &&
2754                        !!*got_sub_ptr >= !!sub->num_rects);
2755
2756             if (sub->num_rects && !sub->end_display_time && avpkt->duration &&
2757                 avctx->pkt_timebase.num) {
2758                 AVRational ms = { 1, 1000 };
2759                 sub->end_display_time = av_rescale_q(avpkt->duration,
2760                                                      avctx->pkt_timebase, ms);
2761             }
2762
2763             for (i = 0; i < sub->num_rects; i++) {
2764                 if (sub->rects[i]->ass && !utf8_check(sub->rects[i]->ass)) {
2765                     av_log(avctx, AV_LOG_ERROR,
2766                            "Invalid UTF-8 in decoded subtitles text; "
2767                            "maybe missing -sub_charenc option\n");
2768                     avsubtitle_free(sub);
2769                     return AVERROR_INVALIDDATA;
2770                 }
2771             }
2772
2773             if (tmp.data != pkt_recoded.data) { // did we recode?
2774                 /* prevent from destroying side data from original packet */
2775                 pkt_recoded.side_data = NULL;
2776                 pkt_recoded.side_data_elems = 0;
2777
2778                 av_free_packet(&pkt_recoded);
2779             }
2780             if (avctx->codec_descriptor->props & AV_CODEC_PROP_BITMAP_SUB)
2781                 sub->format = 0;
2782             else if (avctx->codec_descriptor->props & AV_CODEC_PROP_TEXT_SUB)
2783                 sub->format = 1;
2784             avctx->internal->pkt = NULL;
2785         }
2786
2787         if (did_split) {
2788             av_packet_free_side_data(&tmp);
2789             if(ret == tmp.size)
2790                 ret = avpkt->size;
2791         }
2792
2793         if (*got_sub_ptr)
2794             avctx->frame_number++;
2795     }
2796
2797     return ret;
2798 }
2799
2800 void avsubtitle_free(AVSubtitle *sub)
2801 {
2802     int i;
2803
2804     for (i = 0; i < sub->num_rects; i++) {
2805         av_freep(&sub->rects[i]->pict.data[0]);
2806         av_freep(&sub->rects[i]->pict.data[1]);
2807         av_freep(&sub->rects[i]->pict.data[2]);
2808         av_freep(&sub->rects[i]->pict.data[3]);
2809         av_freep(&sub->rects[i]->text);
2810         av_freep(&sub->rects[i]->ass);
2811         av_freep(&sub->rects[i]);
2812     }
2813
2814     av_freep(&sub->rects);
2815
2816     memset(sub, 0, sizeof(AVSubtitle));
2817 }
2818
2819 av_cold int avcodec_close(AVCodecContext *avctx)
2820 {
2821     if (!avctx)
2822         return 0;
2823
2824     if (avcodec_is_open(avctx)) {
2825         FramePool *pool = avctx->internal->pool;
2826         int i;
2827         if (CONFIG_FRAME_THREAD_ENCODER &&
2828             avctx->internal->frame_thread_encoder && avctx->thread_count > 1) {
2829             ff_frame_thread_encoder_free(avctx);
2830         }
2831         if (HAVE_THREADS && avctx->internal->thread_ctx)
2832             ff_thread_free(avctx);
2833         if (avctx->codec && avctx->codec->close)
2834             avctx->codec->close(avctx);
2835         avctx->coded_frame = NULL;
2836         avctx->internal->byte_buffer_size = 0;
2837         av_freep(&avctx->internal->byte_buffer);
2838         av_frame_free(&avctx->internal->to_free);
2839         for (i = 0; i < FF_ARRAY_ELEMS(pool->pools); i++)
2840             av_buffer_pool_uninit(&pool->pools[i]);
2841         av_freep(&avctx->internal->pool);
2842
2843         if (avctx->hwaccel && avctx->hwaccel->uninit)
2844             avctx->hwaccel->uninit(avctx);
2845         av_freep(&avctx->internal->hwaccel_priv_data);
2846
2847         av_freep(&avctx->internal);
2848     }
2849
2850     if (avctx->priv_data && avctx->codec && avctx->codec->priv_class)
2851         av_opt_free(avctx->priv_data);
2852     av_opt_free(avctx);
2853     av_freep(&avctx->priv_data);
2854     if (av_codec_is_encoder(avctx->codec))
2855         av_freep(&avctx->extradata);
2856     avctx->codec = NULL;
2857     avctx->active_thread_type = 0;
2858
2859     return 0;
2860 }
2861
2862 static enum AVCodecID remap_deprecated_codec_id(enum AVCodecID id)
2863 {
2864     switch(id){
2865         //This is for future deprecatec codec ids, its empty since
2866         //last major bump but will fill up again over time, please don't remove it
2867 //         case AV_CODEC_ID_UTVIDEO_DEPRECATED: return AV_CODEC_ID_UTVIDEO;
2868         case AV_CODEC_ID_BRENDER_PIX_DEPRECATED         : return AV_CODEC_ID_BRENDER_PIX;
2869         case AV_CODEC_ID_OPUS_DEPRECATED                : return AV_CODEC_ID_OPUS;
2870         case AV_CODEC_ID_TAK_DEPRECATED                 : return AV_CODEC_ID_TAK;
2871         case AV_CODEC_ID_PAF_AUDIO_DEPRECATED           : return AV_CODEC_ID_PAF_AUDIO;
2872         case AV_CODEC_ID_PCM_S24LE_PLANAR_DEPRECATED    : return AV_CODEC_ID_PCM_S24LE_PLANAR;
2873         case AV_CODEC_ID_PCM_S32LE_PLANAR_DEPRECATED    : return AV_CODEC_ID_PCM_S32LE_PLANAR;
2874         case AV_CODEC_ID_ADPCM_VIMA_DEPRECATED          : return AV_CODEC_ID_ADPCM_VIMA;
2875         case AV_CODEC_ID_ESCAPE130_DEPRECATED           : return AV_CODEC_ID_ESCAPE130;
2876         case AV_CODEC_ID_EXR_DEPRECATED                 : return AV_CODEC_ID_EXR;
2877         case AV_CODEC_ID_G2M_DEPRECATED                 : return AV_CODEC_ID_G2M;
2878         case AV_CODEC_ID_PAF_VIDEO_DEPRECATED           : return AV_CODEC_ID_PAF_VIDEO;
2879         case AV_CODEC_ID_WEBP_DEPRECATED                : return AV_CODEC_ID_WEBP;
2880         case AV_CODEC_ID_HEVC_DEPRECATED                : return AV_CODEC_ID_HEVC;
2881         case AV_CODEC_ID_MVC1_DEPRECATED                : return AV_CODEC_ID_MVC1;
2882         case AV_CODEC_ID_MVC2_DEPRECATED                : return AV_CODEC_ID_MVC2;
2883         case AV_CODEC_ID_SANM_DEPRECATED                : return AV_CODEC_ID_SANM;
2884         case AV_CODEC_ID_SGIRLE_DEPRECATED              : return AV_CODEC_ID_SGIRLE;
2885         case AV_CODEC_ID_VP7_DEPRECATED                 : return AV_CODEC_ID_VP7;
2886         default                                         : return id;
2887     }
2888 }
2889
2890 static AVCodec *find_encdec(enum AVCodecID id, int encoder)
2891 {
2892     AVCodec *p, *experimental = NULL;
2893     p = first_avcodec;
2894     id= remap_deprecated_codec_id(id);
2895     while (p) {
2896         if ((encoder ? av_codec_is_encoder(p) : av_codec_is_decoder(p)) &&
2897             p->id == id) {
2898             if (p->capabilities & CODEC_CAP_EXPERIMENTAL && !experimental) {
2899                 experimental = p;
2900             } else
2901                 return p;
2902         }
2903         p = p->next;
2904     }
2905     return experimental;
2906 }
2907
2908 AVCodec *avcodec_find_encoder(enum AVCodecID id)
2909 {
2910     return find_encdec(id, 1);
2911 }
2912
2913 AVCodec *avcodec_find_encoder_by_name(const char *name)
2914 {
2915     AVCodec *p;
2916     if (!name)
2917         return NULL;
2918     p = first_avcodec;
2919     while (p) {
2920         if (av_codec_is_encoder(p) && strcmp(name, p->name) == 0)
2921             return p;
2922         p = p->next;
2923     }
2924     return NULL;
2925 }
2926
2927 AVCodec *avcodec_find_decoder(enum AVCodecID id)
2928 {
2929     return find_encdec(id, 0);
2930 }
2931
2932 AVCodec *avcodec_find_decoder_by_name(const char *name)
2933 {
2934     AVCodec *p;
2935     if (!name)
2936         return NULL;
2937     p = first_avcodec;
2938     while (p) {
2939         if (av_codec_is_decoder(p) && strcmp(name, p->name) == 0)
2940             return p;
2941         p = p->next;
2942     }
2943     return NULL;
2944 }
2945
2946 const char *avcodec_get_name(enum AVCodecID id)
2947 {
2948     const AVCodecDescriptor *cd;
2949     AVCodec *codec;
2950
2951     if (id == AV_CODEC_ID_NONE)
2952         return "none";
2953     cd = avcodec_descriptor_get(id);
2954     if (cd)
2955         return cd->name;
2956     av_log(NULL, AV_LOG_WARNING, "Codec 0x%x is not in the full list.\n", id);
2957     codec = avcodec_find_decoder(id);
2958     if (codec)
2959         return codec->name;
2960     codec = avcodec_find_encoder(id);
2961     if (codec)
2962         return codec->name;
2963     return "unknown_codec";
2964 }
2965
2966 size_t av_get_codec_tag_string(char *buf, size_t buf_size, unsigned int codec_tag)
2967 {
2968     int i, len, ret = 0;
2969
2970 #define TAG_PRINT(x)                                              \
2971     (((x) >= '0' && (x) <= '9') ||                                \
2972      ((x) >= 'a' && (x) <= 'z') || ((x) >= 'A' && (x) <= 'Z') ||  \
2973      ((x) == '.' || (x) == ' ' || (x) == '-' || (x) == '_'))
2974
2975     for (i = 0; i < 4; i++) {
2976         len = snprintf(buf, buf_size,
2977                        TAG_PRINT(codec_tag & 0xFF) ? "%c" : "[%d]", codec_tag & 0xFF);
2978         buf        += len;
2979         buf_size    = buf_size > len ? buf_size - len : 0;
2980         ret        += len;
2981         codec_tag >>= 8;
2982     }
2983     return ret;
2984 }
2985
2986 void avcodec_string(char *buf, int buf_size, AVCodecContext *enc, int encode)
2987 {
2988     const char *codec_type;
2989     const char *codec_name;
2990     const char *profile = NULL;
2991     const AVCodec *p;
2992     int bitrate;
2993     int new_line = 0;
2994     AVRational display_aspect_ratio;
2995     const char *separator = enc->dump_separator ? (const char *)enc->dump_separator : ", ";
2996
2997     if (!buf || buf_size <= 0)
2998         return;
2999     codec_type = av_get_media_type_string(enc->codec_type);
3000     codec_name = avcodec_get_name(enc->codec_id);
3001     if (enc->profile != FF_PROFILE_UNKNOWN) {
3002         if (enc->codec)
3003             p = enc->codec;
3004         else
3005             p = encode ? avcodec_find_encoder(enc->codec_id) :
3006                         avcodec_find_decoder(enc->codec_id);
3007         if (p)
3008             profile = av_get_profile_name(p, enc->profile);
3009     }
3010
3011     snprintf(buf, buf_size, "%s: %s", codec_type ? codec_type : "unknown",
3012              codec_name);
3013     buf[0] ^= 'a' ^ 'A'; /* first letter in uppercase */
3014
3015     if (enc->codec && strcmp(enc->codec->name, codec_name))
3016         snprintf(buf + strlen(buf), buf_size - strlen(buf), " (%s)", enc->codec->name);
3017
3018     if (profile)
3019         snprintf(buf + strlen(buf), buf_size - strlen(buf), " (%s)", profile);
3020
3021     if (enc->codec_tag) {
3022         char tag_buf[32];
3023         av_get_codec_tag_string(tag_buf, sizeof(tag_buf), enc->codec_tag);
3024         snprintf(buf + strlen(buf), buf_size - strlen(buf),
3025                  " (%s / 0x%04X)", tag_buf, enc->codec_tag);
3026     }
3027
3028     switch (enc->codec_type) {
3029     case AVMEDIA_TYPE_VIDEO:
3030         {
3031             char detail[256] = "(";
3032
3033             av_strlcat(buf, separator, buf_size);
3034
3035             snprintf(buf + strlen(buf), buf_size - strlen(buf),
3036                  "%s", enc->pix_fmt == AV_PIX_FMT_NONE ? "none" :
3037                      av_get_pix_fmt_name(enc->pix_fmt));
3038             if (enc->bits_per_raw_sample && enc->pix_fmt != AV_PIX_FMT_NONE &&
3039                 enc->bits_per_raw_sample <= av_pix_fmt_desc_get(enc->pix_fmt)->comp[0].depth_minus1)
3040                 av_strlcatf(detail, sizeof(detail), "%d bpc, ", enc->bits_per_raw_sample);
3041             if (enc->color_range != AVCOL_RANGE_UNSPECIFIED)
3042                 av_strlcatf(detail, sizeof(detail), "%s, ",
3043                             av_color_range_name(enc->color_range));
3044
3045             if (enc->colorspace != AVCOL_SPC_UNSPECIFIED ||
3046                 enc->color_primaries != AVCOL_PRI_UNSPECIFIED ||
3047                 enc->color_trc != AVCOL_TRC_UNSPECIFIED) {
3048                 if (enc->colorspace != (int)enc->color_primaries ||
3049                     enc->colorspace != (int)enc->color_trc) {
3050                     new_line = 1;
3051                     av_strlcatf(detail, sizeof(detail), "%s/%s/%s, ",
3052                                 av_color_space_name(enc->colorspace),
3053                                 av_color_primaries_name(enc->color_primaries),
3054                                 av_color_transfer_name(enc->color_trc));
3055                 } else
3056                     av_strlcatf(detail, sizeof(detail), "%s, ",
3057                                 av_get_colorspace_name(enc->colorspace));
3058             }
3059
3060             if (av_log_get_level() >= AV_LOG_DEBUG &&
3061                 enc->chroma_sample_location != AVCHROMA_LOC_UNSPECIFIED)
3062                 av_strlcatf(detail, sizeof(detail), "%s, ",
3063                             av_chroma_location_name(enc->chroma_sample_location));
3064
3065             if (strlen(detail) > 1) {
3066                 detail[strlen(detail) - 2] = 0;
3067                 av_strlcatf(buf, buf_size, "%s)", detail);
3068             }
3069         }
3070
3071         if (enc->width) {
3072             av_strlcat(buf, new_line ? separator : ", ", buf_size);
3073
3074             snprintf(buf + strlen(buf), buf_size - strlen(buf),
3075                      "%dx%d",
3076                      enc->width, enc->height);
3077
3078             if (av_log_get_level() >= AV_LOG_VERBOSE &&
3079                 (enc->width != enc->coded_width ||
3080                  enc->height != enc->coded_height))
3081                 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3082                          " (%dx%d)", enc->coded_width, enc->coded_height);
3083
3084             if (enc->sample_aspect_ratio.num) {
3085                 av_reduce(&display_aspect_ratio.num, &display_aspect_ratio.den,
3086                           enc->width * enc->sample_aspect_ratio.num,
3087                           enc->height * enc->sample_aspect_ratio.den,
3088                           1024 * 1024);
3089                 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3090                          " [SAR %d:%d DAR %d:%d]",
3091                          enc->sample_aspect_ratio.num, enc->sample_aspect_ratio.den,
3092                          display_aspect_ratio.num, display_aspect_ratio.den);
3093             }
3094             if (av_log_get_level() >= AV_LOG_DEBUG) {
3095                 int g = av_gcd(enc->time_base.num, enc->time_base.den);
3096                 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3097                          ", %d/%d",
3098                          enc->time_base.num / g, enc->time_base.den / g);
3099             }
3100         }
3101         if (encode) {
3102             snprintf(buf + strlen(buf), buf_size - strlen(buf),
3103                      ", q=%d-%d", enc->qmin, enc->qmax);
3104         }
3105         break;
3106     case AVMEDIA_TYPE_AUDIO:
3107         av_strlcat(buf, separator, buf_size);
3108
3109         if (enc->sample_rate) {
3110             snprintf(buf + strlen(buf), buf_size - strlen(buf),
3111                      "%d Hz, ", enc->sample_rate);
3112         }
3113         av_get_channel_layout_string(buf + strlen(buf), buf_size - strlen(buf), enc->channels, enc->channel_layout);
3114         if (enc->sample_fmt != AV_SAMPLE_FMT_NONE) {
3115             snprintf(buf + strlen(buf), buf_size - strlen(buf),
3116                      ", %s", av_get_sample_fmt_name(enc->sample_fmt));
3117         }
3118         if (   enc->bits_per_raw_sample > 0
3119             && enc->bits_per_raw_sample != av_get_bytes_per_sample(enc->sample_fmt) * 8)
3120             snprintf(buf + strlen(buf), buf_size - strlen(buf),
3121                      " (%d bit)", enc->bits_per_raw_sample);
3122         break;
3123     case AVMEDIA_TYPE_DATA:
3124         if (av_log_get_level() >= AV_LOG_DEBUG) {
3125             int g = av_gcd(enc->time_base.num, enc->time_base.den);
3126             if (g)
3127                 snprintf(buf + strlen(buf), buf_size - strlen(buf),
3128                          ", %d/%d",
3129                          enc->time_base.num / g, enc->time_base.den / g);
3130         }
3131         break;
3132     case AVMEDIA_TYPE_SUBTITLE:
3133         if (enc->width)
3134             snprintf(buf + strlen(buf), buf_size - strlen(buf),
3135                      ", %dx%d", enc->width, enc->height);
3136         break;
3137     default:
3138         return;
3139     }
3140     if (encode) {
3141         if (enc->flags & CODEC_FLAG_PASS1)
3142             snprintf(buf + strlen(buf), buf_size - strlen(buf),
3143                      ", pass 1");
3144         if (enc->flags & CODEC_FLAG_PASS2)
3145             snprintf(buf + strlen(buf), buf_size - strlen(buf),
3146                      ", pass 2");
3147     }
3148     bitrate = get_bit_rate(enc);
3149     if (bitrate != 0) {
3150         snprintf(buf + strlen(buf), buf_size - strlen(buf),
3151                  ", %d kb/s", bitrate / 1000);
3152     } else if (enc->rc_max_rate > 0) {
3153         snprintf(buf + strlen(buf), buf_size - strlen(buf),
3154                  ", max. %d kb/s", enc->rc_max_rate / 1000);
3155     }
3156 }
3157
3158 const char *av_get_profile_name(const AVCodec *codec, int profile)
3159 {
3160     const AVProfile *p;
3161     if (profile == FF_PROFILE_UNKNOWN || !codec->profiles)
3162         return NULL;
3163
3164     for (p = codec->profiles; p->profile != FF_PROFILE_UNKNOWN; p++)
3165         if (p->profile == profile)
3166             return p->name;
3167
3168     return NULL;
3169 }
3170
3171 unsigned avcodec_version(void)
3172 {
3173 //    av_assert0(AV_CODEC_ID_V410==164);
3174     av_assert0(AV_CODEC_ID_PCM_S8_PLANAR==65563);
3175     av_assert0(AV_CODEC_ID_ADPCM_G722==69660);
3176 //     av_assert0(AV_CODEC_ID_BMV_AUDIO==86071);
3177     av_assert0(AV_CODEC_ID_SRT==94216);
3178     av_assert0(LIBAVCODEC_VERSION_MICRO >= 100);
3179
3180     av_assert0(CODEC_ID_CLLC == AV_CODEC_ID_CLLC);
3181     av_assert0(CODEC_ID_PCM_S8_PLANAR == AV_CODEC_ID_PCM_S8_PLANAR);
3182     av_assert0(CODEC_ID_ADPCM_IMA_APC == AV_CODEC_ID_ADPCM_IMA_APC);
3183     av_assert0(CODEC_ID_ILBC == AV_CODEC_ID_ILBC);
3184     av_assert0(CODEC_ID_SRT == AV_CODEC_ID_SRT);
3185     return LIBAVCODEC_VERSION_INT;
3186 }
3187
3188 const char *avcodec_configuration(void)
3189 {
3190     return FFMPEG_CONFIGURATION;
3191 }
3192
3193 const char *avcodec_license(void)
3194 {
3195 #define LICENSE_PREFIX "libavcodec license: "
3196     return LICENSE_PREFIX FFMPEG_LICENSE + sizeof(LICENSE_PREFIX) - 1;
3197 }
3198
3199 void avcodec_flush_buffers(AVCodecContext *avctx)
3200 {
3201     if (HAVE_THREADS && avctx->active_thread_type & FF_THREAD_FRAME)
3202         ff_thread_flush(avctx);
3203     else if (avctx->codec->flush)
3204         avctx->codec->flush(avctx);
3205
3206     avctx->pts_correction_last_pts =
3207     avctx->pts_correction_last_dts = INT64_MIN;
3208
3209     if (!avctx->refcounted_frames)
3210         av_frame_unref(avctx->internal->to_free);
3211 }
3212
3213 int av_get_exact_bits_per_sample(enum AVCodecID codec_id)
3214 {
3215     switch (codec_id) {
3216     case AV_CODEC_ID_8SVX_EXP:
3217     case AV_CODEC_ID_8SVX_FIB:
3218     case AV_CODEC_ID_ADPCM_CT:
3219     case AV_CODEC_ID_ADPCM_IMA_APC:
3220     case AV_CODEC_ID_ADPCM_IMA_EA_SEAD:
3221     case AV_CODEC_ID_ADPCM_IMA_OKI:
3222     case AV_CODEC_ID_ADPCM_IMA_WS:
3223     case AV_CODEC_ID_ADPCM_G722:
3224     case AV_CODEC_ID_ADPCM_YAMAHA:
3225         return 4;
3226     case AV_CODEC_ID_DSD_LSBF:
3227     case AV_CODEC_ID_DSD_MSBF:
3228     case AV_CODEC_ID_DSD_LSBF_PLANAR:
3229     case AV_CODEC_ID_DSD_MSBF_PLANAR:
3230     case AV_CODEC_ID_PCM_ALAW:
3231     case AV_CODEC_ID_PCM_MULAW:
3232     case AV_CODEC_ID_PCM_S8:
3233     case AV_CODEC_ID_PCM_S8_PLANAR:
3234     case AV_CODEC_ID_PCM_U8:
3235     case AV_CODEC_ID_PCM_ZORK:
3236         return 8;
3237     case AV_CODEC_ID_PCM_S16BE:
3238     case AV_CODEC_ID_PCM_S16BE_PLANAR:
3239     case AV_CODEC_ID_PCM_S16LE:
3240     case AV_CODEC_ID_PCM_S16LE_PLANAR:
3241     case AV_CODEC_ID_PCM_U16BE:
3242     case AV_CODEC_ID_PCM_U16LE:
3243         return 16;
3244     case AV_CODEC_ID_PCM_S24DAUD:
3245     case AV_CODEC_ID_PCM_S24BE:
3246     case AV_CODEC_ID_PCM_S24LE:
3247     case AV_CODEC_ID_PCM_S24LE_PLANAR:
3248     case AV_CODEC_ID_PCM_U24BE:
3249     case AV_CODEC_ID_PCM_U24LE:
3250         return 24;
3251     case AV_CODEC_ID_PCM_S32BE:
3252     case AV_CODEC_ID_PCM_S32LE:
3253     case AV_CODEC_ID_PCM_S32LE_PLANAR:
3254     case AV_CODEC_ID_PCM_U32BE:
3255     case AV_CODEC_ID_PCM_U32LE:
3256     case AV_CODEC_ID_PCM_F32BE:
3257     case AV_CODEC_ID_PCM_F32LE:
3258         return 32;
3259     case AV_CODEC_ID_PCM_F64BE:
3260     case AV_CODEC_ID_PCM_F64LE:
3261         return 64;
3262     default:
3263         return 0;
3264     }
3265 }
3266
3267 enum AVCodecID av_get_pcm_codec(enum AVSampleFormat fmt, int be)
3268 {
3269     static const enum AVCodecID map[AV_SAMPLE_FMT_NB][2] = {
3270         [AV_SAMPLE_FMT_U8  ] = { AV_CODEC_ID_PCM_U8,    AV_CODEC_ID_PCM_U8    },
3271         [AV_SAMPLE_FMT_S16 ] = { AV_CODEC_ID_PCM_S16LE, AV_CODEC_ID_PCM_S16BE },
3272         [AV_SAMPLE_FMT_S32 ] = { AV_CODEC_ID_PCM_S32LE, AV_CODEC_ID_PCM_S32BE },
3273         [AV_SAMPLE_FMT_FLT ] = { AV_CODEC_ID_PCM_F32LE, AV_CODEC_ID_PCM_F32BE },
3274         [AV_SAMPLE_FMT_DBL ] = { AV_CODEC_ID_PCM_F64LE, AV_CODEC_ID_PCM_F64BE },
3275         [AV_SAMPLE_FMT_U8P ] = { AV_CODEC_ID_PCM_U8,    AV_CODEC_ID_PCM_U8    },
3276         [AV_SAMPLE_FMT_S16P] = { AV_CODEC_ID_PCM_S16LE, AV_CODEC_ID_PCM_S16BE },
3277         [AV_SAMPLE_FMT_S32P] = { AV_CODEC_ID_PCM_S32LE, AV_CODEC_ID_PCM_S32BE },
3278         [AV_SAMPLE_FMT_FLTP] = { AV_CODEC_ID_PCM_F32LE, AV_CODEC_ID_PCM_F32BE },
3279         [AV_SAMPLE_FMT_DBLP] = { AV_CODEC_ID_PCM_F64LE, AV_CODEC_ID_PCM_F64BE },
3280     };
3281     if (fmt < 0 || fmt >= AV_SAMPLE_FMT_NB)
3282         return AV_CODEC_ID_NONE;
3283     if (be < 0 || be > 1)
3284         be = AV_NE(1, 0);
3285     return map[fmt][be];
3286 }
3287
3288 int av_get_bits_per_sample(enum AVCodecID codec_id)
3289 {
3290     switch (codec_id) {
3291     case AV_CODEC_ID_ADPCM_SBPRO_2:
3292         return 2;
3293     case AV_CODEC_ID_ADPCM_SBPRO_3:
3294         return 3;
3295     case AV_CODEC_ID_ADPCM_SBPRO_4:
3296     case AV_CODEC_ID_ADPCM_IMA_WAV:
3297     case AV_CODEC_ID_ADPCM_IMA_QT:
3298     case AV_CODEC_ID_ADPCM_SWF:
3299     case AV_CODEC_ID_ADPCM_MS:
3300         return 4;
3301     default:
3302         return av_get_exact_bits_per_sample(codec_id);
3303     }
3304 }
3305
3306 int av_get_audio_frame_duration(AVCodecContext *avctx, int frame_bytes)
3307 {
3308     int id, sr, ch, ba, tag, bps;
3309
3310     id  = avctx->codec_id;
3311     sr  = avctx->sample_rate;
3312     ch  = avctx->channels;
3313     ba  = avctx->block_align;
3314     tag = avctx->codec_tag;
3315     bps = av_get_exact_bits_per_sample(avctx->codec_id);
3316
3317     /* codecs with an exact constant bits per sample */
3318     if (bps > 0 && ch > 0 && frame_bytes > 0 && ch < 32768 && bps < 32768)
3319         return (frame_bytes * 8LL) / (bps * ch);
3320     bps = avctx->bits_per_coded_sample;
3321
3322     /* codecs with a fixed packet duration */
3323     switch (id) {
3324     case AV_CODEC_ID_ADPCM_ADX:    return   32;
3325     case AV_CODEC_ID_ADPCM_IMA_QT: return   64;
3326     case AV_CODEC_ID_ADPCM_EA_XAS: return  128;
3327     case AV_CODEC_ID_AMR_NB:
3328     case AV_CODEC_ID_EVRC:
3329     case AV_CODEC_ID_GSM:
3330     case AV_CODEC_ID_QCELP:
3331     case AV_CODEC_ID_RA_288:       return  160;
3332     case AV_CODEC_ID_AMR_WB:
3333     case AV_CODEC_ID_GSM_MS:       return  320;
3334     case AV_CODEC_ID_MP1:          return  384;
3335     case AV_CODEC_ID_ATRAC1:       return  512;
3336     case AV_CODEC_ID_ATRAC3:       return 1024;
3337     case AV_CODEC_ID_ATRAC3P:      return 2048;
3338     case AV_CODEC_ID_MP2:
3339     case AV_CODEC_ID_MUSEPACK7:    return 1152;
3340     case AV_CODEC_ID_AC3:          return 1536;
3341     }
3342
3343     if (sr > 0) {
3344         /* calc from sample rate */
3345         if (id == AV_CODEC_ID_TTA)
3346             return 256 * sr / 245;
3347
3348         if (ch > 0) {
3349             /* calc from sample rate and channels */
3350             if (id == AV_CODEC_ID_BINKAUDIO_DCT)
3351                 return (480 << (sr / 22050)) / ch;
3352         }
3353     }
3354
3355     if (ba > 0) {
3356         /* calc from block_align */
3357         if (id == AV_CODEC_ID_SIPR) {
3358             switch (ba) {
3359             case 20: return 160;
3360             case 19: return 144;
3361             case 29: return 288;
3362             case 37: return 480;
3363             }
3364         } else if (id == AV_CODEC_ID_ILBC) {
3365             switch (ba) {
3366             case 38: return 160;
3367             case 50: return 240;
3368             }
3369         }
3370     }
3371
3372     if (frame_bytes > 0) {
3373         /* calc from frame_bytes only */
3374         if (id == AV_CODEC_ID_TRUESPEECH)
3375             return 240 * (frame_bytes / 32);
3376         if (id == AV_CODEC_ID_NELLYMOSER)
3377             return 256 * (frame_bytes / 64);
3378         if (id == AV_CODEC_ID_RA_144)
3379             return 160 * (frame_bytes / 20);
3380         if (id == AV_CODEC_ID_G723_1)
3381             return 240 * (frame_bytes / 24);
3382
3383         if (bps > 0) {
3384             /* calc from frame_bytes and bits_per_coded_sample */
3385             if (id == AV_CODEC_ID_ADPCM_G726)
3386                 return frame_bytes * 8 / bps;
3387         }
3388
3389         if (ch > 0) {
3390             /* calc from frame_bytes and channels */
3391             switch (id) {
3392             case AV_CODEC_ID_ADPCM_AFC:
3393                 return frame_bytes / (9 * ch) * 16;
3394             case AV_CODEC_ID_ADPCM_DTK:
3395                 return frame_bytes / (16 * ch) * 28;
3396             case AV_CODEC_ID_ADPCM_4XM:
3397             case AV_CODEC_ID_ADPCM_IMA_ISS:
3398                 return (frame_bytes - 4 * ch) * 2 / ch;
3399             case AV_CODEC_ID_ADPCM_IMA_SMJPEG:
3400                 return (frame_bytes - 4) * 2 / ch;
3401             case AV_CODEC_ID_ADPCM_IMA_AMV:
3402                 return (frame_bytes - 8) * 2 / ch;
3403             case AV_CODEC_ID_ADPCM_XA:
3404                 return (frame_bytes / 128) * 224 / ch;
3405             case AV_CODEC_ID_INTERPLAY_DPCM:
3406                 return (frame_bytes - 6 - ch) / ch;
3407             case AV_CODEC_ID_ROQ_DPCM:
3408                 return (frame_bytes - 8) / ch;
3409             case AV_CODEC_ID_XAN_DPCM:
3410                 return (frame_bytes - 2 * ch) / ch;
3411             case AV_CODEC_ID_MACE3:
3412                 return 3 * frame_bytes / ch;
3413             case AV_CODEC_ID_MACE6:
3414                 return 6 * frame_bytes / ch;
3415             case AV_CODEC_ID_PCM_LXF:
3416                 return 2 * (frame_bytes / (5 * ch));
3417             case AV_CODEC_ID_IAC:
3418             case AV_CODEC_ID_IMC:
3419                 return 4 * frame_bytes / ch;
3420             }
3421
3422             if (tag) {
3423                 /* calc from frame_bytes, channels, and codec_tag */
3424                 if (id == AV_CODEC_ID_SOL_DPCM) {
3425                     if (tag == 3)
3426                         return frame_bytes / ch;
3427                     else
3428                         return frame_bytes * 2 / ch;
3429                 }
3430             }
3431
3432             if (ba > 0) {
3433                 /* calc from frame_bytes, channels, and block_align */
3434                 int blocks = frame_bytes / ba;
3435                 switch (avctx->codec_id) {
3436                 case AV_CODEC_ID_ADPCM_IMA_WAV:
3437                     if (bps < 2 || bps > 5)
3438                         return 0;
3439                     return blocks * (1 + (ba - 4 * ch) / (bps * ch) * 8);
3440                 case AV_CODEC_ID_ADPCM_IMA_DK3:
3441                     return blocks * (((ba - 16) * 2 / 3 * 4) / ch);
3442                 case AV_CODEC_ID_ADPCM_IMA_DK4:
3443                     return blocks * (1 + (ba - 4 * ch) * 2 / ch);
3444                 case AV_CODEC_ID_ADPCM_IMA_RAD:
3445                     return blocks * ((ba - 4 * ch) * 2 / ch);
3446                 case AV_CODEC_ID_ADPCM_MS:
3447                     return blocks * (2 + (ba - 7 * ch) * 2 / ch);
3448                 }
3449             }
3450
3451             if (bps > 0) {
3452                 /* calc from frame_bytes, channels, and bits_per_coded_sample */
3453                 switch (avctx->codec_id) {
3454                 case AV_CODEC_ID_PCM_DVD:
3455                     if(bps<4)
3456                         return 0;
3457                     return 2 * (frame_bytes / ((bps * 2 / 8) * ch));
3458                 case AV_CODEC_ID_PCM_BLURAY:
3459                     if(bps<4)
3460                         return 0;
3461                     return frame_bytes / ((FFALIGN(ch, 2) * bps) / 8);
3462                 case AV_CODEC_ID_S302M:
3463                     return 2 * (frame_bytes / ((bps + 4) / 4)) / ch;
3464                 }
3465             }
3466         }
3467     }
3468
3469     /* Fall back on using frame_size */
3470     if (avctx->frame_size > 1 && frame_bytes)
3471         return avctx->frame_size;
3472
3473     //For WMA we currently have no other means to calculate duration thus we
3474     //do it here by assuming CBR, which is true for all known cases.
3475     if (avctx->bit_rate>0 && frame_bytes>0 && avctx->sample_rate>0 && avctx->block_align>1) {
3476         if (avctx->codec_id == AV_CODEC_ID_WMAV1 || avctx->codec_id == AV_CODEC_ID_WMAV2)
3477             return  (frame_bytes * 8LL * avctx->sample_rate) / avctx->bit_rate;
3478     }
3479
3480     return 0;
3481 }
3482
3483 #if !HAVE_THREADS
3484 int ff_thread_init(AVCodecContext *s)
3485 {
3486     return -1;
3487 }
3488
3489 #endif
3490
3491 unsigned int av_xiphlacing(unsigned char *s, unsigned int v)
3492 {
3493     unsigned int n = 0;
3494
3495     while (v >= 0xff) {
3496         *s++ = 0xff;
3497         v -= 0xff;
3498         n++;
3499     }
3500     *s = v;
3501     n++;
3502     return n;
3503 }
3504
3505 int ff_match_2uint16(const uint16_t(*tab)[2], int size, int a, int b)
3506 {
3507     int i;
3508     for (i = 0; i < size && !(tab[i][0] == a && tab[i][1] == b); i++) ;
3509     return i;
3510 }
3511
3512 #if FF_API_MISSING_SAMPLE
3513 FF_DISABLE_DEPRECATION_WARNINGS
3514 void av_log_missing_feature(void *avc, const char *feature, int want_sample)
3515 {
3516     av_log(avc, AV_LOG_WARNING, "%s is not implemented. Update your FFmpeg "
3517             "version to the newest one from Git. If the problem still "
3518             "occurs, it means that your file has a feature which has not "
3519             "been implemented.\n", feature);
3520     if(want_sample)
3521         av_log_ask_for_sample(avc, NULL);
3522 }
3523
3524 void av_log_ask_for_sample(void *avc, const char *msg, ...)
3525 {
3526     va_list argument_list;
3527
3528     va_start(argument_list, msg);
3529
3530     if (msg)
3531         av_vlog(avc, AV_LOG_WARNING, msg, argument_list);
3532     av_log(avc, AV_LOG_WARNING, "If you want to help, upload a sample "
3533             "of this file to ftp://upload.ffmpeg.org/incoming/ "
3534             "and contact the ffmpeg-devel mailing list. (ffmpeg-devel@ffmpeg.org)\n");
3535
3536     va_end(argument_list);
3537 }
3538 FF_ENABLE_DEPRECATION_WARNINGS
3539 #endif /* FF_API_MISSING_SAMPLE */
3540
3541 static AVHWAccel *first_hwaccel = NULL;
3542 static AVHWAccel **last_hwaccel = &first_hwaccel;
3543
3544 void av_register_hwaccel(AVHWAccel *hwaccel)
3545 {
3546     AVHWAccel **p = last_hwaccel;
3547     hwaccel->next = NULL;
3548     while(*p || avpriv_atomic_ptr_cas((void * volatile *)p, NULL, hwaccel))
3549         p = &(*p)->next;
3550     last_hwaccel = &hwaccel->next;
3551 }
3552
3553 AVHWAccel *av_hwaccel_next(const AVHWAccel *hwaccel)
3554 {
3555     return hwaccel ? hwaccel->next : first_hwaccel;
3556 }
3557
3558 int av_lockmgr_register(int (*cb)(void **mutex, enum AVLockOp op))
3559 {
3560     if (lockmgr_cb) {
3561         // There is no good way to rollback a failure to destroy the
3562         // mutex, so we ignore failures.
3563         lockmgr_cb(&codec_mutex,    AV_LOCK_DESTROY);
3564         lockmgr_cb(&avformat_mutex, AV_LOCK_DESTROY);
3565         lockmgr_cb     = NULL;
3566         codec_mutex    = NULL;
3567         avformat_mutex = NULL;
3568     }
3569
3570     if (cb) {
3571         void *new_codec_mutex    = NULL;
3572         void *new_avformat_mutex = NULL;
3573         int err;
3574         if (err = cb(&new_codec_mutex, AV_LOCK_CREATE)) {
3575             return err > 0 ? AVERROR_UNKNOWN : err;
3576         }
3577         if (err = cb(&new_avformat_mutex, AV_LOCK_CREATE)) {
3578             // Ignore failures to destroy the newly created mutex.
3579             cb(&new_codec_mutex, AV_LOCK_DESTROY);
3580             return err > 0 ? AVERROR_UNKNOWN : err;
3581         }
3582         lockmgr_cb     = cb;
3583         codec_mutex    = new_codec_mutex;
3584         avformat_mutex = new_avformat_mutex;
3585     }
3586
3587     return 0;
3588 }
3589
3590 int ff_lock_avcodec(AVCodecContext *log_ctx)
3591 {
3592     if (lockmgr_cb) {
3593         if ((*lockmgr_cb)(&codec_mutex, AV_LOCK_OBTAIN))
3594             return -1;
3595     }
3596     entangled_thread_counter++;
3597     if (entangled_thread_counter != 1) {
3598         av_log(log_ctx, AV_LOG_ERROR, "Insufficient thread locking around avcodec_open/close()\n");
3599         if (!lockmgr_cb)
3600             av_log(log_ctx, AV_LOG_ERROR, "No lock manager is set, please see av_lockmgr_register()\n");
3601         ff_avcodec_locked = 1;
3602         ff_unlock_avcodec();
3603         return AVERROR(EINVAL);
3604     }
3605     av_assert0(!ff_avcodec_locked);
3606     ff_avcodec_locked = 1;
3607     return 0;
3608 }
3609
3610 int ff_unlock_avcodec(void)
3611 {
3612     av_assert0(ff_avcodec_locked);
3613     ff_avcodec_locked = 0;
3614     entangled_thread_counter--;
3615     if (lockmgr_cb) {
3616         if ((*lockmgr_cb)(&codec_mutex, AV_LOCK_RELEASE))
3617             return -1;
3618     }
3619
3620     return 0;
3621 }
3622
3623 int avpriv_lock_avformat(void)
3624 {
3625     if (lockmgr_cb) {
3626         if ((*lockmgr_cb)(&avformat_mutex, AV_LOCK_OBTAIN))
3627             return -1;
3628     }
3629     return 0;
3630 }
3631
3632 int avpriv_unlock_avformat(void)
3633 {
3634     if (lockmgr_cb) {
3635         if ((*lockmgr_cb)(&avformat_mutex, AV_LOCK_RELEASE))
3636             return -1;
3637     }
3638     return 0;
3639 }
3640
3641 unsigned int avpriv_toupper4(unsigned int x)
3642 {
3643     return av_toupper(x & 0xFF) +
3644           (av_toupper((x >>  8) & 0xFF) << 8)  +
3645           (av_toupper((x >> 16) & 0xFF) << 16) +
3646 ((unsigned)av_toupper((x >> 24) & 0xFF) << 24);
3647 }
3648
3649 int ff_thread_ref_frame(ThreadFrame *dst, ThreadFrame *src)
3650 {
3651     int ret;
3652
3653     dst->owner = src->owner;
3654
3655     ret = av_frame_ref(dst->f, src->f);
3656     if (ret < 0)
3657         return ret;
3658
3659     if (src->progress &&
3660         !(dst->progress = av_buffer_ref(src->progress))) {
3661         ff_thread_release_buffer(dst->owner, dst);
3662         return AVERROR(ENOMEM);
3663     }
3664
3665     return 0;
3666 }
3667
3668 #if !HAVE_THREADS
3669
3670 enum AVPixelFormat ff_thread_get_format(AVCodecContext *avctx, const enum AVPixelFormat *fmt)
3671 {
3672     return ff_get_format(avctx, fmt);
3673 }
3674
3675 int ff_thread_get_buffer(AVCodecContext *avctx, ThreadFrame *f, int flags)
3676 {
3677     f->owner = avctx;
3678     return ff_get_buffer(avctx, f->f, flags);
3679 }
3680
3681 void ff_thread_release_buffer(AVCodecContext *avctx, ThreadFrame *f)
3682 {
3683     if (f->f)
3684         av_frame_unref(f->f);
3685 }
3686
3687 void ff_thread_finish_setup(AVCodecContext *avctx)
3688 {
3689 }
3690
3691 void ff_thread_report_progress(ThreadFrame *f, int progress, int field)
3692 {
3693 }
3694
3695 void ff_thread_await_progress(ThreadFrame *f, int progress, int field)
3696 {
3697 }
3698
3699 int ff_thread_can_start_frame(AVCodecContext *avctx)
3700 {
3701     return 1;
3702 }
3703
3704 int ff_alloc_entries(AVCodecContext *avctx, int count)
3705 {
3706     return 0;
3707 }
3708
3709 void ff_reset_entries(AVCodecContext *avctx)
3710 {
3711 }
3712
3713 void ff_thread_await_progress2(AVCodecContext *avctx, int field, int thread, int shift)
3714 {
3715 }
3716
3717 void ff_thread_report_progress2(AVCodecContext *avctx, int field, int thread, int n)
3718 {
3719 }
3720
3721 #endif
3722
3723 enum AVMediaType avcodec_get_type(enum AVCodecID codec_id)
3724 {
3725     AVCodec *c= avcodec_find_decoder(codec_id);
3726     if(!c)
3727         c= avcodec_find_encoder(codec_id);
3728     if(c)
3729         return c->type;
3730
3731     if (codec_id <= AV_CODEC_ID_NONE)
3732         return AVMEDIA_TYPE_UNKNOWN;
3733     else if (codec_id < AV_CODEC_ID_FIRST_AUDIO)
3734         return AVMEDIA_TYPE_VIDEO;
3735     else if (codec_id < AV_CODEC_ID_FIRST_SUBTITLE)
3736         return AVMEDIA_TYPE_AUDIO;
3737     else if (codec_id < AV_CODEC_ID_FIRST_UNKNOWN)
3738         return AVMEDIA_TYPE_SUBTITLE;
3739
3740     return AVMEDIA_TYPE_UNKNOWN;
3741 }
3742
3743 int avcodec_is_open(AVCodecContext *s)
3744 {
3745     return !!s->internal;
3746 }
3747
3748 int avpriv_bprint_to_extradata(AVCodecContext *avctx, struct AVBPrint *buf)
3749 {
3750     int ret;
3751     char *str;
3752
3753     ret = av_bprint_finalize(buf, &str);
3754     if (ret < 0)
3755         return ret;
3756     if (!av_bprint_is_complete(buf)) {
3757         av_free(str);
3758         return AVERROR(ENOMEM);
3759     }
3760
3761     avctx->extradata = str;
3762     /* Note: the string is NUL terminated (so extradata can be read as a
3763      * string), but the ending character is not accounted in the size (in
3764      * binary formats you are likely not supposed to mux that character). When
3765      * extradata is copied, it is also padded with FF_INPUT_BUFFER_PADDING_SIZE
3766      * zeros. */
3767     avctx->extradata_size = buf->len;
3768     return 0;
3769 }
3770
3771 const uint8_t *avpriv_find_start_code(const uint8_t *av_restrict p,
3772                                       const uint8_t *end,
3773                                       uint32_t *av_restrict state)
3774 {
3775     int i;
3776
3777     av_assert0(p <= end);
3778     if (p >= end)
3779         return end;
3780
3781     for (i = 0; i < 3; i++) {
3782         uint32_t tmp = *state << 8;
3783         *state = tmp + *(p++);
3784         if (tmp == 0x100 || p == end)
3785             return p;
3786     }
3787
3788     while (p < end) {
3789         if      (p[-1] > 1      ) p += 3;
3790         else if (p[-2]          ) p += 2;
3791         else if (p[-3]|(p[-1]-1)) p++;
3792         else {
3793             p++;
3794             break;
3795         }
3796     }
3797
3798     p = FFMIN(p, end) - 4;
3799     *state = AV_RB32(p);
3800
3801     return p + 4;
3802 }