Home | History | Annotate | Download | only in codec
      1 // Copyright 2014 PDFium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
      6 
      7 #include <algorithm>
      8 #include <limits>
      9 #include <vector>
     10 
     11 #include "codec_int.h"
     12 #include "core/include/fpdfapi/fpdf_resource.h"
     13 #include "core/include/fxcodec/fx_codec.h"
     14 #include "core/include/fxcrt/fx_safe_types.h"
     15 #include "third_party/lcms2-2.6/include/lcms2.h"
     16 #include "third_party/libopenjpeg20/openjpeg.h"
     17 
     18 static void fx_error_callback(const char* msg, void* client_data) {
     19   (void)client_data;
     20 }
     21 static void fx_warning_callback(const char* msg, void* client_data) {
     22   (void)client_data;
     23 }
     24 static void fx_info_callback(const char* msg, void* client_data) {
     25   (void)client_data;
     26 }
     27 OPJ_SIZE_T opj_read_from_memory(void* p_buffer,
     28                                 OPJ_SIZE_T nb_bytes,
     29                                 void* p_user_data) {
     30   DecodeData* srcData = static_cast<DecodeData*>(p_user_data);
     31   if (!srcData || !srcData->src_data || srcData->src_size == 0) {
     32     return -1;
     33   }
     34   // Reads at EOF return an error code.
     35   if (srcData->offset >= srcData->src_size) {
     36     return -1;
     37   }
     38   OPJ_SIZE_T bufferLength = srcData->src_size - srcData->offset;
     39   OPJ_SIZE_T readlength = nb_bytes < bufferLength ? nb_bytes : bufferLength;
     40   memcpy(p_buffer, &srcData->src_data[srcData->offset], readlength);
     41   srcData->offset += readlength;
     42   return readlength;
     43 }
     44 OPJ_SIZE_T opj_write_from_memory(void* p_buffer,
     45                                  OPJ_SIZE_T nb_bytes,
     46                                  void* p_user_data) {
     47   DecodeData* srcData = static_cast<DecodeData*>(p_user_data);
     48   if (!srcData || !srcData->src_data || srcData->src_size == 0) {
     49     return -1;
     50   }
     51   // Writes at EOF return an error code.
     52   if (srcData->offset >= srcData->src_size) {
     53     return -1;
     54   }
     55   OPJ_SIZE_T bufferLength = srcData->src_size - srcData->offset;
     56   OPJ_SIZE_T writeLength = nb_bytes < bufferLength ? nb_bytes : bufferLength;
     57   memcpy(&srcData->src_data[srcData->offset], p_buffer, writeLength);
     58   srcData->offset += writeLength;
     59   return writeLength;
     60 }
     61 OPJ_OFF_T opj_skip_from_memory(OPJ_OFF_T nb_bytes, void* p_user_data) {
     62   DecodeData* srcData = static_cast<DecodeData*>(p_user_data);
     63   if (!srcData || !srcData->src_data || srcData->src_size == 0) {
     64     return -1;
     65   }
     66   // Offsets are signed and may indicate a negative skip. Do not support this
     67   // because of the strange return convention where either bytes skipped or
     68   // -1 is returned. Following that convention, a successful relative seek of
     69   // -1 bytes would be required to to give the same result as the error case.
     70   if (nb_bytes < 0) {
     71     return -1;
     72   }
     73   // FIXME: use std::make_unsigned<OPJ_OFF_T>::type once c++11 lib is OK'd.
     74   uint64_t unsignedNbBytes = static_cast<uint64_t>(nb_bytes);
     75   // Additionally, the offset may take us beyond the range of a size_t (e.g.
     76   // 32-bit platforms). If so, just clamp at EOF.
     77   if (unsignedNbBytes >
     78       std::numeric_limits<OPJ_SIZE_T>::max() - srcData->offset) {
     79     srcData->offset = srcData->src_size;
     80   } else {
     81     OPJ_SIZE_T checkedNbBytes = static_cast<OPJ_SIZE_T>(unsignedNbBytes);
     82     // Otherwise, mimic fseek() semantics to always succeed, even past EOF,
     83     // clamping at EOF.  We can get away with this since we don't actually
     84     // provide negative relative skips from beyond EOF back to inside the
     85     // data, which would be the only reason to need to know exactly how far
     86     // beyond EOF we are.
     87     srcData->offset =
     88         std::min(srcData->offset + checkedNbBytes, srcData->src_size);
     89   }
     90   return nb_bytes;
     91 }
     92 OPJ_BOOL opj_seek_from_memory(OPJ_OFF_T nb_bytes, void* p_user_data) {
     93   DecodeData* srcData = static_cast<DecodeData*>(p_user_data);
     94   if (!srcData || !srcData->src_data || srcData->src_size == 0) {
     95     return OPJ_FALSE;
     96   }
     97   // Offsets are signed and may indicate a negative position, which would
     98   // be before the start of the file. Do not support this.
     99   if (nb_bytes < 0) {
    100     return OPJ_FALSE;
    101   }
    102   // FIXME: use std::make_unsigned<OPJ_OFF_T>::type once c++11 lib is OK'd.
    103   uint64_t unsignedNbBytes = static_cast<uint64_t>(nb_bytes);
    104   // Additionally, the offset may take us beyond the range of a size_t (e.g.
    105   // 32-bit platforms). If so, just clamp at EOF.
    106   if (unsignedNbBytes > std::numeric_limits<OPJ_SIZE_T>::max()) {
    107     srcData->offset = srcData->src_size;
    108   } else {
    109     OPJ_SIZE_T checkedNbBytes = static_cast<OPJ_SIZE_T>(nb_bytes);
    110     // Otherwise, mimic fseek() semantics to always succeed, even past EOF,
    111     // again clamping at EOF.
    112     srcData->offset = std::min(checkedNbBytes, srcData->src_size);
    113   }
    114   return OPJ_TRUE;
    115 }
    116 opj_stream_t* fx_opj_stream_create_memory_stream(DecodeData* data,
    117                                                  OPJ_SIZE_T p_size,
    118                                                  OPJ_BOOL p_is_read_stream) {
    119   opj_stream_t* l_stream = 00;
    120   if (!data || !data->src_data || data->src_size <= 0) {
    121     return NULL;
    122   }
    123   l_stream = opj_stream_create(p_size, p_is_read_stream);
    124   if (!l_stream) {
    125     return NULL;
    126   }
    127   opj_stream_set_user_data(l_stream, data, NULL);
    128   opj_stream_set_user_data_length(l_stream, data->src_size);
    129   opj_stream_set_read_function(l_stream, opj_read_from_memory);
    130   opj_stream_set_write_function(l_stream, opj_write_from_memory);
    131   opj_stream_set_skip_function(l_stream, opj_skip_from_memory);
    132   opj_stream_set_seek_function(l_stream, opj_seek_from_memory);
    133   return l_stream;
    134 }
    135 static void sycc_to_rgb(int offset,
    136                         int upb,
    137                         int y,
    138                         int cb,
    139                         int cr,
    140                         int* out_r,
    141                         int* out_g,
    142                         int* out_b) {
    143   int r, g, b;
    144   cb -= offset;
    145   cr -= offset;
    146   r = y + (int)(1.402 * (float)cr);
    147   if (r < 0) {
    148     r = 0;
    149   } else if (r > upb) {
    150     r = upb;
    151   }
    152   *out_r = r;
    153   g = y - (int)(0.344 * (float)cb + 0.714 * (float)cr);
    154   if (g < 0) {
    155     g = 0;
    156   } else if (g > upb) {
    157     g = upb;
    158   }
    159   *out_g = g;
    160   b = y + (int)(1.772 * (float)cb);
    161   if (b < 0) {
    162     b = 0;
    163   } else if (b > upb) {
    164     b = upb;
    165   }
    166   *out_b = b;
    167 }
    168 static void sycc444_to_rgb(opj_image_t* img) {
    169   int prec = img->comps[0].prec;
    170   int offset = 1 << (prec - 1);
    171   int upb = (1 << prec) - 1;
    172   OPJ_UINT32 maxw =
    173       std::min(std::min(img->comps[0].w, img->comps[1].w), img->comps[2].w);
    174   OPJ_UINT32 maxh =
    175       std::min(std::min(img->comps[0].h, img->comps[1].h), img->comps[2].h);
    176   FX_SAFE_SIZE_T max_size = maxw;
    177   max_size *= maxh;
    178   if (!max_size.IsValid())
    179     return;
    180 
    181   const int* y = img->comps[0].data;
    182   const int* cb = img->comps[1].data;
    183   const int* cr = img->comps[2].data;
    184   int *d0, *d1, *d2, *r, *g, *b;
    185   d0 = r = FX_Alloc(int, max_size.ValueOrDie());
    186   d1 = g = FX_Alloc(int, max_size.ValueOrDie());
    187   d2 = b = FX_Alloc(int, max_size.ValueOrDie());
    188   for (size_t i = 0; i < max_size.ValueOrDie(); ++i) {
    189     sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
    190     ++y;
    191     ++cb;
    192     ++cr;
    193     ++r;
    194     ++g;
    195     ++b;
    196   }
    197   FX_Free(img->comps[0].data);
    198   img->comps[0].data = d0;
    199   FX_Free(img->comps[1].data);
    200   img->comps[1].data = d1;
    201   FX_Free(img->comps[2].data);
    202   img->comps[2].data = d2;
    203 }
    204 static void sycc422_to_rgb(opj_image_t* img) {
    205   int prec = img->comps[0].prec;
    206   int offset = 1 << (prec - 1);
    207   int upb = (1 << prec) - 1;
    208   OPJ_UINT32 maxw =
    209       std::min(std::min(img->comps[0].w, img->comps[1].w), img->comps[2].w);
    210   OPJ_UINT32 maxh =
    211       std::min(std::min(img->comps[0].h, img->comps[1].h), img->comps[2].h);
    212   FX_SAFE_SIZE_T max_size = maxw;
    213   max_size *= maxh;
    214   if (!max_size.IsValid())
    215     return;
    216 
    217   const int* y = img->comps[0].data;
    218   const int* cb = img->comps[1].data;
    219   const int* cr = img->comps[2].data;
    220   int *d0, *d1, *d2, *r, *g, *b;
    221   d0 = r = FX_Alloc(int, max_size.ValueOrDie());
    222   d1 = g = FX_Alloc(int, max_size.ValueOrDie());
    223   d2 = b = FX_Alloc(int, max_size.ValueOrDie());
    224   for (uint32_t i = 0; i < maxh; ++i) {
    225     OPJ_UINT32 j;
    226     for (j = 0; j < (maxw & ~static_cast<OPJ_UINT32>(1)); j += 2) {
    227       sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
    228       ++y;
    229       ++r;
    230       ++g;
    231       ++b;
    232       sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
    233       ++y;
    234       ++r;
    235       ++g;
    236       ++b;
    237       ++cb;
    238       ++cr;
    239     }
    240     if (j < maxw) {
    241       sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
    242       ++y;
    243       ++r;
    244       ++g;
    245       ++b;
    246       ++cb;
    247       ++cr;
    248     }
    249   }
    250   FX_Free(img->comps[0].data);
    251   img->comps[0].data = d0;
    252   FX_Free(img->comps[1].data);
    253   img->comps[1].data = d1;
    254   FX_Free(img->comps[2].data);
    255   img->comps[2].data = d2;
    256   img->comps[1].w = maxw;
    257   img->comps[1].h = maxh;
    258   img->comps[2].w = maxw;
    259   img->comps[2].h = maxh;
    260   img->comps[1].dx = img->comps[0].dx;
    261   img->comps[2].dx = img->comps[0].dx;
    262   img->comps[1].dy = img->comps[0].dy;
    263   img->comps[2].dy = img->comps[0].dy;
    264 }
    265 static bool sycc420_size_is_valid(OPJ_UINT32 y, OPJ_UINT32 cbcr) {
    266   if (!y || !cbcr)
    267     return false;
    268 
    269   return (cbcr == y / 2) || ((y & 1) && (cbcr == y / 2 + 1));
    270 }
    271 static bool sycc420_must_extend_cbcr(OPJ_UINT32 y, OPJ_UINT32 cbcr) {
    272   return (y & 1) && (cbcr == y / 2);
    273 }
    274 void sycc420_to_rgb(opj_image_t* img) {
    275   OPJ_UINT32 prec = img->comps[0].prec;
    276   if (!prec)
    277     return;
    278   OPJ_UINT32 offset = 1 << (prec - 1);
    279   OPJ_UINT32 upb = (1 << prec) - 1;
    280   OPJ_UINT32 yw = img->comps[0].w;
    281   OPJ_UINT32 yh = img->comps[0].h;
    282   OPJ_UINT32 cbw = img->comps[1].w;
    283   OPJ_UINT32 cbh = img->comps[1].h;
    284   OPJ_UINT32 crw = img->comps[2].w;
    285   OPJ_UINT32 crh = img->comps[2].h;
    286   if (cbw != crw || cbh != crh)
    287     return;
    288   if (!sycc420_size_is_valid(yw, cbw) || !sycc420_size_is_valid(yh, cbh))
    289     return;
    290   bool extw = sycc420_must_extend_cbcr(yw, cbw);
    291   bool exth = sycc420_must_extend_cbcr(yh, cbh);
    292   FX_SAFE_DWORD safeSize = yw;
    293   safeSize *= yh;
    294   if (!safeSize.IsValid())
    295     return;
    296   int* r = FX_Alloc(int, safeSize.ValueOrDie());
    297   int* g = FX_Alloc(int, safeSize.ValueOrDie());
    298   int* b = FX_Alloc(int, safeSize.ValueOrDie());
    299   int* d0 = r;
    300   int* d1 = g;
    301   int* d2 = b;
    302   const int* y = img->comps[0].data;
    303   const int* cb = img->comps[1].data;
    304   const int* cr = img->comps[2].data;
    305   const int* ny = nullptr;
    306   int* nr = nullptr;
    307   int* ng = nullptr;
    308   int* nb = nullptr;
    309   OPJ_UINT32 i = 0;
    310   OPJ_UINT32 j = 0;
    311   for (i = 0; i < (yh & ~(OPJ_UINT32)1); i += 2) {
    312     ny = y + yw;
    313     nr = r + yw;
    314     ng = g + yw;
    315     nb = b + yw;
    316     for (j = 0; j < (yw & ~(OPJ_UINT32)1); j += 2) {
    317       sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
    318       ++y;
    319       ++r;
    320       ++g;
    321       ++b;
    322       sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
    323       ++y;
    324       ++r;
    325       ++g;
    326       ++b;
    327       sycc_to_rgb(offset, upb, *ny, *cb, *cr, nr, ng, nb);
    328       ++ny;
    329       ++nr;
    330       ++ng;
    331       ++nb;
    332       sycc_to_rgb(offset, upb, *ny, *cb, *cr, nr, ng, nb);
    333       ++ny;
    334       ++nr;
    335       ++ng;
    336       ++nb;
    337       ++cb;
    338       ++cr;
    339     }
    340     if (j < yw) {
    341       if (extw) {
    342         --cb;
    343         --cr;
    344       }
    345       sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
    346       ++y;
    347       ++r;
    348       ++g;
    349       ++b;
    350       sycc_to_rgb(offset, upb, *ny, *cb, *cr, nr, ng, nb);
    351       ++ny;
    352       ++nr;
    353       ++ng;
    354       ++nb;
    355       ++cb;
    356       ++cr;
    357     }
    358     y += yw;
    359     r += yw;
    360     g += yw;
    361     b += yw;
    362   }
    363   if (i < yh) {
    364     if (exth) {
    365       cb -= cbw;
    366       cr -= crw;
    367     }
    368     for (j = 0; j < (yw & ~(OPJ_UINT32)1); j += 2) {
    369       sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
    370       ++y;
    371       ++r;
    372       ++g;
    373       ++b;
    374       sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
    375       ++y;
    376       ++r;
    377       ++g;
    378       ++b;
    379       ++cb;
    380       ++cr;
    381     }
    382     if (j < yw) {
    383       if (extw) {
    384         --cb;
    385         --cr;
    386       }
    387       sycc_to_rgb(offset, upb, *y, *cb, *cr, r, g, b);
    388     }
    389   }
    390 
    391   FX_Free(img->comps[0].data);
    392   img->comps[0].data = d0;
    393   FX_Free(img->comps[1].data);
    394   img->comps[1].data = d1;
    395   FX_Free(img->comps[2].data);
    396   img->comps[2].data = d2;
    397   img->comps[1].w = yw;
    398   img->comps[1].h = yh;
    399   img->comps[2].w = yw;
    400   img->comps[2].h = yh;
    401   img->comps[1].w = yw;
    402   img->comps[1].h = yh;
    403   img->comps[2].w = yw;
    404   img->comps[2].h = yh;
    405   img->comps[1].dx = img->comps[0].dx;
    406   img->comps[2].dx = img->comps[0].dx;
    407   img->comps[1].dy = img->comps[0].dy;
    408   img->comps[2].dy = img->comps[0].dy;
    409 }
    410 void color_sycc_to_rgb(opj_image_t* img) {
    411   if (img->numcomps < 3) {
    412     img->color_space = OPJ_CLRSPC_GRAY;
    413     return;
    414   }
    415   if ((img->comps[0].dx == 1) && (img->comps[1].dx == 2) &&
    416       (img->comps[2].dx == 2) && (img->comps[0].dy == 1) &&
    417       (img->comps[1].dy == 2) && (img->comps[2].dy == 2)) {
    418     sycc420_to_rgb(img);
    419   } else if ((img->comps[0].dx == 1) && (img->comps[1].dx == 2) &&
    420              (img->comps[2].dx == 2) && (img->comps[0].dy == 1) &&
    421              (img->comps[1].dy == 1) && (img->comps[2].dy == 1)) {
    422     sycc422_to_rgb(img);
    423   } else if ((img->comps[0].dx == 1) && (img->comps[1].dx == 1) &&
    424              (img->comps[2].dx == 1) && (img->comps[0].dy == 1) &&
    425              (img->comps[1].dy == 1) && (img->comps[2].dy == 1)) {
    426     sycc444_to_rgb(img);
    427   } else {
    428     return;
    429   }
    430   img->color_space = OPJ_CLRSPC_SRGB;
    431 }
    432 void color_apply_icc_profile(opj_image_t* image) {
    433   cmsHPROFILE out_prof;
    434   cmsUInt32Number in_type;
    435   cmsUInt32Number out_type;
    436   int* r;
    437   int* g;
    438   int* b;
    439   int max;
    440   cmsHPROFILE in_prof =
    441       cmsOpenProfileFromMem(image->icc_profile_buf, image->icc_profile_len);
    442   if (!in_prof) {
    443     return;
    444   }
    445   cmsColorSpaceSignature out_space = cmsGetColorSpace(in_prof);
    446   cmsUInt32Number intent = cmsGetHeaderRenderingIntent(in_prof);
    447   int max_w = (int)image->comps[0].w;
    448   int max_h = (int)image->comps[0].h;
    449   int prec = (int)image->comps[0].prec;
    450   OPJ_COLOR_SPACE oldspace = image->color_space;
    451   if (out_space == cmsSigRgbData) {
    452     if (prec <= 8) {
    453       in_type = TYPE_RGB_8;
    454       out_type = TYPE_RGB_8;
    455     } else {
    456       in_type = TYPE_RGB_16;
    457       out_type = TYPE_RGB_16;
    458     }
    459     out_prof = cmsCreate_sRGBProfile();
    460     image->color_space = OPJ_CLRSPC_SRGB;
    461   } else if (out_space == cmsSigGrayData) {
    462     if (prec <= 8) {
    463       in_type = TYPE_GRAY_8;
    464       out_type = TYPE_RGB_8;
    465     } else {
    466       in_type = TYPE_GRAY_16;
    467       out_type = TYPE_RGB_16;
    468     }
    469     out_prof = cmsCreate_sRGBProfile();
    470     image->color_space = OPJ_CLRSPC_SRGB;
    471   } else if (out_space == cmsSigYCbCrData) {
    472     in_type = TYPE_YCbCr_16;
    473     out_type = TYPE_RGB_16;
    474     out_prof = cmsCreate_sRGBProfile();
    475     image->color_space = OPJ_CLRSPC_SRGB;
    476   } else {
    477     return;
    478   }
    479   cmsHTRANSFORM transform =
    480       cmsCreateTransform(in_prof, in_type, out_prof, out_type, intent, 0);
    481   cmsCloseProfile(in_prof);
    482   cmsCloseProfile(out_prof);
    483   if (!transform) {
    484     image->color_space = oldspace;
    485     return;
    486   }
    487   if (image->numcomps > 2) {
    488     if (prec <= 8) {
    489       unsigned char *inbuf, *outbuf, *in, *out;
    490       max = max_w * max_h;
    491       cmsUInt32Number nr_samples = max * 3 * sizeof(unsigned char);
    492       in = inbuf = FX_Alloc(unsigned char, nr_samples);
    493       out = outbuf = FX_Alloc(unsigned char, nr_samples);
    494       r = image->comps[0].data;
    495       g = image->comps[1].data;
    496       b = image->comps[2].data;
    497       for (int i = 0; i < max; ++i) {
    498         *in++ = (unsigned char)*r++;
    499         *in++ = (unsigned char)*g++;
    500         *in++ = (unsigned char)*b++;
    501       }
    502       cmsDoTransform(transform, inbuf, outbuf, (cmsUInt32Number)max);
    503       r = image->comps[0].data;
    504       g = image->comps[1].data;
    505       b = image->comps[2].data;
    506       for (int i = 0; i < max; ++i) {
    507         *r++ = (int)*out++;
    508         *g++ = (int)*out++;
    509         *b++ = (int)*out++;
    510       }
    511       FX_Free(inbuf);
    512       FX_Free(outbuf);
    513     } else {
    514       unsigned short *inbuf, *outbuf, *in, *out;
    515       max = max_w * max_h;
    516       cmsUInt32Number nr_samples = max * 3 * sizeof(unsigned short);
    517       in = inbuf = FX_Alloc(unsigned short, nr_samples);
    518       out = outbuf = FX_Alloc(unsigned short, nr_samples);
    519       r = image->comps[0].data;
    520       g = image->comps[1].data;
    521       b = image->comps[2].data;
    522       for (int i = 0; i < max; ++i) {
    523         *in++ = (unsigned short)*r++;
    524         *in++ = (unsigned short)*g++;
    525         *in++ = (unsigned short)*b++;
    526       }
    527       cmsDoTransform(transform, inbuf, outbuf, (cmsUInt32Number)max);
    528       r = image->comps[0].data;
    529       g = image->comps[1].data;
    530       b = image->comps[2].data;
    531       for (int i = 0; i < max; ++i) {
    532         *r++ = (int)*out++;
    533         *g++ = (int)*out++;
    534         *b++ = (int)*out++;
    535       }
    536       FX_Free(inbuf);
    537       FX_Free(outbuf);
    538     }
    539   } else {
    540     unsigned char *in, *inbuf, *out, *outbuf;
    541     max = max_w * max_h;
    542     cmsUInt32Number nr_samples =
    543         (cmsUInt32Number)max * 3 * sizeof(unsigned char);
    544     in = inbuf = FX_Alloc(unsigned char, nr_samples);
    545     out = outbuf = FX_Alloc(unsigned char, nr_samples);
    546     image->comps = (opj_image_comp_t*)realloc(
    547         image->comps, (image->numcomps + 2) * sizeof(opj_image_comp_t));
    548     if (image->numcomps == 2) {
    549       image->comps[3] = image->comps[1];
    550     }
    551     image->comps[1] = image->comps[0];
    552     image->comps[2] = image->comps[0];
    553     image->comps[1].data = FX_Alloc(int, (size_t)max);
    554     FXSYS_memset(image->comps[1].data, 0, sizeof(int) * (size_t)max);
    555     image->comps[2].data = FX_Alloc(int, (size_t)max);
    556     FXSYS_memset(image->comps[2].data, 0, sizeof(int) * (size_t)max);
    557     image->numcomps += 2;
    558     r = image->comps[0].data;
    559     for (int i = 0; i < max; ++i) {
    560       *in++ = (unsigned char)*r++;
    561     }
    562     cmsDoTransform(transform, inbuf, outbuf, (cmsUInt32Number)max);
    563     r = image->comps[0].data;
    564     g = image->comps[1].data;
    565     b = image->comps[2].data;
    566     for (int i = 0; i < max; ++i) {
    567       *r++ = (int)*out++;
    568       *g++ = (int)*out++;
    569       *b++ = (int)*out++;
    570     }
    571     FX_Free(inbuf);
    572     FX_Free(outbuf);
    573   }
    574   cmsDeleteTransform(transform);
    575 }
    576 void color_apply_conversion(opj_image_t* image) {
    577   int* row;
    578   int enumcs, numcomps;
    579   numcomps = image->numcomps;
    580   if (numcomps < 3) {
    581     return;
    582   }
    583   row = (int*)image->icc_profile_buf;
    584   enumcs = row[0];
    585   if (enumcs == 14) {
    586     int *L, *a, *b, *red, *green, *blue, *src0, *src1, *src2;
    587     double rl, ol, ra, oa, rb, ob, prec0, prec1, prec2;
    588     double minL, maxL, mina, maxa, minb, maxb;
    589     unsigned int default_type;
    590     unsigned int i, max;
    591     cmsHPROFILE in, out;
    592     cmsHTRANSFORM transform;
    593     cmsUInt16Number RGB[3];
    594     cmsCIELab Lab;
    595     in = cmsCreateLab4Profile(NULL);
    596     out = cmsCreate_sRGBProfile();
    597     transform = cmsCreateTransform(in, TYPE_Lab_DBL, out, TYPE_RGB_16,
    598                                    INTENT_PERCEPTUAL, 0);
    599     cmsCloseProfile(in);
    600     cmsCloseProfile(out);
    601     if (!transform) {
    602       return;
    603     }
    604     prec0 = (double)image->comps[0].prec;
    605     prec1 = (double)image->comps[1].prec;
    606     prec2 = (double)image->comps[2].prec;
    607     default_type = row[1];
    608     if (default_type == 0x44454600) {
    609       rl = 100;
    610       ra = 170;
    611       rb = 200;
    612       ol = 0;
    613       oa = pow(2, prec1 - 1);
    614       ob = pow(2, prec2 - 2) + pow(2, prec2 - 3);
    615     } else {
    616       rl = row[2];
    617       ra = row[4];
    618       rb = row[6];
    619       ol = row[3];
    620       oa = row[5];
    621       ob = row[7];
    622     }
    623     L = src0 = image->comps[0].data;
    624     a = src1 = image->comps[1].data;
    625     b = src2 = image->comps[2].data;
    626     max = image->comps[0].w * image->comps[0].h;
    627     red = FX_Alloc(int, max);
    628     image->comps[0].data = red;
    629     green = FX_Alloc(int, max);
    630     image->comps[1].data = green;
    631     blue = FX_Alloc(int, max);
    632     image->comps[2].data = blue;
    633     minL = -(rl * ol) / (pow(2, prec0) - 1);
    634     maxL = minL + rl;
    635     mina = -(ra * oa) / (pow(2, prec1) - 1);
    636     maxa = mina + ra;
    637     minb = -(rb * ob) / (pow(2, prec2) - 1);
    638     maxb = minb + rb;
    639     for (i = 0; i < max; ++i) {
    640       Lab.L = minL + (double)(*L) * (maxL - minL) / (pow(2, prec0) - 1);
    641       ++L;
    642       Lab.a = mina + (double)(*a) * (maxa - mina) / (pow(2, prec1) - 1);
    643       ++a;
    644       Lab.b = minb + (double)(*b) * (maxb - minb) / (pow(2, prec2) - 1);
    645       ++b;
    646       cmsDoTransform(transform, &Lab, RGB, 1);
    647       *red++ = RGB[0];
    648       *green++ = RGB[1];
    649       *blue++ = RGB[2];
    650     }
    651     cmsDeleteTransform(transform);
    652     FX_Free(src0);
    653     FX_Free(src1);
    654     FX_Free(src2);
    655     image->color_space = OPJ_CLRSPC_SRGB;
    656     image->comps[0].prec = 16;
    657     image->comps[1].prec = 16;
    658     image->comps[2].prec = 16;
    659     return;
    660   }
    661 }
    662 class CJPX_Decoder {
    663  public:
    664   explicit CJPX_Decoder(CPDF_ColorSpace* cs);
    665   ~CJPX_Decoder();
    666   FX_BOOL Init(const unsigned char* src_data, FX_DWORD src_size);
    667   void GetInfo(FX_DWORD* width, FX_DWORD* height, FX_DWORD* components);
    668   bool Decode(uint8_t* dest_buf,
    669               int pitch,
    670               const std::vector<uint8_t>& offsets);
    671 
    672  private:
    673   const uint8_t* m_SrcData;
    674   FX_DWORD m_SrcSize;
    675   opj_image_t* image;
    676   opj_codec_t* l_codec;
    677   opj_stream_t* l_stream;
    678   const CPDF_ColorSpace* const m_ColorSpace;
    679 };
    680 
    681 CJPX_Decoder::CJPX_Decoder(CPDF_ColorSpace* cs)
    682     : image(nullptr), l_codec(nullptr), l_stream(nullptr), m_ColorSpace(cs) {}
    683 
    684 CJPX_Decoder::~CJPX_Decoder() {
    685   if (l_codec) {
    686     opj_destroy_codec(l_codec);
    687   }
    688   if (l_stream) {
    689     opj_stream_destroy(l_stream);
    690   }
    691   if (image) {
    692     opj_image_destroy(image);
    693   }
    694 }
    695 
    696 FX_BOOL CJPX_Decoder::Init(const unsigned char* src_data, FX_DWORD src_size) {
    697   static const unsigned char szJP2Header[] = {
    698       0x00, 0x00, 0x00, 0x0c, 0x6a, 0x50, 0x20, 0x20, 0x0d, 0x0a, 0x87, 0x0a};
    699   if (!src_data || src_size < sizeof(szJP2Header))
    700     return FALSE;
    701 
    702   image = NULL;
    703   m_SrcData = src_data;
    704   m_SrcSize = src_size;
    705   DecodeData srcData(const_cast<unsigned char*>(src_data), src_size);
    706   l_stream = fx_opj_stream_create_memory_stream(&srcData,
    707                                                 OPJ_J2K_STREAM_CHUNK_SIZE, 1);
    708   if (!l_stream) {
    709     return FALSE;
    710   }
    711   opj_dparameters_t parameters;
    712   opj_set_default_decoder_parameters(&parameters);
    713   parameters.decod_format = 0;
    714   parameters.cod_format = 3;
    715   if (FXSYS_memcmp(m_SrcData, szJP2Header, sizeof(szJP2Header)) == 0) {
    716     l_codec = opj_create_decompress(OPJ_CODEC_JP2);
    717     parameters.decod_format = 1;
    718   } else {
    719     l_codec = opj_create_decompress(OPJ_CODEC_J2K);
    720   }
    721   if (!l_codec) {
    722     return FALSE;
    723   }
    724   if (m_ColorSpace && m_ColorSpace->GetFamily() == PDFCS_INDEXED)
    725     parameters.flags |= OPJ_DPARAMETERS_IGNORE_PCLR_CMAP_CDEF_FLAG;
    726   opj_set_info_handler(l_codec, fx_info_callback, 00);
    727   opj_set_warning_handler(l_codec, fx_warning_callback, 00);
    728   opj_set_error_handler(l_codec, fx_error_callback, 00);
    729   if (!opj_setup_decoder(l_codec, &parameters)) {
    730     return FALSE;
    731   }
    732   if (!opj_read_header(l_stream, l_codec, &image)) {
    733     image = NULL;
    734     return FALSE;
    735   }
    736   image->pdfium_use_colorspace = !!m_ColorSpace;
    737 
    738   if (!parameters.nb_tile_to_decode) {
    739     if (!opj_set_decode_area(l_codec, image, parameters.DA_x0, parameters.DA_y0,
    740                              parameters.DA_x1, parameters.DA_y1)) {
    741       opj_image_destroy(image);
    742       image = NULL;
    743       return FALSE;
    744     }
    745     if (!(opj_decode(l_codec, l_stream, image) &&
    746           opj_end_decompress(l_codec, l_stream))) {
    747       opj_image_destroy(image);
    748       image = NULL;
    749       return FALSE;
    750     }
    751   } else {
    752     if (!opj_get_decoded_tile(l_codec, l_stream, image,
    753                               parameters.tile_index)) {
    754       return FALSE;
    755     }
    756   }
    757   opj_stream_destroy(l_stream);
    758   l_stream = NULL;
    759   if (image->color_space != OPJ_CLRSPC_SYCC && image->numcomps == 3 &&
    760       image->comps[0].dx == image->comps[0].dy && image->comps[1].dx != 1) {
    761     image->color_space = OPJ_CLRSPC_SYCC;
    762   } else if (image->numcomps <= 2) {
    763     image->color_space = OPJ_CLRSPC_GRAY;
    764   }
    765   if (image->color_space == OPJ_CLRSPC_SYCC) {
    766     color_sycc_to_rgb(image);
    767   }
    768   if (image->icc_profile_buf) {
    769     FX_Free(image->icc_profile_buf);
    770     image->icc_profile_buf = NULL;
    771     image->icc_profile_len = 0;
    772   }
    773   if (!image) {
    774     return FALSE;
    775   }
    776   return TRUE;
    777 }
    778 
    779 void CJPX_Decoder::GetInfo(FX_DWORD* width,
    780                            FX_DWORD* height,
    781                            FX_DWORD* components) {
    782   *width = (FX_DWORD)image->x1;
    783   *height = (FX_DWORD)image->y1;
    784   *components = (FX_DWORD)image->numcomps;
    785 }
    786 
    787 bool CJPX_Decoder::Decode(uint8_t* dest_buf,
    788                           int pitch,
    789                           const std::vector<uint8_t>& offsets) {
    790   if (image->comps[0].w != image->x1 || image->comps[0].h != image->y1)
    791     return false;
    792 
    793   if (pitch<(int)(image->comps[0].w * 8 * image->numcomps + 31)>> 5 << 2)
    794     return false;
    795 
    796   FXSYS_memset(dest_buf, 0xff, image->y1 * pitch);
    797   std::vector<uint8_t*> channel_bufs(image->numcomps);
    798   std::vector<int> adjust_comps(image->numcomps);
    799   for (uint32_t i = 0; i < image->numcomps; i++) {
    800     channel_bufs[i] = dest_buf + offsets[i];
    801     adjust_comps[i] = image->comps[i].prec - 8;
    802     if (i > 0) {
    803       if (image->comps[i].dx != image->comps[i - 1].dx ||
    804           image->comps[i].dy != image->comps[i - 1].dy ||
    805           image->comps[i].prec != image->comps[i - 1].prec) {
    806         return false;
    807       }
    808     }
    809   }
    810   int width = image->comps[0].w;
    811   int height = image->comps[0].h;
    812   for (uint32_t channel = 0; channel < image->numcomps; ++channel) {
    813     uint8_t* pChannel = channel_bufs[channel];
    814     if (adjust_comps[channel] < 0) {
    815       for (int row = 0; row < height; ++row) {
    816         uint8_t* pScanline = pChannel + row * pitch;
    817         for (int col = 0; col < width; ++col) {
    818           uint8_t* pPixel = pScanline + col * image->numcomps;
    819           int src = image->comps[channel].data[row * width + col];
    820           src += image->comps[channel].sgnd
    821                      ? 1 << (image->comps[channel].prec - 1)
    822                      : 0;
    823           if (adjust_comps[channel] > 0) {
    824             *pPixel = 0;
    825           } else {
    826             *pPixel = (uint8_t)(src << -adjust_comps[channel]);
    827           }
    828         }
    829       }
    830     } else {
    831       for (int row = 0; row < height; ++row) {
    832         uint8_t* pScanline = pChannel + row * pitch;
    833         for (int col = 0; col < width; ++col) {
    834           uint8_t* pPixel = pScanline + col * image->numcomps;
    835           if (!image->comps[channel].data) {
    836             continue;
    837           }
    838           int src = image->comps[channel].data[row * width + col];
    839           src += image->comps[channel].sgnd
    840                      ? 1 << (image->comps[channel].prec - 1)
    841                      : 0;
    842           if (adjust_comps[channel] - 1 < 0) {
    843             *pPixel = (uint8_t)((src >> adjust_comps[channel]));
    844           } else {
    845             int tmpPixel = (src >> adjust_comps[channel]) +
    846                            ((src >> (adjust_comps[channel] - 1)) % 2);
    847             if (tmpPixel > 255) {
    848               tmpPixel = 255;
    849             } else if (tmpPixel < 0) {
    850               tmpPixel = 0;
    851             }
    852             *pPixel = (uint8_t)tmpPixel;
    853           }
    854         }
    855       }
    856     }
    857   }
    858   return true;
    859 }
    860 
    861 CCodec_JpxModule::CCodec_JpxModule() {}
    862 CCodec_JpxModule::~CCodec_JpxModule() {
    863 }
    864 
    865 CJPX_Decoder* CCodec_JpxModule::CreateDecoder(const uint8_t* src_buf,
    866                                               FX_DWORD src_size,
    867                                               CPDF_ColorSpace* cs) {
    868   std::unique_ptr<CJPX_Decoder> decoder(new CJPX_Decoder(cs));
    869   return decoder->Init(src_buf, src_size) ? decoder.release() : nullptr;
    870 }
    871 
    872 void CCodec_JpxModule::GetImageInfo(CJPX_Decoder* pDecoder,
    873                                     FX_DWORD* width,
    874                                     FX_DWORD* height,
    875                                     FX_DWORD* components) {
    876   pDecoder->GetInfo(width, height, components);
    877 }
    878 
    879 bool CCodec_JpxModule::Decode(CJPX_Decoder* pDecoder,
    880                               uint8_t* dest_data,
    881                               int pitch,
    882                               const std::vector<uint8_t>& offsets) {
    883   return pDecoder->Decode(dest_data, pitch, offsets);
    884 }
    885 
    886 void CCodec_JpxModule::DestroyDecoder(CJPX_Decoder* pDecoder) {
    887   delete pDecoder;
    888 }
    889