5af3515f09bbc90032545f206ff71fdb20e01f40
[ffmpeg.git] / 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 "libavutil/avassert.h"
29 #include "libavutil/avstring.h"
30 #include "libavutil/crc.h"
31 #include "libavutil/mathematics.h"
32 #include "libavutil/pixdesc.h"
33 #include "libavutil/audioconvert.h"
34 #include "libavutil/imgutils.h"
35 #include "libavutil/samplefmt.h"
36 #include "libavutil/dict.h"
37 #include "libavutil/avassert.h"
38 #include "avcodec.h"
39 #include "dsputil.h"
40 #include "libavutil/opt.h"
41 #include "imgconvert.h"
42 #include "thread.h"
43 #include "audioconvert.h"
44 #include "internal.h"
45 #include "bytestream.h"
46 #include <stdlib.h>
47 #include <stdarg.h>
48 #include <limits.h>
49 #include <float.h>
50
51 static int volatile entangled_thread_counter=0;
52 static int (*ff_lockmgr_cb)(void **mutex, enum AVLockOp op);
53 static void *codec_mutex;
54 static void *avformat_mutex;
55
56 void *av_fast_realloc(void *ptr, unsigned int *size, size_t min_size)
57 {
58     if(min_size < *size)
59         return ptr;
60
61     min_size= FFMAX(17*min_size/16 + 32, min_size);
62
63     ptr= av_realloc(ptr, min_size);
64     if(!ptr) //we could set this to the unmodified min_size but this is safer if the user lost the ptr and uses NULL now
65         min_size= 0;
66
67     *size= min_size;
68
69     return ptr;
70 }
71
72 static inline int ff_fast_malloc(void *ptr, unsigned int *size, size_t min_size, int zero_realloc)
73 {
74     void **p = ptr;
75     if (min_size < *size)
76         return 0;
77     min_size= FFMAX(17*min_size/16 + 32, min_size);
78     av_free(*p);
79     *p = zero_realloc ? av_mallocz(min_size) : av_malloc(min_size);
80     if (!*p) min_size = 0;
81     *size= min_size;
82     return 1;
83 }
84
85 void av_fast_malloc(void *ptr, unsigned int *size, size_t min_size)
86 {
87     ff_fast_malloc(ptr, size, min_size, 0);
88 }
89
90 void av_fast_padded_malloc(void *ptr, unsigned int *size, size_t min_size)
91 {
92     uint8_t **p = ptr;
93     if (min_size > SIZE_MAX - FF_INPUT_BUFFER_PADDING_SIZE) {
94         av_freep(p);
95         *size = 0;
96         return;
97     }
98     if (!ff_fast_malloc(p, size, min_size + FF_INPUT_BUFFER_PADDING_SIZE, 1))
99         memset(*p + min_size, 0, FF_INPUT_BUFFER_PADDING_SIZE);
100 }
101
102 /* encoder management */
103 static AVCodec *first_avcodec = NULL;
104
105 AVCodec *av_codec_next(AVCodec *c){
106     if(c) return c->next;
107     else  return first_avcodec;
108 }
109
110 static void avcodec_init(void)
111 {
112     static int initialized = 0;
113
114     if (initialized != 0)
115         return;
116     initialized = 1;
117
118     ff_dsputil_static_init();
119 }
120
121 int av_codec_is_encoder(AVCodec *codec)
122 {
123     return codec && (codec->encode || codec->encode2);
124 }
125
126 int av_codec_is_decoder(AVCodec *codec)
127 {
128     return codec && codec->decode;
129 }
130
131 void avcodec_register(AVCodec *codec)
132 {
133     AVCodec **p;
134     avcodec_init();
135     p = &first_avcodec;
136     while (*p != NULL) p = &(*p)->next;
137     *p = codec;
138     codec->next = NULL;
139
140     if (codec->init_static_data)
141         codec->init_static_data(codec);
142 }
143
144 unsigned avcodec_get_edge_width(void)
145 {
146     return EDGE_WIDTH;
147 }
148
149 void avcodec_set_dimensions(AVCodecContext *s, int width, int height){
150     s->coded_width = width;
151     s->coded_height= height;
152     s->width  = width;
153     s->height = height;
154 }
155
156 #define INTERNAL_BUFFER_SIZE (32+1)
157
158 void avcodec_align_dimensions2(AVCodecContext *s, int *width, int *height,
159                                int linesize_align[AV_NUM_DATA_POINTERS])
160 {
161     int i;
162     int w_align= 1;
163     int h_align= 1;
164
165     switch(s->pix_fmt){
166     case PIX_FMT_YUV420P:
167     case PIX_FMT_YUYV422:
168     case PIX_FMT_UYVY422:
169     case PIX_FMT_YUV422P:
170     case PIX_FMT_YUV440P:
171     case PIX_FMT_YUV444P:
172     case PIX_FMT_GBRP:
173     case PIX_FMT_GRAY8:
174     case PIX_FMT_GRAY16BE:
175     case PIX_FMT_GRAY16LE:
176     case PIX_FMT_YUVJ420P:
177     case PIX_FMT_YUVJ422P:
178     case PIX_FMT_YUVJ440P:
179     case PIX_FMT_YUVJ444P:
180     case PIX_FMT_YUVA420P:
181     case PIX_FMT_YUVA444P:
182     case PIX_FMT_YUV420P9LE:
183     case PIX_FMT_YUV420P9BE:
184     case PIX_FMT_YUV420P10LE:
185     case PIX_FMT_YUV420P10BE:
186     case PIX_FMT_YUV422P9LE:
187     case PIX_FMT_YUV422P9BE:
188     case PIX_FMT_YUV422P10LE:
189     case PIX_FMT_YUV422P10BE:
190     case PIX_FMT_YUV444P9LE:
191     case PIX_FMT_YUV444P9BE:
192     case PIX_FMT_YUV444P10LE:
193     case PIX_FMT_YUV444P10BE:
194     case PIX_FMT_GBRP9LE:
195     case PIX_FMT_GBRP9BE:
196     case PIX_FMT_GBRP10LE:
197     case PIX_FMT_GBRP10BE:
198         w_align = 16; //FIXME assume 16 pixel per macroblock
199         h_align = 16 * 2; // interlaced needs 2 macroblocks height
200         break;
201     case PIX_FMT_YUV411P:
202     case PIX_FMT_UYYVYY411:
203         w_align=32;
204         h_align=8;
205         break;
206     case PIX_FMT_YUV410P:
207         if(s->codec_id == CODEC_ID_SVQ1){
208             w_align=64;
209             h_align=64;
210         }
211     case PIX_FMT_RGB555:
212         if(s->codec_id == CODEC_ID_RPZA){
213             w_align=4;
214             h_align=4;
215         }
216     case PIX_FMT_PAL8:
217     case PIX_FMT_BGR8:
218     case PIX_FMT_RGB8:
219         if(s->codec_id == CODEC_ID_SMC){
220             w_align=4;
221             h_align=4;
222         }
223         break;
224     case PIX_FMT_BGR24:
225         if((s->codec_id == CODEC_ID_MSZH) || (s->codec_id == CODEC_ID_ZLIB)){
226             w_align=4;
227             h_align=4;
228         }
229         break;
230     default:
231         w_align= 1;
232         h_align= 1;
233         break;
234     }
235
236     if(s->codec_id == CODEC_ID_IFF_ILBM || s->codec_id == CODEC_ID_IFF_BYTERUN1){
237         w_align= FFMAX(w_align, 8);
238     }
239
240     *width = FFALIGN(*width , w_align);
241     *height= FFALIGN(*height, h_align);
242     if (s->codec_id == CODEC_ID_H264)
243         *height+=2; // some of the optimized chroma MC reads one line too much
244
245     for (i = 0; i < 4; i++)
246         linesize_align[i] = STRIDE_ALIGN;
247 }
248
249 void avcodec_align_dimensions(AVCodecContext *s, int *width, int *height){
250     int chroma_shift = av_pix_fmt_descriptors[s->pix_fmt].log2_chroma_w;
251     int linesize_align[AV_NUM_DATA_POINTERS];
252     int align;
253     avcodec_align_dimensions2(s, width, height, linesize_align);
254     align = FFMAX(linesize_align[0], linesize_align[3]);
255     linesize_align[1] <<= chroma_shift;
256     linesize_align[2] <<= chroma_shift;
257     align = FFMAX3(align, linesize_align[1], linesize_align[2]);
258     *width=FFALIGN(*width, align);
259 }
260
261 void ff_init_buffer_info(AVCodecContext *s, AVFrame *pic)
262 {
263     if (s->pkt) {
264         pic->pkt_pts = s->pkt->pts;
265         pic->pkt_pos = s->pkt->pos;
266     } else {
267         pic->pkt_pts = AV_NOPTS_VALUE;
268         pic->pkt_pos = -1;
269     }
270     pic->reordered_opaque= s->reordered_opaque;
271     pic->sample_aspect_ratio = s->sample_aspect_ratio;
272     pic->width               = s->width;
273     pic->height              = s->height;
274     pic->format              = s->pix_fmt;
275 }
276
277 int avcodec_fill_audio_frame(AVFrame *frame, int nb_channels,
278                              enum AVSampleFormat sample_fmt, const uint8_t *buf,
279                              int buf_size, int align)
280 {
281     int ch, planar, needed_size, ret = 0;
282
283     needed_size = av_samples_get_buffer_size(NULL, nb_channels,
284                                              frame->nb_samples, sample_fmt,
285                                              align);
286     if (buf_size < needed_size)
287         return AVERROR(EINVAL);
288
289     planar = av_sample_fmt_is_planar(sample_fmt);
290     if (planar && nb_channels > AV_NUM_DATA_POINTERS) {
291         if (!(frame->extended_data = av_mallocz(nb_channels *
292                                                 sizeof(*frame->extended_data))))
293             return AVERROR(ENOMEM);
294     } else {
295         frame->extended_data = frame->data;
296     }
297
298     if ((ret = av_samples_fill_arrays(frame->extended_data, &frame->linesize[0],
299                                       (uint8_t *)(intptr_t)buf, nb_channels, frame->nb_samples,
300                                       sample_fmt, align)) < 0) {
301         if (frame->extended_data != frame->data)
302             av_freep(&frame->extended_data);
303         return ret;
304     }
305     if (frame->extended_data != frame->data) {
306         for (ch = 0; ch < AV_NUM_DATA_POINTERS; ch++)
307             frame->data[ch] = frame->extended_data[ch];
308     }
309
310     return ret;
311 }
312
313 static int audio_get_buffer(AVCodecContext *avctx, AVFrame *frame)
314 {
315     AVCodecInternal *avci = avctx->internal;
316     InternalBuffer *buf;
317     int buf_size, ret;
318
319     buf_size = av_samples_get_buffer_size(NULL, avctx->channels,
320                                           frame->nb_samples, avctx->sample_fmt,
321                                           0);
322     if (buf_size < 0)
323         return AVERROR(EINVAL);
324
325     /* allocate InternalBuffer if needed */
326     if (!avci->buffer) {
327         avci->buffer = av_mallocz(sizeof(InternalBuffer));
328         if (!avci->buffer)
329             return AVERROR(ENOMEM);
330     }
331     buf = avci->buffer;
332
333     /* if there is a previously-used internal buffer, check its size and
334        channel count to see if we can reuse it */
335     if (buf->extended_data) {
336         /* if current buffer is too small, free it */
337         if (buf->extended_data[0] && buf_size > buf->audio_data_size) {
338             av_free(buf->extended_data[0]);
339             if (buf->extended_data != buf->data)
340                 av_freep(&buf->extended_data);
341             buf->extended_data = NULL;
342             buf->data[0] = NULL;
343         }
344         /* if number of channels has changed, reset and/or free extended data
345            pointers but leave data buffer in buf->data[0] for reuse */
346         if (buf->nb_channels != avctx->channels) {
347             if (buf->extended_data != buf->data)
348                 av_free(buf->extended_data);
349             buf->extended_data = NULL;
350         }
351     }
352
353     /* if there is no previous buffer or the previous buffer cannot be used
354        as-is, allocate a new buffer and/or rearrange the channel pointers */
355     if (!buf->extended_data) {
356         if (!buf->data[0]) {
357             if (!(buf->data[0] = av_mallocz(buf_size)))
358                 return AVERROR(ENOMEM);
359             buf->audio_data_size = buf_size;
360         }
361         if ((ret = avcodec_fill_audio_frame(frame, avctx->channels,
362                                             avctx->sample_fmt, buf->data[0],
363                                             buf->audio_data_size, 0)))
364             return ret;
365
366         if (frame->extended_data == frame->data)
367             buf->extended_data = buf->data;
368         else
369             buf->extended_data = frame->extended_data;
370         memcpy(buf->data, frame->data, sizeof(frame->data));
371         buf->linesize[0] = frame->linesize[0];
372         buf->nb_channels = avctx->channels;
373     } else {
374         /* copy InternalBuffer info to the AVFrame */
375         frame->extended_data = buf->extended_data;
376         frame->linesize[0]   = buf->linesize[0];
377         memcpy(frame->data, buf->data, sizeof(frame->data));
378     }
379
380     frame->type          = FF_BUFFER_TYPE_INTERNAL;
381
382     if (avctx->pkt) {
383         frame->pkt_pts = avctx->pkt->pts;
384         frame->pkt_pos = avctx->pkt->pos;
385     } else {
386         frame->pkt_pts = AV_NOPTS_VALUE;
387         frame->pkt_pos = -1;
388     }
389
390     frame->reordered_opaque = avctx->reordered_opaque;
391
392     if (avctx->debug & FF_DEBUG_BUFFERS)
393         av_log(avctx, AV_LOG_DEBUG, "default_get_buffer called on frame %p, "
394                "internal audio buffer used\n", frame);
395
396     return 0;
397 }
398
399 static int video_get_buffer(AVCodecContext *s, AVFrame *pic)
400 {
401     int i;
402     int w= s->width;
403     int h= s->height;
404     InternalBuffer *buf;
405     AVCodecInternal *avci = s->internal;
406
407     if(pic->data[0]!=NULL) {
408         av_log(s, AV_LOG_ERROR, "pic->data[0]!=NULL in avcodec_default_get_buffer\n");
409         return -1;
410     }
411     if(avci->buffer_count >= INTERNAL_BUFFER_SIZE) {
412         av_log(s, AV_LOG_ERROR, "buffer_count overflow (missing release_buffer?)\n");
413         return -1;
414     }
415
416     if(av_image_check_size(w, h, 0, s) || s->pix_fmt<0)
417         return -1;
418
419     if (!avci->buffer) {
420         avci->buffer = av_mallocz((INTERNAL_BUFFER_SIZE+1) *
421                                   sizeof(InternalBuffer));
422     }
423
424     buf = &avci->buffer[avci->buffer_count];
425
426     if(buf->base[0] && (buf->width != w || buf->height != h || buf->pix_fmt != s->pix_fmt)){
427         for (i = 0; i < AV_NUM_DATA_POINTERS; i++) {
428             av_freep(&buf->base[i]);
429             buf->data[i]= NULL;
430         }
431     }
432
433     if (!buf->base[0]) {
434         int h_chroma_shift, v_chroma_shift;
435         int size[4] = {0};
436         int tmpsize;
437         int unaligned;
438         AVPicture picture;
439         int stride_align[AV_NUM_DATA_POINTERS];
440         const int pixel_size = av_pix_fmt_descriptors[s->pix_fmt].comp[0].step_minus1+1;
441
442         avcodec_get_chroma_sub_sample(s->pix_fmt, &h_chroma_shift, &v_chroma_shift);
443
444         avcodec_align_dimensions2(s, &w, &h, stride_align);
445
446         if(!(s->flags&CODEC_FLAG_EMU_EDGE)){
447             w+= EDGE_WIDTH*2;
448             h+= EDGE_WIDTH*2;
449         }
450
451         do {
452             // NOTE: do not align linesizes individually, this breaks e.g. assumptions
453             // that linesize[0] == 2*linesize[1] in the MPEG-encoder for 4:2:2
454             av_image_fill_linesizes(picture.linesize, s->pix_fmt, w);
455             // increase alignment of w for next try (rhs gives the lowest bit set in w)
456             w += w & ~(w-1);
457
458             unaligned = 0;
459             for (i=0; i<4; i++){
460                 unaligned |= picture.linesize[i] % stride_align[i];
461             }
462         } while (unaligned);
463
464         tmpsize = av_image_fill_pointers(picture.data, s->pix_fmt, h, NULL, picture.linesize);
465         if (tmpsize < 0)
466             return -1;
467
468         for (i=0; i<3 && picture.data[i+1]; i++)
469             size[i] = picture.data[i+1] - picture.data[i];
470         size[i] = tmpsize - (picture.data[i] - picture.data[0]);
471
472         memset(buf->base, 0, sizeof(buf->base));
473         memset(buf->data, 0, sizeof(buf->data));
474
475         for(i=0; i<4 && size[i]; i++){
476             const int h_shift= i==0 ? 0 : h_chroma_shift;
477             const int v_shift= i==0 ? 0 : v_chroma_shift;
478
479             buf->linesize[i]= picture.linesize[i];
480
481             buf->base[i]= av_malloc(size[i]+16); //FIXME 16
482             if(buf->base[i]==NULL) return -1;
483             memset(buf->base[i], 128, size[i]);
484
485             // no edge if EDGE EMU or not planar YUV
486             if((s->flags&CODEC_FLAG_EMU_EDGE) || !size[2])
487                 buf->data[i] = buf->base[i];
488             else
489                 buf->data[i] = buf->base[i] + FFALIGN((buf->linesize[i]*EDGE_WIDTH>>v_shift) + (pixel_size*EDGE_WIDTH>>h_shift), stride_align[i]);
490         }
491         for (; i < AV_NUM_DATA_POINTERS; i++) {
492             buf->base[i] = buf->data[i] = NULL;
493             buf->linesize[i] = 0;
494         }
495         if(size[1] && !size[2])
496             ff_set_systematic_pal2((uint32_t*)buf->data[1], s->pix_fmt);
497         buf->width  = s->width;
498         buf->height = s->height;
499         buf->pix_fmt= s->pix_fmt;
500     }
501     pic->type= FF_BUFFER_TYPE_INTERNAL;
502
503     for (i = 0; i < AV_NUM_DATA_POINTERS; i++) {
504         pic->base[i]= buf->base[i];
505         pic->data[i]= buf->data[i];
506         pic->linesize[i]= buf->linesize[i];
507     }
508     pic->extended_data = pic->data;
509     avci->buffer_count++;
510     pic->width  = buf->width;
511     pic->height = buf->height;
512     pic->format = buf->pix_fmt;
513     pic->sample_aspect_ratio = s->sample_aspect_ratio;
514
515     if (s->pkt) {
516         pic->pkt_pts = s->pkt->pts;
517         pic->pkt_pos = s->pkt->pos;
518     } else {
519         pic->pkt_pts = AV_NOPTS_VALUE;
520         pic->pkt_pos = -1;
521     }
522     pic->reordered_opaque= s->reordered_opaque;
523     pic->sample_aspect_ratio = s->sample_aspect_ratio;
524     pic->width               = s->width;
525     pic->height              = s->height;
526     pic->format              = s->pix_fmt;
527
528     if(s->debug&FF_DEBUG_BUFFERS)
529         av_log(s, AV_LOG_DEBUG, "default_get_buffer called on pic %p, %d "
530                "buffers used\n", pic, avci->buffer_count);
531
532     return 0;
533 }
534
535 int avcodec_default_get_buffer(AVCodecContext *avctx, AVFrame *frame)
536 {
537     switch (avctx->codec_type) {
538     case AVMEDIA_TYPE_VIDEO:
539         return video_get_buffer(avctx, frame);
540     case AVMEDIA_TYPE_AUDIO:
541         return audio_get_buffer(avctx, frame);
542     default:
543         return -1;
544     }
545 }
546
547 void avcodec_default_release_buffer(AVCodecContext *s, AVFrame *pic){
548     int i;
549     InternalBuffer *buf, *last;
550     AVCodecInternal *avci = s->internal;
551
552     assert(s->codec_type == AVMEDIA_TYPE_VIDEO);
553
554     assert(pic->type==FF_BUFFER_TYPE_INTERNAL);
555     assert(avci->buffer_count);
556
557     if (avci->buffer) {
558         buf = NULL; /* avoids warning */
559         for (i = 0; i < avci->buffer_count; i++) { //just 3-5 checks so is not worth to optimize
560             buf = &avci->buffer[i];
561             if (buf->data[0] == pic->data[0])
562                 break;
563         }
564         assert(i < avci->buffer_count);
565         avci->buffer_count--;
566         last = &avci->buffer[avci->buffer_count];
567
568         if (buf != last)
569             FFSWAP(InternalBuffer, *buf, *last);
570     }
571
572     for (i = 0; i < AV_NUM_DATA_POINTERS; i++) {
573         pic->data[i]=NULL;
574 //        pic->base[i]=NULL;
575     }
576 //printf("R%X\n", pic->opaque);
577
578     if(s->debug&FF_DEBUG_BUFFERS)
579         av_log(s, AV_LOG_DEBUG, "default_release_buffer called on pic %p, %d "
580                "buffers used\n", pic, avci->buffer_count);
581 }
582
583 int avcodec_default_reget_buffer(AVCodecContext *s, AVFrame *pic){
584     AVFrame temp_pic;
585     int i;
586
587     assert(s->codec_type == AVMEDIA_TYPE_VIDEO);
588
589     if (pic->data[0] && (pic->width != s->width || pic->height != s->height || pic->format != s->pix_fmt)) {
590         av_log(s, AV_LOG_WARNING, "Picture changed from size:%dx%d fmt:%s to size:%dx%d fmt:%s in reget buffer()\n",
591                pic->width, pic->height, av_get_pix_fmt_name(pic->format), s->width, s->height, av_get_pix_fmt_name(s->pix_fmt));
592         s->release_buffer(s, pic);
593     }
594
595     ff_init_buffer_info(s, pic);
596
597     /* If no picture return a new buffer */
598     if(pic->data[0] == NULL) {
599         /* We will copy from buffer, so must be readable */
600         pic->buffer_hints |= FF_BUFFER_HINTS_READABLE;
601         return s->get_buffer(s, pic);
602     }
603
604     assert(s->pix_fmt == pic->format);
605
606     /* If internal buffer type return the same buffer */
607     if(pic->type == FF_BUFFER_TYPE_INTERNAL) {
608         return 0;
609     }
610
611     /*
612      * Not internal type and reget_buffer not overridden, emulate cr buffer
613      */
614     temp_pic = *pic;
615     for(i = 0; i < AV_NUM_DATA_POINTERS; i++)
616         pic->data[i] = pic->base[i] = NULL;
617     pic->opaque = NULL;
618     /* Allocate new frame */
619     if (s->get_buffer(s, pic))
620         return -1;
621     /* Copy image data from old buffer to new buffer */
622     av_picture_copy((AVPicture*)pic, (AVPicture*)&temp_pic, s->pix_fmt, s->width,
623              s->height);
624     s->release_buffer(s, &temp_pic); // Release old frame
625     return 0;
626 }
627
628 int avcodec_default_execute(AVCodecContext *c, int (*func)(AVCodecContext *c2, void *arg2),void *arg, int *ret, int count, int size){
629     int i;
630
631     for(i=0; i<count; i++){
632         int r= func(c, (char*)arg + i*size);
633         if(ret) ret[i]= r;
634     }
635     return 0;
636 }
637
638 int avcodec_default_execute2(AVCodecContext *c, int (*func)(AVCodecContext *c2, void *arg2, int jobnr, int threadnr),void *arg, int *ret, int count){
639     int i;
640
641     for(i=0; i<count; i++){
642         int r= func(c, arg, i, 0);
643         if(ret) ret[i]= r;
644     }
645     return 0;
646 }
647
648 enum PixelFormat avcodec_default_get_format(struct AVCodecContext *s, const enum PixelFormat *fmt){
649     while (*fmt != PIX_FMT_NONE && ff_is_hwaccel_pix_fmt(*fmt))
650         ++fmt;
651     return fmt[0];
652 }
653
654 void avcodec_get_frame_defaults(AVFrame *pic){
655     memset(pic, 0, sizeof(AVFrame));
656
657     pic->pts = pic->pkt_dts = pic->pkt_pts = pic->best_effort_timestamp = AV_NOPTS_VALUE;
658     pic->pkt_pos = -1;
659     pic->key_frame= 1;
660     pic->sample_aspect_ratio = (AVRational){0, 1};
661     pic->format = -1;           /* unknown */
662 }
663
664 AVFrame *avcodec_alloc_frame(void){
665     AVFrame *pic= av_malloc(sizeof(AVFrame));
666
667     if(pic==NULL) return NULL;
668
669     avcodec_get_frame_defaults(pic);
670
671     return pic;
672 }
673
674 static void avcodec_get_subtitle_defaults(AVSubtitle *sub)
675 {
676     memset(sub, 0, sizeof(*sub));
677     sub->pts = AV_NOPTS_VALUE;
678 }
679
680 static int get_bit_rate(AVCodecContext *ctx)
681 {
682     int bit_rate;
683     int bits_per_sample;
684
685     switch(ctx->codec_type) {
686     case AVMEDIA_TYPE_VIDEO:
687     case AVMEDIA_TYPE_DATA:
688     case AVMEDIA_TYPE_SUBTITLE:
689     case AVMEDIA_TYPE_ATTACHMENT:
690         bit_rate = ctx->bit_rate;
691         break;
692     case AVMEDIA_TYPE_AUDIO:
693         bits_per_sample = av_get_bits_per_sample(ctx->codec_id);
694         bit_rate = bits_per_sample ? ctx->sample_rate * ctx->channels * bits_per_sample : ctx->bit_rate;
695         break;
696     default:
697         bit_rate = 0;
698         break;
699     }
700     return bit_rate;
701 }
702
703 #if FF_API_AVCODEC_OPEN
704 int attribute_align_arg avcodec_open(AVCodecContext *avctx, AVCodec *codec)
705 {
706     return avcodec_open2(avctx, codec, NULL);
707 }
708 #endif
709
710 int attribute_align_arg avcodec_open2(AVCodecContext *avctx, AVCodec *codec, AVDictionary **options)
711 {
712     int ret = 0;
713     AVDictionary *tmp = NULL;
714
715     if (avcodec_is_open(avctx))
716         return 0;
717
718     if ((!codec && !avctx->codec)) {
719         av_log(avctx, AV_LOG_ERROR, "No codec provided to avcodec_open2().\n");
720         return AVERROR(EINVAL);
721     }
722     if ((codec && avctx->codec && codec != avctx->codec)) {
723         av_log(avctx, AV_LOG_ERROR, "This AVCodecContext was allocated for %s, "
724                "but %s passed to avcodec_open2().\n", avctx->codec->name, codec->name);
725         return AVERROR(EINVAL);
726     }
727     if (!codec)
728         codec = avctx->codec;
729
730     if (avctx->extradata_size < 0 || avctx->extradata_size >= FF_MAX_EXTRADATA_SIZE)
731         return AVERROR(EINVAL);
732
733     if (options)
734         av_dict_copy(&tmp, *options, 0);
735
736     /* If there is a user-supplied mutex locking routine, call it. */
737     if (ff_lockmgr_cb) {
738         if ((*ff_lockmgr_cb)(&codec_mutex, AV_LOCK_OBTAIN))
739             return -1;
740     }
741
742     entangled_thread_counter++;
743     if(entangled_thread_counter != 1){
744         av_log(avctx, AV_LOG_ERROR, "insufficient thread locking around avcodec_open/close()\n");
745         ret = -1;
746         goto end;
747     }
748
749     avctx->internal = av_mallocz(sizeof(AVCodecInternal));
750     if (!avctx->internal) {
751         ret = AVERROR(ENOMEM);
752         goto end;
753     }
754
755     if (codec->priv_data_size > 0) {
756       if(!avctx->priv_data){
757         avctx->priv_data = av_mallocz(codec->priv_data_size);
758         if (!avctx->priv_data) {
759             ret = AVERROR(ENOMEM);
760             goto end;
761         }
762         if (codec->priv_class) {
763             *(const AVClass**)avctx->priv_data= codec->priv_class;
764             av_opt_set_defaults(avctx->priv_data);
765         }
766       }
767       if (codec->priv_class && (ret = av_opt_set_dict(avctx->priv_data, &tmp)) < 0)
768           goto free_and_end;
769     } else {
770         avctx->priv_data = NULL;
771     }
772     if ((ret = av_opt_set_dict(avctx, &tmp)) < 0)
773         goto free_and_end;
774
775     if (codec->capabilities & CODEC_CAP_EXPERIMENTAL)
776         if (avctx->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) {
777             av_log(avctx, AV_LOG_ERROR, "Codec is experimental but experimental codecs are not enabled, see -strict -2\n");
778             ret = -1;
779             goto free_and_end;
780         }
781
782     //We only call avcodec_set_dimensions() for non h264 codecs so as not to overwrite previously setup dimensions
783     if(!( avctx->coded_width && avctx->coded_height && avctx->width && avctx->height && avctx->codec_id == CODEC_ID_H264)){
784     if(avctx->coded_width && avctx->coded_height)
785         avcodec_set_dimensions(avctx, avctx->coded_width, avctx->coded_height);
786     else if(avctx->width && avctx->height)
787         avcodec_set_dimensions(avctx, avctx->width, avctx->height);
788     }
789
790     if ((avctx->coded_width || avctx->coded_height || avctx->width || avctx->height)
791         && (  av_image_check_size(avctx->coded_width, avctx->coded_height, 0, avctx) < 0
792            || av_image_check_size(avctx->width,       avctx->height,       0, avctx) < 0)) {
793         av_log(avctx, AV_LOG_WARNING, "ignoring invalid width/height values\n");
794         avcodec_set_dimensions(avctx, 0, 0);
795     }
796
797     /* if the decoder init function was already called previously,
798        free the already allocated subtitle_header before overwriting it */
799     if (av_codec_is_decoder(codec))
800         av_freep(&avctx->subtitle_header);
801
802 #define SANE_NB_CHANNELS 128U
803     if (avctx->channels > SANE_NB_CHANNELS) {
804         ret = AVERROR(EINVAL);
805         goto free_and_end;
806     }
807
808     avctx->codec = codec;
809     if ((avctx->codec_type == AVMEDIA_TYPE_UNKNOWN || avctx->codec_type == codec->type) &&
810         avctx->codec_id == CODEC_ID_NONE) {
811         avctx->codec_type = codec->type;
812         avctx->codec_id   = codec->id;
813     }
814     if (avctx->codec_id != codec->id || (avctx->codec_type != codec->type
815                            && avctx->codec_type != AVMEDIA_TYPE_ATTACHMENT)) {
816         av_log(avctx, AV_LOG_ERROR, "codec type or id mismatches\n");
817         ret = AVERROR(EINVAL);
818         goto free_and_end;
819     }
820     avctx->frame_number = 0;
821
822     if (avctx->codec_type == AVMEDIA_TYPE_AUDIO &&
823         (!avctx->time_base.num || !avctx->time_base.den)) {
824         avctx->time_base.num = 1;
825         avctx->time_base.den = avctx->sample_rate;
826     }
827
828     if (!HAVE_THREADS)
829         av_log(avctx, AV_LOG_WARNING, "Warning: not compiled with thread support, using thread emulation\n");
830
831     if (HAVE_THREADS && !avctx->thread_opaque) {
832         ret = ff_thread_init(avctx);
833         if (ret < 0) {
834             goto free_and_end;
835         }
836     }
837     if (!HAVE_THREADS && !(codec->capabilities & CODEC_CAP_AUTO_THREADS))
838         avctx->thread_count = 1;
839
840     if (avctx->codec->max_lowres < avctx->lowres || avctx->lowres < 0) {
841         av_log(avctx, AV_LOG_ERROR, "The maximum value for lowres supported by the decoder is %d\n",
842                avctx->codec->max_lowres);
843         ret = AVERROR(EINVAL);
844         goto free_and_end;
845     }
846
847     if (av_codec_is_encoder(avctx->codec)) {
848         int i;
849         if (avctx->codec->sample_fmts) {
850             for (i = 0; avctx->codec->sample_fmts[i] != AV_SAMPLE_FMT_NONE; i++)
851                 if (avctx->sample_fmt == avctx->codec->sample_fmts[i])
852                     break;
853             if (avctx->codec->sample_fmts[i] == AV_SAMPLE_FMT_NONE) {
854                 av_log(avctx, AV_LOG_ERROR, "Specified sample_fmt is not supported.\n");
855                 ret = AVERROR(EINVAL);
856                 goto free_and_end;
857             }
858         }
859         if (avctx->codec->pix_fmts) {
860             for (i = 0; avctx->codec->pix_fmts[i] != PIX_FMT_NONE; i++)
861                 if (avctx->pix_fmt == avctx->codec->pix_fmts[i])
862                     break;
863             if (avctx->codec->pix_fmts[i] == PIX_FMT_NONE) {
864                 av_log(avctx, AV_LOG_ERROR, "Specified pix_fmt is not supported\n");
865                 ret = AVERROR(EINVAL);
866                 goto free_and_end;
867             }
868         }
869         if (avctx->codec->supported_samplerates) {
870             for (i = 0; avctx->codec->supported_samplerates[i] != 0; i++)
871                 if (avctx->sample_rate == avctx->codec->supported_samplerates[i])
872                     break;
873             if (avctx->codec->supported_samplerates[i] == 0) {
874                 av_log(avctx, AV_LOG_ERROR, "Specified sample_rate is not supported\n");
875                 ret = AVERROR(EINVAL);
876                 goto free_and_end;
877             }
878         }
879         if (avctx->codec->channel_layouts) {
880             if (!avctx->channel_layout) {
881                 av_log(avctx, AV_LOG_WARNING, "channel_layout not specified\n");
882             } else {
883                 for (i = 0; avctx->codec->channel_layouts[i] != 0; i++)
884                     if (avctx->channel_layout == avctx->codec->channel_layouts[i])
885                         break;
886                 if (avctx->codec->channel_layouts[i] == 0) {
887                     av_log(avctx, AV_LOG_ERROR, "Specified channel_layout is not supported\n");
888                     ret = AVERROR(EINVAL);
889                     goto free_and_end;
890                 }
891             }
892         }
893         if (avctx->channel_layout && avctx->channels) {
894             if (av_get_channel_layout_nb_channels(avctx->channel_layout) != avctx->channels) {
895                 av_log(avctx, AV_LOG_ERROR, "channel layout does not match number of channels\n");
896                 ret = AVERROR(EINVAL);
897                 goto free_and_end;
898             }
899         } else if (avctx->channel_layout) {
900             avctx->channels = av_get_channel_layout_nb_channels(avctx->channel_layout);
901         }
902     }
903
904     avctx->pts_correction_num_faulty_pts =
905     avctx->pts_correction_num_faulty_dts = 0;
906     avctx->pts_correction_last_pts =
907     avctx->pts_correction_last_dts = INT64_MIN;
908
909     if(avctx->codec->init && !(avctx->active_thread_type&FF_THREAD_FRAME)){
910         ret = avctx->codec->init(avctx);
911         if (ret < 0) {
912             goto free_and_end;
913         }
914     }
915
916     ret=0;
917
918     if (av_codec_is_decoder(avctx->codec)) {
919         if (!avctx->bit_rate)
920             avctx->bit_rate = get_bit_rate(avctx);
921         /* validate channel layout from the decoder */
922         if (avctx->channel_layout &&
923             av_get_channel_layout_nb_channels(avctx->channel_layout) != avctx->channels) {
924             av_log(avctx, AV_LOG_WARNING, "channel layout does not match number of channels\n");
925             avctx->channel_layout = 0;
926         }
927     }
928 end:
929     entangled_thread_counter--;
930
931     /* Release any user-supplied mutex. */
932     if (ff_lockmgr_cb) {
933         (*ff_lockmgr_cb)(&codec_mutex, AV_LOCK_RELEASE);
934     }
935     if (options) {
936         av_dict_free(options);
937         *options = tmp;
938     }
939
940     return ret;
941 free_and_end:
942     av_dict_free(&tmp);
943     av_freep(&avctx->priv_data);
944     av_freep(&avctx->internal);
945     avctx->codec= NULL;
946     goto end;
947 }
948
949 int ff_alloc_packet2(AVCodecContext *avctx, AVPacket *avpkt, int size)
950 {
951     if (size < 0 || avpkt->size < 0 || size > INT_MAX - FF_INPUT_BUFFER_PADDING_SIZE) {
952         av_log(avctx, AV_LOG_ERROR, "Size %d invalid\n", size);
953         return AVERROR(EINVAL);
954     }
955
956     av_assert0(!avpkt->data || avpkt->data != avctx->internal->byte_buffer);
957     if (!avpkt->data || avpkt->size < size) {
958         av_fast_padded_malloc(&avctx->internal->byte_buffer, &avctx->internal->byte_buffer_size, size);
959         avpkt->data = avctx->internal->byte_buffer;
960         avpkt->size = avctx->internal->byte_buffer_size;
961         avpkt->destruct = NULL;
962     }
963
964     if (avpkt->data) {
965         void *destruct = avpkt->destruct;
966
967         if (avpkt->size < size) {
968             av_log(avctx, AV_LOG_ERROR, "User packet is too small (%d < %d)\n", avpkt->size, size);
969             return AVERROR(EINVAL);
970         }
971
972         av_init_packet(avpkt);
973         avpkt->destruct = destruct;
974         avpkt->size = size;
975         return 0;
976     } else {
977         int ret = av_new_packet(avpkt, size);
978         if (ret < 0)
979             av_log(avctx, AV_LOG_ERROR, "Failed to allocate packet of size %d\n", size);
980         return ret;
981     }
982 }
983
984 int ff_alloc_packet(AVPacket *avpkt, int size)
985 {
986     return ff_alloc_packet2(NULL, avpkt, size);
987 }
988
989 int attribute_align_arg avcodec_encode_audio2(AVCodecContext *avctx,
990                                               AVPacket *avpkt,
991                                               const AVFrame *frame,
992                                               int *got_packet_ptr)
993 {
994     int ret;
995     AVPacket user_pkt = *avpkt;
996     int nb_samples;
997     int needs_realloc = !user_pkt.data;
998
999     *got_packet_ptr = 0;
1000
1001     if (!(avctx->codec->capabilities & CODEC_CAP_DELAY) && !frame) {
1002         av_free_packet(avpkt);
1003         av_init_packet(avpkt);
1004         avpkt->size = 0;
1005         return 0;
1006     }
1007
1008     /* check for valid frame size */
1009     if (frame) {
1010         nb_samples = frame->nb_samples;
1011         if (avctx->codec->capabilities & CODEC_CAP_SMALL_LAST_FRAME) {
1012             if (nb_samples > avctx->frame_size)
1013                 return AVERROR(EINVAL);
1014         } else if (!(avctx->codec->capabilities & CODEC_CAP_VARIABLE_FRAME_SIZE)) {
1015             if (nb_samples != avctx->frame_size)
1016                 return AVERROR(EINVAL);
1017         }
1018     } else {
1019         nb_samples = avctx->frame_size;
1020     }
1021
1022     if (avctx->codec->encode2) {
1023         ret = avctx->codec->encode2(avctx, avpkt, frame, got_packet_ptr);
1024         if (!ret && *got_packet_ptr) {
1025             if (!(avctx->codec->capabilities & CODEC_CAP_DELAY)) {
1026                 if (avpkt->pts == AV_NOPTS_VALUE)
1027                     avpkt->pts = frame->pts;
1028                 if (!avpkt->duration)
1029                     avpkt->duration = ff_samples_to_time_base(avctx,
1030                                                               frame->nb_samples);
1031             }
1032             avpkt->dts = avpkt->pts;
1033         } else {
1034             avpkt->size = 0;
1035         }
1036     } else {
1037         /* for compatibility with encoders not supporting encode2(), we need to
1038            allocate a packet buffer if the user has not provided one or check
1039            the size otherwise */
1040         int fs_tmp   = 0;
1041         int buf_size = avpkt->size;
1042         if (!user_pkt.data) {
1043             if (avctx->codec->capabilities & CODEC_CAP_VARIABLE_FRAME_SIZE) {
1044                 av_assert0(av_get_bits_per_sample(avctx->codec_id) != 0);
1045                 if (!frame)
1046                     return AVERROR(EINVAL);
1047                 buf_size = nb_samples * avctx->channels *
1048                            av_get_bits_per_sample(avctx->codec_id) / 8;
1049             } else {
1050                 /* this is a guess as to the required size.
1051                    if an encoder needs more than this, it should probably
1052                    implement encode2() */
1053                 buf_size = 2 * avctx->frame_size * avctx->channels *
1054                            av_get_bytes_per_sample(avctx->sample_fmt);
1055                 buf_size += 2*FF_MIN_BUFFER_SIZE;
1056             }
1057         }
1058         if ((ret = ff_alloc_packet2(avctx, avpkt, buf_size)))
1059             return ret;
1060
1061         /* Encoders using AVCodec.encode() that support
1062            CODEC_CAP_SMALL_LAST_FRAME require avctx->frame_size to be set to
1063            the smaller size when encoding the last frame.
1064            This code can be removed once all encoders supporting
1065            CODEC_CAP_SMALL_LAST_FRAME use encode2() */
1066         if ((avctx->codec->capabilities & CODEC_CAP_SMALL_LAST_FRAME) &&
1067             nb_samples < avctx->frame_size) {
1068             fs_tmp = avctx->frame_size;
1069             avctx->frame_size = nb_samples;
1070         }
1071
1072         /* encode the frame */
1073         ret = avctx->codec->encode(avctx, avpkt->data, avpkt->size,
1074                                    frame ? frame->data[0] : NULL);
1075         if (ret >= 0) {
1076             if (!ret) {
1077                 /* no output. if the packet data was allocated by libavcodec,
1078                    free it */
1079                 if (!user_pkt.data && avpkt->data != avctx->internal->byte_buffer)
1080                     av_freep(&avpkt->data);
1081             } else {
1082                 if (avctx->coded_frame)
1083                     avpkt->pts = avpkt->dts = avctx->coded_frame->pts;
1084                 /* Set duration for final small packet. This can be removed
1085                    once all encoders supporting CODEC_CAP_SMALL_LAST_FRAME use
1086                    encode2() */
1087                 if (fs_tmp) {
1088                     avpkt->duration = ff_samples_to_time_base(avctx,
1089                                                               avctx->frame_size);
1090                 }
1091             }
1092             avpkt->size = ret;
1093             *got_packet_ptr = (ret > 0);
1094             ret = 0;
1095         }
1096
1097         if (fs_tmp)
1098             avctx->frame_size = fs_tmp;
1099     }
1100     if (avpkt->data && avpkt->data == avctx->internal->byte_buffer) {
1101         needs_realloc = 0;
1102         if (user_pkt.data) {
1103             if (user_pkt.size >= avpkt->size) {
1104                 memcpy(user_pkt.data, avpkt->data, avpkt->size);
1105             } else {
1106                 av_log(avctx, AV_LOG_ERROR, "Provided packet is too small, needs to be %d\n", avpkt->size);
1107                 avpkt->size = user_pkt.size;
1108                 ret = -1;
1109             }
1110             avpkt->data     = user_pkt.data;
1111             avpkt->destruct = user_pkt.destruct;
1112         } else {
1113             if (av_dup_packet(avpkt) < 0) {
1114                 ret = AVERROR(ENOMEM);
1115             }
1116         }
1117     }
1118
1119     if (!ret) {
1120         if (needs_realloc && avpkt->data) {
1121             uint8_t *new_data = av_realloc(avpkt->data, avpkt->size + FF_INPUT_BUFFER_PADDING_SIZE);
1122             if (new_data)
1123                 avpkt->data = new_data;
1124         }
1125
1126         avctx->frame_number++;
1127     }
1128
1129     if (ret < 0 || !*got_packet_ptr)
1130         av_free_packet(avpkt);
1131
1132     /* NOTE: if we add any audio encoders which output non-keyframe packets,
1133              this needs to be moved to the encoders, but for now we can do it
1134              here to simplify things */
1135     avpkt->flags |= AV_PKT_FLAG_KEY;
1136
1137     return ret;
1138 }
1139
1140 #if FF_API_OLD_DECODE_AUDIO
1141 int attribute_align_arg avcodec_encode_audio(AVCodecContext *avctx,
1142                                              uint8_t *buf, int buf_size,
1143                                              const short *samples)
1144 {
1145     AVPacket pkt;
1146     AVFrame frame0;
1147     AVFrame *frame;
1148     int ret, samples_size, got_packet;
1149
1150     av_init_packet(&pkt);
1151     pkt.data = buf;
1152     pkt.size = buf_size;
1153
1154     if (samples) {
1155         frame = &frame0;
1156         avcodec_get_frame_defaults(frame);
1157
1158         if (avctx->frame_size) {
1159             frame->nb_samples = avctx->frame_size;
1160         } else {
1161             /* if frame_size is not set, the number of samples must be
1162                calculated from the buffer size */
1163             int64_t nb_samples;
1164             if (!av_get_bits_per_sample(avctx->codec_id)) {
1165                 av_log(avctx, AV_LOG_ERROR, "avcodec_encode_audio() does not "
1166                        "support this codec\n");
1167                 return AVERROR(EINVAL);
1168             }
1169             nb_samples = (int64_t)buf_size * 8 /
1170                          (av_get_bits_per_sample(avctx->codec_id) *
1171                          avctx->channels);
1172             if (nb_samples >= INT_MAX)
1173                 return AVERROR(EINVAL);
1174             frame->nb_samples = nb_samples;
1175         }
1176
1177         /* it is assumed that the samples buffer is large enough based on the
1178            relevant parameters */
1179         samples_size = av_samples_get_buffer_size(NULL, avctx->channels,
1180                                                   frame->nb_samples,
1181                                                   avctx->sample_fmt, 1);
1182         if ((ret = avcodec_fill_audio_frame(frame, avctx->channels,
1183                                             avctx->sample_fmt,
1184                                             (const uint8_t *)samples, samples_size, 1)))
1185             return ret;
1186
1187         /* fabricate frame pts from sample count.
1188            this is needed because the avcodec_encode_audio() API does not have
1189            a way for the user to provide pts */
1190         if(avctx->sample_rate && avctx->time_base.num)
1191             frame->pts = ff_samples_to_time_base(avctx,
1192                                                 avctx->internal->sample_count);
1193         else
1194             frame->pts = AV_NOPTS_VALUE;
1195         avctx->internal->sample_count += frame->nb_samples;
1196     } else {
1197         frame = NULL;
1198     }
1199
1200     got_packet = 0;
1201     ret = avcodec_encode_audio2(avctx, &pkt, frame, &got_packet);
1202     if (!ret && got_packet && avctx->coded_frame) {
1203         avctx->coded_frame->pts       = pkt.pts;
1204         avctx->coded_frame->key_frame = !!(pkt.flags & AV_PKT_FLAG_KEY);
1205     }
1206     /* free any side data since we cannot return it */
1207     ff_packet_free_side_data(&pkt);
1208
1209     if (frame && frame->extended_data != frame->data)
1210         av_freep(&frame->extended_data);
1211
1212     return ret ? ret : pkt.size;
1213 }
1214 #endif
1215
1216 #if FF_API_OLD_ENCODE_VIDEO
1217 int attribute_align_arg avcodec_encode_video(AVCodecContext *avctx, uint8_t *buf, int buf_size,
1218                          const AVFrame *pict)
1219 {
1220     AVPacket pkt;
1221     int ret, got_packet = 0;
1222
1223     if(buf_size < FF_MIN_BUFFER_SIZE){
1224         av_log(avctx, AV_LOG_ERROR, "buffer smaller than minimum size\n");
1225         return -1;
1226     }
1227
1228     av_init_packet(&pkt);
1229     pkt.data = buf;
1230     pkt.size = buf_size;
1231
1232     ret = avcodec_encode_video2(avctx, &pkt, pict, &got_packet);
1233     if (!ret && got_packet && avctx->coded_frame) {
1234         avctx->coded_frame->pts       = pkt.pts;
1235         avctx->coded_frame->key_frame = !!(pkt.flags & AV_PKT_FLAG_KEY);
1236     }
1237
1238     /* free any side data since we cannot return it */
1239     if (pkt.side_data_elems > 0) {
1240         int i;
1241         for (i = 0; i < pkt.side_data_elems; i++)
1242             av_free(pkt.side_data[i].data);
1243         av_freep(&pkt.side_data);
1244         pkt.side_data_elems = 0;
1245     }
1246
1247     return ret ? ret : pkt.size;
1248 }
1249 #endif
1250
1251 int attribute_align_arg avcodec_encode_video2(AVCodecContext *avctx,
1252                                               AVPacket *avpkt,
1253                                               const AVFrame *frame,
1254                                               int *got_packet_ptr)
1255 {
1256     int ret;
1257     AVPacket user_pkt = *avpkt;
1258     int needs_realloc = !user_pkt.data;
1259
1260     *got_packet_ptr = 0;
1261
1262     if (!(avctx->codec->capabilities & CODEC_CAP_DELAY) && !frame) {
1263         av_free_packet(avpkt);
1264         av_init_packet(avpkt);
1265         avpkt->size     = 0;
1266         return 0;
1267     }
1268
1269     if (av_image_check_size(avctx->width, avctx->height, 0, avctx))
1270         return AVERROR(EINVAL);
1271
1272     av_assert0(avctx->codec->encode2);
1273
1274     ret = avctx->codec->encode2(avctx, avpkt, frame, got_packet_ptr);
1275     av_assert0(ret <= 0);
1276
1277     if (avpkt->data && avpkt->data == avctx->internal->byte_buffer) {
1278         needs_realloc = 0;
1279         if (user_pkt.data) {
1280             if (user_pkt.size >= avpkt->size) {
1281                 memcpy(user_pkt.data, avpkt->data, avpkt->size);
1282             } else {
1283                 av_log(avctx, AV_LOG_ERROR, "Provided packet is too small, needs to be %d\n", avpkt->size);
1284                 avpkt->size = user_pkt.size;
1285                 ret = -1;
1286             }
1287             avpkt->data     = user_pkt.data;
1288             avpkt->destruct = user_pkt.destruct;
1289         } else {
1290             if (av_dup_packet(avpkt) < 0) {
1291                 ret = AVERROR(ENOMEM);
1292             }
1293         }
1294     }
1295
1296     if (!ret) {
1297         if (!*got_packet_ptr)
1298             avpkt->size = 0;
1299         else if (!(avctx->codec->capabilities & CODEC_CAP_DELAY))
1300             avpkt->pts = avpkt->dts = frame->pts;
1301
1302         if (needs_realloc && avpkt->data &&
1303             avpkt->destruct == av_destruct_packet) {
1304             uint8_t *new_data = av_realloc(avpkt->data, avpkt->size + FF_INPUT_BUFFER_PADDING_SIZE);
1305             if (new_data)
1306                 avpkt->data = new_data;
1307         }
1308
1309         avctx->frame_number++;
1310     }
1311
1312     if (ret < 0 || !*got_packet_ptr)
1313         av_free_packet(avpkt);
1314
1315     emms_c();
1316     return ret;
1317 }
1318
1319 int avcodec_encode_subtitle(AVCodecContext *avctx, uint8_t *buf, int buf_size,
1320                             const AVSubtitle *sub)
1321 {
1322     int ret;
1323     if(sub->start_display_time) {
1324         av_log(avctx, AV_LOG_ERROR, "start_display_time must be 0.\n");
1325         return -1;
1326     }
1327
1328     ret = avctx->codec->encode(avctx, buf, buf_size, (void *)(intptr_t)sub);
1329     avctx->frame_number++;
1330     return ret;
1331 }
1332
1333 /**
1334  * Attempt to guess proper monotonic timestamps for decoded video frames
1335  * which might have incorrect times. Input timestamps may wrap around, in
1336  * which case the output will as well.
1337  *
1338  * @param pts the pts field of the decoded AVPacket, as passed through
1339  * AVFrame.pkt_pts
1340  * @param dts the dts field of the decoded AVPacket
1341  * @return one of the input values, may be AV_NOPTS_VALUE
1342  */
1343 static int64_t guess_correct_pts(AVCodecContext *ctx,
1344                                  int64_t reordered_pts, int64_t dts)
1345 {
1346     int64_t pts = AV_NOPTS_VALUE;
1347
1348     if (dts != AV_NOPTS_VALUE) {
1349         ctx->pts_correction_num_faulty_dts += dts <= ctx->pts_correction_last_dts;
1350         ctx->pts_correction_last_dts = dts;
1351     }
1352     if (reordered_pts != AV_NOPTS_VALUE) {
1353         ctx->pts_correction_num_faulty_pts += reordered_pts <= ctx->pts_correction_last_pts;
1354         ctx->pts_correction_last_pts = reordered_pts;
1355     }
1356     if ((ctx->pts_correction_num_faulty_pts<=ctx->pts_correction_num_faulty_dts || dts == AV_NOPTS_VALUE)
1357        && reordered_pts != AV_NOPTS_VALUE)
1358         pts = reordered_pts;
1359     else
1360         pts = dts;
1361
1362     return pts;
1363 }
1364
1365 static void apply_param_change(AVCodecContext *avctx, AVPacket *avpkt)
1366 {
1367     int size = 0;
1368     const uint8_t *data;
1369     uint32_t flags;
1370
1371     if (!(avctx->codec->capabilities & CODEC_CAP_PARAM_CHANGE))
1372         return;
1373
1374     data = av_packet_get_side_data(avpkt, AV_PKT_DATA_PARAM_CHANGE, &size);
1375     if (!data || size < 4)
1376         return;
1377     flags = bytestream_get_le32(&data);
1378     size -= 4;
1379     if (size < 4) /* Required for any of the changes */
1380         return;
1381     if (flags & AV_SIDE_DATA_PARAM_CHANGE_CHANNEL_COUNT) {
1382         avctx->channels = bytestream_get_le32(&data);
1383         size -= 4;
1384     }
1385     if (flags & AV_SIDE_DATA_PARAM_CHANGE_CHANNEL_LAYOUT) {
1386         if (size < 8)
1387             return;
1388         avctx->channel_layout = bytestream_get_le64(&data);
1389         size -= 8;
1390     }
1391     if (size < 4)
1392         return;
1393     if (flags & AV_SIDE_DATA_PARAM_CHANGE_SAMPLE_RATE) {
1394         avctx->sample_rate = bytestream_get_le32(&data);
1395         size -= 4;
1396     }
1397     if (flags & AV_SIDE_DATA_PARAM_CHANGE_DIMENSIONS) {
1398         if (size < 8)
1399             return;
1400         avctx->width  = bytestream_get_le32(&data);
1401         avctx->height = bytestream_get_le32(&data);
1402         avcodec_set_dimensions(avctx, avctx->width, avctx->height);
1403         size -= 8;
1404     }
1405 }
1406
1407 int attribute_align_arg avcodec_decode_video2(AVCodecContext *avctx, AVFrame *picture,
1408                          int *got_picture_ptr,
1409                          const AVPacket *avpkt)
1410 {
1411     int ret;
1412     // copy to ensure we do not change avpkt
1413     AVPacket tmp = *avpkt;
1414
1415     if (avctx->codec->type != AVMEDIA_TYPE_VIDEO) {
1416         av_log(avctx, AV_LOG_ERROR, "Invalid media type for video\n");
1417         return AVERROR(EINVAL);
1418     }
1419
1420     *got_picture_ptr= 0;
1421     if((avctx->coded_width||avctx->coded_height) && av_image_check_size(avctx->coded_width, avctx->coded_height, 0, avctx))
1422         return -1;
1423
1424     if((avctx->codec->capabilities & CODEC_CAP_DELAY) || avpkt->size || (avctx->active_thread_type&FF_THREAD_FRAME)){
1425         int did_split = av_packet_split_side_data(&tmp);
1426         apply_param_change(avctx, &tmp);
1427         avctx->pkt = &tmp;
1428         if (HAVE_THREADS && avctx->active_thread_type&FF_THREAD_FRAME)
1429              ret = ff_thread_decode_frame(avctx, picture, got_picture_ptr,
1430                                           &tmp);
1431         else {
1432             ret = avctx->codec->decode(avctx, picture, got_picture_ptr,
1433                               &tmp);
1434             picture->pkt_dts= avpkt->dts;
1435
1436             if(!avctx->has_b_frames){
1437             picture->pkt_pos= avpkt->pos;
1438             }
1439             //FIXME these should be under if(!avctx->has_b_frames)
1440             if (!picture->sample_aspect_ratio.num)
1441                 picture->sample_aspect_ratio = avctx->sample_aspect_ratio;
1442             if (!picture->width)
1443                 picture->width = avctx->width;
1444             if (!picture->height)
1445                 picture->height = avctx->height;
1446             if (picture->format == PIX_FMT_NONE)
1447                 picture->format = avctx->pix_fmt;
1448         }
1449
1450         emms_c(); //needed to avoid an emms_c() call before every return;
1451
1452         avctx->pkt = NULL;
1453         if (did_split)
1454             ff_packet_free_side_data(&tmp);
1455
1456         if (*got_picture_ptr){
1457             avctx->frame_number++;
1458             picture->best_effort_timestamp = guess_correct_pts(avctx,
1459                                                             picture->pkt_pts,
1460                                                             picture->pkt_dts);
1461         }
1462     }else
1463         ret= 0;
1464
1465     return ret;
1466 }
1467
1468 #if FF_API_OLD_DECODE_AUDIO
1469 int attribute_align_arg avcodec_decode_audio3(AVCodecContext *avctx, int16_t *samples,
1470                          int *frame_size_ptr,
1471                          AVPacket *avpkt)
1472 {
1473     AVFrame frame;
1474     int ret, got_frame = 0;
1475
1476     if (avctx->get_buffer != avcodec_default_get_buffer) {
1477         av_log(avctx, AV_LOG_ERROR, "Custom get_buffer() for use with"
1478                "avcodec_decode_audio3() detected. Overriding with avcodec_default_get_buffer\n");
1479         av_log(avctx, AV_LOG_ERROR, "Please port your application to "
1480                "avcodec_decode_audio4()\n");
1481         avctx->get_buffer = avcodec_default_get_buffer;
1482         avctx->release_buffer = avcodec_default_release_buffer;
1483     }
1484
1485     ret = avcodec_decode_audio4(avctx, &frame, &got_frame, avpkt);
1486
1487     if (ret >= 0 && got_frame) {
1488         int ch, plane_size;
1489         int planar = av_sample_fmt_is_planar(avctx->sample_fmt);
1490         int data_size = av_samples_get_buffer_size(&plane_size, avctx->channels,
1491                                                    frame.nb_samples,
1492                                                    avctx->sample_fmt, 1);
1493         if (*frame_size_ptr < data_size) {
1494             av_log(avctx, AV_LOG_ERROR, "output buffer size is too small for "
1495                    "the current frame (%d < %d)\n", *frame_size_ptr, data_size);
1496             return AVERROR(EINVAL);
1497         }
1498
1499         memcpy(samples, frame.extended_data[0], plane_size);
1500
1501         if (planar && avctx->channels > 1) {
1502             uint8_t *out = ((uint8_t *)samples) + plane_size;
1503             for (ch = 1; ch < avctx->channels; ch++) {
1504                 memcpy(out, frame.extended_data[ch], plane_size);
1505                 out += plane_size;
1506             }
1507         }
1508         *frame_size_ptr = data_size;
1509     } else {
1510         *frame_size_ptr = 0;
1511     }
1512     return ret;
1513 }
1514 #endif
1515
1516 int attribute_align_arg avcodec_decode_audio4(AVCodecContext *avctx,
1517                                               AVFrame *frame,
1518                                               int *got_frame_ptr,
1519                                               const AVPacket *avpkt)
1520 {
1521     int ret = 0;
1522
1523     *got_frame_ptr = 0;
1524
1525     if (!avpkt->data && avpkt->size) {
1526         av_log(avctx, AV_LOG_ERROR, "invalid packet: NULL data, size != 0\n");
1527         return AVERROR(EINVAL);
1528     }
1529     if (avctx->codec->type != AVMEDIA_TYPE_AUDIO) {
1530         av_log(avctx, AV_LOG_ERROR, "Invalid media type for audio\n");
1531         return AVERROR(EINVAL);
1532     }
1533
1534     if ((avctx->codec->capabilities & CODEC_CAP_DELAY) || avpkt->size) {
1535         // copy to ensure we do not change avpkt
1536         AVPacket tmp = *avpkt;
1537         int did_split = av_packet_split_side_data(&tmp);
1538         apply_param_change(avctx, &tmp);
1539
1540         avctx->pkt = &tmp;
1541         ret = avctx->codec->decode(avctx, frame, got_frame_ptr, &tmp);
1542         if (ret >= 0 && *got_frame_ptr) {
1543             avctx->frame_number++;
1544             frame->pkt_dts = avpkt->dts;
1545             if (frame->format == AV_SAMPLE_FMT_NONE)
1546                 frame->format = avctx->sample_fmt;
1547         }
1548
1549         avctx->pkt = NULL;
1550         if (did_split)
1551             ff_packet_free_side_data(&tmp);
1552     }
1553     return ret;
1554 }
1555
1556 int avcodec_decode_subtitle2(AVCodecContext *avctx, AVSubtitle *sub,
1557                             int *got_sub_ptr,
1558                             AVPacket *avpkt)
1559 {
1560     int ret;
1561
1562     if (avctx->codec->type != AVMEDIA_TYPE_SUBTITLE) {
1563         av_log(avctx, AV_LOG_ERROR, "Invalid media type for subtitles\n");
1564         return AVERROR(EINVAL);
1565     }
1566
1567     avctx->pkt = avpkt;
1568     *got_sub_ptr = 0;
1569     avcodec_get_subtitle_defaults(sub);
1570     ret = avctx->codec->decode(avctx, sub, got_sub_ptr, avpkt);
1571     if (*got_sub_ptr)
1572         avctx->frame_number++;
1573     return ret;
1574 }
1575
1576 void avsubtitle_free(AVSubtitle *sub)
1577 {
1578     int i;
1579
1580     for (i = 0; i < sub->num_rects; i++)
1581     {
1582         av_freep(&sub->rects[i]->pict.data[0]);
1583         av_freep(&sub->rects[i]->pict.data[1]);
1584         av_freep(&sub->rects[i]->pict.data[2]);
1585         av_freep(&sub->rects[i]->pict.data[3]);
1586         av_freep(&sub->rects[i]->text);
1587         av_freep(&sub->rects[i]->ass);
1588         av_freep(&sub->rects[i]);
1589     }
1590
1591     av_freep(&sub->rects);
1592
1593     memset(sub, 0, sizeof(AVSubtitle));
1594 }
1595
1596 av_cold int avcodec_close(AVCodecContext *avctx)
1597 {
1598     /* If there is a user-supplied mutex locking routine, call it. */
1599     if (ff_lockmgr_cb) {
1600         if ((*ff_lockmgr_cb)(&codec_mutex, AV_LOCK_OBTAIN))
1601             return -1;
1602     }
1603
1604     entangled_thread_counter++;
1605     if(entangled_thread_counter != 1){
1606         av_log(avctx, AV_LOG_ERROR, "insufficient thread locking around avcodec_open/close()\n");
1607         entangled_thread_counter--;
1608         return -1;
1609     }
1610
1611     if (avcodec_is_open(avctx)) {
1612         if (HAVE_THREADS && avctx->thread_opaque)
1613             ff_thread_free(avctx);
1614         if (avctx->codec && avctx->codec->close)
1615             avctx->codec->close(avctx);
1616         avcodec_default_free_buffers(avctx);
1617         avctx->coded_frame = NULL;
1618         avctx->internal->byte_buffer_size = 0;
1619         av_freep(&avctx->internal->byte_buffer);
1620         av_freep(&avctx->internal);
1621     }
1622
1623     if (avctx->priv_data && avctx->codec && avctx->codec->priv_class)
1624         av_opt_free(avctx->priv_data);
1625     av_opt_free(avctx);
1626     av_freep(&avctx->priv_data);
1627     if (av_codec_is_encoder(avctx->codec))
1628         av_freep(&avctx->extradata);
1629     avctx->codec = NULL;
1630     avctx->active_thread_type = 0;
1631     entangled_thread_counter--;
1632
1633     /* Release any user-supplied mutex. */
1634     if (ff_lockmgr_cb) {
1635         (*ff_lockmgr_cb)(&codec_mutex, AV_LOCK_RELEASE);
1636     }
1637     return 0;
1638 }
1639
1640 static enum CodecID remap_deprecated_codec_id(enum CodecID id)
1641 {
1642     switch(id){
1643         //This is for future deprecatec codec ids, its empty since
1644         //last major bump but will fill up again over time, please dont remove it
1645 //         case CODEC_ID_UTVIDEO_DEPRECATED: return CODEC_ID_UTVIDEO;
1646         default                         : return id;
1647     }
1648 }
1649
1650 AVCodec *avcodec_find_encoder(enum CodecID id)
1651 {
1652     AVCodec *p, *experimental=NULL;
1653     p = first_avcodec;
1654     id= remap_deprecated_codec_id(id);
1655     while (p) {
1656         if (av_codec_is_encoder(p) && p->id == id) {
1657             if (p->capabilities & CODEC_CAP_EXPERIMENTAL && !experimental) {
1658                 experimental = p;
1659             } else
1660                 return p;
1661         }
1662         p = p->next;
1663     }
1664     return experimental;
1665 }
1666
1667 AVCodec *avcodec_find_encoder_by_name(const char *name)
1668 {
1669     AVCodec *p;
1670     if (!name)
1671         return NULL;
1672     p = first_avcodec;
1673     while (p) {
1674         if (av_codec_is_encoder(p) && strcmp(name,p->name) == 0)
1675             return p;
1676         p = p->next;
1677     }
1678     return NULL;
1679 }
1680
1681 AVCodec *avcodec_find_decoder(enum CodecID id)
1682 {
1683     AVCodec *p, *experimental=NULL;
1684     p = first_avcodec;
1685     id= remap_deprecated_codec_id(id);
1686     while (p) {
1687         if (av_codec_is_decoder(p) && p->id == id) {
1688             if (p->capabilities & CODEC_CAP_EXPERIMENTAL && !experimental) {
1689                 experimental = p;
1690             } else
1691                 return p;
1692         }
1693         p = p->next;
1694     }
1695     return experimental;
1696 }
1697
1698 AVCodec *avcodec_find_decoder_by_name(const char *name)
1699 {
1700     AVCodec *p;
1701     if (!name)
1702         return NULL;
1703     p = first_avcodec;
1704     while (p) {
1705         if (av_codec_is_decoder(p) && strcmp(name,p->name) == 0)
1706             return p;
1707         p = p->next;
1708     }
1709     return NULL;
1710 }
1711
1712 const char *avcodec_get_name(enum CodecID id)
1713 {
1714     AVCodec *codec;
1715
1716 #if !CONFIG_SMALL
1717     switch (id) {
1718 #include "libavcodec/codec_names.h"
1719     }
1720     av_log(NULL, AV_LOG_WARNING, "Codec 0x%x is not in the full list.\n", id);
1721 #endif
1722     codec = avcodec_find_decoder(id);
1723     if (codec)
1724         return codec->name;
1725     codec = avcodec_find_encoder(id);
1726     if (codec)
1727         return codec->name;
1728     return "unknown_codec";
1729 }
1730
1731 size_t av_get_codec_tag_string(char *buf, size_t buf_size, unsigned int codec_tag)
1732 {
1733     int i, len, ret = 0;
1734
1735     for (i = 0; i < 4; i++) {
1736         len = snprintf(buf, buf_size,
1737                        isprint(codec_tag&0xFF) ? "%c" : "[%d]", codec_tag&0xFF);
1738         buf      += len;
1739         buf_size  = buf_size > len ? buf_size - len : 0;
1740         ret      += len;
1741         codec_tag>>=8;
1742     }
1743     return ret;
1744 }
1745
1746 void avcodec_string(char *buf, int buf_size, AVCodecContext *enc, int encode)
1747 {
1748     const char *codec_type;
1749     const char *codec_name;
1750     const char *profile = NULL;
1751     AVCodec *p;
1752     int bitrate;
1753     AVRational display_aspect_ratio;
1754
1755     if (!buf || buf_size <= 0)
1756         return;
1757     codec_type = av_get_media_type_string(enc->codec_type);
1758     codec_name = avcodec_get_name(enc->codec_id);
1759     if (enc->profile != FF_PROFILE_UNKNOWN) {
1760         if (enc->codec)
1761             p = enc->codec;
1762         else
1763             p = encode ? avcodec_find_encoder(enc->codec_id) :
1764                         avcodec_find_decoder(enc->codec_id);
1765         if (p)
1766             profile = av_get_profile_name(p, enc->profile);
1767     }
1768
1769     snprintf(buf, buf_size, "%s: %s%s", codec_type ? codec_type : "unknown",
1770              codec_name, enc->mb_decision ? " (hq)" : "");
1771     buf[0] ^= 'a' ^ 'A'; /* first letter in uppercase */
1772     if (profile)
1773         snprintf(buf + strlen(buf), buf_size - strlen(buf), " (%s)", profile);
1774     if (enc->codec_tag) {
1775         char tag_buf[32];
1776         av_get_codec_tag_string(tag_buf, sizeof(tag_buf), enc->codec_tag);
1777         snprintf(buf + strlen(buf), buf_size - strlen(buf),
1778                  " (%s / 0x%04X)", tag_buf, enc->codec_tag);
1779     }
1780     switch(enc->codec_type) {
1781     case AVMEDIA_TYPE_VIDEO:
1782         if (enc->pix_fmt != PIX_FMT_NONE) {
1783             snprintf(buf + strlen(buf), buf_size - strlen(buf),
1784                      ", %s",
1785                      av_get_pix_fmt_name(enc->pix_fmt));
1786         }
1787         if (enc->width) {
1788             snprintf(buf + strlen(buf), buf_size - strlen(buf),
1789                      ", %dx%d",
1790                      enc->width, enc->height);
1791             if (enc->sample_aspect_ratio.num) {
1792                 av_reduce(&display_aspect_ratio.num, &display_aspect_ratio.den,
1793                           enc->width*enc->sample_aspect_ratio.num,
1794                           enc->height*enc->sample_aspect_ratio.den,
1795                           1024*1024);
1796                 snprintf(buf + strlen(buf), buf_size - strlen(buf),
1797                          " [SAR %d:%d DAR %d:%d]",
1798                          enc->sample_aspect_ratio.num, enc->sample_aspect_ratio.den,
1799                          display_aspect_ratio.num, display_aspect_ratio.den);
1800             }
1801             if(av_log_get_level() >= AV_LOG_DEBUG){
1802                 int g= av_gcd(enc->time_base.num, enc->time_base.den);
1803                 snprintf(buf + strlen(buf), buf_size - strlen(buf),
1804                      ", %d/%d",
1805                      enc->time_base.num/g, enc->time_base.den/g);
1806             }
1807         }
1808         if (encode) {
1809             snprintf(buf + strlen(buf), buf_size - strlen(buf),
1810                      ", q=%d-%d", enc->qmin, enc->qmax);
1811         }
1812         break;
1813     case AVMEDIA_TYPE_AUDIO:
1814         if (enc->sample_rate) {
1815             snprintf(buf + strlen(buf), buf_size - strlen(buf),
1816                      ", %d Hz", enc->sample_rate);
1817         }
1818         av_strlcat(buf, ", ", buf_size);
1819         av_get_channel_layout_string(buf + strlen(buf), buf_size - strlen(buf), enc->channels, enc->channel_layout);
1820         if (enc->sample_fmt != AV_SAMPLE_FMT_NONE) {
1821             snprintf(buf + strlen(buf), buf_size - strlen(buf),
1822                      ", %s", av_get_sample_fmt_name(enc->sample_fmt));
1823         }
1824         break;
1825     default:
1826         return;
1827     }
1828     if (encode) {
1829         if (enc->flags & CODEC_FLAG_PASS1)
1830             snprintf(buf + strlen(buf), buf_size - strlen(buf),
1831                      ", pass 1");
1832         if (enc->flags & CODEC_FLAG_PASS2)
1833             snprintf(buf + strlen(buf), buf_size - strlen(buf),
1834                      ", pass 2");
1835     }
1836     bitrate = get_bit_rate(enc);
1837     if (bitrate != 0) {
1838         snprintf(buf + strlen(buf), buf_size - strlen(buf),
1839                  ", %d kb/s", bitrate / 1000);
1840     }
1841 }
1842
1843 const char *av_get_profile_name(const AVCodec *codec, int profile)
1844 {
1845     const AVProfile *p;
1846     if (profile == FF_PROFILE_UNKNOWN || !codec->profiles)
1847         return NULL;
1848
1849     for (p = codec->profiles; p->profile != FF_PROFILE_UNKNOWN; p++)
1850         if (p->profile == profile)
1851             return p->name;
1852
1853     return NULL;
1854 }
1855
1856 unsigned avcodec_version( void )
1857 {
1858 //    av_assert0(CODEC_ID_V410==164);
1859     av_assert0(CODEC_ID_PCM_S8_PLANAR==65563);
1860     av_assert0(CODEC_ID_ADPCM_G722==69660);
1861 //     av_assert0(CODEC_ID_BMV_AUDIO==86071);
1862     av_assert0(CODEC_ID_SRT==94216);
1863     av_assert0(LIBAVCODEC_VERSION_MICRO >= 100);
1864
1865   return LIBAVCODEC_VERSION_INT;
1866 }
1867
1868 const char *avcodec_configuration(void)
1869 {
1870     return FFMPEG_CONFIGURATION;
1871 }
1872
1873 const char *avcodec_license(void)
1874 {
1875 #define LICENSE_PREFIX "libavcodec license: "
1876     return LICENSE_PREFIX FFMPEG_LICENSE + sizeof(LICENSE_PREFIX) - 1;
1877 }
1878
1879 void avcodec_flush_buffers(AVCodecContext *avctx)
1880 {
1881     if(HAVE_THREADS && avctx->active_thread_type&FF_THREAD_FRAME)
1882         ff_thread_flush(avctx);
1883     else if(avctx->codec->flush)
1884         avctx->codec->flush(avctx);
1885
1886     avctx->pts_correction_last_pts =
1887     avctx->pts_correction_last_dts = INT64_MIN;
1888 }
1889
1890 static void video_free_buffers(AVCodecContext *s)
1891 {
1892     AVCodecInternal *avci = s->internal;
1893     int i, j;
1894
1895     if (!avci->buffer)
1896         return;
1897
1898     if (avci->buffer_count)
1899         av_log(s, AV_LOG_WARNING, "Found %i unreleased buffers!\n",
1900                avci->buffer_count);
1901     for(i=0; i<INTERNAL_BUFFER_SIZE; i++){
1902         InternalBuffer *buf = &avci->buffer[i];
1903         for(j=0; j<4; j++){
1904             av_freep(&buf->base[j]);
1905             buf->data[j]= NULL;
1906         }
1907     }
1908     av_freep(&avci->buffer);
1909
1910     avci->buffer_count=0;
1911 }
1912
1913 static void audio_free_buffers(AVCodecContext *avctx)
1914 {
1915     AVCodecInternal *avci = avctx->internal;
1916     InternalBuffer *buf;
1917
1918     if (!avci->buffer)
1919         return;
1920     buf = avci->buffer;
1921
1922     if (buf->extended_data) {
1923         av_free(buf->extended_data[0]);
1924         if (buf->extended_data != buf->data)
1925             av_freep(&buf->extended_data);
1926     }
1927     av_freep(&avci->buffer);
1928 }
1929
1930 void avcodec_default_free_buffers(AVCodecContext *avctx)
1931 {
1932     switch (avctx->codec_type) {
1933     case AVMEDIA_TYPE_VIDEO:
1934         video_free_buffers(avctx);
1935         break;
1936     case AVMEDIA_TYPE_AUDIO:
1937         audio_free_buffers(avctx);
1938         break;
1939     default:
1940         break;
1941     }
1942 }
1943
1944 int av_get_exact_bits_per_sample(enum CodecID codec_id)
1945 {
1946     switch(codec_id){
1947     case CODEC_ID_ADPCM_CT:
1948     case CODEC_ID_ADPCM_IMA_APC:
1949     case CODEC_ID_ADPCM_IMA_EA_SEAD:
1950     case CODEC_ID_ADPCM_IMA_WS:
1951     case CODEC_ID_ADPCM_G722:
1952     case CODEC_ID_ADPCM_YAMAHA:
1953         return 4;
1954     case CODEC_ID_PCM_ALAW:
1955     case CODEC_ID_PCM_MULAW:
1956     case CODEC_ID_PCM_S8:
1957     case CODEC_ID_PCM_U8:
1958     case CODEC_ID_PCM_ZORK:
1959         return 8;
1960     case CODEC_ID_PCM_S16BE:
1961     case CODEC_ID_PCM_S16LE:
1962     case CODEC_ID_PCM_S16LE_PLANAR:
1963     case CODEC_ID_PCM_U16BE:
1964     case CODEC_ID_PCM_U16LE:
1965         return 16;
1966     case CODEC_ID_PCM_S24DAUD:
1967     case CODEC_ID_PCM_S24BE:
1968     case CODEC_ID_PCM_S24LE:
1969     case CODEC_ID_PCM_U24BE:
1970     case CODEC_ID_PCM_U24LE:
1971         return 24;
1972     case CODEC_ID_PCM_S32BE:
1973     case CODEC_ID_PCM_S32LE:
1974     case CODEC_ID_PCM_U32BE:
1975     case CODEC_ID_PCM_U32LE:
1976     case CODEC_ID_PCM_F32BE:
1977     case CODEC_ID_PCM_F32LE:
1978         return 32;
1979     case CODEC_ID_PCM_F64BE:
1980     case CODEC_ID_PCM_F64LE:
1981         return 64;
1982     default:
1983         return 0;
1984     }
1985 }
1986
1987 enum CodecID av_get_pcm_codec(enum AVSampleFormat fmt, int be)
1988 {
1989     static const enum CodecID map[AV_SAMPLE_FMT_NB][2] = {
1990         [AV_SAMPLE_FMT_U8  ] = { CODEC_ID_PCM_U8,    CODEC_ID_PCM_U8    },
1991         [AV_SAMPLE_FMT_S16 ] = { CODEC_ID_PCM_S16LE, CODEC_ID_PCM_S16BE },
1992         [AV_SAMPLE_FMT_S32 ] = { CODEC_ID_PCM_S32LE, CODEC_ID_PCM_S32BE },
1993         [AV_SAMPLE_FMT_FLT ] = { CODEC_ID_PCM_F32LE, CODEC_ID_PCM_F32BE },
1994         [AV_SAMPLE_FMT_DBL ] = { CODEC_ID_PCM_F64LE, CODEC_ID_PCM_F64BE },
1995         [AV_SAMPLE_FMT_U8P ] = { CODEC_ID_PCM_U8,    CODEC_ID_PCM_U8    },
1996         [AV_SAMPLE_FMT_S16P] = { CODEC_ID_PCM_S16LE, CODEC_ID_PCM_S16BE },
1997         [AV_SAMPLE_FMT_S32P] = { CODEC_ID_PCM_S32LE, CODEC_ID_PCM_S32BE },
1998         [AV_SAMPLE_FMT_FLTP] = { CODEC_ID_PCM_F32LE, CODEC_ID_PCM_F32BE },
1999         [AV_SAMPLE_FMT_DBLP] = { CODEC_ID_PCM_F64LE, CODEC_ID_PCM_F64BE },
2000     };
2001     if (fmt < 0 || fmt >= AV_SAMPLE_FMT_NB)
2002         return CODEC_ID_NONE;
2003     if (be < 0 || be > 1)
2004         be = AV_NE(1, 0);
2005     return map[fmt][be];
2006 }
2007
2008 int av_get_bits_per_sample(enum CodecID codec_id)
2009 {
2010     switch (codec_id) {
2011     case CODEC_ID_ADPCM_SBPRO_2:
2012         return 2;
2013     case CODEC_ID_ADPCM_SBPRO_3:
2014         return 3;
2015     case CODEC_ID_ADPCM_SBPRO_4:
2016     case CODEC_ID_ADPCM_IMA_WAV:
2017     case CODEC_ID_ADPCM_IMA_QT:
2018     case CODEC_ID_ADPCM_SWF:
2019     case CODEC_ID_ADPCM_MS:
2020         return 4;
2021     default:
2022         return av_get_exact_bits_per_sample(codec_id);
2023     }
2024 }
2025
2026 int av_get_audio_frame_duration(AVCodecContext *avctx, int frame_bytes)
2027 {
2028     int id, sr, ch, ba, tag, bps;
2029
2030     id  = avctx->codec_id;
2031     sr  = avctx->sample_rate;
2032     ch  = avctx->channels;
2033     ba  = avctx->block_align;
2034     tag = avctx->codec_tag;
2035     bps = av_get_exact_bits_per_sample(avctx->codec_id);
2036
2037     /* codecs with an exact constant bits per sample */
2038     if (bps > 0 && ch > 0 && frame_bytes > 0)
2039         return (frame_bytes * 8) / (bps * ch);
2040     bps = avctx->bits_per_coded_sample;
2041
2042     /* codecs with a fixed packet duration */
2043     switch (id) {
2044     case CODEC_ID_ADPCM_ADX:    return   32;
2045     case CODEC_ID_ADPCM_IMA_QT: return   64;
2046     case CODEC_ID_ADPCM_EA_XAS: return  128;
2047     case CODEC_ID_AMR_NB:
2048     case CODEC_ID_GSM:
2049     case CODEC_ID_QCELP:
2050     case CODEC_ID_RA_144:
2051     case CODEC_ID_RA_288:       return  160;
2052     case CODEC_ID_IMC:          return  256;
2053     case CODEC_ID_AMR_WB:
2054     case CODEC_ID_GSM_MS:       return  320;
2055     case CODEC_ID_MP1:          return  384;
2056     case CODEC_ID_ATRAC1:       return  512;
2057     case CODEC_ID_ATRAC3:       return 1024;
2058     case CODEC_ID_MP2:
2059     case CODEC_ID_MUSEPACK7:    return 1152;
2060     case CODEC_ID_AC3:          return 1536;
2061     }
2062
2063     if (sr > 0) {
2064         /* calc from sample rate */
2065         if (id == CODEC_ID_TTA)
2066             return 256 * sr / 245;
2067
2068         if (ch > 0) {
2069             /* calc from sample rate and channels */
2070             if (id == CODEC_ID_BINKAUDIO_DCT)
2071                 return (480 << (sr / 22050)) / ch;
2072         }
2073     }
2074
2075     if (ba > 0) {
2076         /* calc from block_align */
2077         if (id == CODEC_ID_SIPR) {
2078             switch (ba) {
2079             case 20: return 160;
2080             case 19: return 144;
2081             case 29: return 288;
2082             case 37: return 480;
2083             }
2084         }
2085     }
2086
2087     if (frame_bytes > 0) {
2088         /* calc from frame_bytes only */
2089         if (id == CODEC_ID_TRUESPEECH)
2090             return 240 * (frame_bytes / 32);
2091         if (id == CODEC_ID_NELLYMOSER)
2092             return 256 * (frame_bytes / 64);
2093
2094         if (bps > 0) {
2095             /* calc from frame_bytes and bits_per_coded_sample */
2096             if (id == CODEC_ID_ADPCM_G726)
2097                 return frame_bytes * 8 / bps;
2098         }
2099
2100         if (ch > 0) {
2101             /* calc from frame_bytes and channels */
2102             switch (id) {
2103             case CODEC_ID_ADPCM_4XM:
2104             case CODEC_ID_ADPCM_IMA_ISS:
2105                 return (frame_bytes - 4 * ch) * 2 / ch;
2106             case CODEC_ID_ADPCM_IMA_SMJPEG:
2107                 return (frame_bytes - 4) * 2 / ch;
2108             case CODEC_ID_ADPCM_IMA_AMV:
2109                 return (frame_bytes - 8) * 2 / ch;
2110             case CODEC_ID_ADPCM_XA:
2111                 return (frame_bytes / 128) * 224 / ch;
2112             case CODEC_ID_INTERPLAY_DPCM:
2113                 return (frame_bytes - 6 - ch) / ch;
2114             case CODEC_ID_ROQ_DPCM:
2115                 return (frame_bytes - 8) / ch;
2116             case CODEC_ID_XAN_DPCM:
2117                 return (frame_bytes - 2 * ch) / ch;
2118             case CODEC_ID_MACE3:
2119                 return 3 * frame_bytes / ch;
2120             case CODEC_ID_MACE6:
2121                 return 6 * frame_bytes / ch;
2122             case CODEC_ID_PCM_LXF:
2123                 return 2 * (frame_bytes / (5 * ch));
2124             }
2125
2126             if (tag) {
2127                 /* calc from frame_bytes, channels, and codec_tag */
2128                 if (id == CODEC_ID_SOL_DPCM) {
2129                     if (tag == 3)
2130                         return frame_bytes / ch;
2131                     else
2132                         return frame_bytes * 2 / ch;
2133                 }
2134             }
2135
2136             if (ba > 0) {
2137                 /* calc from frame_bytes, channels, and block_align */
2138                 int blocks = frame_bytes / ba;
2139                 switch (avctx->codec_id) {
2140                 case CODEC_ID_ADPCM_IMA_WAV:
2141                     return blocks * (1 + (ba - 4 * ch) / (4 * ch) * 8);
2142                 case CODEC_ID_ADPCM_IMA_DK3:
2143                     return blocks * (((ba - 16) * 2 / 3 * 4) / ch);
2144                 case CODEC_ID_ADPCM_IMA_DK4:
2145                     return blocks * (1 + (ba - 4 * ch) * 2 / ch);
2146                 case CODEC_ID_ADPCM_MS:
2147                     return blocks * (2 + (ba - 7 * ch) * 2 / ch);
2148                 }
2149             }
2150
2151             if (bps > 0) {
2152                 /* calc from frame_bytes, channels, and bits_per_coded_sample */
2153                 switch (avctx->codec_id) {
2154                 case CODEC_ID_PCM_DVD:
2155                     return 2 * (frame_bytes / ((bps * 2 / 8) * ch));
2156                 case CODEC_ID_PCM_BLURAY:
2157                     return frame_bytes / ((FFALIGN(ch, 2) * bps) / 8);
2158                 case CODEC_ID_S302M:
2159                     return 2 * (frame_bytes / ((bps + 4) / 4)) / ch;
2160                 }
2161             }
2162         }
2163     }
2164
2165     return 0;
2166 }
2167
2168 #if !HAVE_THREADS
2169 int ff_thread_init(AVCodecContext *s){
2170     return -1;
2171 }
2172 #endif
2173
2174 unsigned int av_xiphlacing(unsigned char *s, unsigned int v)
2175 {
2176     unsigned int n = 0;
2177
2178     while(v >= 0xff) {
2179         *s++ = 0xff;
2180         v -= 0xff;
2181         n++;
2182     }
2183     *s = v;
2184     n++;
2185     return n;
2186 }
2187
2188 int ff_match_2uint16(const uint16_t (*tab)[2], int size, int a, int b){
2189     int i;
2190     for(i=0; i<size && !(tab[i][0]==a && tab[i][1]==b); i++);
2191     return i;
2192 }
2193
2194 void av_log_missing_feature(void *avc, const char *feature, int want_sample)
2195 {
2196     av_log(avc, AV_LOG_WARNING, "%s not implemented. Update your FFmpeg "
2197             "version to the newest one from Git. If the problem still "
2198             "occurs, it means that your file has a feature which has not "
2199             "been implemented.\n", feature);
2200     if(want_sample)
2201         av_log_ask_for_sample(avc, NULL);
2202 }
2203
2204 void av_log_ask_for_sample(void *avc, const char *msg, ...)
2205 {
2206     va_list argument_list;
2207
2208     va_start(argument_list, msg);
2209
2210     if (msg)
2211         av_vlog(avc, AV_LOG_WARNING, msg, argument_list);
2212     av_log(avc, AV_LOG_WARNING, "If you want to help, upload a sample "
2213             "of this file to ftp://upload.ffmpeg.org/MPlayer/incoming/ "
2214             "and contact the ffmpeg-devel mailing list.\n");
2215
2216     va_end(argument_list);
2217 }
2218
2219 static AVHWAccel *first_hwaccel = NULL;
2220
2221 void av_register_hwaccel(AVHWAccel *hwaccel)
2222 {
2223     AVHWAccel **p = &first_hwaccel;
2224     while (*p)
2225         p = &(*p)->next;
2226     *p = hwaccel;
2227     hwaccel->next = NULL;
2228 }
2229
2230 AVHWAccel *av_hwaccel_next(AVHWAccel *hwaccel)
2231 {
2232     return hwaccel ? hwaccel->next : first_hwaccel;
2233 }
2234
2235 AVHWAccel *ff_find_hwaccel(enum CodecID codec_id, enum PixelFormat pix_fmt)
2236 {
2237     AVHWAccel *hwaccel=NULL;
2238
2239     while((hwaccel= av_hwaccel_next(hwaccel))){
2240         if (   hwaccel->id      == codec_id
2241             && hwaccel->pix_fmt == pix_fmt)
2242             return hwaccel;
2243     }
2244     return NULL;
2245 }
2246
2247 int av_lockmgr_register(int (*cb)(void **mutex, enum AVLockOp op))
2248 {
2249     if (ff_lockmgr_cb) {
2250         if (ff_lockmgr_cb(&codec_mutex, AV_LOCK_DESTROY))
2251             return -1;
2252         if (ff_lockmgr_cb(&avformat_mutex, AV_LOCK_DESTROY))
2253             return -1;
2254     }
2255
2256     ff_lockmgr_cb = cb;
2257
2258     if (ff_lockmgr_cb) {
2259         if (ff_lockmgr_cb(&codec_mutex, AV_LOCK_CREATE))
2260             return -1;
2261         if (ff_lockmgr_cb(&avformat_mutex, AV_LOCK_CREATE))
2262             return -1;
2263     }
2264     return 0;
2265 }
2266
2267 int avpriv_lock_avformat(void)
2268 {
2269     if (ff_lockmgr_cb) {
2270         if ((*ff_lockmgr_cb)(&avformat_mutex, AV_LOCK_OBTAIN))
2271             return -1;
2272     }
2273     return 0;
2274 }
2275
2276 int avpriv_unlock_avformat(void)
2277 {
2278     if (ff_lockmgr_cb) {
2279         if ((*ff_lockmgr_cb)(&avformat_mutex, AV_LOCK_RELEASE))
2280             return -1;
2281     }
2282     return 0;
2283 }
2284
2285 unsigned int avpriv_toupper4(unsigned int x)
2286 {
2287     return     toupper( x     &0xFF)
2288             + (toupper((x>>8 )&0xFF)<<8 )
2289             + (toupper((x>>16)&0xFF)<<16)
2290             + (toupper((x>>24)&0xFF)<<24);
2291 }
2292
2293 #if !HAVE_THREADS
2294
2295 int ff_thread_get_buffer(AVCodecContext *avctx, AVFrame *f)
2296 {
2297     f->owner = avctx;
2298
2299     ff_init_buffer_info(avctx, f);
2300
2301     return avctx->get_buffer(avctx, f);
2302 }
2303
2304 void ff_thread_release_buffer(AVCodecContext *avctx, AVFrame *f)
2305 {
2306     f->owner->release_buffer(f->owner, f);
2307 }
2308
2309 void ff_thread_finish_setup(AVCodecContext *avctx)
2310 {
2311 }
2312
2313 void ff_thread_report_progress(AVFrame *f, int progress, int field)
2314 {
2315 }
2316
2317 void ff_thread_await_progress(AVFrame *f, int progress, int field)
2318 {
2319 }
2320
2321 int ff_thread_can_start_frame(AVCodecContext *avctx)
2322 {
2323     return 1;
2324 }
2325
2326 #endif
2327
2328 enum AVMediaType avcodec_get_type(enum CodecID codec_id)
2329 {
2330     AVCodec *c= avcodec_find_decoder(codec_id);
2331     if(!c)
2332         c= avcodec_find_encoder(codec_id);
2333     if(c)
2334         return c->type;
2335
2336     if (codec_id <= CODEC_ID_NONE)
2337         return AVMEDIA_TYPE_UNKNOWN;
2338     else if (codec_id < CODEC_ID_FIRST_AUDIO)
2339         return AVMEDIA_TYPE_VIDEO;
2340     else if (codec_id < CODEC_ID_FIRST_SUBTITLE)
2341         return AVMEDIA_TYPE_AUDIO;
2342     else if (codec_id < CODEC_ID_FIRST_UNKNOWN)
2343         return AVMEDIA_TYPE_SUBTITLE;
2344
2345     return AVMEDIA_TYPE_UNKNOWN;
2346 }
2347
2348 int avcodec_is_open(AVCodecContext *s)
2349 {
2350     return !!s->internal;
2351 }