// 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 <algorithm>
|
#include <limits>
|
#include <vector>
|
|
#include "core/fpdfapi/fpdf_page/include/cpdf_colorspace.h"
|
#include "core/fxcodec/codec/codec_int.h"
|
#include "core/fxcodec/include/fx_codec.h"
|
#include "core/fxcrt/include/fx_safe_types.h"
|
#include "third_party/lcms2-2.6/include/lcms2.h"
|
#include "third_party/libopenjpeg20/openjpeg.h"
|
|
static void fx_error_callback(const char* msg, void* client_data) {
|
(void)client_data;
|
}
|
static void fx_warning_callback(const char* msg, void* client_data) {
|
(void)client_data;
|
}
|
static void fx_info_callback(const char* msg, void* client_data) {
|
(void)client_data;
|
}
|
|
OPJ_SIZE_T opj_read_from_memory(void* p_buffer,
|
OPJ_SIZE_T nb_bytes,
|
void* p_user_data) {
|
DecodeData* srcData = static_cast<DecodeData*>(p_user_data);
|
if (!srcData || !srcData->src_data || srcData->src_size == 0) {
|
return static_cast<OPJ_SIZE_T>(-1);
|
}
|
// Reads at EOF return an error code.
|
if (srcData->offset >= srcData->src_size) {
|
return static_cast<OPJ_SIZE_T>(-1);
|
}
|
OPJ_SIZE_T bufferLength = srcData->src_size - srcData->offset;
|
OPJ_SIZE_T readlength = nb_bytes < bufferLength ? nb_bytes : bufferLength;
|
memcpy(p_buffer, &srcData->src_data[srcData->offset], readlength);
|
srcData->offset += readlength;
|
return readlength;
|
}
|
|
OPJ_SIZE_T opj_write_from_memory(void* p_buffer,
|
OPJ_SIZE_T nb_bytes,
|
void* p_user_data) {
|
DecodeData* srcData = static_cast<DecodeData*>(p_user_data);
|
if (!srcData || !srcData->src_data || srcData->src_size == 0) {
|
return static_cast<OPJ_SIZE_T>(-1);
|
}
|
// Writes at EOF return an error code.
|
if (srcData->offset >= srcData->src_size) {
|
return static_cast<OPJ_SIZE_T>(-1);
|
}
|
OPJ_SIZE_T bufferLength = srcData->src_size - srcData->offset;
|
OPJ_SIZE_T writeLength = nb_bytes < bufferLength ? nb_bytes : bufferLength;
|
memcpy(&srcData->src_data[srcData->offset], p_buffer, writeLength);
|
srcData->offset += writeLength;
|
return writeLength;
|
}
|
|
OPJ_OFF_T opj_skip_from_memory(OPJ_OFF_T nb_bytes, void* p_user_data) {
|
DecodeData* srcData = static_cast<DecodeData*>(p_user_data);
|
if (!srcData || !srcData->src_data || srcData->src_size == 0) {
|
return static_cast<OPJ_OFF_T>(-1);
|
}
|
// Offsets are signed and may indicate a negative skip. Do not support this
|
// because of the strange return convention where either bytes skipped or
|
// -1 is returned. Following that convention, a successful relative seek of
|
// -1 bytes would be required to to give the same result as the error case.
|
if (nb_bytes < 0) {
|
return static_cast<OPJ_OFF_T>(-1);
|
}
|
// FIXME: use std::make_unsigned<OPJ_OFF_T>::type once c++11 lib is OK'd.
|
uint64_t unsignedNbBytes = static_cast<uint64_t>(nb_bytes);
|
// Additionally, the offset may take us beyond the range of a size_t (e.g.
|
// 32-bit platforms). If so, just clamp at EOF.
|
if (unsignedNbBytes >
|
std::numeric_limits<OPJ_SIZE_T>::max() - srcData->offset) {
|
srcData->offset = srcData->src_size;
|
} else {
|
OPJ_SIZE_T checkedNbBytes = static_cast<OPJ_SIZE_T>(unsignedNbBytes);
|
// Otherwise, mimic fseek() semantics to always succeed, even past EOF,
|
// clamping at EOF. We can get away with this since we don't actually
|
// provide negative relative skips from beyond EOF back to inside the
|
// data, which would be the only reason to need to know exactly how far
|
// beyond EOF we are.
|
srcData->offset =
|
std::min(srcData->offset + checkedNbBytes, srcData->src_size);
|
}
|
return nb_bytes;
|
}
|
|
OPJ_BOOL opj_seek_from_memory(OPJ_OFF_T nb_bytes, void* p_user_data) {
|
DecodeData* srcData = static_cast<DecodeData*>(p_user_data);
|
if (!srcData || !srcData->src_data || srcData->src_size == 0) {
|
return OPJ_FALSE;
|
}
|
// Offsets are signed and may indicate a negative position, which would
|
// be before the start of the file. Do not support this.
|
if (nb_bytes < 0) {
|
return OPJ_FALSE;
|
}
|
// FIXME: use std::make_unsigned<OPJ_OFF_T>::type once c++11 lib is OK'd.
|
uint64_t unsignedNbBytes = static_cast<uint64_t>(nb_bytes);
|
// Additionally, the offset may take us beyond the range of a size_t (e.g.
|
// 32-bit platforms). If so, just clamp at EOF.
|
if (unsignedNbBytes > std::numeric_limits<OPJ_SIZE_T>::max()) {
|
srcData->offset = srcData->src_size;
|
} else {
|
OPJ_SIZE_T checkedNbBytes = static_cast<OPJ_SIZE_T>(nb_bytes);
|
// Otherwise, mimic fseek() semantics to always succeed, even past EOF,
|
// again clamping at EOF.
|
srcData->offset = std::min(checkedNbBytes, srcData->src_size);
|
}
|
return OPJ_TRUE;
|
}
|
opj_stream_t* fx_opj_stream_create_memory_stream(DecodeData* data,
|
OPJ_SIZE_T p_size,
|
OPJ_BOOL p_is_read_stream) {
|
opj_stream_t* l_stream = 00;
|
if (!data || !data->src_data || data->src_size <= 0) {
|
return nullptr;
|
}
|
l_stream = opj_stream_create(p_size, p_is_read_stream);
|
if (!l_stream) {
|
return nullptr;
|
}
|
opj_stream_set_user_data(l_stream, data, nullptr);
|
opj_stream_set_user_data_length(l_stream, data->src_size);
|
opj_stream_set_read_function(l_stream, opj_read_from_memory);
|
opj_stream_set_write_function(l_stream, opj_write_from_memory);
|
opj_stream_set_skip_function(l_stream, opj_skip_from_memory);
|
opj_stream_set_seek_function(l_stream, opj_seek_from_memory);
|
return l_stream;
|
}
|
static void sycc_to_rgb(int offset,
|
int upb,
|
int y,
|
int cb,
|
int cr,
|
int* out_r,
|
int* out_g,
|
int* out_b) {
|
int r, g, b;
|
cb -= offset;
|
cr -= offset;
|
r = y + (int)(1.402 * (float)cr);
|
if (r < 0) {
|
r = 0;
|
} else if (r > upb) {
|
r = upb;
|
}
|
*out_r = r;
|
g = y - (int)(0.344 * (float)cb + 0.714 * (float)cr);
|
if (g < 0) {
|
g = 0;
|
} else if (g > upb) {
|
g = upb;
|
}
|
*out_g = g;
|
b = y + (int)(1.772 * (float)cb);
|
if (b < 0) {
|
b = 0;
|
} else if (b > upb) {
|
b = upb;
|
}
|
*out_b = b;
|
}
|
|
static void sycc444_to_rgb(opj_image_t* img) {
|
int prec = img->comps[0].prec;
|
int offset = 1 << (prec - 1);
|
int upb = (1 << prec) - 1;
|
OPJ_UINT32 maxw =
|
std::min({img->comps[0].w, img->comps[1].w, img->comps[2].w});
|
OPJ_UINT32 maxh =
|
std::min({img->comps[0].h, img->comps[1].h, img->comps[2].h});
|
FX_SAFE_SIZE_T max_size = maxw;
|
max_size *= maxh;
|
if (!max_size.IsValid())
|
return;
|
|
const int* y = img->comps[0].data;
|
const int* cb = img->comps[1].data;
|
const int* cr = img->comps[2].data;
|
if (!y || !cb || !cr)
|
return;
|
|
int* r = FX_Alloc(int, max_size.ValueOrDie());
|
int* g = FX_Alloc(int, max_size.ValueOrDie());
|
int* b = FX_Alloc(int, max_size.ValueOrDie());
|
int* d0 = r;
|
int* d1 = g;
|
int* d2 = b;
|
for (size_t i = 0; i < max_size.ValueOrDie(); ++i) {
|
sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
|
++y;
|
++cb;
|
++cr;
|
++r;
|
++g;
|
++b;
|
}
|
FX_Free(img->comps[0].data);
|
FX_Free(img->comps[1].data);
|
FX_Free(img->comps[2].data);
|
img->comps[0].data = d0;
|
img->comps[1].data = d1;
|
img->comps[2].data = d2;
|
}
|
|
static bool sycc420_422_size_is_valid(opj_image_t* img) {
|
return (img && img->comps[0].w != std::numeric_limits<OPJ_UINT32>::max() &&
|
(img->comps[0].w + 1) / 2 == img->comps[1].w &&
|
img->comps[1].w == img->comps[2].w &&
|
img->comps[1].h == img->comps[2].h);
|
}
|
static bool sycc420_size_is_valid(opj_image_t* img) {
|
return (sycc420_422_size_is_valid(img) &&
|
img->comps[0].h != std::numeric_limits<OPJ_UINT32>::max() &&
|
(img->comps[0].h + 1) / 2 == img->comps[1].h);
|
}
|
static bool sycc422_size_is_valid(opj_image_t* img) {
|
return (sycc420_422_size_is_valid(img) && img->comps[0].h == img->comps[1].h);
|
}
|
static void sycc422_to_rgb(opj_image_t* img) {
|
if (!sycc422_size_is_valid(img))
|
return;
|
|
int prec = img->comps[0].prec;
|
int offset = 1 << (prec - 1);
|
int upb = (1 << prec) - 1;
|
|
OPJ_UINT32 maxw = img->comps[0].w;
|
OPJ_UINT32 maxh = img->comps[0].h;
|
FX_SAFE_SIZE_T max_size = maxw;
|
max_size *= maxh;
|
if (!max_size.IsValid())
|
return;
|
|
const int* y = img->comps[0].data;
|
const int* cb = img->comps[1].data;
|
const int* cr = img->comps[2].data;
|
int *d0, *d1, *d2, *r, *g, *b;
|
d0 = r = FX_Alloc(int, max_size.ValueOrDie());
|
d1 = g = FX_Alloc(int, max_size.ValueOrDie());
|
d2 = b = FX_Alloc(int, max_size.ValueOrDie());
|
for (uint32_t i = 0; i < maxh; ++i) {
|
OPJ_UINT32 j;
|
for (j = 0; j < (maxw & ~static_cast<OPJ_UINT32>(1)); j += 2) {
|
sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
|
++y;
|
++r;
|
++g;
|
++b;
|
sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
|
++y;
|
++r;
|
++g;
|
++b;
|
++cb;
|
++cr;
|
}
|
if (j < maxw) {
|
sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
|
++y;
|
++r;
|
++g;
|
++b;
|
++cb;
|
++cr;
|
}
|
}
|
FX_Free(img->comps[0].data);
|
img->comps[0].data = d0;
|
FX_Free(img->comps[1].data);
|
img->comps[1].data = d1;
|
FX_Free(img->comps[2].data);
|
img->comps[2].data = d2;
|
img->comps[1].w = maxw;
|
img->comps[1].h = maxh;
|
img->comps[2].w = maxw;
|
img->comps[2].h = maxh;
|
img->comps[1].dx = img->comps[0].dx;
|
img->comps[2].dx = img->comps[0].dx;
|
img->comps[1].dy = img->comps[0].dy;
|
img->comps[2].dy = img->comps[0].dy;
|
}
|
static bool sycc420_must_extend_cbcr(OPJ_UINT32 y, OPJ_UINT32 cbcr) {
|
return (y & 1) && (cbcr == y / 2);
|
}
|
void sycc420_to_rgb(opj_image_t* img) {
|
if (!sycc420_size_is_valid(img))
|
return;
|
|
OPJ_UINT32 prec = img->comps[0].prec;
|
if (!prec)
|
return;
|
OPJ_UINT32 offset = 1 << (prec - 1);
|
OPJ_UINT32 upb = (1 << prec) - 1;
|
OPJ_UINT32 yw = img->comps[0].w;
|
OPJ_UINT32 yh = img->comps[0].h;
|
OPJ_UINT32 cbw = img->comps[1].w;
|
OPJ_UINT32 cbh = img->comps[1].h;
|
OPJ_UINT32 crw = img->comps[2].w;
|
bool extw = sycc420_must_extend_cbcr(yw, cbw);
|
bool exth = sycc420_must_extend_cbcr(yh, cbh);
|
FX_SAFE_UINT32 safeSize = yw;
|
safeSize *= yh;
|
if (!safeSize.IsValid())
|
return;
|
int* r = FX_Alloc(int, safeSize.ValueOrDie());
|
int* g = FX_Alloc(int, safeSize.ValueOrDie());
|
int* b = FX_Alloc(int, safeSize.ValueOrDie());
|
int* d0 = r;
|
int* d1 = g;
|
int* d2 = b;
|
const int* y = img->comps[0].data;
|
const int* cb = img->comps[1].data;
|
const int* cr = img->comps[2].data;
|
const int* ny = nullptr;
|
int* nr = nullptr;
|
int* ng = nullptr;
|
int* nb = nullptr;
|
OPJ_UINT32 i = 0;
|
OPJ_UINT32 j = 0;
|
for (i = 0; i < (yh & ~(OPJ_UINT32)1); i += 2) {
|
ny = y + yw;
|
nr = r + yw;
|
ng = g + yw;
|
nb = b + yw;
|
for (j = 0; j < (yw & ~(OPJ_UINT32)1); j += 2) {
|
sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
|
++y;
|
++r;
|
++g;
|
++b;
|
sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
|
++y;
|
++r;
|
++g;
|
++b;
|
sycc_to_rgb(offset, upb, *ny, *cb, *cr, nr, ng, nb);
|
++ny;
|
++nr;
|
++ng;
|
++nb;
|
sycc_to_rgb(offset, upb, *ny, *cb, *cr, nr, ng, nb);
|
++ny;
|
++nr;
|
++ng;
|
++nb;
|
++cb;
|
++cr;
|
}
|
if (j < yw) {
|
if (extw) {
|
--cb;
|
--cr;
|
}
|
sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
|
++y;
|
++r;
|
++g;
|
++b;
|
sycc_to_rgb(offset, upb, *ny, *cb, *cr, nr, ng, nb);
|
++ny;
|
++nr;
|
++ng;
|
++nb;
|
++cb;
|
++cr;
|
}
|
y += yw;
|
r += yw;
|
g += yw;
|
b += yw;
|
}
|
if (i < yh) {
|
if (exth) {
|
cb -= cbw;
|
cr -= crw;
|
}
|
for (j = 0; j < (yw & ~(OPJ_UINT32)1); j += 2) {
|
sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
|
++y;
|
++r;
|
++g;
|
++b;
|
sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
|
++y;
|
++r;
|
++g;
|
++b;
|
++cb;
|
++cr;
|
}
|
if (j < yw) {
|
if (extw) {
|
--cb;
|
--cr;
|
}
|
sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
|
}
|
}
|
|
FX_Free(img->comps[0].data);
|
img->comps[0].data = d0;
|
FX_Free(img->comps[1].data);
|
img->comps[1].data = d1;
|
FX_Free(img->comps[2].data);
|
img->comps[2].data = d2;
|
img->comps[1].w = yw;
|
img->comps[1].h = yh;
|
img->comps[2].w = yw;
|
img->comps[2].h = yh;
|
img->comps[1].w = yw;
|
img->comps[1].h = yh;
|
img->comps[2].w = yw;
|
img->comps[2].h = yh;
|
img->comps[1].dx = img->comps[0].dx;
|
img->comps[2].dx = img->comps[0].dx;
|
img->comps[1].dy = img->comps[0].dy;
|
img->comps[2].dy = img->comps[0].dy;
|
}
|
void color_sycc_to_rgb(opj_image_t* img) {
|
if (img->numcomps < 3) {
|
img->color_space = OPJ_CLRSPC_GRAY;
|
return;
|
}
|
if ((img->comps[0].dx == 1) && (img->comps[1].dx == 2) &&
|
(img->comps[2].dx == 2) && (img->comps[0].dy == 1) &&
|
(img->comps[1].dy == 2) && (img->comps[2].dy == 2)) {
|
sycc420_to_rgb(img);
|
} else if ((img->comps[0].dx == 1) && (img->comps[1].dx == 2) &&
|
(img->comps[2].dx == 2) && (img->comps[0].dy == 1) &&
|
(img->comps[1].dy == 1) && (img->comps[2].dy == 1)) {
|
sycc422_to_rgb(img);
|
} else if ((img->comps[0].dx == 1) && (img->comps[1].dx == 1) &&
|
(img->comps[2].dx == 1) && (img->comps[0].dy == 1) &&
|
(img->comps[1].dy == 1) && (img->comps[2].dy == 1)) {
|
sycc444_to_rgb(img);
|
} else {
|
return;
|
}
|
img->color_space = OPJ_CLRSPC_SRGB;
|
}
|
void color_apply_icc_profile(opj_image_t* image) {
|
cmsHPROFILE out_prof;
|
cmsUInt32Number in_type;
|
cmsUInt32Number out_type;
|
int* r;
|
int* g;
|
int* b;
|
int max;
|
cmsHPROFILE in_prof =
|
cmsOpenProfileFromMem(image->icc_profile_buf, image->icc_profile_len);
|
if (!in_prof) {
|
return;
|
}
|
cmsColorSpaceSignature out_space = cmsGetColorSpace(in_prof);
|
cmsUInt32Number intent = cmsGetHeaderRenderingIntent(in_prof);
|
int max_w = (int)image->comps[0].w;
|
int max_h = (int)image->comps[0].h;
|
int prec = (int)image->comps[0].prec;
|
OPJ_COLOR_SPACE oldspace = image->color_space;
|
if (out_space == cmsSigRgbData) {
|
if (prec <= 8) {
|
in_type = TYPE_RGB_8;
|
out_type = TYPE_RGB_8;
|
} else {
|
in_type = TYPE_RGB_16;
|
out_type = TYPE_RGB_16;
|
}
|
out_prof = cmsCreate_sRGBProfile();
|
image->color_space = OPJ_CLRSPC_SRGB;
|
} else if (out_space == cmsSigGrayData) {
|
if (prec <= 8) {
|
in_type = TYPE_GRAY_8;
|
out_type = TYPE_RGB_8;
|
} else {
|
in_type = TYPE_GRAY_16;
|
out_type = TYPE_RGB_16;
|
}
|
out_prof = cmsCreate_sRGBProfile();
|
image->color_space = OPJ_CLRSPC_SRGB;
|
} else if (out_space == cmsSigYCbCrData) {
|
in_type = TYPE_YCbCr_16;
|
out_type = TYPE_RGB_16;
|
out_prof = cmsCreate_sRGBProfile();
|
image->color_space = OPJ_CLRSPC_SRGB;
|
} else {
|
return;
|
}
|
cmsHTRANSFORM transform =
|
cmsCreateTransform(in_prof, in_type, out_prof, out_type, intent, 0);
|
cmsCloseProfile(in_prof);
|
cmsCloseProfile(out_prof);
|
if (!transform) {
|
image->color_space = oldspace;
|
return;
|
}
|
if (image->numcomps > 2) {
|
if (prec <= 8) {
|
unsigned char *inbuf, *outbuf, *in, *out;
|
max = max_w * max_h;
|
cmsUInt32Number nr_samples = max * 3 * sizeof(unsigned char);
|
in = inbuf = FX_Alloc(unsigned char, nr_samples);
|
out = outbuf = FX_Alloc(unsigned char, nr_samples);
|
r = image->comps[0].data;
|
g = image->comps[1].data;
|
b = image->comps[2].data;
|
for (int i = 0; i < max; ++i) {
|
*in++ = (unsigned char)*r++;
|
*in++ = (unsigned char)*g++;
|
*in++ = (unsigned char)*b++;
|
}
|
cmsDoTransform(transform, inbuf, outbuf, (cmsUInt32Number)max);
|
r = image->comps[0].data;
|
g = image->comps[1].data;
|
b = image->comps[2].data;
|
for (int i = 0; i < max; ++i) {
|
*r++ = (int)*out++;
|
*g++ = (int)*out++;
|
*b++ = (int)*out++;
|
}
|
FX_Free(inbuf);
|
FX_Free(outbuf);
|
} else {
|
unsigned short *inbuf, *outbuf, *in, *out;
|
max = max_w * max_h;
|
cmsUInt32Number nr_samples = max * 3 * sizeof(unsigned short);
|
in = inbuf = FX_Alloc(unsigned short, nr_samples);
|
out = outbuf = FX_Alloc(unsigned short, nr_samples);
|
r = image->comps[0].data;
|
g = image->comps[1].data;
|
b = image->comps[2].data;
|
for (int i = 0; i < max; ++i) {
|
*in++ = (unsigned short)*r++;
|
*in++ = (unsigned short)*g++;
|
*in++ = (unsigned short)*b++;
|
}
|
cmsDoTransform(transform, inbuf, outbuf, (cmsUInt32Number)max);
|
r = image->comps[0].data;
|
g = image->comps[1].data;
|
b = image->comps[2].data;
|
for (int i = 0; i < max; ++i) {
|
*r++ = (int)*out++;
|
*g++ = (int)*out++;
|
*b++ = (int)*out++;
|
}
|
FX_Free(inbuf);
|
FX_Free(outbuf);
|
}
|
} else {
|
unsigned char *in, *inbuf, *out, *outbuf;
|
max = max_w * max_h;
|
cmsUInt32Number nr_samples =
|
(cmsUInt32Number)max * 3 * sizeof(unsigned char);
|
in = inbuf = FX_Alloc(unsigned char, nr_samples);
|
out = outbuf = FX_Alloc(unsigned char, nr_samples);
|
image->comps = (opj_image_comp_t*)realloc(
|
image->comps, (image->numcomps + 2) * sizeof(opj_image_comp_t));
|
if (image->numcomps == 2) {
|
image->comps[3] = image->comps[1];
|
}
|
image->comps[1] = image->comps[0];
|
image->comps[2] = image->comps[0];
|
image->comps[1].data = FX_Alloc(int, (size_t)max);
|
FXSYS_memset(image->comps[1].data, 0, sizeof(int) * (size_t)max);
|
image->comps[2].data = FX_Alloc(int, (size_t)max);
|
FXSYS_memset(image->comps[2].data, 0, sizeof(int) * (size_t)max);
|
image->numcomps += 2;
|
r = image->comps[0].data;
|
for (int i = 0; i < max; ++i) {
|
*in++ = (unsigned char)*r++;
|
}
|
cmsDoTransform(transform, inbuf, outbuf, (cmsUInt32Number)max);
|
r = image->comps[0].data;
|
g = image->comps[1].data;
|
b = image->comps[2].data;
|
for (int i = 0; i < max; ++i) {
|
*r++ = (int)*out++;
|
*g++ = (int)*out++;
|
*b++ = (int)*out++;
|
}
|
FX_Free(inbuf);
|
FX_Free(outbuf);
|
}
|
cmsDeleteTransform(transform);
|
}
|
void color_apply_conversion(opj_image_t* image) {
|
int* row;
|
int enumcs, numcomps;
|
numcomps = image->numcomps;
|
if (numcomps < 3) {
|
return;
|
}
|
row = (int*)image->icc_profile_buf;
|
enumcs = row[0];
|
if (enumcs == 14) {
|
int *L, *a, *b, *red, *green, *blue, *src0, *src1, *src2;
|
double rl, ol, ra, oa, rb, ob, prec0, prec1, prec2;
|
double minL, maxL, mina, maxa, minb, maxb;
|
unsigned int default_type;
|
unsigned int i, max;
|
cmsHPROFILE in, out;
|
cmsHTRANSFORM transform;
|
cmsUInt16Number RGB[3];
|
cmsCIELab Lab;
|
in = cmsCreateLab4Profile(nullptr);
|
out = cmsCreate_sRGBProfile();
|
transform = cmsCreateTransform(in, TYPE_Lab_DBL, out, TYPE_RGB_16,
|
INTENT_PERCEPTUAL, 0);
|
cmsCloseProfile(in);
|
cmsCloseProfile(out);
|
if (!transform) {
|
return;
|
}
|
prec0 = (double)image->comps[0].prec;
|
prec1 = (double)image->comps[1].prec;
|
prec2 = (double)image->comps[2].prec;
|
default_type = row[1];
|
if (default_type == 0x44454600) {
|
rl = 100;
|
ra = 170;
|
rb = 200;
|
ol = 0;
|
oa = pow(2, prec1 - 1);
|
ob = pow(2, prec2 - 2) + pow(2, prec2 - 3);
|
} else {
|
rl = row[2];
|
ra = row[4];
|
rb = row[6];
|
ol = row[3];
|
oa = row[5];
|
ob = row[7];
|
}
|
L = src0 = image->comps[0].data;
|
a = src1 = image->comps[1].data;
|
b = src2 = image->comps[2].data;
|
max = image->comps[0].w * image->comps[0].h;
|
red = FX_Alloc(int, max);
|
image->comps[0].data = red;
|
green = FX_Alloc(int, max);
|
image->comps[1].data = green;
|
blue = FX_Alloc(int, max);
|
image->comps[2].data = blue;
|
minL = -(rl * ol) / (pow(2, prec0) - 1);
|
maxL = minL + rl;
|
mina = -(ra * oa) / (pow(2, prec1) - 1);
|
maxa = mina + ra;
|
minb = -(rb * ob) / (pow(2, prec2) - 1);
|
maxb = minb + rb;
|
for (i = 0; i < max; ++i) {
|
Lab.L = minL + (double)(*L) * (maxL - minL) / (pow(2, prec0) - 1);
|
++L;
|
Lab.a = mina + (double)(*a) * (maxa - mina) / (pow(2, prec1) - 1);
|
++a;
|
Lab.b = minb + (double)(*b) * (maxb - minb) / (pow(2, prec2) - 1);
|
++b;
|
cmsDoTransform(transform, &Lab, RGB, 1);
|
*red++ = RGB[0];
|
*green++ = RGB[1];
|
*blue++ = RGB[2];
|
}
|
cmsDeleteTransform(transform);
|
FX_Free(src0);
|
FX_Free(src1);
|
FX_Free(src2);
|
image->color_space = OPJ_CLRSPC_SRGB;
|
image->comps[0].prec = 16;
|
image->comps[1].prec = 16;
|
image->comps[2].prec = 16;
|
return;
|
}
|
}
|
class CJPX_Decoder {
|
public:
|
explicit CJPX_Decoder(CPDF_ColorSpace* cs);
|
~CJPX_Decoder();
|
FX_BOOL Init(const unsigned char* src_data, uint32_t src_size);
|
void GetInfo(uint32_t* width, uint32_t* height, uint32_t* components);
|
bool Decode(uint8_t* dest_buf,
|
int pitch,
|
const std::vector<uint8_t>& offsets);
|
|
private:
|
const uint8_t* m_SrcData;
|
uint32_t m_SrcSize;
|
opj_image_t* image;
|
opj_codec_t* l_codec;
|
opj_stream_t* l_stream;
|
const CPDF_ColorSpace* const m_ColorSpace;
|
};
|
|
CJPX_Decoder::CJPX_Decoder(CPDF_ColorSpace* cs)
|
: image(nullptr), l_codec(nullptr), l_stream(nullptr), m_ColorSpace(cs) {}
|
|
CJPX_Decoder::~CJPX_Decoder() {
|
if (l_codec) {
|
opj_destroy_codec(l_codec);
|
}
|
if (l_stream) {
|
opj_stream_destroy(l_stream);
|
}
|
if (image) {
|
opj_image_destroy(image);
|
}
|
}
|
|
FX_BOOL CJPX_Decoder::Init(const unsigned char* src_data, uint32_t src_size) {
|
static const unsigned char szJP2Header[] = {
|
0x00, 0x00, 0x00, 0x0c, 0x6a, 0x50, 0x20, 0x20, 0x0d, 0x0a, 0x87, 0x0a};
|
if (!src_data || src_size < sizeof(szJP2Header))
|
return FALSE;
|
|
image = nullptr;
|
m_SrcData = src_data;
|
m_SrcSize = src_size;
|
DecodeData srcData(const_cast<unsigned char*>(src_data), src_size);
|
l_stream = fx_opj_stream_create_memory_stream(&srcData,
|
OPJ_J2K_STREAM_CHUNK_SIZE, 1);
|
if (!l_stream) {
|
return FALSE;
|
}
|
opj_dparameters_t parameters;
|
opj_set_default_decoder_parameters(¶meters);
|
parameters.decod_format = 0;
|
parameters.cod_format = 3;
|
if (FXSYS_memcmp(m_SrcData, szJP2Header, sizeof(szJP2Header)) == 0) {
|
l_codec = opj_create_decompress(OPJ_CODEC_JP2);
|
parameters.decod_format = 1;
|
} else {
|
l_codec = opj_create_decompress(OPJ_CODEC_J2K);
|
}
|
if (!l_codec) {
|
return FALSE;
|
}
|
if (m_ColorSpace && m_ColorSpace->GetFamily() == PDFCS_INDEXED)
|
parameters.flags |= OPJ_DPARAMETERS_IGNORE_PCLR_CMAP_CDEF_FLAG;
|
opj_set_info_handler(l_codec, fx_info_callback, 00);
|
opj_set_warning_handler(l_codec, fx_warning_callback, 00);
|
opj_set_error_handler(l_codec, fx_error_callback, 00);
|
if (!opj_setup_decoder(l_codec, ¶meters)) {
|
return FALSE;
|
}
|
if (!opj_read_header(l_stream, l_codec, &image)) {
|
image = nullptr;
|
return FALSE;
|
}
|
image->pdfium_use_colorspace = !!m_ColorSpace;
|
|
if (!parameters.nb_tile_to_decode) {
|
if (!opj_set_decode_area(l_codec, image, parameters.DA_x0, parameters.DA_y0,
|
parameters.DA_x1, parameters.DA_y1)) {
|
opj_image_destroy(image);
|
image = nullptr;
|
return FALSE;
|
}
|
if (!(opj_decode(l_codec, l_stream, image) &&
|
opj_end_decompress(l_codec, l_stream))) {
|
opj_image_destroy(image);
|
image = nullptr;
|
return FALSE;
|
}
|
} else {
|
if (!opj_get_decoded_tile(l_codec, l_stream, image,
|
parameters.tile_index)) {
|
return FALSE;
|
}
|
}
|
opj_stream_destroy(l_stream);
|
l_stream = nullptr;
|
if (image->color_space != OPJ_CLRSPC_SYCC && image->numcomps == 3 &&
|
image->comps[0].dx == image->comps[0].dy && image->comps[1].dx != 1) {
|
image->color_space = OPJ_CLRSPC_SYCC;
|
} else if (image->numcomps <= 2) {
|
image->color_space = OPJ_CLRSPC_GRAY;
|
}
|
if (image->color_space == OPJ_CLRSPC_SYCC) {
|
color_sycc_to_rgb(image);
|
}
|
if (image->icc_profile_buf) {
|
FX_Free(image->icc_profile_buf);
|
image->icc_profile_buf = nullptr;
|
image->icc_profile_len = 0;
|
}
|
if (!image) {
|
return FALSE;
|
}
|
return TRUE;
|
}
|
|
void CJPX_Decoder::GetInfo(uint32_t* width,
|
uint32_t* height,
|
uint32_t* components) {
|
*width = (uint32_t)image->x1;
|
*height = (uint32_t)image->y1;
|
*components = (uint32_t)image->numcomps;
|
}
|
|
bool CJPX_Decoder::Decode(uint8_t* dest_buf,
|
int pitch,
|
const std::vector<uint8_t>& offsets) {
|
if (image->comps[0].w != image->x1 || image->comps[0].h != image->y1)
|
return false;
|
|
if (pitch<(int)(image->comps[0].w * 8 * image->numcomps + 31)>> 5 << 2)
|
return false;
|
|
FXSYS_memset(dest_buf, 0xff, image->y1 * pitch);
|
std::vector<uint8_t*> channel_bufs(image->numcomps);
|
std::vector<int> adjust_comps(image->numcomps);
|
for (uint32_t i = 0; i < image->numcomps; i++) {
|
channel_bufs[i] = dest_buf + offsets[i];
|
adjust_comps[i] = image->comps[i].prec - 8;
|
if (i > 0) {
|
if (image->comps[i].dx != image->comps[i - 1].dx ||
|
image->comps[i].dy != image->comps[i - 1].dy ||
|
image->comps[i].prec != image->comps[i - 1].prec) {
|
return false;
|
}
|
}
|
}
|
int width = image->comps[0].w;
|
int height = image->comps[0].h;
|
for (uint32_t channel = 0; channel < image->numcomps; ++channel) {
|
uint8_t* pChannel = channel_bufs[channel];
|
if (adjust_comps[channel] < 0) {
|
for (int row = 0; row < height; ++row) {
|
uint8_t* pScanline = pChannel + row * pitch;
|
for (int col = 0; col < width; ++col) {
|
uint8_t* pPixel = pScanline + col * image->numcomps;
|
int src = image->comps[channel].data[row * width + col];
|
src += image->comps[channel].sgnd
|
? 1 << (image->comps[channel].prec - 1)
|
: 0;
|
if (adjust_comps[channel] > 0) {
|
*pPixel = 0;
|
} else {
|
*pPixel = (uint8_t)(src << -adjust_comps[channel]);
|
}
|
}
|
}
|
} else {
|
for (int row = 0; row < height; ++row) {
|
uint8_t* pScanline = pChannel + row * pitch;
|
for (int col = 0; col < width; ++col) {
|
uint8_t* pPixel = pScanline + col * image->numcomps;
|
if (!image->comps[channel].data) {
|
continue;
|
}
|
int src = image->comps[channel].data[row * width + col];
|
src += image->comps[channel].sgnd
|
? 1 << (image->comps[channel].prec - 1)
|
: 0;
|
if (adjust_comps[channel] - 1 < 0) {
|
*pPixel = (uint8_t)((src >> adjust_comps[channel]));
|
} else {
|
int tmpPixel = (src >> adjust_comps[channel]) +
|
((src >> (adjust_comps[channel] - 1)) % 2);
|
if (tmpPixel > 255) {
|
tmpPixel = 255;
|
} else if (tmpPixel < 0) {
|
tmpPixel = 0;
|
}
|
*pPixel = (uint8_t)tmpPixel;
|
}
|
}
|
}
|
}
|
}
|
return true;
|
}
|
|
CCodec_JpxModule::CCodec_JpxModule() {}
|
CCodec_JpxModule::~CCodec_JpxModule() {}
|
|
CJPX_Decoder* CCodec_JpxModule::CreateDecoder(const uint8_t* src_buf,
|
uint32_t src_size,
|
CPDF_ColorSpace* cs) {
|
std::unique_ptr<CJPX_Decoder> decoder(new CJPX_Decoder(cs));
|
return decoder->Init(src_buf, src_size) ? decoder.release() : nullptr;
|
}
|
|
void CCodec_JpxModule::GetImageInfo(CJPX_Decoder* pDecoder,
|
uint32_t* width,
|
uint32_t* height,
|
uint32_t* components) {
|
pDecoder->GetInfo(width, height, components);
|
}
|
|
bool CCodec_JpxModule::Decode(CJPX_Decoder* pDecoder,
|
uint8_t* dest_data,
|
int pitch,
|
const std::vector<uint8_t>& offsets) {
|
return pDecoder->Decode(dest_data, pitch, offsets);
|
}
|
|
void CCodec_JpxModule::DestroyDecoder(CJPX_Decoder* pDecoder) {
|
delete pDecoder;
|
}
|
