common.h 22.6 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

F
Fabrice Bellard 已提交
36 37 38 39
#ifndef M_PI
#define M_PI    3.14159265358979323846
#endif

40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57
#include <stddef.h>
#ifndef offsetof
# define offsetof(T,F) ((unsigned int)((char *)&((T *)0)->F))
#endif

#define AVOPTION_CODEC_BOOL(name, help, field) \
    { name, help, offsetof(AVCodecContext, field), FF_OPT_TYPE_BOOL }
#define AVOPTION_CODEC_FLAG(name, help, field, flag, defval) \
    { name, help, offsetof(AVCodecContext, field), FF_OPT_TYPE_FLAG, flag, 0, defval }
#define AVOPTION_CODEC_INT(name, help, field, minv, maxv, defval) \
    { name, help, offsetof(AVCodecContext, field), FF_OPT_TYPE_INT, minv, maxv, defval }
#define AVOPTION_CODEC_STRING(name, help, field, str, val) \
    { name, help, offsetof(AVCodecContext, field), FF_OPT_TYPE_STRING, .defval = val, .defstr = str }
#define AVOPTION_CODEC_RCOVERRIDE(name, help, field) \
    { name, help, offsetof(AVCodecContext, field), FF_OPT_TYPE_RCOVERRIDE, .defval = 0, .defstr = NULL }
#define AVOPTION_SUB(ptr) { .name = NULL, .sub = ptr }
#define AVOPTION_END() AVOPTION_SUB(NULL)

58
#endif /* HAVE_AV_CONFIG_H */
F
Fabrice Bellard 已提交
59

60 61
/* Suppress restrict if it was not defined in config.h.  */
#ifndef restrict
62
#    define restrict
63 64
#endif

F
Falk Hüffner 已提交
65
#if defined(__GNUC__) && (__GNUC__ > 3 || __GNUC__ == 3 && __GNUC_MINOR__ > 0)
66
#    define always_inline __attribute__((always_inline)) inline
F
Falk Hüffner 已提交
67
#else
68
#    define always_inline inline
F
Falk Hüffner 已提交
69 70
#endif

F
Fabrice Bellard 已提交
71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89
#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 已提交
90 91
typedef UINT64 uint64_t;
typedef INT64 int64_t;
F
Fabrice Bellard 已提交
92

93 94 95
#    ifndef __MINGW32__
#        define INT64_C(c)     (c ## i64)
#        define UINT64_C(c)    (c ## i64)
F
Fabrice Bellard 已提交
96

97
#        define inline __inline
F
Fabrice Bellard 已提交
98

99 100 101 102
#    else
#        define INT64_C(c)     (c ## LL)
#        define UINT64_C(c)    (c ## ULL)
#    endif /* __MINGW32__ */
F
Fabrice Bellard 已提交
103

104 105 106
#    ifdef _DEBUG
#        define DEBUG
#    endif
F
Fabrice Bellard 已提交
107

108
#    define snprintf _snprintf
F
Fabrice Bellard 已提交
109
#    define vsnprintf _vsnprintf
F
Fabrice Bellard 已提交
110

111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142
/* CONFIG_WIN32 end */
#elif defined (CONFIG_OS2)
/* OS/2 EMX */

#include <inttypes.h>

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

#ifdef HAVE_AV_CONFIG_H

#ifndef INT64_C
#define INT64_C(c)     (c ## LL)
#define UINT64_C(c)    (c ## ULL)
#endif

#ifdef USE_FASTMEMCPY
#include "fastmemcpy.h"
#endif

#include <float.h>

#endif /* HAVE_AV_CONFIG_H */

/* CONFIG_OS2 end */
#else
F
Fabrice Bellard 已提交
143 144 145

/* unix */

146
#    include <inttypes.h>
F
Fabrice Bellard 已提交
147

148
#    ifndef __WINE_WINDEF16_H
149
/* workaround for typedef conflict in MPlayer (wine typedefs) */
F
Fabrice Bellard 已提交
150 151
typedef unsigned short UINT16;
typedef signed short INT16;
152
#    endif
F
Fabrice Bellard 已提交
153 154 155 156 157 158 159 160

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;

161 162 163 164 165
#    ifdef HAVE_AV_CONFIG_H
#        ifndef INT64_C
#            define INT64_C(c)     (c ## LL)
#            define UINT64_C(c)    (c ## ULL)
#        endif
F
Fabrice Bellard 已提交
166

167 168 169 170
#        ifdef USE_FASTMEMCPY
#            include "fastmemcpy.h"
#        endif
#    endif /* HAVE_AV_CONFIG_H */
F
Fabrice Bellard 已提交
171

172
#endif /* !CONFIG_WIN32 && !CONFIG_OS2 */
F
Fabrice Bellard 已提交
173 174 175

#ifdef HAVE_AV_CONFIG_H

176
#    include "bswap.h"
177

178 179 180 181 182 183
#    if defined(__MINGW32__) || defined(__CYGWIN__) || \
        defined(__OS2__) || defined (__OpenBSD__)
#        define MANGLE(a) "_" #a
#    else
#        define MANGLE(a) #a
#    endif
F
Fabrice Bellard 已提交
184

F
Fabrice Bellard 已提交
185 186
/* debug stuff */

187 188 189 190
#    ifndef DEBUG
#        define NDEBUG
#    endif
#    include <assert.h>
F
Fabrice Bellard 已提交
191

F
Fabrice Bellard 已提交
192
/* dprintf macros */
193
#    if defined(CONFIG_WIN32) && !defined(__MINGW32__)
F
Fabrice Bellard 已提交
194 195 196

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

197
#    else
F
Fabrice Bellard 已提交
198

199 200 201 202 203
#        ifdef DEBUG
#            define dprintf(fmt,args...) printf(fmt, ## args)
#        else
#            define dprintf(fmt,args...)
#        endif
F
Fabrice Bellard 已提交
204

205
#    endif /* !CONFIG_WIN32 */
F
Fabrice Bellard 已提交
206

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

209 210
//rounded divison & shift
#define RSHIFT(a,b) ((a) > 0 ? ((a) + (1<<((b)-1)))>>(b) : ((a) + (1<<((b)-1))-1)>>(b))
211 212
/* assume b>0 */
#define ROUNDED_DIV(a,b) (((a)>0 ? (a) + ((b)>>1) : (a) - ((b)>>1))/(b))
M
Michael Niedermayer 已提交
213
#define ABS(a) ((a) >= 0 ? (a) : (-(a)))
214

M
Michael Niedermayer 已提交
215 216
#define FFMAX(a,b) ((a) > (b) ? (a) : (b))
#define FFMIN(a,b) ((a) > (b) ? (b) : (a))
217

218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233
#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;
}
234
#else
235 236
#    define NEG_SSR32(a,s) ((( int32_t)(a))>>(32-(s)))
#    define NEG_USR32(a,s) (((uint32_t)(a))>>(32-(s)))
237 238
#endif

F
Fabrice Bellard 已提交
239 240 241 242 243 244 245
/* bit output */

struct PutBitContext;

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

typedef struct PutBitContext {
246 247 248 249
#ifdef ALT_BITSTREAM_WRITER
    UINT8 *buf, *buf_end;
    int index;
#else
F
Fabrice Bellard 已提交
250
    UINT32 bit_buf;
251
    int bit_left;
252
    UINT8 *buf, *buf_ptr, *buf_end;
253 254
#endif
    INT64 data_out_size; /* in bytes */
F
Fabrice Bellard 已提交
255 256 257 258 259 260
} PutBitContext;

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

F
Fabrice Bellard 已提交
262
INT64 get_bit_count(PutBitContext *s); /* XXX: change function name */
F
Fabrice Bellard 已提交
263 264
void align_put_bits(PutBitContext *s);
void flush_put_bits(PutBitContext *s);
265
void put_string(PutBitContext * pbc, char *s);
F
Fabrice Bellard 已提交
266 267 268 269

/* bit input */

typedef struct GetBitContext {
270
    UINT8 *buffer, *buffer_end;
271 272
#ifdef ALT_BITSTREAM_READER
    int index;
273 274 275 276 277 278 279 280 281
#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;
282
#endif
283
    int size_in_bits;
F
Fabrice Bellard 已提交
284 285
} GetBitContext;

286 287
static inline int get_bits_count(GetBitContext *s);

288 289
#define VLC_TYPE INT16

F
Fabrice Bellard 已提交
290 291
typedef struct VLC {
    int bits;
292
    VLC_TYPE (*table)[2]; // code, bits
F
Fabrice Bellard 已提交
293 294 295
    int table_size, table_allocated;
} VLC;

296 297 298 299 300 301
typedef struct RL_VLC_ELEM {
    int16_t level;
    int8_t len;
    uint8_t run;
} RL_VLC_ELEM;

302 303 304 305
#ifdef ARCH_SPARC64
#define UNALIGNED_STORES_ARE_BAD
#endif

306 307
/* used to avoid missaligned exceptions on some archs (alpha, ...) */
#ifdef ARCH_X86
308
#    define unaligned32(a) (*(UINT32*)(a))
309
#else
310
#    ifdef __GNUC__
311 312 313 314 315 316 317
static inline uint32_t unaligned32(const void *v) {
    struct Unaligned {
	uint32_t i;
    } __attribute__((packed));

    return ((const struct Unaligned *) v)->i;
}
318
#    elif defined(__DECC)
319 320 321
static inline uint32_t unaligned32(const void *v) {
    return *(const __unaligned uint32_t *) v;
}
322
#    else
323 324 325
static inline uint32_t unaligned32(const void *v) {
    return *(const uint32_t *) v;
}
326
#    endif
327 328
#endif //!ARCH_X86

329 330 331 332 333 334 335 336 337 338 339
#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));
340
    
341 342 343 344 345 346 347 348 349 350 351
    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);
352 353 354 355 356 357 358 359
#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
360 361 362 363 364 365 366 367 368 369 370 371 372
        *(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


373
#ifdef ALT_BITSTREAM_WRITER
374
static inline void put_bits(PutBitContext *s, int n, unsigned int value)
375
{
376 377
#    ifdef ALIGNED_BITSTREAM_WRITER
#        ifdef ARCH_X86
378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394
    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"
    );
395
#        else
396 397 398 399 400 401 402 403 404 405
    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;
406 407 408
#        endif
#    else //ALIGNED_BITSTREAM_WRITER
#        ifdef ARCH_X86
409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424
    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"
    );
425
#        else
426 427 428 429 430 431 432 433
    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;
434 435
#        endif
#    endif //!ALIGNED_BITSTREAM_WRITER
436 437 438 439 440 441 442 443 444 445 446 447 448
}
#endif


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

449 450 451
/* Bitstream reader API docs:
name
    abritary name which is used as prefix for the internal variables
F
Fabrice Bellard 已提交
452

453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492
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
*/
493

494
#ifdef ALT_BITSTREAM_READER
495
#   define MIN_CACHE_BITS 25
496

497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523
#   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)\
524
        NEG_USR32(name##_cache, num)
525 526

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

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

static inline int get_bits_count(GetBitContext *s){
    return s->index;
534
}
535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555
#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;\
    }\
556

557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572
#   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)\
573
        NEG_USR32(name##_cache, num)
574 575

#   define SHOW_SBITS(name, gb, num)\
576
        NEG_SSR32(name##_cache, num)
577 578 579 580 581 582

#   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;
583 584
}

585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603
#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 );\
604
        name##_cache0 |= NEG_USR32(next,name##_bit_count);\
605 606 607 608 609
        name##_cache1 |= next<<name##_bit_count;\
        name##_buffer_ptr++;\
        name##_bit_count-= 32;\
    }\

610
#ifdef ARCH_X86
611 612 613 614 615 616 617
#   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)\
           );
618
#else
619 620
#   define SKIP_CACHE(name, gb, num)\
        name##_cache0 <<= (num);\
621
        name##_cache0 |= NEG_USR32(name##_cache1,num);\
622
        name##_cache1 <<= (num);
623
#endif
624

625 626 627 628 629 630 631 632 633 634 635 636 637
#   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)\
638
        NEG_USR32(name##_cache0, num)
639 640

#   define SHOW_SBITS(name, gb, num)\
641
        NEG_SSR32(name##_cache0, num)
642 643 644 645 646 647

#   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;
648 649
}

650 651
#endif

652 653 654 655 656 657 658 659 660
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;
}
661

662 663 664 665 666 667 668
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;
669 670
}

671
static inline void skip_bits(GetBitContext *s, int n){
672 673 674 675 676 677 678 679
 //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){
680
#ifdef ALT_BITSTREAM_READER
681 682 683 684 685 686
    int index= s->index;
    uint8_t result= s->buffer[ index>>3 ];
    result<<= (index&0x07);
    result>>= 8 - 1;
    index++;
    s->index= index;
687

688
    return result;
689
#else
690
    return get_bits(s, 1);
691
#endif
692 693
}

694 695
static inline unsigned int show_bits1(GetBitContext *s){
    return show_bits(s, 1);
696 697
}

698 699
static inline void skip_bits1(GetBitContext *s){
    skip_bits(s, 1);
F
Fabrice Bellard 已提交
700
}
701

702 703 704
void init_get_bits(GetBitContext *s,
                   UINT8 *buffer, int buffer_size);

705
int check_marker(GetBitContext *s, const char *msg);
F
Fabrice Bellard 已提交
706 707 708 709 710 711
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);

712 713 714 715 716 717
/**
 *
 * if the vlc code is invalid and max_depth=1 than no bits will be removed
 * if the vlc code is invalid and max_depth>1 than the number of bits removed
 * is undefined
 */
M
Michael Niedermayer 已提交
718 719 720 721 722 723 724 725 726 727 728 729 730 731
#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;\
\
732
        index= SHOW_UBITS(name, gb, nb_bits) + code;\
M
Michael Niedermayer 已提交
733 734 735 736 737 738 739 740
        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;\
\
741
            index= SHOW_UBITS(name, gb, nb_bits) + code;\
M
Michael Niedermayer 已提交
742 743 744 745 746 747 748
            code = table[index][0];\
            n    = table[index][1];\
        }\
    }\
    SKIP_BITS(name, gb, n)\
}

749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770
#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 已提交
771
// deprecated, dont use get_vlc for new code, use get_vlc2 instead or use GET_VLC directly
772 773
static inline int get_vlc(GetBitContext *s, VLC *vlc)
{
M
Michael Niedermayer 已提交
774 775 776
    int code;
    VLC_TYPE (*table)[2]= vlc->table;
    
777 778
    OPEN_READER(re, s)
    UPDATE_CACHE(re, s)
779

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

M
Michael Niedermayer 已提交
782 783 784 785
    CLOSE_READER(re, s)
    return code;
}

F
Falk Hüffner 已提交
786 787
static always_inline int get_vlc2(GetBitContext *s, VLC_TYPE (*table)[2],
                                  int bits, int max_depth)
M
Michael Niedermayer 已提交
788 789 790 791 792 793 794 795
{
    int code;
    
    OPEN_READER(re, s)
    UPDATE_CACHE(re, s)

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

796
    CLOSE_READER(re, s)
797 798 799
    return code;
}

800

F
Fabrice Bellard 已提交
801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825
/* 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 */
M
Michael Niedermayer 已提交
826
extern const uint8_t ff_log2_tab[256];
F
Fabrice Bellard 已提交
827

828
static inline int av_log2(unsigned int v)
F
Fabrice Bellard 已提交
829 830 831 832 833 834 835 836 837 838 839 840
{
    int n;

    n = 0;
    if (v & 0xffff0000) {
        v >>= 16;
        n += 16;
    }
    if (v & 0xff00) {
        v >>= 8;
        n += 8;
    }
M
Michael Niedermayer 已提交
841 842 843 844 845 846 847 848 849 850 851 852 853
    n += ff_log2_tab[v];

    return n;
}

static inline int av_log2_16bit(unsigned int v)
{
    int n;

    n = 0;
    if (v & 0xff00) {
        v >>= 8;
        n += 8;
F
Fabrice Bellard 已提交
854
    }
M
Michael Niedermayer 已提交
855 856
    n += ff_log2_tab[v];

F
Fabrice Bellard 已提交
857 858 859
    return n;
}

M
Michael Niedermayer 已提交
860

861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878
/* 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;
}

879 880 881 882 883 884 885 886 887 888
static inline int clip(int a, int amin, int amax)
{
    if (a < amin)
        return amin;
    else if (a > amax)
        return amax;
    else
        return a;
}

889
/* math */
M
Michael Niedermayer 已提交
890
extern const uint8_t ff_sqrt_tab[128];
M
Michael Niedermayer 已提交
891

892 893
int ff_gcd(int a, int b);

M
sqrt  
Michael Niedermayer 已提交
894 895 896 897 898
static inline int ff_sqrt(int a)
{
    int ret=0;
    int s;
    int ret_sq=0;
M
Michael Niedermayer 已提交
899 900 901
    
    if(a<128) return ff_sqrt_tab[a];
    
M
sqrt  
Michael Niedermayer 已提交
902 903 904 905 906 907 908 909 910
    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;
}
911 912 913 914

/**
 * converts fourcc string to int
 */
Z
Zdenek Kabelac 已提交
915
static inline int ff_get_fourcc(const char *s){
916 917 918 919 920
    assert( strlen(s)==4 );
    
    return (s[0]) + (s[1]<<8) + (s[2]<<16) + (s[3]<<24);
}

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

923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938

#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 已提交
939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957
#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

958 959
#define CLAMP_TO_8BIT(d) ((d > 0xff) ? 0xff : (d < 0) ? 0 : d)

960 961 962 963 964
/* avoid usage of various functions */
#define malloc please_use_av_malloc
#define free please_use_av_free
#define realloc please_use_av_realloc

965 966 967
#endif /* HAVE_AV_CONFIG_H */

#endif /* COMMON_H */