| OLD | NEW |
| 1 // Copyright 2014 PDFium Authors. All rights reserved. | 1 // Copyright 2014 PDFium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com | 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
| 6 | 6 |
| 7 #ifndef CORE_FXCODEC_JBIG2_JBIG2_CONTEXT_H_ | 7 #ifndef CORE_FXCODEC_JBIG2_JBIG2_CONTEXT_H_ |
| 8 #define CORE_FXCODEC_JBIG2_JBIG2_CONTEXT_H_ | 8 #define CORE_FXCODEC_JBIG2_JBIG2_CONTEXT_H_ |
| 9 | 9 |
| 10 #include <list> | 10 #include <list> |
| 11 #include <memory> | 11 #include <memory> |
| 12 #include <utility> | 12 #include <utility> |
| 13 #include <vector> |
| 13 | 14 |
| 14 #include "core/fpdfapi/fpdf_parser/include/cpdf_object.h" | 15 #include "core/fpdfapi/fpdf_parser/include/cpdf_object.h" |
| 15 #include "core/fxcodec/include/fx_codec_def.h" | 16 #include "core/fxcodec/include/fx_codec_def.h" |
| 16 #include "core/fxcodec/jbig2/JBig2_List.h" | 17 #include "core/fxcodec/jbig2/JBig2_List.h" |
| 17 #include "core/fxcodec/jbig2/JBig2_Page.h" | 18 #include "core/fxcodec/jbig2/JBig2_Page.h" |
| 18 #include "core/fxcodec/jbig2/JBig2_Segment.h" | 19 #include "core/fxcodec/jbig2/JBig2_Segment.h" |
| 19 | 20 |
| 20 class CJBig2_ArithDecoder; | 21 class CJBig2_ArithDecoder; |
| 21 class CJBig2_GRDProc; | 22 class CJBig2_GRDProc; |
| 22 class CPDF_StreamAcc; | 23 class CPDF_StreamAcc; |
| (...skipping 10 matching lines...) Expand all Loading... |
| 33 #define JBIG2_ERROR_FATAL -3 | 34 #define JBIG2_ERROR_FATAL -3 |
| 34 #define JBIG2_END_OF_PAGE 2 | 35 #define JBIG2_END_OF_PAGE 2 |
| 35 #define JBIG2_END_OF_FILE 3 | 36 #define JBIG2_END_OF_FILE 3 |
| 36 #define JBIG2_ERROR_FILE_FORMAT -4 | 37 #define JBIG2_ERROR_FILE_FORMAT -4 |
| 37 #define JBIG2_ERROR_STREAM_TYPE -5 | 38 #define JBIG2_ERROR_STREAM_TYPE -5 |
| 38 #define JBIG2_ERROR_LIMIT -6 | 39 #define JBIG2_ERROR_LIMIT -6 |
| 39 #define JBIG2_MIN_SEGMENT_SIZE 11 | 40 #define JBIG2_MIN_SEGMENT_SIZE 11 |
| 40 | 41 |
| 41 class CJBig2_Context { | 42 class CJBig2_Context { |
| 42 public: | 43 public: |
| 43 static CJBig2_Context* CreateContext( | 44 CJBig2_Context(CPDF_StreamAcc* pGlobalStream, |
| 44 CPDF_StreamAcc* pGlobalStream, | 45 CPDF_StreamAcc* pSrcStream, |
| 45 CPDF_StreamAcc* pSrcStream, | 46 std::list<CJBig2_CachePair>* pSymbolDictCache, |
| 46 std::list<CJBig2_CachePair>* pSymbolDictCache, | 47 IFX_Pause* pPause, |
| 47 IFX_Pause* pPause = nullptr); | 48 bool bIsGlobal); |
| 48 | 49 ~CJBig2_Context(); |
| 49 static void DestroyContext(CJBig2_Context* pContext); | |
| 50 | 50 |
| 51 int32_t getFirstPage(uint8_t* pBuf, | 51 int32_t getFirstPage(uint8_t* pBuf, |
| 52 int32_t width, | 52 int32_t width, |
| 53 int32_t height, | 53 int32_t height, |
| 54 int32_t stride, | 54 int32_t stride, |
| 55 IFX_Pause* pPause); | 55 IFX_Pause* pPause); |
| 56 | 56 |
| 57 int32_t Continue(IFX_Pause* pPause); | 57 int32_t Continue(IFX_Pause* pPause); |
| 58 FXCODEC_STATUS GetProcessingStatus() { return m_ProcessingStatus; } | 58 FXCODEC_STATUS GetProcessingStatus() { return m_ProcessingStatus; } |
| 59 | 59 |
| 60 private: | 60 private: |
| 61 CJBig2_Context(CPDF_StreamAcc* pGlobalStream, | |
| 62 CPDF_StreamAcc* pSrcStream, | |
| 63 std::list<CJBig2_CachePair>* pSymbolDictCache, | |
| 64 IFX_Pause* pPause, | |
| 65 bool bIsGlobal); | |
| 66 | |
| 67 ~CJBig2_Context(); | |
| 68 | |
| 69 int32_t decode_SquentialOrgnazation(IFX_Pause* pPause); | 61 int32_t decode_SquentialOrgnazation(IFX_Pause* pPause); |
| 70 | |
| 71 int32_t decode_EmbedOrgnazation(IFX_Pause* pPause); | 62 int32_t decode_EmbedOrgnazation(IFX_Pause* pPause); |
| 72 | |
| 73 int32_t decode_RandomOrgnazation_FirstPage(IFX_Pause* pPause); | 63 int32_t decode_RandomOrgnazation_FirstPage(IFX_Pause* pPause); |
| 74 | |
| 75 int32_t decode_RandomOrgnazation(IFX_Pause* pPause); | 64 int32_t decode_RandomOrgnazation(IFX_Pause* pPause); |
| 76 | 65 |
| 77 CJBig2_Segment* findSegmentByNumber(uint32_t dwNumber); | 66 CJBig2_Segment* findSegmentByNumber(uint32_t dwNumber); |
| 78 | |
| 79 CJBig2_Segment* findReferredSegmentByTypeAndIndex(CJBig2_Segment* pSegment, | 67 CJBig2_Segment* findReferredSegmentByTypeAndIndex(CJBig2_Segment* pSegment, |
| 80 uint8_t cType, | 68 uint8_t cType, |
| 81 int32_t nIndex); | 69 int32_t nIndex); |
| 82 | 70 |
| 83 int32_t parseSegmentHeader(CJBig2_Segment* pSegment); | 71 int32_t parseSegmentHeader(CJBig2_Segment* pSegment); |
| 84 | |
| 85 int32_t parseSegmentData(CJBig2_Segment* pSegment, IFX_Pause* pPause); | 72 int32_t parseSegmentData(CJBig2_Segment* pSegment, IFX_Pause* pPause); |
| 86 int32_t ProcessingParseSegmentData(CJBig2_Segment* pSegment, | 73 int32_t ProcessingParseSegmentData(CJBig2_Segment* pSegment, |
| 87 IFX_Pause* pPause); | 74 IFX_Pause* pPause); |
| 88 | |
| 89 int32_t parseSymbolDict(CJBig2_Segment* pSegment, IFX_Pause* pPause); | 75 int32_t parseSymbolDict(CJBig2_Segment* pSegment, IFX_Pause* pPause); |
| 90 | |
| 91 int32_t parseTextRegion(CJBig2_Segment* pSegment); | 76 int32_t parseTextRegion(CJBig2_Segment* pSegment); |
| 92 | |
| 93 int32_t parsePatternDict(CJBig2_Segment* pSegment, IFX_Pause* pPause); | 77 int32_t parsePatternDict(CJBig2_Segment* pSegment, IFX_Pause* pPause); |
| 94 | |
| 95 int32_t parseHalftoneRegion(CJBig2_Segment* pSegment, IFX_Pause* pPause); | 78 int32_t parseHalftoneRegion(CJBig2_Segment* pSegment, IFX_Pause* pPause); |
| 96 | |
| 97 int32_t parseGenericRegion(CJBig2_Segment* pSegment, IFX_Pause* pPause); | 79 int32_t parseGenericRegion(CJBig2_Segment* pSegment, IFX_Pause* pPause); |
| 98 | |
| 99 int32_t parseGenericRefinementRegion(CJBig2_Segment* pSegment); | 80 int32_t parseGenericRefinementRegion(CJBig2_Segment* pSegment); |
| 100 | |
| 101 int32_t parseTable(CJBig2_Segment* pSegment); | 81 int32_t parseTable(CJBig2_Segment* pSegment); |
| 102 | |
| 103 int32_t parseRegionInfo(JBig2RegionInfo* pRI); | 82 int32_t parseRegionInfo(JBig2RegionInfo* pRI); |
| 104 | 83 |
| 105 JBig2HuffmanCode* decodeSymbolIDHuffmanTable(CJBig2_BitStream* pStream, | 84 JBig2HuffmanCode* decodeSymbolIDHuffmanTable(CJBig2_BitStream* pStream, |
| 106 uint32_t SBNUMSYMS); | 85 uint32_t SBNUMSYMS); |
| 107 | 86 |
| 108 void huffman_assign_code(int* CODES, int* PREFLEN, int NTEMP); | 87 void huffman_assign_code(int* CODES, int* PREFLEN, int NTEMP); |
| 109 | |
| 110 void huffman_assign_code(JBig2HuffmanCode* SBSYMCODES, int NTEMP); | 88 void huffman_assign_code(JBig2HuffmanCode* SBSYMCODES, int NTEMP); |
| 111 | 89 |
| 112 private: | 90 std::unique_ptr<CJBig2_Context> m_pGlobalContext; |
| 113 CJBig2_Context* m_pGlobalContext; | |
| 114 std::unique_ptr<CJBig2_BitStream> m_pStream; | 91 std::unique_ptr<CJBig2_BitStream> m_pStream; |
| 115 CJBig2_List<CJBig2_Segment> m_SegmentList; | 92 CJBig2_List<CJBig2_Segment> m_SegmentList; |
| 116 CJBig2_List<JBig2PageInfo> m_PageInfoList; | 93 CJBig2_List<JBig2PageInfo> m_PageInfoList; |
| 117 std::unique_ptr<CJBig2_Image> m_pPage; | 94 std::unique_ptr<CJBig2_Image> m_pPage; |
| 118 size_t m_nSegmentDecoded; | 95 size_t m_nSegmentDecoded; |
| 119 bool m_bInPage; | 96 bool m_bInPage; |
| 120 bool m_bBufSpecified; | 97 bool m_bBufSpecified; |
| 121 int32_t m_PauseStep; | 98 int32_t m_PauseStep; |
| 122 IFX_Pause* m_pPause; | 99 IFX_Pause* const m_pPause; |
| 123 FXCODEC_STATUS m_ProcessingStatus; | 100 FXCODEC_STATUS m_ProcessingStatus; |
| 101 std::vector<JBig2ArithCtx> m_gbContext; |
| 124 std::unique_ptr<CJBig2_ArithDecoder> m_pArithDecoder; | 102 std::unique_ptr<CJBig2_ArithDecoder> m_pArithDecoder; |
| 125 std::unique_ptr<CJBig2_GRDProc> m_pGRD; | 103 std::unique_ptr<CJBig2_GRDProc> m_pGRD; |
| 126 JBig2ArithCtx* m_gbContext; | |
| 127 std::unique_ptr<CJBig2_Segment> m_pSegment; | 104 std::unique_ptr<CJBig2_Segment> m_pSegment; |
| 128 uint32_t m_dwOffset; | 105 uint32_t m_dwOffset; |
| 129 JBig2RegionInfo m_ri; | 106 JBig2RegionInfo m_ri; |
| 130 std::list<CJBig2_CachePair>* const m_pSymbolDictCache; | 107 std::list<CJBig2_CachePair>* const m_pSymbolDictCache; |
| 131 bool m_bIsGlobal; | 108 bool m_bIsGlobal; |
| 132 }; | 109 }; |
| 133 | 110 |
| 134 #endif // CORE_FXCODEC_JBIG2_JBIG2_CONTEXT_H_ | 111 #endif // CORE_FXCODEC_JBIG2_JBIG2_CONTEXT_H_ |
| OLD | NEW |