| // Copyright 2014 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 "../../../include/fxcodec/fx_codec.h" |
| #include "codec_int.h" |
| |
| CCodec_Jbig2Context::CCodec_Jbig2Context() { |
| FXSYS_memset(this, 0, sizeof(CCodec_Jbig2Context)); |
| } |
| CCodec_Jbig2Module::~CCodec_Jbig2Module() {} |
| void* CCodec_Jbig2Module::CreateJbig2Context() { |
| return new CCodec_Jbig2Context(); |
| } |
| void CCodec_Jbig2Module::DestroyJbig2Context(void* pJbig2Content) { |
| if (pJbig2Content) { |
| CJBig2_Context::DestroyContext( |
| ((CCodec_Jbig2Context*)pJbig2Content)->m_pContext); |
| delete (CCodec_Jbig2Context*)pJbig2Content; |
| } |
| pJbig2Content = NULL; |
| } |
| FX_BOOL CCodec_Jbig2Module::Decode(FX_DWORD width, |
| FX_DWORD height, |
| const uint8_t* src_buf, |
| FX_DWORD src_size, |
| const uint8_t* global_data, |
| FX_DWORD global_size, |
| uint8_t* dest_buf, |
| FX_DWORD dest_pitch) { |
| FXSYS_memset(dest_buf, 0, height * dest_pitch); |
| CJBig2_Context* pContext = CJBig2_Context::CreateContext( |
| &m_Module, (uint8_t*)global_data, global_size, (uint8_t*)src_buf, |
| src_size, JBIG2_EMBED_STREAM, &m_SymbolDictCache); |
| if (pContext == NULL) { |
| return FALSE; |
| } |
| int ret = pContext->getFirstPage(dest_buf, width, height, dest_pitch, NULL); |
| CJBig2_Context::DestroyContext(pContext); |
| if (ret != JBIG2_SUCCESS) { |
| return FALSE; |
| } |
| int dword_size = height * dest_pitch / 4; |
| FX_DWORD* dword_buf = (FX_DWORD*)dest_buf; |
| for (int i = 0; i < dword_size; i++) { |
| dword_buf[i] = ~dword_buf[i]; |
| } |
| return TRUE; |
| } |
| FX_BOOL CCodec_Jbig2Module::Decode(IFX_FileRead* file_ptr, |
| FX_DWORD& width, |
| FX_DWORD& height, |
| FX_DWORD& pitch, |
| uint8_t*& dest_buf) { |
| CJBig2_Context* pContext = NULL; |
| CJBig2_Image* dest_image = NULL; |
| FX_DWORD src_size = (FX_DWORD)file_ptr->GetSize(); |
| uint8_t* src_buf = FX_Alloc(uint8_t, src_size); |
| int ret = 0; |
| if (!file_ptr->ReadBlock(src_buf, 0, src_size)) { |
| goto failed; |
| } |
| pContext = |
| CJBig2_Context::CreateContext(&m_Module, NULL, 0, src_buf, src_size, |
| JBIG2_FILE_STREAM, &m_SymbolDictCache); |
| if (pContext == NULL) { |
| goto failed; |
| } |
| ret = pContext->getFirstPage(&dest_image, NULL); |
| CJBig2_Context::DestroyContext(pContext); |
| if (ret != JBIG2_SUCCESS) { |
| goto failed; |
| } |
| width = (FX_DWORD)dest_image->m_nWidth; |
| height = (FX_DWORD)dest_image->m_nHeight; |
| pitch = (FX_DWORD)dest_image->m_nStride; |
| dest_buf = dest_image->m_pData; |
| dest_image->m_bNeedFree = FALSE; |
| delete dest_image; |
| FX_Free(src_buf); |
| return TRUE; |
| failed: |
| FX_Free(src_buf); |
| return FALSE; |
| } |
| FXCODEC_STATUS CCodec_Jbig2Module::StartDecode(void* pJbig2Context, |
| FX_DWORD width, |
| FX_DWORD height, |
| const uint8_t* src_buf, |
| FX_DWORD src_size, |
| const uint8_t* global_data, |
| FX_DWORD global_size, |
| uint8_t* dest_buf, |
| FX_DWORD dest_pitch, |
| IFX_Pause* pPause) { |
| if (!pJbig2Context) { |
| return FXCODEC_STATUS_ERR_PARAMS; |
| } |
| CCodec_Jbig2Context* m_pJbig2Context = (CCodec_Jbig2Context*)pJbig2Context; |
| m_pJbig2Context->m_width = width; |
| m_pJbig2Context->m_height = height; |
| m_pJbig2Context->m_src_buf = (unsigned char*)src_buf; |
| m_pJbig2Context->m_src_size = src_size; |
| m_pJbig2Context->m_global_data = global_data; |
| m_pJbig2Context->m_global_size = global_size; |
| m_pJbig2Context->m_dest_buf = dest_buf; |
| m_pJbig2Context->m_dest_pitch = dest_pitch; |
| m_pJbig2Context->m_pPause = pPause; |
| m_pJbig2Context->m_bFileReader = FALSE; |
| FXSYS_memset(dest_buf, 0, height * dest_pitch); |
| m_pJbig2Context->m_pContext = CJBig2_Context::CreateContext( |
| &m_Module, (uint8_t*)global_data, global_size, (uint8_t*)src_buf, |
| src_size, JBIG2_EMBED_STREAM, &m_SymbolDictCache, pPause); |
| if (!m_pJbig2Context->m_pContext) { |
| return FXCODEC_STATUS_ERROR; |
| } |
| int ret = m_pJbig2Context->m_pContext->getFirstPage(dest_buf, width, height, |
| dest_pitch, pPause); |
| if (m_pJbig2Context->m_pContext->GetProcessiveStatus() == |
| FXCODEC_STATUS_DECODE_FINISH) { |
| CJBig2_Context::DestroyContext(m_pJbig2Context->m_pContext); |
| m_pJbig2Context->m_pContext = NULL; |
| if (ret != JBIG2_SUCCESS) { |
| return FXCODEC_STATUS_ERROR; |
| } |
| int dword_size = height * dest_pitch / 4; |
| FX_DWORD* dword_buf = (FX_DWORD*)dest_buf; |
| for (int i = 0; i < dword_size; i++) { |
| dword_buf[i] = ~dword_buf[i]; |
| } |
| return FXCODEC_STATUS_DECODE_FINISH; |
| } |
| return m_pJbig2Context->m_pContext->GetProcessiveStatus(); |
| } |
| FXCODEC_STATUS CCodec_Jbig2Module::StartDecode(void* pJbig2Context, |
| IFX_FileRead* file_ptr, |
| FX_DWORD& width, |
| FX_DWORD& height, |
| FX_DWORD& pitch, |
| uint8_t*& dest_buf, |
| IFX_Pause* pPause) { |
| if (!pJbig2Context) { |
| return FXCODEC_STATUS_ERR_PARAMS; |
| } |
| CCodec_Jbig2Context* m_pJbig2Context = (CCodec_Jbig2Context*)pJbig2Context; |
| m_pJbig2Context->m_bFileReader = TRUE; |
| m_pJbig2Context->m_dest_image = NULL; |
| m_pJbig2Context->m_src_size = (FX_DWORD)file_ptr->GetSize(); |
| m_pJbig2Context->m_src_buf = FX_Alloc(uint8_t, m_pJbig2Context->m_src_size); |
| int ret = 0; |
| if (!file_ptr->ReadBlock((void*)m_pJbig2Context->m_src_buf, 0, |
| m_pJbig2Context->m_src_size)) { |
| goto failed; |
| } |
| m_pJbig2Context->m_pContext = CJBig2_Context::CreateContext( |
| &m_Module, NULL, 0, m_pJbig2Context->m_src_buf, |
| m_pJbig2Context->m_src_size, JBIG2_FILE_STREAM, &m_SymbolDictCache, |
| pPause); |
| if (m_pJbig2Context->m_pContext == NULL) { |
| goto failed; |
| } |
| ret = m_pJbig2Context->m_pContext->getFirstPage( |
| &m_pJbig2Context->m_dest_image, pPause); |
| if (m_pJbig2Context->m_pContext->GetProcessiveStatus() == |
| FXCODEC_STATUS_DECODE_TOBECONTINUE) { |
| width = (FX_DWORD)m_pJbig2Context->m_dest_image->m_nWidth; |
| height = (FX_DWORD)m_pJbig2Context->m_dest_image->m_nHeight; |
| pitch = (FX_DWORD)m_pJbig2Context->m_dest_image->m_nStride; |
| dest_buf = m_pJbig2Context->m_dest_image->m_pData; |
| m_pJbig2Context->m_dest_image->m_bNeedFree = FALSE; |
| return FXCODEC_STATUS_DECODE_TOBECONTINUE; |
| } |
| CJBig2_Context::DestroyContext(m_pJbig2Context->m_pContext); |
| m_pJbig2Context->m_pContext = NULL; |
| if (ret != JBIG2_SUCCESS) { |
| goto failed; |
| } |
| width = (FX_DWORD)m_pJbig2Context->m_dest_image->m_nWidth; |
| height = (FX_DWORD)m_pJbig2Context->m_dest_image->m_nHeight; |
| pitch = (FX_DWORD)m_pJbig2Context->m_dest_image->m_nStride; |
| dest_buf = m_pJbig2Context->m_dest_image->m_pData; |
| m_pJbig2Context->m_dest_image->m_bNeedFree = FALSE; |
| delete m_pJbig2Context->m_dest_image; |
| FX_Free(m_pJbig2Context->m_src_buf); |
| return FXCODEC_STATUS_DECODE_FINISH; |
| failed: |
| FX_Free(m_pJbig2Context->m_src_buf); |
| m_pJbig2Context->m_src_buf = NULL; |
| return FXCODEC_STATUS_ERROR; |
| } |
| FXCODEC_STATUS CCodec_Jbig2Module::ContinueDecode(void* pJbig2Context, |
| IFX_Pause* pPause) { |
| CCodec_Jbig2Context* m_pJbig2Context = (CCodec_Jbig2Context*)pJbig2Context; |
| int ret = m_pJbig2Context->m_pContext->Continue(pPause); |
| if (m_pJbig2Context->m_pContext->GetProcessiveStatus() != |
| FXCODEC_STATUS_DECODE_FINISH) { |
| return m_pJbig2Context->m_pContext->GetProcessiveStatus(); |
| } |
| if (m_pJbig2Context->m_bFileReader) { |
| CJBig2_Context::DestroyContext(m_pJbig2Context->m_pContext); |
| m_pJbig2Context->m_pContext = NULL; |
| if (ret != JBIG2_SUCCESS) { |
| FX_Free(m_pJbig2Context->m_src_buf); |
| m_pJbig2Context->m_src_buf = NULL; |
| return FXCODEC_STATUS_ERROR; |
| } |
| delete m_pJbig2Context->m_dest_image; |
| FX_Free(m_pJbig2Context->m_src_buf); |
| return FXCODEC_STATUS_DECODE_FINISH; |
| } |
| CJBig2_Context::DestroyContext(m_pJbig2Context->m_pContext); |
| m_pJbig2Context->m_pContext = NULL; |
| if (ret != JBIG2_SUCCESS) { |
| return FXCODEC_STATUS_ERROR; |
| } |
| int dword_size = |
| m_pJbig2Context->m_height * m_pJbig2Context->m_dest_pitch / 4; |
| FX_DWORD* dword_buf = (FX_DWORD*)m_pJbig2Context->m_dest_buf; |
| for (int i = 0; i < dword_size; i++) { |
| dword_buf[i] = ~dword_buf[i]; |
| } |
| return FXCODEC_STATUS_DECODE_FINISH; |
| } |
