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