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