| // Copyright 2015 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 |
| |
| #include "JBig2_GsidProc.h" |
| |
| #include "../../../../third_party/base/nonstd_unique_ptr.h" |
| #include "../../../include/fxcrt/fx_basic.h" |
| #include "JBig2_BitStream.h" |
| #include "JBig2_GrdProc.h" |
| #include "JBig2_Image.h" |
| |
| FX_DWORD* CJBig2_GSIDProc::decode_Arith(CJBig2_ArithDecoder* pArithDecoder, |
| JBig2ArithCtx* gbContext, |
| IFX_Pause* pPause) { |
| CJBig2_Image** GSPLANES; |
| int32_t J, K; |
| FX_DWORD x, y; |
| FX_DWORD* GSVALS; |
| GSPLANES = FX_Alloc(CJBig2_Image*, GSBPP); |
| GSVALS = FX_Alloc2D(FX_DWORD, GSW, GSH); |
| JBIG2_memset(GSPLANES, 0, sizeof(CJBig2_Image*) * GSBPP); |
| JBIG2_memset(GSVALS, 0, sizeof(FX_DWORD) * GSW * GSH); |
| |
| nonstd::unique_ptr<CJBig2_GRDProc> pGRD(new CJBig2_GRDProc()); |
| pGRD->MMR = GSMMR; |
| pGRD->GBW = GSW; |
| pGRD->GBH = GSH; |
| pGRD->GBTEMPLATE = GSTEMPLATE; |
| pGRD->TPGDON = 0; |
| pGRD->USESKIP = GSUSESKIP; |
| pGRD->SKIP = GSKIP; |
| if (GSTEMPLATE <= 1) { |
| pGRD->GBAT[0] = 3; |
| } else { |
| pGRD->GBAT[0] = 2; |
| } |
| pGRD->GBAT[1] = -1; |
| if (pGRD->GBTEMPLATE == 0) { |
| pGRD->GBAT[2] = -3; |
| pGRD->GBAT[3] = -1; |
| pGRD->GBAT[4] = 2; |
| pGRD->GBAT[5] = -2; |
| pGRD->GBAT[6] = -2; |
| pGRD->GBAT[7] = -2; |
| } |
| FXCODEC_STATUS status = pGRD->Start_decode_Arith( |
| &GSPLANES[GSBPP - 1], pArithDecoder, gbContext, nullptr); |
| while (status == FXCODEC_STATUS_DECODE_TOBECONTINUE) { |
| pGRD->Continue_decode(pPause); |
| } |
| if (!GSPLANES[GSBPP - 1]) { |
| goto failed; |
| } |
| J = GSBPP - 2; |
| while (J >= 0) { |
| FXCODEC_STATUS status = pGRD->Start_decode_Arith( |
| &GSPLANES[J], pArithDecoder, gbContext, nullptr); |
| while (status == FXCODEC_STATUS_DECODE_TOBECONTINUE) { |
| pGRD->Continue_decode(pPause); |
| } |
| if (!GSPLANES[J]) { |
| for (K = GSBPP - 1; K > J; K--) { |
| delete GSPLANES[K]; |
| goto failed; |
| } |
| } |
| GSPLANES[J]->composeFrom(0, 0, GSPLANES[J + 1], JBIG2_COMPOSE_XOR); |
| J = J - 1; |
| } |
| for (y = 0; y < GSH; y++) { |
| for (x = 0; x < GSW; x++) { |
| for (J = 0; J < GSBPP; J++) { |
| GSVALS[y * GSW + x] |= GSPLANES[J]->getPixel(x, y) << J; |
| } |
| } |
| } |
| for (J = 0; J < GSBPP; J++) { |
| delete GSPLANES[J]; |
| } |
| FX_Free(GSPLANES); |
| return GSVALS; |
| failed: |
| FX_Free(GSPLANES); |
| FX_Free(GSVALS); |
| return nullptr; |
| } |
| |
| FX_DWORD* CJBig2_GSIDProc::decode_MMR(CJBig2_BitStream* pStream, |
| IFX_Pause* pPause) { |
| CJBig2_Image** GSPLANES; |
| int32_t J, K; |
| FX_DWORD x, y; |
| FX_DWORD* GSVALS; |
| GSPLANES = FX_Alloc(CJBig2_Image*, GSBPP); |
| GSVALS = FX_Alloc2D(FX_DWORD, GSW, GSH); |
| JBIG2_memset(GSPLANES, 0, sizeof(CJBig2_Image*) * GSBPP); |
| JBIG2_memset(GSVALS, 0, sizeof(FX_DWORD) * GSW * GSH); |
| |
| nonstd::unique_ptr<CJBig2_GRDProc> pGRD(new CJBig2_GRDProc()); |
| pGRD->MMR = GSMMR; |
| pGRD->GBW = GSW; |
| pGRD->GBH = GSH; |
| FXCODEC_STATUS status = |
| pGRD->Start_decode_MMR(&GSPLANES[GSBPP - 1], pStream, nullptr); |
| while (status == FXCODEC_STATUS_DECODE_TOBECONTINUE) { |
| pGRD->Continue_decode(pPause); |
| } |
| if (!GSPLANES[GSBPP - 1]) { |
| goto failed; |
| } |
| pStream->alignByte(); |
| pStream->offset(3); |
| J = GSBPP - 2; |
| while (J >= 0) { |
| FXCODEC_STATUS status = |
| pGRD->Start_decode_MMR(&GSPLANES[J], pStream, nullptr); |
| while (status == FXCODEC_STATUS_DECODE_TOBECONTINUE) { |
| pGRD->Continue_decode(pPause); |
| } |
| if (!GSPLANES[J]) { |
| for (K = GSBPP - 1; K > J; K--) { |
| delete GSPLANES[K]; |
| goto failed; |
| } |
| } |
| pStream->alignByte(); |
| pStream->offset(3); |
| GSPLANES[J]->composeFrom(0, 0, GSPLANES[J + 1], JBIG2_COMPOSE_XOR); |
| J = J - 1; |
| } |
| for (y = 0; y < GSH; y++) { |
| for (x = 0; x < GSW; x++) { |
| for (J = 0; J < GSBPP; J++) { |
| GSVALS[y * GSW + x] |= GSPLANES[J]->getPixel(x, y) << J; |
| } |
| } |
| } |
| for (J = 0; J < GSBPP; J++) { |
| delete GSPLANES[J]; |
| } |
| FX_Free(GSPLANES); |
| return GSVALS; |
| failed: |
| FX_Free(GSPLANES); |
| FX_Free(GSVALS); |
| return nullptr; |
| } |