// 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_CONTEXT_H_
#define _JBIG2_CONTEXT_H_

#include <list>
#include <utility>

#include "JBig2_Module.h"
#include "JBig2_List.h"
#include "JBig2_Segment.h"
#include "JBig2_Page.h"
#include "JBig2_GeneralDecoder.h"
#include "../../../include/fxcodec/fx_codec_def.h"

typedef std::pair<uint8_t*, CJBig2_SymbolDict*> CJBig2_CachePair;
typedef enum {
    JBIG2_OUT_OF_PAGE = 0,
    JBIG2_IN_PAGE,
} JBig2State;
#define JBIG2_SUCCESS			 0
#define JBIG2_FAILED			-1
#define JBIG2_ERROR_TOO_SHORT	-2
#define JBIG2_ERROR_FATAL		-3
#define JBIG2_END_OF_PAGE		 2
#define JBIG2_END_OF_FILE		 3
#define JBIG2_ERROR_FILE_FORMAT -4
#define JBIG2_ERROR_STREAM_TYPE -5
#define JBIG2_ERROR_LIMIT		-6
#define JBIG2_FILE_STREAM			0
#define JBIG2_SQUENTIAL_STREAM		1
#define JBIG2_RANDOM_STREAM			2
#define JBIG2_EMBED_STREAM			3
#define JBIG2_MIN_SEGMENT_SIZE					11
class CJBig2_Context : public CJBig2_Object
{
public:

    static CJBig2_Context *CreateContext(CJBig2_Module *pModule, uint8_t *pGlobalData, FX_DWORD dwGlobalLength,
                                         uint8_t *pData, FX_DWORD dwLength, int32_t nStreamType, std::list<CJBig2_CachePair>* pSymbolDictCache, IFX_Pause* pPause = NULL);

    static void DestroyContext(CJBig2_Context *pContext);

    int32_t getFirstPage(uint8_t *pBuf, int32_t width, int32_t height, int32_t stride, IFX_Pause* pPause);

    int32_t getNextPage(uint8_t *pBuf, int32_t width, int32_t height, int32_t stride, IFX_Pause* pPause);

    int32_t getFirstPage(CJBig2_Image **image, IFX_Pause* pPause);

    int32_t getNextPage(CJBig2_Image **image, IFX_Pause* pPause);
    int32_t Continue(IFX_Pause* pPause);
    FXCODEC_STATUS GetProcessiveStatus()
    {
        return m_ProcessiveStatus;
    };
private:

    CJBig2_Context(uint8_t *pGlobalData, FX_DWORD dwGlobalLength,
                   uint8_t *pData, FX_DWORD dwLength, int32_t nStreamType, std::list<CJBig2_CachePair>* pSymbolDictCache, IFX_Pause* pPause);

    ~CJBig2_Context();

    int32_t decodeFile(IFX_Pause* pPause);

    int32_t decode_SquentialOrgnazation(IFX_Pause* pPause);

    int32_t decode_EmbedOrgnazation(IFX_Pause* pPause);

    int32_t decode_RandomOrgnazation_FirstPage(IFX_Pause* pPause);

    int32_t decode_RandomOrgnazation(IFX_Pause* pPause);

    CJBig2_Segment *findSegmentByNumber(FX_DWORD dwNumber);

    CJBig2_Segment *findReferredSegmentByTypeAndIndex(CJBig2_Segment *pSegment, uint8_t cType, int32_t nIndex);

    int32_t parseSegmentHeader(CJBig2_Segment *pSegment);

    int32_t parseSegmentData(CJBig2_Segment *pSegment, IFX_Pause* pPause);
    int32_t ProcessiveParseSegmentData(CJBig2_Segment *pSegment, IFX_Pause* pPause);

    int32_t parseSymbolDict(CJBig2_Segment *pSegment, IFX_Pause* pPause);

    int32_t parseTextRegion(CJBig2_Segment *pSegment);

    int32_t parsePatternDict(CJBig2_Segment *pSegment, IFX_Pause* pPause);

    int32_t parseHalftoneRegion(CJBig2_Segment *pSegment, IFX_Pause* pPause);

    int32_t parseGenericRegion(CJBig2_Segment *pSegment, IFX_Pause* pPause);

    int32_t parseGenericRefinementRegion(CJBig2_Segment *pSegment);

    int32_t parseTable(CJBig2_Segment *pSegment);

    int32_t parseRegionInfo(JBig2RegionInfo *pRI);


    JBig2HuffmanCode *decodeSymbolIDHuffmanTable(CJBig2_BitStream *pStream, FX_DWORD SBNUMSYMS);

    void huffman_assign_code(int* CODES, int* PREFLEN, int NTEMP);

    void huffman_assign_code(JBig2HuffmanCode *SBSYMCODES, int NTEMP);

private:

    CJBig2_Context *m_pGlobalContext;

    int32_t m_nStreamType;

    CJBig2_BitStream *m_pStream;

    int32_t m_nState;

    CJBig2_List<CJBig2_Segment> *m_pSegmentList;

    CJBig2_List<JBig2PageInfo> *m_pPageInfoList;

    CJBig2_Image *m_pPage;

    bool m_bBufSpecified;

    int32_t m_nSegmentDecoded;
    IFX_Pause*	m_pPause;
    int32_t	m_PauseStep;
    FXCODEC_STATUS m_ProcessiveStatus;
    bool	m_bFirstPage;
    CJBig2_ArithDecoder *m_pArithDecoder;
    CJBig2_GRDProc *m_pGRD;
    JBig2ArithCtx *m_gbContext;
    CJBig2_Segment *m_pSegment;
    FX_DWORD m_dwOffset;
    JBig2RegionInfo m_ri;
    std::list<CJBig2_CachePair>* m_pSymbolDictCache;
};
#endif