Various changes to JBig2 cache:
[pdfium.git] / core / src / fxcodec / jbig2 / JBig2_BitStream.h
index 8ed473a..c7c50b8 100644 (file)
-// Copyright 2014 PDFium Authors. All rights reserved.\r
-// Use of this source code is governed by a BSD-style license that can be\r
-// found in the LICENSE file.\r
\r
-// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com\r
-\r
-#ifndef _JBIG2_BIT_STREAM_H_\r
-#define _JBIG2_BIT_STREAM_H_\r
-#include "JBig2_Define.h"\r
-class CJBig2_BitStream : public CJBig2_Object\r
-{\r
-public:\r
-\r
-    CJBig2_BitStream(FX_BYTE *pBuffer, FX_DWORD dwLength);\r
-\r
-    CJBig2_BitStream(CJBig2_BitStream &bs);\r
-\r
-    ~CJBig2_BitStream();\r
-\r
-    FX_INT32 readNBits(FX_DWORD nBits, FX_DWORD *dwResult);\r
-\r
-    FX_INT32 readNBits(FX_DWORD nBits, FX_INT32 *nResult);\r
-\r
-    FX_INT32 read1Bit(FX_DWORD *dwResult);\r
-\r
-    FX_INT32 read1Bit(FX_BOOL  *bResult);\r
-\r
-    FX_INT32 read1Byte(FX_BYTE *cResult);\r
-\r
-    FX_INT32 readInteger(FX_DWORD *dwResult);\r
-\r
-    FX_INT32 readShortInteger(FX_WORD *wResult);\r
-\r
-    void alignByte();\r
-\r
-    void align4Byte();\r
-\r
-    FX_BYTE getAt(FX_DWORD dwOffset);\r
-\r
-    FX_BYTE getCurByte();\r
-\r
-    FX_BYTE getNextByte();\r
-\r
-    FX_INT32 incByteIdx();\r
-\r
-    FX_BYTE getCurByte_arith();\r
-\r
-    FX_BYTE getNextByte_arith();\r
-\r
-    FX_DWORD getOffset();\r
-\r
-    void setOffset(FX_DWORD dwOffset);\r
-\r
-    FX_DWORD getBitPos();\r
-\r
-    void setBitPos(FX_DWORD dwBitPos);\r
-\r
-    FX_BYTE *getBuf();\r
-\r
-    FX_DWORD getLength()\r
-    {\r
-        return m_dwLength;\r
-    }\r
-\r
-    FX_BYTE *getPointer();\r
-\r
-    void offset(FX_DWORD dwOffset);\r
-\r
-    FX_DWORD getByteLeft();\r
-private:\r
-\r
-    FX_BYTE *m_pBuf;\r
-\r
-    FX_DWORD m_dwLength;\r
-\r
-    FX_DWORD m_dwByteIdx;\r
-\r
-    FX_DWORD m_dwBitIdx;\r
-};\r
-inline CJBig2_BitStream::CJBig2_BitStream(FX_BYTE *pBuffer, FX_DWORD dwLength)\r
-{\r
-    m_pBuf = pBuffer;\r
-    m_dwLength = dwLength;\r
-    m_dwByteIdx = 0;\r
-    m_dwBitIdx  = 0;\r
-    if (m_dwLength > 256 * 1024 * 1024) {\r
-        m_dwLength = 0;\r
-        m_pBuf = NULL;\r
-    }\r
-}\r
-inline CJBig2_BitStream::CJBig2_BitStream(CJBig2_BitStream &bs)\r
-{\r
-    m_pBuf = bs.m_pBuf;\r
-    m_dwLength = bs.m_dwLength;\r
-    m_dwByteIdx = bs.m_dwByteIdx;\r
-    m_dwBitIdx = bs.m_dwBitIdx;\r
-}\r
-inline CJBig2_BitStream::~CJBig2_BitStream()\r
-{\r
-}\r
-inline FX_INT32 CJBig2_BitStream::readNBits(FX_DWORD dwBits, FX_DWORD *dwResult)\r
-{\r
-    FX_DWORD dwTemp = (m_dwByteIdx << 3) + m_dwBitIdx;\r
-    if(dwTemp <= (m_dwLength << 3)) {\r
-        *dwResult = 0;\r
-        if(dwTemp + dwBits <= (m_dwLength << 3)) {\r
-            dwTemp = dwBits;\r
-        } else {\r
-            dwTemp = (m_dwLength << 3) - dwTemp;\r
-        }\r
-        while(dwTemp > 0) {\r
-            *dwResult = (*dwResult << 1) | ((m_pBuf[m_dwByteIdx] >> (7 - m_dwBitIdx)) & 0x01);\r
-            if(m_dwBitIdx == 7) {\r
-                m_dwByteIdx ++;\r
-                m_dwBitIdx = 0;\r
-            } else {\r
-                m_dwBitIdx ++;\r
-            }\r
-            dwTemp --;\r
-        }\r
-        return 0;\r
-    } else {\r
-        return -1;\r
-    }\r
-}\r
-inline FX_INT32 CJBig2_BitStream::readNBits(FX_DWORD dwBits, FX_INT32 *nResult)\r
-{\r
-    FX_DWORD dwTemp = (m_dwByteIdx << 3) + m_dwBitIdx;\r
-    if(dwTemp <= (m_dwLength << 3)) {\r
-        *nResult = 0;\r
-        if(dwTemp + dwBits <= (m_dwLength << 3)) {\r
-            dwTemp = dwBits;\r
-        } else {\r
-            dwTemp = (m_dwLength << 3) - dwTemp;\r
-        }\r
-        while(dwTemp > 0) {\r
-            *nResult = (*nResult << 1) | ((m_pBuf[m_dwByteIdx] >> (7 - m_dwBitIdx)) & 0x01);\r
-            if(m_dwBitIdx == 7) {\r
-                m_dwByteIdx ++;\r
-                m_dwBitIdx = 0;\r
-            } else {\r
-                m_dwBitIdx ++;\r
-            }\r
-            dwTemp --;\r
-        }\r
-        return 0;\r
-    } else {\r
-        return -1;\r
-    }\r
-}\r
-\r
-inline FX_INT32 CJBig2_BitStream::read1Bit(FX_DWORD *dwResult)\r
-{\r
-    if(m_dwByteIdx < m_dwLength) {\r
-        *dwResult = (m_pBuf[m_dwByteIdx] >> (7 - m_dwBitIdx)) & 0x01;\r
-        if(m_dwBitIdx == 7) {\r
-            m_dwByteIdx ++;\r
-            m_dwBitIdx = 0;\r
-        } else {\r
-            m_dwBitIdx ++;\r
-        }\r
-        return 0;\r
-    } else {\r
-        return -1;\r
-    }\r
-}\r
-\r
-inline FX_INT32 CJBig2_BitStream::read1Bit(FX_BOOL *bResult)\r
-{\r
-    if(m_dwByteIdx < m_dwLength) {\r
-        *bResult = (m_pBuf[m_dwByteIdx] >> (7 - m_dwBitIdx)) & 0x01;\r
-        if(m_dwBitIdx == 7) {\r
-            m_dwByteIdx ++;\r
-            m_dwBitIdx = 0;\r
-        } else {\r
-            m_dwBitIdx ++;\r
-        }\r
-        return 0;\r
-    } else {\r
-        return -1;\r
-    }\r
-}\r
-inline FX_INT32 CJBig2_BitStream::read1Byte(FX_BYTE *cResult)\r
-{\r
-    if(m_dwByteIdx < m_dwLength) {\r
-        *cResult = m_pBuf[m_dwByteIdx];\r
-        m_dwByteIdx ++;\r
-        return 0;\r
-    } else {\r
-        return -1;\r
-    }\r
-}\r
-\r
-inline FX_INT32 CJBig2_BitStream::readInteger(FX_DWORD *dwResult)\r
-{\r
-    if(m_dwByteIdx + 3 < m_dwLength) {\r
-        *dwResult = (m_pBuf[m_dwByteIdx] << 24) | (m_pBuf[m_dwByteIdx + 1] << 16)\r
-                    | (m_pBuf[m_dwByteIdx + 2] << 8) | m_pBuf[m_dwByteIdx + 3];\r
-        m_dwByteIdx += 4;\r
-        return 0;\r
-    } else {\r
-        return -1;\r
-    }\r
-}\r
-\r
-inline FX_INT32 CJBig2_BitStream::readShortInteger(FX_WORD *dwResult)\r
-{\r
-    if(m_dwByteIdx + 1 < m_dwLength) {\r
-        *dwResult = (m_pBuf[m_dwByteIdx] << 8) | m_pBuf[m_dwByteIdx + 1];\r
-        m_dwByteIdx += 2;\r
-        return 0;\r
-    } else {\r
-        return -1;\r
-    }\r
-}\r
-inline void CJBig2_BitStream::alignByte()\r
-{\r
-    if(m_dwBitIdx != 0) {\r
-        m_dwByteIdx ++;\r
-        m_dwBitIdx = 0;\r
-    }\r
-}\r
-inline void CJBig2_BitStream::align4Byte()\r
-{\r
-    if(m_dwBitIdx != 0) {\r
-        m_dwByteIdx ++;\r
-        m_dwBitIdx = 0;\r
-    }\r
-    m_dwByteIdx = (m_dwByteIdx + 3) & -4;\r
-}\r
-inline FX_BYTE CJBig2_BitStream::getAt(FX_DWORD dwOffset)\r
-{\r
-    if(dwOffset < m_dwLength) {\r
-        return m_pBuf[dwOffset];\r
-    } else {\r
-        return 0;\r
-    }\r
-}\r
-inline FX_BYTE CJBig2_BitStream::getCurByte()\r
-{\r
-    if(m_dwByteIdx < m_dwLength) {\r
-        return m_pBuf[m_dwByteIdx];\r
-    } else {\r
-        return 0;\r
-    }\r
-}\r
-inline FX_BYTE CJBig2_BitStream::getNextByte()\r
-{\r
-    if(m_dwByteIdx + 1 < m_dwLength) {\r
-        return m_pBuf[m_dwByteIdx + 1];\r
-    } else {\r
-        return 0;\r
-    }\r
-}\r
-inline FX_INT32 CJBig2_BitStream::incByteIdx()\r
-{\r
-    if(m_dwByteIdx < m_dwLength) {\r
-        m_dwByteIdx ++;\r
-        return 0;\r
-    } else {\r
-        return -1;\r
-    }\r
-}\r
-inline FX_BYTE CJBig2_BitStream::getCurByte_arith()\r
-{\r
-    if(m_dwByteIdx < m_dwLength) {\r
-        return m_pBuf[m_dwByteIdx];\r
-    } else {\r
-        return 0xff;\r
-    }\r
-}\r
-inline FX_BYTE CJBig2_BitStream::getNextByte_arith()\r
-{\r
-    if(m_dwByteIdx + 1 < m_dwLength) {\r
-        return m_pBuf[m_dwByteIdx + 1];\r
-    } else {\r
-        return 0xff;\r
-    }\r
-}\r
-inline FX_DWORD CJBig2_BitStream::getOffset()\r
-{\r
-    return m_dwByteIdx;\r
-}\r
-inline void CJBig2_BitStream::setOffset(FX_DWORD dwOffset)\r
-{\r
-    if (dwOffset > m_dwLength) {\r
-        dwOffset = m_dwLength;\r
-    }\r
-    m_dwByteIdx = dwOffset;\r
-}\r
-inline FX_DWORD CJBig2_BitStream::getBitPos()\r
-{\r
-    return (m_dwByteIdx << 3) + m_dwBitIdx;\r
-}\r
-inline void CJBig2_BitStream::setBitPos(FX_DWORD dwBitPos)\r
-{\r
-    m_dwByteIdx = dwBitPos >> 3;\r
-    m_dwBitIdx = dwBitPos & 7;\r
-}\r
-inline FX_BYTE *CJBig2_BitStream::getBuf()\r
-{\r
-    return m_pBuf;\r
-}\r
-inline FX_BYTE *CJBig2_BitStream::getPointer()\r
-{\r
-    return m_pBuf + m_dwByteIdx;\r
-}\r
-inline void CJBig2_BitStream::offset(FX_DWORD dwOffset)\r
-{\r
-    m_dwByteIdx += dwOffset;\r
-}\r
-inline FX_DWORD CJBig2_BitStream::getByteLeft()\r
-{\r
-    return m_dwLength - m_dwByteIdx;\r
-}\r
-#endif\r
+// Copyright 2014 PDFium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
+
+#ifndef CORE_SRC_FXCODEC_JBIG2_JBIG2_BITSTREAM_H_
+#define CORE_SRC_FXCODEC_JBIG2_JBIG2_BITSTREAM_H_
+
+#include "../../../include/fxcrt/fx_basic.h"
+
+class CPDF_StreamAcc;
+
+class CJBig2_BitStream {
+ public:
+  explicit CJBig2_BitStream(CPDF_StreamAcc* pSrcStream);
+  ~CJBig2_BitStream();
+
+  // TODO(thestig): readFoo() should return bool.
+  int32_t readNBits(FX_DWORD nBits, FX_DWORD* dwResult);
+  int32_t readNBits(FX_DWORD nBits, int32_t* nResult);
+  int32_t read1Bit(FX_DWORD* dwResult);
+  int32_t read1Bit(FX_BOOL* bResult);
+  int32_t read1Byte(uint8_t* cResult);
+  int32_t readInteger(FX_DWORD* dwResult);
+  int32_t readShortInteger(FX_WORD* wResult);
+  void alignByte();
+  uint8_t getCurByte() const;
+  void incByteIdx();
+  uint8_t getCurByte_arith() const;
+  uint8_t getNextByte_arith() const;
+  FX_DWORD getOffset() const;
+  void setOffset(FX_DWORD dwOffset);
+  FX_DWORD getBitPos() const;
+  void setBitPos(FX_DWORD dwBitPos);
+  const uint8_t* getBuf() const;
+  FX_DWORD getLength() const { return m_dwLength; }
+  const uint8_t* getPointer() const;
+  void offset(FX_DWORD dwOffset);
+  FX_DWORD getByteLeft() const;
+  FX_DWORD getObjNum() const;
+
+ private:
+  void AdvanceBit();
+  bool IsInBound() const;
+  FX_DWORD LengthInBits() const;
+
+  const uint8_t* m_pBuf;
+  FX_DWORD m_dwLength;
+  FX_DWORD m_dwByteIdx;
+  FX_DWORD m_dwBitIdx;
+  const FX_DWORD m_dwObjNum;
+
+  CJBig2_BitStream(const CJBig2_BitStream&) = delete;
+  void operator=(const CJBig2_BitStream&) = delete;
+};
+
+#endif  // CORE_SRC_FXCODEC_JBIG2_JBIG2_BITSTREAM_H_