Statistics
| Branch: | Revision:

ffmpeg / libavcodec / utils.c @ 11e29a41

History | View | Annotate | Download (20.4 KB)

1 de6d9b64 Fabrice Bellard
/*
2
 * utils for libavcodec
3 ff4ec49e Fabrice Bellard
 * Copyright (c) 2001 Fabrice Bellard.
4 9b879566 Michel Bardiaux
 * Copyright (c) 2003 Michel Bardiaux for the av_log API
5 de6d9b64 Fabrice Bellard
 *
6 ff4ec49e Fabrice Bellard
 * This library is free software; you can redistribute it and/or
7
 * modify it under the terms of the GNU Lesser General Public
8
 * License as published by the Free Software Foundation; either
9
 * version 2 of the License, or (at your option) any later version.
10 de6d9b64 Fabrice Bellard
 *
11 ff4ec49e Fabrice Bellard
 * This library is distributed in the hope that it will be useful,
12 de6d9b64 Fabrice Bellard
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 ff4ec49e Fabrice Bellard
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14
 * Lesser General Public License for more details.
15 de6d9b64 Fabrice Bellard
 *
16 ff4ec49e Fabrice Bellard
 * You should have received a copy of the GNU Lesser General Public
17
 * License along with this library; if not, write to the Free Software
18
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19 de6d9b64 Fabrice Bellard
 */
20 983e3246 Michael Niedermayer
 
21
/**
22
 * @file utils.c
23
 * utils.
24
 */
25
 
26 de6d9b64 Fabrice Bellard
#include "avcodec.h"
27 3123dd79 Fabrice Bellard
#include "dsputil.h"
28 1c2a8c7f Michael Niedermayer
#include "mpegvideo.h"
29 9b879566 Michel Bardiaux
#include <stdarg.h>
30 de6d9b64 Fabrice Bellard
31 18f77016 Zdenek Kabelac
void *av_mallocz(unsigned int size)
32 3123dd79 Fabrice Bellard
{
33
    void *ptr;
34 5b20b732 Michael Niedermayer
    
35 3123dd79 Fabrice Bellard
    ptr = av_malloc(size);
36
    if (!ptr)
37
        return NULL;
38
    memset(ptr, 0, size);
39
    return ptr;
40
}
41
42 8e1e6f31 Fabrice Bellard
char *av_strdup(const char *s)
43
{
44
    char *ptr;
45
    int len;
46
    len = strlen(s) + 1;
47
    ptr = av_malloc(len);
48
    if (!ptr)
49
        return NULL;
50
    memcpy(ptr, s, len);
51
    return ptr;
52
}
53
54
/**
55
 * realloc which does nothing if the block is large enough
56
 */
57 5c91a675 Zdenek Kabelac
void *av_fast_realloc(void *ptr, unsigned int *size, unsigned int min_size)
58 8e1e6f31 Fabrice Bellard
{
59
    if(min_size < *size) 
60
        return ptr;
61
    
62
    *size= min_size + 10*1024;
63
64
    return av_realloc(ptr, *size);
65
}
66
67
68 855ea723 Zdenek Kabelac
/* allocation of static arrays - do not use for normal allocation */
69
static unsigned int last_static = 0;
70
static char*** array_static = NULL;
71
static const unsigned int grow_static = 64; // ^2
72
void *__av_mallocz_static(void** location, unsigned int size)
73
{
74 5c91a675 Zdenek Kabelac
    unsigned int l = (last_static + grow_static) & ~(grow_static - 1);
75 855ea723 Zdenek Kabelac
    void *ptr = av_mallocz(size);
76
    if (!ptr)
77
        return NULL;
78
79
    if (location)
80
    {
81
        if (l > last_static)
82 8e1e6f31 Fabrice Bellard
            array_static = av_realloc(array_static, l);
83 855ea723 Zdenek Kabelac
        array_static[last_static++] = (char**) location;
84
        *location = ptr;
85
    }
86
    return ptr;
87
}
88
/* free all static arrays and reset pointers to 0 */
89 4cfbf61b Falk Hüffner
void av_free_static(void)
90 855ea723 Zdenek Kabelac
{
91
    if (array_static)
92
    {
93
        unsigned i;
94
        for (i = 0; i < last_static; i++)
95
        {
96 8e1e6f31 Fabrice Bellard
            av_free(*array_static[i]);
97 855ea723 Zdenek Kabelac
            *array_static[i] = NULL;
98
        }
99 8e1e6f31 Fabrice Bellard
        av_free(array_static);
100 855ea723 Zdenek Kabelac
        array_static = 0;
101
    }
102
    last_static = 0;
103
}
104
105 4d7a0a05 Fabrice Bellard
/* cannot call it directly because of 'void **' casting is not automatic */
106
void __av_freep(void **ptr)
107
{
108
    av_free(*ptr);
109
    *ptr = NULL;
110
}
111
112 de6d9b64 Fabrice Bellard
/* encoder management */
113
AVCodec *first_avcodec;
114
115
void register_avcodec(AVCodec *format)
116
{
117
    AVCodec **p;
118
    p = &first_avcodec;
119
    while (*p != NULL) p = &(*p)->next;
120
    *p = format;
121
    format->next = NULL;
122
}
123
124 d90cf87b Michael Niedermayer
typedef struct InternalBuffer{
125 1e491e29 Michael Niedermayer
    int last_pic_num;
126 d90cf87b Michael Niedermayer
    uint8_t *base[4];
127 1e491e29 Michael Niedermayer
    uint8_t *data[4];
128 237e4938 Michael Niedermayer
    int linesize[4];
129 d90cf87b Michael Niedermayer
}InternalBuffer;
130
131
#define INTERNAL_BUFFER_SIZE 32
132 1e491e29 Michael Niedermayer
133 f0bbfc4a Michael Niedermayer
#define ALIGN(x, a) (((x)+(a)-1)&~((a)-1))
134
135
void avcodec_align_dimensions(AVCodecContext *s, int *width, int *height){
136
    int w_align= 1;    
137
    int h_align= 1;    
138
    
139
    switch(s->pix_fmt){
140
    case PIX_FMT_YUV420P:
141
    case PIX_FMT_YUV422:
142
    case PIX_FMT_YUV422P:
143
    case PIX_FMT_YUV444P:
144
    case PIX_FMT_GRAY8:
145
    case PIX_FMT_YUVJ420P:
146
    case PIX_FMT_YUVJ422P:
147
    case PIX_FMT_YUVJ444P:
148
        w_align= 16; //FIXME check for non mpeg style codecs and use less alignment
149
        h_align= 16;
150
        break;
151
    case PIX_FMT_YUV411P:
152
        w_align=32;
153
        h_align=8;
154
        break;
155
    case PIX_FMT_YUV410P:
156
        if(s->codec_id == CODEC_ID_SVQ1){
157
            w_align=64;
158
            h_align=64;
159
        }
160
        break;
161
    default:
162
        w_align= 1;
163
        h_align= 1;
164
        break;
165
    }
166
167
    *width = ALIGN(*width , w_align);
168
    *height= ALIGN(*height, h_align);
169
}
170
171 492cd3a9 Michael Niedermayer
int avcodec_default_get_buffer(AVCodecContext *s, AVFrame *pic){
172 1e491e29 Michael Niedermayer
    int i;
173 f0bbfc4a Michael Niedermayer
    int w= s->width;
174
    int h= s->height;
175 d90cf87b Michael Niedermayer
    InternalBuffer *buf;
176 237e4938 Michael Niedermayer
    int *picture_number;
177 4e00e76b Michael Niedermayer
    
178
    assert(pic->data[0]==NULL);
179 d90cf87b Michael Niedermayer
    assert(INTERNAL_BUFFER_SIZE > s->internal_buffer_count);
180 1e491e29 Michael Niedermayer
181 d90cf87b Michael Niedermayer
    if(s->internal_buffer==NULL){
182
        s->internal_buffer= av_mallocz(INTERNAL_BUFFER_SIZE*sizeof(InternalBuffer));
183
    }
184
#if 0
185
    s->internal_buffer= av_fast_realloc(
186
        s->internal_buffer, 
187
        &s->internal_buffer_size, 
188
        sizeof(InternalBuffer)*FFMAX(99,  s->internal_buffer_count+1)/*FIXME*/
189
        );
190
#endif
191
     
192
    buf= &((InternalBuffer*)s->internal_buffer)[s->internal_buffer_count];
193 237e4938 Michael Niedermayer
    picture_number= &(((InternalBuffer*)s->internal_buffer)[INTERNAL_BUFFER_SIZE-1]).last_pic_num; //FIXME ugly hack
194
    (*picture_number)++;
195
    
196 d90cf87b Michael Niedermayer
    if(buf->base[0]){
197 237e4938 Michael Niedermayer
        pic->age= *picture_number - buf->last_pic_num;
198
        buf->last_pic_num= *picture_number;
199 1e491e29 Michael Niedermayer
    }else{
200 f0bbfc4a Michael Niedermayer
        int h_chroma_shift, v_chroma_shift;
201
        int s_align, pixel_size;
202 1e491e29 Michael Niedermayer
        
203
        avcodec_get_chroma_sub_sample(s->pix_fmt, &h_chroma_shift, &v_chroma_shift);
204 f0bbfc4a Michael Niedermayer
        
205 1e491e29 Michael Niedermayer
        switch(s->pix_fmt){
206 ba88675b Michael Niedermayer
        case PIX_FMT_RGB555:
207
        case PIX_FMT_RGB565:
208 1e491e29 Michael Niedermayer
        case PIX_FMT_YUV422:
209
            pixel_size=2;
210
            break;
211
        case PIX_FMT_RGB24:
212
        case PIX_FMT_BGR24:
213
            pixel_size=3;
214
            break;
215
        case PIX_FMT_RGBA32:
216
            pixel_size=4;
217
            break;
218
        default:
219
            pixel_size=1;
220
        }
221 f0bbfc4a Michael Niedermayer
222
        avcodec_align_dimensions(s, &w, &h);
223
#if defined(ARCH_POWERPC) || defined(HAVE_MMI) //FIXME some cleaner check
224
        s_align= 16;
225
#else
226
        s_align= 8;
227
#endif
228
            
229 1e491e29 Michael Niedermayer
        if(!(s->flags&CODEC_FLAG_EMU_EDGE)){
230
            w+= EDGE_WIDTH*2;
231
            h+= EDGE_WIDTH*2;
232
        }
233
        
234 d90cf87b Michael Niedermayer
        buf->last_pic_num= -256*256*256*64;
235 1e491e29 Michael Niedermayer
236
        for(i=0; i<3; i++){
237 2c19981a Michael Niedermayer
            const int h_shift= i==0 ? 0 : h_chroma_shift;
238
            const int v_shift= i==0 ? 0 : v_chroma_shift;
239 1e491e29 Michael Niedermayer
240 237e4938 Michael Niedermayer
            buf->linesize[i]= ALIGN(pixel_size*w>>h_shift, s_align);
241 1e491e29 Michael Niedermayer
242 237e4938 Michael Niedermayer
            buf->base[i]= av_mallocz((buf->linesize[i]*h>>v_shift)+16); //FIXME 16
243 d90cf87b Michael Niedermayer
            if(buf->base[i]==NULL) return -1;
244 237e4938 Michael Niedermayer
            memset(buf->base[i], 128, buf->linesize[i]*h>>v_shift);
245 1e491e29 Michael Niedermayer
        
246
            if(s->flags&CODEC_FLAG_EMU_EDGE)
247 d90cf87b Michael Niedermayer
                buf->data[i] = buf->base[i];
248 1e491e29 Michael Niedermayer
            else
249 237e4938 Michael Niedermayer
                buf->data[i] = buf->base[i] + ALIGN((buf->linesize[i]*EDGE_WIDTH>>v_shift) + (EDGE_WIDTH>>h_shift), s_align);
250 1e491e29 Michael Niedermayer
        }
251
        pic->age= 256*256*256*64;
252
    }
253 237e4938 Michael Niedermayer
    pic->type= FF_BUFFER_TYPE_INTERNAL;
254 1e491e29 Michael Niedermayer
255 d90cf87b Michael Niedermayer
    for(i=0; i<4; i++){
256
        pic->base[i]= buf->base[i];
257
        pic->data[i]= buf->data[i];
258 237e4938 Michael Niedermayer
        pic->linesize[i]= buf->linesize[i];
259 d90cf87b Michael Niedermayer
    }
260
    s->internal_buffer_count++;
261
262 1e491e29 Michael Niedermayer
    return 0;
263
}
264
265 492cd3a9 Michael Niedermayer
void avcodec_default_release_buffer(AVCodecContext *s, AVFrame *pic){
266 1e491e29 Michael Niedermayer
    int i;
267 d90cf87b Michael Niedermayer
    InternalBuffer *buf, *last, temp;
268
269 4e00e76b Michael Niedermayer
    assert(pic->type==FF_BUFFER_TYPE_INTERNAL);
270 59c673d5 Michael Niedermayer
    assert(s->internal_buffer_count);
271 d90cf87b Michael Niedermayer
272 b1609412 Fabrice Bellard
    buf = NULL; /* avoids warning */
273 d90cf87b Michael Niedermayer
    for(i=0; i<s->internal_buffer_count; i++){ //just 3-5 checks so is not worth to optimize
274
        buf= &((InternalBuffer*)s->internal_buffer)[i];
275
        if(buf->data[0] == pic->data[0])
276
            break;
277
    }
278
    assert(i < s->internal_buffer_count);
279
    s->internal_buffer_count--;
280
    last = &((InternalBuffer*)s->internal_buffer)[s->internal_buffer_count];
281
282
    temp= *buf;
283
    *buf= *last;
284
    *last= temp;
285
286
    for(i=0; i<3; i++){
287 1e491e29 Michael Niedermayer
        pic->data[i]=NULL;
288 d90cf87b Michael Niedermayer
//        pic->base[i]=NULL;
289
    }
290 1e491e29 Michael Niedermayer
//printf("R%X\n", pic->opaque);
291
}
292
293 a33c7159 Michael Niedermayer
enum PixelFormat avcodec_default_get_format(struct AVCodecContext *s, enum PixelFormat * fmt){
294
    return fmt[0];
295
}
296
297 a949d72e Michael Niedermayer
void avcodec_get_context_defaults(AVCodecContext *s){
298 e8b62df6 Michael Niedermayer
    s->bit_rate= 800*1000;
299
    s->bit_rate_tolerance= s->bit_rate*10;
300 a949d72e Michael Niedermayer
    s->qmin= 2;
301
    s->qmax= 31;
302 17a70fde Michael Niedermayer
    s->mb_qmin= 2;
303
    s->mb_qmax= 31;
304 a949d72e Michael Niedermayer
    s->rc_eq= "tex^qComp";
305
    s->qcompress= 0.5;
306 e8b62df6 Michael Niedermayer
    s->max_qdiff= 3;
307
    s->b_quant_factor=1.25;
308
    s->b_quant_offset=1.25;
309 b3a391e8 Michael Niedermayer
    s->i_quant_factor=-0.8;
310 e8b62df6 Michael Niedermayer
    s->i_quant_offset=0.0;
311 4d2858de Michael Niedermayer
    s->error_concealment= 3;
312 8d0e42ca Michael Niedermayer
    s->error_resilience= 1;
313 4d2858de Michael Niedermayer
    s->workaround_bugs= FF_BUG_AUTODETECT;
314 14bea432 Michael Niedermayer
    s->frame_rate_base= 1;
315
    s->frame_rate = 25;
316 8d0e42ca Michael Niedermayer
    s->gop_size= 50;
317
    s->me_method= ME_EPZS;
318 1e491e29 Michael Niedermayer
    s->get_buffer= avcodec_default_get_buffer;
319
    s->release_buffer= avcodec_default_release_buffer;
320 a33c7159 Michael Niedermayer
    s->get_format= avcodec_default_get_format;
321 826f429a Michael Niedermayer
    s->me_subpel_quality=8;
322 158c7f05 Michael Niedermayer
    s->lmin= FF_QP2LAMBDA * s->qmin;
323
    s->lmax= FF_QP2LAMBDA * s->qmax;
324 5ff85f1d Michael Niedermayer
    s->sample_aspect_ratio= (AVRational){0,1};
325 1984f635 Michael Niedermayer
    
326
    s->intra_quant_bias= FF_DEFAULT_QUANT_BIAS;
327
    s->inter_quant_bias= FF_DEFAULT_QUANT_BIAS;
328 2a2bbcb0 Mike Melanson
    s->palctrl = NULL;
329 074c4ca7 Roberto Togni
    s->cr_available = 0;
330 a949d72e Michael Niedermayer
}
331
332
/**
333
 * allocates a AVCodecContext and set it to defaults.
334
 * this can be deallocated by simply calling free() 
335
 */
336 7ffbb60e Falk Hüffner
AVCodecContext *avcodec_alloc_context(void){
337 a949d72e Michael Niedermayer
    AVCodecContext *avctx= av_mallocz(sizeof(AVCodecContext));
338
    
339
    if(avctx==NULL) return NULL;
340
    
341
    avcodec_get_context_defaults(avctx);
342
    
343
    return avctx;
344
}
345
346 1e491e29 Michael Niedermayer
/**
347 492cd3a9 Michael Niedermayer
 * allocates a AVPFrame and set it to defaults.
348 1e491e29 Michael Niedermayer
 * this can be deallocated by simply calling free() 
349
 */
350 492cd3a9 Michael Niedermayer
AVFrame *avcodec_alloc_frame(void){
351
    AVFrame *pic= av_mallocz(sizeof(AVFrame));
352 1e491e29 Michael Niedermayer
    
353
    return pic;
354
}
355
356 de6d9b64 Fabrice Bellard
int avcodec_open(AVCodecContext *avctx, AVCodec *codec)
357
{
358
    int ret;
359
360 09770af8 Michael Niedermayer
    if(avctx->codec)
361
        return -1;
362
363 de6d9b64 Fabrice Bellard
    avctx->codec = codec;
364 4cb3ca72 Zdenek Kabelac
    avctx->codec_id = codec->id;
365 de6d9b64 Fabrice Bellard
    avctx->frame_number = 0;
366 0edf8a7a Philip Gladstone
    if (codec->priv_data_size > 0) {
367
        avctx->priv_data = av_mallocz(codec->priv_data_size);
368
        if (!avctx->priv_data) 
369
            return -ENOMEM;
370
    } else {
371
        avctx->priv_data = NULL;
372
    }
373 de6d9b64 Fabrice Bellard
    ret = avctx->codec->init(avctx);
374
    if (ret < 0) {
375 3123dd79 Fabrice Bellard
        av_freep(&avctx->priv_data);
376 de6d9b64 Fabrice Bellard
        return ret;
377
    }
378
    return 0;
379
}
380
381 0c1a9eda Zdenek Kabelac
int avcodec_encode_audio(AVCodecContext *avctx, uint8_t *buf, int buf_size, 
382 de6d9b64 Fabrice Bellard
                         const short *samples)
383
{
384
    int ret;
385
386
    ret = avctx->codec->encode(avctx, buf, buf_size, (void *)samples);
387
    avctx->frame_number++;
388
    return ret;
389
}
390
391 0c1a9eda Zdenek Kabelac
int avcodec_encode_video(AVCodecContext *avctx, uint8_t *buf, int buf_size, 
392 492cd3a9 Michael Niedermayer
                         const AVFrame *pict)
393 de6d9b64 Fabrice Bellard
{
394
    int ret;
395
396
    ret = avctx->codec->encode(avctx, buf, buf_size, (void *)pict);
397 6bb925f4 Michael Niedermayer
    
398
    emms_c(); //needed to avoid a emms_c() call before every return;
399
400 de6d9b64 Fabrice Bellard
    avctx->frame_number++;
401
    return ret;
402
}
403
404 f138f883 Michael Niedermayer
/** 
405
 * decode a frame. 
406
 * @param buf bitstream buffer, must be FF_INPUT_BUFFER_PADDING_SIZE larger then the actual read bytes
407
 * because some optimized bitstream readers read 32 or 64 bit at once and could read over the end
408
 * @param buf_size the size of the buffer in bytes
409
 * @param got_picture_ptr zero if no frame could be decompressed, Otherwise, it is non zero
410
 * @return -1 if error, otherwise return the number of
411
 * bytes used. 
412
 */
413 492cd3a9 Michael Niedermayer
int avcodec_decode_video(AVCodecContext *avctx, AVFrame *picture, 
414 de6d9b64 Fabrice Bellard
                         int *got_picture_ptr,
415 0c1a9eda Zdenek Kabelac
                         uint8_t *buf, int buf_size)
416 de6d9b64 Fabrice Bellard
{
417
    int ret;
418 1e491e29 Michael Niedermayer
    
419 de6d9b64 Fabrice Bellard
    ret = avctx->codec->decode(avctx, picture, got_picture_ptr, 
420
                               buf, buf_size);
421 6bb925f4 Michael Niedermayer
422
    emms_c(); //needed to avoid a emms_c() call before every return;
423 1e491e29 Michael Niedermayer
    
424 1cb0edb4 Juanjo
    if (*got_picture_ptr)                           
425
        avctx->frame_number++;
426 de6d9b64 Fabrice Bellard
    return ret;
427
}
428
429
/* decode an audio frame. return -1 if error, otherwise return the
430
   *number of bytes used. If no frame could be decompressed,
431
   *frame_size_ptr is zero. Otherwise, it is the decompressed frame
432
   *size in BYTES. */
433 0c1a9eda Zdenek Kabelac
int avcodec_decode_audio(AVCodecContext *avctx, int16_t *samples, 
434 de6d9b64 Fabrice Bellard
                         int *frame_size_ptr,
435 0c1a9eda Zdenek Kabelac
                         uint8_t *buf, int buf_size)
436 de6d9b64 Fabrice Bellard
{
437
    int ret;
438
439
    ret = avctx->codec->decode(avctx, samples, frame_size_ptr, 
440
                               buf, buf_size);
441
    avctx->frame_number++;
442
    return ret;
443
}
444
445
int avcodec_close(AVCodecContext *avctx)
446
{
447
    if (avctx->codec->close)
448
        avctx->codec->close(avctx);
449 3123dd79 Fabrice Bellard
    av_freep(&avctx->priv_data);
450 de6d9b64 Fabrice Bellard
    avctx->codec = NULL;
451
    return 0;
452
}
453
454
AVCodec *avcodec_find_encoder(enum CodecID id)
455
{
456
    AVCodec *p;
457
    p = first_avcodec;
458
    while (p) {
459
        if (p->encode != NULL && p->id == id)
460
            return p;
461
        p = p->next;
462
    }
463
    return NULL;
464
}
465
466 98f3b098 Arpi
AVCodec *avcodec_find_encoder_by_name(const char *name)
467
{
468
    AVCodec *p;
469
    p = first_avcodec;
470
    while (p) {
471
        if (p->encode != NULL && strcmp(name,p->name) == 0)
472
            return p;
473
        p = p->next;
474
    }
475
    return NULL;
476
}
477
478 de6d9b64 Fabrice Bellard
AVCodec *avcodec_find_decoder(enum CodecID id)
479
{
480
    AVCodec *p;
481
    p = first_avcodec;
482
    while (p) {
483
        if (p->decode != NULL && p->id == id)
484
            return p;
485
        p = p->next;
486
    }
487
    return NULL;
488
}
489
490
AVCodec *avcodec_find_decoder_by_name(const char *name)
491
{
492
    AVCodec *p;
493
    p = first_avcodec;
494
    while (p) {
495
        if (p->decode != NULL && strcmp(name,p->name) == 0)
496
            return p;
497
        p = p->next;
498
    }
499
    return NULL;
500
}
501
502
AVCodec *avcodec_find(enum CodecID id)
503
{
504
    AVCodec *p;
505
    p = first_avcodec;
506
    while (p) {
507
        if (p->id == id)
508
            return p;
509
        p = p->next;
510
    }
511
    return NULL;
512
}
513
514
void avcodec_string(char *buf, int buf_size, AVCodecContext *enc, int encode)
515
{
516
    const char *codec_name;
517
    AVCodec *p;
518
    char buf1[32];
519 623563c0 Zdenek Kabelac
    char channels_str[100];
520 a96b68b7 Fabrice Bellard
    int bitrate;
521 de6d9b64 Fabrice Bellard
522
    if (encode)
523
        p = avcodec_find_encoder(enc->codec_id);
524
    else
525
        p = avcodec_find_decoder(enc->codec_id);
526
527
    if (p) {
528
        codec_name = p->name;
529 98ce5991 Fabrice Bellard
        if (!encode && enc->codec_id == CODEC_ID_MP3) {
530
            if (enc->sub_id == 2)
531
                codec_name = "mp2";
532
            else if (enc->sub_id == 1)
533
                codec_name = "mp1";
534
        }
535 985180a1 Fabrice Bellard
    } else if (enc->codec_id == CODEC_ID_MPEG2TS) {
536
        /* fake mpeg2 transport stream codec (currently not
537
           registered) */
538
        codec_name = "mpeg2ts";
539 de6d9b64 Fabrice Bellard
    } else if (enc->codec_name[0] != '\0') {
540
        codec_name = enc->codec_name;
541
    } else {
542
        /* output avi tags */
543
        if (enc->codec_type == CODEC_TYPE_VIDEO) {
544
            snprintf(buf1, sizeof(buf1), "%c%c%c%c", 
545
                     enc->codec_tag & 0xff,
546
                     (enc->codec_tag >> 8) & 0xff,
547
                     (enc->codec_tag >> 16) & 0xff,
548
                     (enc->codec_tag >> 24) & 0xff);
549
        } else {
550
            snprintf(buf1, sizeof(buf1), "0x%04x", enc->codec_tag);
551
        }
552
        codec_name = buf1;
553
    }
554
555
    switch(enc->codec_type) {
556
    case CODEC_TYPE_VIDEO:
557
        snprintf(buf, buf_size,
558
                 "Video: %s%s",
559 7d1c3fc1 Michael Niedermayer
                 codec_name, enc->mb_decision ? " (hq)" : "");
560 cf087595 Fabrice Bellard
        if (enc->codec_id == CODEC_ID_RAWVIDEO) {
561
            snprintf(buf + strlen(buf), buf_size - strlen(buf),
562
                     ", %s",
563 9bbffbb1 Fabrice Bellard
                     avcodec_get_pix_fmt_name(enc->pix_fmt));
564 cf087595 Fabrice Bellard
        }
565 de6d9b64 Fabrice Bellard
        if (enc->width) {
566
            snprintf(buf + strlen(buf), buf_size - strlen(buf),
567
                     ", %dx%d, %0.2f fps",
568
                     enc->width, enc->height, 
569 14bea432 Michael Niedermayer
                     (float)enc->frame_rate / enc->frame_rate_base);
570 de6d9b64 Fabrice Bellard
        }
571 4bfad535 Fabrice Bellard
        if (encode) {
572
            snprintf(buf + strlen(buf), buf_size - strlen(buf),
573
                     ", q=%d-%d", enc->qmin, enc->qmax);
574
        }
575 a96b68b7 Fabrice Bellard
        bitrate = enc->bit_rate;
576 de6d9b64 Fabrice Bellard
        break;
577
    case CODEC_TYPE_AUDIO:
578
        snprintf(buf, buf_size,
579
                 "Audio: %s",
580
                 codec_name);
581 e0d2714a Juanjo
        switch (enc->channels) {
582
            case 1:
583 623563c0 Zdenek Kabelac
                strcpy(channels_str, "mono");
584 e0d2714a Juanjo
                break;
585
            case 2:
586 623563c0 Zdenek Kabelac
                strcpy(channels_str, "stereo");
587 e0d2714a Juanjo
                break;
588
            case 6:
589 623563c0 Zdenek Kabelac
                strcpy(channels_str, "5:1");
590 e0d2714a Juanjo
                break;
591
            default:
592
                sprintf(channels_str, "%d channels", enc->channels);
593
                break;
594
        }
595 de6d9b64 Fabrice Bellard
        if (enc->sample_rate) {
596
            snprintf(buf + strlen(buf), buf_size - strlen(buf),
597
                     ", %d Hz, %s",
598
                     enc->sample_rate,
599 e0d2714a Juanjo
                     channels_str);
600 de6d9b64 Fabrice Bellard
        }
601 e0d2714a Juanjo
        
602 a96b68b7 Fabrice Bellard
        /* for PCM codecs, compute bitrate directly */
603
        switch(enc->codec_id) {
604
        case CODEC_ID_PCM_S16LE:
605
        case CODEC_ID_PCM_S16BE:
606
        case CODEC_ID_PCM_U16LE:
607
        case CODEC_ID_PCM_U16BE:
608 a190b7e9 Fabrice Bellard
            bitrate = enc->sample_rate * enc->channels * 16;
609 a96b68b7 Fabrice Bellard
            break;
610
        case CODEC_ID_PCM_S8:
611
        case CODEC_ID_PCM_U8:
612
        case CODEC_ID_PCM_ALAW:
613
        case CODEC_ID_PCM_MULAW:
614 a190b7e9 Fabrice Bellard
            bitrate = enc->sample_rate * enc->channels * 8;
615 a96b68b7 Fabrice Bellard
            break;
616
        default:
617
            bitrate = enc->bit_rate;
618
            break;
619
        }
620 de6d9b64 Fabrice Bellard
        break;
621 985180a1 Fabrice Bellard
    case CODEC_TYPE_DATA:
622
        snprintf(buf, buf_size, "Data: %s", codec_name);
623
        bitrate = enc->bit_rate;
624
        break;
625 de6d9b64 Fabrice Bellard
    default:
626 b71472eb Philip Gladstone
        av_abort();
627 de6d9b64 Fabrice Bellard
    }
628 4bfad535 Fabrice Bellard
    if (encode) {
629
        if (enc->flags & CODEC_FLAG_PASS1)
630
            snprintf(buf + strlen(buf), buf_size - strlen(buf),
631
                     ", pass 1");
632
        if (enc->flags & CODEC_FLAG_PASS2)
633
            snprintf(buf + strlen(buf), buf_size - strlen(buf),
634
                     ", pass 2");
635
    }
636 a96b68b7 Fabrice Bellard
    if (bitrate != 0) {
637 de6d9b64 Fabrice Bellard
        snprintf(buf + strlen(buf), buf_size - strlen(buf), 
638 a96b68b7 Fabrice Bellard
                 ", %d kb/s", bitrate / 1000);
639 de6d9b64 Fabrice Bellard
    }
640
}
641
642 156e5023 Nick Kurshev
unsigned avcodec_version( void )
643
{
644
  return LIBAVCODEC_VERSION_INT;
645
}
646 cf087595 Fabrice Bellard
647 8bceb6af Nick Kurshev
unsigned avcodec_build( void )
648
{
649
  return LIBAVCODEC_BUILD;
650
}
651
652 de6d9b64 Fabrice Bellard
/* must be called before any other functions */
653
void avcodec_init(void)
654
{
655 0344cd0a Alex Beregszaszi
    static int inited = 0;
656
657
    if (inited != 0)
658
        return;
659
    inited = 1;
660
661 59cf08ce Fabrice Bellard
    dsputil_static_init();
662 de6d9b64 Fabrice Bellard
}
663
664 7a06ff14 Michael Niedermayer
/**
665
 * Flush buffers, should be called when seeking or when swicthing to a different stream.
666
 */
667 1c2a8c7f Michael Niedermayer
void avcodec_flush_buffers(AVCodecContext *avctx)
668
{
669 7a06ff14 Michael Niedermayer
    if(avctx->codec->flush)
670
        avctx->codec->flush(avctx);
671 1c2a8c7f Michael Niedermayer
}
672
673 d90cf87b Michael Niedermayer
void avcodec_default_free_buffers(AVCodecContext *s){
674
    int i, j;
675
676
    if(s->internal_buffer==NULL) return;
677
    
678
    for(i=0; i<INTERNAL_BUFFER_SIZE; i++){
679
        InternalBuffer *buf= &((InternalBuffer*)s->internal_buffer)[i];
680
        for(j=0; j<4; j++){
681
            av_freep(&buf->base[j]);
682
            buf->data[j]= NULL;
683
        }
684
    }
685
    av_freep(&s->internal_buffer);
686
    
687
    s->internal_buffer_count=0;
688
}
689
690 d8085ea7 Michael Niedermayer
char av_get_pict_type_char(int pict_type){
691
    switch(pict_type){
692
    case I_TYPE: return 'I'; 
693
    case P_TYPE: return 'P'; 
694
    case B_TYPE: return 'B'; 
695
    case S_TYPE: return 'S'; 
696
    case SI_TYPE:return 'i'; 
697
    case SP_TYPE:return 'p'; 
698
    default:     return '?';
699
    }
700
}
701
702 14bea432 Michael Niedermayer
int av_reduce(int *dst_nom, int *dst_den, int64_t nom, int64_t den, int64_t max){
703
    int exact=1, sign=0;
704 5ed9f2e5 Michael Niedermayer
    int64_t gcd;
705 14bea432 Michael Niedermayer
706
    assert(den != 0);
707
708
    if(den < 0){
709
        den= -den;
710
        nom= -nom;
711
    }
712
    
713
    if(nom < 0){
714
        nom= -nom;
715
        sign= 1;
716
    }
717
    
718 5ed9f2e5 Michael Niedermayer
    gcd = ff_gcd(nom, den);
719
    nom /= gcd;
720
    den /= gcd;
721 14bea432 Michael Niedermayer
    
722 5ed9f2e5 Michael Niedermayer
    if(nom > max || den > max){
723
        AVRational a0={0,1}, a1={1,0};
724
        exact=0;
725
726
        for(;;){
727
            int64_t x= nom / den;
728
            int64_t a2n= x*a1.num + a0.num;
729
            int64_t a2d= x*a1.den + a0.den;
730
731
            if(a2n > max || a2d > max) break;
732
733
            nom %= den;
734
        
735
            a0= a1;
736
            a1= (AVRational){a2n, a2d};
737
            if(nom==0) break;
738
            x= nom; nom=den; den=x;
739
        }
740
        nom= a1.num;
741
        den= a1.den;
742 14bea432 Michael Niedermayer
    }
743
    
744 5ed9f2e5 Michael Niedermayer
    assert(ff_gcd(nom, den) == 1);
745
    
746 14bea432 Michael Niedermayer
    if(sign) nom= -nom;
747
    
748
    *dst_nom = nom;
749
    *dst_den = den;
750
    
751
    return exact;
752
}
753
754
int64_t av_rescale(int64_t a, int b, int c){
755
    uint64_t h, l;
756
    assert(c > 0);
757
    assert(b >=0);
758
    
759
    if(a<0) return -av_rescale(-a, b, c);
760
    
761
    h= a>>32;
762
    if(h==0) return a*b/c;
763
    
764
    l= a&0xFFFFFFFF;
765
    l *= b;
766
    h *= b;
767
768
    l += (h%c)<<32;
769
770
    return ((h/c)<<32) + l/c;
771
}
772 9b879566 Michel Bardiaux
773
/* av_log API */
774
775
#ifdef AV_LOG_TRAP_PRINTF
776
#undef stderr
777
#undef fprintf
778
#endif
779
780
static int av_log_level = AV_LOG_DEBUG;
781
782
static void av_log_default_callback(AVCodecContext* avctx, int level, const char* fmt, va_list vl)
783
{
784 89b9441a Michael Niedermayer
    static int print_prefix=1;
785
786 9b879566 Michel Bardiaux
    if(level>av_log_level)
787
            return;
788 89b9441a Michael Niedermayer
    if(avctx && print_prefix)
789 9b879566 Michel Bardiaux
        fprintf(stderr, "[%s @ %p]", avctx->codec->name, avctx);
790 89b9441a Michael Niedermayer
        
791
    print_prefix= (int)strstr(fmt, "\n");
792
        
793 9b879566 Michel Bardiaux
    vfprintf(stderr, fmt, vl);
794
}
795
796
static void (*av_log_callback)(AVCodecContext*, int, const char*, va_list) = av_log_default_callback;
797
798
void av_log(AVCodecContext* avctx, int level, const char *fmt, ...)
799
{
800
    va_list vl;
801
    va_start(vl, fmt);
802
    av_vlog(avctx, level, fmt, vl);
803
    va_end(vl);
804
}
805
806
void av_vlog(AVCodecContext* avctx, int level, const char *fmt, va_list vl)
807
{
808
    av_log_callback(avctx, level, fmt, vl);
809
}
810
811
int av_log_get_level(void)
812
{
813
    return av_log_level;
814
}
815
816
void av_log_set_level(int level)
817
{
818
    av_log_level = level;
819
}
820
821
void av_log_set_callback(void (*callback)(AVCodecContext*, int, const char*, va_list))
822
{
823
    av_log_callback = callback;
824
}