Cygwin's mangling by Felix Buenemann <atmosfear@users.sourceforge.net>
[ffmpeg.git] / libavcodec / common.h
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__) && !defined(__CYGWIN__)
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