Make ContentStreamParser members private, fix constructor
[pdfium.git] / core / src / fpdfapi / fpdf_page / fpdf_page_pattern.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 "pageint.h"
9
10 CPDF_Pattern::CPDF_Pattern(const CFX_AffineMatrix* pParentMatrix) :
11     m_pPatternObj(NULL), m_PatternType(PATTERN_TILING), m_pDocument(NULL), m_bForceClear(FALSE)
12 {
13     if (pParentMatrix) {
14         m_ParentMatrix = *pParentMatrix;
15     }
16 }
17 CPDF_Pattern::~CPDF_Pattern()
18 {
19 }
20 CPDF_TilingPattern::CPDF_TilingPattern(CPDF_Document* pDoc, CPDF_Object* pPatternObj, const CFX_AffineMatrix* parentMatrix) :
21     CPDF_Pattern(parentMatrix)
22 {
23     m_PatternType = PATTERN_TILING;
24     m_pPatternObj = pPatternObj;
25     m_pDocument = pDoc;
26     CPDF_Dictionary* pDict = m_pPatternObj->GetDict();
27     ASSERT(pDict != NULL);
28     m_Pattern2Form = pDict->GetMatrix(FX_BSTRC("Matrix"));
29     m_bColored = pDict->GetInteger(FX_BSTRC("PaintType")) == 1;
30     if (parentMatrix) {
31         m_Pattern2Form.Concat(*parentMatrix);
32     }
33     m_pForm = NULL;
34 }
35 CPDF_TilingPattern::~CPDF_TilingPattern()
36 {
37     delete m_pForm;
38     m_pForm = NULL;
39 }
40 FX_BOOL CPDF_TilingPattern::Load()
41 {
42     if (m_pForm != NULL) {
43         return TRUE;
44     }
45     CPDF_Dictionary* pDict = m_pPatternObj->GetDict();
46     if (pDict == NULL) {
47         return FALSE;
48     }
49     m_bColored = pDict->GetInteger(FX_BSTRC("PaintType")) == 1;
50     m_XStep = (FX_FLOAT)FXSYS_fabs(pDict->GetNumber(FX_BSTRC("XStep")));
51     m_YStep = (FX_FLOAT)FXSYS_fabs(pDict->GetNumber(FX_BSTRC("YStep")));
52     if (m_pPatternObj->GetType() != PDFOBJ_STREAM) {
53         return FALSE;
54     }
55     CPDF_Stream* pStream = (CPDF_Stream*)m_pPatternObj;
56     m_pForm = new CPDF_Form(m_pDocument, NULL, pStream);
57     m_pForm->ParseContent(NULL, &m_ParentMatrix, NULL, NULL);
58     m_BBox = pDict->GetRect(FX_BSTRC("BBox"));
59     return TRUE;
60 }
61 CPDF_ShadingPattern::CPDF_ShadingPattern(CPDF_Document* pDoc, CPDF_Object* pPatternObj, FX_BOOL bShading, const CFX_AffineMatrix* parentMatrix) : CPDF_Pattern(parentMatrix)
62 {
63     m_PatternType = PATTERN_SHADING;
64     m_pPatternObj = bShading ? NULL : pPatternObj;
65     m_pDocument = pDoc;
66     m_bShadingObj = bShading;
67     if (!bShading) {
68         CPDF_Dictionary* pDict = m_pPatternObj->GetDict();
69         ASSERT(pDict != NULL);
70         m_Pattern2Form = pDict->GetMatrix(FX_BSTRC("Matrix"));
71         m_pShadingObj = pDict->GetElementValue(FX_BSTRC("Shading"));
72         if (parentMatrix) {
73             m_Pattern2Form.Concat(*parentMatrix);
74         }
75     } else {
76         m_pShadingObj = pPatternObj;
77     }
78     m_ShadingType = 0;
79     m_pCS = NULL;
80     m_nFuncs = 0;
81     for (int i = 0; i < 4; i ++) {
82         m_pFunctions[i] = NULL;
83     }
84     m_pCountedCS = NULL;
85 }
86 CPDF_ShadingPattern::~CPDF_ShadingPattern()
87 {
88     Clear();
89 }
90 void CPDF_ShadingPattern::Clear()
91 {
92     for (int i = 0; i < m_nFuncs; i ++) {
93         delete m_pFunctions[i];
94         m_pFunctions[i] = NULL;
95     }
96     CPDF_ColorSpace* pCS = m_pCountedCS ? m_pCountedCS->get() : NULL;
97     if (pCS && m_pDocument) {
98         m_pDocument->GetPageData()->ReleaseColorSpace(pCS->GetArray());
99     }
100     m_ShadingType = 0;
101     m_pCS = NULL;
102     m_pCountedCS = NULL;
103     m_nFuncs = 0;
104 }
105 FX_BOOL CPDF_ShadingPattern::Load()
106 {
107     if (m_ShadingType != 0) {
108         return TRUE;
109     }
110     CPDF_Dictionary* pShadingDict = m_pShadingObj ? m_pShadingObj->GetDict() : NULL;
111     if (pShadingDict == NULL) {
112         return FALSE;
113     }
114     if (m_nFuncs) {
115         for (int i = 0; i < m_nFuncs; i ++)
116             delete m_pFunctions[i];
117         m_nFuncs = 0;
118     }
119     CPDF_Object* pFunc = pShadingDict->GetElementValue(FX_BSTRC("Function"));
120     if (pFunc) {
121         if (pFunc->GetType() == PDFOBJ_ARRAY) {
122             m_nFuncs = ((CPDF_Array*)pFunc)->GetCount();
123             if (m_nFuncs > 4) {
124                 m_nFuncs = 4;
125             }
126             for (int i = 0; i < m_nFuncs; i ++) {
127                 m_pFunctions[i] = CPDF_Function::Load(((CPDF_Array*)pFunc)->GetElementValue(i));
128             }
129         } else {
130             m_pFunctions[0] = CPDF_Function::Load(pFunc);
131             m_nFuncs = 1;
132         }
133     }
134     CPDF_Object* pCSObj = pShadingDict->GetElementValue(FX_BSTRC("ColorSpace"));
135     if (pCSObj == NULL) {
136         return FALSE;
137     }
138     CPDF_DocPageData* pDocPageData = m_pDocument->GetPageData();
139     m_pCS = pDocPageData->GetColorSpace(pCSObj, NULL);
140     if (m_pCS) {
141         m_pCountedCS = pDocPageData->FindColorSpacePtr(m_pCS->GetArray());
142     }
143     m_ShadingType = pShadingDict->GetInteger(FX_BSTRC("ShadingType"));
144     return TRUE;
145 }
146 FX_BOOL CPDF_ShadingPattern::Reload()
147 {
148     Clear();
149     return Load();
150 }
151 FX_BOOL CPDF_MeshStream::Load(CPDF_Stream* pShadingStream, CPDF_Function** pFuncs, int nFuncs, CPDF_ColorSpace* pCS)
152 {
153     m_Stream.LoadAllData(pShadingStream);
154     m_BitStream.Init(m_Stream.GetData(), m_Stream.GetSize());
155     m_pFuncs = pFuncs;
156     m_nFuncs = nFuncs;
157     m_pCS = pCS;
158     CPDF_Dictionary* pDict = pShadingStream->GetDict();
159     m_nCoordBits = pDict->GetInteger(FX_BSTRC("BitsPerCoordinate"));
160     m_nCompBits = pDict->GetInteger(FX_BSTRC("BitsPerComponent"));
161     m_nFlagBits = pDict->GetInteger(FX_BSTRC("BitsPerFlag"));
162     if (!m_nCoordBits || !m_nCompBits) {
163         return FALSE;
164     }
165     int nComps = pCS->CountComponents();
166     if (nComps > 8) {
167         return FALSE;
168     }
169     m_nComps = nFuncs ? 1 : nComps;
170     if (((int)m_nComps < 0) || m_nComps > 8) {
171         return FALSE;
172     }
173     m_CoordMax = m_nCoordBits == 32 ? -1 : (1 << m_nCoordBits) - 1;
174     m_CompMax = (1 << m_nCompBits) - 1;
175     CPDF_Array* pDecode = pDict->GetArray(FX_BSTRC("Decode"));
176     if (pDecode == NULL || pDecode->GetCount() != 4 + m_nComps * 2) {
177         return FALSE;
178     }
179     m_xmin = pDecode->GetNumber(0);
180     m_xmax = pDecode->GetNumber(1);
181     m_ymin = pDecode->GetNumber(2);
182     m_ymax = pDecode->GetNumber(3);
183     for (FX_DWORD i = 0; i < m_nComps; i ++) {
184         m_ColorMin[i] = pDecode->GetNumber(i * 2 + 4);
185         m_ColorMax[i] = pDecode->GetNumber(i * 2 + 5);
186     }
187     return TRUE;
188 }
189 FX_DWORD CPDF_MeshStream::GetFlag()
190 {
191     return m_BitStream.GetBits(m_nFlagBits) & 0x03;
192 }
193 void CPDF_MeshStream::GetCoords(FX_FLOAT& x, FX_FLOAT& y)
194 {
195     if (m_nCoordBits == 32) {
196         x = m_xmin + (FX_FLOAT)(m_BitStream.GetBits(m_nCoordBits) * (m_xmax - m_xmin) / (double)m_CoordMax);
197         y = m_ymin + (FX_FLOAT)(m_BitStream.GetBits(m_nCoordBits) * (m_ymax - m_ymin) / (double)m_CoordMax);
198     } else {
199         x = m_xmin + m_BitStream.GetBits(m_nCoordBits) * (m_xmax - m_xmin) / m_CoordMax;
200         y = m_ymin + m_BitStream.GetBits(m_nCoordBits) * (m_ymax - m_ymin) / m_CoordMax;
201     }
202 }
203 void CPDF_MeshStream::GetColor(FX_FLOAT& r, FX_FLOAT& g, FX_FLOAT& b)
204 {
205     FX_DWORD i;
206     FX_FLOAT color_value[8];
207     for (i = 0; i < m_nComps; i ++) {
208         color_value[i] = m_ColorMin[i] + m_BitStream.GetBits(m_nCompBits) * (m_ColorMax[i] - m_ColorMin[i]) / m_CompMax;
209     }
210     if (m_nFuncs) {
211         static const int kMaxResults = 8;
212         FX_FLOAT result[kMaxResults];
213         int nResults;
214         FXSYS_memset(result, 0, sizeof(result));
215         for (FX_DWORD i = 0; i < m_nFuncs; i ++) {
216             if (m_pFuncs[i] && m_pFuncs[i]->CountOutputs() <= kMaxResults) {
217                 m_pFuncs[i]->Call(color_value, 1, result, nResults);
218             }
219         }
220         m_pCS->GetRGB(result, r, g, b);
221     } else {
222         m_pCS->GetRGB(color_value, r, g, b);
223     }
224 }
225 FX_DWORD CPDF_MeshStream::GetVertex(CPDF_MeshVertex& vertex, CFX_AffineMatrix* pObject2Bitmap)
226 {
227     FX_DWORD flag = GetFlag();
228     GetCoords(vertex.x, vertex.y);
229     pObject2Bitmap->Transform(vertex.x, vertex.y);
230     GetColor(vertex.r, vertex.g, vertex.b);
231     m_BitStream.ByteAlign();
232     return flag;
233 }
234 FX_BOOL CPDF_MeshStream::GetVertexRow(CPDF_MeshVertex* vertex, int count, CFX_AffineMatrix* pObject2Bitmap)
235 {
236     for (int i = 0; i < count; i ++) {
237         if (m_BitStream.IsEOF()) {
238             return FALSE;
239         }
240         GetCoords(vertex[i].x, vertex[i].y);
241         pObject2Bitmap->Transform(vertex[i].x, vertex[i].y);
242         GetColor(vertex[i].r, vertex[i].g, vertex[i].b);
243         m_BitStream.ByteAlign();
244     }
245     return TRUE;
246 }
247 CFX_FloatRect _GetShadingBBox(CPDF_Stream* pStream, int type, const CFX_AffineMatrix* pMatrix,
248                               CPDF_Function** pFuncs, int nFuncs, CPDF_ColorSpace* pCS)
249 {
250     if (pStream == NULL || pStream->GetType() != PDFOBJ_STREAM || pFuncs == NULL || pCS == NULL) {
251         return CFX_FloatRect(0, 0, 0, 0);
252     }
253     CPDF_MeshStream stream;
254     if (!stream.Load(pStream, pFuncs, nFuncs, pCS)) {
255         return CFX_FloatRect(0, 0, 0, 0);
256     }
257     CFX_FloatRect rect;
258     FX_BOOL bStarted = FALSE;
259     FX_BOOL bGouraud = type == 4 || type == 5;
260     int full_point_count = type == 7 ? 16 : (type == 6 ? 12 : 1);
261     int full_color_count = (type == 6 || type == 7) ? 4 : 1;
262     while (!stream.m_BitStream.IsEOF()) {
263         FX_DWORD flag = 0;
264         if (type != 5) {
265             flag = stream.GetFlag();
266         }
267         int point_count = full_point_count, color_count = full_color_count;
268         if (!bGouraud && flag) {
269             point_count -= 4;
270             color_count -= 2;
271         }
272         for (int i = 0; i < point_count; i ++) {
273             FX_FLOAT x, y;
274             stream.GetCoords(x, y);
275             if (bStarted) {
276                 rect.UpdateRect(x, y);
277             } else {
278                 rect.InitRect(x, y);
279                 bStarted = TRUE;
280             }
281         }
282         stream.m_BitStream.SkipBits(stream.m_nComps * stream.m_nCompBits * color_count);
283         if (bGouraud) {
284             stream.m_BitStream.ByteAlign();
285         }
286     }
287     rect.Transform(pMatrix);
288     return rect;
289 }