blob: fc91a6c1621e1268b8759b9d59cd9304826ad926 [file] [log] [blame]
// 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 "../../../include/fxge/fx_dib.h"
#include "codec_int.h"
extern "C" {
static void _JpegScanSOI(const FX_BYTE*& src_buf, FX_DWORD& src_size)
{
if (src_size == 0) {
return;
}
FX_DWORD offset = 0;
while (offset < src_size - 1) {
if (src_buf[offset] == 0xff && src_buf[offset + 1] == 0xd8) {
src_buf += offset;
src_size -= offset;
return;
}
offset ++;
}
}
};
extern "C" {
#undef FAR
#include "../../fx_jpeglib.h"
}
extern "C" {
static void _src_do_nothing(struct jpeg_decompress_struct* cinfo) {}
};
extern "C" {
static void _error_fatal(j_common_ptr cinfo)
{
longjmp(*(jmp_buf*)cinfo->client_data, -1);
}
};
extern "C" {
static void _src_skip_data(struct jpeg_decompress_struct* cinfo, long num)
{
if (num > (long)cinfo->src->bytes_in_buffer) {
_error_fatal((j_common_ptr)cinfo);
}
cinfo->src->next_input_byte += num;
cinfo->src->bytes_in_buffer -= num;
}
};
extern "C" {
static boolean _src_fill_buffer(j_decompress_ptr cinfo)
{
return 0;
}
};
extern "C" {
static boolean _src_resync(j_decompress_ptr cinfo, int desired)
{
return 0;
}
};
extern "C" {
static void _error_do_nothing(j_common_ptr cinfo) {}
};
extern "C" {
static void _error_do_nothing1(j_common_ptr cinfo, int) {}
};
extern "C" {
static void _error_do_nothing2(j_common_ptr cinfo, char*) {}
};
#define JPEG_MARKER_EXIF (JPEG_APP0 + 1)
#define JPEG_MARKER_ICC (JPEG_APP0 + 2)
#define JPEG_MARKER_AUTHORTIME (JPEG_APP0 + 3)
#define JPEG_MARKER_MAXSIZE 0xFFFF
#define JPEG_OVERHEAD_LEN 14
static FX_BOOL _JpegIsIccMarker(jpeg_saved_marker_ptr marker)
{
if (marker->marker == JPEG_MARKER_ICC &&
marker->data_length >= JPEG_OVERHEAD_LEN &&
(FXSYS_memcmp32(marker->data, "\x49\x43\x43\x5f\x50\x52\x4f\x46\x49\x4c\x45\x00", 12) == 0)) {
return TRUE;
}
return FALSE;
}
static FX_BOOL _JpegLoadIccProfile(j_decompress_ptr cinfo, FX_LPBYTE* icc_buf_ptr, FX_DWORD* icc_length)
{
if(icc_buf_ptr == NULL || icc_length == NULL) {
return FALSE;
}
*icc_buf_ptr = NULL;
*icc_length = 0;
FX_LPBYTE icc_data_ptr = NULL;
FX_DWORD icc_data_len = 0;
FX_BYTE count_icc_marker = 0;
FX_BYTE num_icc_marker = 0;
jpeg_saved_marker_ptr marker_list[256] = {NULL};
for (jpeg_saved_marker_ptr cur_marker = cinfo->marker_list;
cur_marker != NULL;
cur_marker = cur_marker->next) {
if(_JpegIsIccMarker(cur_marker)) {
if(count_icc_marker == 0) {
num_icc_marker = cur_marker->data[13];
} else if(num_icc_marker != cur_marker->data[13]) {
return FALSE;
}
int sn = cur_marker->data[12] - 1;
if(sn < 0 || sn >= num_icc_marker) {
return FALSE;
}
if(marker_list[sn] == NULL) {
marker_list[sn] = cur_marker;
} else {
return FALSE;
}
count_icc_marker ++;
icc_data_len += (cur_marker->data_length - JPEG_OVERHEAD_LEN);
}
}
if(count_icc_marker != num_icc_marker) {
return FALSE;
}
if(num_icc_marker == 0) {
return TRUE;
}
icc_data_ptr = FX_Alloc(FX_BYTE, icc_data_len);
if(icc_buf_ptr == NULL) {
return FALSE;
}
*icc_buf_ptr = icc_data_ptr;
*icc_length = icc_data_len;
for (int idx = 0; idx < num_icc_marker; idx++) {
icc_data_len = marker_list[idx]->data_length - JPEG_OVERHEAD_LEN;
FXSYS_memcpy32(icc_data_ptr, marker_list[idx]->data + JPEG_OVERHEAD_LEN, icc_data_len);
icc_data_ptr += icc_data_len;
}
return TRUE;
}
static FX_BOOL _JpegEmbedIccProfile(j_compress_ptr cinfo, FX_LPCBYTE icc_buf_ptr, FX_DWORD icc_length)
{
if(icc_buf_ptr == NULL || icc_length == 0) {
return FALSE;
}
FX_DWORD icc_segment_size = (JPEG_MARKER_MAXSIZE - 2 - JPEG_OVERHEAD_LEN);
FX_DWORD icc_segment_num = (icc_length / icc_segment_size) + 1;
if (icc_segment_num > 255) {
return FALSE;
}
FX_DWORD icc_data_length = JPEG_OVERHEAD_LEN + (icc_segment_num > 1 ? icc_segment_size : icc_length);
FX_LPBYTE icc_data = FX_Alloc(FX_BYTE, icc_data_length);
if (icc_data == NULL) {
return FALSE;
}
FXSYS_memcpy32(icc_data, "\x49\x43\x43\x5f\x50\x52\x4f\x46\x49\x4c\x45\x00", 12);
icc_data[13] = (FX_BYTE)icc_segment_num;
for (FX_BYTE i = 0; i < (icc_segment_num - 1); i++) {
icc_data[12] = i + 1;
FXSYS_memcpy32(icc_data + JPEG_OVERHEAD_LEN, icc_buf_ptr + i * icc_segment_size, icc_segment_size);
jpeg_write_marker(cinfo, JPEG_MARKER_ICC, icc_data, icc_data_length);
}
icc_data[12] = (FX_BYTE)icc_segment_num;
FX_DWORD icc_size = (icc_segment_num - 1) * icc_segment_size;
FXSYS_memcpy32(icc_data + JPEG_OVERHEAD_LEN, icc_buf_ptr + icc_size, icc_length - icc_size);
jpeg_write_marker(cinfo, JPEG_MARKER_ICC, icc_data, JPEG_OVERHEAD_LEN + icc_length - icc_size);
FX_Free(icc_data);
return TRUE;
}
extern "C" {
static void _dest_do_nothing(j_compress_ptr cinfo) {}
};
extern "C" {
static boolean _dest_empty(j_compress_ptr cinfo)
{
return FALSE;
}
};
#define JPEG_BLOCK_SIZE 1048576
static void _JpegEncode(const CFX_DIBSource* pSource, FX_LPBYTE& dest_buf, FX_STRSIZE& dest_size, int quality, FX_LPCBYTE icc_buf, FX_DWORD icc_length)
{
struct jpeg_compress_struct cinfo;
struct jpeg_error_mgr jerr;
jerr.error_exit = _error_do_nothing;
jerr.emit_message = _error_do_nothing1;
jerr.output_message = _error_do_nothing;
jerr.format_message = _error_do_nothing2;
jerr.reset_error_mgr = _error_do_nothing;
cinfo.err = &jerr;
jpeg_create_compress(&cinfo);
int Bpp = pSource->GetBPP() / 8;
int nComponents = Bpp >= 3 ? (pSource->IsCmykImage() ? 4 : 3) : 1;
int pitch = pSource->GetPitch();
int width = pSource->GetWidth();
int height = pSource->GetHeight();
FX_DWORD dest_buf_length = width * height * nComponents + 1024 + (icc_length ? (icc_length + 255 * 18) : 0);
dest_buf = FX_Alloc(FX_BYTE, dest_buf_length);
while (dest_buf == NULL) {
dest_buf_length >>= 1;
dest_buf = FX_Alloc(FX_BYTE, dest_buf_length);
}
FXSYS_memset32(dest_buf, 0, dest_buf_length);
struct jpeg_destination_mgr dest;
dest.init_destination = _dest_do_nothing;
dest.term_destination = _dest_do_nothing;
dest.empty_output_buffer = _dest_empty;
dest.next_output_byte = dest_buf;
dest.free_in_buffer = dest_buf_length;
cinfo.dest = &dest;
cinfo.image_width = width;
cinfo.image_height = height;
cinfo.input_components = nComponents;
if (nComponents == 1) {
cinfo.in_color_space = JCS_GRAYSCALE;
} else if (nComponents == 3) {
cinfo.in_color_space = JCS_RGB;
} else {
cinfo.in_color_space = JCS_CMYK;
}
FX_LPBYTE line_buf = NULL;
if (nComponents > 1) {
line_buf = FX_Alloc(FX_BYTE, width * nComponents);
if (line_buf == NULL) {
return;
}
}
jpeg_set_defaults(&cinfo);
if(quality != 75) {
jpeg_set_quality(&cinfo, quality, TRUE);
}
jpeg_start_compress(&cinfo, TRUE);
_JpegEmbedIccProfile(&cinfo, icc_buf, icc_length);
JSAMPROW row_pointer[1];
JDIMENSION row;
while (cinfo.next_scanline < cinfo.image_height) {
FX_LPCBYTE src_scan = pSource->GetScanline(cinfo.next_scanline);
if (nComponents > 1) {
FX_LPBYTE dest_scan = line_buf;
if (nComponents == 3) {
for (int i = 0; i < width; i ++) {
dest_scan[0] = src_scan[2];
dest_scan[1] = src_scan[1];
dest_scan[2] = src_scan[0];
dest_scan += 3;
src_scan += Bpp;
}
} else {
for (int i = 0; i < pitch; i ++) {
*dest_scan++ = ~*src_scan++;
}
}
row_pointer[0] = line_buf;
} else {
row_pointer[0] = (FX_LPBYTE)src_scan;
}
row = cinfo.next_scanline;
jpeg_write_scanlines(&cinfo, row_pointer, 1);
if (cinfo.next_scanline == row) {
dest_buf = FX_Realloc(FX_BYTE, dest_buf, dest_buf_length + JPEG_BLOCK_SIZE);
if (dest_buf == NULL) {
FX_Free(line_buf);
return;
}
dest.next_output_byte = dest_buf + dest_buf_length - dest.free_in_buffer;
dest_buf_length += JPEG_BLOCK_SIZE;
dest.free_in_buffer += JPEG_BLOCK_SIZE;
}
}
jpeg_finish_compress(&cinfo);
jpeg_destroy_compress(&cinfo);
if (line_buf) {
FX_Free(line_buf);
}
dest_size = dest_buf_length - (FX_STRSIZE)dest.free_in_buffer;
}
static FX_BOOL _JpegLoadInfo(FX_LPCBYTE src_buf, FX_DWORD src_size, int& width, int& height,
int& num_components, int& bits_per_components, FX_BOOL& color_transform,
FX_LPBYTE* icc_buf_ptr, FX_DWORD* icc_length)
{
_JpegScanSOI(src_buf, src_size);
struct jpeg_decompress_struct cinfo;
struct jpeg_error_mgr jerr;
jerr.error_exit = _error_fatal;
jerr.emit_message = _error_do_nothing1;
jerr.output_message = _error_do_nothing;
jerr.format_message = _error_do_nothing2;
jerr.reset_error_mgr = _error_do_nothing;
cinfo.err = &jerr;
jmp_buf mark;
cinfo.client_data = &mark;
if (setjmp(mark) == -1) {
return FALSE;
}
jpeg_create_decompress(&cinfo);
struct jpeg_source_mgr src;
src.init_source = _src_do_nothing;
src.term_source = _src_do_nothing;
src.skip_input_data = _src_skip_data;
src.fill_input_buffer = _src_fill_buffer;
src.resync_to_restart = _src_resync;
src.bytes_in_buffer = src_size;
src.next_input_byte = src_buf;
cinfo.src = &src;
if (setjmp(mark) == -1) {
jpeg_destroy_decompress(&cinfo);
return FALSE;
}
if(icc_buf_ptr && icc_length) {
jpeg_save_markers(&cinfo, JPEG_MARKER_ICC, JPEG_MARKER_MAXSIZE);
}
int ret = jpeg_read_header(&cinfo, TRUE);
if (ret != JPEG_HEADER_OK) {
jpeg_destroy_decompress(&cinfo);
return FALSE;
}
width = cinfo.image_width;
height = cinfo.image_height;
num_components = cinfo.num_components;
color_transform = cinfo.jpeg_color_space == JCS_YCbCr || cinfo.jpeg_color_space == JCS_YCCK;
bits_per_components = cinfo.data_precision;
if(icc_buf_ptr != NULL) {
*icc_buf_ptr = NULL;
}
if(icc_length != NULL) {
*icc_length = 0;
}
jpeg_destroy_decompress(&cinfo);
return TRUE;
}
class CCodec_JpegDecoder : public CCodec_ScanlineDecoder
{
public:
CCodec_JpegDecoder();
~CCodec_JpegDecoder();
FX_BOOL Create(FX_LPCBYTE src_buf, FX_DWORD src_size, int width, int height, int nComps,
FX_BOOL ColorTransform, IFX_JpegProvider* pJP);
virtual void Destroy()
{
delete this;
}
virtual void v_DownScale(int dest_width, int dest_height);
virtual FX_BOOL v_Rewind();
virtual FX_LPBYTE v_GetNextLine();
virtual FX_DWORD GetSrcOffset();
jmp_buf m_JmpBuf;
struct jpeg_decompress_struct cinfo;
struct jpeg_error_mgr jerr;
struct jpeg_source_mgr src;
FX_LPCBYTE m_SrcBuf;
FX_DWORD m_SrcSize;
FX_LPBYTE m_pScanlineBuf;
FX_BOOL InitDecode();
FX_BOOL m_bInited, m_bStarted, m_bJpegTransform;
protected:
IFX_JpegProvider* m_pExtProvider;
void* m_pExtContext;
FX_DWORD m_nDefaultScaleDenom;
};
CCodec_JpegDecoder::CCodec_JpegDecoder()
{
m_pScanlineBuf = NULL;
m_DownScale = 1;
m_bStarted = FALSE;
m_bInited = FALSE;
m_pExtProvider = NULL;
m_pExtContext = NULL;
FXSYS_memset32(&cinfo, 0, sizeof(cinfo));
FXSYS_memset32(&jerr, 0, sizeof(jerr));
FXSYS_memset32(&src, 0, sizeof(src));
m_nDefaultScaleDenom = 1;
}
CCodec_JpegDecoder::~CCodec_JpegDecoder()
{
if (m_pExtProvider) {
m_pExtProvider->DestroyDecoder(m_pExtContext);
return;
}
if (m_pScanlineBuf) {
FX_Free(m_pScanlineBuf);
}
if (m_bInited) {
jpeg_destroy_decompress(&cinfo);
}
}
FX_BOOL CCodec_JpegDecoder::InitDecode()
{
cinfo.err = &jerr;
cinfo.client_data = &m_JmpBuf;
if (setjmp(m_JmpBuf) == -1) {
return FALSE;
}
jpeg_create_decompress(&cinfo);
m_bInited = TRUE;
cinfo.src = &src;
src.bytes_in_buffer = m_SrcSize;
src.next_input_byte = m_SrcBuf;
if (setjmp(m_JmpBuf) == -1) {
jpeg_destroy_decompress(&cinfo);
m_bInited = FALSE;
return FALSE;
}
cinfo.image_width = m_OrigWidth;
cinfo.image_height = m_OrigHeight;
int ret = jpeg_read_header(&cinfo, TRUE);
if (ret != JPEG_HEADER_OK) {
return FALSE;
}
if (cinfo.saw_Adobe_marker) {
m_bJpegTransform = TRUE;
}
if (cinfo.num_components == 3 && !m_bJpegTransform) {
cinfo.out_color_space = cinfo.jpeg_color_space;
}
m_OrigWidth = cinfo.image_width;
m_OrigHeight = cinfo.image_height;
m_OutputWidth = m_OrigWidth;
m_OutputHeight = m_OutputHeight;
m_nDefaultScaleDenom = cinfo.scale_denom;
return TRUE;
}
FX_BOOL CCodec_JpegDecoder::Create(FX_LPCBYTE src_buf, FX_DWORD src_size, int width, int height,
int nComps, FX_BOOL ColorTransform, IFX_JpegProvider* pJP)
{
if (pJP) {
m_pExtProvider = pJP;
m_pExtContext = m_pExtProvider->CreateDecoder(src_buf, src_size, width, height, nComps, ColorTransform);
return m_pExtContext != NULL;
}
_JpegScanSOI(src_buf, src_size);
m_SrcBuf = src_buf;
m_SrcSize = src_size;
jerr.error_exit = _error_fatal;
jerr.emit_message = _error_do_nothing1;
jerr.output_message = _error_do_nothing;
jerr.format_message = _error_do_nothing2;
jerr.reset_error_mgr = _error_do_nothing;
src.init_source = _src_do_nothing;
src.term_source = _src_do_nothing;
src.skip_input_data = _src_skip_data;
src.fill_input_buffer = _src_fill_buffer;
src.resync_to_restart = _src_resync;
m_bJpegTransform = ColorTransform;
if(src_size > 1 && FXSYS_memcmp32(src_buf + src_size - 2, "\xFF\xD9", 2) != 0) {
((FX_LPBYTE)src_buf)[src_size - 2] = 0xFF;
((FX_LPBYTE)src_buf)[src_size - 1] = 0xD9;
}
m_OutputWidth = m_OrigWidth = width;
m_OutputHeight = m_OrigHeight = height;
if (!InitDecode()) {
return FALSE;
}
if (cinfo.num_components < nComps) {
return FALSE;
}
if ((int)cinfo.image_width < width) {
return FALSE;
}
m_Pitch = (cinfo.image_width * cinfo.num_components + 3) / 4 * 4;
m_pScanlineBuf = FX_Alloc(FX_BYTE, m_Pitch);
if (m_pScanlineBuf == NULL) {
return FALSE;
}
m_nComps = cinfo.num_components;
m_bpc = 8;
m_bColorTransformed = FALSE;
m_bStarted = FALSE;
return TRUE;
}
extern "C" {
FX_INT32 FX_GetDownsampleRatio(FX_INT32 originWidth, FX_INT32 originHeight, FX_INT32 downsampleWidth, FX_INT32 downsampleHeight)
{
int iratio_w = originWidth / downsampleWidth;
int iratio_h = originHeight / downsampleHeight;
int ratio = (iratio_w > iratio_h) ? iratio_h : iratio_w;
if (ratio >= 8) {
return 8;
} else if (ratio >= 4) {
return 4;
} else if (ratio >= 2) {
return 2;
}
return 1;
}
}
void CCodec_JpegDecoder::v_DownScale(int dest_width, int dest_height)
{
if (m_pExtProvider) {
m_pExtProvider->DownScale(m_pExtContext, dest_width, dest_height);
return;
}
int old_scale = m_DownScale;
m_DownScale = FX_GetDownsampleRatio(m_OrigWidth, m_OrigHeight, dest_width, dest_height);
m_OutputWidth = (m_OrigWidth + m_DownScale - 1) / m_DownScale;
m_OutputHeight = (m_OrigHeight + m_DownScale - 1) / m_DownScale;
m_Pitch = (m_OutputWidth * m_nComps + 3) / 4 * 4;
if (old_scale != m_DownScale) {
m_NextLine = -1;
}
}
FX_BOOL CCodec_JpegDecoder::v_Rewind()
{
if (m_pExtProvider) {
return m_pExtProvider->Rewind(m_pExtContext);
}
if (m_bStarted) {
jpeg_destroy_decompress(&cinfo);
if (!InitDecode()) {
return FALSE;
}
}
if (setjmp(m_JmpBuf) == -1) {
return FALSE;
}
cinfo.scale_denom = m_nDefaultScaleDenom * m_DownScale;
m_OutputWidth = (m_OrigWidth + m_DownScale - 1) / m_DownScale;
m_OutputHeight = (m_OrigHeight + m_DownScale - 1) / m_DownScale;
if (!jpeg_start_decompress(&cinfo)) {
jpeg_destroy_decompress(&cinfo);
return FALSE;
}
if ((int)cinfo.output_width > m_OrigWidth) {
FXSYS_assert(FALSE);
return FALSE;
}
m_bStarted = TRUE;
return TRUE;
}
FX_LPBYTE CCodec_JpegDecoder::v_GetNextLine()
{
if (m_pExtProvider) {
return m_pExtProvider->GetNextLine(m_pExtContext);
}
int nlines = jpeg_read_scanlines(&cinfo, &m_pScanlineBuf, 1);
if (nlines < 1) {
return NULL;
}
return m_pScanlineBuf;
}
FX_DWORD CCodec_JpegDecoder::GetSrcOffset()
{
if (m_pExtProvider) {
return m_pExtProvider->GetSrcOffset(m_pExtContext);
}
return (FX_DWORD)(m_SrcSize - src.bytes_in_buffer);
}
ICodec_ScanlineDecoder* CCodec_JpegModule::CreateDecoder(FX_LPCBYTE src_buf, FX_DWORD src_size,
int width, int height, int nComps, FX_BOOL ColorTransform)
{
if (src_buf == NULL || src_size == 0) {
return NULL;
}
CCodec_JpegDecoder* pDecoder = FX_NEW CCodec_JpegDecoder;
if (pDecoder == NULL) {
return NULL;
}
if (!pDecoder->Create(src_buf, src_size, width, height, nComps, ColorTransform, m_pExtProvider)) {
delete pDecoder;
return NULL;
}
return pDecoder;
}
FX_BOOL CCodec_JpegModule::LoadInfo(FX_LPCBYTE src_buf, FX_DWORD src_size, int& width, int& height,
int& num_components, int& bits_per_components, FX_BOOL& color_transform,
FX_LPBYTE* icc_buf_ptr, FX_DWORD* icc_length)
{
if (m_pExtProvider) {
return m_pExtProvider->LoadInfo(src_buf, src_size, width, height,
num_components, bits_per_components, color_transform,
icc_buf_ptr, icc_length);
}
return _JpegLoadInfo(src_buf, src_size, width, height, num_components, bits_per_components, color_transform, icc_buf_ptr, icc_length);
}
FX_BOOL CCodec_JpegModule::Encode(const CFX_DIBSource* pSource, FX_LPBYTE& dest_buf, FX_STRSIZE& dest_size, int quality, FX_LPCBYTE icc_buf, FX_DWORD icc_length)
{
if (m_pExtProvider) {
return m_pExtProvider->Encode(pSource, dest_buf, dest_size, quality, icc_buf, icc_length);
}
if(pSource->GetBPP() < 8 || pSource->GetPalette() != NULL) {
ASSERT(pSource->GetBPP() >= 8 && pSource->GetPalette() == NULL);
return FALSE;
}
_JpegEncode(pSource, dest_buf, dest_size, quality, icc_buf, icc_length);
return TRUE;
}
struct FXJPEG_Context {
jmp_buf m_JumpMark;
jpeg_decompress_struct m_Info;
jpeg_error_mgr m_ErrMgr;
jpeg_source_mgr m_SrcMgr;
unsigned int m_SkipSize;
void* (*m_AllocFunc)(unsigned int);
void (*m_FreeFunc)(void*);
};
extern "C" {
static void _error_fatal1(j_common_ptr cinfo)
{
longjmp(((FXJPEG_Context*)cinfo->client_data)->m_JumpMark, -1);
}
};
extern "C" {
static void _src_skip_data1(struct jpeg_decompress_struct* cinfo, long num)
{
if (cinfo->src->bytes_in_buffer < (size_t)num) {
((FXJPEG_Context*)cinfo->client_data)->m_SkipSize = (unsigned int)(num - cinfo->src->bytes_in_buffer);
cinfo->src->bytes_in_buffer = 0;
} else {
cinfo->src->next_input_byte += num;
cinfo->src->bytes_in_buffer -= num;
}
}
};
static void* jpeg_alloc_func(unsigned int size)
{
return FX_Alloc(char, size);
}
static void jpeg_free_func(void* p)
{
FX_Free(p);
}
void* CCodec_JpegModule::Start()
{
if (m_pExtProvider) {
return m_pExtProvider->Start();
}
FXJPEG_Context* p = (FXJPEG_Context*)FX_Alloc(FX_BYTE, sizeof(FXJPEG_Context));
if (p == NULL) {
return NULL;
}
p->m_AllocFunc = jpeg_alloc_func;
p->m_FreeFunc = jpeg_free_func;
p->m_ErrMgr.error_exit = _error_fatal1;
p->m_ErrMgr.emit_message = _error_do_nothing1;
p->m_ErrMgr.output_message = _error_do_nothing;
p->m_ErrMgr.format_message = _error_do_nothing2;
p->m_ErrMgr.reset_error_mgr = _error_do_nothing;
p->m_SrcMgr.init_source = _src_do_nothing;
p->m_SrcMgr.term_source = _src_do_nothing;
p->m_SrcMgr.skip_input_data = _src_skip_data1;
p->m_SrcMgr.fill_input_buffer = _src_fill_buffer;
p->m_SrcMgr.resync_to_restart = _src_resync;
p->m_Info.client_data = p;
p->m_Info.err = &p->m_ErrMgr;
if (setjmp(p->m_JumpMark) == -1) {
return 0;
}
jpeg_create_decompress(&p->m_Info);
p->m_Info.src = &p->m_SrcMgr;
p->m_SkipSize = 0;
return p;
}
void CCodec_JpegModule::Finish(void* pContext)
{
if (m_pExtProvider) {
m_pExtProvider->Finish(pContext);
return;
}
FXJPEG_Context* p = (FXJPEG_Context*)pContext;
jpeg_destroy_decompress(&p->m_Info);
p->m_FreeFunc(p);
}
void CCodec_JpegModule::Input(void* pContext, const unsigned char* src_buf, FX_DWORD src_size)
{
if (m_pExtProvider) {
m_pExtProvider->Input(pContext, src_buf, src_size);
return;
}
FXJPEG_Context* p = (FXJPEG_Context*)pContext;
if (p->m_SkipSize) {
if (p->m_SkipSize > src_size) {
p->m_SrcMgr.bytes_in_buffer = 0;
p->m_SkipSize -= src_size;
return;
}
src_size -= p->m_SkipSize;
src_buf += p->m_SkipSize;
p->m_SkipSize = 0;
}
p->m_SrcMgr.next_input_byte = src_buf;
p->m_SrcMgr.bytes_in_buffer = src_size;
}
int CCodec_JpegModule::ReadHeader(void* pContext, int* width, int* height, int* nComps)
{
if (m_pExtProvider) {
return m_pExtProvider->ReadHeader(pContext, width, height, nComps);
}
FXJPEG_Context* p = (FXJPEG_Context*)pContext;
if (setjmp(p->m_JumpMark) == -1) {
return 1;
}
int ret = jpeg_read_header(&p->m_Info, true);
if (ret == JPEG_SUSPENDED) {
return 2;
}
if (ret != JPEG_HEADER_OK) {
return 1;
}
*width = p->m_Info.image_width;
*height = p->m_Info.image_height;
*nComps = p->m_Info.num_components;
return 0;
}
FX_BOOL CCodec_JpegModule::StartScanline(void* pContext, int down_scale)
{
if (m_pExtProvider) {
return m_pExtProvider->StartScanline(pContext, down_scale);
}
FXJPEG_Context* p = (FXJPEG_Context*)pContext;
if (setjmp(p->m_JumpMark) == -1) {
return FALSE;
}
p->m_Info.scale_denom = down_scale;
return jpeg_start_decompress(&p->m_Info);
}
FX_BOOL CCodec_JpegModule::ReadScanline(void* pContext, unsigned char* dest_buf)
{
if (m_pExtProvider) {
return m_pExtProvider->ReadScanline(pContext, dest_buf);
}
FXJPEG_Context* p = (FXJPEG_Context*)pContext;
if (setjmp(p->m_JumpMark) == -1) {
return FALSE;
}
int nlines = jpeg_read_scanlines(&p->m_Info, &dest_buf, 1);
return nlines == 1;
}
FX_DWORD CCodec_JpegModule::GetAvailInput(void* pContext, FX_LPBYTE* avail_buf_ptr)
{
if (m_pExtProvider) {
return m_pExtProvider->GetAvailInput(pContext, avail_buf_ptr);
}
if(avail_buf_ptr != NULL) {
*avail_buf_ptr = NULL;
if(((FXJPEG_Context*)pContext)->m_SrcMgr.bytes_in_buffer > 0) {
*avail_buf_ptr = (FX_LPBYTE)((FXJPEG_Context*)pContext)->m_SrcMgr.next_input_byte;
}
}
return (FX_DWORD)((FXJPEG_Context*)pContext)->m_SrcMgr.bytes_in_buffer;
}