summaryrefslogtreecommitdiffstats
path: root/libFDK/src/FDK_bitbuffer.cpp
diff options
context:
space:
mode:
authorDave Burke <daveburke@google.com>2012-04-17 09:51:45 -0700
committerDave Burke <daveburke@google.com>2012-04-17 23:04:43 -0700
commit9bf37cc9712506b2483650c82d3c41152337ef7e (patch)
tree77db44e2bae06e3d144b255628be2b7a55c581d3 /libFDK/src/FDK_bitbuffer.cpp
parenta37315fe10ee143d6d0b28c19d41a476a23e63ea (diff)
downloadfdk-aac-dabplus-9bf37cc9712506b2483650c82d3c41152337ef7e.tar.gz
fdk-aac-dabplus-9bf37cc9712506b2483650c82d3c41152337ef7e.tar.bz2
fdk-aac-dabplus-9bf37cc9712506b2483650c82d3c41152337ef7e.zip
Fraunhofer AAC codec.
License boilerplate update to follow. Change-Id: I2810460c11a58b6d148d84673cc031f3685e79b5
Diffstat (limited to 'libFDK/src/FDK_bitbuffer.cpp')
-rw-r--r--libFDK/src/FDK_bitbuffer.cpp427
1 files changed, 427 insertions, 0 deletions
diff --git a/libFDK/src/FDK_bitbuffer.cpp b/libFDK/src/FDK_bitbuffer.cpp
new file mode 100644
index 0000000..75a0375
--- /dev/null
+++ b/libFDK/src/FDK_bitbuffer.cpp
@@ -0,0 +1,427 @@
+/*************************** Fraunhofer IIS FDK Tools ***********************
+
+ (C) Copyright Fraunhofer IIS (2005)
+ All Rights Reserved
+
+ Please be advised that this software and/or program delivery is
+ Confidential Information of Fraunhofer and subject to and covered by the
+
+ Fraunhofer IIS Software Evaluation Agreement
+ between Google Inc. and Fraunhofer
+ effective and in full force since March 1, 2012.
+
+ You may use this software and/or program only under the terms and
+ conditions described in the above mentioned Fraunhofer IIS Software
+ Evaluation Agreement. Any other and/or further use requires a separate agreement.
+
+
+ $Id$
+ Author(s): M. Lohwasser
+ Description: common bitbuffer read/write routines
+
+ This software and/or program is protected by copyright law and international
+ treaties. Any reproduction or distribution of this software and/or program,
+ or any portion of it, may result in severe civil and criminal penalties, and
+ will be prosecuted to the maximum extent possible under law.
+
+******************************************************************************/
+
+#include "FDK_bitbuffer.h"
+
+
+
+
+#include "genericStds.h"
+#include "common_fix.h"
+#include "fixminmax.h"
+
+const UINT BitMask [32+1] =
+{
+ 0x0, 0x1, 0x3, 0x7,
+ 0xf, 0x1f, 0x3f, 0x7f,
+ 0xff, 0x1ff, 0x3ff, 0x7ff,
+ 0xfff, 0x1fff, 0x3fff, 0x7fff,
+ 0xffff, 0x1ffff, 0x3ffff, 0x7ffff,
+ 0xfffff, 0x1fffff, 0x3fffff, 0x7fffff,
+ 0xffffff, 0x1ffffff, 0x3ffffff, 0x7ffffff,
+ 0xfffffff, 0x1fffffff, 0x3fffffff, 0x7fffffff,
+ 0xffffffff
+};
+
+const UINT *const RESTRICT pBitMask = BitMask;
+
+void FDK_CreateBitBuffer (HANDLE_FDK_BITBUF *hBitBuf, UCHAR *pBuffer,
+ UINT bufSize)
+{
+ //*hBitBuf = (HANDLE_FDK_BITBUF) FDKcalloc(sizeof(FDK_BITBUF),1);
+
+ FDK_InitBitBuffer (*hBitBuf, pBuffer, bufSize, 0);
+
+ FDKmemclear((*hBitBuf)->Buffer, bufSize*sizeof(UCHAR));
+}
+
+void FDK_DeleteBitBuffer (HANDLE_FDK_BITBUF hBitBuf) { /*FDKfree((HANDLE_FDK_BITBUF)hBitBuf) */; }
+
+void FDK_InitBitBuffer (HANDLE_FDK_BITBUF hBitBuf, UCHAR *pBuffer,
+ UINT bufSize, UINT validBits)
+{
+ hBitBuf->ValidBits = validBits ;
+ hBitBuf->ReadOffset = 0 ;
+ hBitBuf->WriteOffset = 0 ;
+ hBitBuf->BitCnt = 0 ;
+ hBitBuf->BitNdx = 0 ;
+
+ hBitBuf->Buffer = pBuffer ;
+ hBitBuf->bufSize = bufSize ;
+ hBitBuf->bufBits = (bufSize << 3) ;
+#if defined(FDK_DEBUG) || defined(DEBUG)
+ /*assure bufsize (2^n) */
+ if (bufSize!=0) {
+ UINT x = 0, n=bufSize;
+ for (x=0; n>0; x++,n>>=1) {}
+ if ( bufSize != (1<<(x-1)) ) {
+ FDKprintfErr("Error: bufSizein FDK_InitBitBuffer() != (2^n), %d\n", bufSize);
+ }
+ }
+#endif
+}
+
+void FDK_ResetBitBuffer ( HANDLE_FDK_BITBUF hBitBuf )
+{
+ hBitBuf->ValidBits = 0 ;
+ hBitBuf->ReadOffset = 0 ;
+ hBitBuf->WriteOffset = 0 ;
+ hBitBuf->BitCnt = 0 ;
+ hBitBuf->BitNdx = 0 ;
+}
+
+INT FDK_get (HANDLE_FDK_BITBUF hBitBuf, const UINT numberOfBits)
+{
+ UINT byteOffset = hBitBuf->BitNdx >> 3 ;
+ UINT bitOffset = hBitBuf->BitNdx & 0x07 ;
+
+ hBitBuf->BitNdx = (hBitBuf->BitNdx + numberOfBits) & (hBitBuf->bufBits - 1) ;
+ hBitBuf->BitCnt += numberOfBits ;
+ hBitBuf->ValidBits -= numberOfBits ;
+
+ UINT byteMask = hBitBuf->bufSize - 1 ;
+
+ UINT tx = (hBitBuf->Buffer [ byteOffset & byteMask] << 24) |
+ (hBitBuf->Buffer [(byteOffset+1) & byteMask] << 16) |
+ (hBitBuf->Buffer [(byteOffset+2) & byteMask] << 8) |
+ hBitBuf->Buffer [(byteOffset+3) & byteMask];
+
+ if (bitOffset)
+ {
+ tx <<= bitOffset;
+ tx |= hBitBuf->Buffer [(byteOffset+4) & byteMask] >> (8-bitOffset);
+ }
+
+ return (tx >> (32 - numberOfBits)) ;
+}
+
+/* 01-DEC_2009 A. Tritthart
+ New routine added that reads 32 bits in order to allow a simple update of the 32-bit cache.
+ Due to push-back features, the read operations are not byte-aligned, nor 32-bit aligned,
+ all bytes need to be shifted according to the current read bit position.
+ Usage: bitstream->cache = FDK_get32 (bitstream);
+*/
+
+INT FDK_get32 (HANDLE_FDK_BITBUF hBitBuf)
+{
+ UINT BitNdx = hBitBuf->BitNdx + 32;
+ if (BitNdx <= hBitBuf->bufBits)
+ {
+ hBitBuf->BitNdx = BitNdx;
+ hBitBuf->BitCnt += 32;
+ hBitBuf->ValidBits -= 32;
+
+ UINT byteOffset = (BitNdx-1) >> 3;
+ UINT cache = (hBitBuf->Buffer[(byteOffset-3)] << 24) |
+ (hBitBuf->Buffer[(byteOffset-2)] << 16) |
+ (hBitBuf->Buffer[(byteOffset-1)] << 8) |
+ hBitBuf->Buffer[(byteOffset-0)];
+
+ if ( (BitNdx = (BitNdx&7)) != 0 ) {
+ cache = (cache >> (8-BitNdx)) | ((UINT)hBitBuf->Buffer [byteOffset-4] << (24+BitNdx));
+ }
+ return (cache) ;
+ }
+ else
+ {
+ /* exotic path, used only at the end of the buffer, when wrapping around */
+ int nBits = (INT)hBitBuf->bufBits-(INT)hBitBuf->BitNdx;
+
+ UINT cache = FDK_get (hBitBuf,nBits)<< (32-nBits);
+ cache |= (FDK_get (hBitBuf,32-nBits));
+ return (cache);
+ }
+}
+
+INT FDK_getBwd (HANDLE_FDK_BITBUF hBitBuf, const UINT numberOfBits)
+{
+ UINT byteOffset = hBitBuf->BitNdx >> 3 ;
+ UINT bitOffset = hBitBuf->BitNdx & 0x07 ;
+ UINT byteMask = hBitBuf->bufSize - 1 ;
+ int i;
+
+ hBitBuf->BitNdx = (hBitBuf->BitNdx - numberOfBits) & (hBitBuf->bufBits - 1) ;
+ hBitBuf->BitCnt -= numberOfBits ;
+ hBitBuf->ValidBits += numberOfBits ;
+
+ UINT tx = hBitBuf->Buffer [(byteOffset-3) & byteMask] << 24 |
+ hBitBuf->Buffer [(byteOffset-2) & byteMask] << 16 |
+ hBitBuf->Buffer [(byteOffset-1) & byteMask] << 8 |
+ hBitBuf->Buffer [ byteOffset & byteMask] ;
+ UINT txa = 0x0;
+
+ tx >>= (8 - bitOffset) ;
+
+ if (bitOffset && numberOfBits > 24)
+ {
+ tx |= hBitBuf->Buffer [(byteOffset-4) & byteMask] << (24 + bitOffset) ;
+ }
+
+ /* in place turn around */
+ for (i = 0; i < 16; i++) {
+ UINT bitMaskR = 0x00000001 << i;
+ UINT bitMaskL = 0x80000000 >> i;
+
+ txa |= (tx & bitMaskR) << (31 - (i<<1));
+ txa |= (tx & bitMaskL) >> (31 - (i<<1));
+ }
+
+ return (txa >> (32 - numberOfBits)) ;
+}
+
+void FDK_put (HANDLE_FDK_BITBUF hBitBuf, UINT value, const UINT numberOfBits)
+{
+ UINT byteOffset = hBitBuf->BitNdx >> 3 ;
+ UINT bitOffset = hBitBuf->BitNdx & 0x07 ;
+
+ hBitBuf->BitNdx = (hBitBuf->BitNdx + numberOfBits) & (hBitBuf->bufBits - 1) ;
+ hBitBuf->BitCnt += numberOfBits ;
+ hBitBuf->ValidBits += numberOfBits ;
+
+ UINT byteMask = hBitBuf->bufSize - 1 ;
+
+ UINT tmp = value<<(32-numberOfBits)>>bitOffset;
+ UINT mask = ~BitMask[(32-bitOffset)] | (BitMask [(32-numberOfBits)] >> bitOffset) ;
+
+ hBitBuf->Buffer [ byteOffset & byteMask] = (hBitBuf->Buffer [ byteOffset & byteMask]&(mask>>24)) | (UCHAR)(tmp>>24);
+ hBitBuf->Buffer [(byteOffset+1) & byteMask] = (hBitBuf->Buffer [(byteOffset+1) & byteMask]&(mask>>16)) | (UCHAR)(tmp>>16);
+ hBitBuf->Buffer [(byteOffset+2) & byteMask] = (hBitBuf->Buffer [(byteOffset+2) & byteMask]&(mask>>8)) | (UCHAR)(tmp>>8);
+ hBitBuf->Buffer [(byteOffset+3) & byteMask] = (hBitBuf->Buffer [(byteOffset+3) & byteMask]&(mask)) | (UCHAR)(tmp);
+
+ if (bitOffset && numberOfBits > 24)
+ {
+ hBitBuf->Buffer [(byteOffset+4) & byteMask] = (UCHAR)(value<<(40-numberOfBits)>>bitOffset) |
+ ( hBitBuf->Buffer [(byteOffset+4) & byteMask] & BitMask[(40-numberOfBits-bitOffset)] ) ;
+ }
+}
+
+void FDK_putBwd (HANDLE_FDK_BITBUF hBitBuf, UINT value, const UINT numberOfBits)
+{
+ UINT byteOffset = hBitBuf->BitNdx >> 3 ;
+ UINT bitOffset = 7 - (hBitBuf->BitNdx & 0x07) ;
+ UINT byteMask = hBitBuf->bufSize - 1 ;
+
+ UINT mask = ~(BitMask[numberOfBits] << bitOffset) ;
+ UINT tmp = 0x0000;
+ int i;
+
+ hBitBuf->BitNdx = (hBitBuf->BitNdx - numberOfBits) & (hBitBuf->bufBits - 1) ;
+ hBitBuf->BitCnt -= numberOfBits ;
+ hBitBuf->ValidBits -= numberOfBits ;
+
+ /* in place turn around */
+ for (i = 0; i < 16; i++) {
+ UINT bitMaskR = 0x00000001 << i;
+ UINT bitMaskL = 0x80000000 >> i;
+
+ tmp |= (value & bitMaskR) << (31 - (i<<1));
+ tmp |= (value & bitMaskL) >> (31 - (i<<1));
+ }
+ value = tmp;
+ tmp = value>>(32-numberOfBits)<<bitOffset;
+
+ hBitBuf->Buffer [ byteOffset & byteMask] = (hBitBuf->Buffer [ byteOffset & byteMask]&(mask)) | (UCHAR)(tmp);
+ hBitBuf->Buffer [(byteOffset-1) & byteMask] = (hBitBuf->Buffer [(byteOffset-1) & byteMask]&(mask>>8)) | (UCHAR)(tmp>>8);
+ hBitBuf->Buffer [(byteOffset-2) & byteMask] = (hBitBuf->Buffer [(byteOffset-2) & byteMask]&(mask>>16)) | (UCHAR)(tmp>>16);
+ hBitBuf->Buffer [(byteOffset-3) & byteMask] = (hBitBuf->Buffer [(byteOffset-3) & byteMask]&(mask>>24)) | (UCHAR)(tmp>>24);
+
+ if ((bitOffset + numberOfBits) > 32)
+ {
+ hBitBuf->Buffer [(byteOffset-4) & byteMask] = (UCHAR)(value>>(64-numberOfBits-bitOffset)) |
+ ( hBitBuf->Buffer [(byteOffset-4) & byteMask] & ~(BitMask[bitOffset] >> (32-numberOfBits)) ) ;
+ }
+}
+
+
+void FDK_pushBack (HANDLE_FDK_BITBUF hBitBuf, const UINT numberOfBits, UCHAR config)
+{
+ hBitBuf->BitCnt -= numberOfBits ;
+ hBitBuf->ValidBits += (config==0) ? numberOfBits : (-(INT)numberOfBits) ;
+ hBitBuf->BitNdx = (hBitBuf->BitNdx - numberOfBits) & (hBitBuf->bufBits - 1) ;
+}
+
+void FDK_pushForward (HANDLE_FDK_BITBUF hBitBuf, const UINT numberOfBits, UCHAR config)
+{
+ hBitBuf->BitCnt += numberOfBits ;
+ hBitBuf->ValidBits -= (config==0) ? numberOfBits : (-(INT)numberOfBits) ;
+ hBitBuf->BitNdx = (hBitBuf->BitNdx + numberOfBits) & (hBitBuf->bufBits - 1) ;
+}
+
+
+void FDK_byteAlign (HANDLE_FDK_BITBUF hBitBuf, UCHAR config)
+{
+ INT alignment = hBitBuf->BitCnt & 0x07 ;
+
+ if (alignment)
+ {
+ if (config==0)
+ FDK_pushForward (hBitBuf, 8 - alignment, config) ; /* BS_READER */
+ else
+ FDK_put (hBitBuf,0 , 8 - alignment) ; /* BS_WRITER */
+ }
+
+ hBitBuf->BitCnt = 0 ;
+}
+
+UINT FDK_getValidBits (HANDLE_FDK_BITBUF hBitBuf)
+{
+ return hBitBuf->ValidBits;
+}
+
+INT FDK_getFreeBits (HANDLE_FDK_BITBUF hBitBuf)
+{
+ return (hBitBuf->bufBits - hBitBuf->ValidBits) ;
+}
+
+void FDK_setBitCnt (HANDLE_FDK_BITBUF hBitBuf, const UINT value)
+{
+ hBitBuf->BitCnt = value ;
+}
+
+INT FDK_getBitCnt (HANDLE_FDK_BITBUF hBitBuf)
+{
+ return hBitBuf->BitCnt ;
+}
+
+void FDK_Feed(HANDLE_FDK_BITBUF hBitBuf,
+ UCHAR *RESTRICT inputBuffer,
+ const UINT bufferSize,
+ UINT *bytesValid)
+{
+ inputBuffer = &inputBuffer [bufferSize - *bytesValid] ;
+
+ UINT bTotal = 0 ;
+
+ UINT bToRead = (hBitBuf->bufBits - hBitBuf->ValidBits) >> 3 ;
+ UINT noOfBytes = fMin(bToRead, *bytesValid); //(bToRead < *bytesValid) ? bToRead : *bytesValid ;
+
+ while (noOfBytes > 0)
+ {
+ /* split read to buffer size */
+ bToRead = hBitBuf->bufSize - hBitBuf->ReadOffset ;
+ bToRead = fMin(bToRead, noOfBytes); //(bToRead < noOfBytes) ? bToRead : noOfBytes ;
+
+ /* copy 'bToRead' bytes from 'ptr' to inputbuffer */
+ FDKmemcpy(&hBitBuf->Buffer[hBitBuf->ReadOffset], inputBuffer, bToRead*sizeof(UCHAR));
+
+ /* add noOfBits to number of valid bits in buffer */
+ hBitBuf->ValidBits += bToRead << 3 ;
+ bTotal += bToRead ;
+ inputBuffer += bToRead ;
+
+ hBitBuf->ReadOffset = (hBitBuf->ReadOffset + bToRead) & (hBitBuf->bufSize - 1) ;
+ noOfBytes -= bToRead ;
+ }
+
+ *bytesValid -= bTotal ;
+}
+
+void CopyAlignedBlock (HANDLE_FDK_BITBUF h_BitBufSrc, UCHAR *RESTRICT dstBuffer, UINT bToRead)
+{
+ UINT byteOffset = h_BitBufSrc->BitNdx >> 3 ;
+ const UINT byteMask = h_BitBufSrc->bufSize - 1 ;
+
+ UCHAR *RESTRICT pBBB = h_BitBufSrc->Buffer;
+ for (UINT i = 0 ; i < bToRead ; i++)
+ {
+ dstBuffer[i] = pBBB[(byteOffset+i) & byteMask] ;
+ }
+
+ bToRead <<= 3 ;
+
+ h_BitBufSrc->BitNdx = (h_BitBufSrc->BitNdx + bToRead) & (h_BitBufSrc->bufBits - 1) ;
+ h_BitBufSrc->BitCnt += bToRead ;
+ h_BitBufSrc->ValidBits -= bToRead ;
+}
+
+void FDK_Copy (HANDLE_FDK_BITBUF h_BitBufDst, HANDLE_FDK_BITBUF h_BitBufSrc, UINT *bytesValid)
+{
+ INT bTotal = 0;
+
+ /* limit noOfBytes to valid bytes in src buffer and available bytes in dst buffer */
+ UINT bToRead = h_BitBufSrc->ValidBits >> 3 ;
+ UINT noOfBytes = fMin(bToRead, *bytesValid); //(*bytesValid < bToRead) ? *bytesValid : bToRead ;
+ bToRead = FDK_getFreeBits(h_BitBufDst);
+ noOfBytes = fMin(bToRead, noOfBytes); //(bToRead < noOfBytes) ? bToRead : noOfBytes;
+
+ while (noOfBytes > 0)
+ {
+ /* Split Read to buffer size */
+ bToRead = h_BitBufDst->bufSize - h_BitBufDst->ReadOffset ;
+ bToRead = fMin(noOfBytes, bToRead); //(noOfBytes < bToRead) ? noOfBytes : bToRead ;
+
+ /* copy 'bToRead' bytes from buffer to buffer */
+ if (!(h_BitBufSrc->BitNdx & 0x07)) {
+ CopyAlignedBlock (h_BitBufSrc, h_BitBufDst->Buffer + h_BitBufDst->ReadOffset, bToRead) ;
+ } else {
+ for (UINT i = 0; i < bToRead; i++)
+ {
+ h_BitBufDst->Buffer [h_BitBufDst->ReadOffset + i] = (UCHAR)FDK_get(h_BitBufSrc,8);
+ }
+ }
+
+ /* add noOfBits to number of valid bits in buffer */
+ h_BitBufDst->ValidBits += bToRead << 3 ;
+ bTotal += bToRead;
+
+ h_BitBufDst->ReadOffset = (h_BitBufDst->ReadOffset + bToRead) & (h_BitBufDst->bufSize-1);
+ noOfBytes -= bToRead;
+ }
+
+ *bytesValid -=bTotal;
+}
+
+void FDK_Fetch (HANDLE_FDK_BITBUF hBitBuf, UCHAR *outBuf, UINT *writeBytes)
+{
+ UCHAR *RESTRICT outputBuffer = outBuf;
+ UINT bTotal = 0 ;
+
+ UINT bToWrite = (hBitBuf->ValidBits) >> 3 ;
+ UINT noOfBytes = fMin(bToWrite, *writeBytes); //(bToWrite < *writeBytes) ? bToWrite : *writeBytes ;
+
+ while (noOfBytes > 0)
+ {
+ /* split write to buffer size */
+ bToWrite = hBitBuf->bufSize - hBitBuf->WriteOffset ;
+ bToWrite = fMin(bToWrite, noOfBytes); //(bToWrite < noOfBytes) ? bToWrite : noOfBytes ;
+
+ /* copy 'bToWrite' bytes from bitbuffer to outputbuffer */
+ FDKmemcpy(outputBuffer, &hBitBuf->Buffer[hBitBuf->WriteOffset], bToWrite*sizeof(UCHAR));
+
+ /* sub noOfBits from number of valid bits in buffer */
+ hBitBuf->ValidBits -= bToWrite << 3 ;
+ bTotal += bToWrite ;
+ outputBuffer += bToWrite ;
+
+ hBitBuf->WriteOffset = (hBitBuf->WriteOffset + bToWrite) & (hBitBuf->bufSize - 1) ;
+ noOfBytes -= bToWrite ;
+ }
+
+ *writeBytes = bTotal ;
+}
+