use av_realloc()
[libav.git] / libavcodec / utils.c
CommitLineData
de6d9b64
FB
1/*
2 * utils for libavcodec
ff4ec49e 3 * Copyright (c) 2001 Fabrice Bellard.
de6d9b64 4 *
ff4ec49e
FB
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.
de6d9b64 9 *
ff4ec49e 10 * This library is distributed in the hope that it will be useful,
de6d9b64 11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
ff4ec49e
FB
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
de6d9b64 14 *
ff4ec49e
FB
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
de6d9b64 18 */
de6d9b64 19#include "avcodec.h"
3123dd79 20#include "dsputil.h"
1c2a8c7f 21#include "mpegvideo.h"
de6d9b64 22
18f77016 23void *av_mallocz(unsigned int size)
3123dd79
FB
24{
25 void *ptr;
5b20b732
MN
26
27 if(size == 0) fprintf(stderr, "Warning, allocating 0 bytes\n");
28
3123dd79
FB
29 ptr = av_malloc(size);
30 if (!ptr)
31 return NULL;
32 memset(ptr, 0, size);
33 return ptr;
34}
35
855ea723
ZK
36/* allocation of static arrays - do not use for normal allocation */
37static unsigned int last_static = 0;
38static char*** array_static = NULL;
39static const unsigned int grow_static = 64; // ^2
40void *__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 */
57void 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
4d7a0a05
FB
73/* cannot call it directly because of 'void **' casting is not automatic */
74void __av_freep(void **ptr)
75{
76 av_free(*ptr);
77 *ptr = NULL;
78}
79
de6d9b64
FB
80/* encoder management */
81AVCodec *first_avcodec;
82
83void 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
1e491e29
MN
92typedef struct DefaultPicOpaque{
93 int last_pic_num;
94 uint8_t *data[4];
95}DefaultPicOpaque;
96
492cd3a9 97int avcodec_default_get_buffer(AVCodecContext *s, AVFrame *pic){
1e491e29
MN
98 int i;
99 const int width = s->width;
100 const int height= s->height;
101 DefaultPicOpaque *opaque;
4e00e76b
MN
102
103 assert(pic->data[0]==NULL);
104 assert(pic->type==0 || pic->type==FF_TYPE_INTERNAL);
1e491e29
MN
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;
1e491e29
MN
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)
4e00e76b 165 pic->data[i] = pic->base[i] + 16; //FIXME 16
1e491e29 166 else
4e00e76b 167 pic->data[i] = pic->base[i] + (pic->linesize[i]*EDGE_WIDTH>>v_shift) + (EDGE_WIDTH>>h_shift) + 16; //FIXME 16
1e491e29
MN
168
169 opaque->data[i]= pic->data[i];
170 }
171 pic->age= 256*256*256*64;
4e00e76b 172 pic->type= FF_BUFFER_TYPE_INTERNAL;
1e491e29
MN
173 }
174
175 return 0;
176}
177
492cd3a9 178void avcodec_default_release_buffer(AVCodecContext *s, AVFrame *pic){
1e491e29
MN
179 int i;
180
4e00e76b
MN
181 assert(pic->type==FF_BUFFER_TYPE_INTERNAL);
182
1e491e29
MN
183 for(i=0; i<3; i++)
184 pic->data[i]=NULL;
185//printf("R%X\n", pic->opaque);
186}
187
a33c7159
MN
188enum PixelFormat avcodec_default_get_format(struct AVCodecContext *s, enum PixelFormat * fmt){
189 return fmt[0];
190}
191
a949d72e 192void avcodec_get_context_defaults(AVCodecContext *s){
e8b62df6
MN
193 s->bit_rate= 800*1000;
194 s->bit_rate_tolerance= s->bit_rate*10;
a949d72e
MN
195 s->qmin= 2;
196 s->qmax= 31;
17a70fde
MN
197 s->mb_qmin= 2;
198 s->mb_qmax= 31;
a949d72e
MN
199 s->rc_eq= "tex^qComp";
200 s->qcompress= 0.5;
e8b62df6
MN
201 s->max_qdiff= 3;
202 s->b_quant_factor=1.25;
203 s->b_quant_offset=1.25;
b3a391e8 204 s->i_quant_factor=-0.8;
e8b62df6 205 s->i_quant_offset=0.0;
4d2858de 206 s->error_concealment= 3;
8d0e42ca 207 s->error_resilience= 1;
4d2858de 208 s->workaround_bugs= FF_BUG_AUTODETECT;
8d0e42ca
MN
209 s->frame_rate = 25 * FRAME_RATE_BASE;
210 s->gop_size= 50;
211 s->me_method= ME_EPZS;
1e491e29
MN
212 s->get_buffer= avcodec_default_get_buffer;
213 s->release_buffer= avcodec_default_release_buffer;
a33c7159 214 s->get_format= avcodec_default_get_format;
826f429a 215 s->me_subpel_quality=8;
a949d72e
MN
216}
217
218/**
219 * allocates a AVCodecContext and set it to defaults.
220 * this can be deallocated by simply calling free()
221 */
7ffbb60e 222AVCodecContext *avcodec_alloc_context(void){
a949d72e
MN
223 AVCodecContext *avctx= av_mallocz(sizeof(AVCodecContext));
224
225 if(avctx==NULL) return NULL;
226
227 avcodec_get_context_defaults(avctx);
228
229 return avctx;
230}
231
1e491e29 232/**
492cd3a9 233 * allocates a AVPFrame and set it to defaults.
1e491e29
MN
234 * this can be deallocated by simply calling free()
235 */
492cd3a9
MN
236AVFrame *avcodec_alloc_frame(void){
237 AVFrame *pic= av_mallocz(sizeof(AVFrame));
1e491e29
MN
238
239 return pic;
240}
241
de6d9b64
FB
242int avcodec_open(AVCodecContext *avctx, AVCodec *codec)
243{
244 int ret;
245
246 avctx->codec = codec;
4cb3ca72 247 avctx->codec_id = codec->id;
de6d9b64 248 avctx->frame_number = 0;
0edf8a7a
PG
249 if (codec->priv_data_size > 0) {
250 avctx->priv_data = av_mallocz(codec->priv_data_size);
251 if (!avctx->priv_data)
252 return -ENOMEM;
253 } else {
254 avctx->priv_data = NULL;
255 }
de6d9b64
FB
256 ret = avctx->codec->init(avctx);
257 if (ret < 0) {
3123dd79 258 av_freep(&avctx->priv_data);
de6d9b64
FB
259 return ret;
260 }
261 return 0;
262}
263
264int avcodec_encode_audio(AVCodecContext *avctx, UINT8 *buf, int buf_size,
265 const short *samples)
266{
267 int ret;
268
269 ret = avctx->codec->encode(avctx, buf, buf_size, (void *)samples);
270 avctx->frame_number++;
271 return ret;
272}
273
274int avcodec_encode_video(AVCodecContext *avctx, UINT8 *buf, int buf_size,
492cd3a9 275 const AVFrame *pict)
de6d9b64
FB
276{
277 int ret;
278
279 ret = avctx->codec->encode(avctx, buf, buf_size, (void *)pict);
6bb925f4
MN
280
281 emms_c(); //needed to avoid a emms_c() call before every return;
282
de6d9b64
FB
283 avctx->frame_number++;
284 return ret;
285}
286
287/* decode a frame. return -1 if error, otherwise return the number of
288 bytes used. If no frame could be decompressed, *got_picture_ptr is
289 zero. Otherwise, it is non zero */
492cd3a9 290int avcodec_decode_video(AVCodecContext *avctx, AVFrame *picture,
de6d9b64
FB
291 int *got_picture_ptr,
292 UINT8 *buf, int buf_size)
293{
294 int ret;
1e491e29 295
de6d9b64
FB
296 ret = avctx->codec->decode(avctx, picture, got_picture_ptr,
297 buf, buf_size);
6bb925f4
MN
298
299 emms_c(); //needed to avoid a emms_c() call before every return;
1e491e29 300
1cb0edb4
J
301 if (*got_picture_ptr)
302 avctx->frame_number++;
de6d9b64
FB
303 return ret;
304}
305
306/* decode an audio frame. return -1 if error, otherwise return the
307 *number of bytes used. If no frame could be decompressed,
308 *frame_size_ptr is zero. Otherwise, it is the decompressed frame
309 *size in BYTES. */
310int avcodec_decode_audio(AVCodecContext *avctx, INT16 *samples,
311 int *frame_size_ptr,
312 UINT8 *buf, int buf_size)
313{
314 int ret;
315
316 ret = avctx->codec->decode(avctx, samples, frame_size_ptr,
317 buf, buf_size);
318 avctx->frame_number++;
319 return ret;
320}
321
322int avcodec_close(AVCodecContext *avctx)
323{
324 if (avctx->codec->close)
325 avctx->codec->close(avctx);
3123dd79 326 av_freep(&avctx->priv_data);
de6d9b64
FB
327 avctx->codec = NULL;
328 return 0;
329}
330
331AVCodec *avcodec_find_encoder(enum CodecID id)
332{
333 AVCodec *p;
334 p = first_avcodec;
335 while (p) {
336 if (p->encode != NULL && p->id == id)
337 return p;
338 p = p->next;
339 }
340 return NULL;
341}
342
98f3b098
A
343AVCodec *avcodec_find_encoder_by_name(const char *name)
344{
345 AVCodec *p;
346 p = first_avcodec;
347 while (p) {
348 if (p->encode != NULL && strcmp(name,p->name) == 0)
349 return p;
350 p = p->next;
351 }
352 return NULL;
353}
354
de6d9b64
FB
355AVCodec *avcodec_find_decoder(enum CodecID id)
356{
357 AVCodec *p;
358 p = first_avcodec;
359 while (p) {
360 if (p->decode != NULL && p->id == id)
361 return p;
362 p = p->next;
363 }
364 return NULL;
365}
366
367AVCodec *avcodec_find_decoder_by_name(const char *name)
368{
369 AVCodec *p;
370 p = first_avcodec;
371 while (p) {
372 if (p->decode != NULL && strcmp(name,p->name) == 0)
373 return p;
374 p = p->next;
375 }
376 return NULL;
377}
378
379AVCodec *avcodec_find(enum CodecID id)
380{
381 AVCodec *p;
382 p = first_avcodec;
383 while (p) {
384 if (p->id == id)
385 return p;
386 p = p->next;
387 }
388 return NULL;
389}
390
391void avcodec_string(char *buf, int buf_size, AVCodecContext *enc, int encode)
392{
393 const char *codec_name;
394 AVCodec *p;
395 char buf1[32];
623563c0 396 char channels_str[100];
a96b68b7 397 int bitrate;
de6d9b64
FB
398
399 if (encode)
400 p = avcodec_find_encoder(enc->codec_id);
401 else
402 p = avcodec_find_decoder(enc->codec_id);
403
404 if (p) {
405 codec_name = p->name;
406 } else if (enc->codec_name[0] != '\0') {
407 codec_name = enc->codec_name;
408 } else {
409 /* output avi tags */
410 if (enc->codec_type == CODEC_TYPE_VIDEO) {
411 snprintf(buf1, sizeof(buf1), "%c%c%c%c",
412 enc->codec_tag & 0xff,
413 (enc->codec_tag >> 8) & 0xff,
414 (enc->codec_tag >> 16) & 0xff,
415 (enc->codec_tag >> 24) & 0xff);
416 } else {
417 snprintf(buf1, sizeof(buf1), "0x%04x", enc->codec_tag);
418 }
419 codec_name = buf1;
420 }
421
422 switch(enc->codec_type) {
423 case CODEC_TYPE_VIDEO:
424 snprintf(buf, buf_size,
425 "Video: %s%s",
426 codec_name, enc->flags & CODEC_FLAG_HQ ? " (hq)" : "");
cf087595
FB
427 if (enc->codec_id == CODEC_ID_RAWVIDEO) {
428 snprintf(buf + strlen(buf), buf_size - strlen(buf),
429 ", %s",
9bbffbb1 430 avcodec_get_pix_fmt_name(enc->pix_fmt));
cf087595 431 }
de6d9b64
FB
432 if (enc->width) {
433 snprintf(buf + strlen(buf), buf_size - strlen(buf),
434 ", %dx%d, %0.2f fps",
435 enc->width, enc->height,
436 (float)enc->frame_rate / FRAME_RATE_BASE);
437 }
4bfad535
FB
438 if (encode) {
439 snprintf(buf + strlen(buf), buf_size - strlen(buf),
440 ", q=%d-%d", enc->qmin, enc->qmax);
441 }
a96b68b7 442 bitrate = enc->bit_rate;
de6d9b64
FB
443 break;
444 case CODEC_TYPE_AUDIO:
445 snprintf(buf, buf_size,
446 "Audio: %s",
447 codec_name);
e0d2714a
J
448 switch (enc->channels) {
449 case 1:
623563c0 450 strcpy(channels_str, "mono");
e0d2714a
J
451 break;
452 case 2:
623563c0 453 strcpy(channels_str, "stereo");
e0d2714a
J
454 break;
455 case 6:
623563c0 456 strcpy(channels_str, "5:1");
e0d2714a
J
457 break;
458 default:
459 sprintf(channels_str, "%d channels", enc->channels);
460 break;
461 }
de6d9b64
FB
462 if (enc->sample_rate) {
463 snprintf(buf + strlen(buf), buf_size - strlen(buf),
464 ", %d Hz, %s",
465 enc->sample_rate,
e0d2714a 466 channels_str);
de6d9b64 467 }
e0d2714a 468
a96b68b7
FB
469 /* for PCM codecs, compute bitrate directly */
470 switch(enc->codec_id) {
471 case CODEC_ID_PCM_S16LE:
472 case CODEC_ID_PCM_S16BE:
473 case CODEC_ID_PCM_U16LE:
474 case CODEC_ID_PCM_U16BE:
a190b7e9 475 bitrate = enc->sample_rate * enc->channels * 16;
a96b68b7
FB
476 break;
477 case CODEC_ID_PCM_S8:
478 case CODEC_ID_PCM_U8:
479 case CODEC_ID_PCM_ALAW:
480 case CODEC_ID_PCM_MULAW:
a190b7e9 481 bitrate = enc->sample_rate * enc->channels * 8;
a96b68b7
FB
482 break;
483 default:
484 bitrate = enc->bit_rate;
485 break;
486 }
de6d9b64
FB
487 break;
488 default:
b71472eb 489 av_abort();
de6d9b64 490 }
4bfad535
FB
491 if (encode) {
492 if (enc->flags & CODEC_FLAG_PASS1)
493 snprintf(buf + strlen(buf), buf_size - strlen(buf),
494 ", pass 1");
495 if (enc->flags & CODEC_FLAG_PASS2)
496 snprintf(buf + strlen(buf), buf_size - strlen(buf),
497 ", pass 2");
498 }
a96b68b7 499 if (bitrate != 0) {
de6d9b64 500 snprintf(buf + strlen(buf), buf_size - strlen(buf),
a96b68b7 501 ", %d kb/s", bitrate / 1000);
de6d9b64
FB
502 }
503}
504
156e5023
NK
505unsigned avcodec_version( void )
506{
507 return LIBAVCODEC_VERSION_INT;
508}
cf087595 509
8bceb6af
NK
510unsigned avcodec_build( void )
511{
512 return LIBAVCODEC_BUILD;
513}
514
de6d9b64
FB
515/* must be called before any other functions */
516void avcodec_init(void)
517{
0344cd0a
AB
518 static int inited = 0;
519
520 if (inited != 0)
521 return;
522 inited = 1;
523
c6af86c2 524 //dsputil_init();
de6d9b64
FB
525}
526
4e00e76b 527/* this can be called after seeking and before trying to decode the next keyframe */
1c2a8c7f
MN
528void avcodec_flush_buffers(AVCodecContext *avctx)
529{
4e00e76b 530 int i;
1c2a8c7f 531 MpegEncContext *s = avctx->priv_data;
4e00e76b
MN
532
533 switch(avctx->codec_id){
534 case CODEC_ID_MPEG1VIDEO:
535 case CODEC_ID_H263:
536 case CODEC_ID_RV10:
537 case CODEC_ID_MJPEG:
538 case CODEC_ID_MJPEGB:
539 case CODEC_ID_MPEG4:
540 case CODEC_ID_MSMPEG4V1:
541 case CODEC_ID_MSMPEG4V2:
542 case CODEC_ID_MSMPEG4V3:
543 case CODEC_ID_WMV1:
544 case CODEC_ID_WMV2:
545 case CODEC_ID_H263P:
546 case CODEC_ID_H263I:
547 case CODEC_ID_SVQ1:
548 for(i=0; i<MAX_PICTURE_COUNT; i++){
549 if(s->picture[i].data[0] && ( s->picture[i].type == FF_BUFFER_TYPE_INTERNAL
550 || s->picture[i].type == FF_BUFFER_TYPE_USER))
492cd3a9 551 avctx->release_buffer(avctx, (AVFrame*)&s->picture[i]);
87fb268f
ZK
552 }
553 s->last_picture.data[0] = s->next_picture.data[0] = NULL;
4e00e76b
MN
554 break;
555 default:
556 //FIXME
557 break;
558 }
1c2a8c7f
MN
559}
560
cd4af68a 561static int raw_encode_init(AVCodecContext *s)
de6d9b64
FB
562{
563 return 0;
564}
565
cd4af68a
ZK
566static int raw_decode_frame(AVCodecContext *avctx,
567 void *data, int *data_size,
568 UINT8 *buf, int buf_size)
de6d9b64
FB
569{
570 return -1;
571}
572
cd4af68a
ZK
573static int raw_encode_frame(AVCodecContext *avctx,
574 unsigned char *frame, int buf_size, void *data)
de6d9b64
FB
575{
576 return -1;
577}
578
de6d9b64
FB
579AVCodec rawvideo_codec = {
580 "rawvideo",
581 CODEC_TYPE_VIDEO,
582 CODEC_ID_RAWVIDEO,
583 0,
cd4af68a
ZK
584 raw_encode_init,
585 raw_encode_frame,
de6d9b64 586 NULL,
cd4af68a 587 raw_decode_frame,
de6d9b64 588};