1 // Copyright 2017 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 #include "public/fpdf_edit.h"
6
7 #include <utility>
8 #include <vector>
9
10 #include "core/fpdfapi/page/cpdf_path.h"
11 #include "core/fpdfapi/page/cpdf_pathobject.h"
12 #include "core/fxcrt/fx_system.h"
13 #include "fpdfsdk/cpdfsdk_helpers.h"
14 #include "third_party/base/ptr_util.h"
15 #include "third_party/base/span.h"
16 #include "third_party/base/stl_util.h"
17
18 // These checks are here because core/ and public/ cannot depend on each other.
19 static_assert(CFX_GraphStateData::LineCapButt == FPDF_LINECAP_BUTT,
20 "CFX_GraphStateData::LineCapButt value mismatch");
21 static_assert(CFX_GraphStateData::LineCapRound == FPDF_LINECAP_ROUND,
22 "CFX_GraphStateData::LineCapRound value mismatch");
23 static_assert(CFX_GraphStateData::LineCapSquare ==
24 FPDF_LINECAP_PROJECTING_SQUARE,
25 "CFX_GraphStateData::LineCapSquare value mismatch");
26
27 static_assert(CFX_GraphStateData::LineJoinMiter == FPDF_LINEJOIN_MITER,
28 "CFX_GraphStateData::LineJoinMiter value mismatch");
29 static_assert(CFX_GraphStateData::LineJoinRound == FPDF_LINEJOIN_ROUND,
30 "CFX_GraphStateData::LineJoinRound value mismatch");
31 static_assert(CFX_GraphStateData::LineJoinBevel == FPDF_LINEJOIN_BEVEL,
32 "CFX_GraphStateData::LineJoinBevel value mismatch");
33
34 static_assert(static_cast<int>(FXPT_TYPE::LineTo) == FPDF_SEGMENT_LINETO,
35 "FXPT_TYPE::LineTo value mismatch");
36 static_assert(static_cast<int>(FXPT_TYPE::BezierTo) == FPDF_SEGMENT_BEZIERTO,
37 "FXPT_TYPE::BezierTo value mismatch");
38 static_assert(static_cast<int>(FXPT_TYPE::MoveTo) == FPDF_SEGMENT_MOVETO,
39 "FXPT_TYPE::MoveTo value mismatch");
40
41 namespace {
42
CPDFPathObjectFromFPDFPageObject(FPDF_PAGEOBJECT page_object)43 CPDF_PathObject* CPDFPathObjectFromFPDFPageObject(FPDF_PAGEOBJECT page_object) {
44 auto* obj = CPDFPageObjectFromFPDFPageObject(page_object);
45 return obj ? obj->AsPath() : nullptr;
46 }
47
48 } // namespace
49
FPDFPageObj_CreateNewPath(float x,float y)50 FPDF_EXPORT FPDF_PAGEOBJECT FPDF_CALLCONV FPDFPageObj_CreateNewPath(float x,
51 float y) {
52 auto pPathObj = pdfium::MakeUnique<CPDF_PathObject>();
53 pPathObj->path().AppendPoint(CFX_PointF(x, y), FXPT_TYPE::MoveTo, false);
54 pPathObj->DefaultStates();
55
56 // Caller takes ownership.
57 return FPDFPageObjectFromCPDFPageObject(pPathObj.release());
58 }
59
FPDFPageObj_CreateNewRect(float x,float y,float w,float h)60 FPDF_EXPORT FPDF_PAGEOBJECT FPDF_CALLCONV FPDFPageObj_CreateNewRect(float x,
61 float y,
62 float w,
63 float h) {
64 auto pPathObj = pdfium::MakeUnique<CPDF_PathObject>();
65 pPathObj->path().AppendRect(x, y, x + w, y + h);
66 pPathObj->DefaultStates();
67
68 // Caller takes ownership.
69 return FPDFPageObjectFromCPDFPageObject(pPathObj.release());
70 }
71
FPDFPath_CountSegments(FPDF_PAGEOBJECT path)72 FPDF_EXPORT int FPDF_CALLCONV FPDFPath_CountSegments(FPDF_PAGEOBJECT path) {
73 auto* pPathObj = CPDFPathObjectFromFPDFPageObject(path);
74 if (!pPathObj)
75 return -1;
76 return pdfium::CollectionSize<int>(pPathObj->path().GetPoints());
77 }
78
79 FPDF_EXPORT FPDF_PATHSEGMENT FPDF_CALLCONV
FPDFPath_GetPathSegment(FPDF_PAGEOBJECT path,int index)80 FPDFPath_GetPathSegment(FPDF_PAGEOBJECT path, int index) {
81 auto* pPathObj = CPDFPathObjectFromFPDFPageObject(path);
82 if (!pPathObj)
83 return nullptr;
84
85 pdfium::span<const FX_PATHPOINT> points = pPathObj->path().GetPoints();
86 if (!pdfium::IndexInBounds(points, index))
87 return nullptr;
88
89 return FPDFPathSegmentFromFXPathPoint(&points[index]);
90 }
91
FPDFPath_MoveTo(FPDF_PAGEOBJECT path,float x,float y)92 FPDF_EXPORT FPDF_BOOL FPDF_CALLCONV FPDFPath_MoveTo(FPDF_PAGEOBJECT path,
93 float x,
94 float y) {
95 auto* pPathObj = CPDFPathObjectFromFPDFPageObject(path);
96 if (!pPathObj)
97 return false;
98
99 pPathObj->path().AppendPoint(CFX_PointF(x, y), FXPT_TYPE::MoveTo, false);
100 pPathObj->SetDirty(true);
101 return true;
102 }
103
FPDFPath_LineTo(FPDF_PAGEOBJECT path,float x,float y)104 FPDF_EXPORT FPDF_BOOL FPDF_CALLCONV FPDFPath_LineTo(FPDF_PAGEOBJECT path,
105 float x,
106 float y) {
107 auto* pPathObj = CPDFPathObjectFromFPDFPageObject(path);
108 if (!pPathObj)
109 return false;
110
111 pPathObj->path().AppendPoint(CFX_PointF(x, y), FXPT_TYPE::LineTo, false);
112 pPathObj->SetDirty(true);
113 return true;
114 }
115
FPDFPath_BezierTo(FPDF_PAGEOBJECT path,float x1,float y1,float x2,float y2,float x3,float y3)116 FPDF_EXPORT FPDF_BOOL FPDF_CALLCONV FPDFPath_BezierTo(FPDF_PAGEOBJECT path,
117 float x1,
118 float y1,
119 float x2,
120 float y2,
121 float x3,
122 float y3) {
123 auto* pPathObj = CPDFPathObjectFromFPDFPageObject(path);
124 if (!pPathObj)
125 return false;
126
127 CPDF_Path& cpath = pPathObj->path();
128 cpath.AppendPoint(CFX_PointF(x1, y1), FXPT_TYPE::BezierTo, false);
129 cpath.AppendPoint(CFX_PointF(x2, y2), FXPT_TYPE::BezierTo, false);
130 cpath.AppendPoint(CFX_PointF(x3, y3), FXPT_TYPE::BezierTo, false);
131 pPathObj->SetDirty(true);
132 return true;
133 }
134
FPDFPath_Close(FPDF_PAGEOBJECT path)135 FPDF_EXPORT FPDF_BOOL FPDF_CALLCONV FPDFPath_Close(FPDF_PAGEOBJECT path) {
136 auto* pPathObj = CPDFPathObjectFromFPDFPageObject(path);
137 if (!pPathObj)
138 return false;
139
140 CPDF_Path& cpath = pPathObj->path();
141 if (cpath.GetPoints().empty())
142 return false;
143
144 cpath.ClosePath();
145 pPathObj->SetDirty(true);
146 return true;
147 }
148
FPDFPath_SetDrawMode(FPDF_PAGEOBJECT path,int fillmode,FPDF_BOOL stroke)149 FPDF_EXPORT FPDF_BOOL FPDF_CALLCONV FPDFPath_SetDrawMode(FPDF_PAGEOBJECT path,
150 int fillmode,
151 FPDF_BOOL stroke) {
152 auto* pPathObj = CPDFPathObjectFromFPDFPageObject(path);
153 if (!pPathObj)
154 return false;
155
156 pPathObj->set_stroke(!!stroke);
157 if (fillmode == FPDF_FILLMODE_ALTERNATE)
158 pPathObj->set_alternate_filltype();
159 else if (fillmode == FPDF_FILLMODE_WINDING)
160 pPathObj->set_winding_filltype();
161 else
162 pPathObj->set_no_filltype();
163 pPathObj->SetDirty(true);
164 return true;
165 }
166
FPDFPath_GetDrawMode(FPDF_PAGEOBJECT path,int * fillmode,FPDF_BOOL * stroke)167 FPDF_EXPORT FPDF_BOOL FPDF_CALLCONV FPDFPath_GetDrawMode(FPDF_PAGEOBJECT path,
168 int* fillmode,
169 FPDF_BOOL* stroke) {
170 auto* pPathObj = CPDFPathObjectFromFPDFPageObject(path);
171 if (!pPathObj || !fillmode || !stroke)
172 return false;
173
174 if (pPathObj->has_alternate_filltype())
175 *fillmode = FPDF_FILLMODE_ALTERNATE;
176 else if (pPathObj->has_winding_filltype())
177 *fillmode = FPDF_FILLMODE_WINDING;
178 else
179 *fillmode = FPDF_FILLMODE_NONE;
180
181 *stroke = pPathObj->stroke();
182 return true;
183 }
184
FPDFPath_GetMatrix(FPDF_PAGEOBJECT path,FS_MATRIX * matrix)185 FPDF_EXPORT FPDF_BOOL FPDF_CALLCONV FPDFPath_GetMatrix(FPDF_PAGEOBJECT path,
186 FS_MATRIX* matrix) {
187 if (!path || !matrix)
188 return false;
189
190 CPDF_PathObject* pPathObj = CPDFPathObjectFromFPDFPageObject(path);
191 if (!pPathObj)
192 return false;
193
194 *matrix = FSMatrixFromCFXMatrix(pPathObj->matrix());
195 return true;
196 }
197
198 FPDF_EXPORT FPDF_BOOL FPDF_CALLCONV
FPDFPath_SetMatrix(FPDF_PAGEOBJECT path,const FS_MATRIX * matrix)199 FPDFPath_SetMatrix(FPDF_PAGEOBJECT path, const FS_MATRIX* matrix) {
200 if (!matrix)
201 return false;
202
203 CPDF_PathObject* pPathObj = CPDFPathObjectFromFPDFPageObject(path);
204 if (!pPathObj)
205 return false;
206
207 pPathObj->set_matrix(CFXMatrixFromFSMatrix(*matrix));
208 pPathObj->SetDirty(true);
209 return true;
210 }
211
212 FPDF_EXPORT FPDF_BOOL FPDF_CALLCONV
FPDFPathSegment_GetPoint(FPDF_PATHSEGMENT segment,float * x,float * y)213 FPDFPathSegment_GetPoint(FPDF_PATHSEGMENT segment, float* x, float* y) {
214 auto* pPathPoint = FXPathPointFromFPDFPathSegment(segment);
215 if (!pPathPoint || !x || !y)
216 return false;
217
218 *x = pPathPoint->m_Point.x;
219 *y = pPathPoint->m_Point.y;
220 return true;
221 }
222
223 FPDF_EXPORT int FPDF_CALLCONV
FPDFPathSegment_GetType(FPDF_PATHSEGMENT segment)224 FPDFPathSegment_GetType(FPDF_PATHSEGMENT segment) {
225 auto* pPathPoint = FXPathPointFromFPDFPathSegment(segment);
226 return pPathPoint ? static_cast<int>(pPathPoint->m_Type)
227 : FPDF_SEGMENT_UNKNOWN;
228 }
229
230 FPDF_EXPORT FPDF_BOOL FPDF_CALLCONV
FPDFPathSegment_GetClose(FPDF_PATHSEGMENT segment)231 FPDFPathSegment_GetClose(FPDF_PATHSEGMENT segment) {
232 auto* pPathPoint = FXPathPointFromFPDFPathSegment(segment);
233 return pPathPoint && pPathPoint->m_CloseFigure;
234 }
235