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