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