123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261 |
-
- #include "bitstream.h"
- void SetBitstreamPointer(BitStreamInfo *bsi, int nBytes, unsigned char *buf)
- {
-
- bsi->bytePtr = buf;
- bsi->iCache = 0;
- bsi->cachedBits = 0;
- bsi->nBytes = nBytes;
- }
- static __inline void RefillBitstreamCache(BitStreamInfo *bsi)
- {
- int nBytes = bsi->nBytes;
- if (nBytes >= 4) {
-
- bsi->iCache = (*bsi->bytePtr++) << 24;
- bsi->iCache |= (*bsi->bytePtr++) << 16;
- bsi->iCache |= (*bsi->bytePtr++) << 8;
- bsi->iCache |= (*bsi->bytePtr++);
-
- bsi->cachedBits = 32;
- bsi->nBytes -= 4;
- } else {
- bsi->iCache = 0;
- while (nBytes--) {
- bsi->iCache |= (*bsi->bytePtr++);
- bsi->iCache <<= 8;
- }
- bsi->iCache <<= ((3 - bsi->nBytes)*8);
- bsi->cachedBits = 8*bsi->nBytes;
- bsi->nBytes = 0;
- }
- }
- unsigned int GetBits(BitStreamInfo *bsi, int nBits)
- {
- unsigned int data, lowBits;
- nBits &= 0x1f;
- data = bsi->iCache >> (31 - nBits);
- data >>= 1;
- bsi->iCache <<= nBits;
- bsi->cachedBits -= nBits;
-
- if (bsi->cachedBits < 0) {
- lowBits = -bsi->cachedBits;
- RefillBitstreamCache(bsi);
- data |= bsi->iCache >> (32 - lowBits);
-
- bsi->cachedBits -= lowBits;
- bsi->iCache <<= lowBits;
- }
- return data;
- }
- unsigned int GetBitsNoAdvance(BitStreamInfo *bsi, int nBits)
- {
- unsigned char *buf;
- unsigned int data, iCache;
- signed int lowBits;
- nBits &= 0x1f;
- data = bsi->iCache >> (31 - nBits);
- data >>= 1;
- lowBits = nBits - bsi->cachedBits;
-
- if (lowBits > 0) {
- iCache = 0;
- buf = bsi->bytePtr;
- while (lowBits > 0) {
- iCache <<= 8;
- if (buf < bsi->bytePtr + bsi->nBytes)
- iCache |= (unsigned int)*buf++;
- lowBits -= 8;
- }
- lowBits = -lowBits;
- data |= iCache >> lowBits;
- }
- return data;
- }
- void AdvanceBitstream(BitStreamInfo *bsi, int nBits)
- {
- nBits &= 0x1f;
- if (nBits > bsi->cachedBits) {
- nBits -= bsi->cachedBits;
- RefillBitstreamCache(bsi);
- }
- bsi->iCache <<= nBits;
- bsi->cachedBits -= nBits;
- }
- int CalcBitsUsed(BitStreamInfo *bsi, unsigned char *startBuf, int startOffset)
- {
- int bitsUsed;
- bitsUsed = (bsi->bytePtr - startBuf) * 8;
- bitsUsed -= bsi->cachedBits;
- bitsUsed -= startOffset;
- return bitsUsed;
- }
- void ByteAlignBitstream(BitStreamInfo *bsi)
- {
- int offset;
- offset = bsi->cachedBits & 0x07;
- AdvanceBitstream(bsi, offset);
- }
|