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