1 // Copyright 2015 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 #include "JBig2_PddProc.h"
8
9 #include <memory>
10
11 #include "JBig2_GrdProc.h"
12 #include "JBig2_Image.h"
13 #include "JBig2_PatternDict.h"
14
decode_Arith(CJBig2_ArithDecoder * pArithDecoder,JBig2ArithCtx * gbContext,IFX_Pause * pPause)15 CJBig2_PatternDict* CJBig2_PDDProc::decode_Arith(
16 CJBig2_ArithDecoder* pArithDecoder,
17 JBig2ArithCtx* gbContext,
18 IFX_Pause* pPause) {
19 FX_DWORD GRAY;
20 CJBig2_Image* BHDC = nullptr;
21 std::unique_ptr<CJBig2_PatternDict> pDict(new CJBig2_PatternDict());
22 pDict->NUMPATS = GRAYMAX + 1;
23 pDict->HDPATS = FX_Alloc(CJBig2_Image*, pDict->NUMPATS);
24 JBIG2_memset(pDict->HDPATS, 0, sizeof(CJBig2_Image*) * pDict->NUMPATS);
25
26 std::unique_ptr<CJBig2_GRDProc> pGRD(new CJBig2_GRDProc());
27 pGRD->MMR = HDMMR;
28 pGRD->GBW = (GRAYMAX + 1) * HDPW;
29 pGRD->GBH = HDPH;
30 pGRD->GBTEMPLATE = HDTEMPLATE;
31 pGRD->TPGDON = 0;
32 pGRD->USESKIP = 0;
33 pGRD->GBAT[0] = -(int32_t)HDPW;
34 pGRD->GBAT[1] = 0;
35 if (pGRD->GBTEMPLATE == 0) {
36 pGRD->GBAT[2] = -3;
37 pGRD->GBAT[3] = -1;
38 pGRD->GBAT[4] = 2;
39 pGRD->GBAT[5] = -2;
40 pGRD->GBAT[6] = -2;
41 pGRD->GBAT[7] = -2;
42 }
43 FXCODEC_STATUS status =
44 pGRD->Start_decode_Arith(&BHDC, pArithDecoder, gbContext, nullptr);
45 while (status == FXCODEC_STATUS_DECODE_TOBECONTINUE) {
46 pGRD->Continue_decode(pPause);
47 }
48 if (!BHDC)
49 return nullptr;
50
51 GRAY = 0;
52 while (GRAY <= GRAYMAX) {
53 pDict->HDPATS[GRAY] = BHDC->subImage(HDPW * GRAY, 0, HDPW, HDPH);
54 GRAY = GRAY + 1;
55 }
56 delete BHDC;
57 return pDict.release();
58 }
59
decode_MMR(CJBig2_BitStream * pStream,IFX_Pause * pPause)60 CJBig2_PatternDict* CJBig2_PDDProc::decode_MMR(CJBig2_BitStream* pStream,
61 IFX_Pause* pPause) {
62 FX_DWORD GRAY;
63 CJBig2_Image* BHDC = nullptr;
64 std::unique_ptr<CJBig2_PatternDict> pDict(new CJBig2_PatternDict());
65 pDict->NUMPATS = GRAYMAX + 1;
66 pDict->HDPATS = FX_Alloc(CJBig2_Image*, pDict->NUMPATS);
67 JBIG2_memset(pDict->HDPATS, 0, sizeof(CJBig2_Image*) * pDict->NUMPATS);
68
69 std::unique_ptr<CJBig2_GRDProc> pGRD(new CJBig2_GRDProc());
70 pGRD->MMR = HDMMR;
71 pGRD->GBW = (GRAYMAX + 1) * HDPW;
72 pGRD->GBH = HDPH;
73 FXCODEC_STATUS status = pGRD->Start_decode_MMR(&BHDC, pStream, nullptr);
74 while (status == FXCODEC_STATUS_DECODE_TOBECONTINUE) {
75 pGRD->Continue_decode(pPause);
76 }
77 if (!BHDC)
78 return nullptr;
79
80 GRAY = 0;
81 while (GRAY <= GRAYMAX) {
82 pDict->HDPATS[GRAY] = BHDC->subImage(HDPW * GRAY, 0, HDPW, HDPH);
83 GRAY = GRAY + 1;
84 }
85 delete BHDC;
86 return pDict.release();
87 }
88