common.h 20.7 KB
Newer Older
F
Fabrice Bellard 已提交
1 2 3
#ifndef COMMON_H
#define COMMON_H

F
Fabrice Bellard 已提交
4 5
#define FFMPEG_VERSION_INT 0x000406
#define FFMPEG_VERSION     "0.4.6"
F
Fabrice Bellard 已提交
6

7
#if defined(WIN32) && !defined(__MINGW32__) && !defined(__CYGWIN__)
8
#    define CONFIG_WIN32
F
Fabrice Bellard 已提交
9 10
#endif

11
//#define ALT_BITSTREAM_WRITER
12
//#define ALIGNED_BITSTREAM_WRITER
13

14 15 16 17
#define ALT_BITSTREAM_READER
//#define LIBMPEG2_BITSTREAM_READER
//#define A32_BITSTREAM_READER

F
Fabrice Bellard 已提交
18
#ifdef HAVE_AV_CONFIG_H
F
Fabrice Bellard 已提交
19
/* only include the following when compiling package */
20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
#    include "config.h"

#    include <stdlib.h>
#    include <stdio.h>
#    include <string.h>
#    ifndef __BEOS__
#        include <errno.h>
#    else
#        include "berrno.h"
#    endif
#    include <math.h>

#    ifndef ENODATA
#        define ENODATA  61
#    endif
F
Fabrice Bellard 已提交
35

36
#endif /* HAVE_AV_CONFIG_H */
F
Fabrice Bellard 已提交
37

38 39
/* Suppress restrict if it was not defined in config.h.  */
#ifndef restrict
40
#    define restrict
41 42
#endif

F
Falk Hüffner 已提交
43
#if defined(__GNUC__) && (__GNUC__ > 3 || __GNUC__ == 3 && __GNUC_MINOR__ > 0)
44
#    define always_inline __attribute__((always_inline)) inline
F
Falk Hüffner 已提交
45
#else
46
#    define always_inline inline
F
Falk Hüffner 已提交
47 48
#endif

F
Fabrice Bellard 已提交
49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67
#ifdef CONFIG_WIN32

/* windows */

typedef unsigned short UINT16;
typedef signed short INT16;
typedef unsigned char UINT8;
typedef unsigned int UINT32;
typedef unsigned __int64 UINT64;
typedef signed char INT8;
typedef signed int INT32;
typedef signed __int64 INT64;

typedef UINT8 uint8_t;
typedef INT8 int8_t;
typedef UINT16 uint16_t;
typedef INT16 int16_t;
typedef UINT32 uint32_t;
typedef INT32 int32_t;
F
Fabrice Bellard 已提交
68 69
typedef UINT64 uint64_t;
typedef INT64 int64_t;
F
Fabrice Bellard 已提交
70

71 72 73
#    ifndef __MINGW32__
#        define INT64_C(c)     (c ## i64)
#        define UINT64_C(c)    (c ## i64)
F
Fabrice Bellard 已提交
74

75
#        define inline __inline
F
Fabrice Bellard 已提交
76

77 78 79 80
#    else
#        define INT64_C(c)     (c ## LL)
#        define UINT64_C(c)    (c ## ULL)
#    endif /* __MINGW32__ */
F
Fabrice Bellard 已提交
81

82 83
#    define M_PI    3.14159265358979323846
#    define M_SQRT2 1.41421356237309504880  /* sqrt(2) */
F
Fabrice Bellard 已提交
84

85 86 87
#    ifdef _DEBUG
#        define DEBUG
#    endif
F
Fabrice Bellard 已提交
88

89
#    define snprintf _snprintf
F
Fabrice Bellard 已提交
90

F
Fabrice Bellard 已提交
91
#else /* CONFIG_WIN32 */
F
Fabrice Bellard 已提交
92 93 94

/* unix */

95
#    include <inttypes.h>
F
Fabrice Bellard 已提交
96

97
#    ifndef __WINE_WINDEF16_H
98
/* workaround for typedef conflict in MPlayer (wine typedefs) */
F
Fabrice Bellard 已提交
99 100
typedef unsigned short UINT16;
typedef signed short INT16;
101
#    endif
F
Fabrice Bellard 已提交
102 103 104 105 106 107 108 109

typedef unsigned char UINT8;
typedef unsigned int UINT32;
typedef unsigned long long UINT64;
typedef signed char INT8;
typedef signed int INT32;
typedef signed long long INT64;

110 111 112 113 114
#    ifdef HAVE_AV_CONFIG_H
#        ifndef INT64_C
#            define INT64_C(c)     (c ## LL)
#            define UINT64_C(c)    (c ## ULL)
#        endif
F
Fabrice Bellard 已提交
115

116 117 118 119
#        ifdef USE_FASTMEMCPY
#            include "fastmemcpy.h"
#        endif
#    endif /* HAVE_AV_CONFIG_H */
F
Fabrice Bellard 已提交
120 121 122 123 124

#endif /* !CONFIG_WIN32 */

#ifdef HAVE_AV_CONFIG_H

125
#    include "bswap.h"
126

127 128 129 130 131 132
#    if defined(__MINGW32__) || defined(__CYGWIN__) || \
        defined(__OS2__) || defined (__OpenBSD__)
#        define MANGLE(a) "_" #a
#    else
#        define MANGLE(a) #a
#    endif
F
Fabrice Bellard 已提交
133

F
Fabrice Bellard 已提交
134 135
/* debug stuff */

136 137 138 139
#    ifndef DEBUG
#        define NDEBUG
#    endif
#    include <assert.h>
F
Fabrice Bellard 已提交
140

F
Fabrice Bellard 已提交
141
/* dprintf macros */
142
#    if defined(CONFIG_WIN32) && !defined(__MINGW32__)
F
Fabrice Bellard 已提交
143 144 145

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

146
#    else
F
Fabrice Bellard 已提交
147

148 149 150 151 152
#        ifdef DEBUG
#            define dprintf(fmt,args...) printf(fmt, ## args)
#        else
#            define dprintf(fmt,args...)
#        endif
F
Fabrice Bellard 已提交
153

154
#    endif /* !CONFIG_WIN32 */
F
Fabrice Bellard 已提交
155

156
#    define av_abort()      do { fprintf(stderr, "Abort at %s:%d\n", __FILE__, __LINE__); abort(); } while (0)
157

158 159
//rounded divison & shift
#define RSHIFT(a,b) ((a) > 0 ? ((a) + (1<<((b)-1)))>>(b) : ((a) + (1<<((b)-1))-1)>>(b))
160 161
/* assume b>0 */
#define ROUNDED_DIV(a,b) (((a)>0 ? (a) + ((b)>>1) : (a) - ((b)>>1))/(b))
M
Michael Niedermayer 已提交
162
#define ABS(a) ((a) >= 0 ? (a) : (-(a)))
163

M
Michael Niedermayer 已提交
164 165
#define FFMAX(a,b) ((a) > (b) ? (a) : (b))
#define FFMIN(a,b) ((a) > (b) ? (b) : (a))
166

167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182
#ifdef ARCH_X86
// avoid +32 for shift optimization (gcc should do that ...)
static inline  int32_t NEG_SSR32( int32_t a, int8_t s){
    asm ("sarl %1, %0\n\t"
         : "+r" (a)
         : "ic" ((uint8_t)(-s))
    );
    return a;
}
static inline uint32_t NEG_USR32(uint32_t a, int8_t s){
    asm ("shrl %1, %0\n\t"
         : "+r" (a)
         : "ic" ((uint8_t)(-s))
    );
    return a;
}
183
#else
184 185
#    define NEG_SSR32(a,s) ((( int32_t)(a))>>(32-(s)))
#    define NEG_USR32(a,s) (((uint32_t)(a))>>(32-(s)))
186 187
#endif

F
Fabrice Bellard 已提交
188 189 190 191 192 193 194
/* bit output */

struct PutBitContext;

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

typedef struct PutBitContext {
195 196 197 198
#ifdef ALT_BITSTREAM_WRITER
    UINT8 *buf, *buf_end;
    int index;
#else
F
Fabrice Bellard 已提交
199
    UINT32 bit_buf;
200
    int bit_left;
201
    UINT8 *buf, *buf_ptr, *buf_end;
202 203
#endif
    INT64 data_out_size; /* in bytes */
F
Fabrice Bellard 已提交
204 205 206 207 208 209
} PutBitContext;

void init_put_bits(PutBitContext *s, 
                   UINT8 *buffer, int buffer_size,
                   void *opaque,
                   void (*write_data)(void *, UINT8 *, int));
210

F
Fabrice Bellard 已提交
211
INT64 get_bit_count(PutBitContext *s); /* XXX: change function name */
F
Fabrice Bellard 已提交
212 213
void align_put_bits(PutBitContext *s);
void flush_put_bits(PutBitContext *s);
214
void put_string(PutBitContext * pbc, char *s);
F
Fabrice Bellard 已提交
215 216 217 218

/* bit input */

typedef struct GetBitContext {
219
    UINT8 *buffer, *buffer_end;
220 221
#ifdef ALT_BITSTREAM_READER
    int index;
222 223 224 225 226 227 228 229 230
#elif defined LIBMPEG2_BITSTREAM_READER
    UINT8 *buffer_ptr;
    UINT32 cache;
    int bit_count;
#elif defined A32_BITSTREAM_READER
    UINT32 *buffer_ptr;
    UINT32 cache0;
    UINT32 cache1;
    int bit_count;
231
#endif
232
    int size;
F
Fabrice Bellard 已提交
233 234
} GetBitContext;

235 236
static inline int get_bits_count(GetBitContext *s);

237 238
#define VLC_TYPE INT16

F
Fabrice Bellard 已提交
239 240
typedef struct VLC {
    int bits;
241
    VLC_TYPE (*table)[2]; // code, bits
F
Fabrice Bellard 已提交
242 243 244
    int table_size, table_allocated;
} VLC;

245 246 247 248 249 250
typedef struct RL_VLC_ELEM {
    int16_t level;
    int8_t len;
    uint8_t run;
} RL_VLC_ELEM;

251 252 253 254
#ifdef ARCH_SPARC64
#define UNALIGNED_STORES_ARE_BAD
#endif

255 256
/* used to avoid missaligned exceptions on some archs (alpha, ...) */
#ifdef ARCH_X86
257
#    define unaligned32(a) (*(UINT32*)(a))
258
#else
259
#    ifdef __GNUC__
260 261 262 263 264 265 266
static inline uint32_t unaligned32(const void *v) {
    struct Unaligned {
	uint32_t i;
    } __attribute__((packed));

    return ((const struct Unaligned *) v)->i;
}
267
#    elif defined(__DECC)
268 269 270
static inline uint32_t unaligned32(const void *v) {
    return *(const __unaligned uint32_t *) v;
}
271
#    else
272 273 274
static inline uint32_t unaligned32(const void *v) {
    return *(const uint32_t *) v;
}
275
#    endif
276 277
#endif //!ARCH_X86

278 279 280 281 282 283 284 285 286 287 288
#ifndef ALT_BITSTREAM_WRITER
static inline void put_bits(PutBitContext *s, int n, unsigned int value)
{
    unsigned int bit_buf;
    int bit_left;

#ifdef STATS
    st_out_bit_counts[st_current_index] += n;
#endif
    //    printf("put_bits=%d %x\n", n, value);
    assert(n == 32 || value < (1U << n));
289
    
290 291 292 293 294 295 296 297 298 299 300
    bit_buf = s->bit_buf;
    bit_left = s->bit_left;

    //    printf("n=%d value=%x cnt=%d buf=%x\n", n, value, bit_cnt, bit_buf);
    /* XXX: optimize */
    if (n < bit_left) {
        bit_buf = (bit_buf<<n) | value;
        bit_left-=n;
    } else {
	bit_buf<<=bit_left;
        bit_buf |= value >> (n - bit_left);
301 302 303 304 305 306 307 308
#ifdef UNALIGNED_STORES_ARE_BAD
        if (3 & (int) s->buf_ptr) {
            s->buf_ptr[0] = bit_buf >> 24;
            s->buf_ptr[1] = bit_buf >> 16;
            s->buf_ptr[2] = bit_buf >>  8;
            s->buf_ptr[3] = bit_buf      ;
        } else
#endif
309 310 311 312 313 314 315 316 317 318 319 320 321
        *(UINT32 *)s->buf_ptr = be2me_32(bit_buf);
        //printf("bitbuf = %08x\n", bit_buf);
        s->buf_ptr+=4;
	bit_left+=32 - n;
        bit_buf = value;
    }

    s->bit_buf = bit_buf;
    s->bit_left = bit_left;
}
#endif


322
#ifdef ALT_BITSTREAM_WRITER
323
static inline void put_bits(PutBitContext *s, int n, unsigned int value)
324
{
325 326
#    ifdef ALIGNED_BITSTREAM_WRITER
#        ifdef ARCH_X86
327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343
    asm volatile(
	"movl %0, %%ecx			\n\t"
	"xorl %%eax, %%eax		\n\t"
	"shrdl %%cl, %1, %%eax		\n\t"
	"shrl %%cl, %1			\n\t"
	"movl %0, %%ecx			\n\t"
	"shrl $3, %%ecx			\n\t"
	"andl $0xFFFFFFFC, %%ecx	\n\t"
	"bswapl %1			\n\t"
	"orl %1, (%2, %%ecx)		\n\t"
	"bswapl %%eax			\n\t"
	"addl %3, %0			\n\t"
	"movl %%eax, 4(%2, %%ecx)	\n\t"
	: "=&r" (s->index), "=&r" (value)
	: "r" (s->buf), "r" (n), "0" (s->index), "1" (value<<(-n))
	: "%eax", "%ecx"
    );
344
#        else
345 346 347 348 349 350 351 352 353 354
    int index= s->index;
    uint32_t *ptr= ((uint32_t *)s->buf)+(index>>5);
    
    value<<= 32-n; 
    
    ptr[0] |= be2me_32(value>>(index&31));
    ptr[1]  = be2me_32(value<<(32-(index&31)));
//if(n>24) printf("%d %d\n", n, value);
    index+= n;
    s->index= index;
355 356 357
#        endif
#    else //ALIGNED_BITSTREAM_WRITER
#        ifdef ARCH_X86
358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373
    asm volatile(
	"movl $7, %%ecx			\n\t"
	"andl %0, %%ecx			\n\t"
	"addl %3, %%ecx			\n\t"
	"negl %%ecx			\n\t"
	"shll %%cl, %1			\n\t"
	"bswapl %1			\n\t"
	"movl %0, %%ecx			\n\t"
	"shrl $3, %%ecx			\n\t"
	"orl %1, (%%ecx, %2)		\n\t"
	"addl %3, %0			\n\t"
	"movl $0, 4(%%ecx, %2)		\n\t"
	: "=&r" (s->index), "=&r" (value)
	: "r" (s->buf), "r" (n), "0" (s->index), "1" (value)
	: "%ecx"
    );
374
#        else
375 376 377 378 379 380 381 382
    int index= s->index;
    uint32_t *ptr= (uint32_t*)(((uint8_t *)s->buf)+(index>>3));
    
    ptr[0] |= be2me_32(value<<(32-n-(index&7) ));
    ptr[1] = 0;
//if(n>24) printf("%d %d\n", n, value);
    index+= n;
    s->index= index;
383 384
#        endif
#    endif //!ALIGNED_BITSTREAM_WRITER
385 386 387 388 389 390 391 392 393 394 395 396 397
}
#endif


static inline uint8_t* pbBufPtr(PutBitContext *s)
{
#ifdef ALT_BITSTREAM_WRITER
	return s->buf + (s->index>>3);
#else
	return s->buf_ptr;
#endif
}

398 399 400
/* Bitstream reader API docs:
name
    abritary name which is used as prefix for the internal variables
F
Fabrice Bellard 已提交
401

402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441
gb
    getbitcontext

OPEN_READER(name, gb)
    loads gb into local variables

CLOSE_READER(name, gb)
    stores local vars in gb

UPDATE_CACHE(name, gb)
    refills the internal cache from the bitstream
    after this call at least MIN_CACHE_BITS will be available,

GET_CACHE(name, gb)
    will output the contents of the internal cache, next bit is MSB of 32 or 64 bit (FIXME 64bit)

SHOW_UBITS(name, gb, num)
    will return the nest num bits

SHOW_SBITS(name, gb, num)
    will return the nest num bits and do sign extension

SKIP_BITS(name, gb, num)
    will skip over the next num bits
    note, this is equinvalent to SKIP_CACHE; SKIP_COUNTER

SKIP_CACHE(name, gb, num)
    will remove the next num bits from the cache (note SKIP_COUNTER MUST be called before UPDATE_CACHE / CLOSE_READER)

SKIP_COUNTER(name, gb, num)
    will increment the internal bit counter (see SKIP_CACHE & SKIP_BITS)

LAST_SKIP_CACHE(name, gb, num)
    will remove the next num bits from the cache if it is needed for UPDATE_CACHE otherwise it will do nothing

LAST_SKIP_BITS(name, gb, num)
    is equinvalent to SKIP_LAST_CACHE; SKIP_COUNTER

for examples see get_bits, show_bits, skip_bits, get_vlc
*/
442

443
#ifdef ALT_BITSTREAM_READER
444
#   define MIN_CACHE_BITS 25
445

446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472
#   define OPEN_READER(name, gb)\
        int name##_index= (gb)->index;\
        int name##_cache= 0;\

#   define CLOSE_READER(name, gb)\
        (gb)->index= name##_index;\

#   define UPDATE_CACHE(name, gb)\
        name##_cache= be2me_32( unaligned32( ((uint8_t *)(gb)->buffer)+(name##_index>>3) ) ) << (name##_index&0x07);\

#   define SKIP_CACHE(name, gb, num)\
        name##_cache <<= (num);\

// FIXME name?
#   define SKIP_COUNTER(name, gb, num)\
        name##_index += (num);\

#   define SKIP_BITS(name, gb, num)\
        {\
            SKIP_CACHE(name, gb, num)\
            SKIP_COUNTER(name, gb, num)\
        }\

#   define LAST_SKIP_BITS(name, gb, num) SKIP_COUNTER(name, gb, num)
#   define LAST_SKIP_CACHE(name, gb, num) ;

#   define SHOW_UBITS(name, gb, num)\
473
        NEG_USR32(name##_cache, num)
474 475

#   define SHOW_SBITS(name, gb, num)\
476
        NEG_SSR32(name##_cache, num)
477 478 479 480 481 482

#   define GET_CACHE(name, gb)\
        ((uint32_t)name##_cache)

static inline int get_bits_count(GetBitContext *s){
    return s->index;
483
}
484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504
#elif defined LIBMPEG2_BITSTREAM_READER
//libmpeg2 like reader

#   define MIN_CACHE_BITS 16

#   define OPEN_READER(name, gb)\
        int name##_bit_count=(gb)->bit_count;\
        int name##_cache= (gb)->cache;\
        uint8_t * name##_buffer_ptr=(gb)->buffer_ptr;\

#   define CLOSE_READER(name, gb)\
        (gb)->bit_count= name##_bit_count;\
        (gb)->cache= name##_cache;\
        (gb)->buffer_ptr= name##_buffer_ptr;\

#   define UPDATE_CACHE(name, gb)\
    if(name##_bit_count > 0){\
        name##_cache+= ((name##_buffer_ptr[0]<<8) + name##_buffer_ptr[1]) << name##_bit_count;\
        name##_buffer_ptr+=2;\
        name##_bit_count-= 16;\
    }\
505

506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521
#   define SKIP_CACHE(name, gb, num)\
        name##_cache <<= (num);\

#   define SKIP_COUNTER(name, gb, num)\
        name##_bit_count += (num);\

#   define SKIP_BITS(name, gb, num)\
        {\
            SKIP_CACHE(name, gb, num)\
            SKIP_COUNTER(name, gb, num)\
        }\

#   define LAST_SKIP_BITS(name, gb, num) SKIP_BITS(name, gb, num)
#   define LAST_SKIP_CACHE(name, gb, num) SKIP_CACHE(name, gb, num)

#   define SHOW_UBITS(name, gb, num)\
522
        NEG_USR32(name##_cache, num)
523 524

#   define SHOW_SBITS(name, gb, num)\
525
        NEG_SSR32(name##_cache, num)
526 527 528 529 530 531

#   define GET_CACHE(name, gb)\
        ((uint32_t)name##_cache)

static inline int get_bits_count(GetBitContext *s){
    return (s->buffer_ptr - s->buffer)*8 - 16 + s->bit_count;
532 533
}

534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552
#elif defined A32_BITSTREAM_READER

#   define MIN_CACHE_BITS 32

#   define OPEN_READER(name, gb)\
        int name##_bit_count=(gb)->bit_count;\
        uint32_t name##_cache0= (gb)->cache0;\
        uint32_t name##_cache1= (gb)->cache1;\
        uint32_t * name##_buffer_ptr=(gb)->buffer_ptr;\

#   define CLOSE_READER(name, gb)\
        (gb)->bit_count= name##_bit_count;\
        (gb)->cache0= name##_cache0;\
        (gb)->cache1= name##_cache1;\
        (gb)->buffer_ptr= name##_buffer_ptr;\

#   define UPDATE_CACHE(name, gb)\
    if(name##_bit_count > 0){\
        const uint32_t next= be2me_32( *name##_buffer_ptr );\
553
        name##_cache0 |= NEG_USR32(next,name##_bit_count);\
554 555 556 557 558
        name##_cache1 |= next<<name##_bit_count;\
        name##_buffer_ptr++;\
        name##_bit_count-= 32;\
    }\

559
#ifdef ARCH_X86
560 561 562 563 564 565 566
#   define SKIP_CACHE(name, gb, num)\
        asm(\
            "shldl %2, %1, %0		\n\t"\
            "shll %2, %1		\n\t"\
            : "+r" (name##_cache0), "+r" (name##_cache1)\
            : "Ic" ((uint8_t)num)\
           );
567
#else
568 569
#   define SKIP_CACHE(name, gb, num)\
        name##_cache0 <<= (num);\
570
        name##_cache0 |= NEG_USR32(name##_cache1,num);\
571
        name##_cache1 <<= (num);
572
#endif
573

574 575 576 577 578 579 580 581 582 583 584 585 586
#   define SKIP_COUNTER(name, gb, num)\
        name##_bit_count += (num);\

#   define SKIP_BITS(name, gb, num)\
        {\
            SKIP_CACHE(name, gb, num)\
            SKIP_COUNTER(name, gb, num)\
        }\

#   define LAST_SKIP_BITS(name, gb, num) SKIP_BITS(name, gb, num)
#   define LAST_SKIP_CACHE(name, gb, num) SKIP_CACHE(name, gb, num)

#   define SHOW_UBITS(name, gb, num)\
587
        NEG_USR32(name##_cache0, num)
588 589

#   define SHOW_SBITS(name, gb, num)\
590
        NEG_SSR32(name##_cache0, num)
591 592 593 594 595 596

#   define GET_CACHE(name, gb)\
        (name##_cache0)

static inline int get_bits_count(GetBitContext *s){
    return ((uint8_t*)s->buffer_ptr - s->buffer)*8 - 32 + s->bit_count;
597 598
}

599 600
#endif

601 602 603 604 605 606 607 608 609
static inline unsigned int get_bits(GetBitContext *s, int n){
    register int tmp;
    OPEN_READER(re, s)
    UPDATE_CACHE(re, s)
    tmp= SHOW_UBITS(re, s, n);
    LAST_SKIP_BITS(re, s, n)
    CLOSE_READER(re, s)
    return tmp;
}
610

611 612 613 614 615 616 617
static inline unsigned int show_bits(GetBitContext *s, int n){
    register int tmp;
    OPEN_READER(re, s)
    UPDATE_CACHE(re, s)
    tmp= SHOW_UBITS(re, s, n);
//    CLOSE_READER(re, s)
    return tmp;
618 619
}

620
static inline void skip_bits(GetBitContext *s, int n){
621 622 623 624 625 626 627 628
 //Note gcc seems to optimize this to s->index+=n for the ALT_READER :))
    OPEN_READER(re, s)
    UPDATE_CACHE(re, s)
    LAST_SKIP_BITS(re, s, n)
    CLOSE_READER(re, s)
}

static inline unsigned int get_bits1(GetBitContext *s){
629
#ifdef ALT_BITSTREAM_READER
630 631 632 633 634 635
    int index= s->index;
    uint8_t result= s->buffer[ index>>3 ];
    result<<= (index&0x07);
    result>>= 8 - 1;
    index++;
    s->index= index;
636

637
    return result;
638
#else
639
    return get_bits(s, 1);
640
#endif
641 642
}

643 644
static inline unsigned int show_bits1(GetBitContext *s){
    return show_bits(s, 1);
645 646
}

647 648
static inline void skip_bits1(GetBitContext *s){
    skip_bits(s, 1);
F
Fabrice Bellard 已提交
649
}
650

651 652 653
void init_get_bits(GetBitContext *s,
                   UINT8 *buffer, int buffer_size);

654
int check_marker(GetBitContext *s, const char *msg);
F
Fabrice Bellard 已提交
655 656 657 658 659 660
void align_get_bits(GetBitContext *s);
int init_vlc(VLC *vlc, int nb_bits, int nb_codes,
             const void *bits, int bits_wrap, int bits_size,
             const void *codes, int codes_wrap, int codes_size);
void free_vlc(VLC *vlc);

M
Michael Niedermayer 已提交
661 662 663 664 665 666 667 668 669 670 671 672 673 674
#define GET_VLC(code, name, gb, table, bits, max_depth)\
{\
    int n, index, nb_bits;\
\
    index= SHOW_UBITS(name, gb, bits);\
    code = table[index][0];\
    n    = table[index][1];\
\
    if(max_depth > 1 && n < 0){\
        LAST_SKIP_BITS(name, gb, bits)\
        UPDATE_CACHE(name, gb)\
\
        nb_bits = -n;\
\
675
        index= SHOW_UBITS(name, gb, nb_bits) + code;\
M
Michael Niedermayer 已提交
676 677 678 679 680 681 682 683
        code = table[index][0];\
        n    = table[index][1];\
        if(max_depth > 2 && n < 0){\
            LAST_SKIP_BITS(name, gb, nb_bits)\
            UPDATE_CACHE(name, gb)\
\
            nb_bits = -n;\
\
684
            index= SHOW_UBITS(name, gb, nb_bits) + code;\
M
Michael Niedermayer 已提交
685 686 687 688 689 690 691
            code = table[index][0];\
            n    = table[index][1];\
        }\
    }\
    SKIP_BITS(name, gb, n)\
}

692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713
#define GET_RL_VLC(level, run, name, gb, table, bits, max_depth)\
{\
    int n, index, nb_bits;\
\
    index= SHOW_UBITS(name, gb, bits);\
    level = table[index].level;\
    n     = table[index].len;\
\
    if(max_depth > 1 && n < 0){\
        LAST_SKIP_BITS(name, gb, bits)\
        UPDATE_CACHE(name, gb)\
\
        nb_bits = -n;\
\
        index= SHOW_UBITS(name, gb, nb_bits) + level;\
        level = table[index].level;\
        n     = table[index].len;\
    }\
    run= table[index].run;\
    SKIP_BITS(name, gb, n)\
}

M
Michael Niedermayer 已提交
714
// deprecated, dont use get_vlc for new code, use get_vlc2 instead or use GET_VLC directly
715 716
static inline int get_vlc(GetBitContext *s, VLC *vlc)
{
M
Michael Niedermayer 已提交
717 718 719
    int code;
    VLC_TYPE (*table)[2]= vlc->table;
    
720 721
    OPEN_READER(re, s)
    UPDATE_CACHE(re, s)
722

M
Michael Niedermayer 已提交
723
    GET_VLC(code, re, s, table, vlc->bits, 3)    
724

M
Michael Niedermayer 已提交
725 726 727 728
    CLOSE_READER(re, s)
    return code;
}

F
Falk Hüffner 已提交
729 730
static always_inline int get_vlc2(GetBitContext *s, VLC_TYPE (*table)[2],
                                  int bits, int max_depth)
M
Michael Niedermayer 已提交
731 732 733 734 735 736 737 738
{
    int code;
    
    OPEN_READER(re, s)
    UPDATE_CACHE(re, s)

    GET_VLC(code, re, s, table, bits, max_depth)

739
    CLOSE_READER(re, s)
740 741 742
    return code;
}

743

F
Fabrice Bellard 已提交
744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769
/* define it to include statistics code (useful only for optimizing
   codec efficiency */
//#define STATS

#ifdef STATS

enum {
    ST_UNKNOWN,
    ST_DC,
    ST_INTRA_AC,
    ST_INTER_AC,
    ST_INTRA_MB,
    ST_INTER_MB,
    ST_MV,
    ST_NB,
};

extern int st_current_index;
extern unsigned int st_bit_counts[ST_NB];
extern unsigned int st_out_bit_counts[ST_NB];

void print_stats(void);
#endif

/* misc math functions */

770
static inline int av_log2(unsigned int v)
F
Fabrice Bellard 已提交
771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796
{
    int n;

    n = 0;
    if (v & 0xffff0000) {
        v >>= 16;
        n += 16;
    }
    if (v & 0xff00) {
        v >>= 8;
        n += 8;
    }
    if (v & 0xf0) {
        v >>= 4;
        n += 4;
    }
    if (v & 0xc) {
        v >>= 2;
        n += 2;
    }
    if (v & 0x2) {
        n++;
    }
    return n;
}

797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814
/* median of 3 */
static inline int mid_pred(int a, int b, int c)
{
    int vmin, vmax;
    vmax = vmin = a;
    if (b < vmin)
        vmin = b;
    else
	vmax = b;

    if (c < vmin)
        vmin = c;
    else if (c > vmax)
        vmax = c;

    return a + b + c - vmin - vmax;
}

815 816 817 818 819 820 821 822 823 824
static inline int clip(int a, int amin, int amax)
{
    if (a < amin)
        return amin;
    else if (a > amax)
        return amax;
    else
        return a;
}

825
/* math */
M
Michael Niedermayer 已提交
826 827
extern const UINT8 ff_sqrt_tab[128];

828 829
int ff_gcd(int a, int b);

M
sqrt  
Michael Niedermayer 已提交
830 831 832 833 834
static inline int ff_sqrt(int a)
{
    int ret=0;
    int s;
    int ret_sq=0;
M
Michael Niedermayer 已提交
835 836 837
    
    if(a<128) return ff_sqrt_tab[a];
    
M
sqrt  
Michael Niedermayer 已提交
838 839 840 841 842 843 844 845 846
    for(s=15; s>=0; s--){
        int b= ret_sq + (1<<(s*2)) + (ret<<s)*2;
        if(b<=a){
            ret_sq=b;
            ret+= 1<<s;
        }
    }
    return ret;
}
847 848 849 850

/**
 * converts fourcc string to int
 */
Z
Zdenek Kabelac 已提交
851
static inline int ff_get_fourcc(const char *s){
852 853 854 855 856
    assert( strlen(s)==4 );
    
    return (s[0]) + (s[1]<<8) + (s[2]<<16) + (s[3]<<24);
}

M
Michael Niedermayer 已提交
857 858
void ff_float2fraction(int *nom_arg, int *denom_arg, double f, int max);

859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874

#ifdef ARCH_X86
#define MASK_ABS(mask, level)\
            asm volatile(\
		"cdq			\n\t"\
		"xorl %1, %0		\n\t"\
		"subl %1, %0		\n\t"\
		: "+a" (level), "=&d" (mask)\
	    );
#else
#define MASK_ABS(mask, level)\
            mask= level>>31;\
            level= (level^mask)-mask;
#endif


M
sqrt  
Michael Niedermayer 已提交
875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893
#if __CPU__ >= 686 && !defined(RUNTIME_CPUDETECT)
#define COPY3_IF_LT(x,y,a,b,c,d)\
asm volatile (\
    "cmpl %0, %3	\n\t"\
    "cmovl %3, %0	\n\t"\
    "cmovl %4, %1	\n\t"\
    "cmovl %5, %2	\n\t"\
    : "+r" (x), "+r" (a), "+r" (c)\
    : "r" (y), "r" (b), "r" (d)\
);
#else
#define COPY3_IF_LT(x,y,a,b,c,d)\
if((y)<(x)){\
     (x)=(y);\
     (a)=(b);\
     (c)=(d);\
}
#endif

894 895
#define CLAMP_TO_8BIT(d) ((d > 0xff) ? 0xff : (d < 0) ? 0 : d)

896 897 898
#endif /* HAVE_AV_CONFIG_H */

#endif /* COMMON_H */