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

#include "JBig2_ArithIntDecoder.h"
CJBig2_ArithIntDecoder::CJBig2_ArithIntDecoder() {
  IAx = (JBig2ArithCtx*)m_pModule->JBig2_Malloc2(sizeof(JBig2ArithCtx), 512);
  JBIG2_memset(IAx, 0, sizeof(JBig2ArithCtx) * 512);
}
CJBig2_ArithIntDecoder::~CJBig2_ArithIntDecoder() {
  m_pModule->JBig2_Free(IAx);
}
int CJBig2_ArithIntDecoder::decode(CJBig2_ArithDecoder* pArithDecoder,
                                   int* nResult) {
  int PREV, V;
  int S, D;
  int nNeedBits, nTemp, i;
  PREV = 1;
  S = pArithDecoder->DECODE(IAx + PREV);
  PREV = (PREV << 1) | S;
  D = pArithDecoder->DECODE(IAx + PREV);
  PREV = (PREV << 1) | D;
  if (D) {
    D = pArithDecoder->DECODE(IAx + PREV);
    PREV = (PREV << 1) | D;
    if (D) {
      D = pArithDecoder->DECODE(IAx + PREV);
      PREV = (PREV << 1) | D;
      if (D) {
        D = pArithDecoder->DECODE(IAx + PREV);
        PREV = (PREV << 1) | D;
        if (D) {
          D = pArithDecoder->DECODE(IAx + PREV);
          PREV = (PREV << 1) | D;
          if (D) {
            nNeedBits = 32;
            V = 4436;
          } else {
            nNeedBits = 12;
            V = 340;
          }
        } else {
          nNeedBits = 8;
          V = 84;
        }
      } else {
        nNeedBits = 6;
        V = 20;
      }
    } else {
      nNeedBits = 4;
      V = 4;
    }
  } else {
    nNeedBits = 2;
    V = 0;
  }
  nTemp = 0;
  for (i = 0; i < nNeedBits; i++) {
    D = pArithDecoder->DECODE(IAx + PREV);
    if (PREV < 256) {
      PREV = (PREV << 1) | D;
    } else {
      PREV = (((PREV << 1) | D) & 511) | 256;
    }
    nTemp = (nTemp << 1) | D;
  }
  V += nTemp;
  if (S == 1 && V > 0) {
    V = -V;
  }
  *nResult = V;
  if (S == 1 && V == 0) {
    return JBIG2_OOB;
  }
  return 0;
}
CJBig2_ArithIaidDecoder::CJBig2_ArithIaidDecoder(unsigned char SBSYMCODELENA) {
  SBSYMCODELEN = SBSYMCODELENA;
  IAID = (JBig2ArithCtx*)m_pModule->JBig2_Malloc2(sizeof(JBig2ArithCtx),
                                                  (1 << SBSYMCODELEN));
  JBIG2_memset(IAID, 0, sizeof(JBig2ArithCtx) * (int)(1 << SBSYMCODELEN));
}
CJBig2_ArithIaidDecoder::~CJBig2_ArithIaidDecoder() {
  m_pModule->JBig2_Free(IAID);
}
int CJBig2_ArithIaidDecoder::decode(CJBig2_ArithDecoder* pArithDecoder,
                                    int* nResult) {
  int PREV;
  int D;
  int i;
  PREV = 1;
  for (i = 0; i < SBSYMCODELEN; i++) {
    D = pArithDecoder->DECODE(IAID + PREV);
    PREV = (PREV << 1) | D;
  }
  PREV = PREV - (1 << SBSYMCODELEN);
  *nResult = PREV;
  return 0;
}