1 // Copyright 2014 PDFium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com 6 7 #ifndef CORE_FXCODEC_JBIG2_JBIG2_BITSTREAM_H_ 8 #define CORE_FXCODEC_JBIG2_JBIG2_BITSTREAM_H_ 9 10 #include "core/fxcrt/retain_ptr.h" 11 #include "third_party/base/span.h" 12 13 class CJBig2_BitStream { 14 public: 15 CJBig2_BitStream(pdfium::span<const uint8_t> pSrcStream, uint32_t dwObjNum); 16 CJBig2_BitStream(const CJBig2_BitStream&) = delete; 17 CJBig2_BitStream& operator=(const CJBig2_BitStream&) = delete; 18 ~CJBig2_BitStream(); 19 20 // TODO(thestig): readFoo() should return bool. 21 int32_t readNBits(uint32_t dwBits, uint32_t* dwResult); 22 int32_t readNBits(uint32_t dwBits, int32_t* nResult); 23 int32_t read1Bit(uint32_t* dwResult); 24 int32_t read1Bit(bool* bResult); 25 int32_t read1Byte(uint8_t* cResult); 26 int32_t readInteger(uint32_t* dwResult); 27 int32_t readShortInteger(uint16_t* wResult); 28 void alignByte(); 29 uint8_t getCurByte() const; 30 void incByteIdx(); 31 uint8_t getCurByte_arith() const; 32 uint8_t getNextByte_arith() const; 33 uint32_t getOffset() const; 34 void setOffset(uint32_t dwOffset); 35 uint32_t getBitPos() const; 36 void setBitPos(uint32_t dwBitPos); 37 const uint8_t* getBuf() const; getLength()38 uint32_t getLength() const { return m_Span.size(); } 39 const uint8_t* getPointer() const; 40 void offset(uint32_t dwOffset); 41 uint32_t getByteLeft() const; 42 uint32_t getObjNum() const; 43 bool IsInBounds() const; 44 45 private: 46 void AdvanceBit(); 47 uint32_t LengthInBits() const; 48 49 const pdfium::span<const uint8_t> m_Span; 50 uint32_t m_dwByteIdx = 0; 51 uint32_t m_dwBitIdx = 0; 52 const uint32_t m_dwObjNum; 53 }; 54 55 #endif // CORE_FXCODEC_JBIG2_JBIG2_BITSTREAM_H_ 56