OLD | NEW |
1 // Copyright 2014 PDFium Authors. All rights reserved. | 1 // Copyright 2014 PDFium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com | 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
6 | 6 |
7 #include "core/fpdfapi/fpdf_page/pageint.h" | 7 #include "core/fpdfapi/fpdf_page/pageint.h" |
8 | 8 |
9 #include <algorithm> | 9 #include <algorithm> |
10 | 10 |
| 11 #include "core/fpdfapi/fpdf_page/cpdf_meshstream.h" |
| 12 #include "core/fpdfapi/fpdf_page/cpdf_shadingpattern.h" |
11 #include "core/fpdfapi/fpdf_page/include/cpdf_form.h" | 13 #include "core/fpdfapi/fpdf_page/include/cpdf_form.h" |
12 #include "core/fpdfapi/fpdf_parser/include/cpdf_array.h" | 14 #include "core/fpdfapi/fpdf_parser/include/cpdf_array.h" |
13 #include "core/fpdfapi/fpdf_parser/include/cpdf_dictionary.h" | 15 #include "core/fpdfapi/fpdf_parser/include/cpdf_dictionary.h" |
14 #include "core/fpdfapi/fpdf_parser/include/cpdf_document.h" | 16 #include "core/fpdfapi/fpdf_parser/include/cpdf_document.h" |
| 17 #include "core/fpdfapi/fpdf_parser/include/cpdf_stream.h" |
15 | 18 |
16 namespace { | 19 namespace { |
17 | 20 |
18 const int kSingleCoordinatePair = 1; | 21 const int kSingleCoordinatePair = 1; |
19 const int kTensorCoordinatePairs = 16; | 22 const int kTensorCoordinatePairs = 16; |
20 const int kCoonsCoordinatePairs = 12; | 23 const int kCoonsCoordinatePairs = 12; |
21 | 24 |
22 const int kSingleColorPerPatch = 1; | 25 const int kSingleColorPerPatch = 1; |
23 const int kQuadColorsPerPatch = 4; | 26 const int kQuadColorsPerPatch = 4; |
24 | 27 |
25 ShadingType ToShadingType(int type) { | |
26 return (type > static_cast<int>(kInvalidShading) && | |
27 type < static_cast<int>(kMaxShading)) | |
28 ? static_cast<ShadingType>(type) | |
29 : kInvalidShading; | |
30 } | |
31 | |
32 } // namespace | 28 } // namespace |
33 | 29 |
34 CPDF_Pattern::CPDF_Pattern(PatternType type, | |
35 CPDF_Document* pDoc, | |
36 CPDF_Object* pObj, | |
37 const CFX_Matrix* pParentMatrix) | |
38 : m_PatternType(type), | |
39 m_pDocument(pDoc), | |
40 m_pPatternObj(pObj), | |
41 m_bForceClear(FALSE) { | |
42 if (pParentMatrix) | |
43 m_ParentMatrix = *pParentMatrix; | |
44 } | |
45 CPDF_Pattern::~CPDF_Pattern() {} | |
46 CPDF_TilingPattern::CPDF_TilingPattern(CPDF_Document* pDoc, | |
47 CPDF_Object* pPatternObj, | |
48 const CFX_Matrix* parentMatrix) | |
49 : CPDF_Pattern(TILING, pDoc, pPatternObj, parentMatrix) { | |
50 CPDF_Dictionary* pDict = m_pPatternObj->GetDict(); | |
51 m_Pattern2Form = pDict->GetMatrixBy("Matrix"); | |
52 m_bColored = pDict->GetIntegerBy("PaintType") == 1; | |
53 if (parentMatrix) { | |
54 m_Pattern2Form.Concat(*parentMatrix); | |
55 } | |
56 m_pForm = NULL; | |
57 } | |
58 CPDF_TilingPattern::~CPDF_TilingPattern() { | |
59 delete m_pForm; | |
60 m_pForm = NULL; | |
61 } | |
62 FX_BOOL CPDF_TilingPattern::Load() { | |
63 if (m_pForm) | |
64 return TRUE; | |
65 | 30 |
66 CPDF_Dictionary* pDict = m_pPatternObj->GetDict(); | |
67 if (!pDict) | |
68 return FALSE; | |
69 | |
70 m_bColored = pDict->GetIntegerBy("PaintType") == 1; | |
71 m_XStep = (FX_FLOAT)FXSYS_fabs(pDict->GetNumberBy("XStep")); | |
72 m_YStep = (FX_FLOAT)FXSYS_fabs(pDict->GetNumberBy("YStep")); | |
73 | |
74 CPDF_Stream* pStream = m_pPatternObj->AsStream(); | |
75 if (!pStream) | |
76 return FALSE; | |
77 | |
78 m_pForm = new CPDF_Form(m_pDocument, NULL, pStream); | |
79 m_pForm->ParseContent(NULL, &m_ParentMatrix, NULL, NULL); | |
80 m_BBox = pDict->GetRectBy("BBox"); | |
81 return TRUE; | |
82 } | |
83 CPDF_ShadingPattern::CPDF_ShadingPattern(CPDF_Document* pDoc, | |
84 CPDF_Object* pPatternObj, | |
85 FX_BOOL bShading, | |
86 const CFX_Matrix* parentMatrix) | |
87 : CPDF_Pattern(SHADING, | |
88 pDoc, | |
89 bShading ? nullptr : pPatternObj, | |
90 parentMatrix), | |
91 m_ShadingType(kInvalidShading), | |
92 m_bShadingObj(bShading), | |
93 m_pShadingObj(pPatternObj), | |
94 m_pCS(nullptr), | |
95 m_pCountedCS(nullptr), | |
96 m_nFuncs(0) { | |
97 if (!bShading) { | |
98 CPDF_Dictionary* pDict = m_pPatternObj->GetDict(); | |
99 m_Pattern2Form = pDict->GetMatrixBy("Matrix"); | |
100 m_pShadingObj = pDict->GetElementValue("Shading"); | |
101 if (parentMatrix) | |
102 m_Pattern2Form.Concat(*parentMatrix); | |
103 } | |
104 for (int i = 0; i < FX_ArraySize(m_pFunctions); ++i) | |
105 m_pFunctions[i] = nullptr; | |
106 } | |
107 | |
108 CPDF_ShadingPattern::~CPDF_ShadingPattern() { | |
109 for (int i = 0; i < m_nFuncs; ++i) | |
110 delete m_pFunctions[i]; | |
111 | |
112 CPDF_ColorSpace* pCS = m_pCountedCS ? m_pCountedCS->get() : NULL; | |
113 if (pCS && m_pDocument) | |
114 m_pDocument->GetPageData()->ReleaseColorSpace(pCS->GetArray()); | |
115 } | |
116 | |
117 FX_BOOL CPDF_ShadingPattern::Load() { | |
118 if (m_ShadingType != kInvalidShading) | |
119 return TRUE; | |
120 | |
121 CPDF_Dictionary* pShadingDict = | |
122 m_pShadingObj ? m_pShadingObj->GetDict() : NULL; | |
123 if (!pShadingDict) { | |
124 return FALSE; | |
125 } | |
126 if (m_nFuncs) { | |
127 for (int i = 0; i < m_nFuncs; i++) | |
128 delete m_pFunctions[i]; | |
129 m_nFuncs = 0; | |
130 } | |
131 CPDF_Object* pFunc = pShadingDict->GetElementValue("Function"); | |
132 if (pFunc) { | |
133 if (CPDF_Array* pArray = pFunc->AsArray()) { | |
134 m_nFuncs = std::min<int>(pArray->GetCount(), 4); | |
135 | |
136 for (int i = 0; i < m_nFuncs; i++) { | |
137 m_pFunctions[i] = CPDF_Function::Load(pArray->GetElementValue(i)); | |
138 } | |
139 } else { | |
140 m_pFunctions[0] = CPDF_Function::Load(pFunc); | |
141 m_nFuncs = 1; | |
142 } | |
143 } | |
144 CPDF_Object* pCSObj = pShadingDict->GetElementValue("ColorSpace"); | |
145 if (!pCSObj) { | |
146 return FALSE; | |
147 } | |
148 CPDF_DocPageData* pDocPageData = m_pDocument->GetPageData(); | |
149 m_pCS = pDocPageData->GetColorSpace(pCSObj, NULL); | |
150 if (m_pCS) { | |
151 m_pCountedCS = pDocPageData->FindColorSpacePtr(m_pCS->GetArray()); | |
152 } | |
153 | |
154 m_ShadingType = ToShadingType(pShadingDict->GetIntegerBy("ShadingType")); | |
155 | |
156 // We expect to have a stream if our shading type is a mesh. | |
157 if (IsMeshShading() && !ToStream(m_pShadingObj)) | |
158 return FALSE; | |
159 | |
160 return TRUE; | |
161 } | |
162 FX_BOOL CPDF_MeshStream::Load(CPDF_Stream* pShadingStream, | |
163 CPDF_Function** pFuncs, | |
164 int nFuncs, | |
165 CPDF_ColorSpace* pCS) { | |
166 m_Stream.LoadAllData(pShadingStream); | |
167 m_BitStream.Init(m_Stream.GetData(), m_Stream.GetSize()); | |
168 m_pFuncs = pFuncs; | |
169 m_nFuncs = nFuncs; | |
170 m_pCS = pCS; | |
171 CPDF_Dictionary* pDict = pShadingStream->GetDict(); | |
172 m_nCoordBits = pDict->GetIntegerBy("BitsPerCoordinate"); | |
173 m_nCompBits = pDict->GetIntegerBy("BitsPerComponent"); | |
174 m_nFlagBits = pDict->GetIntegerBy("BitsPerFlag"); | |
175 if (!m_nCoordBits || !m_nCompBits) { | |
176 return FALSE; | |
177 } | |
178 FX_DWORD nComps = pCS->CountComponents(); | |
179 if (nComps > 8) { | |
180 return FALSE; | |
181 } | |
182 m_nComps = nFuncs ? 1 : nComps; | |
183 if (((int)m_nComps < 0) || m_nComps > 8) { | |
184 return FALSE; | |
185 } | |
186 m_CoordMax = m_nCoordBits == 32 ? -1 : (1 << m_nCoordBits) - 1; | |
187 m_CompMax = (1 << m_nCompBits) - 1; | |
188 CPDF_Array* pDecode = pDict->GetArrayBy("Decode"); | |
189 if (!pDecode || pDecode->GetCount() != 4 + m_nComps * 2) { | |
190 return FALSE; | |
191 } | |
192 m_xmin = pDecode->GetNumberAt(0); | |
193 m_xmax = pDecode->GetNumberAt(1); | |
194 m_ymin = pDecode->GetNumberAt(2); | |
195 m_ymax = pDecode->GetNumberAt(3); | |
196 for (FX_DWORD i = 0; i < m_nComps; i++) { | |
197 m_ColorMin[i] = pDecode->GetNumberAt(i * 2 + 4); | |
198 m_ColorMax[i] = pDecode->GetNumberAt(i * 2 + 5); | |
199 } | |
200 return TRUE; | |
201 } | |
202 FX_DWORD CPDF_MeshStream::GetFlag() { | |
203 return m_BitStream.GetBits(m_nFlagBits) & 0x03; | |
204 } | |
205 void CPDF_MeshStream::GetCoords(FX_FLOAT& x, FX_FLOAT& y) { | |
206 if (m_nCoordBits == 32) { | |
207 x = m_xmin + (FX_FLOAT)(m_BitStream.GetBits(m_nCoordBits) * | |
208 (m_xmax - m_xmin) / (double)m_CoordMax); | |
209 y = m_ymin + (FX_FLOAT)(m_BitStream.GetBits(m_nCoordBits) * | |
210 (m_ymax - m_ymin) / (double)m_CoordMax); | |
211 } else { | |
212 x = m_xmin + | |
213 m_BitStream.GetBits(m_nCoordBits) * (m_xmax - m_xmin) / m_CoordMax; | |
214 y = m_ymin + | |
215 m_BitStream.GetBits(m_nCoordBits) * (m_ymax - m_ymin) / m_CoordMax; | |
216 } | |
217 } | |
218 void CPDF_MeshStream::GetColor(FX_FLOAT& r, FX_FLOAT& g, FX_FLOAT& b) { | |
219 FX_DWORD i; | |
220 FX_FLOAT color_value[8]; | |
221 for (i = 0; i < m_nComps; i++) { | |
222 color_value[i] = m_ColorMin[i] + | |
223 m_BitStream.GetBits(m_nCompBits) * | |
224 (m_ColorMax[i] - m_ColorMin[i]) / m_CompMax; | |
225 } | |
226 if (m_nFuncs) { | |
227 static const int kMaxResults = 8; | |
228 FX_FLOAT result[kMaxResults]; | |
229 int nResults; | |
230 FXSYS_memset(result, 0, sizeof(result)); | |
231 for (FX_DWORD i = 0; i < m_nFuncs; i++) { | |
232 if (m_pFuncs[i] && m_pFuncs[i]->CountOutputs() <= kMaxResults) { | |
233 m_pFuncs[i]->Call(color_value, 1, result, nResults); | |
234 } | |
235 } | |
236 m_pCS->GetRGB(result, r, g, b); | |
237 } else { | |
238 m_pCS->GetRGB(color_value, r, g, b); | |
239 } | |
240 } | |
241 FX_DWORD CPDF_MeshStream::GetVertex(CPDF_MeshVertex& vertex, | |
242 CFX_Matrix* pObject2Bitmap) { | |
243 FX_DWORD flag = GetFlag(); | |
244 GetCoords(vertex.x, vertex.y); | |
245 pObject2Bitmap->Transform(vertex.x, vertex.y); | |
246 GetColor(vertex.r, vertex.g, vertex.b); | |
247 m_BitStream.ByteAlign(); | |
248 return flag; | |
249 } | |
250 FX_BOOL CPDF_MeshStream::GetVertexRow(CPDF_MeshVertex* vertex, | |
251 int count, | |
252 CFX_Matrix* pObject2Bitmap) { | |
253 for (int i = 0; i < count; i++) { | |
254 if (m_BitStream.IsEOF()) { | |
255 return FALSE; | |
256 } | |
257 GetCoords(vertex[i].x, vertex[i].y); | |
258 pObject2Bitmap->Transform(vertex[i].x, vertex[i].y); | |
259 GetColor(vertex[i].r, vertex[i].g, vertex[i].b); | |
260 m_BitStream.ByteAlign(); | |
261 } | |
262 return TRUE; | |
263 } | |
264 | 31 |
265 CFX_FloatRect GetShadingBBox(CPDF_Stream* pStream, | 32 CFX_FloatRect GetShadingBBox(CPDF_Stream* pStream, |
266 ShadingType type, | 33 ShadingType type, |
267 const CFX_Matrix* pMatrix, | 34 const CFX_Matrix* pMatrix, |
268 CPDF_Function** pFuncs, | 35 CPDF_Function** pFuncs, |
269 int nFuncs, | 36 int nFuncs, |
270 CPDF_ColorSpace* pCS) { | 37 CPDF_ColorSpace* pCS) { |
271 if (!pStream || !pStream->IsStream() || !pFuncs || !pCS) | 38 if (!pStream || !pStream->IsStream() || !pFuncs || !pCS) |
272 return CFX_FloatRect(0, 0, 0, 0); | 39 return CFX_FloatRect(0, 0, 0, 0); |
273 | 40 |
(...skipping 37 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
311 } | 78 } |
312 } | 79 } |
313 stream.m_BitStream.SkipBits(stream.m_nComps * stream.m_nCompBits * | 80 stream.m_BitStream.SkipBits(stream.m_nComps * stream.m_nCompBits * |
314 color_count); | 81 color_count); |
315 if (bGouraud) | 82 if (bGouraud) |
316 stream.m_BitStream.ByteAlign(); | 83 stream.m_BitStream.ByteAlign(); |
317 } | 84 } |
318 rect.Transform(pMatrix); | 85 rect.Transform(pMatrix); |
319 return rect; | 86 return rect; |
320 } | 87 } |
OLD | NEW |