X-Git-Url: http://git.tuebingen.mpg.de/?p=paraslash.git;a=blobdiff_plain;f=bitstream.h;h=d052337c1a1825d29bc700c8a918177220dc4f74;hp=fe85aba7475684168d13b0f3614c93953f7d957e;hb=ae9a1e8863223938de42debb38a20b2060e0c181;hpb=18651668774f785d38189eee9d8389fe69428380 diff --git a/bitstream.h b/bitstream.h index fe85aba7..d052337c 100644 --- a/bitstream.h +++ b/bitstream.h @@ -47,7 +47,7 @@ struct vlc { * Refill the internal cache from the bitstream. */ #define UPDATE_CACHE(name, gb)\ - name##_cache= AV_RB32( ((const uint8_t *)(gb)->buffer) \ + name##_cache= AV_RB32( ((gb)->buffer) \ + (name##_index >> 3) ) << (name##_index & 0x07);\ /** @@ -77,9 +77,6 @@ struct vlc { SKIP_COUNTER(name, gb, num)\ }\ -/** This is equivalent to SKIP_LAST_CACHE; SKIP_COUNTER. */ -#define LAST_SKIP_BITS(name, gb, num) SKIP_COUNTER(name, gb, num) - /** Return the next num bits. */ #define SHOW_UBITS(name, gb, num)\ NEG_USR32(name##_cache, num) @@ -98,7 +95,7 @@ static inline unsigned int get_bits(struct getbit_context *s, int n) OPEN_READER(re, s) UPDATE_CACHE(re, s) tmp = SHOW_UBITS(re, s, n); - LAST_SKIP_BITS(re, s, n) + SKIP_COUNTER(re, s, n) CLOSE_READER(re, s) return tmp; } @@ -108,7 +105,7 @@ static inline void skip_bits(struct getbit_context *s, int n) /* 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) + SKIP_COUNTER(re, s, n) CLOSE_READER(re, s) } @@ -131,18 +128,13 @@ static inline unsigned int get_bits1(struct getbit_context *s) * bytes larger then the actual read bits * because some optimized bitstream * readers read 32 or 64 bit at once and could read over the end. * - * \param bit_size The size of the buffer in bits. + * \param bit_size The size of the buffer in bytes. */ static inline void init_get_bits(struct getbit_context *s, - const uint8_t *buffer, int bit_size) + const uint8_t *buffer, int size) { - int buffer_size = (bit_size + 7) >> 3; - if (buffer_size < 0 || bit_size < 0) { - buffer_size = bit_size = 0; - buffer = NULL; - } s->buffer = buffer; - s->buffer_end = buffer + buffer_size; + s->buffer_end = buffer + size; s->index = 0; } @@ -152,38 +144,41 @@ void init_vlc(struct vlc *vlc, int nb_bits, int nb_codes, const void *bits, void free_vlc(struct vlc *vlc); /** + * Parse a vlc code. + * + * \param bits The number of bits which will be read at once, must be + * identical to nb_bits in init_vlc() * - * 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 + * \param max_depth The number of times bits bits must be read to completely + * read the longest vlc code = (max_vlc_length + bits - 1) / bits. */ -#define GET_VLC(code, gb, table, bits, max_depth)\ -{\ - int n, index, nb_bits;\ -\ - index= SHOW_UBITS(re, gb, bits);\ - code = table[index][0];\ - n = table[index][1];\ -\ - if(max_depth > 1 && n < 0){\ - LAST_SKIP_BITS(re, gb, bits)\ - UPDATE_CACHE(re, gb)\ -\ - nb_bits = -n;\ -\ - index= SHOW_UBITS(re, gb, nb_bits) + code;\ - code = table[index][0];\ - n = table[index][1];\ - if(max_depth > 2 && n < 0){\ - LAST_SKIP_BITS(re, gb, nb_bits)\ - UPDATE_CACHE(re, gb)\ -\ - nb_bits = -n;\ -\ - index= SHOW_UBITS(re, gb, nb_bits) + code;\ - code = table[index][0];\ - n = table[index][1];\ - }\ - }\ - SKIP_BITS(re, gb, n)\ +static inline int get_vlc(struct getbit_context *gb, VLC_TYPE(*table)[2], + int bits, int max_depth) +{ + int n, idx, nb_bits, code; + + OPEN_READER(re, gb) + UPDATE_CACHE(re, gb) + idx = SHOW_UBITS(re, gb, bits); + code = table[idx][0]; + n = table[idx][1]; + if (max_depth > 1 && n < 0) { + SKIP_COUNTER(re, gb, bits) + UPDATE_CACHE(re, gb) + nb_bits = -n; + idx = SHOW_UBITS(re, gb, nb_bits) + code; + code = table[idx][0]; + n = table[idx][1]; + if (max_depth > 2 && n < 0) { + SKIP_COUNTER(re, gb, nb_bits) + UPDATE_CACHE(re, gb) + nb_bits = -n; + idx = SHOW_UBITS(re, gb, nb_bits) + code; + code = table[idx][0]; + n = table[idx][1]; + } + } + SKIP_BITS(re, gb, n) + CLOSE_READER(re, gb) + return code; }