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