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