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