common.h 22.3 KB
Newer Older
M
Michael Niedermayer 已提交
1 2 3 4 5
/**
 * @file common.h
 * common internal api header.
 */

F
Fabrice Bellard 已提交
6 7 8
#ifndef COMMON_H
#define COMMON_H

F
Fabrice Bellard 已提交
9 10
#define FFMPEG_VERSION_INT 0x000406
#define FFMPEG_VERSION     "0.4.6"
F
Fabrice Bellard 已提交
11

12
#if defined(WIN32) && !defined(__MINGW32__) && !defined(__CYGWIN__)
13
#    define CONFIG_WIN32
F
Fabrice Bellard 已提交
14 15
#endif

16
//#define ALT_BITSTREAM_WRITER
17
//#define ALIGNED_BITSTREAM_WRITER
18

19 20 21 22
#define ALT_BITSTREAM_READER
//#define LIBMPEG2_BITSTREAM_READER
//#define A32_BITSTREAM_READER

F
Fabrice Bellard 已提交
23
#ifdef HAVE_AV_CONFIG_H
F
Fabrice Bellard 已提交
24
/* only include the following when compiling package */
25 26 27 28 29 30 31 32 33 34 35 36 37 38 39
#    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 已提交
40

F
Fabrice Bellard 已提交
41 42 43 44
#ifndef M_PI
#define M_PI    3.14159265358979323846
#endif

45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62
#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)

63
#endif /* HAVE_AV_CONFIG_H */
F
Fabrice Bellard 已提交
64

65 66
/* Suppress restrict if it was not defined in config.h.  */
#ifndef restrict
67
#    define restrict
68 69
#endif

F
Falk Hüffner 已提交
70
#if defined(__GNUC__) && (__GNUC__ > 3 || __GNUC__ == 3 && __GNUC_MINOR__ > 0)
71
#    define always_inline __attribute__((always_inline)) inline
F
Falk Hüffner 已提交
72
#else
73
#    define always_inline inline
F
Falk Hüffner 已提交
74 75
#endif

F
Fabrice Bellard 已提交
76 77 78 79
#ifdef CONFIG_WIN32

/* windows */

80 81 82 83 84 85 86 87
typedef unsigned short uint16_t;
typedef signed short int16_t;
typedef unsigned char uint8_t;
typedef unsigned int uint32_t;
typedef unsigned __int64 uint64_t;
typedef signed char int8_t;
typedef signed int int32_t;
typedef signed __int64 int64_t;
F
Fabrice Bellard 已提交
88

89
#    ifndef __MINGW32__
90 91
#        define int64_t_C(c)     (c ## i64)
#        define uint64_t_C(c)    (c ## i64)
F
Fabrice Bellard 已提交
92

93
#        define inline __inline
F
Fabrice Bellard 已提交
94

95
#    else
96 97
#        define int64_t_C(c)     (c ## LL)
#        define uint64_t_C(c)    (c ## ULL)
98
#    endif /* __MINGW32__ */
F
Fabrice Bellard 已提交
99

100 101 102
#    ifdef _DEBUG
#        define DEBUG
#    endif
F
Fabrice Bellard 已提交
103

104
#    define snprintf _snprintf
F
Fabrice Bellard 已提交
105
#    define vsnprintf _vsnprintf
F
Fabrice Bellard 已提交
106

107 108 109 110 111 112 113 114
/* CONFIG_WIN32 end */
#elif defined (CONFIG_OS2)
/* OS/2 EMX */

#include <inttypes.h>

#ifdef HAVE_AV_CONFIG_H

115 116 117
#ifndef int64_t_C
#define int64_t_C(c)     (c ## LL)
#define uint64_t_C(c)    (c ## ULL)
118 119 120 121 122 123 124 125 126 127 128 129
#endif

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

#include <float.h>

#endif /* HAVE_AV_CONFIG_H */

/* CONFIG_OS2 end */
#else
F
Fabrice Bellard 已提交
130 131 132

/* unix */

133
#    include <inttypes.h>
F
Fabrice Bellard 已提交
134

135
#    ifdef HAVE_AV_CONFIG_H
136 137 138
#        ifndef int64_t_C
#            define int64_t_C(c)     (c ## LL)
#            define uint64_t_C(c)    (c ## ULL)
139
#        endif
F
Fabrice Bellard 已提交
140

141 142 143 144
#        ifdef USE_FASTMEMCPY
#            include "fastmemcpy.h"
#        endif
#    endif /* HAVE_AV_CONFIG_H */
F
Fabrice Bellard 已提交
145

146
#endif /* !CONFIG_WIN32 && !CONFIG_OS2 */
F
Fabrice Bellard 已提交
147 148 149

#ifdef HAVE_AV_CONFIG_H

150
#    include "bswap.h"
151

152 153 154 155 156 157
#    if defined(__MINGW32__) || defined(__CYGWIN__) || \
        defined(__OS2__) || defined (__OpenBSD__)
#        define MANGLE(a) "_" #a
#    else
#        define MANGLE(a) #a
#    endif
F
Fabrice Bellard 已提交
158

F
Fabrice Bellard 已提交
159 160
/* debug stuff */

161 162 163 164
#    ifndef DEBUG
#        define NDEBUG
#    endif
#    include <assert.h>
F
Fabrice Bellard 已提交
165

F
Fabrice Bellard 已提交
166
/* dprintf macros */
167
#    if defined(CONFIG_WIN32) && !defined(__MINGW32__)
F
Fabrice Bellard 已提交
168 169 170

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

171
#    else
F
Fabrice Bellard 已提交
172

173 174 175 176 177
#        ifdef DEBUG
#            define dprintf(fmt,args...) printf(fmt, ## args)
#        else
#            define dprintf(fmt,args...)
#        endif
F
Fabrice Bellard 已提交
178

179
#    endif /* !CONFIG_WIN32 */
F
Fabrice Bellard 已提交
180

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

183 184
//rounded divison & shift
#define RSHIFT(a,b) ((a) > 0 ? ((a) + (1<<((b)-1)))>>(b) : ((a) + (1<<((b)-1))-1)>>(b))
185 186
/* assume b>0 */
#define ROUNDED_DIV(a,b) (((a)>0 ? (a) + ((b)>>1) : (a) - ((b)>>1))/(b))
M
Michael Niedermayer 已提交
187
#define ABS(a) ((a) >= 0 ? (a) : (-(a)))
188

M
Michael Niedermayer 已提交
189 190
#define FFMAX(a,b) ((a) > (b) ? (a) : (b))
#define FFMIN(a,b) ((a) > (b) ? (b) : (a))
191

192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207
#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;
}
208
#else
209 210
#    define NEG_SSR32(a,s) ((( int32_t)(a))>>(32-(s)))
#    define NEG_USR32(a,s) (((uint32_t)(a))>>(32-(s)))
211 212
#endif

F
Fabrice Bellard 已提交
213 214 215 216
/* bit output */

struct PutBitContext;

217
typedef void (*WriteDataFunc)(void *, uint8_t *, int);
F
Fabrice Bellard 已提交
218 219

typedef struct PutBitContext {
220
#ifdef ALT_BITSTREAM_WRITER
221
    uint8_t *buf, *buf_end;
222 223
    int index;
#else
224
    uint32_t bit_buf;
225
    int bit_left;
226
    uint8_t *buf, *buf_ptr, *buf_end;
227
#endif
228
    int64_t data_out_size; /* in bytes */
F
Fabrice Bellard 已提交
229 230 231
} PutBitContext;

void init_put_bits(PutBitContext *s, 
232
                   uint8_t *buffer, int buffer_size,
F
Fabrice Bellard 已提交
233
                   void *opaque,
234
                   void (*write_data)(void *, uint8_t *, int));
235

236
int64_t get_bit_count(PutBitContext *s); /* XXX: change function name */
F
Fabrice Bellard 已提交
237 238
void align_put_bits(PutBitContext *s);
void flush_put_bits(PutBitContext *s);
239
void put_string(PutBitContext * pbc, char *s);
F
Fabrice Bellard 已提交
240 241 242 243

/* bit input */

typedef struct GetBitContext {
244
    const uint8_t *buffer, *buffer_end;
245 246
#ifdef ALT_BITSTREAM_READER
    int index;
247
#elif defined LIBMPEG2_BITSTREAM_READER
248 249
    uint8_t *buffer_ptr;
    uint32_t cache;
250 251
    int bit_count;
#elif defined A32_BITSTREAM_READER
252 253 254
    uint32_t *buffer_ptr;
    uint32_t cache0;
    uint32_t cache1;
255
    int bit_count;
256
#endif
257
    int size_in_bits;
F
Fabrice Bellard 已提交
258 259
} GetBitContext;

260 261
static inline int get_bits_count(GetBitContext *s);

262
#define VLC_TYPE int16_t
263

F
Fabrice Bellard 已提交
264 265
typedef struct VLC {
    int bits;
266
    VLC_TYPE (*table)[2]; ///< code, bits
F
Fabrice Bellard 已提交
267 268 269
    int table_size, table_allocated;
} VLC;

270 271 272 273 274 275
typedef struct RL_VLC_ELEM {
    int16_t level;
    int8_t len;
    uint8_t run;
} RL_VLC_ELEM;

276 277 278 279
#ifdef ARCH_SPARC64
#define UNALIGNED_STORES_ARE_BAD
#endif

280 281
/* used to avoid missaligned exceptions on some archs (alpha, ...) */
#ifdef ARCH_X86
282
#    define unaligned32(a) (*(uint32_t*)(a))
283
#else
284
#    ifdef __GNUC__
285 286 287 288 289 290 291
static inline uint32_t unaligned32(const void *v) {
    struct Unaligned {
	uint32_t i;
    } __attribute__((packed));

    return ((const struct Unaligned *) v)->i;
}
292
#    elif defined(__DECC)
293 294 295
static inline uint32_t unaligned32(const void *v) {
    return *(const __unaligned uint32_t *) v;
}
296
#    else
297 298 299
static inline uint32_t unaligned32(const void *v) {
    return *(const uint32_t *) v;
}
300
#    endif
301 302
#endif //!ARCH_X86

303 304 305 306 307 308 309 310 311 312 313
#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));
314
    
315 316 317 318 319 320 321 322 323 324 325
    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);
326 327 328 329 330 331 332 333
#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
334
        *(uint32_t *)s->buf_ptr = be2me_32(bit_buf);
335 336 337 338 339 340 341 342 343 344 345 346
        //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


347
#ifdef ALT_BITSTREAM_WRITER
348
static inline void put_bits(PutBitContext *s, int n, unsigned int value)
349
{
350 351
#    ifdef ALIGNED_BITSTREAM_WRITER
#        ifdef ARCH_X86
352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368
    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"
    );
369
#        else
370 371 372 373 374 375 376 377 378 379
    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;
380 381 382
#        endif
#    else //ALIGNED_BITSTREAM_WRITER
#        ifdef ARCH_X86
383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398
    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"
    );
399
#        else
400 401 402 403 404 405 406 407
    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;
408 409
#        endif
#    endif //!ALIGNED_BITSTREAM_WRITER
410 411 412 413 414 415 416 417 418 419 420 421 422
}
#endif


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

423 424 425
/* Bitstream reader API docs:
name
    abritary name which is used as prefix for the internal variables
F
Fabrice Bellard 已提交
426

427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466
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
*/
467

468
#ifdef ALT_BITSTREAM_READER
469
#   define MIN_CACHE_BITS 25
470

471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497
#   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)\
498
        NEG_USR32(name##_cache, num)
499 500

#   define SHOW_SBITS(name, gb, num)\
501
        NEG_SSR32(name##_cache, num)
502 503 504 505 506 507

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

static inline int get_bits_count(GetBitContext *s){
    return s->index;
508
}
509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529
#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;\
    }\
530

531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546
#   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)\
547
        NEG_USR32(name##_cache, num)
548 549

#   define SHOW_SBITS(name, gb, num)\
550
        NEG_SSR32(name##_cache, num)
551 552 553 554 555 556

#   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;
557 558
}

559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577
#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 );\
578
        name##_cache0 |= NEG_USR32(next,name##_bit_count);\
579 580 581 582 583
        name##_cache1 |= next<<name##_bit_count;\
        name##_buffer_ptr++;\
        name##_bit_count-= 32;\
    }\

584
#ifdef ARCH_X86
585 586 587 588 589 590 591
#   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)\
           );
592
#else
593 594
#   define SKIP_CACHE(name, gb, num)\
        name##_cache0 <<= (num);\
595
        name##_cache0 |= NEG_USR32(name##_cache1,num);\
596
        name##_cache1 <<= (num);
597
#endif
598

599 600 601 602 603 604 605 606 607 608 609 610 611
#   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)\
612
        NEG_USR32(name##_cache0, num)
613 614

#   define SHOW_SBITS(name, gb, num)\
615
        NEG_SSR32(name##_cache0, num)
616 617 618 619 620 621

#   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;
622 623
}

624 625
#endif

626 627 628 629 630 631 632 633 634
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;
}
635

636 637 638 639 640 641 642
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;
643 644
}

645
static inline void skip_bits(GetBitContext *s, int n){
646 647 648 649 650 651 652 653
 //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){
654
#ifdef ALT_BITSTREAM_READER
655 656 657 658 659 660
    int index= s->index;
    uint8_t result= s->buffer[ index>>3 ];
    result<<= (index&0x07);
    result>>= 8 - 1;
    index++;
    s->index= index;
661

662
    return result;
663
#else
664
    return get_bits(s, 1);
665
#endif
666 667
}

668 669
static inline unsigned int show_bits1(GetBitContext *s){
    return show_bits(s, 1);
670 671
}

672 673
static inline void skip_bits1(GetBitContext *s){
    skip_bits(s, 1);
F
Fabrice Bellard 已提交
674
}
675

676
void init_get_bits(GetBitContext *s,
677
                   const uint8_t *buffer, int buffer_size);
678

679
int check_marker(GetBitContext *s, const char *msg);
F
Fabrice Bellard 已提交
680 681 682 683 684 685
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);

686 687 688 689 690 691
/**
 *
 * 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 已提交
692 693 694 695 696 697 698 699 700 701 702 703 704 705
#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;\
\
706
        index= SHOW_UBITS(name, gb, nb_bits) + code;\
M
Michael Niedermayer 已提交
707 708 709 710 711 712 713 714
        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;\
\
715
            index= SHOW_UBITS(name, gb, nb_bits) + code;\
M
Michael Niedermayer 已提交
716 717 718 719 720 721 722
            code = table[index][0];\
            n    = table[index][1];\
        }\
    }\
    SKIP_BITS(name, gb, n)\
}

723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744
#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 已提交
745
// deprecated, dont use get_vlc for new code, use get_vlc2 instead or use GET_VLC directly
746 747
static inline int get_vlc(GetBitContext *s, VLC *vlc)
{
M
Michael Niedermayer 已提交
748 749 750
    int code;
    VLC_TYPE (*table)[2]= vlc->table;
    
751 752
    OPEN_READER(re, s)
    UPDATE_CACHE(re, s)
753

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

M
Michael Niedermayer 已提交
756 757 758 759
    CLOSE_READER(re, s)
    return code;
}

M
Michael Niedermayer 已提交
760 761 762 763 764 765 766 767
/**
 * parses a vlc code, faster then get_vlc()
 * @param bits is the number of bits which will be read at once, must be 
 *             identical to nb_bits in init_vlc()
 * @param max_depth is the number of times bits bits must be readed to completly
 *                  read the longest vlc code 
 *                  = (max_vlc_length + bits - 1) / bits
 */
F
Falk Hüffner 已提交
768 769
static always_inline int get_vlc2(GetBitContext *s, VLC_TYPE (*table)[2],
                                  int bits, int max_depth)
M
Michael Niedermayer 已提交
770 771 772 773 774 775 776 777
{
    int code;
    
    OPEN_READER(re, s)
    UPDATE_CACHE(re, s)

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

778
    CLOSE_READER(re, s)
779 780 781
    return code;
}

782

F
Fabrice Bellard 已提交
783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807
/* 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 已提交
808
extern const uint8_t ff_log2_tab[256];
F
Fabrice Bellard 已提交
809

810
static inline int av_log2(unsigned int v)
F
Fabrice Bellard 已提交
811 812 813 814 815 816 817 818 819 820 821 822
{
    int n;

    n = 0;
    if (v & 0xffff0000) {
        v >>= 16;
        n += 16;
    }
    if (v & 0xff00) {
        v >>= 8;
        n += 8;
    }
M
Michael Niedermayer 已提交
823 824 825 826 827 828 829 830 831 832 833 834 835
    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 已提交
836
    }
M
Michael Niedermayer 已提交
837 838
    n += ff_log2_tab[v];

F
Fabrice Bellard 已提交
839 840 841
    return n;
}

M
Michael Niedermayer 已提交
842

843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860
/* 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;
}

861 862 863 864 865 866 867 868 869 870
static inline int clip(int a, int amin, int amax)
{
    if (a < amin)
        return amin;
    else if (a > amax)
        return amax;
    else
        return a;
}

871
/* math */
M
Michael Niedermayer 已提交
872
extern const uint8_t ff_sqrt_tab[128];
M
Michael Niedermayer 已提交
873

874 875
int ff_gcd(int a, int b);

M
sqrt  
Michael Niedermayer 已提交
876 877 878 879 880
static inline int ff_sqrt(int a)
{
    int ret=0;
    int s;
    int ret_sq=0;
M
Michael Niedermayer 已提交
881 882 883
    
    if(a<128) return ff_sqrt_tab[a];
    
M
sqrt  
Michael Niedermayer 已提交
884 885 886 887 888 889 890 891 892
    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;
}
893 894 895 896

/**
 * converts fourcc string to int
 */
Z
Zdenek Kabelac 已提交
897
static inline int ff_get_fourcc(const char *s){
898 899 900 901 902
    assert( strlen(s)==4 );
    
    return (s[0]) + (s[1]<<8) + (s[2]<<16) + (s[3]<<24);
}

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

905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920

#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 已提交
921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939
#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

940 941
#define CLAMP_TO_8BIT(d) ((d > 0xff) ? 0xff : (d < 0) ? 0 : d)

942 943 944 945 946
/* avoid usage of various functions */
#define malloc please_use_av_malloc
#define free please_use_av_free
#define realloc please_use_av_realloc

947 948 949
#endif /* HAVE_AV_CONFIG_H */

#endif /* COMMON_H */