Remove trailing whitespaces in core.
[pdfium.git] / core / src / fpdfapi / fpdf_render / fpdf_render_loadimage.cpp
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 "../../../../third_party/base/nonstd_unique_ptr.h"
8 #include "../../../include/fpdfapi/fpdf_module.h"
9 #include "../../../include/fpdfapi/fpdf_pageobj.h"
10 #include "../../../include/fpdfapi/fpdf_render.h"
11 #include "../../../include/fxcodec/fx_codec.h"
12 #include "../../../include/fxcrt/fx_safe_types.h"
13 #include "../../../include/fxge/fx_ge.h"
14 #include "../fpdf_page/pageint.h"
15 #include "render_int.h"
16
17 namespace {
18
19 unsigned int _GetBits8(const uint8_t* pData, int bitpos, int nbits)
20 {
21     unsigned int byte = pData[bitpos / 8];
22     if (nbits == 8) {
23         return byte;
24     } else if (nbits == 4) {
25         return (bitpos % 8) ? (byte & 0x0f) : (byte >> 4);
26     } else if (nbits == 2) {
27         return (byte >> (6 - bitpos % 8)) & 0x03;
28     } else if (nbits == 1) {
29         return (byte >> (7 - bitpos % 8)) & 0x01;
30     } else if (nbits == 16) {
31         return byte * 256 + pData[bitpos / 8 + 1];
32     }
33     return 0;
34 }
35
36 FX_SAFE_DWORD CalculatePitch8(FX_DWORD bpc,
37                               FX_DWORD components,
38                               int width,
39                               int height)
40 {
41     FX_SAFE_DWORD pitch = bpc;
42     pitch *= components;
43     pitch *= width;
44     pitch += 7;
45     pitch /= 8;
46     pitch *= height;
47     return pitch;
48 }
49
50 FX_SAFE_DWORD CalculatePitch32(int bpp, int width)
51 {
52     FX_SAFE_DWORD pitch = bpp;
53     pitch *= width;
54     pitch += 31;
55     pitch /= 8;
56     return pitch;
57 }
58
59 // Wrapper class to hold objects allocated in CPDF_DIBSource::LoadJpxBitmap(),
60 // because nonstd::unique_ptr does not support custom deleters yet.
61 class JpxBitMapContext
62 {
63   public:
64     explicit JpxBitMapContext(ICodec_JpxModule* jpx_module)
65         : jpx_module_(jpx_module),
66           ctx_(nullptr),
67           output_offsets_(nullptr) {}
68
69     ~JpxBitMapContext() {
70         FX_Free(output_offsets_);
71         jpx_module_->DestroyDecoder(ctx_);
72     }
73
74     // Takes ownership of |ctx|.
75     void set_context(void* ctx) {
76         ctx_ = ctx;
77     }
78
79     void* context() {
80         return ctx_;
81     }
82
83     // Takes ownership of |output_offsets|.
84     void set_output_offsets(unsigned char* output_offsets) {
85       output_offsets_ = output_offsets;
86     }
87
88     unsigned char* output_offsets() {
89       return output_offsets_;
90     }
91
92   private:
93     ICodec_JpxModule* jpx_module_;  // Weak pointer.
94     void* ctx_;  // Decoder context, owned.
95     unsigned char* output_offsets_;  // Output offsets for decoding, owned.
96
97     // Disallow evil constructors
98     JpxBitMapContext(const JpxBitMapContext&);
99     void operator=(const JpxBitMapContext&);
100 };
101
102 }  // namespace
103
104 CFX_DIBSource* CPDF_Image::LoadDIBSource(CFX_DIBSource** ppMask, FX_DWORD* pMatteColor, FX_BOOL bStdCS, FX_DWORD GroupFamily, FX_BOOL bLoadMask) const
105 {
106     CPDF_DIBSource* pSource = new CPDF_DIBSource;
107     if (pSource->Load(m_pDocument, m_pStream, (CPDF_DIBSource**)ppMask, pMatteColor, NULL, NULL, bStdCS, GroupFamily, bLoadMask)) {
108         return pSource;
109     }
110     delete pSource;
111     return NULL;
112 }
113 CFX_DIBSource* CPDF_Image::DetachBitmap()
114 {
115     CFX_DIBSource* pBitmap = m_pDIBSource;
116     m_pDIBSource = NULL;
117     return pBitmap;
118 }
119 CFX_DIBSource* CPDF_Image::DetachMask()
120 {
121     CFX_DIBSource* pBitmap = m_pMask;
122     m_pMask = NULL;
123     return pBitmap;
124 }
125 FX_BOOL CPDF_Image::StartLoadDIBSource(CPDF_Dictionary* pFormResource, CPDF_Dictionary* pPageResource, FX_BOOL bStdCS, FX_DWORD GroupFamily, FX_BOOL bLoadMask)
126 {
127     m_pDIBSource = new CPDF_DIBSource;
128     int ret = ((CPDF_DIBSource*)m_pDIBSource)->StartLoadDIBSource(m_pDocument, m_pStream, TRUE, pFormResource, pPageResource, bStdCS, GroupFamily, bLoadMask);
129     if (ret == 2) {
130         return TRUE;
131     }
132     if (!ret) {
133         delete m_pDIBSource;
134         m_pDIBSource = NULL;
135         return FALSE;
136     }
137     m_pMask = ((CPDF_DIBSource*)m_pDIBSource)->DetachMask();
138     m_MatteColor = ((CPDF_DIBSource*)m_pDIBSource)->m_MatteColor;
139     return FALSE;
140 }
141 FX_BOOL CPDF_Image::Continue(IFX_Pause* pPause)
142 {
143     int ret = ((CPDF_DIBSource*)m_pDIBSource)->ContinueLoadDIBSource(pPause);
144     if (ret == 2) {
145         return TRUE;
146     }
147     if (!ret) {
148         delete m_pDIBSource;
149         m_pDIBSource = NULL;
150         return FALSE;
151     }
152     m_pMask = ((CPDF_DIBSource*)m_pDIBSource)->DetachMask();
153     m_MatteColor = ((CPDF_DIBSource*)m_pDIBSource)->m_MatteColor;
154     return FALSE;
155 }
156 CPDF_DIBSource::CPDF_DIBSource()
157 {
158     m_pDocument = NULL;
159     m_pStreamAcc = NULL;
160     m_pDict = NULL;
161     m_bpp = 0;
162     m_Width = m_Height = 0;
163     m_pColorSpace = NULL;
164     m_bDefaultDecode = TRUE;
165     m_bImageMask = FALSE;
166     m_bDoBpcCheck = TRUE;
167     m_pPalette = NULL;
168     m_pCompData = NULL;
169     m_bColorKey = FALSE;
170     m_pMaskedLine = m_pLineBuf = NULL;
171     m_pDecoder = NULL;
172     m_nComponents = 0;
173     m_bpc = 0;
174     m_bLoadMask = FALSE;
175     m_Family = 0;
176     m_pMask = NULL;
177     m_MatteColor = 0;
178     m_pJbig2Context = NULL;
179     m_pGlobalStream = NULL;
180     m_bStdCS = FALSE;
181     m_pMaskStream = NULL;
182     m_Status = 0;
183     m_bHasMask = FALSE;
184 }
185 CPDF_DIBSource::~CPDF_DIBSource()
186 {
187     delete m_pStreamAcc;
188     if (m_pMaskedLine) {
189         FX_Free(m_pMaskedLine);
190     }
191     if (m_pLineBuf) {
192         FX_Free(m_pLineBuf);
193     }
194     m_pCachedBitmap.reset();
195     delete m_pDecoder;
196     if (m_pCompData) {
197         FX_Free(m_pCompData);
198     }
199     CPDF_ColorSpace* pCS = m_pColorSpace;
200     if (pCS && m_pDocument) {
201         m_pDocument->GetPageData()->ReleaseColorSpace(pCS->GetArray());
202     }
203     if (m_pJbig2Context) {
204         ICodec_Jbig2Module* pJbig2Module = CPDF_ModuleMgr::Get()->GetJbig2Module();
205         pJbig2Module->DestroyJbig2Context(m_pJbig2Context);
206     }
207     delete m_pGlobalStream;
208 }
209 CFX_DIBitmap* CPDF_DIBSource::GetBitmap() const
210 {
211     return m_pCachedBitmap ? m_pCachedBitmap.get() : Clone();
212 }
213 void CPDF_DIBSource::ReleaseBitmap(CFX_DIBitmap* pBitmap) const
214 {
215     if (pBitmap && pBitmap != m_pCachedBitmap) {
216         delete pBitmap;
217     }
218 }
219 FX_BOOL CPDF_DIBSource::Load(CPDF_Document* pDoc, const CPDF_Stream* pStream, CPDF_DIBSource** ppMask,
220                              FX_DWORD* pMatteColor, CPDF_Dictionary* pFormResources, CPDF_Dictionary* pPageResources, FX_BOOL bStdCS, FX_DWORD GroupFamily, FX_BOOL bLoadMask)
221 {
222     if (pStream == NULL) {
223         return FALSE;
224     }
225     m_pDocument = pDoc;
226     m_pDict = pStream->GetDict();
227     if (m_pDict == NULL) {
228         return FALSE;
229     }
230     m_pStream = pStream;
231     m_Width = m_pDict->GetInteger(FX_BSTRC("Width"));
232     m_Height = m_pDict->GetInteger(FX_BSTRC("Height"));
233     if (m_Width <= 0 || m_Height <= 0 || m_Width > 0x01ffff || m_Height > 0x01ffff) {
234         return FALSE;
235     }
236     m_GroupFamily = GroupFamily;
237     m_bLoadMask = bLoadMask;
238     if (!LoadColorInfo(m_pStream->GetObjNum() != 0 ? NULL : pFormResources, pPageResources)) {
239         return FALSE;
240     }
241     if (m_bDoBpcCheck && (m_bpc == 0 || m_nComponents == 0)) {
242         return FALSE;
243     }
244     FX_SAFE_DWORD src_pitch =
245         CalculatePitch8(m_bpc, m_nComponents, m_Width, m_Height);
246     if (!src_pitch.IsValid()) {
247         return FALSE;
248     }
249     m_pStreamAcc = new CPDF_StreamAcc;
250     m_pStreamAcc->LoadAllData(pStream, FALSE, src_pitch.ValueOrDie(), TRUE);
251     if (m_pStreamAcc->GetSize() == 0 || m_pStreamAcc->GetData() == NULL) {
252         return FALSE;
253     }
254     if (!CreateDecoder()) {
255         return FALSE;
256     }
257     if (m_bImageMask) {
258         m_bpp = 1;
259         m_bpc = 1;
260         m_nComponents = 1;
261         m_AlphaFlag = 1;
262     } else if (m_bpc * m_nComponents == 1) {
263         m_bpp = 1;
264     } else if (m_bpc * m_nComponents <= 8) {
265         m_bpp = 8;
266     } else {
267         m_bpp = 24;
268     }
269     FX_SAFE_DWORD pitch = CalculatePitch32(m_bpp, m_Width);
270     if (!pitch.IsValid()) {
271         return FALSE;
272     }
273     m_pLineBuf = FX_Alloc(uint8_t, pitch.ValueOrDie());
274     if (m_pColorSpace && bStdCS) {
275         m_pColorSpace->EnableStdConversion(TRUE);
276     }
277     LoadPalette();
278     if (m_bColorKey) {
279         m_bpp = 32;
280         m_AlphaFlag = 2;
281         pitch = CalculatePitch32(m_bpp, m_Width);
282         if (!pitch.IsValid()) {
283             return FALSE;
284         }
285         m_pMaskedLine = FX_Alloc(uint8_t, pitch.ValueOrDie());
286     }
287     m_Pitch = pitch.ValueOrDie();
288     if (ppMask) {
289         *ppMask = LoadMask(*pMatteColor);
290     }
291     if (m_pColorSpace && bStdCS) {
292         m_pColorSpace->EnableStdConversion(FALSE);
293     }
294     return TRUE;
295 }
296 int     CPDF_DIBSource::ContinueToLoadMask()
297 {
298     if (m_bImageMask) {
299         m_bpp = 1;
300         m_bpc = 1;
301         m_nComponents = 1;
302         m_AlphaFlag = 1;
303     } else if (m_bpc * m_nComponents == 1) {
304         m_bpp = 1;
305     } else if (m_bpc * m_nComponents <= 8) {
306         m_bpp = 8;
307     } else {
308         m_bpp = 24;
309     }
310     if (!m_bpc || !m_nComponents) {
311         return 0;
312     }
313     FX_SAFE_DWORD pitch = CalculatePitch32(m_bpp, m_Width);
314     if (!pitch.IsValid()) {
315         return 0;
316     }
317     m_pLineBuf = FX_Alloc(uint8_t, pitch.ValueOrDie());
318     if (m_pColorSpace && m_bStdCS) {
319         m_pColorSpace->EnableStdConversion(TRUE);
320     }
321     LoadPalette();
322     if (m_bColorKey) {
323         m_bpp = 32;
324         m_AlphaFlag = 2;
325         pitch = CalculatePitch32(m_bpp, m_Width);
326         if (!pitch.IsValid()) {
327             return 0;
328         }
329         m_pMaskedLine = FX_Alloc(uint8_t, pitch.ValueOrDie());
330     }
331     m_Pitch = pitch.ValueOrDie();
332     return 1;
333 }
334 int     CPDF_DIBSource::StartLoadDIBSource(CPDF_Document* pDoc, const CPDF_Stream* pStream, FX_BOOL bHasMask,
335                                        CPDF_Dictionary* pFormResources, CPDF_Dictionary* pPageResources,
336                                        FX_BOOL bStdCS, FX_DWORD GroupFamily, FX_BOOL bLoadMask)
337 {
338     if (pStream == NULL) {
339         return 0;
340     }
341     m_pDocument = pDoc;
342     m_pDict = pStream->GetDict();
343     m_pStream = pStream;
344     m_bStdCS = bStdCS;
345     m_bHasMask = bHasMask;
346     m_Width = m_pDict->GetInteger(FX_BSTRC("Width"));
347     m_Height = m_pDict->GetInteger(FX_BSTRC("Height"));
348     if (m_Width <= 0 || m_Height <= 0 || m_Width > 0x01ffff || m_Height > 0x01ffff) {
349         return 0;
350     }
351     m_GroupFamily = GroupFamily;
352     m_bLoadMask = bLoadMask;
353     if (!LoadColorInfo(m_pStream->GetObjNum() != 0 ? NULL : pFormResources, pPageResources)) {
354         return 0;
355     }
356     if (m_bDoBpcCheck && (m_bpc == 0 || m_nComponents == 0)) {
357         return 0;
358     }
359     FX_SAFE_DWORD src_pitch =
360         CalculatePitch8(m_bpc, m_nComponents, m_Width, m_Height);
361     if (!src_pitch.IsValid()) {
362         return 0;
363     }
364     m_pStreamAcc = new CPDF_StreamAcc;
365     m_pStreamAcc->LoadAllData(pStream, FALSE, src_pitch.ValueOrDie(), TRUE);
366     if (m_pStreamAcc->GetSize() == 0 || m_pStreamAcc->GetData() == NULL) {
367         return 0;
368     }
369     int ret = CreateDecoder();
370     if (ret != 1) {
371         if (!ret) {
372             return ret;
373         }
374         if (!ContinueToLoadMask()) {
375             return 0;
376         }
377         if (m_bHasMask) {
378             StratLoadMask();
379         }
380         return ret;
381     }
382     if (!ContinueToLoadMask()) {
383         return 0;
384     }
385     if (m_bHasMask) {
386         ret = StratLoadMask();
387     }
388     if (ret == 2) {
389         return ret;
390     }
391     if (m_pColorSpace && m_bStdCS) {
392         m_pColorSpace->EnableStdConversion(FALSE);
393     }
394     return ret;
395 }
396 int     CPDF_DIBSource::ContinueLoadDIBSource(IFX_Pause* pPause)
397 {
398     FXCODEC_STATUS ret;
399     if (m_Status == 1) {
400         const CFX_ByteString& decoder = m_pStreamAcc->GetImageDecoder();
401         if (decoder == FX_BSTRC("JPXDecode")) {
402             return 0;
403         }
404         ICodec_Jbig2Module* pJbig2Module = CPDF_ModuleMgr::Get()->GetJbig2Module();
405         if (m_pJbig2Context == NULL) {
406             m_pJbig2Context = pJbig2Module->CreateJbig2Context();
407             if (m_pStreamAcc->GetImageParam()) {
408                 CPDF_Stream* pGlobals = m_pStreamAcc->GetImageParam()->GetStream(FX_BSTRC("JBIG2Globals"));
409                 if (pGlobals) {
410                     m_pGlobalStream = new CPDF_StreamAcc;
411                     m_pGlobalStream->LoadAllData(pGlobals, FALSE);
412                 }
413             }
414             ret = pJbig2Module->StartDecode(m_pJbig2Context, m_Width, m_Height, m_pStreamAcc->GetData(), m_pStreamAcc->GetSize(),
415                                             m_pGlobalStream ? m_pGlobalStream->GetData() : NULL, m_pGlobalStream ? m_pGlobalStream->GetSize() : 0, m_pCachedBitmap->GetBuffer(),
416                                             m_pCachedBitmap->GetPitch(), pPause);
417             if (ret < 0) {
418                 m_pCachedBitmap.reset();
419                 delete m_pGlobalStream;
420                 m_pGlobalStream = NULL;
421                 pJbig2Module->DestroyJbig2Context(m_pJbig2Context);
422                 m_pJbig2Context = NULL;
423                 return 0;
424             }
425             if (ret == FXCODEC_STATUS_DECODE_TOBECONTINUE) {
426                 return 2;
427             }
428             int ret1 = 1;
429             if (m_bHasMask) {
430                 ret1 = ContinueLoadMaskDIB(pPause);
431                 m_Status = 2;
432             }
433             if (ret1 == 2) {
434                 return ret1;
435             }
436             if (m_pColorSpace && m_bStdCS) {
437                 m_pColorSpace->EnableStdConversion(FALSE);
438             }
439             return ret1;
440         }
441         FXCODEC_STATUS ret = pJbig2Module->ContinueDecode(m_pJbig2Context, pPause);
442         if (ret < 0) {
443             m_pCachedBitmap.reset();
444             delete m_pGlobalStream;
445             m_pGlobalStream = NULL;
446             pJbig2Module->DestroyJbig2Context(m_pJbig2Context);
447             m_pJbig2Context = NULL;
448             return 0;
449         }
450         if (ret == FXCODEC_STATUS_DECODE_TOBECONTINUE) {
451             return 2;
452         }
453         int ret1 = 1;
454         if (m_bHasMask) {
455             ret1 = ContinueLoadMaskDIB(pPause);
456             m_Status = 2;
457         }
458         if (ret1 == 2) {
459             return ret1;
460         }
461         if (m_pColorSpace && m_bStdCS) {
462             m_pColorSpace->EnableStdConversion(FALSE);
463         }
464         return ret1;
465     } else if (m_Status == 2) {
466         return ContinueLoadMaskDIB(pPause);
467     }
468     return 0;
469 }
470 FX_BOOL CPDF_DIBSource::LoadColorInfo(CPDF_Dictionary* pFormResources, CPDF_Dictionary* pPageResources)
471 {
472     m_bpc_orig = m_pDict->GetInteger(FX_BSTRC("BitsPerComponent"));
473     if (m_pDict->GetInteger("ImageMask")) {
474         m_bImageMask = TRUE;
475     }
476     if (m_bImageMask || !m_pDict->KeyExist(FX_BSTRC("ColorSpace"))) {
477         if (!m_bImageMask) {
478             CPDF_Object* pFilter = m_pDict->GetElementValue(FX_BSTRC("Filter"));
479             if (pFilter) {
480                 CFX_ByteString filter;
481                 if (pFilter->GetType() == PDFOBJ_NAME) {
482                     filter = pFilter->GetString();
483                     if (filter == FX_BSTRC("JPXDecode")) {
484                         m_bDoBpcCheck = FALSE;
485                         return TRUE;
486                     }
487                 } else if (pFilter->GetType() == PDFOBJ_ARRAY) {
488                     CPDF_Array* pArray = (CPDF_Array*)pFilter;
489                     if (pArray->GetString(pArray->GetCount() - 1) == FX_BSTRC("JPXDecode")) {
490                         m_bDoBpcCheck = FALSE;
491                         return TRUE;
492                     }
493                 }
494             }
495         }
496         m_bImageMask = TRUE;
497         m_bpc = m_nComponents = 1;
498         CPDF_Array* pDecode = m_pDict->GetArray(FX_BSTRC("Decode"));
499         m_bDefaultDecode = pDecode == NULL || pDecode->GetInteger(0) == 0;
500         return TRUE;
501     }
502     CPDF_Object* pCSObj = m_pDict->GetElementValue(FX_BSTRC("ColorSpace"));
503     if (pCSObj == NULL) {
504         return FALSE;
505     }
506     CPDF_DocPageData* pDocPageData = m_pDocument->GetPageData();
507     if (pFormResources) {
508         m_pColorSpace = pDocPageData->GetColorSpace(pCSObj, pFormResources);
509     }
510     if (m_pColorSpace == NULL) {
511         m_pColorSpace = pDocPageData->GetColorSpace(pCSObj, pPageResources);
512     }
513     if (m_pColorSpace == NULL) {
514         return FALSE;
515     }
516     m_Family = m_pColorSpace->GetFamily();
517     m_nComponents = m_pColorSpace->CountComponents();
518     if (m_Family == PDFCS_ICCBASED && pCSObj->GetType() == PDFOBJ_NAME) {
519         CFX_ByteString cs = pCSObj->GetString();
520         if (cs == FX_BSTRC("DeviceGray")) {
521             m_nComponents = 1;
522         } else if (cs == FX_BSTRC("DeviceRGB")) {
523             m_nComponents = 3;
524         } else if (cs == FX_BSTRC("DeviceCMYK")) {
525             m_nComponents = 4;
526         }
527     }
528     ValidateDictParam();
529     m_pCompData = GetDecodeAndMaskArray(m_bDefaultDecode, m_bColorKey);
530     if (m_pCompData == NULL) {
531         return FALSE;
532     }
533     return TRUE;
534 }
535 DIB_COMP_DATA* CPDF_DIBSource::GetDecodeAndMaskArray(FX_BOOL& bDefaultDecode, FX_BOOL& bColorKey)
536 {
537     if (m_pColorSpace == NULL) {
538         return NULL;
539     }
540     DIB_COMP_DATA* pCompData = FX_Alloc(DIB_COMP_DATA, m_nComponents);
541     int max_data = (1 << m_bpc) - 1;
542     CPDF_Array* pDecode = m_pDict->GetArray(FX_BSTRC("Decode"));
543     if (pDecode) {
544         for (FX_DWORD i = 0; i < m_nComponents; i ++) {
545             pCompData[i].m_DecodeMin = pDecode->GetNumber(i * 2);
546             FX_FLOAT max = pDecode->GetNumber(i * 2 + 1);
547             pCompData[i].m_DecodeStep = (max - pCompData[i].m_DecodeMin) / max_data;
548             FX_FLOAT def_value, def_min, def_max;
549             m_pColorSpace->GetDefaultValue(i, def_value, def_min, def_max);
550             if (m_Family == PDFCS_INDEXED) {
551                 def_max = (FX_FLOAT)max_data;
552             }
553             if (def_min != pCompData[i].m_DecodeMin || def_max != max) {
554                 bDefaultDecode = FALSE;
555             }
556         }
557     } else {
558         for (FX_DWORD i = 0; i < m_nComponents; i ++) {
559             FX_FLOAT def_value;
560             m_pColorSpace->GetDefaultValue(i, def_value, pCompData[i].m_DecodeMin, pCompData[i].m_DecodeStep);
561             if (m_Family == PDFCS_INDEXED) {
562                 pCompData[i].m_DecodeStep = (FX_FLOAT)max_data;
563             }
564             pCompData[i].m_DecodeStep = (pCompData[i].m_DecodeStep - pCompData[i].m_DecodeMin) / max_data;
565         }
566     }
567     if (!m_pDict->KeyExist(FX_BSTRC("SMask"))) {
568         CPDF_Object* pMask = m_pDict->GetElementValue(FX_BSTRC("Mask"));
569         if (pMask == NULL) {
570             return pCompData;
571         }
572         if (pMask->GetType() == PDFOBJ_ARRAY) {
573             CPDF_Array* pArray = (CPDF_Array*)pMask;
574             if (pArray->GetCount() >= m_nComponents * 2) {
575                 for (FX_DWORD i = 0; i < m_nComponents; i++) {
576                     int min_num = pArray->GetInteger(i * 2);
577                     int max_num = pArray->GetInteger(i * 2 + 1);
578                     pCompData[i].m_ColorKeyMin = FX_MAX(min_num, 0);
579                     pCompData[i].m_ColorKeyMax = FX_MIN(max_num, max_data);
580                 }
581             }
582             bColorKey = TRUE;
583         }
584     }
585     return pCompData;
586 }
587 ICodec_ScanlineDecoder* FPDFAPI_CreateFaxDecoder(const uint8_t* src_buf, FX_DWORD src_size, int width, int height,
588         const CPDF_Dictionary* pParams);
589 ICodec_ScanlineDecoder* FPDFAPI_CreateFlateDecoder(const uint8_t* src_buf, FX_DWORD src_size, int width, int height,
590         int nComps, int bpc, const CPDF_Dictionary* pParams);
591 int CPDF_DIBSource::CreateDecoder()
592 {
593     const CFX_ByteString& decoder = m_pStreamAcc->GetImageDecoder();
594     if (decoder.IsEmpty()) {
595         return 1;
596     }
597     if (m_bDoBpcCheck && m_bpc == 0) {
598         return 0;
599     }
600     const uint8_t* src_data = m_pStreamAcc->GetData();
601     FX_DWORD src_size = m_pStreamAcc->GetSize();
602     const CPDF_Dictionary* pParams = m_pStreamAcc->GetImageParam();
603     if (decoder == FX_BSTRC("CCITTFaxDecode")) {
604         m_pDecoder = FPDFAPI_CreateFaxDecoder(src_data, src_size, m_Width, m_Height, pParams);
605     } else if (decoder == FX_BSTRC("DCTDecode")) {
606         m_pDecoder = CPDF_ModuleMgr::Get()->GetJpegModule()->CreateDecoder(
607             src_data, src_size, m_Width, m_Height, m_nComponents,
608             pParams ? pParams->GetInteger("ColorTransform", 1) : 1);
609         if (!m_pDecoder) {
610             FX_BOOL bTransform = FALSE;
611             int comps, bpc;
612             ICodec_JpegModule* pJpegModule = CPDF_ModuleMgr::Get()->GetJpegModule();
613             if (pJpegModule->LoadInfo(src_data, src_size, m_Width, m_Height, comps, bpc, bTransform)) {
614                 if (m_nComponents != comps) {
615                     FX_Free(m_pCompData);
616                     m_nComponents = comps;
617                     if (m_Family == PDFCS_LAB && m_nComponents != 3) {
618                         m_pCompData = NULL;
619                         return 0;
620                     }
621                     m_pCompData = GetDecodeAndMaskArray(m_bDefaultDecode, m_bColorKey);
622                     if (m_pCompData == NULL) {
623                         return 0;
624                     }
625                 }
626                 m_bpc = bpc;
627                 m_pDecoder = CPDF_ModuleMgr::Get()->GetJpegModule()->CreateDecoder(src_data, src_size, m_Width, m_Height,
628                              m_nComponents, bTransform);
629             }
630         }
631     } else if (decoder == FX_BSTRC("FlateDecode")) {
632         m_pDecoder = FPDFAPI_CreateFlateDecoder(src_data, src_size, m_Width, m_Height, m_nComponents, m_bpc, pParams);
633     } else if (decoder == FX_BSTRC("JPXDecode")) {
634         LoadJpxBitmap();
635         return m_pCachedBitmap ? 1 : 0;
636     } else if (decoder == FX_BSTRC("JBIG2Decode")) {
637         m_pCachedBitmap.reset(new CFX_DIBitmap);
638         if (!m_pCachedBitmap->Create(m_Width, m_Height, m_bImageMask ? FXDIB_1bppMask : FXDIB_1bppRgb)) {
639             m_pCachedBitmap.reset();
640             return 0;
641         }
642         m_Status = 1;
643         return 2;
644     } else if (decoder == FX_BSTRC("RunLengthDecode")) {
645         m_pDecoder = CPDF_ModuleMgr::Get()->GetCodecModule()->GetBasicModule()->CreateRunLengthDecoder(src_data, src_size, m_Width, m_Height, m_nComponents, m_bpc);
646     }
647     if (!m_pDecoder)
648         return 0;
649
650     FX_SAFE_DWORD requested_pitch =
651         CalculatePitch8(m_bpc, m_nComponents, m_Width, 1);
652     if (!requested_pitch.IsValid()) {
653         return 0;
654     }
655     FX_SAFE_DWORD provided_pitch = CalculatePitch8(m_pDecoder->GetBPC(),
656                                                    m_pDecoder->CountComps(),
657                                                    m_pDecoder->GetWidth(),
658                                                    1);
659     if (!provided_pitch.IsValid()) {
660         return 0;
661     }
662     if (provided_pitch.ValueOrDie() < requested_pitch.ValueOrDie()) {
663         return 0;
664     }
665     return 1;
666 }
667 void CPDF_DIBSource::LoadJpxBitmap()
668 {
669     ICodec_JpxModule* pJpxModule = CPDF_ModuleMgr::Get()->GetJpxModule();
670     if (!pJpxModule)
671         return;
672
673     nonstd::unique_ptr<JpxBitMapContext> context(
674         new JpxBitMapContext(pJpxModule));
675     context->set_context(pJpxModule->CreateDecoder(m_pStreamAcc->GetData(),
676                                                    m_pStreamAcc->GetSize(),
677                                                    m_pColorSpace != nullptr));
678     if (!context->context())
679         return;
680
681     FX_DWORD width = 0;
682     FX_DWORD height = 0;
683     FX_DWORD codestream_nComps = 0;
684     FX_DWORD image_nComps = 0;
685     pJpxModule->GetImageInfo(context->context(), width, height,
686                              codestream_nComps, image_nComps);
687     if ((int)width < m_Width || (int)height < m_Height)
688         return;
689
690     int output_nComps;
691     FX_BOOL bTranslateColor;
692     FX_BOOL bSwapRGB = FALSE;
693     if (m_pColorSpace) {
694         if (codestream_nComps != (FX_DWORD)m_pColorSpace->CountComponents())
695             return;
696         output_nComps = codestream_nComps;
697         bTranslateColor = FALSE;
698         if (m_pColorSpace == CPDF_ColorSpace::GetStockCS(PDFCS_DEVICERGB)) {
699             bSwapRGB = TRUE;
700             m_pColorSpace = nullptr;
701         }
702     } else {
703         bTranslateColor = TRUE;
704         if (image_nComps) {
705             output_nComps = image_nComps;
706         } else {
707             output_nComps = codestream_nComps;
708         }
709         if (output_nComps == 3) {
710             bSwapRGB = TRUE;
711         } else if (output_nComps == 4) {
712             m_pColorSpace = CPDF_ColorSpace::GetStockCS(PDFCS_DEVICECMYK);
713             bTranslateColor = FALSE;
714         }
715         m_nComponents = output_nComps;
716     }
717     FXDIB_Format format;
718     if (output_nComps == 1) {
719         format = FXDIB_8bppRgb;
720     } else if (output_nComps <= 3) {
721         format = FXDIB_Rgb;
722     } else if (output_nComps == 4) {
723         format = FXDIB_Rgb32;
724     } else {
725         width = (width * output_nComps + 2) / 3;
726         format = FXDIB_Rgb;
727     }
728     m_pCachedBitmap.reset(new CFX_DIBitmap);
729     if (!m_pCachedBitmap->Create(width, height, format)) {
730         m_pCachedBitmap.reset();
731         return;
732     }
733     m_pCachedBitmap->Clear(0xFFFFFFFF);
734     context->set_output_offsets(FX_Alloc(uint8_t, output_nComps));
735     for (int i = 0; i < output_nComps; ++i)
736         context->output_offsets()[i] = i;
737     if (bSwapRGB) {
738         context->output_offsets()[0] = 2;
739         context->output_offsets()[2] = 0;
740     }
741     if (!pJpxModule->Decode(context->context(),
742                             m_pCachedBitmap->GetBuffer(),
743                             m_pCachedBitmap->GetPitch(),
744                             bTranslateColor,
745                             context->output_offsets())) {
746         m_pCachedBitmap.reset();
747         return;
748     }
749     if (m_pColorSpace &&
750         m_pColorSpace->GetFamily() == PDFCS_INDEXED &&
751         m_bpc < 8) {
752         int scale = 8 - m_bpc;
753         for (FX_DWORD row = 0; row < height; ++row) {
754             uint8_t* scanline = (uint8_t*)m_pCachedBitmap->GetScanline(row);
755             for (FX_DWORD col = 0; col < width; ++col) {
756                 *scanline = (*scanline) >> scale;
757                 ++scanline;
758             }
759         }
760     }
761     m_bpc = 8;
762 }
763 CPDF_DIBSource* CPDF_DIBSource::LoadMask(FX_DWORD& MatteColor)
764 {
765     MatteColor = 0xffffffff;
766     CPDF_Stream* pSoftMask = m_pDict->GetStream(FX_BSTRC("SMask"));
767     if (pSoftMask) {
768         CPDF_Array* pMatte = pSoftMask->GetDict()->GetArray(FX_BSTRC("Matte"));
769         if (pMatte != NULL && m_pColorSpace && (FX_DWORD)m_pColorSpace->CountComponents() <= m_nComponents) {
770             FX_FLOAT* pColor = FX_Alloc(FX_FLOAT, m_nComponents);
771             for (FX_DWORD i = 0; i < m_nComponents; i ++) {
772                 pColor[i] = pMatte->GetFloat(i);
773             }
774             FX_FLOAT R, G, B;
775             m_pColorSpace->GetRGB(pColor, R, G, B);
776             FX_Free(pColor);
777             MatteColor = FXARGB_MAKE(0, FXSYS_round(R * 255), FXSYS_round(G * 255), FXSYS_round(B * 255));
778         }
779         return LoadMaskDIB(pSoftMask);
780     }
781     CPDF_Object* pMask = m_pDict->GetElementValue(FX_BSTRC("Mask"));
782     if (pMask == NULL) {
783         return NULL;
784     }
785     if (pMask->GetType() == PDFOBJ_STREAM) {
786         return LoadMaskDIB((CPDF_Stream*)pMask);
787     }
788     return NULL;
789 }
790 int     CPDF_DIBSource::StratLoadMask()
791 {
792     m_MatteColor = 0xffffffff;
793     m_pMaskStream = m_pDict->GetStream(FX_BSTRC("SMask"));
794     if (m_pMaskStream) {
795         CPDF_Array* pMatte = m_pMaskStream->GetDict()->GetArray(FX_BSTRC("Matte"));
796         if (pMatte != NULL && m_pColorSpace && (FX_DWORD)m_pColorSpace->CountComponents() <= m_nComponents) {
797             FX_FLOAT R, G, B;
798             FX_FLOAT* pColor = FX_Alloc(FX_FLOAT, m_nComponents);
799             for (FX_DWORD i = 0; i < m_nComponents; i ++) {
800                 pColor[i] = pMatte->GetFloat(i);
801             }
802             m_pColorSpace->GetRGB(pColor, R, G, B);
803             FX_Free(pColor);
804             m_MatteColor = FXARGB_MAKE(0, FXSYS_round(R * 255), FXSYS_round(G * 255), FXSYS_round(B * 255));
805         }
806         return StartLoadMaskDIB();
807     }
808     m_pMaskStream = m_pDict->GetElementValue(FX_BSTRC("Mask"));
809     if (m_pMaskStream == NULL) {
810         return 1;
811     }
812     if (m_pMaskStream->GetType() == PDFOBJ_STREAM) {
813         return StartLoadMaskDIB();
814     }
815     return 1;
816 }
817 int     CPDF_DIBSource::ContinueLoadMaskDIB(IFX_Pause* pPause)
818 {
819     if (m_pMask == NULL) {
820         return 1;
821     }
822     int ret = m_pMask->ContinueLoadDIBSource(pPause);
823     if (ret == 2) {
824         return ret;
825     }
826     if (m_pColorSpace && m_bStdCS) {
827         m_pColorSpace->EnableStdConversion(FALSE);
828     }
829     if (!ret) {
830         delete m_pMask;
831         m_pMask = NULL;
832         return ret;
833     }
834     return 1;
835 }
836 CPDF_DIBSource* CPDF_DIBSource::DetachMask()
837 {
838     CPDF_DIBSource* pDIBSource = m_pMask;
839     m_pMask = NULL;
840     return pDIBSource;
841 }
842 CPDF_DIBSource* CPDF_DIBSource::LoadMaskDIB(CPDF_Stream* pMask)
843 {
844     CPDF_DIBSource* pMaskSource = new CPDF_DIBSource;
845     if (!pMaskSource->Load(m_pDocument, pMask, NULL, NULL, NULL, NULL, TRUE)) {
846         delete pMaskSource;
847         return NULL;
848     }
849     return pMaskSource;
850 }
851 int CPDF_DIBSource::StartLoadMaskDIB()
852 {
853     m_pMask = new CPDF_DIBSource;
854     int ret = m_pMask->StartLoadDIBSource(m_pDocument, (CPDF_Stream*)m_pMaskStream, FALSE, NULL, NULL, TRUE);
855     if (ret == 2) {
856         if (m_Status == 0) {
857             m_Status = 2;
858         }
859         return 2;
860     }
861     if (!ret) {
862         delete m_pMask;
863         m_pMask = NULL;
864         return 1;
865     }
866     return 1;
867 }
868 void CPDF_DIBSource::LoadPalette()
869 {
870     if (m_bpc == 0) {
871         return;
872     }
873     if (m_bpc * m_nComponents > 8) {
874         return;
875     }
876     if (m_pColorSpace == NULL) {
877         return;
878     }
879     if (m_bpc * m_nComponents == 1) {
880         if (m_bDefaultDecode && (m_Family == PDFCS_DEVICEGRAY || m_Family == PDFCS_DEVICERGB)) {
881             return;
882         }
883         if (m_pColorSpace->CountComponents() > 3) {
884             return;
885         }
886         FX_FLOAT color_values[3];
887         color_values[0] = m_pCompData[0].m_DecodeMin;
888         color_values[1] = color_values[2] = color_values[0];
889         FX_FLOAT R=0.0f, G=0.0f, B=0.0f;
890         m_pColorSpace->GetRGB(color_values, R, G, B);
891         FX_ARGB argb0 = ArgbEncode(255, FXSYS_round(R * 255), FXSYS_round(G * 255), FXSYS_round(B * 255));
892         color_values[0] += m_pCompData[0].m_DecodeStep;
893         color_values[1] += m_pCompData[0].m_DecodeStep;
894         color_values[2] += m_pCompData[0].m_DecodeStep;
895         m_pColorSpace->GetRGB(color_values, R, G, B);
896         FX_ARGB argb1 = ArgbEncode(255, FXSYS_round(R * 255), FXSYS_round(G * 255), FXSYS_round(B * 255));
897         if (argb0 != 0xFF000000 || argb1 != 0xFFFFFFFF) {
898             SetPaletteArgb(0, argb0);
899             SetPaletteArgb(1, argb1);
900         }
901         return;
902     }
903     if (m_pColorSpace == CPDF_ColorSpace::GetStockCS(PDFCS_DEVICEGRAY) && m_bpc == 8 && m_bDefaultDecode) {
904     } else {
905         int palette_count = 1 << (m_bpc * m_nComponents);
906         CFX_FixedBufGrow<FX_FLOAT, 16> color_values(m_nComponents);
907         FX_FLOAT* color_value = color_values;
908         for (int i = 0; i < palette_count; i ++) {
909             int color_data = i;
910             for (FX_DWORD j = 0; j < m_nComponents; j ++) {
911                 int encoded_component = color_data % (1 << m_bpc);
912                 color_data /= 1 << m_bpc;
913                 color_value[j] = m_pCompData[j].m_DecodeMin + m_pCompData[j].m_DecodeStep * encoded_component;
914             }
915             FX_FLOAT R = 0, G = 0, B = 0;
916             if (m_nComponents == 1 && m_Family == PDFCS_ICCBASED && m_pColorSpace->CountComponents() > 1) {
917                 int nComponents = m_pColorSpace->CountComponents();
918                 FX_FLOAT* temp_buf = FX_Alloc(FX_FLOAT, nComponents);
919                 for (int i = 0; i < nComponents; i++) {
920                     temp_buf[i] = *color_value;
921                 }
922                 m_pColorSpace->GetRGB(temp_buf, R, G, B);
923                 FX_Free(temp_buf);
924             } else {
925                 m_pColorSpace->GetRGB(color_value, R, G, B);
926             }
927             SetPaletteArgb(i, ArgbEncode(255, FXSYS_round(R * 255), FXSYS_round(G * 255), FXSYS_round(B * 255)));
928         }
929     }
930 }
931 void CPDF_DIBSource::ValidateDictParam()
932 {
933     m_bpc = m_bpc_orig;
934     CPDF_Object * pFilter = m_pDict->GetElementValue(FX_BSTRC("Filter"));
935     if (pFilter) {
936         if (pFilter->GetType() == PDFOBJ_NAME) {
937             CFX_ByteString filter = pFilter->GetString();
938             if (filter == FX_BSTRC("CCITTFaxDecode") || filter == FX_BSTRC("JBIG2Decode")) {
939                 m_bpc = 1;
940                 m_nComponents = 1;
941             }
942             if (filter == FX_BSTRC("RunLengthDecode") || filter == FX_BSTRC("DCTDecode")) {
943                 m_bpc = 8;
944             }
945         } else if (pFilter->GetType() == PDFOBJ_ARRAY) {
946             CPDF_Array *pArray = (CPDF_Array *)pFilter;
947             if (pArray->GetString(pArray->GetCount() - 1) == FX_BSTRC("CCITTFaxDecode") ||
948                     pArray->GetString(pArray->GetCount() - 1) == FX_BSTRC("JBIG2Decode")) {
949                 m_bpc = 1;
950                 m_nComponents = 1;
951             }
952             if (pArray->GetString(pArray->GetCount() - 1) == FX_BSTRC("DCTDecode")) {
953                 // Previously, pArray->GetString(pArray->GetCount() - 1) == FX_BSTRC("RunLengthDecode") was checked in the "if" statement as well,
954                 // but too many documents don't conform to it.
955                 m_bpc = 8;
956             }
957         }
958     }
959     if (m_bpc != 1 && m_bpc != 2 && m_bpc != 4 && m_bpc != 8 && m_bpc != 16) {
960         m_bpc = 0;
961     }
962 }
963 #define NORMALCOLOR_MAX(color, max) (color) > (max) ? (max) : (color) < 0 ? 0 : (color);
964 void CPDF_DIBSource::TranslateScanline24bpp(uint8_t* dest_scan, const uint8_t* src_scan) const
965 {
966     if (m_bpc == 0) {
967         return;
968     }
969     int max_data = (1 << m_bpc) - 1;
970     if (m_bDefaultDecode) {
971         if (m_Family == PDFCS_DEVICERGB || m_Family == PDFCS_CALRGB) {
972             const uint8_t* src_pos = src_scan;
973             switch (m_bpc) {
974                 case 16:
975                     for (int col = 0; col < m_Width; col ++) {
976                         *dest_scan++ = src_pos[4];
977                         *dest_scan++ = src_pos[2];
978                         *dest_scan++ = *src_pos;
979                         src_pos += 6;
980                     }
981                     break;
982                 case 8:
983                     for (int column = 0; column < m_Width; column ++) {
984                         *dest_scan++ = src_pos[2];
985                         *dest_scan++ = src_pos[1];
986                         *dest_scan++ = *src_pos;
987                         src_pos += 3;
988                     }
989                     break;
990                 default:
991                     int src_bit_pos = 0;
992                     int dest_byte_pos = 0;
993                     for (int column = 0; column < m_Width; column ++) {
994                         int R = _GetBits8(src_scan, src_bit_pos, m_bpc);
995                         src_bit_pos += m_bpc;
996                         int G = _GetBits8(src_scan, src_bit_pos, m_bpc);
997                         src_bit_pos += m_bpc;
998                         int B = _GetBits8(src_scan, src_bit_pos, m_bpc);
999                         src_bit_pos += m_bpc;
1000                         R = NORMALCOLOR_MAX(R, max_data);
1001                         G = NORMALCOLOR_MAX(G, max_data);
1002                         B = NORMALCOLOR_MAX(B, max_data);
1003                         dest_scan[dest_byte_pos] = B * 255 / max_data;
1004                         dest_scan[dest_byte_pos + 1] = G * 255 / max_data;
1005                         dest_scan[dest_byte_pos + 2] = R * 255 / max_data;
1006                         dest_byte_pos += 3;
1007                     }
1008                     break;
1009             }
1010             return;
1011         } else if (m_bpc == 8) {
1012             if (m_nComponents == m_pColorSpace->CountComponents())
1013                 m_pColorSpace->TranslateImageLine(dest_scan, src_scan, m_Width, m_Width, m_Height,
1014                                                   m_bLoadMask && m_GroupFamily == PDFCS_DEVICECMYK && m_Family == PDFCS_DEVICECMYK);
1015             return;
1016         }
1017     }
1018     CFX_FixedBufGrow<FX_FLOAT, 16> color_values1(m_nComponents);
1019     FX_FLOAT* color_values = color_values1;
1020     FX_FLOAT R = 0.0f, G = 0.0f, B = 0.0f;
1021     if (m_bpc == 8) {
1022         int src_byte_pos = 0;
1023         int dest_byte_pos = 0;
1024         for (int column = 0; column < m_Width; column ++) {
1025             for (FX_DWORD color = 0; color < m_nComponents; color ++) {
1026                 int data = src_scan[src_byte_pos ++];
1027                 color_values[color] = m_pCompData[color].m_DecodeMin +
1028                                       m_pCompData[color].m_DecodeStep * data;
1029             }
1030             if (m_bLoadMask && m_GroupFamily == PDFCS_DEVICECMYK && m_Family == PDFCS_DEVICECMYK) {
1031                 FX_FLOAT k = 1.0f - color_values[3];
1032                 R = (1.0f - color_values[0]) * k;
1033                 G = (1.0f - color_values[1]) * k;
1034                 B = (1.0f - color_values[2]) * k;
1035             } else {
1036                 m_pColorSpace->GetRGB(color_values, R, G, B);
1037             }
1038             R = NORMALCOLOR_MAX(R, 1);
1039             G = NORMALCOLOR_MAX(G, 1);
1040             B = NORMALCOLOR_MAX(B, 1);
1041             dest_scan[dest_byte_pos] = (int32_t)(B * 255);
1042             dest_scan[dest_byte_pos + 1] = (int32_t)(G * 255);
1043             dest_scan[dest_byte_pos + 2] = (int32_t)(R * 255);
1044             dest_byte_pos += 3;
1045         }
1046     } else {
1047         int src_bit_pos = 0;
1048         int dest_byte_pos = 0;
1049         for (int column = 0; column < m_Width; column ++) {
1050             for (FX_DWORD color = 0; color < m_nComponents; color ++) {
1051                 int data = _GetBits8(src_scan, src_bit_pos, m_bpc);
1052                 color_values[color] = m_pCompData[color].m_DecodeMin +
1053                                       m_pCompData[color].m_DecodeStep * data;
1054                 src_bit_pos += m_bpc;
1055             }
1056             if (m_bLoadMask && m_GroupFamily == PDFCS_DEVICECMYK && m_Family == PDFCS_DEVICECMYK) {
1057                 FX_FLOAT k = 1.0f - color_values[3];
1058                 R = (1.0f - color_values[0]) * k;
1059                 G = (1.0f - color_values[1]) * k;
1060                 B = (1.0f - color_values[2]) * k;
1061             } else {
1062                 m_pColorSpace->GetRGB(color_values, R, G, B);
1063             }
1064             R = NORMALCOLOR_MAX(R, 1);
1065             G = NORMALCOLOR_MAX(G, 1);
1066             B = NORMALCOLOR_MAX(B, 1);
1067             dest_scan[dest_byte_pos] = (int32_t)(B * 255);
1068             dest_scan[dest_byte_pos + 1] = (int32_t)(G * 255);
1069             dest_scan[dest_byte_pos + 2] = (int32_t)(R * 255);
1070             dest_byte_pos += 3;
1071         }
1072     }
1073 }
1074 uint8_t* CPDF_DIBSource::GetBuffer() const
1075 {
1076     if (m_pCachedBitmap) {
1077         return m_pCachedBitmap->GetBuffer();
1078     }
1079     return NULL;
1080 }
1081 const uint8_t* CPDF_DIBSource::GetScanline(int line) const
1082 {
1083     if (m_bpc == 0) {
1084         return NULL;
1085     }
1086     FX_SAFE_DWORD src_pitch = CalculatePitch8(m_bpc, m_nComponents, m_Width, 1);
1087     if (!src_pitch.IsValid())
1088         return NULL;
1089     FX_DWORD src_pitch_value = src_pitch.ValueOrDie();
1090     const uint8_t* pSrcLine = NULL;
1091     if (m_pCachedBitmap) {
1092         if (line >= m_pCachedBitmap->GetHeight()) {
1093             line = m_pCachedBitmap->GetHeight() - 1;
1094         }
1095         pSrcLine = m_pCachedBitmap->GetScanline(line);
1096     } else if (m_pDecoder) {
1097         pSrcLine = m_pDecoder->GetScanline(line);
1098     } else {
1099         if (m_pStreamAcc->GetSize() >= (line + 1) * src_pitch_value) {
1100             pSrcLine = m_pStreamAcc->GetData() + line * src_pitch_value;
1101         }
1102     }
1103     if (pSrcLine == NULL) {
1104         uint8_t* pLineBuf = m_pMaskedLine ? m_pMaskedLine : m_pLineBuf;
1105         FXSYS_memset(pLineBuf, 0xff, m_Pitch);
1106         return pLineBuf;
1107     }
1108     if (m_bpc * m_nComponents == 1) {
1109         if (m_bImageMask && m_bDefaultDecode) {
1110             for (FX_DWORD i = 0; i < src_pitch_value; i++) {
1111                 m_pLineBuf[i] = ~pSrcLine[i];
1112             }
1113         } else if (m_bColorKey) {
1114             FX_DWORD reset_argb, set_argb;
1115             reset_argb = m_pPalette ? m_pPalette[0] : 0xff000000;
1116             set_argb = m_pPalette ? m_pPalette[1] : 0xffffffff;
1117             if (m_pCompData[0].m_ColorKeyMin == 0) {
1118                 reset_argb = 0;
1119             }
1120             if (m_pCompData[0].m_ColorKeyMax == 1) {
1121                 set_argb = 0;
1122             }
1123             set_argb = FXARGB_TODIB(set_argb);
1124             reset_argb = FXARGB_TODIB(reset_argb);
1125             FX_DWORD* dest_scan = (FX_DWORD*)m_pMaskedLine;
1126             for (int col = 0; col < m_Width; col ++) {
1127                 if (pSrcLine[col / 8] & (1 << (7 - col % 8))) {
1128                     *dest_scan = set_argb;
1129                 } else {
1130                     *dest_scan = reset_argb;
1131                 }
1132                 dest_scan ++;
1133             }
1134             return m_pMaskedLine;
1135         } else {
1136             FXSYS_memcpy(m_pLineBuf, pSrcLine, src_pitch_value);
1137         }
1138         return m_pLineBuf;
1139     }
1140     if (m_bpc * m_nComponents <= 8) {
1141         if (m_bpc == 8) {
1142             FXSYS_memcpy(m_pLineBuf, pSrcLine, src_pitch_value);
1143         } else {
1144             int src_bit_pos = 0;
1145             for (int col = 0; col < m_Width; col ++) {
1146                 int color_index = 0;
1147                 for (FX_DWORD color = 0; color < m_nComponents; color ++) {
1148                     int data = _GetBits8(pSrcLine, src_bit_pos, m_bpc);
1149                     color_index |= data << (color * m_bpc);
1150                     src_bit_pos += m_bpc;
1151                 }
1152                 m_pLineBuf[col] = color_index;
1153             }
1154         }
1155         if (m_bColorKey) {
1156             uint8_t* pDestPixel = m_pMaskedLine;
1157             const uint8_t* pSrcPixel = m_pLineBuf;
1158             for (int col = 0; col < m_Width; col ++) {
1159                 uint8_t index = *pSrcPixel++;
1160                 if (m_pPalette) {
1161                     *pDestPixel++ = FXARGB_B(m_pPalette[index]);
1162                     *pDestPixel++ = FXARGB_G(m_pPalette[index]);
1163                     *pDestPixel++ = FXARGB_R(m_pPalette[index]);
1164                 } else {
1165                     *pDestPixel++ = index;
1166                     *pDestPixel++ = index;
1167                     *pDestPixel++ = index;
1168                 }
1169                 *pDestPixel = (index < m_pCompData[0].m_ColorKeyMin || index > m_pCompData[0].m_ColorKeyMax) ? 0xff : 0;
1170                 pDestPixel ++ ;
1171             }
1172             return m_pMaskedLine;
1173         }
1174         return m_pLineBuf;
1175     }
1176     if (m_bColorKey) {
1177         if (m_nComponents == 3 && m_bpc == 8) {
1178             uint8_t* alpha_channel = m_pMaskedLine + 3;
1179             for (int col = 0; col < m_Width; col ++) {
1180                 const uint8_t* pPixel = pSrcLine + col * 3;
1181                 alpha_channel[col * 4] = (pPixel[0] < m_pCompData[0].m_ColorKeyMin || pPixel[0] > m_pCompData[0].m_ColorKeyMax ||
1182                                           pPixel[1] < m_pCompData[1].m_ColorKeyMin || pPixel[1] > m_pCompData[1].m_ColorKeyMax ||
1183                                           pPixel[2] < m_pCompData[2].m_ColorKeyMin || pPixel[2] > m_pCompData[2].m_ColorKeyMax) ? 0xff : 0;
1184             }
1185         } else {
1186             FXSYS_memset(m_pMaskedLine, 0xff, m_Pitch);
1187         }
1188     }
1189     if (m_pColorSpace) {
1190         TranslateScanline24bpp(m_pLineBuf, pSrcLine);
1191         pSrcLine = m_pLineBuf;
1192     }
1193     if (m_bColorKey) {
1194         const uint8_t* pSrcPixel = pSrcLine;
1195         uint8_t* pDestPixel = m_pMaskedLine;
1196         for (int col = 0; col < m_Width; col ++) {
1197             *pDestPixel++ = *pSrcPixel++;
1198             *pDestPixel++ = *pSrcPixel++;
1199             *pDestPixel++ = *pSrcPixel++;
1200             pDestPixel ++;
1201         }
1202         return m_pMaskedLine;
1203     }
1204     return pSrcLine;
1205 }
1206 FX_BOOL CPDF_DIBSource::SkipToScanline(int line, IFX_Pause* pPause) const
1207 {
1208     if (m_pDecoder) {
1209         return m_pDecoder->SkipToScanline(line, pPause);
1210     }
1211     return FALSE;
1212 }
1213 void CPDF_DIBSource::DownSampleScanline(int line, uint8_t* dest_scan, int dest_bpp,
1214                                         int dest_width, FX_BOOL bFlipX, int clip_left, int clip_width) const
1215 {
1216     if (line < 0 || dest_scan == NULL || dest_bpp <= 0 ||
1217         dest_width <= 0 || clip_left < 0 || clip_width <= 0) {
1218         return;
1219     }
1220
1221     FX_DWORD src_width = m_Width;
1222     FX_SAFE_DWORD pitch = CalculatePitch8(m_bpc, m_nComponents, m_Width, 1);
1223     if (!pitch.IsValid()) {
1224         return;
1225     }
1226
1227     const uint8_t* pSrcLine = NULL;
1228     if (m_pCachedBitmap) {
1229         pSrcLine = m_pCachedBitmap->GetScanline(line);
1230     } else if (m_pDecoder) {
1231         pSrcLine = m_pDecoder->GetScanline(line);
1232     } else {
1233         FX_DWORD src_pitch = pitch.ValueOrDie();
1234         pitch *= (line+1);
1235         if (!pitch.IsValid()) {
1236             return;
1237         }
1238
1239         if (m_pStreamAcc->GetSize() >= pitch.ValueOrDie()) {
1240             pSrcLine = m_pStreamAcc->GetData() + line * src_pitch;
1241         }
1242     }
1243     int orig_Bpp = m_bpc * m_nComponents / 8;
1244     int dest_Bpp = dest_bpp / 8;
1245     if (pSrcLine == NULL) {
1246         FXSYS_memset(dest_scan, 0xff, dest_Bpp * clip_width);
1247         return;
1248     }
1249
1250     FX_SAFE_INT32 max_src_x = clip_left;
1251     max_src_x += clip_width - 1;
1252     max_src_x *= src_width;
1253     max_src_x /= dest_width;
1254     if (!max_src_x.IsValid()) {
1255         return;
1256     }
1257
1258     CFX_FixedBufGrow<uint8_t, 128> temp(orig_Bpp);
1259     if (m_bpc * m_nComponents == 1) {
1260         FX_DWORD set_argb = (FX_DWORD) - 1, reset_argb = 0;
1261         if (m_bImageMask) {
1262             if (m_bDefaultDecode) {
1263                 set_argb = 0;
1264                 reset_argb = (FX_DWORD) - 1;
1265             }
1266         } else if (m_bColorKey) {
1267             reset_argb = m_pPalette ? m_pPalette[0] : 0xff000000;
1268             set_argb = m_pPalette ? m_pPalette[1] : 0xffffffff;
1269             if (m_pCompData[0].m_ColorKeyMin == 0) {
1270                 reset_argb = 0;
1271             }
1272             if (m_pCompData[0].m_ColorKeyMax == 1) {
1273                 set_argb = 0;
1274             }
1275             set_argb = FXARGB_TODIB(set_argb);
1276             reset_argb = FXARGB_TODIB(reset_argb);
1277             for (int i = 0; i < clip_width; i ++) {
1278                 FX_DWORD src_x = (clip_left + i) * src_width / dest_width;
1279                 if (bFlipX) {
1280                     src_x = src_width - src_x - 1;
1281                 }
1282                 src_x %= src_width;
1283                 if (pSrcLine[src_x / 8] & (1 << (7 - src_x % 8))) {
1284                     ((FX_DWORD*)dest_scan)[i] = set_argb;
1285                 } else {
1286                     ((FX_DWORD*)dest_scan)[i] = reset_argb;
1287                 }
1288             }
1289             return;
1290         } else {
1291             if (dest_Bpp == 1) {
1292             } else if (m_pPalette) {
1293                 reset_argb = m_pPalette[0];
1294                 set_argb = m_pPalette[1];
1295             }
1296         }
1297         for (int i = 0; i < clip_width; i ++) {
1298             FX_DWORD src_x = (clip_left + i) * src_width / dest_width;
1299             if (bFlipX) {
1300                 src_x = src_width - src_x - 1;
1301             }
1302             src_x %= src_width;
1303             int dest_pos = i * dest_Bpp;
1304             if (pSrcLine[src_x / 8] & (1 << (7 - src_x % 8))) {
1305                 if (dest_Bpp == 1) {
1306                     dest_scan[dest_pos] = (uint8_t)set_argb;
1307                 } else if (dest_Bpp == 3) {
1308                     dest_scan[dest_pos] = FXARGB_B(set_argb);
1309                     dest_scan[dest_pos + 1] = FXARGB_G(set_argb);
1310                     dest_scan[dest_pos + 2] = FXARGB_R(set_argb);
1311                 } else {
1312                     *(FX_DWORD*)(dest_scan + dest_pos) = set_argb;
1313                 }
1314             } else {
1315                 if (dest_Bpp == 1) {
1316                     dest_scan[dest_pos] = (uint8_t)reset_argb;
1317                 } else if (dest_Bpp == 3) {
1318                     dest_scan[dest_pos] = FXARGB_B(reset_argb);
1319                     dest_scan[dest_pos + 1] = FXARGB_G(reset_argb);
1320                     dest_scan[dest_pos + 2] = FXARGB_R(reset_argb);
1321                 } else {
1322                     *(FX_DWORD*)(dest_scan + dest_pos) = reset_argb;
1323                 }
1324             }
1325         }
1326         return;
1327     } else if (m_bpc * m_nComponents <= 8) {
1328         if (m_bpc < 8) {
1329             int src_bit_pos = 0;
1330             for (FX_DWORD col = 0; col < src_width; col ++) {
1331                 int color_index = 0;
1332                 for (FX_DWORD color = 0; color < m_nComponents; color ++) {
1333                     int data = _GetBits8(pSrcLine, src_bit_pos, m_bpc);
1334                     color_index |= data << (color * m_bpc);
1335                     src_bit_pos += m_bpc;
1336                 }
1337                 m_pLineBuf[col] = color_index;
1338             }
1339             pSrcLine = m_pLineBuf;
1340         }
1341         if (m_bColorKey) {
1342             for (int i = 0; i < clip_width; i ++) {
1343                 FX_DWORD src_x = (clip_left + i) * src_width / dest_width;
1344                 if (bFlipX) {
1345                     src_x = src_width - src_x - 1;
1346                 }
1347                 src_x %= src_width;
1348                 uint8_t* pDestPixel = dest_scan + i * 4;
1349                 uint8_t index = pSrcLine[src_x];
1350                 if (m_pPalette) {
1351                     *pDestPixel++ = FXARGB_B(m_pPalette[index]);
1352                     *pDestPixel++ = FXARGB_G(m_pPalette[index]);
1353                     *pDestPixel++ = FXARGB_R(m_pPalette[index]);
1354                 } else {
1355                     *pDestPixel++ = index;
1356                     *pDestPixel++ = index;
1357                     *pDestPixel++ = index;
1358                 }
1359                 *pDestPixel = (index < m_pCompData[0].m_ColorKeyMin || index > m_pCompData[0].m_ColorKeyMax) ? 0xff : 0;
1360             }
1361             return;
1362         }
1363         for (int i = 0; i < clip_width; i ++) {
1364             FX_DWORD src_x = (clip_left + i) * src_width / dest_width;
1365             if (bFlipX) {
1366                 src_x = src_width - src_x - 1;
1367             }
1368             src_x %= src_width;
1369             uint8_t index = pSrcLine[src_x];
1370             if (dest_Bpp == 1) {
1371                 dest_scan[i] = index;
1372             } else {
1373                 int dest_pos = i * dest_Bpp;
1374                 FX_ARGB argb = m_pPalette[index];
1375                 dest_scan[dest_pos] = FXARGB_B(argb);
1376                 dest_scan[dest_pos + 1] = FXARGB_G(argb);
1377                 dest_scan[dest_pos + 2] = FXARGB_R(argb);
1378             }
1379         }
1380         return;
1381     } else {
1382         int last_src_x = -1;
1383         FX_ARGB last_argb;
1384         FX_FLOAT orig_Not8Bpp = (FX_FLOAT)m_bpc * (FX_FLOAT)m_nComponents / 8.0f;
1385         FX_FLOAT unit_To8Bpc = 255.0f / ((1 << m_bpc) - 1);
1386         for (int i = 0; i < clip_width; i ++) {
1387             int dest_x = clip_left + i;
1388             FX_DWORD src_x = (bFlipX ? (dest_width - dest_x - 1) : dest_x) * (int64_t)src_width / dest_width;
1389             src_x %= src_width;
1390             const uint8_t* pSrcPixel = NULL;
1391             if (m_bpc % 8 == 0) {
1392                 pSrcPixel = pSrcLine + src_x * orig_Bpp;
1393             } else {
1394                 pSrcPixel = pSrcLine + (int)(src_x * orig_Not8Bpp);
1395             }
1396             uint8_t* pDestPixel = dest_scan + i * dest_Bpp;
1397             FX_ARGB argb;
1398             if (src_x == last_src_x) {
1399                 argb = last_argb;
1400             } else {
1401                 if (m_pColorSpace) {
1402                     uint8_t color[4];
1403                     if (!m_bDefaultDecode) {
1404                         for (int i = 0; i < m_nComponents; i ++) {
1405                             int color_value = (int)((m_pCompData[i].m_DecodeMin + m_pCompData[i].m_DecodeStep * (FX_FLOAT)pSrcPixel[i]) * 255.0f + 0.5f);
1406                             temp[i] = color_value > 255 ? 255 : (color_value < 0 ? 0 : color_value);
1407                         }
1408                         m_pColorSpace->TranslateImageLine(color, temp, 1, 0, 0, m_bLoadMask && m_GroupFamily == PDFCS_DEVICECMYK && m_Family == PDFCS_DEVICECMYK);
1409                     } else {
1410                         if (m_bpc < 8) {
1411                             int src_bit_pos = 0;
1412                             if (src_x % 2) {
1413                                 src_bit_pos = 4;
1414                             }
1415                             for (FX_DWORD i = 0; i < m_nComponents; i ++) {
1416                                 temp[i] = (uint8_t)(_GetBits8(pSrcPixel, src_bit_pos, m_bpc) * unit_To8Bpc);
1417                                 src_bit_pos += m_bpc;
1418                             }
1419                             m_pColorSpace->TranslateImageLine(color, temp, 1, 0, 0, m_bLoadMask && m_GroupFamily == PDFCS_DEVICECMYK && m_Family == PDFCS_DEVICECMYK);
1420                         } else {
1421                             m_pColorSpace->TranslateImageLine(color, pSrcPixel, 1, 0, 0, m_bLoadMask && m_GroupFamily == PDFCS_DEVICECMYK && m_Family == PDFCS_DEVICECMYK);
1422                         }
1423                     }
1424                     argb = FXARGB_MAKE(0xff, color[2], color[1], color[0]);
1425                 } else {
1426                     argb = FXARGB_MAKE(0xff, pSrcPixel[2], pSrcPixel[1], pSrcPixel[0]);
1427                 }
1428                 if (m_bColorKey) {
1429                     int alpha = 0xff;
1430                     if (m_nComponents == 3 && m_bpc == 8) {
1431                         alpha = (pSrcPixel[0] < m_pCompData[0].m_ColorKeyMin ||
1432                                  pSrcPixel[0] > m_pCompData[0].m_ColorKeyMax ||
1433                                  pSrcPixel[1] < m_pCompData[1].m_ColorKeyMin ||
1434                                  pSrcPixel[1] > m_pCompData[1].m_ColorKeyMax ||
1435                                  pSrcPixel[2] < m_pCompData[2].m_ColorKeyMin ||
1436                                  pSrcPixel[2] > m_pCompData[2].m_ColorKeyMax) ? 0xff : 0;
1437                     }
1438                     argb &= 0xffffff;
1439                     argb |= alpha << 24;
1440                 }
1441                 last_src_x = src_x;
1442                 last_argb = argb;
1443             }
1444             if (dest_Bpp == 4) {
1445                 *(FX_DWORD*)pDestPixel = FXARGB_TODIB(argb);
1446             } else {
1447                 *pDestPixel++ = FXARGB_B(argb);
1448                 *pDestPixel++ = FXARGB_G(argb);
1449                 *pDestPixel = FXARGB_R(argb);
1450             }
1451         }
1452     }
1453 }
1454 void CPDF_DIBSource::SetDownSampleSize(int dest_width, int dest_height) const
1455 {
1456     if (m_pDecoder) {
1457         m_pDecoder->DownScale(dest_width, dest_height);
1458         ((CPDF_DIBSource*)this)->m_Width = m_pDecoder->GetWidth();
1459         ((CPDF_DIBSource*)this)->m_Height = m_pDecoder->GetHeight();
1460     }
1461 }
1462 void CPDF_DIBSource::ClearImageData()
1463 {
1464     if (m_pDecoder) {
1465         m_pDecoder->ClearImageData();
1466     }
1467 }
1468 CPDF_ProgressiveImageLoaderHandle::CPDF_ProgressiveImageLoaderHandle()
1469 {
1470     m_pImageLoader = NULL;
1471     m_pCache = NULL;
1472     m_pImage = NULL;
1473 }
1474 CPDF_ProgressiveImageLoaderHandle::~CPDF_ProgressiveImageLoaderHandle()
1475 {
1476 }
1477 FX_BOOL CPDF_ProgressiveImageLoaderHandle::Start(CPDF_ImageLoader* pImageLoader, const CPDF_ImageObject* pImage, CPDF_PageRenderCache* pCache, FX_BOOL bStdCS, FX_DWORD GroupFamily, FX_BOOL bLoadMask, CPDF_RenderStatus* pRenderStatus, int32_t nDownsampleWidth, int32_t nDownsampleHeight)
1478 {
1479     m_pImageLoader = pImageLoader;
1480     m_pCache = pCache;
1481     m_pImage = (CPDF_ImageObject*)pImage;
1482     m_nDownsampleWidth = nDownsampleWidth;
1483     m_nDownsampleHeight = nDownsampleHeight;
1484     FX_BOOL ret;
1485     if (pCache) {
1486         ret = pCache->StartGetCachedBitmap(pImage->m_pImage->GetStream(), bStdCS, GroupFamily, bLoadMask, pRenderStatus, m_nDownsampleWidth, m_nDownsampleHeight);
1487         if (ret == FALSE) {
1488             m_pImageLoader->m_bCached = TRUE;
1489             m_pImageLoader->m_pBitmap = pCache->m_pCurImageCache->DetachBitmap();
1490             m_pImageLoader->m_pMask = pCache->m_pCurImageCache->DetachMask();
1491             m_pImageLoader->m_MatteColor = pCache->m_pCurImageCache->m_MatteColor;
1492         }
1493     } else {
1494         ret = pImage->m_pImage->StartLoadDIBSource(pRenderStatus->m_pFormResource, pRenderStatus->m_pPageResource, bStdCS, GroupFamily, bLoadMask);
1495         if (ret == FALSE) {
1496             m_pImageLoader->m_bCached = FALSE;
1497             m_pImageLoader->m_pBitmap = m_pImage->m_pImage->DetachBitmap();
1498             m_pImageLoader->m_pMask = m_pImage->m_pImage->DetachMask();
1499             m_pImageLoader->m_MatteColor = m_pImage->m_pImage->m_MatteColor;
1500         }
1501     }
1502     return ret;
1503 }
1504 FX_BOOL CPDF_ProgressiveImageLoaderHandle::Continue(IFX_Pause* pPause)
1505 {
1506     FX_BOOL ret;
1507     if (m_pCache) {
1508         ret = m_pCache->Continue(pPause);
1509         if (ret == FALSE) {
1510             m_pImageLoader->m_bCached = TRUE;
1511             m_pImageLoader->m_pBitmap = m_pCache->m_pCurImageCache->DetachBitmap();
1512             m_pImageLoader->m_pMask = m_pCache->m_pCurImageCache->DetachMask();
1513             m_pImageLoader->m_MatteColor = m_pCache->m_pCurImageCache->m_MatteColor;
1514         }
1515     } else {
1516         ret = m_pImage->m_pImage->Continue(pPause);
1517         if (ret == FALSE) {
1518             m_pImageLoader->m_bCached = FALSE;
1519             m_pImageLoader->m_pBitmap = m_pImage->m_pImage->DetachBitmap();
1520             m_pImageLoader->m_pMask = m_pImage->m_pImage->DetachMask();
1521             m_pImageLoader->m_MatteColor = m_pImage->m_pImage->m_MatteColor;
1522         }
1523     }
1524     return ret;
1525 }
1526 FX_BOOL CPDF_ImageLoader::Load(const CPDF_ImageObject* pImage, CPDF_PageRenderCache* pCache, FX_BOOL bStdCS, FX_DWORD GroupFamily, FX_BOOL bLoadMask, CPDF_RenderStatus* pRenderStatus)
1527 {
1528     if (pImage == NULL) {
1529         return FALSE;
1530     }
1531     if (pCache) {
1532         pCache->GetCachedBitmap(pImage->m_pImage->GetStream(), m_pBitmap, m_pMask, m_MatteColor, bStdCS, GroupFamily, bLoadMask, pRenderStatus, m_nDownsampleWidth, m_nDownsampleHeight);
1533         m_bCached = TRUE;
1534     } else {
1535         m_pBitmap = pImage->m_pImage->LoadDIBSource(&m_pMask, &m_MatteColor, bStdCS, GroupFamily, bLoadMask);
1536         m_bCached = FALSE;
1537     }
1538     return FALSE;
1539 }
1540 FX_BOOL CPDF_ImageLoader::StartLoadImage(const CPDF_ImageObject* pImage, CPDF_PageRenderCache* pCache, void*& LoadHandle, FX_BOOL bStdCS, FX_DWORD GroupFamily, FX_BOOL bLoadMask, CPDF_RenderStatus* pRenderStatus, int32_t nDownsampleWidth, int32_t nDownsampleHeight)
1541 {
1542     m_nDownsampleWidth = nDownsampleWidth;
1543     m_nDownsampleHeight = nDownsampleHeight;
1544     CPDF_ProgressiveImageLoaderHandle* pLoaderHandle = new CPDF_ProgressiveImageLoaderHandle;
1545     FX_BOOL ret = pLoaderHandle->Start(this, pImage, pCache, bStdCS, GroupFamily, bLoadMask, pRenderStatus, m_nDownsampleWidth, m_nDownsampleHeight);
1546     LoadHandle = pLoaderHandle;
1547     return ret;
1548 }
1549 FX_BOOL CPDF_ImageLoader::Continue(void* LoadHandle, IFX_Pause* pPause)
1550 {
1551     return ((CPDF_ProgressiveImageLoaderHandle*)LoadHandle)->Continue(pPause);
1552 }
1553 CPDF_ImageLoader::~CPDF_ImageLoader()
1554 {
1555     if (!m_bCached) {
1556         delete m_pBitmap;
1557         delete m_pMask;
1558     }
1559 }