| Index: core/fxcodec/codec/fx_codec_jpeg.cpp
|
| diff --git a/core/fxcodec/codec/fx_codec_jpeg.cpp b/core/fxcodec/codec/fx_codec_jpeg.cpp
|
| index c41dde943f040e9ad866dfa4950e70b9845c00f3..cbfea59329075ffd4e45e0217909e7cca0def771 100644
|
| --- a/core/fxcodec/codec/fx_codec_jpeg.cpp
|
| +++ b/core/fxcodec/codec/fx_codec_jpeg.cpp
|
| @@ -23,15 +23,15 @@ extern "C" {
|
| }
|
|
|
| extern "C" {
|
| -static void _JpegScanSOI(const uint8_t*& src_buf, uint32_t& src_size) {
|
| - if (src_size == 0) {
|
| +static void JpegScanSOI(const uint8_t** src_buf, uint32_t* src_size) {
|
| + if (*src_size == 0)
|
| return;
|
| - }
|
| +
|
| uint32_t offset = 0;
|
| - while (offset < src_size - 1) {
|
| - if (src_buf[offset] == 0xff && src_buf[offset + 1] == 0xd8) {
|
| - src_buf += offset;
|
| - src_size -= offset;
|
| + while (offset < *src_size - 1) {
|
| + if ((*src_buf)[offset] == 0xff && (*src_buf)[offset + 1] == 0xd8) {
|
| + *src_buf += offset;
|
| + *src_size -= offset;
|
| return;
|
| }
|
| offset++;
|
| @@ -74,187 +74,31 @@ 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 _JpegEmbedIccProfile(j_compress_ptr cinfo,
|
| - const uint8_t* icc_buf_ptr,
|
| - uint32_t icc_length) {
|
| - if (!icc_buf_ptr || icc_length == 0) {
|
| - return FALSE;
|
| - }
|
| - uint32_t icc_segment_size = (JPEG_MARKER_MAXSIZE - 2 - JPEG_OVERHEAD_LEN);
|
| - uint32_t icc_segment_num = (icc_length / icc_segment_size) + 1;
|
| - if (icc_segment_num > 255) {
|
| - return FALSE;
|
| - }
|
| - uint32_t icc_data_length =
|
| - JPEG_OVERHEAD_LEN + (icc_segment_num > 1 ? icc_segment_size : icc_length);
|
| - uint8_t* icc_data = FX_Alloc(uint8_t, icc_data_length);
|
| - FXSYS_memcpy(icc_data, "\x49\x43\x43\x5f\x50\x52\x4f\x46\x49\x4c\x45\x00",
|
| - 12);
|
| - icc_data[13] = (uint8_t)icc_segment_num;
|
| - for (uint8_t i = 0; i < (icc_segment_num - 1); i++) {
|
| - icc_data[12] = i + 1;
|
| - FXSYS_memcpy(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] = (uint8_t)icc_segment_num;
|
| - uint32_t icc_size = (icc_segment_num - 1) * icc_segment_size;
|
| - FXSYS_memcpy(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,
|
| - uint8_t*& dest_buf,
|
| - FX_STRSIZE& dest_size,
|
| - int quality,
|
| - const uint8_t* icc_buf,
|
| - uint32_t icc_length) {
|
| - 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;
|
|
|
| - struct jpeg_compress_struct cinfo;
|
| - memset(&cinfo, 0, sizeof(cinfo));
|
| - cinfo.err = &jerr;
|
| - jpeg_create_compress(&cinfo);
|
| - int Bpp = pSource->GetBPP() / 8;
|
| - uint32_t nComponents = Bpp >= 3 ? (pSource->IsCmykImage() ? 4 : 3) : 1;
|
| - uint32_t pitch = pSource->GetPitch();
|
| - uint32_t width = pdfium::base::checked_cast<uint32_t>(pSource->GetWidth());
|
| - uint32_t height = pdfium::base::checked_cast<uint32_t>(pSource->GetHeight());
|
| - FX_SAFE_UINT32 safe_buf_len = width;
|
| - safe_buf_len *= height;
|
| - safe_buf_len *= nComponents;
|
| - safe_buf_len += 1024;
|
| - if (icc_length) {
|
| - safe_buf_len += 255 * 18;
|
| - safe_buf_len += icc_length;
|
| - }
|
| - uint32_t dest_buf_length = 0;
|
| - if (!safe_buf_len.IsValid()) {
|
| - dest_buf = nullptr;
|
| - } else {
|
| - dest_buf_length = safe_buf_len.ValueOrDie();
|
| - dest_buf = FX_TryAlloc(uint8_t, dest_buf_length);
|
| - const int MIN_TRY_BUF_LEN = 1024;
|
| - while (!dest_buf && dest_buf_length > MIN_TRY_BUF_LEN) {
|
| - dest_buf_length >>= 1;
|
| - dest_buf = FX_TryAlloc(uint8_t, dest_buf_length);
|
| - }
|
| - }
|
| - if (!dest_buf) {
|
| - FX_OutOfMemoryTerminate();
|
| - }
|
| - 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;
|
| - }
|
| - uint8_t* line_buf = NULL;
|
| - if (nComponents > 1) {
|
| - line_buf = FX_Alloc2D(uint8_t, width, nComponents);
|
| - }
|
| - 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) {
|
| - const uint8_t* src_scan = pSource->GetScanline(cinfo.next_scanline);
|
| - if (nComponents > 1) {
|
| - uint8_t* dest_scan = line_buf;
|
| - if (nComponents == 3) {
|
| - for (uint32_t 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 (uint32_t i = 0; i < pitch; i++) {
|
| - *dest_scan++ = ~*src_scan++;
|
| - }
|
| - }
|
| - row_pointer[0] = line_buf;
|
| - } else {
|
| - row_pointer[0] = (uint8_t*)src_scan;
|
| - }
|
| - row = cinfo.next_scanline;
|
| - jpeg_write_scanlines(&cinfo, row_pointer, 1);
|
| - if (cinfo.next_scanline == row) {
|
| - dest_buf =
|
| - FX_Realloc(uint8_t, dest_buf, dest_buf_length + JPEG_BLOCK_SIZE);
|
| - 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);
|
| - FX_Free(line_buf);
|
| - dest_size = dest_buf_length - (FX_STRSIZE)dest.free_in_buffer;
|
| -}
|
|
|
| #ifdef PDF_ENABLE_XFA
|
| -static void _JpegLoadAttribute(struct jpeg_decompress_struct* pInfo,
|
| - CFX_DIBAttribute* pAttribute) {
|
| - if (pInfo == NULL || pAttribute == NULL) {
|
| +static void JpegLoadAttribute(struct jpeg_decompress_struct* pInfo,
|
| + CFX_DIBAttribute* pAttribute) {
|
| + if (!pInfo || !pAttribute)
|
| return;
|
| - }
|
| - if (pAttribute) {
|
| - pAttribute->m_nXDPI = pInfo->X_density;
|
| - pAttribute->m_nYDPI = pInfo->Y_density;
|
| - pAttribute->m_wDPIUnit = pInfo->density_unit;
|
| - }
|
| +
|
| + pAttribute->m_nXDPI = pInfo->X_density;
|
| + pAttribute->m_nYDPI = pInfo->Y_density;
|
| + pAttribute->m_wDPIUnit = pInfo->density_unit;
|
| }
|
| #endif // PDF_ENABLE_XFA
|
|
|
| -static FX_BOOL _JpegLoadInfo(const uint8_t* src_buf,
|
| - uint32_t src_size,
|
| - int& width,
|
| - int& height,
|
| - int& num_components,
|
| - int& bits_per_components,
|
| - FX_BOOL& color_transform,
|
| - uint8_t** icc_buf_ptr,
|
| - uint32_t* icc_length) {
|
| - _JpegScanSOI(src_buf, src_size);
|
| +static bool JpegLoadInfo(const uint8_t* src_buf,
|
| + uint32_t src_size,
|
| + int* width,
|
| + int* height,
|
| + int* num_components,
|
| + int* bits_per_components,
|
| + bool* color_transform) {
|
| + JpegScanSOI(&src_buf, &src_size);
|
| struct jpeg_decompress_struct cinfo;
|
| struct jpeg_error_mgr jerr;
|
| jerr.error_exit = _error_fatal;
|
| @@ -266,9 +110,9 @@ static FX_BOOL _JpegLoadInfo(const uint8_t* src_buf,
|
| cinfo.err = &jerr;
|
| jmp_buf mark;
|
| cinfo.client_data = &mark;
|
| - if (setjmp(mark) == -1) {
|
| - return FALSE;
|
| - }
|
| + if (setjmp(mark) == -1)
|
| + return false;
|
| +
|
| jpeg_create_decompress(&cinfo);
|
| struct jpeg_source_mgr src;
|
| src.init_source = _src_do_nothing;
|
| @@ -281,30 +125,21 @@ static FX_BOOL _JpegLoadInfo(const uint8_t* 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);
|
| + return false;
|
| }
|
| int ret = jpeg_read_header(&cinfo, TRUE);
|
| if (ret != JPEG_HEADER_OK) {
|
| jpeg_destroy_decompress(&cinfo);
|
| - return FALSE;
|
| + return false;
|
| }
|
| - width = cinfo.image_width;
|
| - height = cinfo.image_height;
|
| - num_components = cinfo.num_components;
|
| - color_transform =
|
| + *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) {
|
| - *icc_buf_ptr = NULL;
|
| - }
|
| - if (icc_length) {
|
| - *icc_length = 0;
|
| - }
|
| + *bits_per_components = cinfo.data_precision;
|
| jpeg_destroy_decompress(&cinfo);
|
| - return TRUE;
|
| + return true;
|
| }
|
|
|
| class CCodec_JpegDecoder : public CCodec_ScanlineDecoder {
|
| @@ -351,18 +186,19 @@ CCodec_JpegDecoder::CCodec_JpegDecoder() {
|
| FXSYS_memset(&src, 0, sizeof(src));
|
| m_nDefaultScaleDenom = 1;
|
| }
|
| +
|
| CCodec_JpegDecoder::~CCodec_JpegDecoder() {
|
| FX_Free(m_pScanlineBuf);
|
| - if (m_bInited) {
|
| + 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) {
|
| + if (setjmp(m_JmpBuf) == -1)
|
| return FALSE;
|
| - }
|
| +
|
| jpeg_create_decompress(&cinfo);
|
| m_bInited = TRUE;
|
| cinfo.src = &src;
|
| @@ -376,15 +212,15 @@ FX_BOOL CCodec_JpegDecoder::InitDecode() {
|
| cinfo.image_width = m_OrigWidth;
|
| cinfo.image_height = m_OrigHeight;
|
| int ret = jpeg_read_header(&cinfo, TRUE);
|
| - if (ret != JPEG_HEADER_OK) {
|
| + if (ret != JPEG_HEADER_OK)
|
| return FALSE;
|
| - }
|
| - if (cinfo.saw_Adobe_marker) {
|
| +
|
| + if (cinfo.saw_Adobe_marker)
|
| m_bJpegTransform = TRUE;
|
| - }
|
| - if (cinfo.num_components == 3 && !m_bJpegTransform) {
|
| +
|
| + 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;
|
| @@ -392,13 +228,14 @@ FX_BOOL CCodec_JpegDecoder::InitDecode() {
|
| m_nDefaultScaleDenom = cinfo.scale_denom;
|
| return TRUE;
|
| }
|
| +
|
| FX_BOOL CCodec_JpegDecoder::Create(const uint8_t* src_buf,
|
| uint32_t src_size,
|
| int width,
|
| int height,
|
| int nComps,
|
| FX_BOOL ColorTransform) {
|
| - _JpegScanSOI(src_buf, src_size);
|
| + JpegScanSOI(&src_buf, &src_size);
|
| m_SrcBuf = src_buf;
|
| m_SrcSize = src_size;
|
| jerr.error_exit = _error_fatal;
|
| @@ -419,15 +256,15 @@ FX_BOOL CCodec_JpegDecoder::Create(const uint8_t* src_buf,
|
| }
|
| m_OutputWidth = m_OrigWidth = width;
|
| m_OutputHeight = m_OrigHeight = height;
|
| - if (!InitDecode()) {
|
| + if (!InitDecode())
|
| return FALSE;
|
| - }
|
| - if (cinfo.num_components < nComps) {
|
| +
|
| + if (cinfo.num_components < nComps)
|
| return FALSE;
|
| - }
|
| - if ((int)cinfo.image_width < width) {
|
| +
|
| + if ((int)cinfo.image_width < width)
|
| return FALSE;
|
| - }
|
| +
|
| m_Pitch =
|
| (static_cast<uint32_t>(cinfo.image_width) * cinfo.num_components + 3) /
|
| 4 * 4;
|
| @@ -462,19 +299,19 @@ FX_BOOL CCodec_JpegDecoder::v_Rewind() {
|
| m_bStarted = TRUE;
|
| return TRUE;
|
| }
|
| +
|
| uint8_t* CCodec_JpegDecoder::v_GetNextLine() {
|
| if (setjmp(m_JmpBuf) == -1)
|
| return nullptr;
|
|
|
| int nlines = jpeg_read_scanlines(&cinfo, &m_pScanlineBuf, 1);
|
| - if (nlines < 1) {
|
| - return nullptr;
|
| - }
|
| - return m_pScanlineBuf;
|
| + return nlines > 0 ? m_pScanlineBuf : nullptr;
|
| }
|
| +
|
| uint32_t CCodec_JpegDecoder::GetSrcOffset() {
|
| return (uint32_t)(m_SrcSize - src.bytes_in_buffer);
|
| }
|
| +
|
| CCodec_ScanlineDecoder* CCodec_JpegModule::CreateDecoder(
|
| const uint8_t* src_buf,
|
| uint32_t src_size,
|
| @@ -482,42 +319,28 @@ CCodec_ScanlineDecoder* CCodec_JpegModule::CreateDecoder(
|
| int height,
|
| int nComps,
|
| FX_BOOL ColorTransform) {
|
| - if (!src_buf || src_size == 0) {
|
| - return NULL;
|
| - }
|
| - CCodec_JpegDecoder* pDecoder = new CCodec_JpegDecoder;
|
| + if (!src_buf || src_size == 0)
|
| + return nullptr;
|
| +
|
| + std::unique_ptr<CCodec_JpegDecoder> pDecoder(new CCodec_JpegDecoder);
|
| if (!pDecoder->Create(src_buf, src_size, width, height, nComps,
|
| ColorTransform)) {
|
| - delete pDecoder;
|
| - return NULL;
|
| + return nullptr;
|
| }
|
| - return pDecoder;
|
| -}
|
| -FX_BOOL CCodec_JpegModule::LoadInfo(const uint8_t* src_buf,
|
| - uint32_t src_size,
|
| - int& width,
|
| - int& height,
|
| - int& num_components,
|
| - int& bits_per_components,
|
| - FX_BOOL& color_transform,
|
| - uint8_t** icc_buf_ptr,
|
| - uint32_t* icc_length) {
|
| - return _JpegLoadInfo(src_buf, src_size, width, height, num_components,
|
| - bits_per_components, color_transform, icc_buf_ptr,
|
| - icc_length);
|
| + return pDecoder.release();
|
| }
|
| -FX_BOOL CCodec_JpegModule::Encode(const CFX_DIBSource* pSource,
|
| - uint8_t*& dest_buf,
|
| - FX_STRSIZE& dest_size,
|
| - int quality,
|
| - const uint8_t* icc_buf,
|
| - uint32_t icc_length) {
|
| - if (pSource->GetBPP() < 8 || pSource->GetPalette())
|
| - return FALSE;
|
|
|
| - _JpegEncode(pSource, dest_buf, dest_size, quality, icc_buf, icc_length);
|
| - return TRUE;
|
| +bool CCodec_JpegModule::LoadInfo(const uint8_t* src_buf,
|
| + uint32_t src_size,
|
| + int* width,
|
| + int* height,
|
| + int* num_components,
|
| + int* bits_per_components,
|
| + bool* color_transform) {
|
| + return JpegLoadInfo(src_buf, src_size, width, height, num_components,
|
| + bits_per_components, color_transform);
|
| }
|
| +
|
| struct FXJPEG_Context {
|
| jmp_buf m_JumpMark;
|
| jpeg_decompress_struct m_Info;
|
| @@ -622,7 +445,7 @@ int CCodec_JpegModule::ReadHeader(FXJPEG_Context* ctx,
|
| *height = ctx->m_Info.image_height;
|
| *nComps = ctx->m_Info.num_components;
|
| #ifdef PDF_ENABLE_XFA
|
| - _JpegLoadAttribute(&ctx->m_Info, pAttribute);
|
| + JpegLoadAttribute(&ctx->m_Info, pAttribute);
|
| #endif
|
| return 0;
|
| }
|
|
|