sse16 & pix_norm1 optimization patch by (Felix von Leitner <felix-ffmpeg at fefe...
[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
a949d72e 188void avcodec_get_context_defaults(AVCodecContext *s){
e8b62df6
MN
189 s->bit_rate= 800*1000;
190 s->bit_rate_tolerance= s->bit_rate*10;
a949d72e
MN
191 s->qmin= 2;
192 s->qmax= 31;
17a70fde
MN
193 s->mb_qmin= 2;
194 s->mb_qmax= 31;
a949d72e
MN
195 s->rc_eq= "tex^qComp";
196 s->qcompress= 0.5;
e8b62df6
MN
197 s->max_qdiff= 3;
198 s->b_quant_factor=1.25;
199 s->b_quant_offset=1.25;
b3a391e8 200 s->i_quant_factor=-0.8;
e8b62df6 201 s->i_quant_offset=0.0;
4d2858de 202 s->error_concealment= 3;
8d0e42ca 203 s->error_resilience= 1;
4d2858de 204 s->workaround_bugs= FF_BUG_AUTODETECT;
8d0e42ca
MN
205 s->frame_rate = 25 * FRAME_RATE_BASE;
206 s->gop_size= 50;
207 s->me_method= ME_EPZS;
1e491e29
MN
208 s->get_buffer= avcodec_default_get_buffer;
209 s->release_buffer= avcodec_default_release_buffer;
826f429a 210 s->me_subpel_quality=8;
a949d72e
MN
211}
212
213/**
214 * allocates a AVCodecContext and set it to defaults.
215 * this can be deallocated by simply calling free()
216 */
7ffbb60e 217AVCodecContext *avcodec_alloc_context(void){
a949d72e
MN
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
1e491e29 227/**
492cd3a9 228 * allocates a AVPFrame and set it to defaults.
1e491e29
MN
229 * this can be deallocated by simply calling free()
230 */
492cd3a9
MN
231AVFrame *avcodec_alloc_frame(void){
232 AVFrame *pic= av_mallocz(sizeof(AVFrame));
1e491e29
MN
233
234 return pic;
235}
236
de6d9b64
FB
237int avcodec_open(AVCodecContext *avctx, AVCodec *codec)
238{
239 int ret;
240
241 avctx->codec = codec;
4cb3ca72 242 avctx->codec_id = codec->id;
de6d9b64 243 avctx->frame_number = 0;
0edf8a7a
PG
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 }
de6d9b64
FB
251 ret = avctx->codec->init(avctx);
252 if (ret < 0) {
3123dd79 253 av_freep(&avctx->priv_data);
de6d9b64
FB
254 return ret;
255 }
256 return 0;
257}
258
259int 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
269int avcodec_encode_video(AVCodecContext *avctx, UINT8 *buf, int buf_size,
492cd3a9 270 const AVFrame *pict)
de6d9b64
FB
271{
272 int ret;
273
274 ret = avctx->codec->encode(avctx, buf, buf_size, (void *)pict);
6bb925f4
MN
275
276 emms_c(); //needed to avoid a emms_c() call before every return;
277
de6d9b64
FB
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 */
492cd3a9 285int avcodec_decode_video(AVCodecContext *avctx, AVFrame *picture,
de6d9b64
FB
286 int *got_picture_ptr,
287 UINT8 *buf, int buf_size)
288{
289 int ret;
1e491e29 290
de6d9b64
FB
291 ret = avctx->codec->decode(avctx, picture, got_picture_ptr,
292 buf, buf_size);
6bb925f4
MN
293
294 emms_c(); //needed to avoid a emms_c() call before every return;
1e491e29 295
1cb0edb4
J
296 if (*got_picture_ptr)
297 avctx->frame_number++;
de6d9b64
FB
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. */
305int 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
317int avcodec_close(AVCodecContext *avctx)
318{
319 if (avctx->codec->close)
320 avctx->codec->close(avctx);
3123dd79 321 av_freep(&avctx->priv_data);
de6d9b64
FB
322 avctx->codec = NULL;
323 return 0;
324}
325
326AVCodec *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
98f3b098
A
338AVCodec *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
de6d9b64
FB
350AVCodec *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
362AVCodec *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
374AVCodec *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
386void avcodec_string(char *buf, int buf_size, AVCodecContext *enc, int encode)
387{
388 const char *codec_name;
389 AVCodec *p;
390 char buf1[32];
623563c0 391 char channels_str[100];
a96b68b7 392 int bitrate;
de6d9b64
FB
393
394 if (encode)
395 p = avcodec_find_encoder(enc->codec_id);
396 else
397 p = avcodec_find_decoder(enc->codec_id);
398
399 if (p) {
400 codec_name = p->name;
401 } else if (enc->codec_name[0] != '\0') {
402 codec_name = enc->codec_name;
403 } else {
404 /* output avi tags */
405 if (enc->codec_type == CODEC_TYPE_VIDEO) {
406 snprintf(buf1, sizeof(buf1), "%c%c%c%c",
407 enc->codec_tag & 0xff,
408 (enc->codec_tag >> 8) & 0xff,
409 (enc->codec_tag >> 16) & 0xff,
410 (enc->codec_tag >> 24) & 0xff);
411 } else {
412 snprintf(buf1, sizeof(buf1), "0x%04x", enc->codec_tag);
413 }
414 codec_name = buf1;
415 }
416
417 switch(enc->codec_type) {
418 case CODEC_TYPE_VIDEO:
419 snprintf(buf, buf_size,
420 "Video: %s%s",
421 codec_name, enc->flags & CODEC_FLAG_HQ ? " (hq)" : "");
cf087595
FB
422 if (enc->codec_id == CODEC_ID_RAWVIDEO) {
423 snprintf(buf + strlen(buf), buf_size - strlen(buf),
424 ", %s",
9bbffbb1 425 avcodec_get_pix_fmt_name(enc->pix_fmt));
cf087595 426 }
de6d9b64
FB
427 if (enc->width) {
428 snprintf(buf + strlen(buf), buf_size - strlen(buf),
429 ", %dx%d, %0.2f fps",
430 enc->width, enc->height,
431 (float)enc->frame_rate / FRAME_RATE_BASE);
432 }
4bfad535
FB
433 if (encode) {
434 snprintf(buf + strlen(buf), buf_size - strlen(buf),
435 ", q=%d-%d", enc->qmin, enc->qmax);
436 }
a96b68b7 437 bitrate = enc->bit_rate;
de6d9b64
FB
438 break;
439 case CODEC_TYPE_AUDIO:
440 snprintf(buf, buf_size,
441 "Audio: %s",
442 codec_name);
e0d2714a
J
443 switch (enc->channels) {
444 case 1:
623563c0 445 strcpy(channels_str, "mono");
e0d2714a
J
446 break;
447 case 2:
623563c0 448 strcpy(channels_str, "stereo");
e0d2714a
J
449 break;
450 case 6:
623563c0 451 strcpy(channels_str, "5:1");
e0d2714a
J
452 break;
453 default:
454 sprintf(channels_str, "%d channels", enc->channels);
455 break;
456 }
de6d9b64
FB
457 if (enc->sample_rate) {
458 snprintf(buf + strlen(buf), buf_size - strlen(buf),
459 ", %d Hz, %s",
460 enc->sample_rate,
e0d2714a 461 channels_str);
de6d9b64 462 }
e0d2714a 463
a96b68b7
FB
464 /* for PCM codecs, compute bitrate directly */
465 switch(enc->codec_id) {
466 case CODEC_ID_PCM_S16LE:
467 case CODEC_ID_PCM_S16BE:
468 case CODEC_ID_PCM_U16LE:
469 case CODEC_ID_PCM_U16BE:
a190b7e9 470 bitrate = enc->sample_rate * enc->channels * 16;
a96b68b7
FB
471 break;
472 case CODEC_ID_PCM_S8:
473 case CODEC_ID_PCM_U8:
474 case CODEC_ID_PCM_ALAW:
475 case CODEC_ID_PCM_MULAW:
a190b7e9 476 bitrate = enc->sample_rate * enc->channels * 8;
a96b68b7
FB
477 break;
478 default:
479 bitrate = enc->bit_rate;
480 break;
481 }
de6d9b64
FB
482 break;
483 default:
b71472eb 484 av_abort();
de6d9b64 485 }
4bfad535
FB
486 if (encode) {
487 if (enc->flags & CODEC_FLAG_PASS1)
488 snprintf(buf + strlen(buf), buf_size - strlen(buf),
489 ", pass 1");
490 if (enc->flags & CODEC_FLAG_PASS2)
491 snprintf(buf + strlen(buf), buf_size - strlen(buf),
492 ", pass 2");
493 }
a96b68b7 494 if (bitrate != 0) {
de6d9b64 495 snprintf(buf + strlen(buf), buf_size - strlen(buf),
a96b68b7 496 ", %d kb/s", bitrate / 1000);
de6d9b64
FB
497 }
498}
499
156e5023
NK
500unsigned avcodec_version( void )
501{
502 return LIBAVCODEC_VERSION_INT;
503}
cf087595 504
8bceb6af
NK
505unsigned avcodec_build( void )
506{
507 return LIBAVCODEC_BUILD;
508}
509
de6d9b64
FB
510/* must be called before any other functions */
511void avcodec_init(void)
512{
0344cd0a
AB
513 static int inited = 0;
514
515 if (inited != 0)
516 return;
517 inited = 1;
518
c6af86c2 519 //dsputil_init();
de6d9b64
FB
520}
521
4e00e76b 522/* this can be called after seeking and before trying to decode the next keyframe */
1c2a8c7f
MN
523void avcodec_flush_buffers(AVCodecContext *avctx)
524{
4e00e76b 525 int i;
1c2a8c7f 526 MpegEncContext *s = avctx->priv_data;
4e00e76b
MN
527
528 switch(avctx->codec_id){
529 case CODEC_ID_MPEG1VIDEO:
530 case CODEC_ID_H263:
531 case CODEC_ID_RV10:
532 case CODEC_ID_MJPEG:
533 case CODEC_ID_MJPEGB:
534 case CODEC_ID_MPEG4:
535 case CODEC_ID_MSMPEG4V1:
536 case CODEC_ID_MSMPEG4V2:
537 case CODEC_ID_MSMPEG4V3:
538 case CODEC_ID_WMV1:
539 case CODEC_ID_WMV2:
540 case CODEC_ID_H263P:
541 case CODEC_ID_H263I:
542 case CODEC_ID_SVQ1:
543 for(i=0; i<MAX_PICTURE_COUNT; i++){
544 if(s->picture[i].data[0] && ( s->picture[i].type == FF_BUFFER_TYPE_INTERNAL
545 || s->picture[i].type == FF_BUFFER_TYPE_USER))
492cd3a9 546 avctx->release_buffer(avctx, (AVFrame*)&s->picture[i]);
87fb268f
ZK
547 }
548 s->last_picture.data[0] = s->next_picture.data[0] = NULL;
4e00e76b
MN
549 break;
550 default:
551 //FIXME
552 break;
553 }
1c2a8c7f
MN
554}
555
cd4af68a 556static int raw_encode_init(AVCodecContext *s)
de6d9b64
FB
557{
558 return 0;
559}
560
cd4af68a
ZK
561static int raw_decode_frame(AVCodecContext *avctx,
562 void *data, int *data_size,
563 UINT8 *buf, int buf_size)
de6d9b64
FB
564{
565 return -1;
566}
567
cd4af68a
ZK
568static int raw_encode_frame(AVCodecContext *avctx,
569 unsigned char *frame, int buf_size, void *data)
de6d9b64
FB
570{
571 return -1;
572}
573
de6d9b64
FB
574AVCodec rawvideo_codec = {
575 "rawvideo",
576 CODEC_TYPE_VIDEO,
577 CODEC_ID_RAWVIDEO,
578 0,
cd4af68a
ZK
579 raw_encode_init,
580 raw_encode_frame,
de6d9b64 581 NULL,
cd4af68a 582 raw_decode_frame,
de6d9b64 583};