Stack-buffer-overflow in IccLib_Translate
[pdfium.git] / core / src / fpdfapi / fpdf_page / fpdf_page_colors.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/fpdfapi/fpdf_page.h"
8 #include "../../../include/fpdfapi/fpdf_module.h"
9 #include "../../../include/fxcodec/fx_codec.h"
10 #include "pageint.h"
11 #include <limits.h>
12 void sRGB_to_AdobeCMYK(FX_FLOAT R, FX_FLOAT G, FX_FLOAT B, FX_FLOAT& c, FX_FLOAT& m, FX_FLOAT& y, FX_FLOAT& k)
13 {
14     c = 1.0f - R;
15     m = 1.0f - G;
16     y = 1.0f - B;
17     k = c;
18     if (m < k) {
19         k = m;
20     }
21     if (y < k) {
22         k = y;
23     }
24 }
25 CPDF_DeviceCS::CPDF_DeviceCS(int family)
26 {
27     m_Family = family;
28     if (m_Family == PDFCS_DEVICERGB) {
29         m_nComponents = 3;
30     } else if (m_Family == PDFCS_DEVICEGRAY) {
31         m_nComponents = 1;
32     } else {
33         m_nComponents = 4;
34     }
35 }
36 FX_BOOL CPDF_DeviceCS::GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const
37 {
38     if (m_Family == PDFCS_DEVICERGB) {
39         R = pBuf[0];
40         if (R < 0) {
41             R = 0;
42         } else if (R > 1) {
43             R = 1;
44         }
45         G = pBuf[1];
46         if (G < 0) {
47             G = 0;
48         } else if (G > 1) {
49             G = 1;
50         }
51         B = pBuf[2];
52         if (B < 0) {
53             B = 0;
54         } else if (B > 1) {
55             B = 1;
56         }
57     } else if (m_Family == PDFCS_DEVICEGRAY) {
58         R = *pBuf;
59         if (R < 0) {
60             R = 0;
61         } else if (R > 1) {
62             R = 1;
63         }
64         G = B = R;
65     } else if (m_Family == PDFCS_DEVICECMYK) {
66         if (!m_dwStdConversion) {
67             AdobeCMYK_to_sRGB(pBuf[0], pBuf[1], pBuf[2], pBuf[3], R, G, B);
68         } else {
69             FX_FLOAT k = pBuf[3];
70             R = 1.0f - FX_MIN(1.0f, pBuf[0] + k);
71             G = 1.0f - FX_MIN(1.0f, pBuf[1] + k);
72             B = 1.0f - FX_MIN(1.0f, pBuf[2] + k);
73         }
74     } else {
75         ASSERT(m_Family == PDFCS_PATTERN);
76         R = G = B = 0;
77         return FALSE;
78     }
79     return TRUE;
80 }
81 FX_BOOL CPDF_DeviceCS::v_GetCMYK(FX_FLOAT* pBuf, FX_FLOAT& c, FX_FLOAT& m, FX_FLOAT& y, FX_FLOAT& k) const
82 {
83     if (m_Family != PDFCS_DEVICECMYK) {
84         return FALSE;
85     }
86     c = pBuf[0];
87     m = pBuf[1];
88     y = pBuf[2];
89     k = pBuf[3];
90     return TRUE;
91 }
92 FX_BOOL CPDF_DeviceCS::SetRGB(FX_FLOAT* pBuf, FX_FLOAT R, FX_FLOAT G, FX_FLOAT B) const
93 {
94     if (m_Family == PDFCS_DEVICERGB) {
95         pBuf[0] = R;
96         pBuf[1] = G;
97         pBuf[2] = B;
98         return TRUE;
99     } else if (m_Family == PDFCS_DEVICEGRAY) {
100         if (R == G && R == B) {
101             *pBuf = R;
102             return TRUE;
103         } else {
104             return FALSE;
105         }
106     } else if (m_Family == PDFCS_DEVICECMYK) {
107         sRGB_to_AdobeCMYK(R, G, B, pBuf[0], pBuf[1], pBuf[2], pBuf[3]);
108         return TRUE;
109     }
110     return FALSE;
111 }
112 FX_BOOL CPDF_DeviceCS::v_SetCMYK(FX_FLOAT* pBuf, FX_FLOAT c, FX_FLOAT m, FX_FLOAT y, FX_FLOAT k) const
113 {
114     if (m_Family == PDFCS_DEVICERGB) {
115         AdobeCMYK_to_sRGB(c, m, y, k, pBuf[0], pBuf[1], pBuf[2]);
116         return TRUE;
117     } else if (m_Family == PDFCS_DEVICEGRAY) {
118         return FALSE;
119     } else if (m_Family == PDFCS_DEVICECMYK) {
120         pBuf[0] = c;
121         pBuf[1] = m;
122         pBuf[2] = y;
123         pBuf[3] = k;
124         return TRUE;
125     }
126     return FALSE;
127 }
128 static void ReverseRGB(FX_LPBYTE pDestBuf, FX_LPCBYTE pSrcBuf, int pixels)
129 {
130     if (pDestBuf == pSrcBuf)
131         for (int i = 0; i < pixels; i ++) {
132             FX_BYTE temp = pDestBuf[2];
133             pDestBuf[2] = pDestBuf[0];
134             pDestBuf[0] = temp;
135             pDestBuf += 3;
136         }
137     else
138         for (int i = 0; i < pixels; i ++) {
139             *pDestBuf ++ = pSrcBuf[2];
140             *pDestBuf ++ = pSrcBuf[1];
141             *pDestBuf ++ = pSrcBuf[0];
142             pSrcBuf += 3;
143         }
144 }
145 void CPDF_DeviceCS::TranslateImageLine(FX_LPBYTE pDestBuf, FX_LPCBYTE pSrcBuf, int pixels, int image_width, int image_height, FX_BOOL bTransMask) const
146 {
147     if (bTransMask && m_Family == PDFCS_DEVICECMYK) {
148         for (int i = 0; i < pixels; i ++) {
149             int k = 255 - pSrcBuf[3];
150             pDestBuf[0] = ((255 - pSrcBuf[0]) * k) / 255;
151             pDestBuf[1] = ((255 - pSrcBuf[1]) * k) / 255;
152             pDestBuf[2] = ((255 - pSrcBuf[2]) * k) / 255;
153             pDestBuf += 3;
154             pSrcBuf += 4;
155         }
156         return;
157     }
158     if (m_Family == PDFCS_DEVICERGB) {
159         ReverseRGB(pDestBuf, pSrcBuf, pixels);
160     } else if (m_Family == PDFCS_DEVICEGRAY) {
161         for (int i = 0; i < pixels; i ++) {
162             *pDestBuf ++ = pSrcBuf[i];
163             *pDestBuf ++ = pSrcBuf[i];
164             *pDestBuf ++ = pSrcBuf[i];
165         }
166     } else {
167         for (int i = 0; i < pixels; i ++) {
168             if (!m_dwStdConversion) {
169                 AdobeCMYK_to_sRGB1(pSrcBuf[0], pSrcBuf[1], pSrcBuf[2], pSrcBuf[3], pDestBuf[2], pDestBuf[1], pDestBuf[0]);
170             } else {
171                 FX_BYTE k = pSrcBuf[3];
172                 pDestBuf[2] = 255 - FX_MIN(255, pSrcBuf[0] + k);
173                 pDestBuf[1] = 255 - FX_MIN(255, pSrcBuf[1] + k);
174                 pDestBuf[0] = 255 - FX_MIN(255, pSrcBuf[2] + k);
175             }
176             pSrcBuf += 4;
177             pDestBuf += 3;
178         }
179     }
180 }
181 const FX_BYTE g_sRGBSamples1[] = {
182     0,   3,   6,  10,  13,  15,  18,  20,  22,  23,  25,  27,  28,  30,  31,  32,
183     34,  35,  36,  37,  38,  39,  40,  41,  42,  43,  44,  45,  46,  47,  48,  49,
184     49,  50,  51,  52,  53,  53,  54,  55,  56,  56,  57,  58,  58,  59,  60,  61,
185     61,  62,  62,  63,  64,  64,  65,  66,  66,  67,  67,  68,  68,  69,  70,  70,
186     71,  71,  72,  72,  73,  73,  74,  74,  75,  76,  76,  77,  77,  78,  78,  79,
187     79,  79,  80,  80,  81,  81,  82,  82,  83,  83,  84,  84,  85,  85,  85,  86,
188     86,  87,  87,  88,  88,  88,  89,  89,  90,  90,  91,  91,  91,  92,  92,  93,
189     93,  93,  94,  94,  95,  95,  95,  96,  96,  97,  97,  97,  98,  98,  98,  99,
190     99,  99, 100, 100, 101, 101, 101, 102, 102, 102, 103, 103, 103, 104, 104, 104,
191     105, 105, 106, 106, 106, 107, 107, 107, 108, 108, 108, 109, 109, 109, 110, 110,
192     110, 110, 111, 111, 111, 112, 112, 112, 113, 113, 113, 114, 114, 114, 115, 115,
193     115, 115, 116, 116, 116, 117, 117, 117, 118, 118, 118, 118, 119, 119, 119, 120,
194 };
195 const FX_BYTE g_sRGBSamples2[] = {
196     120, 121, 122, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136,
197     137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 148, 149, 150, 151,
198     152, 153, 154, 155, 155, 156, 157, 158, 159, 159, 160, 161, 162, 163, 163, 164,
199     165, 166, 167, 167, 168, 169, 170, 170, 171, 172, 173, 173, 174, 175, 175, 176,
200     177, 178, 178, 179, 180, 180, 181, 182, 182, 183, 184, 185, 185, 186, 187, 187,
201     188, 189, 189, 190, 190, 191, 192, 192, 193, 194, 194, 195, 196, 196, 197, 197,
202     198, 199, 199, 200, 200, 201, 202, 202, 203, 203, 204, 205, 205, 206, 206, 207,
203     208, 208, 209, 209, 210, 210, 211, 212, 212, 213, 213, 214, 214, 215, 215, 216,
204     216, 217, 218, 218, 219, 219, 220, 220, 221, 221, 222, 222, 223, 223, 224, 224,
205     225, 226, 226, 227, 227, 228, 228, 229, 229, 230, 230, 231, 231, 232, 232, 233,
206     233, 234, 234, 235, 235, 236, 236, 237, 237, 238, 238, 238, 239, 239, 240, 240,
207     241, 241, 242, 242, 243, 243, 244, 244, 245, 245, 246, 246, 246, 247, 247, 248,
208     248, 249, 249, 250, 250, 251, 251, 251, 252, 252, 253, 253, 254, 254, 255, 255,
209 };
210 static void XYZ_to_sRGB(FX_FLOAT X, FX_FLOAT Y, FX_FLOAT Z, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B)
211 {
212     FX_FLOAT R1 = 3.2410f * X - 1.5374f * Y - 0.4986f * Z;
213     FX_FLOAT G1 = -0.9692f * X + 1.8760f * Y + 0.0416f * Z;
214     FX_FLOAT B1 =  0.0556f * X - 0.2040f * Y + 1.0570f * Z;
215     if (R1 > 1) {
216         R1 = 1;
217     }
218     if (R1 < 0) {
219         R1 = 0;
220     }
221     if (G1 > 1) {
222         G1 = 1;
223     }
224     if (G1 < 0) {
225         G1 = 0;
226     }
227     if (B1 > 1) {
228         B1 = 1;
229     }
230     if (B1 < 0) {
231         B1 = 0;
232     }
233     int scale = (int)(R1 * 1023);
234     if (scale < 0) {
235         scale = 0;
236     }
237     if (scale < 192) {
238         R = (g_sRGBSamples1[scale] / 255.0f);
239     } else {
240         R = (g_sRGBSamples2[scale / 4 - 48] / 255.0f);
241     }
242     scale = (int)(G1 * 1023);
243     if (scale < 0) {
244         scale = 0;
245     }
246     if (scale < 192) {
247         G = (g_sRGBSamples1[scale] / 255.0f);
248     } else {
249         G = (g_sRGBSamples2[scale / 4 - 48] / 255.0f);
250     }
251     scale = (int)(B1 * 1023);
252     if (scale < 0) {
253         scale = 0;
254     }
255     if (scale < 192) {
256         B = (g_sRGBSamples1[scale] / 255.0f);
257     } else {
258         B = (g_sRGBSamples2[scale / 4 - 48] / 255.0f);
259     }
260 }
261 class CPDF_CalGray : public CPDF_ColorSpace
262 {
263 public:
264     CPDF_CalGray();
265     virtual FX_BOOL             v_Load(CPDF_Document* pDoc, CPDF_Array* pArray);
266     virtual FX_BOOL             GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const;
267     virtual void                TranslateImageLine(FX_LPBYTE pDestBuf, FX_LPCBYTE pSrcBuf, int pixels, int image_width, int image_height, FX_BOOL bTransMask = FALSE) const;
268     FX_BOOL                             SetRGB(FX_FLOAT* pBuf, FX_FLOAT R, FX_FLOAT G, FX_FLOAT B) const;
269     FX_FLOAT                    m_WhitePoint[3];
270     FX_FLOAT                    m_BlackPoint[3];
271     FX_FLOAT                    m_Gamma;
272 };
273 CPDF_CalGray::CPDF_CalGray()
274 {
275     m_Family = PDFCS_CALGRAY;
276     m_nComponents = 1;
277 }
278 FX_BOOL CPDF_CalGray::v_Load(CPDF_Document* pDoc, CPDF_Array* pArray)
279 {
280     CPDF_Dictionary* pDict = pArray->GetDict(1);
281     CPDF_Array* pParam = pDict->GetArray(FX_BSTRC("WhitePoint"));
282     int i;
283     for (i = 0; i < 3; i ++) {
284         m_WhitePoint[i] = pParam->GetNumber(i);
285     }
286     pParam = pDict->GetArray(FX_BSTRC("BlackPoint"));
287     for (i = 0; i < 3; i ++) {
288         m_BlackPoint[i] = pParam ? pParam->GetNumber(i) : 0;
289     }
290     m_Gamma = pDict->GetNumber(FX_BSTRC("Gamma"));
291     if (m_Gamma == 0) {
292         m_Gamma = 1.0f;
293     }
294     return TRUE;
295 }
296 FX_BOOL CPDF_CalGray::GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const
297 {
298     R = G = B = *pBuf;
299     return TRUE;
300 }
301 FX_BOOL CPDF_CalGray::SetRGB(FX_FLOAT* pBuf, FX_FLOAT R, FX_FLOAT G, FX_FLOAT B) const
302 {
303     if (R == G && R == B) {
304         *pBuf = R;
305         return TRUE;
306     } else {
307         return FALSE;
308     }
309 }
310 void CPDF_CalGray::TranslateImageLine(FX_LPBYTE pDestBuf, FX_LPCBYTE pSrcBuf, int pixels, int image_width, int image_height, FX_BOOL bTransMask) const
311 {
312     for (int i = 0; i < pixels; i ++) {
313         *pDestBuf ++ = pSrcBuf[i];
314         *pDestBuf ++ = pSrcBuf[i];
315         *pDestBuf ++ = pSrcBuf[i];
316     }
317 }
318 class CPDF_CalRGB : public CPDF_ColorSpace
319 {
320 public:
321     CPDF_CalRGB();
322     virtual FX_BOOL             v_Load(CPDF_Document* pDoc, CPDF_Array* pArray);
323     virtual FX_BOOL             GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const;
324     virtual void                TranslateImageLine(FX_LPBYTE pDestBuf, FX_LPCBYTE pSrcBuf, int pixels, int image_width, int image_height, FX_BOOL bTransMask = FALSE) const;
325     FX_BOOL                             SetRGB(FX_FLOAT* pBuf, FX_FLOAT R, FX_FLOAT G, FX_FLOAT B) const;
326     FX_FLOAT                    m_WhitePoint[3];
327     FX_FLOAT                    m_BlackPoint[3];
328     FX_FLOAT                    m_Gamma[3];
329     FX_FLOAT                    m_Matrix[9];
330     FX_BOOL                             m_bGamma, m_bMatrix;
331 };
332 CPDF_CalRGB::CPDF_CalRGB()
333 {
334     m_Family = PDFCS_CALRGB;
335     m_nComponents = 3;
336 }
337 FX_BOOL CPDF_CalRGB::v_Load(CPDF_Document* pDoc, CPDF_Array* pArray)
338 {
339     CPDF_Dictionary* pDict = pArray->GetDict(1);
340     CPDF_Array* pParam = pDict->GetArray(FX_BSTRC("WhitePoint"));
341     int i;
342     for (i = 0; i < 3; i ++) {
343         m_WhitePoint[i] = pParam->GetNumber(i);
344     }
345     pParam = pDict->GetArray(FX_BSTRC("BlackPoint"));
346     for (i = 0; i < 3; i ++) {
347         m_BlackPoint[i] = pParam ? pParam->GetNumber(i) : 0;
348     }
349     pParam = pDict->GetArray(FX_BSTRC("Gamma"));
350     if (pParam) {
351         m_bGamma = TRUE;
352         for (i = 0; i < 3; i ++) {
353             m_Gamma[i] = pParam->GetNumber(i);
354         }
355     } else {
356         m_bGamma = FALSE;
357     }
358     pParam = pDict->GetArray(FX_BSTRC("Matrix"));
359     if (pParam) {
360         m_bMatrix = TRUE;
361         for (i = 0; i < 9; i ++) {
362             m_Matrix[i] = pParam->GetNumber(i);
363         }
364     } else {
365         m_bMatrix = FALSE;
366     }
367     return TRUE;
368 }
369 FX_BOOL CPDF_CalRGB::GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const
370 {
371     FX_FLOAT A_ = pBuf[0];
372     FX_FLOAT B_ = pBuf[1];
373     FX_FLOAT C_ = pBuf[2];
374     if (m_bGamma) {
375         A_ = (FX_FLOAT)FXSYS_pow(A_, m_Gamma[0]);
376         B_ = (FX_FLOAT)FXSYS_pow(B_, m_Gamma[1]);
377         C_ = (FX_FLOAT)FXSYS_pow(C_, m_Gamma[2]);
378     }
379     FX_FLOAT X, Y, Z;
380     if (m_bMatrix) {
381         X = m_Matrix[0] * A_ + m_Matrix[3] * B_ + m_Matrix[6] * C_;
382         Y = m_Matrix[1] * A_ + m_Matrix[4] * B_ + m_Matrix[7] * C_;
383         Z = m_Matrix[2] * A_ + m_Matrix[5] * B_ + m_Matrix[8] * C_;
384     } else {
385         X = A_;
386         Y = B_;
387         Z = C_;
388     }
389     XYZ_to_sRGB(X, Y, Z, R, G, B);
390     return TRUE;
391 }
392 FX_BOOL CPDF_CalRGB::SetRGB(FX_FLOAT* pBuf, FX_FLOAT R, FX_FLOAT G, FX_FLOAT B) const
393 {
394     pBuf[0] = R;
395     pBuf[1] = G;
396     pBuf[2] = B;
397     return TRUE;
398 }
399 void CPDF_CalRGB::TranslateImageLine(FX_LPBYTE pDestBuf, FX_LPCBYTE pSrcBuf, int pixels, int image_width, int image_height, FX_BOOL bTransMask) const
400 {
401     if (bTransMask) {
402         FX_FLOAT Cal[3];
403         FX_FLOAT R, G, B;
404         for(int i = 0; i < pixels; i ++) {
405             Cal[0] = ((FX_FLOAT)pSrcBuf[2]) / 255;
406             Cal[1] = ((FX_FLOAT)pSrcBuf[1]) / 255;
407             Cal[2] = ((FX_FLOAT)pSrcBuf[0]) / 255;
408             GetRGB(Cal, R, G, B);
409             pDestBuf[0] = FXSYS_round(B * 255);
410             pDestBuf[1] = FXSYS_round(G * 255);
411             pDestBuf[2] = FXSYS_round(R * 255);
412             pSrcBuf += 3;
413             pDestBuf += 3;
414         }
415     }
416     ReverseRGB(pDestBuf, pSrcBuf, pixels);
417 }
418 class CPDF_LabCS : public CPDF_ColorSpace
419 {
420 public:
421     CPDF_LabCS()
422     {
423         m_Family = PDFCS_LAB;
424         m_nComponents = 3;
425     }
426     virtual FX_BOOL             v_Load(CPDF_Document* pDoc, CPDF_Array* pArray);
427     virtual void                GetDefaultValue(int iComponent, FX_FLOAT& value, FX_FLOAT& min, FX_FLOAT& max) const;
428     virtual FX_BOOL             GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const;
429     FX_BOOL             SetRGB(FX_FLOAT* pBuf, FX_FLOAT R, FX_FLOAT G, FX_FLOAT B) const;
430     virtual void                TranslateImageLine(FX_LPBYTE pDestBuf, FX_LPCBYTE pSrcBuf, int pixels, int image_width, int image_height, FX_BOOL bTransMask = FALSE) const;
431     FX_FLOAT    m_WhitePoint[3];
432     FX_FLOAT    m_BlackPoint[3];
433     FX_FLOAT    m_Ranges[4];
434 };
435 FX_BOOL CPDF_LabCS::v_Load(CPDF_Document* pDoc, CPDF_Array* pArray)
436 {
437     CPDF_Dictionary* pDict = pArray->GetDict(1);
438     CPDF_Array* pParam = pDict->GetArray(FX_BSTRC("WhitePoint"));
439     int i;
440     for (i = 0; i < 3; i ++) {
441         m_WhitePoint[i] = pParam->GetNumber(i);
442     }
443     pParam = pDict->GetArray(FX_BSTRC("BlackPoint"));
444     for (i = 0; i < 3; i ++) {
445         m_BlackPoint[i] = pParam ? pParam->GetNumber(i) : 0;
446     }
447     pParam = pDict->GetArray(FX_BSTRC("Range"));
448     const FX_FLOAT def_ranges[4] = { -100 * 1.0f, 100 * 1.0f, -100 * 1.0f, 100 * 1.0f};
449     for (i = 0; i < 4; i ++) {
450         m_Ranges[i] = pParam ? pParam->GetNumber(i) : def_ranges[i];
451     }
452     return TRUE;
453 }
454 void CPDF_LabCS::GetDefaultValue(int iComponent, FX_FLOAT& value, FX_FLOAT& min, FX_FLOAT& max) const
455 {
456     value = 0;
457     if (iComponent == 0) {
458         min = 0;
459         max = 100 * 1.0f;
460     } else {
461         min = m_Ranges[iComponent * 2 - 2];
462         max = m_Ranges[iComponent * 2 - 1];
463         if (value < min) {
464             value = min;
465         } else if (value > max) {
466             value = max;
467         }
468     }
469 }
470 FX_BOOL CPDF_LabCS::GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const
471 {
472     FX_FLOAT Lstar = pBuf[0];
473     FX_FLOAT astar = pBuf[1];
474     FX_FLOAT bstar = pBuf[2];
475     FX_FLOAT M = (Lstar + 16.0f) / 116.0f;
476     FX_FLOAT L = M + astar / 500.0f;
477     FX_FLOAT N = M - bstar / 200.0f;
478     FX_FLOAT X, Y, Z;
479     if (L < 0.2069f) {
480         X = 0.957f * 0.12842f * (L - 0.1379f);
481     } else {
482         X = 0.957f * L * L * L;
483     }
484     if (M < 0.2069f) {
485         Y = 0.12842f * (M - 0.1379f);
486     } else {
487         Y = M * M * M;
488     }
489     if (N < 0.2069f) {
490         Z = 1.0889f * 0.12842f * (N - 0.1379f);
491     } else {
492         Z = 1.0889f * N * N * N;
493     }
494     XYZ_to_sRGB(X, Y, Z, R, G, B);
495     return TRUE;
496 }
497 FX_BOOL CPDF_LabCS::SetRGB(FX_FLOAT* pBuf, FX_FLOAT R, FX_FLOAT G, FX_FLOAT B) const
498 {
499     return FALSE;
500 }
501 void CPDF_LabCS::TranslateImageLine(FX_LPBYTE pDestBuf, FX_LPCBYTE pSrcBuf, int pixels, int image_width, int image_height, FX_BOOL bTransMask) const
502 {
503     for (int i = 0; i < pixels; i ++) {
504         FX_FLOAT lab[3];
505         FX_FLOAT R, G, B;
506         lab[0] = (pSrcBuf[0] * 100 / 255.0f);
507         lab[1] = (FX_FLOAT)(pSrcBuf[1] - 128);
508         lab[2] = (FX_FLOAT)(pSrcBuf[2] - 128);
509         GetRGB(lab, R, G, B);
510         pDestBuf[0] = (FX_INT32)(B * 255);
511         pDestBuf[1] = (FX_INT32)(G * 255);
512         pDestBuf[2] = (FX_INT32)(R * 255);
513         pDestBuf += 3;
514         pSrcBuf += 3;
515     }
516 }
517 CPDF_IccProfile::CPDF_IccProfile(FX_LPCBYTE pData, FX_DWORD dwSize, int nComponents)
518 {
519     m_bsRGB = nComponents == 3 && dwSize == 3144 && FXSYS_memcmp32(pData + 0x190, "sRGB IEC61966-2.1", 17) == 0;
520     m_pTransform = NULL;
521     if (!m_bsRGB && CPDF_ModuleMgr::Get()->GetIccModule()) {
522         m_pTransform = CPDF_ModuleMgr::Get()->GetIccModule()->CreateTransform_sRGB(pData, dwSize, nComponents);
523     }
524 }
525 CPDF_IccProfile::~CPDF_IccProfile()
526 {
527     if (m_pTransform) {
528         CPDF_ModuleMgr::Get()->GetIccModule()->DestroyTransform(m_pTransform);
529     }
530 }
531 class CPDF_ICCBasedCS : public CPDF_ColorSpace
532 {
533 public:
534     CPDF_ICCBasedCS();
535     virtual ~CPDF_ICCBasedCS();
536     virtual FX_BOOL             v_Load(CPDF_Document* pDoc, CPDF_Array* pArray);
537     void                                GetDefaultValue(int i, FX_FLOAT& min, FX_FLOAT& max) const
538     {
539         min = m_pRanges[i * 2];
540         max = m_pRanges[i * 2 + 1];
541     }
542     virtual FX_BOOL             GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const;
543     FX_BOOL                             v_GetCMYK(FX_FLOAT* pBuf, FX_FLOAT& c, FX_FLOAT& m, FX_FLOAT& y, FX_FLOAT& k) const;
544     FX_BOOL                             SetRGB(FX_FLOAT* pBuf, FX_FLOAT R, FX_FLOAT G, FX_FLOAT B) const;
545     virtual void                EnableStdConversion(FX_BOOL bEnabled);
546     virtual void                TranslateImageLine(FX_LPBYTE pDestBuf, FX_LPCBYTE pSrcBuf, int pixels, int image_width, int image_height, FX_BOOL bTransMask = FALSE) const;
547     FX_FLOAT*           m_pRanges;
548     CPDF_IccProfile*    m_pProfile;
549     CPDF_ColorSpace*    m_pAlterCS;
550     FX_LPBYTE                   m_pCache;
551     FX_BOOL                             m_bOwn;
552 };
553 CPDF_ICCBasedCS::CPDF_ICCBasedCS()
554 {
555     m_pAlterCS = NULL;
556     m_pProfile = NULL;
557     m_Family = PDFCS_ICCBASED;
558     m_pCache = NULL;
559     m_pRanges = NULL;
560     m_bOwn = FALSE;
561 }
562 CPDF_ICCBasedCS::~CPDF_ICCBasedCS()
563 {
564     if (m_pCache) {
565         FX_Free(m_pCache);
566     }
567     if (m_pRanges) {
568         FX_Free(m_pRanges);
569     }
570     if (m_pAlterCS && m_bOwn) {
571         m_pAlterCS->ReleaseCS();
572     }
573     if (m_pProfile && m_pDocument) {
574         m_pDocument->GetPageData()->ReleaseIccProfile(NULL, m_pProfile);
575     }
576 }
577 FX_BOOL CPDF_ICCBasedCS::v_Load(CPDF_Document* pDoc, CPDF_Array* pArray)
578 {
579     CPDF_Stream* pStream = pArray->GetStream(1);
580     if (pStream == NULL) {
581         return FALSE;
582     }
583     m_nComponents = pStream->GetDict()->GetInteger(FX_BSTRC("N"));
584     if (m_nComponents < 0 || m_nComponents > (1 << 16)) {
585         return FALSE;
586     }
587     CPDF_Array* pRanges = pStream->GetDict()->GetArray(FX_BSTRC("Range"));
588     m_pRanges = FX_Alloc(FX_FLOAT, m_nComponents * 2);
589     for (int i = 0; i < m_nComponents * 2; i ++) {
590         if (pRanges) {
591             m_pRanges[i] = pRanges->GetNumber(i);
592         } else if (i % 2) {
593             m_pRanges[i] = 1.0f;
594         } else {
595             m_pRanges[i] = 0;
596         }
597     }
598     m_pProfile = pDoc->LoadIccProfile(pStream, m_nComponents);
599     if (!m_pProfile) {
600         return FALSE;
601     }
602     if (m_pProfile->m_pTransform == NULL) {
603         CPDF_Object* pAlterCSObj = pStream->GetDict()->GetElementValue(FX_BSTRC("Alternate"));
604         if (pAlterCSObj) {
605             CPDF_ColorSpace* alter_cs = CPDF_ColorSpace::Load(pDoc, pAlterCSObj);
606             if (alter_cs) {
607                 if (alter_cs->CountComponents() > m_nComponents) {
608                     alter_cs->ReleaseCS();
609                 } else {
610                     m_pAlterCS = alter_cs;
611                     m_bOwn = TRUE;
612                 }
613             }
614         }
615         if (!m_pAlterCS) {
616             if (m_nComponents == 3) {
617                 m_pAlterCS = GetStockCS(PDFCS_DEVICERGB);
618             } else if (m_nComponents == 4) {
619                 m_pAlterCS = GetStockCS(PDFCS_DEVICECMYK);
620             } else {
621                 m_pAlterCS = GetStockCS(PDFCS_DEVICEGRAY);
622             }
623         }
624     }
625     return TRUE;
626 }
627 FX_BOOL CPDF_ICCBasedCS::GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const
628 {
629     if (m_pProfile && m_pProfile->m_bsRGB) {
630         R = pBuf[0];
631         G = pBuf[1];
632         B = pBuf[2];
633         return TRUE;
634     }
635     ICodec_IccModule *pIccModule = CPDF_ModuleMgr::Get()->GetIccModule();
636     if (m_pProfile->m_pTransform == NULL || pIccModule == NULL) {
637         if (m_pAlterCS) {
638             m_pAlterCS->GetRGB(pBuf, R, G, B);
639         } else {
640             R = G = B = 0.0f;
641         }
642         return TRUE;
643     }
644     FX_FLOAT rgb[3];
645     pIccModule->SetComponents(m_nComponents);
646     pIccModule->Translate(m_pProfile->m_pTransform, pBuf, rgb);
647     R = rgb[0];
648     G = rgb[1];
649     B = rgb[2];
650     return TRUE;
651 }
652 FX_BOOL CPDF_ICCBasedCS::v_GetCMYK(FX_FLOAT* pBuf, FX_FLOAT& c, FX_FLOAT& m, FX_FLOAT& y, FX_FLOAT& k) const
653 {
654     if (m_nComponents != 4) {
655         return FALSE;
656     }
657     c = pBuf[0];
658     m = pBuf[1];
659     y = pBuf[2];
660     k = pBuf[3];
661     return TRUE;
662 }
663 FX_BOOL CPDF_ICCBasedCS::SetRGB(FX_FLOAT* pBuf, FX_FLOAT R, FX_FLOAT G, FX_FLOAT B) const
664 {
665     return FALSE;
666 }
667 void CPDF_ICCBasedCS::EnableStdConversion(FX_BOOL bEnabled)
668 {
669     CPDF_ColorSpace::EnableStdConversion(bEnabled);
670     if (m_pAlterCS) {
671         m_pAlterCS->EnableStdConversion(bEnabled);
672     }
673 }
674 void CPDF_ICCBasedCS::TranslateImageLine(FX_LPBYTE pDestBuf, FX_LPCBYTE pSrcBuf, int pixels, int image_width, int image_height, FX_BOOL bTransMask) const
675 {
676     if (m_pProfile->m_bsRGB) {
677         ReverseRGB(pDestBuf, pSrcBuf, pixels);
678     } else if (m_pProfile->m_pTransform) {
679         int nMaxColors = 1;
680         for (int i = 0; i < m_nComponents; i ++) {
681             nMaxColors *= 52;
682         }
683         if (m_nComponents > 3 || image_width * image_height < nMaxColors * 3 / 2) {
684             CPDF_ModuleMgr::Get()->GetIccModule()->TranslateScanline(m_pProfile->m_pTransform, pDestBuf, pSrcBuf, pixels);
685         } else {
686             if (m_pCache == NULL) {
687                 ((CPDF_ICCBasedCS*)this)->m_pCache = FX_Alloc(FX_BYTE, nMaxColors * 3);
688                 FX_LPBYTE temp_src = FX_Alloc(FX_BYTE, nMaxColors * m_nComponents);
689                 FX_LPBYTE pSrc = temp_src;
690                 for (int i = 0; i < nMaxColors; i ++) {
691                     FX_DWORD color = i;
692                     FX_DWORD order = nMaxColors / 52;
693                     for (int c = 0; c < m_nComponents; c ++) {
694                         *pSrc++ = (FX_BYTE)(color / order * 5);
695                         color %= order;
696                         order /= 52;
697                     }
698                 }
699                 CPDF_ModuleMgr::Get()->GetIccModule()->TranslateScanline(m_pProfile->m_pTransform, m_pCache, temp_src, nMaxColors);
700                 FX_Free(temp_src);
701             }
702             for (int i = 0; i < pixels; i ++) {
703                 int index = 0;
704                 for (int c = 0; c < m_nComponents; c ++) {
705                     index = index * 52 + (*pSrcBuf) / 5;
706                     pSrcBuf ++;
707                 }
708                 index *= 3;
709                 *pDestBuf++ = m_pCache[index];
710                 *pDestBuf++ = m_pCache[index + 1];
711                 *pDestBuf++ = m_pCache[index + 2];
712             }
713         }
714     } else if (m_pAlterCS) {
715         m_pAlterCS->TranslateImageLine(pDestBuf, pSrcBuf, pixels, image_width, image_height);
716     }
717 }
718 class CPDF_IndexedCS : public CPDF_ColorSpace
719 {
720 public:
721     CPDF_IndexedCS();
722     virtual ~CPDF_IndexedCS();
723     virtual FX_BOOL             v_Load(CPDF_Document* pDoc, CPDF_Array* pArray);
724     void                                GetDefaultValue(int iComponent, FX_FLOAT& min, FX_FLOAT& max) const
725     {
726         min = 0;
727         max = (FX_FLOAT)m_MaxIndex;
728     }
729     virtual FX_BOOL             GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const;
730     virtual CPDF_ColorSpace*    GetBaseCS() const
731     {
732         return m_pBaseCS;
733     }
734     virtual void                EnableStdConversion(FX_BOOL bEnabled);
735     CPDF_ColorSpace*    m_pBaseCS;
736     int                                 m_nBaseComponents;
737     int                                 m_MaxIndex;
738     CFX_ByteString              m_Table;
739     FX_FLOAT*           m_pCompMinMax;
740 };
741 CPDF_IndexedCS::CPDF_IndexedCS()
742 {
743     m_pBaseCS = NULL;
744     m_Family = PDFCS_INDEXED;
745     m_nComponents = 1;
746     m_pCompMinMax = NULL;
747 }
748 CPDF_IndexedCS::~CPDF_IndexedCS()
749 {
750     if (m_pCompMinMax) {
751         FX_Free(m_pCompMinMax);
752     }
753     CPDF_ColorSpace* pCS = m_pBaseCS;
754     if (pCS && m_pDocument) {
755         m_pDocument->GetPageData()->ReleaseColorSpace(pCS->GetArray());
756     }
757 }
758 FX_BOOL CPDF_IndexedCS::v_Load(CPDF_Document* pDoc, CPDF_Array* pArray)
759 {
760     if (pArray->GetCount() < 4) {
761         return FALSE;
762     }
763     CPDF_Object* pBaseObj = pArray->GetElementValue(1);
764     if (pBaseObj == m_pArray) {
765         return FALSE;
766     }
767     CPDF_DocPageData* pDocPageData = pDoc->GetPageData();
768     m_pBaseCS = pDocPageData->GetColorSpace(pBaseObj, NULL);
769     if (m_pBaseCS == NULL) {
770         return FALSE;
771     }
772     m_nBaseComponents = m_pBaseCS->CountComponents();
773     m_pCompMinMax = FX_Alloc(FX_FLOAT, m_nBaseComponents * 2);
774     FX_FLOAT defvalue;
775     for (int i = 0; i < m_nBaseComponents; i ++) {
776         m_pBaseCS->GetDefaultValue(i, defvalue, m_pCompMinMax[i * 2], m_pCompMinMax[i * 2 + 1]);
777         m_pCompMinMax[i * 2 + 1] -= m_pCompMinMax[i * 2];
778     }
779     m_MaxIndex = pArray->GetInteger(2);
780     CPDF_Object* pTableObj = pArray->GetElementValue(3);
781     if (pTableObj == NULL) {
782         return FALSE;
783     }
784     FX_LPCBYTE pTable = NULL;
785     FX_DWORD size = 0;
786     CPDF_StreamAcc* pStreamAcc = NULL;
787     if (pTableObj->GetType() == PDFOBJ_STRING) {
788         m_Table = ((CPDF_String*)pTableObj)->GetString();
789     } else if (pTableObj->GetType() == PDFOBJ_STREAM) {
790         CPDF_StreamAcc acc;
791         acc.LoadAllData((CPDF_Stream*)pTableObj, FALSE);
792         m_Table = CFX_ByteStringC(acc.GetData(), acc.GetSize());
793     }
794     return TRUE;
795 }
796 FX_BOOL CPDF_IndexedCS::GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const
797 {
798     int index = (FX_INT32)(*pBuf);
799     if (index < 0 || index > m_MaxIndex) {
800         return FALSE;
801     }
802     if (m_nBaseComponents) {
803         if (index == INT_MAX || (index + 1) > INT_MAX / m_nBaseComponents ||
804                 (index + 1)*m_nBaseComponents > (int)m_Table.GetLength()) {
805             R = G = B = 0;
806             return FALSE;
807         }
808     }
809     CFX_FixedBufGrow<FX_FLOAT, 16> Comps(m_nBaseComponents);
810     FX_FLOAT* comps = Comps;
811     FX_LPCBYTE pTable = m_Table;
812     for (int i = 0; i < m_nBaseComponents; i ++) {
813         comps[i] = m_pCompMinMax[i * 2] + m_pCompMinMax[i * 2 + 1] * pTable[index * m_nBaseComponents + i] / 255;
814     }
815     m_pBaseCS->GetRGB(comps, R, G, B);
816     return TRUE;
817 }
818 void CPDF_IndexedCS::EnableStdConversion(FX_BOOL bEnabled)
819 {
820     CPDF_ColorSpace::EnableStdConversion(bEnabled);
821     if (m_pBaseCS) {
822         m_pBaseCS->EnableStdConversion(bEnabled);
823     }
824 }
825 #define MAX_PATTERN_COLORCOMPS  16
826 typedef struct _PatternValue {
827     CPDF_Pattern*       m_pPattern;
828     int                         m_nComps;
829     FX_FLOAT            m_Comps[MAX_PATTERN_COLORCOMPS];
830 } PatternValue;
831 CPDF_PatternCS::CPDF_PatternCS()
832 {
833     m_Family = PDFCS_PATTERN;
834     m_pBaseCS = NULL;
835     m_nComponents = 1;
836 }
837 CPDF_PatternCS::~CPDF_PatternCS()
838 {
839     CPDF_ColorSpace* pCS = m_pBaseCS;
840     if (pCS && m_pDocument) {
841         m_pDocument->GetPageData()->ReleaseColorSpace(pCS->GetArray());
842     }
843 }
844 FX_BOOL CPDF_PatternCS::v_Load(CPDF_Document* pDoc, CPDF_Array* pArray)
845 {
846     CPDF_Object* pBaseCS = pArray->GetElementValue(1);
847     if (pBaseCS == m_pArray) {
848         return FALSE;
849     }
850     CPDF_DocPageData* pDocPageData = pDoc->GetPageData();
851     m_pBaseCS = pDocPageData->GetColorSpace(pBaseCS, NULL);
852     if (m_pBaseCS) {
853         m_nComponents = m_pBaseCS->CountComponents() + 1;
854         if (m_pBaseCS->CountComponents() > MAX_PATTERN_COLORCOMPS) {
855             return FALSE;
856         }
857     } else {
858         m_nComponents = 1;
859     }
860     return TRUE;
861 }
862 FX_BOOL CPDF_PatternCS::GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const
863 {
864     if (m_pBaseCS) {
865         PatternValue* pvalue = (PatternValue*)pBuf;
866         m_pBaseCS->GetRGB(pvalue->m_Comps, R, G, B);
867         return TRUE;
868     }
869     R = G = B = 0.75f;
870     return FALSE;
871 }
872 class CPDF_SeparationCS : public CPDF_ColorSpace
873 {
874 public:
875     CPDF_SeparationCS();
876     virtual ~CPDF_SeparationCS();
877     virtual void                GetDefaultValue(int iComponent, FX_FLOAT& value, FX_FLOAT& min, FX_FLOAT& max) const
878     {
879         value = 1.0f;
880         min = 0;
881         max = 1.0f;
882     }
883     virtual FX_BOOL             v_Load(CPDF_Document* pDoc, CPDF_Array* pArray);
884     virtual FX_BOOL             GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const;
885     virtual void                EnableStdConversion(FX_BOOL bEnabled);
886     CPDF_ColorSpace*    m_pAltCS;
887     CPDF_Function*              m_pFunc;
888     enum {None, All, Colorant}  m_Type;
889 };
890 CPDF_SeparationCS::CPDF_SeparationCS()
891 {
892     m_Family = PDFCS_SEPARATION;
893     m_pAltCS = NULL;
894     m_pFunc = NULL;
895     m_nComponents = 1;
896 }
897 CPDF_SeparationCS::~CPDF_SeparationCS()
898 {
899     if (m_pAltCS) {
900         m_pAltCS->ReleaseCS();
901     }
902     if (m_pFunc) {
903         delete m_pFunc;
904     }
905 }
906 FX_BOOL CPDF_SeparationCS::v_Load(CPDF_Document* pDoc, CPDF_Array* pArray)
907 {
908     CFX_ByteString name = pArray->GetString(1);
909     if (name == FX_BSTRC("None")) {
910         m_Type = None;
911     } else {
912         m_Type = Colorant;
913         CPDF_Object* pAltCS = pArray->GetElementValue(2);
914         if (pAltCS == m_pArray) {
915             return FALSE;
916         }
917         m_pAltCS = Load(pDoc, pAltCS);
918         CPDF_Object* pFuncObj = pArray->GetElementValue(3);
919         if (pFuncObj && pFuncObj->GetType() != PDFOBJ_NAME) {
920             m_pFunc = CPDF_Function::Load(pFuncObj);
921         }
922         if (m_pFunc && m_pAltCS && m_pFunc->CountOutputs() < m_pAltCS->CountComponents()) {
923             delete m_pFunc;
924             m_pFunc = NULL;
925         }
926     }
927     return TRUE;
928 }
929 FX_BOOL CPDF_SeparationCS::GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const
930 {
931     if (m_Type == None) {
932         return FALSE;
933     }
934     if (m_pFunc == NULL) {
935         if (m_pAltCS == NULL) {
936             return FALSE;
937         }
938         int nComps = m_pAltCS->CountComponents();
939         CFX_FixedBufGrow<FX_FLOAT, 16> results(nComps);
940         for (int i = 0; i < nComps; i ++) {
941             results[i] = *pBuf;
942         }
943         m_pAltCS->GetRGB(results, R, G, B);
944         return TRUE;
945     }
946     CFX_FixedBufGrow<FX_FLOAT, 16> results(m_pFunc->CountOutputs());
947     int nresults;
948     m_pFunc->Call(pBuf, 1, results, nresults);
949     if (nresults == 0) {
950         return FALSE;
951     }
952     if (m_pAltCS) {
953         m_pAltCS->GetRGB(results, R, G, B);
954         return TRUE;
955     } else {
956         R = G = B = 0;
957         return FALSE;
958     }
959 }
960 void CPDF_SeparationCS::EnableStdConversion(FX_BOOL bEnabled)
961 {
962     CPDF_ColorSpace::EnableStdConversion(bEnabled);
963     if (m_pAltCS) {
964         m_pAltCS->EnableStdConversion(bEnabled);
965     }
966 }
967 class CPDF_DeviceNCS : public CPDF_ColorSpace
968 {
969 public:
970     CPDF_DeviceNCS();
971     virtual ~CPDF_DeviceNCS();
972     virtual void                GetDefaultValue(int iComponent, FX_FLOAT& value, FX_FLOAT& min, FX_FLOAT& max) const
973     {
974         value = 1.0f;
975         min = 0;
976         max = 1.0f;
977     }
978     virtual FX_BOOL     v_Load(CPDF_Document* pDoc, CPDF_Array* pArray);
979     virtual FX_BOOL     GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const;
980     virtual void        EnableStdConversion(FX_BOOL bEnabled);
981     CPDF_ColorSpace*    m_pAltCS;
982     CPDF_Function*              m_pFunc;
983 };
984 CPDF_DeviceNCS::CPDF_DeviceNCS()
985 {
986     m_Family = PDFCS_DEVICEN;
987     m_pAltCS = NULL;
988     m_pFunc = NULL;
989 }
990 CPDF_DeviceNCS::~CPDF_DeviceNCS()
991 {
992     if (m_pFunc) {
993         delete m_pFunc;
994     }
995     if (m_pAltCS) {
996         m_pAltCS->ReleaseCS();
997     }
998 }
999 FX_BOOL CPDF_DeviceNCS::v_Load(CPDF_Document* pDoc, CPDF_Array* pArray)
1000 {
1001     CPDF_Object* pObj = pArray->GetElementValue(1);
1002     if (!pObj) {
1003         return FALSE;
1004     }
1005     if (pObj->GetType() != PDFOBJ_ARRAY) {
1006         return FALSE;
1007     }
1008     m_nComponents = ((CPDF_Array*)pObj)->GetCount();
1009     CPDF_Object* pAltCS = pArray->GetElementValue(2);
1010     if (!pAltCS || pAltCS == m_pArray) {
1011         return FALSE;
1012     }
1013     m_pAltCS = Load(pDoc, pAltCS);
1014     m_pFunc = CPDF_Function::Load(pArray->GetElementValue(3));
1015     if (m_pAltCS == NULL || m_pFunc == NULL) {
1016         return FALSE;
1017     }
1018     if (m_pFunc->CountOutputs() < m_pAltCS->CountComponents()) {
1019         return FALSE;
1020     }
1021     return TRUE;
1022 }
1023 FX_BOOL CPDF_DeviceNCS::GetRGB(FX_FLOAT* pBuf, FX_FLOAT& R, FX_FLOAT& G, FX_FLOAT& B) const
1024 {
1025     if (m_pFunc == NULL) {
1026         return FALSE;
1027     }
1028     CFX_FixedBufGrow<FX_FLOAT, 16> results(m_pFunc->CountOutputs());
1029     int nresults;
1030     m_pFunc->Call(pBuf, m_nComponents, results, nresults);
1031     if (nresults == 0) {
1032         return FALSE;
1033     }
1034     m_pAltCS->GetRGB(results, R, G, B);
1035     return TRUE;
1036 }
1037 void CPDF_DeviceNCS::EnableStdConversion(FX_BOOL bEnabled)
1038 {
1039     CPDF_ColorSpace::EnableStdConversion(bEnabled);
1040     if (m_pAltCS) {
1041         m_pAltCS->EnableStdConversion(bEnabled);
1042     }
1043 }
1044 CPDF_ColorSpace* CPDF_ColorSpace::GetStockCS(int family)
1045 {
1046     return CPDF_ModuleMgr::Get()->GetPageModule()->GetStockCS(family);;
1047 }
1048 CPDF_ColorSpace* _CSFromName(const CFX_ByteString& name)
1049 {
1050     if (name == FX_BSTRC("DeviceRGB") || name == FX_BSTRC("RGB")) {
1051         return CPDF_ColorSpace::GetStockCS(PDFCS_DEVICERGB);
1052     }
1053     if (name == FX_BSTRC("DeviceGray") || name == FX_BSTRC("G")) {
1054         return CPDF_ColorSpace::GetStockCS(PDFCS_DEVICEGRAY);
1055     }
1056     if (name == FX_BSTRC("DeviceCMYK") || name == FX_BSTRC("CMYK")) {
1057         return CPDF_ColorSpace::GetStockCS(PDFCS_DEVICECMYK);
1058     }
1059     if (name == FX_BSTRC("Pattern")) {
1060         return CPDF_ColorSpace::GetStockCS(PDFCS_PATTERN);
1061     }
1062     return NULL;
1063 }
1064 CPDF_ColorSpace* CPDF_ColorSpace::Load(CPDF_Document* pDoc, CPDF_Object* pObj)
1065 {
1066     if (pObj == NULL) {
1067         return NULL;
1068     }
1069     if (pObj->GetType() == PDFOBJ_NAME) {
1070         return _CSFromName(pObj->GetString());
1071     }
1072     if (pObj->GetType() == PDFOBJ_STREAM) {
1073         CPDF_Dictionary *pDict = ((CPDF_Stream *)pObj)->GetDict();
1074         if (!pDict) {
1075             return NULL;
1076         }
1077         CPDF_ColorSpace *pRet = NULL;
1078         FX_POSITION pos = pDict->GetStartPos();
1079         while (pos) {
1080             CFX_ByteString bsKey;
1081             CPDF_Object *pValue = pDict->GetNextElement(pos, bsKey);
1082             if (pValue->GetType() == PDFOBJ_NAME) {
1083                 pRet = _CSFromName(pValue->GetString());
1084             }
1085             if (pRet) {
1086                 return pRet;
1087             }
1088         }
1089         return NULL;
1090     }
1091     if (pObj->GetType() != PDFOBJ_ARRAY) {
1092         return NULL;
1093     }
1094     CPDF_Array* pArray = (CPDF_Array*)pObj;
1095     if (pArray->GetCount() == 0) {
1096         return NULL;
1097     }
1098     CFX_ByteString familyname = pArray->GetElementValue(0)->GetString();
1099     if (pArray->GetCount() == 1) {
1100         return _CSFromName(familyname);
1101     }
1102     CPDF_ColorSpace* pCS = NULL;
1103     FX_DWORD id = familyname.GetID();
1104     if (id == FXBSTR_ID('C', 'a', 'l', 'G')) {
1105         pCS = FX_NEW CPDF_CalGray();
1106     } else if (id == FXBSTR_ID('C', 'a', 'l', 'R')) {
1107         pCS = FX_NEW CPDF_CalRGB();
1108     } else if (id == FXBSTR_ID('L', 'a', 'b', 0)) {
1109         pCS = FX_NEW CPDF_LabCS();
1110     } else if (id == FXBSTR_ID('I', 'C', 'C', 'B')) {
1111         pCS = FX_NEW CPDF_ICCBasedCS();
1112     } else if (id == FXBSTR_ID('I', 'n', 'd', 'e') || id == FXBSTR_ID('I', 0, 0, 0)) {
1113         pCS = FX_NEW CPDF_IndexedCS();
1114     } else if (id == FXBSTR_ID('S', 'e', 'p', 'a')) {
1115         pCS = FX_NEW CPDF_SeparationCS();
1116     } else if (id == FXBSTR_ID('D', 'e', 'v', 'i')) {
1117         pCS = FX_NEW CPDF_DeviceNCS();
1118     } else if (id == FXBSTR_ID('P', 'a', 't', 't')) {
1119         pCS = FX_NEW CPDF_PatternCS();
1120     } else {
1121         return NULL;
1122     }
1123     pCS->m_pDocument = pDoc;
1124     pCS->m_pArray = pArray;
1125     if (!pCS->v_Load(pDoc, pArray)) {
1126         pCS->ReleaseCS();
1127         return NULL;
1128     }
1129     return pCS;
1130 }
1131 CPDF_ColorSpace::CPDF_ColorSpace()
1132 {
1133     m_Family = 0;
1134     m_pArray = NULL;
1135     m_dwStdConversion = 0;
1136     m_pDocument = NULL;
1137 }
1138 void CPDF_ColorSpace::ReleaseCS()
1139 {
1140     if (this == GetStockCS(PDFCS_DEVICERGB)) {
1141         return;
1142     }
1143     if (this == GetStockCS(PDFCS_DEVICEGRAY)) {
1144         return;
1145     }
1146     if (this == GetStockCS(PDFCS_DEVICECMYK)) {
1147         return;
1148     }
1149     if (this == GetStockCS(PDFCS_PATTERN)) {
1150         return;
1151     }
1152     delete this;
1153 }
1154 int CPDF_ColorSpace::GetBufSize() const
1155 {
1156     if (m_Family == PDFCS_PATTERN) {
1157         return sizeof(PatternValue);
1158     }
1159     return m_nComponents * sizeof(FX_FLOAT);
1160 }
1161 FX_FLOAT* CPDF_ColorSpace::CreateBuf()
1162 {
1163     int size = GetBufSize();
1164     FX_BYTE* pBuf = FX_Alloc(FX_BYTE, size);
1165     FXSYS_memset32(pBuf, 0, size);
1166     return (FX_FLOAT*)pBuf;
1167 }
1168 FX_BOOL CPDF_ColorSpace::sRGB() const
1169 {
1170     if (m_Family == PDFCS_DEVICERGB) {
1171         return TRUE;
1172     }
1173     if (m_Family != PDFCS_ICCBASED) {
1174         return FALSE;
1175     }
1176     CPDF_ICCBasedCS* pCS = (CPDF_ICCBasedCS*)this;
1177     return pCS->m_pProfile->m_bsRGB;
1178 }
1179 FX_BOOL CPDF_ColorSpace::GetCMYK(FX_FLOAT* pBuf, FX_FLOAT& c, FX_FLOAT& m, FX_FLOAT& y, FX_FLOAT& k) const
1180 {
1181     if (v_GetCMYK(pBuf, c, m, y, k)) {
1182         return TRUE;
1183     }
1184     FX_FLOAT R, G, B;
1185     if (!GetRGB(pBuf, R, G, B)) {
1186         return FALSE;
1187     }
1188     sRGB_to_AdobeCMYK(R, G, B, c, m, y, k);
1189     return TRUE;
1190 }
1191 FX_BOOL CPDF_ColorSpace::SetCMYK(FX_FLOAT* pBuf, FX_FLOAT c, FX_FLOAT m, FX_FLOAT y, FX_FLOAT k) const
1192 {
1193     if (v_SetCMYK(pBuf, c, m, y, k)) {
1194         return TRUE;
1195     }
1196     FX_FLOAT R, G, B;
1197     AdobeCMYK_to_sRGB(c, m, y, k, R, G, B);
1198     return SetRGB(pBuf, R, G, B);
1199 }
1200 void CPDF_ColorSpace::GetDefaultColor(FX_FLOAT* buf) const
1201 {
1202     if (buf == NULL || m_Family == PDFCS_PATTERN) {
1203         return;
1204     }
1205     FX_FLOAT min, max;
1206     for (int i = 0; i < m_nComponents; i ++) {
1207         GetDefaultValue(i, buf[i], min, max);
1208     }
1209 }
1210 int CPDF_ColorSpace::GetMaxIndex() const
1211 {
1212     if (m_Family != PDFCS_INDEXED) {
1213         return 0;
1214     }
1215     CPDF_IndexedCS* pCS = (CPDF_IndexedCS*)this;
1216     return pCS->m_MaxIndex;
1217 }
1218 void CPDF_ColorSpace::TranslateImageLine(FX_LPBYTE dest_buf, FX_LPCBYTE src_buf, int pixels, int image_width, int image_height, FX_BOOL bTransMask) const
1219 {
1220     CFX_FixedBufGrow<FX_FLOAT, 16> srcbuf(m_nComponents);
1221     FX_FLOAT* src = srcbuf;
1222     FX_FLOAT R, G, B;
1223     for (int i = 0; i < pixels; i ++) {
1224         for (int j = 0; j < m_nComponents; j ++)
1225             if (m_Family == PDFCS_INDEXED) {
1226                 src[j] = (FX_FLOAT)(*src_buf ++);
1227             } else {
1228                 src[j] = (FX_FLOAT)(*src_buf ++) / 255;
1229             }
1230         GetRGB(src, R, G, B);
1231         *dest_buf ++ = (FX_INT32)(B * 255);
1232         *dest_buf ++ = (FX_INT32)(G * 255);
1233         *dest_buf ++ = (FX_INT32)(R * 255);
1234     }
1235 }
1236 void CPDF_ColorSpace::EnableStdConversion(FX_BOOL bEnabled)
1237 {
1238     if (bEnabled) {
1239         m_dwStdConversion ++;
1240     } else if (m_dwStdConversion) {
1241         m_dwStdConversion --;
1242     }
1243 }
1244 CPDF_Color::CPDF_Color(int family)
1245 {
1246     m_pCS = CPDF_ColorSpace::GetStockCS(family);
1247     int nComps = 3;
1248     if (family == PDFCS_DEVICEGRAY) {
1249         nComps = 1;
1250     } else if (family == PDFCS_DEVICECMYK) {
1251         nComps = 4;
1252     }
1253     m_pBuffer = FX_Alloc(FX_FLOAT, nComps);
1254     for (int i = 0; i < nComps; i ++) {
1255         m_pBuffer[i] = 0;
1256     }
1257 }
1258 CPDF_Color::~CPDF_Color()
1259 {
1260     ReleaseBuffer();
1261     ReleaseColorSpace();
1262 }
1263 void CPDF_Color::ReleaseBuffer()
1264 {
1265     if (!m_pBuffer) {
1266         return;
1267     }
1268     if (m_pCS->GetFamily() == PDFCS_PATTERN) {
1269         PatternValue* pvalue = (PatternValue*)m_pBuffer;
1270         CPDF_Pattern* pPattern = pvalue->m_pPattern;
1271         if (pPattern && pPattern->m_pDocument) {
1272             pPattern->m_pDocument->GetPageData()->ReleasePattern(pPattern->m_pPatternObj);
1273         }
1274     }
1275     FX_Free(m_pBuffer);
1276     m_pBuffer = NULL;
1277 }
1278 void CPDF_Color::ReleaseColorSpace()
1279 {
1280     if (m_pCS && m_pCS->m_pDocument && m_pCS->GetArray()) {
1281         m_pCS->m_pDocument->GetPageData()->ReleaseColorSpace(m_pCS->GetArray());
1282         m_pCS = NULL;
1283     }
1284 }
1285 void CPDF_Color::SetColorSpace(CPDF_ColorSpace* pCS)
1286 {
1287     if (m_pCS == pCS) {
1288         if (m_pBuffer == NULL) {
1289             m_pBuffer = pCS->CreateBuf();
1290         }
1291         ReleaseColorSpace();
1292         m_pCS = pCS;
1293         return;
1294     }
1295     ReleaseBuffer();
1296     ReleaseColorSpace();
1297     m_pCS = pCS;
1298     if (m_pCS) {
1299         m_pBuffer = pCS->CreateBuf();
1300         pCS->GetDefaultColor(m_pBuffer);
1301     }
1302 }
1303 void CPDF_Color::SetValue(FX_FLOAT* comps)
1304 {
1305     if (m_pBuffer == NULL) {
1306         return;
1307     }
1308     if (m_pCS->GetFamily() != PDFCS_PATTERN) {
1309         FXSYS_memcpy32(m_pBuffer, comps, m_pCS->CountComponents() * sizeof(FX_FLOAT));
1310     }
1311 }
1312 void CPDF_Color::SetValue(CPDF_Pattern* pPattern, FX_FLOAT* comps, int ncomps)
1313 {
1314     if (ncomps > MAX_PATTERN_COLORCOMPS) {
1315         return;
1316     }
1317     if (m_pCS == NULL || m_pCS->GetFamily() != PDFCS_PATTERN) {
1318         if (m_pBuffer) {
1319             FX_Free(m_pBuffer);
1320         }
1321         m_pCS = CPDF_ColorSpace::GetStockCS(PDFCS_PATTERN);
1322         m_pBuffer = m_pCS->CreateBuf();
1323     }
1324     CPDF_DocPageData* pDocPageData = NULL;
1325     PatternValue* pvalue = (PatternValue*)m_pBuffer;
1326     if (pvalue->m_pPattern && pvalue->m_pPattern->m_pDocument) {
1327         pDocPageData = pvalue->m_pPattern->m_pDocument->GetPageData();
1328         pDocPageData->ReleasePattern(pvalue->m_pPattern->m_pPatternObj);
1329     }
1330     pvalue->m_nComps = ncomps;
1331     pvalue->m_pPattern = pPattern;
1332     if (ncomps) {
1333         FXSYS_memcpy32(pvalue->m_Comps, comps, ncomps * sizeof(FX_FLOAT));
1334     }
1335 }
1336 void CPDF_Color::Copy(const CPDF_Color* pSrc)
1337 {
1338     ReleaseBuffer();
1339     ReleaseColorSpace();
1340     m_pCS = pSrc->m_pCS;
1341     if (m_pCS && m_pCS->m_pDocument) {
1342         CPDF_Array* pArray = m_pCS->GetArray();
1343         if (pArray) {
1344             m_pCS = m_pCS->m_pDocument->GetPageData()->GetCopiedColorSpace(pArray);
1345         }
1346     }
1347     if (m_pCS == NULL) {
1348         return;
1349     }
1350     m_pBuffer = m_pCS->CreateBuf();
1351     FXSYS_memcpy32(m_pBuffer, pSrc->m_pBuffer, m_pCS->GetBufSize());
1352     if (m_pCS->GetFamily() == PDFCS_PATTERN) {
1353         PatternValue* pvalue = (PatternValue*)m_pBuffer;
1354         if (pvalue->m_pPattern && pvalue->m_pPattern->m_pDocument) {
1355             pvalue->m_pPattern = pvalue->m_pPattern->m_pDocument->GetPageData()->GetPattern(pvalue->m_pPattern->m_pPatternObj, FALSE, &pvalue->m_pPattern->m_ParentMatrix);
1356         }
1357     }
1358 }
1359 FX_BOOL CPDF_Color::GetRGB(int& R, int& G, int& B) const
1360 {
1361     if (m_pCS == NULL || m_pBuffer == NULL) {
1362         return FALSE;
1363     }
1364     FX_FLOAT r, g, b;
1365     if (!m_pCS->GetRGB(m_pBuffer, r, g, b)) {
1366         return FALSE;
1367     }
1368     R = (FX_INT32)(r * 255 + 0.5f);
1369     G = (FX_INT32)(g * 255 + 0.5f);
1370     B = (FX_INT32)(b * 255 + 0.5f);
1371     return TRUE;
1372 }
1373 CPDF_Pattern* CPDF_Color::GetPattern() const
1374 {
1375     if (m_pBuffer == NULL || m_pCS->GetFamily() != PDFCS_PATTERN) {
1376         return NULL;
1377     }
1378     PatternValue* pvalue = (PatternValue*)m_pBuffer;
1379     return pvalue->m_pPattern;
1380 }
1381 CPDF_ColorSpace* CPDF_Color::GetPatternCS() const
1382 {
1383     if (m_pBuffer == NULL || m_pCS->GetFamily() != PDFCS_PATTERN) {
1384         return NULL;
1385     }
1386     return m_pCS->GetBaseCS();
1387 }
1388 FX_FLOAT* CPDF_Color::GetPatternColor() const
1389 {
1390     if (m_pBuffer == NULL || m_pCS->GetFamily() != PDFCS_PATTERN) {
1391         return NULL;
1392     }
1393     PatternValue* pvalue = (PatternValue*)m_pBuffer;
1394     return pvalue->m_nComps ? pvalue->m_Comps : NULL;
1395 }
1396 FX_BOOL CPDF_Color::IsEqual(const CPDF_Color& other) const
1397 {
1398     if (m_pCS != other.m_pCS || m_pCS == NULL) {
1399         return FALSE;
1400     }
1401     return FXSYS_memcmp32(m_pBuffer, other.m_pBuffer, m_pCS->GetBufSize()) == 0;
1402 }