Statistics
| Branch: | Revision:

ffmpeg / libavcodec / common.h @ 05f361f0

History | View | Annotate | Download (13 KB)

1
#ifndef COMMON_H
2
#define COMMON_H
3

    
4
#define FFMPEG_VERSION_INT 0x000406
5
#define FFMPEG_VERSION     "0.4.6"
6

    
7
#if defined(WIN32) && !defined(__MINGW32__)
8
#define CONFIG_WIN32
9
#endif
10

    
11
//#define ALT_BITSTREAM_READER
12
//#define ALIGNED_BITSTREAM
13
#define FAST_GET_FIRST_VLC
14

    
15
#ifdef HAVE_AV_CONFIG_H
16
/* only include the following when compiling package */
17
#include "../config.h"
18

    
19
#include <stdlib.h>
20
#include <stdio.h>
21
#include <string.h>
22
#include <errno.h>
23

    
24
#ifndef ENODATA
25
#define ENODATA  61
26
#endif
27

    
28
#endif
29

    
30
#ifdef CONFIG_WIN32
31

    
32
/* windows */
33

    
34
typedef unsigned short UINT16;
35
typedef signed short INT16;
36
typedef unsigned char UINT8;
37
typedef unsigned int UINT32;
38
typedef unsigned __int64 UINT64;
39
typedef signed char INT8;
40
typedef signed int INT32;
41
typedef signed __int64 INT64;
42

    
43
typedef UINT8 uint8_t;
44
typedef INT8 int8_t;
45
typedef UINT16 uint16_t;
46
typedef INT16 int16_t;
47
typedef UINT32 uint32_t;
48
typedef INT32 int32_t;
49

    
50
#ifndef __MINGW32__
51
#define INT64_C(c)     (c ## i64)
52
#define UINT64_C(c)    (c ## i64)
53

    
54
#define inline __inline
55

    
56
/*
57
  Disable warning messages:
58
    warning C4244: '=' : conversion from 'double' to 'float', possible loss of data
59
    warning C4305: 'argument' : truncation from 'const double' to 'float'
60
*/
61
#pragma warning( disable : 4244 )
62
#pragma warning( disable : 4305 )
63

    
64
#else
65
#define INT64_C(c)     (c ## LL)
66
#define UINT64_C(c)    (c ## ULL)
67
#endif /* __MINGW32__ */
68

    
69
#define M_PI    3.14159265358979323846
70
#define M_SQRT2 1.41421356237309504880  /* sqrt(2) */
71

    
72
#ifdef _DEBUG
73
#define DEBUG
74
#endif
75

    
76
// code from bits/byteswap.h (C) 1997, 1998 Free Software Foundation, Inc.
77
#define bswap_32(x) \
78
     ((((x) & 0xff000000) >> 24) | (((x) & 0x00ff0000) >>  8) | \
79
      (((x) & 0x0000ff00) <<  8) | (((x) & 0x000000ff) << 24))
80
#define be2me_32(x) bswap_32(x)
81

    
82
#define snprintf _snprintf
83

    
84
#ifndef __MINGW32__
85
/* no config.h with VC */
86
#define CONFIG_ENCODERS 1
87
#define CONFIG_DECODERS 1
88
#define CONFIG_AC3      1
89
#endif
90

    
91
#else
92

    
93
/* unix */
94

    
95
#include <inttypes.h>
96

    
97
#ifndef __WINE_WINDEF16_H
98
/* workaround for typedef conflict in MPlayer (wine typedefs) */
99
typedef unsigned short UINT16;
100
typedef signed short INT16;
101
#endif
102

    
103
typedef unsigned char UINT8;
104
typedef unsigned int UINT32;
105
typedef unsigned long long UINT64;
106
typedef signed char INT8;
107
typedef signed int INT32;
108
typedef signed long long INT64;
109

    
110
#ifdef HAVE_AV_CONFIG_H
111

    
112
#ifdef __FreeBSD__
113
#include <sys/param.h>
114
#endif
115

    
116
#ifndef INT64_C
117
#define INT64_C(c)     (c ## LL)
118
#define UINT64_C(c)    (c ## ULL)
119
#endif
120

    
121
#include "../bswap.h"
122

    
123
#ifdef USE_FASTMEMCPY
124
#include "fastmemcpy.h"
125
#endif
126

    
127
#endif /* HAVE_AV_CONFIG_H */
128

    
129
#endif /* !CONFIG_WIN32 */
130

    
131

    
132
/* debug stuff */
133
#ifdef HAVE_AV_CONFIG_H
134

    
135
#ifndef DEBUG
136
#define NDEBUG
137
#endif
138
#include <assert.h>
139

    
140
/* dprintf macros */
141
#if defined(CONFIG_WIN32) && !defined(__MINGW32__)
142

    
143
inline void dprintf(const char* fmt,...) {}
144

    
145
#else
146

    
147
#ifdef DEBUG
148
#define dprintf(fmt,args...) printf(fmt, ## args)
149
#else
150
#define dprintf(fmt,args...)
151
#endif
152

    
153
#endif /* !CONFIG_WIN32 */
154

    
155
#endif /* HAVE_AV_CONFIG_H */
156

    
157
/* bit output */
158

    
159
struct PutBitContext;
160

    
161
typedef void (*WriteDataFunc)(void *, UINT8 *, int);
162

    
163
typedef struct PutBitContext {
164
    UINT32 bit_buf;
165
    int bit_cnt;
166
    UINT8 *buf, *buf_ptr, *buf_end;
167
    INT64 data_out_size; /* in bytes */
168
    void *opaque;
169
    WriteDataFunc write_data;
170
} PutBitContext;
171

    
172
void init_put_bits(PutBitContext *s, 
173
                   UINT8 *buffer, int buffer_size,
174
                   void *opaque,
175
                   void (*write_data)(void *, UINT8 *, int));
176
void put_bits(PutBitContext *s, int n, unsigned int value);
177
INT64 get_bit_count(PutBitContext *s); /* XXX: change function name */
178
void align_put_bits(PutBitContext *s);
179
void flush_put_bits(PutBitContext *s);
180

    
181
/* jpeg specific put_bits */
182
void jput_bits(PutBitContext *s, int n, unsigned int value);
183
void jflush_put_bits(PutBitContext *s);
184

    
185
/* bit input */
186

    
187
typedef struct GetBitContext {
188
#ifdef ALT_BITSTREAM_READER
189
    int index;
190
    UINT8 *buffer;
191
#else
192
    UINT32 bit_buf;
193
    int bit_cnt;
194
    UINT8 *buf, *buf_ptr, *buf_end;
195
#endif
196
} GetBitContext;
197

    
198
typedef struct VLC {
199
    int bits;
200
    INT16 *table_codes;
201
    INT8 *table_bits;
202
    int table_size, table_allocated;
203
} VLC;
204

    
205
/* used to avoid missaligned exceptions on some archs (alpha, ...) */
206
#ifdef ARCH_X86
207
#define unaligned32(a) (*(UINT32*)(a))
208
#else
209
#ifdef __GNUC__
210
static inline uint32_t unaligned32(const void *v) {
211
    struct Unaligned {
212
        uint32_t i;
213
    } __attribute__((packed));
214

    
215
    return ((const struct Unaligned *) v)->i;
216
}
217
#elif defined(__DECC)
218
static inline uint32_t unaligned32(const void *v) {
219
    return *(const __unaligned uint32_t *) v;
220
}
221
#else
222
static inline uint32_t unaligned32(const void *v) {
223
    return *(const uint32_t *) v;
224
}
225
#endif
226
#endif //!ARCH_X86
227

    
228
void init_get_bits(GetBitContext *s, 
229
                   UINT8 *buffer, int buffer_size);
230

    
231
#ifndef ALT_BITSTREAM_READER
232
unsigned int get_bits_long(GetBitContext *s, int n);
233
unsigned int show_bits_long(GetBitContext *s, int n);
234
#endif
235

    
236
static inline unsigned int get_bits(GetBitContext *s, int n){
237
#ifdef ALT_BITSTREAM_READER
238
#ifdef ALIGNED_BITSTREAM
239
    int index= s->index;
240
    uint32_t result1= be2me_32( ((uint32_t *)s->buffer)[index>>5] );
241
    uint32_t result2= be2me_32( ((uint32_t *)s->buffer)[(index>>5) + 1] );
242
#ifdef ARCH_X86
243
    asm ("shldl %%cl, %2, %0\n\t"
244
         : "=r" (result1)
245
         : "0" (result1), "r" (result2), "c" (index));
246
#else
247
    result1<<= (index&0x1F);
248
    result2= (result2>>1) >> (31-(index&0x1F));
249
    result1|= result2;
250
#endif
251
    result1>>= 32 - n;
252
    index+= n;
253
    s->index= index;
254
    
255
    return result1;
256
#else //ALIGNED_BITSTREAM
257
    int index= s->index;
258
    uint32_t result= be2me_32( unaligned32( ((uint8_t *)s->buffer)+(index>>3) ) );
259

    
260
    result<<= (index&0x07);
261
    result>>= 32 - n;
262
    index+= n;
263
    s->index= index;
264
    
265
    return result;
266
#endif //!ALIGNED_BITSTREAM
267
#else //ALT_BITSTREAM_READER
268
    if(s->bit_cnt>=n){
269
        /* most common case here */
270
        unsigned int val = s->bit_buf >> (32 - n);
271
        s->bit_buf <<= n;
272
        s->bit_cnt -= n;
273
#ifdef STATS
274
        st_bit_counts[st_current_index] += n;
275
#endif
276
        return val;
277
    }
278
    return get_bits_long(s,n);
279
#endif //!ALT_BITSTREAM_READER
280
}
281

    
282
static inline unsigned int get_bits1(GetBitContext *s){
283
#ifdef ALT_BITSTREAM_READER
284
    int index= s->index;
285
    uint8_t result= s->buffer[ index>>3 ];
286
    result<<= (index&0x07);
287
    result>>= 8 - 1;
288
    index++;
289
    s->index= index;
290
    
291
    return result;
292
#else
293
    if(s->bit_cnt>0){
294
        /* most common case here */
295
        unsigned int val = s->bit_buf >> 31;
296
        s->bit_buf <<= 1;
297
        s->bit_cnt--;
298
#ifdef STATS
299
        st_bit_counts[st_current_index]++;
300
#endif
301
        return val;
302
    }
303
    return get_bits_long(s,1);
304
#endif
305
}
306

    
307
/* This function is identical to get_bits(), the only */
308
/* diference is that it doesn't touch the buffer      */
309
/* it is usefull to see the buffer.                   */
310
static inline unsigned int show_bits(GetBitContext *s, int n)
311
{
312
#ifdef ALT_BITSTREAM_READER
313
#ifdef ALIGNED_BITSTREAM
314
    int index= s->index;
315
    uint32_t result1= be2me_32( ((uint32_t *)s->buffer)[index>>5] );
316
    uint32_t result2= be2me_32( ((uint32_t *)s->buffer)[(index>>5) + 1] );
317
#ifdef ARCH_X86
318
    asm ("shldl %%cl, %2, %0\n\t"
319
         : "=r" (result1)
320
         : "0" (result1), "r" (result2), "c" (index));
321
#else
322
    result1<<= (index&0x1F);
323
    result2= (result2>>1) >> (31-(index&0x1F));
324
    result1|= result2;
325
#endif
326
    result1>>= 32 - n;
327
    
328
    return result1;
329
#else //ALIGNED_BITSTREAM
330
    int index= s->index;
331
    uint32_t result= be2me_32( unaligned32( ((uint8_t *)s->buffer)+(index>>3) ) );
332

    
333
    result<<= (index&0x07);
334
    result>>= 32 - n;
335
    
336
    return result;
337
#endif //!ALIGNED_BITSTREAM
338
#else //ALT_BITSTREAM_READER
339
    if(s->bit_cnt>=n) {
340
        /* most common case here */
341
        unsigned int val = s->bit_buf >> (32 - n);
342
        return val;
343
    }
344
    return show_bits_long(s,n);
345
#endif //!ALT_BITSTREAM_READER
346
}
347

    
348
static inline void skip_bits(GetBitContext *s, int n){
349
#ifdef ALT_BITSTREAM_READER
350
    s->index+= n;
351
#else
352
    if(s->bit_cnt>=n){
353
        /* most common case here */
354
        s->bit_buf <<= n;
355
        s->bit_cnt -= n;
356
#ifdef STATS
357
        st_bit_counts[st_current_index] += n;
358
#endif
359
    } else {
360
        get_bits_long(s,n);
361
    }
362
#endif
363
}
364

    
365
static inline void skip_bits1(GetBitContext *s){
366
#ifdef ALT_BITSTREAM_READER
367
    s->index++;
368
#else
369
    if(s->bit_cnt>0){
370
        /* most common case here */
371
        s->bit_buf <<= 1;
372
        s->bit_cnt--;
373
#ifdef STATS
374
        st_bit_counts[st_current_index]++;
375
#endif
376
    } else {
377
        get_bits_long(s,1);
378
    }
379
#endif
380
}
381

    
382
static inline int get_bits_count(GetBitContext *s)
383
{
384
#ifdef ALT_BITSTREAM_READER
385
    return s->index;
386
#else
387
    return (s->buf_ptr - s->buf) * 8 - s->bit_cnt;
388
#endif
389
}
390

    
391
void align_get_bits(GetBitContext *s);
392
int init_vlc(VLC *vlc, int nb_bits, int nb_codes,
393
             const void *bits, int bits_wrap, int bits_size,
394
             const void *codes, int codes_wrap, int codes_size);
395
void free_vlc(VLC *vlc);
396

    
397
#ifdef ALT_BITSTREAM_READER
398
#ifdef ALIGNED_BITSTREAM
399
#ifdef ARCH_X86
400
#define SHOW_BITS(s, val, n) \
401
    val= be2me_32( ((uint32_t *)(s)->buffer)[bit_cnt>>5] );\
402
    {uint32_t result2= be2me_32( ((uint32_t *)(s)->buffer)[(bit_cnt>>5) + 1] );\
403
    asm ("shldl %%cl, %2, %0\n\t"\
404
         : "=r" (val)\
405
         : "0" (val), "r" (result2), "c" (bit_cnt));\
406
    ((uint32_t)val)>>= 32 - n;}
407
#else //ARCH_X86
408
#define SHOW_BITS(s, val, n) \
409
    val= be2me_32( ((uint32_t *)(s)->buffer)[bit_cnt>>5] );\
410
    {uint32_t result2= be2me_32( ((uint32_t *)(s)->buffer)[(bit_cnt>>5) + 1] );\
411
    val<<= (bit_cnt&0x1F);\
412
    result2= (result2>>1) >> (31-(bit_cnt&0x1F));\
413
    val|= result2;\
414
    ((uint32_t)val)>>= 32 - n;}
415
#endif //!ARCH_X86
416
#else //ALIGNED_BITSTREAM
417
#define SHOW_BITS(s, val, n) \
418
    val= be2me_32( unaligned32( ((uint8_t *)(s)->buffer)+(bit_cnt>>3) ) );\
419
    val<<= (bit_cnt&0x07);\
420
    ((uint32_t)val)>>= 32 - n;
421
#endif // !ALIGNED_BITSTREAM
422
#define FLUSH_BITS(n) bit_cnt+=n; 
423
#define SAVE_BITS(s) bit_cnt= (s)->index;
424
#define RESTORE_BITS(s) (s)->index= bit_cnt;
425
#else
426

    
427
/* macro to go faster */
428
/* n must be <= 24 */
429
/* XXX: optimize buffer end test */
430
#define SHOW_BITS(s, val, n)\
431
{\
432
    if (bit_cnt < n && buf_ptr < (s)->buf_end) {\
433
        bit_buf |= *buf_ptr++ << (24 - bit_cnt);\
434
        bit_cnt += 8;\
435
        if (bit_cnt < n && buf_ptr < (s)->buf_end) {\
436
            bit_buf |= *buf_ptr++ << (24 - bit_cnt);\
437
            bit_cnt += 8;\
438
            if (bit_cnt < n && buf_ptr < (s)->buf_end) {\
439
                bit_buf |= *buf_ptr++ << (24 - bit_cnt);\
440
                bit_cnt += 8;\
441
            }\
442
        }\
443
    }\
444
    val = bit_buf >> (32 - n);\
445
}
446

    
447
/* SHOW_BITS with n1 >= n must be been done before */
448
#define FLUSH_BITS(n)\
449
{\
450
    bit_buf <<= n;\
451
    bit_cnt -= n;\
452
}
453

    
454
#define SAVE_BITS(s) \
455
{\
456
    bit_cnt = (s)->bit_cnt;\
457
    bit_buf = (s)->bit_buf;\
458
    buf_ptr = (s)->buf_ptr;\
459
}
460

    
461
#define RESTORE_BITS(s) \
462
{\
463
    (s)->buf_ptr = buf_ptr;\
464
    (s)->bit_buf = bit_buf;\
465
    (s)->bit_cnt = bit_cnt;\
466
}
467
#endif // !ALT_BITSTREAM_READER
468

    
469
static inline int get_vlc(GetBitContext *s, VLC *vlc)
470
{
471
    int code, n, nb_bits, index;
472
    INT16 *table_codes;
473
    INT8 *table_bits;
474
    int bit_cnt;
475
#ifndef ALT_BITSTREAM_READER
476
    UINT32 bit_buf;
477
    UINT8 *buf_ptr;
478
#endif
479

    
480
    SAVE_BITS(s);
481
    nb_bits = vlc->bits;
482
    table_codes = vlc->table_codes;
483
    table_bits = vlc->table_bits;
484

    
485
#ifdef FAST_GET_FIRST_VLC
486
    SHOW_BITS(s, index, nb_bits);
487
    code = table_codes[index];
488
    n = table_bits[index];
489
    if (n > 0) {
490
        /* most common case (90%)*/
491
        FLUSH_BITS(n);
492
        RESTORE_BITS(s);
493
        return code;
494
    } else if (n == 0) {
495
        return -1;
496
    } else {
497
        FLUSH_BITS(nb_bits);
498
        nb_bits = -n;
499
        table_codes = vlc->table_codes + code;
500
        table_bits = vlc->table_bits + code;
501
    }
502
#endif
503
    for(;;) {
504
        SHOW_BITS(s, index, nb_bits);
505
        code = table_codes[index];
506
        n = table_bits[index];
507
        if (n > 0) {
508
            /* most common case */
509
            FLUSH_BITS(n);
510
#ifdef STATS
511
            st_bit_counts[st_current_index] += n;
512
#endif
513
            break;
514
        } else if (n == 0) {
515
            return -1;
516
        } else {
517
            FLUSH_BITS(nb_bits);
518
#ifdef STATS
519
            st_bit_counts[st_current_index] += nb_bits;
520
#endif
521
            nb_bits = -n;
522
            table_codes = vlc->table_codes + code;
523
            table_bits = vlc->table_bits + code;
524
        }
525
    }
526
    RESTORE_BITS(s);
527
    return code;
528
}
529

    
530

    
531
/* define it to include statistics code (useful only for optimizing
532
   codec efficiency */
533
//#define STATS
534

    
535
#ifdef STATS
536

    
537
enum {
538
    ST_UNKNOWN,
539
    ST_DC,
540
    ST_INTRA_AC,
541
    ST_INTER_AC,
542
    ST_INTRA_MB,
543
    ST_INTER_MB,
544
    ST_MV,
545
    ST_NB,
546
};
547

    
548
extern int st_current_index;
549
extern unsigned int st_bit_counts[ST_NB];
550
extern unsigned int st_out_bit_counts[ST_NB];
551

    
552
void print_stats(void);
553
#endif
554

    
555
/* misc math functions */
556

    
557
static inline int av_log2(unsigned int v)
558
{
559
    int n;
560

    
561
    n = 0;
562
    if (v & 0xffff0000) {
563
        v >>= 16;
564
        n += 16;
565
    }
566
    if (v & 0xff00) {
567
        v >>= 8;
568
        n += 8;
569
    }
570
    if (v & 0xf0) {
571
        v >>= 4;
572
        n += 4;
573
    }
574
    if (v & 0xc) {
575
        v >>= 2;
576
        n += 2;
577
    }
578
    if (v & 0x2) {
579
        n++;
580
    }
581
    return n;
582
}
583

    
584
/* memory */
585
void *av_mallocz(int size);
586

    
587
#endif