// 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_memset32(this, 0, sizeof(CCodec_Jbig2Context)); | |
} | |
CCodec_Jbig2Module::~CCodec_Jbig2Module() | |
{ | |
} | |
void* CCodec_Jbig2Module::CreateJbig2Context() | |
{ | |
return FX_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, FX_LPCBYTE src_buf, FX_DWORD src_size, | |
FX_LPCBYTE global_data, FX_DWORD global_size, FX_LPBYTE dest_buf, FX_DWORD dest_pitch) | |
{ | |
FXSYS_memset32(dest_buf, 0, height * dest_pitch); | |
CJBig2_Context* pContext = CJBig2_Context::CreateContext(&m_Module, | |
(FX_LPBYTE)global_data, global_size, (FX_LPBYTE)src_buf, src_size, JBIG2_EMBED_STREAM); | |
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, FX_LPBYTE& dest_buf) | |
{ | |
CJBig2_Context* pContext = NULL; | |
CJBig2_Image* dest_image = NULL; | |
FX_DWORD src_size = (FX_DWORD)file_ptr->GetSize(); | |
FX_LPBYTE src_buf = FX_Alloc(FX_BYTE, src_size); | |
if (src_buf == NULL) { | |
return FALSE; | |
} | |
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); | |
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: | |
if(src_buf) { | |
FX_Free(src_buf); | |
} | |
return FALSE; | |
} | |
FXCODEC_STATUS CCodec_Jbig2Module::StartDecode(void* pJbig2Context, FX_DWORD width, FX_DWORD height, FX_LPCBYTE src_buf, FX_DWORD src_size, | |
FX_LPCBYTE global_data, FX_DWORD global_size, FX_LPBYTE 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_memset32(dest_buf, 0, height * dest_pitch); | |
m_pJbig2Context->m_pContext = CJBig2_Context::CreateContext(&m_Module, | |
(FX_LPBYTE)global_data, global_size, (FX_LPBYTE)src_buf, src_size, JBIG2_EMBED_STREAM, 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, FX_LPBYTE& 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(FX_BYTE, m_pJbig2Context->m_src_size); | |
if (m_pJbig2Context->m_src_buf == NULL) { | |
return FXCODEC_STATUS_ERR_MEMORY; | |
} | |
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, 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: | |
if(m_pJbig2Context->m_src_buf) { | |
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) { | |
if(m_pJbig2Context->m_bFileReader) { | |
CJBig2_Context::DestroyContext(m_pJbig2Context->m_pContext); | |
m_pJbig2Context->m_pContext = NULL; | |
if (ret != JBIG2_SUCCESS) { | |
if(m_pJbig2Context->m_src_buf) { | |
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; | |
} else { | |
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; | |
} | |
} | |
return m_pJbig2Context->m_pContext->GetProcessiveStatus(); | |
} | |