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 #include "core/fxcodec/codec/ccodec_jbig2module.h" 8 9 #include <list> 10 #include <memory> 11 12 #include "core/fpdfapi/parser/cpdf_stream_acc.h" 13 #include "core/fxcodec/JBig2_DocumentContext.h" 14 #include "core/fxcodec/jbig2/JBig2_Context.h" 15 #include "core/fxcodec/jbig2/JBig2_Image.h" 16 #include "core/fxcrt/fx_memory.h" 17 #include "third_party/base/ptr_util.h" 18 19 JBig2_DocumentContext::JBig2_DocumentContext() {} 20 21 JBig2_DocumentContext::~JBig2_DocumentContext() {} 22 23 JBig2_DocumentContext* GetJBig2DocumentContext( 24 std::unique_ptr<JBig2_DocumentContext>* pContextHolder) { 25 if (!pContextHolder->get()) 26 *pContextHolder = pdfium::MakeUnique<JBig2_DocumentContext>(); 27 return pContextHolder->get(); 28 } 29 30 CCodec_Jbig2Context::CCodec_Jbig2Context() 31 : m_width(0), 32 m_height(0), 33 m_pGlobalStream(nullptr), 34 m_pSrcStream(nullptr), 35 m_dest_buf(0), 36 m_dest_pitch(0) {} 37 38 CCodec_Jbig2Context::~CCodec_Jbig2Context() {} 39 40 CCodec_Jbig2Module::~CCodec_Jbig2Module() {} 41 42 FXCODEC_STATUS CCodec_Jbig2Module::StartDecode( 43 CCodec_Jbig2Context* pJbig2Context, 44 std::unique_ptr<JBig2_DocumentContext>* pContextHolder, 45 uint32_t width, 46 uint32_t height, 47 const RetainPtr<CPDF_StreamAcc>& src_stream, 48 const RetainPtr<CPDF_StreamAcc>& global_stream, 49 uint8_t* dest_buf, 50 uint32_t dest_pitch, 51 IFX_PauseIndicator* pPause) { 52 if (!pJbig2Context) 53 return FXCODEC_STATUS_ERR_PARAMS; 54 55 JBig2_DocumentContext* pJBig2DocumentContext = 56 GetJBig2DocumentContext(pContextHolder); 57 pJbig2Context->m_width = width; 58 pJbig2Context->m_height = height; 59 pJbig2Context->m_pSrcStream = src_stream; 60 pJbig2Context->m_pGlobalStream = global_stream; 61 pJbig2Context->m_dest_buf = dest_buf; 62 pJbig2Context->m_dest_pitch = dest_pitch; 63 memset(dest_buf, 0, height * dest_pitch); 64 pJbig2Context->m_pContext = pdfium::MakeUnique<CJBig2_Context>( 65 global_stream, src_stream, pJBig2DocumentContext->GetSymbolDictCache(), 66 false); 67 int ret = pJbig2Context->m_pContext->getFirstPage(dest_buf, width, height, 68 dest_pitch, pPause); 69 return Decode(pJbig2Context, ret); 70 } 71 72 FXCODEC_STATUS CCodec_Jbig2Module::ContinueDecode( 73 CCodec_Jbig2Context* pJbig2Context, 74 IFX_PauseIndicator* pPause) { 75 int ret = pJbig2Context->m_pContext->Continue(pPause); 76 return Decode(pJbig2Context, ret); 77 } 78 79 FXCODEC_STATUS CCodec_Jbig2Module::Decode(CCodec_Jbig2Context* pJbig2Context, 80 int result) { 81 FXCODEC_STATUS status = pJbig2Context->m_pContext->GetProcessingStatus(); 82 if (status != FXCODEC_STATUS_DECODE_FINISH) 83 return status; 84 85 pJbig2Context->m_pContext.reset(); 86 if (result != JBIG2_SUCCESS) 87 return FXCODEC_STATUS_ERROR; 88 89 int dword_size = pJbig2Context->m_height * pJbig2Context->m_dest_pitch / 4; 90 uint32_t* dword_buf = reinterpret_cast<uint32_t*>(pJbig2Context->m_dest_buf); 91 for (int i = 0; i < dword_size; i++) 92 dword_buf[i] = ~dword_buf[i]; 93 return FXCODEC_STATUS_DECODE_FINISH; 94 } 95