// 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_