// 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 _JBIG2_ARITHMETIC_DECODER_H_
#define _JBIG2_ARITHMETIC_DECODER_H_
#include "JBig2_Define.h"
#include "JBig2_BitStream.h"
#include "JBig2_ArithQe.h"
typedef struct {
    unsigned int MPS;
    unsigned int I;
} JBig2ArithCtx;
class CJBig2_ArithDecoder : public CJBig2_Object
{
public:

    CJBig2_ArithDecoder(CJBig2_BitStream *pStream);

    ~CJBig2_ArithDecoder();

    int DECODE(JBig2ArithCtx *pCX);
private:

    void INITDEC();

    void BYTEIN();
    unsigned char B;
    unsigned int C;
    unsigned int A;
    unsigned int CT;
    CJBig2_BitStream *m_pStream;
};
inline CJBig2_ArithDecoder::CJBig2_ArithDecoder(CJBig2_BitStream *pStream)
{
    m_pStream = pStream;
    INITDEC();
}
inline CJBig2_ArithDecoder::~CJBig2_ArithDecoder()
{
}
inline void CJBig2_ArithDecoder::INITDEC()
{
    B = m_pStream->getCurByte_arith();
    C = (B ^ 0xff) << 16;;
    BYTEIN();
    C = C << 7;
    CT = CT - 7;
    A = 0x8000;
}
inline void CJBig2_ArithDecoder::BYTEIN()
{
    unsigned char B1;
    if(B == 0xff) {
        B1 = m_pStream->getNextByte_arith();
        if(B1 > 0x8f) {
            CT = 8;
        } else {
            m_pStream->incByteIdx();
            B = B1;
            C = C + 0xfe00 - (B << 9);
            CT = 7;
        }
    } else {
        m_pStream->incByteIdx();
        B = m_pStream->getCurByte_arith();
        C = C + 0xff00 - (B << 8);
        CT = 8;
    }
}
inline int CJBig2_ArithDecoder::DECODE(JBig2ArithCtx *pCX)
{
    if (!pCX || pCX->I >= JBIG2_QE_NUM) {
        return 0;
    }

    int D;
    const JBig2ArithQe * qe = &QeTable[pCX->I];
    A = A - qe->Qe;
    if((C >> 16) < A) {
        if(A & 0x8000) {
            D = pCX->MPS;
        } else {
            if(A < qe->Qe) {
                D = 1 - pCX->MPS;
                if(qe->nSwitch == 1) {
                    pCX->MPS = 1 - pCX->MPS;
                }
                pCX->I = qe->NLPS;
            } else {
                D = pCX->MPS;
                pCX->I = qe->NMPS;
            }
            do {
                if (CT == 0) {
                    BYTEIN();
                }
                A <<= 1;
                C <<= 1;
                CT--;
            } while ((A & 0x8000) == 0);
        }
    } else {
        C -= A << 16;
        if(A < qe->Qe) {
            A = qe->Qe;
            D = pCX->MPS;
            pCX->I = qe->NMPS;
        } else {
            A = qe->Qe;
            D = 1 - pCX->MPS;
            if(qe->nSwitch == 1) {
                pCX->MPS = 1 - pCX->MPS;
            }
            pCX->I = qe->NLPS;
        }
        do {
            if (CT == 0) {
                BYTEIN();
            }
            A <<= 1;
            C <<= 1;
            CT--;
        } while ((A & 0x8000) == 0);
    }
    return D;
}
#endif