f4a81ce4e96b6ecac50b0c4a7cd014812d97b9b0
[libav.git] / libavcodec / utils.c
1 /*
2 * utils for libavcodec
3 * Copyright (c) 2001 Fabrice Bellard.
4 *
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2 of the License, or (at your option) any later version.
9 *
10 * This library is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
14 *
15 * You should have received a copy of the GNU Lesser General Public
16 * License along with this library; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18 */
19 #include "avcodec.h"
20 #include "dsputil.h"
21 #include "mpegvideo.h"
22
23 void *av_mallocz(unsigned int size)
24 {
25 void *ptr;
26
27 if(size == 0) fprintf(stderr, "Warning, allocating 0 bytes\n");
28
29 ptr = av_malloc(size);
30 if (!ptr)
31 return NULL;
32 memset(ptr, 0, size);
33 return ptr;
34 }
35
36 /* allocation of static arrays - do not use for normal allocation */
37 static unsigned int last_static = 0;
38 static char*** array_static = NULL;
39 static const unsigned int grow_static = 64; // ^2
40 void *__av_mallocz_static(void** location, unsigned int size)
41 {
42 int l = (last_static + grow_static) & ~(grow_static - 1);
43 void *ptr = av_mallocz(size);
44 if (!ptr)
45 return NULL;
46
47 if (location)
48 {
49 if (l > last_static)
50 array_static = realloc(array_static, l);
51 array_static[last_static++] = (char**) location;
52 *location = ptr;
53 }
54 return ptr;
55 }
56 /* free all static arrays and reset pointers to 0 */
57 void av_free_static()
58 {
59 if (array_static)
60 {
61 unsigned i;
62 for (i = 0; i < last_static; i++)
63 {
64 free(*array_static[i]);
65 *array_static[i] = NULL;
66 }
67 free(array_static);
68 array_static = 0;
69 }
70 last_static = 0;
71 }
72
73 /* cannot call it directly because of 'void **' casting is not automatic */
74 void __av_freep(void **ptr)
75 {
76 av_free(*ptr);
77 *ptr = NULL;
78 }
79
80 /* encoder management */
81 AVCodec *first_avcodec;
82
83 void register_avcodec(AVCodec *format)
84 {
85 AVCodec **p;
86 p = &first_avcodec;
87 while (*p != NULL) p = &(*p)->next;
88 *p = format;
89 format->next = NULL;
90 }
91
92 typedef struct DefaultPicOpaque{
93 int last_pic_num;
94 uint8_t *data[4];
95 }DefaultPicOpaque;
96
97 int avcodec_default_get_buffer(AVCodecContext *s, AVFrame *pic){
98 int i;
99 const int width = s->width;
100 const int height= s->height;
101 DefaultPicOpaque *opaque;
102
103 assert(pic->data[0]==NULL);
104 assert(pic->type==0 || pic->type==FF_TYPE_INTERNAL);
105
106 if(pic->opaque){
107 opaque= (DefaultPicOpaque *)pic->opaque;
108 for(i=0; i<3; i++)
109 pic->data[i]= opaque->data[i];
110
111 // printf("get_buffer %X coded_pic_num:%d last:%d\n", pic->opaque, pic->coded_picture_number, opaque->last_pic_num);
112 pic->age= pic->coded_picture_number - opaque->last_pic_num;
113 opaque->last_pic_num= pic->coded_picture_number;
114 //printf("age: %d %d %d\n", pic->age, c->picture_number, pic->coded_picture_number);
115 }else{
116 int align, h_chroma_shift, v_chroma_shift;
117 int w, h, pixel_size;
118
119 avcodec_get_chroma_sub_sample(s->pix_fmt, &h_chroma_shift, &v_chroma_shift);
120
121 switch(s->pix_fmt){
122 case PIX_FMT_YUV422:
123 pixel_size=2;
124 break;
125 case PIX_FMT_RGB24:
126 case PIX_FMT_BGR24:
127 pixel_size=3;
128 break;
129 case PIX_FMT_RGBA32:
130 pixel_size=4;
131 break;
132 default:
133 pixel_size=1;
134 }
135
136 if(s->codec_id==CODEC_ID_SVQ1) align=63;
137 else align=15;
138
139 w= (width +align)&~align;
140 h= (height+align)&~align;
141
142 if(!(s->flags&CODEC_FLAG_EMU_EDGE)){
143 w+= EDGE_WIDTH*2;
144 h+= EDGE_WIDTH*2;
145 }
146
147 opaque= av_mallocz(sizeof(DefaultPicOpaque));
148 if(opaque==NULL) return -1;
149
150 pic->opaque= opaque;
151 opaque->last_pic_num= -256*256*256*64;
152
153 for(i=0; i<3; i++){
154 int h_shift= i==0 ? 0 : h_chroma_shift;
155 int v_shift= i==0 ? 0 : v_chroma_shift;
156
157 pic->linesize[i]= pixel_size*w>>h_shift;
158
159 pic->base[i]= av_mallocz((pic->linesize[i]*h>>v_shift)+16); //FIXME 16
160 if(pic->base[i]==NULL) return -1;
161
162 memset(pic->base[i], 128, pic->linesize[i]*h>>v_shift);
163
164 if(s->flags&CODEC_FLAG_EMU_EDGE)
165 pic->data[i] = pic->base[i] + 16; //FIXME 16
166 else
167 pic->data[i] = pic->base[i] + (pic->linesize[i]*EDGE_WIDTH>>v_shift) + (EDGE_WIDTH>>h_shift) + 16; //FIXME 16
168
169 opaque->data[i]= pic->data[i];
170 }
171 pic->age= 256*256*256*64;
172 pic->type= FF_BUFFER_TYPE_INTERNAL;
173 }
174
175 return 0;
176 }
177
178 void avcodec_default_release_buffer(AVCodecContext *s, AVFrame *pic){
179 int i;
180
181 assert(pic->type==FF_BUFFER_TYPE_INTERNAL);
182
183 for(i=0; i<3; i++)
184 pic->data[i]=NULL;
185 //printf("R%X\n", pic->opaque);
186 }
187
188 void avcodec_get_context_defaults(AVCodecContext *s){
189 s->bit_rate= 800*1000;
190 s->bit_rate_tolerance= s->bit_rate*10;
191 s->qmin= 2;
192 s->qmax= 31;
193 s->mb_qmin= 2;
194 s->mb_qmax= 31;
195 s->rc_eq= "tex^qComp";
196 s->qcompress= 0.5;
197 s->max_qdiff= 3;
198 s->b_quant_factor=1.25;
199 s->b_quant_offset=1.25;
200 s->i_quant_factor=-0.8;
201 s->i_quant_offset=0.0;
202 s->error_concealment= 3;
203 s->error_resilience= 1;
204 s->workaround_bugs= FF_BUG_AUTODETECT;
205 s->frame_rate = 25 * FRAME_RATE_BASE;
206 s->gop_size= 50;
207 s->me_method= ME_EPZS;
208 s->get_buffer= avcodec_default_get_buffer;
209 s->release_buffer= avcodec_default_release_buffer;
210 s->me_subpel_quality=8;
211 }
212
213 /**
214 * allocates a AVCodecContext and set it to defaults.
215 * this can be deallocated by simply calling free()
216 */
217 AVCodecContext *avcodec_alloc_context(void){
218 AVCodecContext *avctx= av_mallocz(sizeof(AVCodecContext));
219
220 if(avctx==NULL) return NULL;
221
222 avcodec_get_context_defaults(avctx);
223
224 return avctx;
225 }
226
227 /**
228 * allocates a AVPFrame and set it to defaults.
229 * this can be deallocated by simply calling free()
230 */
231 AVFrame *avcodec_alloc_frame(void){
232 AVFrame *pic= av_mallocz(sizeof(AVFrame));
233
234 return pic;
235 }
236
237 int avcodec_open(AVCodecContext *avctx, AVCodec *codec)
238 {
239 int ret;
240
241 avctx->codec = codec;
242 avctx->codec_id = codec->id;
243 avctx->frame_number = 0;
244 if (codec->priv_data_size > 0) {
245 avctx->priv_data = av_mallocz(codec->priv_data_size);
246 if (!avctx->priv_data)
247 return -ENOMEM;
248 } else {
249 avctx->priv_data = NULL;
250 }
251 ret = avctx->codec->init(avctx);
252 if (ret < 0) {
253 av_freep(&avctx->priv_data);
254 return ret;
255 }
256 return 0;
257 }
258
259 int avcodec_encode_audio(AVCodecContext *avctx, UINT8 *buf, int buf_size,
260 const short *samples)
261 {
262 int ret;
263
264 ret = avctx->codec->encode(avctx, buf, buf_size, (void *)samples);
265 avctx->frame_number++;
266 return ret;
267 }
268
269 int avcodec_encode_video(AVCodecContext *avctx, UINT8 *buf, int buf_size,
270 const AVFrame *pict)
271 {
272 int ret;
273
274 ret = avctx->codec->encode(avctx, buf, buf_size, (void *)pict);
275
276 emms_c(); //needed to avoid a emms_c() call before every return;
277
278 avctx->frame_number++;
279 return ret;
280 }
281
282 /* decode a frame. return -1 if error, otherwise return the number of
283 bytes used. If no frame could be decompressed, *got_picture_ptr is
284 zero. Otherwise, it is non zero */
285 int avcodec_decode_video(AVCodecContext *avctx, AVFrame *picture,
286 int *got_picture_ptr,
287 UINT8 *buf, int buf_size)
288 {
289 int ret;
290
291 ret = avctx->codec->decode(avctx, picture, got_picture_ptr,
292 buf, buf_size);
293
294 emms_c(); //needed to avoid a emms_c() call before every return;
295
296 if (*got_picture_ptr)
297 avctx->frame_number++;
298 return ret;
299 }
300
301 /* decode an audio frame. return -1 if error, otherwise return the
302 *number of bytes used. If no frame could be decompressed,
303 *frame_size_ptr is zero. Otherwise, it is the decompressed frame
304 *size in BYTES. */
305 int avcodec_decode_audio(AVCodecContext *avctx, INT16 *samples,
306 int *frame_size_ptr,
307 UINT8 *buf, int buf_size)
308 {
309 int ret;
310
311 ret = avctx->codec->decode(avctx, samples, frame_size_ptr,
312 buf, buf_size);
313 avctx->frame_number++;
314 return ret;
315 }
316
317 int avcodec_close(AVCodecContext *avctx)
318 {
319 if (avctx->codec->close)
320 avctx->codec->close(avctx);
321 av_freep(&avctx->priv_data);
322 avctx->codec = NULL;
323 return 0;
324 }
325
326 AVCodec *avcodec_find_encoder(enum CodecID id)
327 {
328 AVCodec *p;
329 p = first_avcodec;
330 while (p) {
331 if (p->encode != NULL && p->id == id)
332 return p;
333 p = p->next;
334 }
335 return NULL;
336 }
337
338 AVCodec *avcodec_find_encoder_by_name(const char *name)
339 {
340 AVCodec *p;
341 p = first_avcodec;
342 while (p) {
343 if (p->encode != NULL && strcmp(name,p->name) == 0)
344 return p;
345 p = p->next;
346 }
347 return NULL;
348 }
349
350 AVCodec *avcodec_find_decoder(enum CodecID id)
351 {
352 AVCodec *p;
353 p = first_avcodec;
354 while (p) {
355 if (p->decode != NULL && p->id == id)
356 return p;
357 p = p->next;
358 }
359 return NULL;
360 }
361
362 AVCodec *avcodec_find_decoder_by_name(const char *name)
363 {
364 AVCodec *p;
365 p = first_avcodec;
366 while (p) {
367 if (p->decode != NULL && strcmp(name,p->name) == 0)
368 return p;
369 p = p->next;
370 }
371 return NULL;
372 }
373
374 AVCodec *avcodec_find(enum CodecID id)
375 {
376 AVCodec *p;
377 p = first_avcodec;
378 while (p) {
379 if (p->id == id)
380 return p;
381 p = p->next;
382 }
383 return NULL;
384 }
385
386 const char *pix_fmt_str[] = {
387 "yuv420p",
388 "yuv422",
389 "rgb24",
390 "bgr24",
391 "yuv422p",
392 "yuv444p",
393 "rgba32",
394 "bgra32",
395 "yuv410p",
396 "yuv411p",
397 };
398
399 void avcodec_string(char *buf, int buf_size, AVCodecContext *enc, int encode)
400 {
401 const char *codec_name;
402 AVCodec *p;
403 char buf1[32];
404 char channels_str[100];
405 int bitrate;
406
407 if (encode)
408 p = avcodec_find_encoder(enc->codec_id);
409 else
410 p = avcodec_find_decoder(enc->codec_id);
411
412 if (p) {
413 codec_name = p->name;
414 } else if (enc->codec_name[0] != '\0') {
415 codec_name = enc->codec_name;
416 } else {
417 /* output avi tags */
418 if (enc->codec_type == CODEC_TYPE_VIDEO) {
419 snprintf(buf1, sizeof(buf1), "%c%c%c%c",
420 enc->codec_tag & 0xff,
421 (enc->codec_tag >> 8) & 0xff,
422 (enc->codec_tag >> 16) & 0xff,
423 (enc->codec_tag >> 24) & 0xff);
424 } else {
425 snprintf(buf1, sizeof(buf1), "0x%04x", enc->codec_tag);
426 }
427 codec_name = buf1;
428 }
429
430 switch(enc->codec_type) {
431 case CODEC_TYPE_VIDEO:
432 snprintf(buf, buf_size,
433 "Video: %s%s",
434 codec_name, enc->flags & CODEC_FLAG_HQ ? " (hq)" : "");
435 if (enc->codec_id == CODEC_ID_RAWVIDEO) {
436 snprintf(buf + strlen(buf), buf_size - strlen(buf),
437 ", %s",
438 avcodec_get_pix_fmt_name(enc->pix_fmt));
439 }
440 if (enc->width) {
441 snprintf(buf + strlen(buf), buf_size - strlen(buf),
442 ", %dx%d, %0.2f fps",
443 enc->width, enc->height,
444 (float)enc->frame_rate / FRAME_RATE_BASE);
445 }
446 if (encode) {
447 snprintf(buf + strlen(buf), buf_size - strlen(buf),
448 ", q=%d-%d", enc->qmin, enc->qmax);
449 }
450 bitrate = enc->bit_rate;
451 break;
452 case CODEC_TYPE_AUDIO:
453 snprintf(buf, buf_size,
454 "Audio: %s",
455 codec_name);
456 switch (enc->channels) {
457 case 1:
458 strcpy(channels_str, "mono");
459 break;
460 case 2:
461 strcpy(channels_str, "stereo");
462 break;
463 case 6:
464 strcpy(channels_str, "5:1");
465 break;
466 default:
467 sprintf(channels_str, "%d channels", enc->channels);
468 break;
469 }
470 if (enc->sample_rate) {
471 snprintf(buf + strlen(buf), buf_size - strlen(buf),
472 ", %d Hz, %s",
473 enc->sample_rate,
474 channels_str);
475 }
476
477 /* for PCM codecs, compute bitrate directly */
478 switch(enc->codec_id) {
479 case CODEC_ID_PCM_S16LE:
480 case CODEC_ID_PCM_S16BE:
481 case CODEC_ID_PCM_U16LE:
482 case CODEC_ID_PCM_U16BE:
483 bitrate = enc->sample_rate * enc->channels * 16;
484 break;
485 case CODEC_ID_PCM_S8:
486 case CODEC_ID_PCM_U8:
487 case CODEC_ID_PCM_ALAW:
488 case CODEC_ID_PCM_MULAW:
489 bitrate = enc->sample_rate * enc->channels * 8;
490 break;
491 default:
492 bitrate = enc->bit_rate;
493 break;
494 }
495 break;
496 default:
497 av_abort();
498 }
499 if (encode) {
500 if (enc->flags & CODEC_FLAG_PASS1)
501 snprintf(buf + strlen(buf), buf_size - strlen(buf),
502 ", pass 1");
503 if (enc->flags & CODEC_FLAG_PASS2)
504 snprintf(buf + strlen(buf), buf_size - strlen(buf),
505 ", pass 2");
506 }
507 if (bitrate != 0) {
508 snprintf(buf + strlen(buf), buf_size - strlen(buf),
509 ", %d kb/s", bitrate / 1000);
510 }
511 }
512
513 /* Picture field are filled with 'ptr' addresses */
514 void avpicture_fill(AVPicture *picture, UINT8 *ptr,
515 int pix_fmt, int width, int height)
516 {
517 int size;
518
519 size = width * height;
520 switch(pix_fmt) {
521 case PIX_FMT_YUV420P:
522 picture->data[0] = ptr;
523 picture->data[1] = picture->data[0] + size;
524 picture->data[2] = picture->data[1] + size / 4;
525 picture->linesize[0] = width;
526 picture->linesize[1] = width / 2;
527 picture->linesize[2] = width / 2;
528 break;
529 case PIX_FMT_YUV422P:
530 picture->data[0] = ptr;
531 picture->data[1] = picture->data[0] + size;
532 picture->data[2] = picture->data[1] + size / 2;
533 picture->linesize[0] = width;
534 picture->linesize[1] = width / 2;
535 picture->linesize[2] = width / 2;
536 break;
537 case PIX_FMT_YUV444P:
538 picture->data[0] = ptr;
539 picture->data[1] = picture->data[0] + size;
540 picture->data[2] = picture->data[1] + size;
541 picture->linesize[0] = width;
542 picture->linesize[1] = width;
543 picture->linesize[2] = width;
544 break;
545 case PIX_FMT_RGB24:
546 case PIX_FMT_BGR24:
547 picture->data[0] = ptr;
548 picture->data[1] = NULL;
549 picture->data[2] = NULL;
550 picture->linesize[0] = width * 3;
551 break;
552 case PIX_FMT_RGBA32:
553 picture->data[0] = ptr;
554 picture->data[1] = NULL;
555 picture->data[2] = NULL;
556 picture->linesize[0] = width * 4;
557 break;
558 case PIX_FMT_YUV422:
559 picture->data[0] = ptr;
560 picture->data[1] = NULL;
561 picture->data[2] = NULL;
562 picture->linesize[0] = width * 2;
563 break;
564 default:
565 picture->data[0] = NULL;
566 picture->data[1] = NULL;
567 picture->data[2] = NULL;
568 break;
569 }
570 }
571
572 int avpicture_get_size(int pix_fmt, int width, int height)
573 {
574 int size;
575
576 size = width * height;
577 switch(pix_fmt) {
578 case PIX_FMT_YUV420P:
579 size = (size * 3) / 2;
580 break;
581 case PIX_FMT_YUV422P:
582 size = (size * 2);
583 break;
584 case PIX_FMT_YUV444P:
585 size = (size * 3);
586 break;
587 case PIX_FMT_RGB24:
588 case PIX_FMT_BGR24:
589 size = (size * 3);
590 break;
591 case PIX_FMT_RGBA32:
592 size = (size * 4);
593 break;
594 case PIX_FMT_YUV422:
595 size = (size * 2);
596 break;
597 default:
598 size = -1;
599 break;
600 }
601 return size;
602 }
603
604 unsigned avcodec_version( void )
605 {
606 return LIBAVCODEC_VERSION_INT;
607 }
608
609 unsigned avcodec_build( void )
610 {
611 return LIBAVCODEC_BUILD;
612 }
613
614 /* must be called before any other functions */
615 void avcodec_init(void)
616 {
617 static int inited = 0;
618
619 if (inited != 0)
620 return;
621 inited = 1;
622
623 //dsputil_init();
624 }
625
626 /* this can be called after seeking and before trying to decode the next keyframe */
627 void avcodec_flush_buffers(AVCodecContext *avctx)
628 {
629 int i;
630 MpegEncContext *s = avctx->priv_data;
631
632 switch(avctx->codec_id){
633 case CODEC_ID_MPEG1VIDEO:
634 case CODEC_ID_H263:
635 case CODEC_ID_RV10:
636 case CODEC_ID_MJPEG:
637 case CODEC_ID_MJPEGB:
638 case CODEC_ID_MPEG4:
639 case CODEC_ID_MSMPEG4V1:
640 case CODEC_ID_MSMPEG4V2:
641 case CODEC_ID_MSMPEG4V3:
642 case CODEC_ID_WMV1:
643 case CODEC_ID_WMV2:
644 case CODEC_ID_H263P:
645 case CODEC_ID_H263I:
646 case CODEC_ID_SVQ1:
647 for(i=0; i<MAX_PICTURE_COUNT; i++){
648 if(s->picture[i].data[0] && ( s->picture[i].type == FF_BUFFER_TYPE_INTERNAL
649 || s->picture[i].type == FF_BUFFER_TYPE_USER))
650 avctx->release_buffer(avctx, (AVFrame*)&s->picture[i]);
651 }
652 s->last_picture.data[0] = s->next_picture.data[0] = NULL;
653 break;
654 default:
655 //FIXME
656 break;
657 }
658 }
659
660 static int raw_encode_init(AVCodecContext *s)
661 {
662 return 0;
663 }
664
665 static int raw_decode_frame(AVCodecContext *avctx,
666 void *data, int *data_size,
667 UINT8 *buf, int buf_size)
668 {
669 return -1;
670 }
671
672 static int raw_encode_frame(AVCodecContext *avctx,
673 unsigned char *frame, int buf_size, void *data)
674 {
675 return -1;
676 }
677
678 AVCodec rawvideo_codec = {
679 "rawvideo",
680 CODEC_TYPE_VIDEO,
681 CODEC_ID_RAWVIDEO,
682 0,
683 raw_encode_init,
684 raw_encode_frame,
685 NULL,
686 raw_decode_frame,
687 };