common.h 22.2 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
#ifdef CONFIG_WIN32

/* windows */

75 76 77 78 79 80 81 82
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 已提交
83

84
#    ifndef __MINGW32__
85 86
#        define int64_t_C(c)     (c ## i64)
#        define uint64_t_C(c)    (c ## i64)
F
Fabrice Bellard 已提交
87

88
#        define inline __inline
F
Fabrice Bellard 已提交
89

90
#    else
91 92
#        define int64_t_C(c)     (c ## LL)
#        define uint64_t_C(c)    (c ## ULL)
93
#    endif /* __MINGW32__ */
F
Fabrice Bellard 已提交
94

95 96 97
#    ifdef _DEBUG
#        define DEBUG
#    endif
F
Fabrice Bellard 已提交
98

99
#    define snprintf _snprintf
F
Fabrice Bellard 已提交
100
#    define vsnprintf _vsnprintf
F
Fabrice Bellard 已提交
101

102 103 104 105 106 107 108 109
/* CONFIG_WIN32 end */
#elif defined (CONFIG_OS2)
/* OS/2 EMX */

#include <inttypes.h>

#ifdef HAVE_AV_CONFIG_H

110 111 112
#ifndef int64_t_C
#define int64_t_C(c)     (c ## LL)
#define uint64_t_C(c)    (c ## ULL)
113 114 115 116 117 118 119 120 121 122 123 124
#endif

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

#include <float.h>

#endif /* HAVE_AV_CONFIG_H */

/* CONFIG_OS2 end */
#else
F
Fabrice Bellard 已提交
125 126 127

/* unix */

128
#    include <inttypes.h>
F
Fabrice Bellard 已提交
129

130
#    ifdef HAVE_AV_CONFIG_H
131 132 133
#        ifndef int64_t_C
#            define int64_t_C(c)     (c ## LL)
#            define uint64_t_C(c)    (c ## ULL)
134
#        endif
F
Fabrice Bellard 已提交
135

136 137 138 139
#        ifdef USE_FASTMEMCPY
#            include "fastmemcpy.h"
#        endif
#    endif /* HAVE_AV_CONFIG_H */
F
Fabrice Bellard 已提交
140

141
#endif /* !CONFIG_WIN32 && !CONFIG_OS2 */
F
Fabrice Bellard 已提交
142 143 144

#ifdef HAVE_AV_CONFIG_H

145
#    include "bswap.h"
146

147 148 149 150 151 152
#    if defined(__MINGW32__) || defined(__CYGWIN__) || \
        defined(__OS2__) || defined (__OpenBSD__)
#        define MANGLE(a) "_" #a
#    else
#        define MANGLE(a) #a
#    endif
F
Fabrice Bellard 已提交
153

F
Fabrice Bellard 已提交
154 155
/* debug stuff */

156 157 158 159
#    ifndef DEBUG
#        define NDEBUG
#    endif
#    include <assert.h>
F
Fabrice Bellard 已提交
160

F
Fabrice Bellard 已提交
161
/* dprintf macros */
162
#    if defined(CONFIG_WIN32) && !defined(__MINGW32__)
F
Fabrice Bellard 已提交
163 164 165

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

166
#    else
F
Fabrice Bellard 已提交
167

168 169 170 171 172
#        ifdef DEBUG
#            define dprintf(fmt,args...) printf(fmt, ## args)
#        else
#            define dprintf(fmt,args...)
#        endif
F
Fabrice Bellard 已提交
173

174
#    endif /* !CONFIG_WIN32 */
F
Fabrice Bellard 已提交
175

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

178 179
//rounded divison & shift
#define RSHIFT(a,b) ((a) > 0 ? ((a) + (1<<((b)-1)))>>(b) : ((a) + (1<<((b)-1))-1)>>(b))
180 181
/* assume b>0 */
#define ROUNDED_DIV(a,b) (((a)>0 ? (a) + ((b)>>1) : (a) - ((b)>>1))/(b))
M
Michael Niedermayer 已提交
182
#define ABS(a) ((a) >= 0 ? (a) : (-(a)))
183

M
Michael Niedermayer 已提交
184 185
#define FFMAX(a,b) ((a) > (b) ? (a) : (b))
#define FFMIN(a,b) ((a) > (b) ? (b) : (a))
186

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

F
Fabrice Bellard 已提交
208 209 210 211
/* bit output */

struct PutBitContext;

212
typedef void (*WriteDataFunc)(void *, uint8_t *, int);
F
Fabrice Bellard 已提交
213 214

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

void init_put_bits(PutBitContext *s, 
227
                   uint8_t *buffer, int buffer_size,
F
Fabrice Bellard 已提交
228
                   void *opaque,
229
                   void (*write_data)(void *, uint8_t *, int));
230

231
int64_t get_bit_count(PutBitContext *s); /* XXX: change function name */
F
Fabrice Bellard 已提交
232 233
void align_put_bits(PutBitContext *s);
void flush_put_bits(PutBitContext *s);
234
void put_string(PutBitContext * pbc, char *s);
F
Fabrice Bellard 已提交
235 236 237 238

/* bit input */

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

255 256
static inline int get_bits_count(GetBitContext *s);

257
#define VLC_TYPE int16_t
258

F
Fabrice Bellard 已提交
259 260
typedef struct VLC {
    int bits;
261
    VLC_TYPE (*table)[2]; // code, bits
F
Fabrice Bellard 已提交
262 263 264
    int table_size, table_allocated;
} VLC;

265 266 267 268 269 270
typedef struct RL_VLC_ELEM {
    int16_t level;
    int8_t len;
    uint8_t run;
} RL_VLC_ELEM;

271 272 273 274
#ifdef ARCH_SPARC64
#define UNALIGNED_STORES_ARE_BAD
#endif

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

    return ((const struct Unaligned *) v)->i;
}
287
#    elif defined(__DECC)
288 289 290
static inline uint32_t unaligned32(const void *v) {
    return *(const __unaligned uint32_t *) v;
}
291
#    else
292 293 294
static inline uint32_t unaligned32(const void *v) {
    return *(const uint32_t *) v;
}
295
#    endif
296 297
#endif //!ARCH_X86

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


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


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

418 419 420
/* Bitstream reader API docs:
name
    abritary name which is used as prefix for the internal variables
F
Fabrice Bellard 已提交
421

422 423 424 425 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
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
*/
462

463
#ifdef ALT_BITSTREAM_READER
464
#   define MIN_CACHE_BITS 25
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
#   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)\
493
        NEG_USR32(name##_cache, num)
494 495

#   define SHOW_SBITS(name, gb, num)\
496
        NEG_SSR32(name##_cache, num)
497 498 499 500 501 502

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

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

526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541
#   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)\
542
        NEG_USR32(name##_cache, num)
543 544

#   define SHOW_SBITS(name, gb, num)\
545
        NEG_SSR32(name##_cache, num)
546 547 548 549 550 551

#   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;
552 553
}

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

579
#ifdef ARCH_X86
580 581 582 583 584 585 586
#   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)\
           );
587
#else
588 589
#   define SKIP_CACHE(name, gb, num)\
        name##_cache0 <<= (num);\
590
        name##_cache0 |= NEG_USR32(name##_cache1,num);\
591
        name##_cache1 <<= (num);
592
#endif
593

594 595 596 597 598 599 600 601 602 603 604 605 606
#   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)\
607
        NEG_USR32(name##_cache0, num)
608 609

#   define SHOW_SBITS(name, gb, num)\
610
        NEG_SSR32(name##_cache0, num)
611 612 613 614 615 616

#   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;
617 618
}

619 620
#endif

621 622 623 624 625 626 627 628 629
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;
}
630

631 632 633 634 635 636 637
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;
638 639
}

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

657
    return result;
658
#else
659
    return get_bits(s, 1);
660
#endif
661 662
}

663 664
static inline unsigned int show_bits1(GetBitContext *s){
    return show_bits(s, 1);
665 666
}

667 668
static inline void skip_bits1(GetBitContext *s){
    skip_bits(s, 1);
F
Fabrice Bellard 已提交
669
}
670

671
void init_get_bits(GetBitContext *s,
672
                   uint8_t *buffer, int buffer_size);
673

674
int check_marker(GetBitContext *s, const char *msg);
F
Fabrice Bellard 已提交
675 676 677 678 679 680
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);

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

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

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

M
Michael Niedermayer 已提交
751 752 753 754
    CLOSE_READER(re, s)
    return code;
}

M
Michael Niedermayer 已提交
755 756 757 758 759 760 761 762
/**
 * 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 已提交
763 764
static always_inline int get_vlc2(GetBitContext *s, VLC_TYPE (*table)[2],
                                  int bits, int max_depth)
M
Michael Niedermayer 已提交
765 766 767 768 769 770 771 772
{
    int code;
    
    OPEN_READER(re, s)
    UPDATE_CACHE(re, s)

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

773
    CLOSE_READER(re, s)
774 775 776
    return code;
}

777

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

805
static inline int av_log2(unsigned int v)
F
Fabrice Bellard 已提交
806 807 808 809 810 811 812 813 814 815 816 817
{
    int n;

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

F
Fabrice Bellard 已提交
834 835 836
    return n;
}

M
Michael Niedermayer 已提交
837

838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855
/* 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;
}

856 857 858 859 860 861 862 863 864 865
static inline int clip(int a, int amin, int amax)
{
    if (a < amin)
        return amin;
    else if (a > amax)
        return amax;
    else
        return a;
}

866
/* math */
M
Michael Niedermayer 已提交
867
extern const uint8_t ff_sqrt_tab[128];
M
Michael Niedermayer 已提交
868

869 870
int ff_gcd(int a, int b);

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

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

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

900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915

#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 已提交
916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934
#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

935 936
#define CLAMP_TO_8BIT(d) ((d > 0xff) ? 0xff : (d < 0) ? 0 : d)

937 938 939 940 941
/* avoid usage of various functions */
#define malloc please_use_av_malloc
#define free please_use_av_free
#define realloc please_use_av_realloc

942 943 944
#endif /* HAVE_AV_CONFIG_H */

#endif /* COMMON_H */