📄 common.h
字号:
#ifndef COMMON_H#define COMMON_H#define FFMPEG_VERSION_INT 0x000406#define FFMPEG_VERSION "0.4.6"#if defined(WIN32) && !defined(__MINGW32__) && !defined(__CYGWIN__)# define CONFIG_WIN32#endif//#define ALT_BITSTREAM_WRITER//#define ALIGNED_BITSTREAM_WRITER#define ALT_BITSTREAM_READER//#define LIBMPEG2_BITSTREAM_READER//#define A32_BITSTREAM_READER#ifdef HAVE_AV_CONFIG_H/* only include the following when compiling package */# 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#endif /* HAVE_AV_CONFIG_H *//* Suppress restrict if it was not defined in config.h. */#ifndef restrict# define restrict#endif#if defined(__GNUC__) && (__GNUC__ > 3 || __GNUC__ == 3 && __GNUC_MINOR__ > 0)# define always_inline __attribute__((always_inline)) inline#else# define always_inline inline#endif#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;typedef UINT64 uint64_t;typedef INT64 int64_t;# ifndef __MINGW32__# define INT64_C(c) (c ## i64)# define UINT64_C(c) (c ## i64)# define inline __inline# else# define INT64_C(c) (c ## LL)# define UINT64_C(c) (c ## ULL)# endif /* __MINGW32__ */# define M_PI 3.14159265358979323846# define M_SQRT2 1.41421356237309504880 /* sqrt(2) */# ifdef _DEBUG# define DEBUG# endif# define snprintf _snprintf#else /* CONFIG_WIN32 *//* unix */# include <inttypes.h># ifndef __WINE_WINDEF16_H/* workaround for typedef conflict in MPlayer (wine typedefs) */typedef unsigned short UINT16;typedef signed short INT16;# endiftypedef 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;# 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# endif /* HAVE_AV_CONFIG_H */#endif /* !CONFIG_WIN32 */#ifdef HAVE_AV_CONFIG_H# include "bswap.h"# if defined(__MINGW32__) || defined(__CYGWIN__) || \ defined(__OS2__) || defined (__OpenBSD__)# define MANGLE(a) "_" #a# else# define MANGLE(a) #a# endif/* debug stuff */# ifndef DEBUG# define NDEBUG# endif# include <assert.h>/* dprintf macros */# if defined(CONFIG_WIN32) && !defined(__MINGW32__)inline void dprintf(const char* fmt,...) {}# else# ifdef DEBUG# define dprintf(fmt,args...) printf(fmt, ## args)# else# define dprintf(fmt,args...)# endif# endif /* !CONFIG_WIN32 */# define av_abort() do { fprintf(stderr, "Abort at %s:%d\n", __FILE__, __LINE__); abort(); } while (0)//rounded divison & shift#define RSHIFT(a,b) ((a) > 0 ? ((a) + (1<<((b)-1)))>>(b) : ((a) + (1<<((b)-1))-1)>>(b))/* assume b>0 */#define ROUNDED_DIV(a,b) (((a)>0 ? (a) + ((b)>>1) : (a) - ((b)>>1))/(b))#define ABS(a) ((a) >= 0 ? (a) : (-(a)))#define FFMAX(a,b) ((a) > (b) ? (a) : (b))#define FFMIN(a,b) ((a) > (b) ? (b) : (a))#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;}#else# define NEG_SSR32(a,s) ((( int32_t)(a))>>(32-(s)))# define NEG_USR32(a,s) (((uint32_t)(a))>>(32-(s)))#endif/* bit output */struct PutBitContext;typedef void (*WriteDataFunc)(void *, UINT8 *, int);typedef struct PutBitContext {#ifdef ALT_BITSTREAM_WRITER UINT8 *buf, *buf_end; int index;#else UINT32 bit_buf; int bit_left; UINT8 *buf, *buf_ptr, *buf_end;#endif INT64 data_out_size; /* in bytes */} PutBitContext;void init_put_bits(PutBitContext *s, UINT8 *buffer, int buffer_size, void *opaque, void (*write_data)(void *, UINT8 *, int));INT64 get_bit_count(PutBitContext *s); /* XXX: change function name */void align_put_bits(PutBitContext *s);void flush_put_bits(PutBitContext *s);void put_string(PutBitContext * pbc, char *s);/* bit input */typedef struct GetBitContext { UINT8 *buffer, *buffer_end;#ifdef ALT_BITSTREAM_READER int index;#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;#endif int size;} GetBitContext;static inline int get_bits_count(GetBitContext *s);#define VLC_TYPE INT16typedef struct VLC { int bits; VLC_TYPE (*table)[2]; // code, bits int table_size, table_allocated;} VLC;typedef struct RL_VLC_ELEM { int16_t level; int8_t len; uint8_t run;} RL_VLC_ELEM;/* used to avoid missaligned exceptions on some archs (alpha, ...) */#ifdef ARCH_X86# define unaligned32(a) (*(UINT32*)(a))#else# ifdef __GNUC__static inline uint32_t unaligned32(const void *v) { struct Unaligned { uint32_t i; } __attribute__((packed)); return ((const struct Unaligned *) v)->i;}# elif defined(__DECC)static inline uint32_t unaligned32(const void *v) { return *(const __unaligned uint32_t *) v;}# elsestatic inline uint32_t unaligned32(const void *v) { return *(const uint32_t *) v;}# endif#endif //!ARCH_X86#ifndef ALT_BITSTREAM_WRITERstatic 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)); 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); *(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#ifdef ALT_BITSTREAM_WRITERstatic inline void put_bits(PutBitContext *s, int n, unsigned int value){# ifdef ALIGNED_BITSTREAM_WRITER# ifdef ARCH_X86 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" );# else 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;# endif# else //ALIGNED_BITSTREAM_WRITER# ifdef ARCH_X86 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" );# else 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;# endif# endif //!ALIGNED_BITSTREAM_WRITER}#endifstatic inline uint8_t* pbBufPtr(PutBitContext *s){#ifdef ALT_BITSTREAM_WRITER return s->buf + (s->index>>3);#else return s->buf_ptr;#endif}/* Bitstream reader API docs:name abritary name which is used as prefix for the internal variablesgb getbitcontextOPEN_READER(name, gb) loads gb into local variablesCLOSE_READER(name, gb) stores local vars in gbUPDATE_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 bitsSHOW_SBITS(name, gb, num) will return the nest num bits and do sign extensionSKIP_BITS(name, gb, num) will skip over the next num bits note, this is equinvalent to SKIP_CACHE; SKIP_COUNTERSKIP_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 nothingLAST_SKIP_BITS(name, gb, num) is equinvalent to SKIP_LAST_CACHE; SKIP_COUNTERfor examples see get_bits, show_bits, skip_bits, get_vlc*/#ifdef ALT_BITSTREAM_READER# define MIN_CACHE_BITS 25# 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)\
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -