blob: 0b66191953bcd5fc143133d8cc12b4e3bdabcce2 [file] [log] [blame]
epoger@google.comec3ed6a2011-07-28 14:26:00 +00001
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00002/*
epoger@google.comec3ed6a2011-07-28 14:26:00 +00003 * Copyright 2011 Google Inc.
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00004 *
epoger@google.comec3ed6a2011-07-28 14:26:00 +00005 * Use of this source code is governed by a BSD-style license that can be
6 * found in the LICENSE file.
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00007 */
8
epoger@google.comec3ed6a2011-07-28 14:26:00 +00009
vandebo@chromium.orgda912d62011-03-08 18:31:02 +000010#include "SkPDFShader.h"
11
vandebo@chromium.org421d6442011-07-20 17:39:01 +000012#include "SkData.h"
halcanaryfb62b3d2015-01-21 09:59:14 -080013#include "SkPDFCanon.h"
vandebo@chromium.orgda912d62011-03-08 18:31:02 +000014#include "SkPDFCatalog.h"
15#include "SkPDFDevice.h"
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +000016#include "SkPDFFormXObject.h"
17#include "SkPDFGraphicState.h"
commit-bot@chromium.org47401352013-07-23 21:49:29 +000018#include "SkPDFResourceDict.h"
vandebo@chromium.orgda912d62011-03-08 18:31:02 +000019#include "SkPDFUtils.h"
20#include "SkScalar.h"
21#include "SkStream.h"
twiz@google.com316338a2011-03-09 23:14:04 +000022#include "SkTemplates.h"
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +000023#include "SkTSet.h"
vandebo@chromium.orgda912d62011-03-08 18:31:02 +000024#include "SkTypes.h"
25
fmalita7b4d4c72015-02-12 13:56:50 -080026static bool inverse_transform_bbox(const SkMatrix& matrix, SkRect* bbox) {
vandebo@chromium.orgda912d62011-03-08 18:31:02 +000027 SkMatrix inverse;
vandebo@chromium.orgb0549902012-04-13 20:45:46 +000028 if (!matrix.invert(&inverse)) {
vandebo@chromium.org386dfc02012-04-17 22:31:52 +000029 return false;
vandebo@chromium.orgb0549902012-04-13 20:45:46 +000030 }
vandebo@chromium.orgda912d62011-03-08 18:31:02 +000031 inverse.mapRect(bbox);
vandebo@chromium.org386dfc02012-04-17 22:31:52 +000032 return true;
vandebo@chromium.orgda912d62011-03-08 18:31:02 +000033}
34
35static void unitToPointsMatrix(const SkPoint pts[2], SkMatrix* matrix) {
36 SkVector vec = pts[1] - pts[0];
37 SkScalar mag = vec.length();
38 SkScalar inv = mag ? SkScalarInvert(mag) : 0;
39
40 vec.scale(inv);
41 matrix->setSinCos(vec.fY, vec.fX);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +000042 matrix->preScale(mag, mag);
commit-bot@chromium.orgace22692013-06-12 21:33:02 +000043 matrix->postTranslate(pts[0].fX, pts[0].fY);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +000044}
45
46/* Assumes t + startOffset is on the stack and does a linear interpolation on t
47 between startOffset and endOffset from prevColor to curColor (for each color
djsollen@google.com53648ab2013-04-03 12:34:16 +000048 component), leaving the result in component order on the stack. It assumes
49 there are always 3 components per color.
vandebo@chromium.orgda912d62011-03-08 18:31:02 +000050 @param range endOffset - startOffset
51 @param curColor[components] The current color components.
52 @param prevColor[components] The previous color components.
53 @param result The result ps function.
54 */
55static void interpolateColorCode(SkScalar range, SkScalar* curColor,
djsollen@google.com53648ab2013-04-03 12:34:16 +000056 SkScalar* prevColor, SkString* result) {
edisonn@google.com81353232013-09-17 17:17:47 +000057 SkASSERT(range != SkIntToScalar(0));
djsollen@google.com53648ab2013-04-03 12:34:16 +000058 static const int kColorComponents = 3;
59
vandebo@chromium.orgda912d62011-03-08 18:31:02 +000060 // Figure out how to scale each color component.
djsollen@google.com53648ab2013-04-03 12:34:16 +000061 SkScalar multiplier[kColorComponents];
62 for (int i = 0; i < kColorComponents; i++) {
vandebo@chromium.orgda912d62011-03-08 18:31:02 +000063 multiplier[i] = SkScalarDiv(curColor[i] - prevColor[i], range);
64 }
65
66 // Calculate when we no longer need to keep a copy of the input parameter t.
67 // If the last component to use t is i, then dupInput[0..i - 1] = true
68 // and dupInput[i .. components] = false.
djsollen@google.com53648ab2013-04-03 12:34:16 +000069 bool dupInput[kColorComponents];
70 dupInput[kColorComponents - 1] = false;
71 for (int i = kColorComponents - 2; i >= 0; i--) {
vandebo@chromium.orgda912d62011-03-08 18:31:02 +000072 dupInput[i] = dupInput[i + 1] || multiplier[i + 1] != 0;
73 }
74
75 if (!dupInput[0] && multiplier[0] == 0) {
76 result->append("pop ");
77 }
78
djsollen@google.com53648ab2013-04-03 12:34:16 +000079 for (int i = 0; i < kColorComponents; i++) {
vandebo@chromium.orgb8622042012-10-22 20:12:40 +000080 // If the next components needs t and this component will consume a
81 // copy, make another copy.
82 if (dupInput[i] && multiplier[i] != 0) {
vandebo@chromium.orgda912d62011-03-08 18:31:02 +000083 result->append("dup ");
84 }
85
86 if (multiplier[i] == 0) {
87 result->appendScalar(prevColor[i]);
88 result->append(" ");
89 } else {
90 if (multiplier[i] != 1) {
91 result->appendScalar(multiplier[i]);
92 result->append(" mul ");
93 }
94 if (prevColor[i] != 0) {
95 result->appendScalar(prevColor[i]);
96 result->append(" add ");
97 }
98 }
99
100 if (dupInput[i]) {
101 result->append("exch\n");
102 }
103 }
104}
105
106/* Generate Type 4 function code to map t=[0,1) to the passed gradient,
107 clamping at the edges of the range. The generated code will be of the form:
108 if (t < 0) {
109 return colorData[0][r,g,b];
110 } else {
111 if (t < info.fColorOffsets[1]) {
112 return linearinterpolation(colorData[0][r,g,b],
113 colorData[1][r,g,b]);
114 } else {
115 if (t < info.fColorOffsets[2]) {
116 return linearinterpolation(colorData[1][r,g,b],
117 colorData[2][r,g,b]);
118 } else {
119
120 ... } else {
121 return colorData[info.fColorCount - 1][r,g,b];
122 }
123 ...
124 }
125 }
126 */
127static void gradientFunctionCode(const SkShader::GradientInfo& info,
128 SkString* result) {
129 /* We want to linearly interpolate from the previous color to the next.
130 Scale the colors from 0..255 to 0..1 and determine the multipliers
131 for interpolation.
132 C{r,g,b}(t, section) = t - offset_(section-1) + t * Multiplier{r,g,b}.
133 */
134 static const int kColorComponents = 3;
twiz@google.com316338a2011-03-09 23:14:04 +0000135 typedef SkScalar ColorTuple[kColorComponents];
136 SkAutoSTMalloc<4, ColorTuple> colorDataAlloc(info.fColorCount);
137 ColorTuple *colorData = colorDataAlloc.get();
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000138 const SkScalar scale = SkScalarInvert(SkIntToScalar(255));
139 for (int i = 0; i < info.fColorCount; i++) {
140 colorData[i][0] = SkScalarMul(SkColorGetR(info.fColors[i]), scale);
141 colorData[i][1] = SkScalarMul(SkColorGetG(info.fColors[i]), scale);
142 colorData[i][2] = SkScalarMul(SkColorGetB(info.fColors[i]), scale);
143 }
144
145 // Clamp the initial color.
146 result->append("dup 0 le {pop ");
147 result->appendScalar(colorData[0][0]);
148 result->append(" ");
149 result->appendScalar(colorData[0][1]);
150 result->append(" ");
151 result->appendScalar(colorData[0][2]);
152 result->append(" }\n");
153
154 // The gradient colors.
edisonn@google.com81353232013-09-17 17:17:47 +0000155 int gradients = 0;
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000156 for (int i = 1 ; i < info.fColorCount; i++) {
edisonn@google.com81353232013-09-17 17:17:47 +0000157 if (info.fColorOffsets[i] == info.fColorOffsets[i - 1]) {
158 continue;
159 }
160 gradients++;
161
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000162 result->append("{dup ");
163 result->appendScalar(info.fColorOffsets[i]);
164 result->append(" le {");
165 if (info.fColorOffsets[i - 1] != 0) {
166 result->appendScalar(info.fColorOffsets[i - 1]);
167 result->append(" sub\n");
168 }
169
170 interpolateColorCode(info.fColorOffsets[i] - info.fColorOffsets[i - 1],
djsollen@google.com53648ab2013-04-03 12:34:16 +0000171 colorData[i], colorData[i - 1], result);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000172 result->append("}\n");
173 }
174
175 // Clamp the final color.
176 result->append("{pop ");
177 result->appendScalar(colorData[info.fColorCount - 1][0]);
178 result->append(" ");
179 result->appendScalar(colorData[info.fColorCount - 1][1]);
180 result->append(" ");
181 result->appendScalar(colorData[info.fColorCount - 1][2]);
182
edisonn@google.com81353232013-09-17 17:17:47 +0000183 for (int i = 0 ; i < gradients + 1; i++) {
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000184 result->append("} ifelse\n");
185 }
186}
187
188/* Map a value of t on the stack into [0, 1) for Repeat or Mirror tile mode. */
189static void tileModeCode(SkShader::TileMode mode, SkString* result) {
190 if (mode == SkShader::kRepeat_TileMode) {
191 result->append("dup truncate sub\n"); // Get the fractional part.
192 result->append("dup 0 le {1 add} if\n"); // Map (-1,0) => (0,1)
193 return;
194 }
195
196 if (mode == SkShader::kMirror_TileMode) {
197 // Map t mod 2 into [0, 1, 1, 0].
198 // Code Stack
199 result->append("abs " // Map negative to positive.
200 "dup " // t.s t.s
201 "truncate " // t.s t
202 "dup " // t.s t t
203 "cvi " // t.s t T
204 "2 mod " // t.s t (i mod 2)
205 "1 eq " // t.s t true|false
206 "3 1 roll " // true|false t.s t
207 "sub " // true|false 0.s
208 "exch " // 0.s true|false
209 "{1 exch sub} if\n"); // 1 - 0.s|0.s
210 }
211}
212
edisonn@google.com83d8eda2013-10-24 13:19:28 +0000213/**
214 * Returns PS function code that applies inverse perspective
215 * to a x, y point.
216 * The function assumes that the stack has at least two elements,
217 * and that the top 2 elements are numeric values.
218 * After executing this code on a PS stack, the last 2 elements are updated
219 * while the rest of the stack is preserved intact.
220 * inversePerspectiveMatrix is the inverse perspective matrix.
221 */
222static SkString apply_perspective_to_coordinates(
223 const SkMatrix& inversePerspectiveMatrix) {
224 SkString code;
225 if (!inversePerspectiveMatrix.hasPerspective()) {
226 return code;
227 }
228
229 // Perspective matrix should be:
230 // 1 0 0
231 // 0 1 0
232 // p0 p1 p2
233
234 const SkScalar p0 = inversePerspectiveMatrix[SkMatrix::kMPersp0];
235 const SkScalar p1 = inversePerspectiveMatrix[SkMatrix::kMPersp1];
236 const SkScalar p2 = inversePerspectiveMatrix[SkMatrix::kMPersp2];
237
238 // y = y / (p2 + p0 x + p1 y)
239 // x = x / (p2 + p0 x + p1 y)
240
241 // Input on stack: x y
242 code.append(" dup "); // x y y
243 code.appendScalar(p1); // x y y p1
244 code.append(" mul " // x y y*p1
245 " 2 index "); // x y y*p1 x
246 code.appendScalar(p0); // x y y p1 x p0
247 code.append(" mul "); // x y y*p1 x*p0
248 code.appendScalar(p2); // x y y p1 x*p0 p2
249 code.append(" add " // x y y*p1 x*p0+p2
250 "add " // x y y*p1+x*p0+p2
251 "3 1 roll " // y*p1+x*p0+p2 x y
252 "2 index " // z x y y*p1+x*p0+p2
253 "div " // y*p1+x*p0+p2 x y/(y*p1+x*p0+p2)
254 "3 1 roll " // y/(y*p1+x*p0+p2) y*p1+x*p0+p2 x
255 "exch " // y/(y*p1+x*p0+p2) x y*p1+x*p0+p2
256 "div " // y/(y*p1+x*p0+p2) x/(y*p1+x*p0+p2)
257 "exch\n"); // x/(y*p1+x*p0+p2) y/(y*p1+x*p0+p2)
258 return code;
259}
260
261static SkString linearCode(const SkShader::GradientInfo& info,
262 const SkMatrix& perspectiveRemover) {
263 SkString function("{");
264
265 function.append(apply_perspective_to_coordinates(perspectiveRemover));
266
267 function.append("pop\n"); // Just ditch the y value.
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000268 tileModeCode(info.fTileMode, &function);
269 gradientFunctionCode(info, &function);
270 function.append("}");
271 return function;
272}
273
edisonn@google.com83d8eda2013-10-24 13:19:28 +0000274static SkString radialCode(const SkShader::GradientInfo& info,
275 const SkMatrix& perspectiveRemover) {
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000276 SkString function("{");
edisonn@google.com83d8eda2013-10-24 13:19:28 +0000277
278 function.append(apply_perspective_to_coordinates(perspectiveRemover));
279
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000280 // Find the distance from the origin.
281 function.append("dup " // x y y
282 "mul " // x y^2
283 "exch " // y^2 x
284 "dup " // y^2 x x
285 "mul " // y^2 x^2
286 "add " // y^2+x^2
287 "sqrt\n"); // sqrt(y^2+x^2)
288
289 tileModeCode(info.fTileMode, &function);
290 gradientFunctionCode(info, &function);
291 function.append("}");
292 return function;
293}
294
295/* The math here is all based on the description in Two_Point_Radial_Gradient,
296 with one simplification, the coordinate space has been scaled so that
297 Dr = 1. This means we don't need to scale the entire equation by 1/Dr^2.
298 */
edisonn@google.com83d8eda2013-10-24 13:19:28 +0000299static SkString twoPointRadialCode(const SkShader::GradientInfo& info,
300 const SkMatrix& perspectiveRemover) {
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000301 SkScalar dx = info.fPoint[0].fX - info.fPoint[1].fX;
302 SkScalar dy = info.fPoint[0].fY - info.fPoint[1].fY;
303 SkScalar sr = info.fRadius[0];
304 SkScalar a = SkScalarMul(dx, dx) + SkScalarMul(dy, dy) - SK_Scalar1;
305 bool posRoot = info.fRadius[1] > info.fRadius[0];
306
307 // We start with a stack of (x y), copy it and then consume one copy in
308 // order to calculate b and the other to calculate c.
309 SkString function("{");
edisonn@google.com83d8eda2013-10-24 13:19:28 +0000310
311 function.append(apply_perspective_to_coordinates(perspectiveRemover));
312
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000313 function.append("2 copy ");
314
315 // Calculate -b and b^2.
316 function.appendScalar(dy);
317 function.append(" mul exch ");
318 function.appendScalar(dx);
319 function.append(" mul add ");
320 function.appendScalar(sr);
321 function.append(" sub 2 mul neg dup dup mul\n");
322
323 // Calculate c
324 function.append("4 2 roll dup mul exch dup mul add ");
325 function.appendScalar(SkScalarMul(sr, sr));
326 function.append(" sub\n");
327
328 // Calculate the determinate
329 function.appendScalar(SkScalarMul(SkIntToScalar(4), a));
330 function.append(" mul sub abs sqrt\n");
331
332 // And then the final value of t.
333 if (posRoot) {
334 function.append("sub ");
335 } else {
336 function.append("add ");
337 }
338 function.appendScalar(SkScalarMul(SkIntToScalar(2), a));
339 function.append(" div\n");
340
341 tileModeCode(info.fTileMode, &function);
342 gradientFunctionCode(info, &function);
343 function.append("}");
344 return function;
345}
346
rileya@google.com62197282012-07-10 20:05:23 +0000347/* Conical gradient shader, based on the Canvas spec for radial gradients
rmistry@google.comd6176b02012-08-23 18:14:13 +0000348 See: http://www.w3.org/TR/2dcontext/#dom-context-2d-createradialgradient
rileya@google.com62197282012-07-10 20:05:23 +0000349 */
edisonn@google.com83d8eda2013-10-24 13:19:28 +0000350static SkString twoPointConicalCode(const SkShader::GradientInfo& info,
351 const SkMatrix& perspectiveRemover) {
rileya@google.com62197282012-07-10 20:05:23 +0000352 SkScalar dx = info.fPoint[1].fX - info.fPoint[0].fX;
353 SkScalar dy = info.fPoint[1].fY - info.fPoint[0].fY;
354 SkScalar r0 = info.fRadius[0];
355 SkScalar dr = info.fRadius[1] - info.fRadius[0];
rmistry@google.comd6176b02012-08-23 18:14:13 +0000356 SkScalar a = SkScalarMul(dx, dx) + SkScalarMul(dy, dy) -
rileya@google.com62197282012-07-10 20:05:23 +0000357 SkScalarMul(dr, dr);
358
359 // First compute t, if the pixel falls outside the cone, then we'll end
360 // with 'false' on the stack, otherwise we'll push 'true' with t below it
361
362 // We start with a stack of (x y), copy it and then consume one copy in
363 // order to calculate b and the other to calculate c.
364 SkString function("{");
edisonn@google.com83d8eda2013-10-24 13:19:28 +0000365
366 function.append(apply_perspective_to_coordinates(perspectiveRemover));
367
rileya@google.com62197282012-07-10 20:05:23 +0000368 function.append("2 copy ");
369
370 // Calculate b and b^2; b = -2 * (y * dy + x * dx + r0 * dr).
371 function.appendScalar(dy);
372 function.append(" mul exch ");
373 function.appendScalar(dx);
374 function.append(" mul add ");
375 function.appendScalar(SkScalarMul(r0, dr));
376 function.append(" add -2 mul dup dup mul\n");
377
378 // c = x^2 + y^2 + radius0^2
379 function.append("4 2 roll dup mul exch dup mul add ");
380 function.appendScalar(SkScalarMul(r0, r0));
381 function.append(" sub dup 4 1 roll\n");
382
383 // Contents of the stack at this point: c, b, b^2, c
384
385 // if a = 0, then we collapse to a simpler linear case
386 if (a == 0) {
387
388 // t = -c/b
389 function.append("pop pop div neg dup ");
390
391 // compute radius(t)
392 function.appendScalar(dr);
393 function.append(" mul ");
394 function.appendScalar(r0);
395 function.append(" add\n");
396
397 // if r(t) < 0, then it's outside the cone
rileya@google.comab2fe822012-07-10 20:41:38 +0000398 function.append("0 lt {pop false} {true} ifelse\n");
rileya@google.com62197282012-07-10 20:05:23 +0000399
400 } else {
401
402 // quadratic case: the Canvas spec wants the largest
403 // root t for which radius(t) > 0
rmistry@google.comd6176b02012-08-23 18:14:13 +0000404
rileya@google.com62197282012-07-10 20:05:23 +0000405 // compute the discriminant (b^2 - 4ac)
406 function.appendScalar(SkScalarMul(SkIntToScalar(4), a));
407 function.append(" mul sub dup\n");
408
409 // if d >= 0, proceed
410 function.append("0 ge {\n");
411
412 // an intermediate value we'll use to compute the roots:
413 // q = -0.5 * (b +/- sqrt(d))
414 function.append("sqrt exch dup 0 lt {exch -1 mul} if");
415 function.append(" add -0.5 mul dup\n");
416
417 // first root = q / a
418 function.appendScalar(a);
419 function.append(" div\n");
420
421 // second root = c / q
422 function.append("3 1 roll div\n");
423
424 // put the larger root on top of the stack
425 function.append("2 copy gt {exch} if\n");
426
427 // compute radius(t) for larger root
428 function.append("dup ");
429 function.appendScalar(dr);
430 function.append(" mul ");
431 function.appendScalar(r0);
432 function.append(" add\n");
433
434 // if r(t) > 0, we have our t, pop off the smaller root and we're done
435 function.append(" 0 gt {exch pop true}\n");
436
437 // otherwise, throw out the larger one and try the smaller root
438 function.append("{pop dup\n");
439 function.appendScalar(dr);
440 function.append(" mul ");
441 function.appendScalar(r0);
442 function.append(" add\n");
443
444 // if r(t) < 0, push false, otherwise the smaller root is our t
rileya@google.comab2fe822012-07-10 20:41:38 +0000445 function.append("0 le {pop false} {true} ifelse\n");
rileya@google.com62197282012-07-10 20:05:23 +0000446 function.append("} ifelse\n");
447
448 // d < 0, clear the stack and push false
rileya@google.comab2fe822012-07-10 20:41:38 +0000449 function.append("} {pop pop pop false} ifelse\n");
rileya@google.com62197282012-07-10 20:05:23 +0000450 }
451
452 // if the pixel is in the cone, proceed to compute a color
453 function.append("{");
454 tileModeCode(info.fTileMode, &function);
455 gradientFunctionCode(info, &function);
456
457 // otherwise, just write black
458 function.append("} {0 0 0} ifelse }");
459
460 return function;
461}
462
edisonn@google.com83d8eda2013-10-24 13:19:28 +0000463static SkString sweepCode(const SkShader::GradientInfo& info,
464 const SkMatrix& perspectiveRemover) {
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000465 SkString function("{exch atan 360 div\n");
466 tileModeCode(info.fTileMode, &function);
467 gradientFunctionCode(info, &function);
468 function.append("}");
469 return function;
470}
471
Florin Malitac54d8db2014-12-10 12:02:16 -0500472static void drawBitmapMatrix(SkCanvas* canvas, const SkBitmap& bm, const SkMatrix& matrix) {
473 SkAutoCanvasRestore acr(canvas, true);
474 canvas->concat(matrix);
475 canvas->drawBitmap(bm, 0, 0);
476}
477
vandebo@chromium.org421d6442011-07-20 17:39:01 +0000478class SkPDFShader::State {
479public:
480 SkShader::GradientType fType;
481 SkShader::GradientInfo fInfo;
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000482 SkAutoFree fColorData; // This provides storage for arrays in fInfo.
vandebo@chromium.org421d6442011-07-20 17:39:01 +0000483 SkMatrix fCanvasTransform;
484 SkMatrix fShaderTransform;
485 SkIRect fBBox;
486
487 SkBitmap fImage;
488 uint32_t fPixelGeneration;
489 SkShader::TileMode fImageTileModes[2];
490
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000491 State(const SkShader& shader, const SkMatrix& canvasTransform,
fmalitac3796c72015-01-13 08:06:11 -0800492 const SkIRect& bbox, SkScalar rasterScale);
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000493
vandebo@chromium.org421d6442011-07-20 17:39:01 +0000494 bool operator==(const State& b) const;
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000495
496 SkPDFShader::State* CreateAlphaToLuminosityState() const;
497 SkPDFShader::State* CreateOpaqueState() const;
498
499 bool GradientHasAlpha() const;
500
501private:
502 State(const State& other);
503 State operator=(const State& rhs);
504 void AllocateGradientInfoStorage();
vandebo@chromium.org421d6442011-07-20 17:39:01 +0000505};
506
halcanary530ea8e2015-01-23 06:17:35 -0800507////////////////////////////////////////////////////////////////////////////////
508
halcanary2e3f9d82015-02-27 12:41:03 -0800509SkPDFFunctionShader::SkPDFFunctionShader(SkPDFShader::State* state)
510 : SkPDFDict("Pattern"), fShaderState(state) {}
halcanary530ea8e2015-01-23 06:17:35 -0800511
halcanary530ea8e2015-01-23 06:17:35 -0800512SkPDFFunctionShader::~SkPDFFunctionShader() {
halcanary530ea8e2015-01-23 06:17:35 -0800513 fResources.unrefAll();
halcanarybc59ac62015-01-23 04:18:53 -0800514}
515
halcanary530ea8e2015-01-23 06:17:35 -0800516bool SkPDFFunctionShader::equals(const SkPDFShader::State& state) const {
517 return state == *fShaderState;
halcanaryfb62b3d2015-01-21 09:59:14 -0800518}
519
halcanary530ea8e2015-01-23 06:17:35 -0800520////////////////////////////////////////////////////////////////////////////////
521
halcanary2e3f9d82015-02-27 12:41:03 -0800522SkPDFAlphaFunctionShader::SkPDFAlphaFunctionShader(SkPDFShader::State* state)
523 : fShaderState(state) {}
halcanary530ea8e2015-01-23 06:17:35 -0800524
525bool SkPDFAlphaFunctionShader::equals(const SkPDFShader::State& state) const {
526 return state == *fShaderState;
527}
528
halcanary2e3f9d82015-02-27 12:41:03 -0800529SkPDFAlphaFunctionShader::~SkPDFAlphaFunctionShader() {}
halcanary530ea8e2015-01-23 06:17:35 -0800530
halcanary530ea8e2015-01-23 06:17:35 -0800531////////////////////////////////////////////////////////////////////////////////
532
halcanary2e3f9d82015-02-27 12:41:03 -0800533SkPDFImageShader::SkPDFImageShader(SkPDFShader::State* state)
534 : fShaderState(state) {}
halcanary530ea8e2015-01-23 06:17:35 -0800535
536bool SkPDFImageShader::equals(const SkPDFShader::State& state) const {
537 return state == *fShaderState;
538}
539
540SkPDFImageShader::~SkPDFImageShader() {
halcanary530ea8e2015-01-23 06:17:35 -0800541 fResources.unrefAll();
542}
543
halcanary530ea8e2015-01-23 06:17:35 -0800544////////////////////////////////////////////////////////////////////////////////
545
546static SkPDFObject* get_pdf_shader_by_state(
halcanary792c80f2015-02-20 07:21:05 -0800547 SkPDFCanon* canon,
548 SkScalar dpi,
halcanarybc59ac62015-01-23 04:18:53 -0800549 SkAutoTDelete<SkPDFShader::State>* autoState) {
halcanary530ea8e2015-01-23 06:17:35 -0800550 const SkPDFShader::State& state = **autoState;
halcanarybc59ac62015-01-23 04:18:53 -0800551 if (state.fType == SkShader::kNone_GradientType && state.fImage.isNull()) {
vandebo@chromium.orgda6c5692012-06-28 21:37:20 +0000552 // TODO(vandebo) This drops SKComposeShader on the floor. We could
553 // handle compose shader by pulling things up to a layer, drawing with
554 // the first shader, applying the xfer mode and drawing again with the
555 // second shader, then applying the layer to the original drawing.
556 return NULL;
halcanary530ea8e2015-01-23 06:17:35 -0800557 } else if (state.fType == SkShader::kNone_GradientType) {
halcanary792c80f2015-02-20 07:21:05 -0800558 SkPDFObject* shader = canon->findImageShader(state);
halcanary530ea8e2015-01-23 06:17:35 -0800559 return shader ? SkRef(shader)
halcanary792c80f2015-02-20 07:21:05 -0800560 : SkPDFImageShader::Create(canon, dpi, autoState);
561 } else if (state.GradientHasAlpha()) {
562 SkPDFObject* shader = canon->findAlphaShader(state);
563 return shader ? SkRef(shader)
564 : SkPDFAlphaFunctionShader::Create(canon, dpi, autoState);
vandebo@chromium.org421d6442011-07-20 17:39:01 +0000565 } else {
halcanary792c80f2015-02-20 07:21:05 -0800566 SkPDFObject* shader = canon->findFunctionShader(state);
567 return shader ? SkRef(shader)
568 : SkPDFFunctionShader::Create(canon, autoState);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000569 }
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000570}
571
572// static
halcanary792c80f2015-02-20 07:21:05 -0800573SkPDFObject* SkPDFShader::GetPDFShader(SkPDFCanon* canon,
574 SkScalar dpi,
575 const SkShader& shader,
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000576 const SkMatrix& matrix,
fmalitac3796c72015-01-13 08:06:11 -0800577 const SkIRect& surfaceBBox,
578 SkScalar rasterScale) {
halcanarybc59ac62015-01-23 04:18:53 -0800579 SkAutoTDelete<SkPDFShader::State> state(
fmalitac3796c72015-01-13 08:06:11 -0800580 SkNEW_ARGS(State, (shader, matrix, surfaceBBox, rasterScale)));
halcanary792c80f2015-02-20 07:21:05 -0800581 return get_pdf_shader_by_state(canon, dpi, &state);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000582}
583
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000584static SkPDFResourceDict* get_gradient_resource_dict(
585 SkPDFObject* functionShader,
586 SkPDFObject* gState) {
587 SkPDFResourceDict* dict = new SkPDFResourceDict();
588
589 if (functionShader != NULL) {
590 dict->insertResourceAsReference(
591 SkPDFResourceDict::kPattern_ResourceType, 0, functionShader);
592 }
593 if (gState != NULL) {
594 dict->insertResourceAsReference(
595 SkPDFResourceDict::kExtGState_ResourceType, 0, gState);
596 }
597
598 return dict;
599}
600
601static void populate_tiling_pattern_dict(SkPDFDict* pattern,
602 SkRect& bbox, SkPDFDict* resources,
603 const SkMatrix& matrix) {
604 const int kTiling_PatternType = 1;
605 const int kColoredTilingPattern_PaintType = 1;
606 const int kConstantSpacing_TilingType = 1;
607
608 pattern->insertName("Type", "Pattern");
609 pattern->insertInt("PatternType", kTiling_PatternType);
610 pattern->insertInt("PaintType", kColoredTilingPattern_PaintType);
611 pattern->insertInt("TilingType", kConstantSpacing_TilingType);
612 pattern->insert("BBox", SkPDFUtils::RectToArray(bbox))->unref();
613 pattern->insertScalar("XStep", bbox.width());
614 pattern->insertScalar("YStep", bbox.height());
615 pattern->insert("Resources", resources);
616 if (!matrix.isIdentity()) {
617 pattern->insert("Matrix", SkPDFUtils::MatrixToArray(matrix))->unref();
618 }
619}
620
621/**
622 * Creates a content stream which fills the pattern P0 across bounds.
623 * @param gsIndex A graphics state resource index to apply, or <0 if no
624 * graphics state to apply.
625 */
626static SkStream* create_pattern_fill_content(int gsIndex, SkRect& bounds) {
627 SkDynamicMemoryWStream content;
628 if (gsIndex >= 0) {
629 SkPDFUtils::ApplyGraphicState(gsIndex, &content);
630 }
631 SkPDFUtils::ApplyPattern(0, &content);
632 SkPDFUtils::AppendRectangle(bounds, &content);
633 SkPDFUtils::PaintPath(SkPaint::kFill_Style, SkPath::kEvenOdd_FillType,
634 &content);
635
636 return content.detachAsStream();
637}
638
639/**
640 * Creates a ExtGState with the SMask set to the luminosityShader in
641 * luminosity mode. The shader pattern extends to the bbox.
642 */
halcanary8de56322015-01-23 07:22:58 -0800643static SkPDFGraphicState* create_smask_graphic_state(
halcanary792c80f2015-02-20 07:21:05 -0800644 SkPDFCanon* canon, SkScalar dpi, const SkPDFShader::State& state) {
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000645 SkRect bbox;
halcanarybc59ac62015-01-23 04:18:53 -0800646 bbox.set(state.fBBox);
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000647
halcanarybc59ac62015-01-23 04:18:53 -0800648 SkAutoTDelete<SkPDFShader::State> alphaToLuminosityState(
649 state.CreateAlphaToLuminosityState());
650 SkAutoTUnref<SkPDFObject> luminosityShader(
halcanary792c80f2015-02-20 07:21:05 -0800651 get_pdf_shader_by_state(canon, dpi, &alphaToLuminosityState));
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000652
scroggoa1193e42015-01-21 12:09:53 -0800653 SkAutoTDelete<SkStream> alphaStream(create_pattern_fill_content(-1, bbox));
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000654
655 SkAutoTUnref<SkPDFResourceDict>
656 resources(get_gradient_resource_dict(luminosityShader, NULL));
657
658 SkAutoTUnref<SkPDFFormXObject> alphaMask(
659 new SkPDFFormXObject(alphaStream.get(), bbox, resources.get()));
660
661 return SkPDFGraphicState::GetSMaskGraphicState(
662 alphaMask.get(), false,
663 SkPDFGraphicState::kLuminosity_SMaskMode);
664}
665
halcanary530ea8e2015-01-23 06:17:35 -0800666SkPDFAlphaFunctionShader* SkPDFAlphaFunctionShader::Create(
halcanary792c80f2015-02-20 07:21:05 -0800667 SkPDFCanon* canon,
668 SkScalar dpi,
halcanarybc59ac62015-01-23 04:18:53 -0800669 SkAutoTDelete<SkPDFShader::State>* autoState) {
670 const SkPDFShader::State& state = **autoState;
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000671 SkRect bbox;
halcanarybc59ac62015-01-23 04:18:53 -0800672 bbox.set(state.fBBox);
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000673
halcanarybc59ac62015-01-23 04:18:53 -0800674 SkAutoTDelete<SkPDFShader::State> opaqueState(state.CreateOpaqueState());
675
halcanary792c80f2015-02-20 07:21:05 -0800676 SkPDFObject* colorShader =
677 get_pdf_shader_by_state(canon, dpi, &opaqueState);
halcanarybc59ac62015-01-23 04:18:53 -0800678 if (!colorShader) {
679 return NULL;
680 }
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000681
682 // Create resource dict with alpha graphics state as G0 and
683 // pattern shader as P0, then write content stream.
halcanary792c80f2015-02-20 07:21:05 -0800684 SkAutoTUnref<SkPDFGraphicState> alphaGs(
685 create_smask_graphic_state(canon, dpi, state));
halcanarybc59ac62015-01-23 04:18:53 -0800686
687 SkPDFAlphaFunctionShader* alphaFunctionShader =
halcanary2e3f9d82015-02-27 12:41:03 -0800688 SkNEW_ARGS(SkPDFAlphaFunctionShader, (autoState->detach()));
halcanarybc59ac62015-01-23 04:18:53 -0800689
690 alphaFunctionShader->fColorShader.reset(colorShader);
691
692 alphaFunctionShader->fResourceDict.reset(get_gradient_resource_dict(
693 alphaFunctionShader->fColorShader.get(), alphaGs.get()));
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000694
scroggoa1193e42015-01-21 12:09:53 -0800695 SkAutoTDelete<SkStream> colorStream(
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000696 create_pattern_fill_content(0, bbox));
halcanarybc59ac62015-01-23 04:18:53 -0800697 alphaFunctionShader->setData(colorStream.get());
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000698
halcanarybc59ac62015-01-23 04:18:53 -0800699 populate_tiling_pattern_dict(alphaFunctionShader, bbox,
700 alphaFunctionShader->fResourceDict.get(),
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000701 SkMatrix::I());
halcanary792c80f2015-02-20 07:21:05 -0800702 canon->addAlphaShader(alphaFunctionShader);
halcanarybc59ac62015-01-23 04:18:53 -0800703 return alphaFunctionShader;
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +0000704}
705
edisonn@google.com83d8eda2013-10-24 13:19:28 +0000706// Finds affine and persp such that in = affine * persp.
707// but it returns the inverse of perspective matrix.
708static bool split_perspective(const SkMatrix in, SkMatrix* affine,
709 SkMatrix* perspectiveInverse) {
710 const SkScalar p2 = in[SkMatrix::kMPersp2];
711
712 if (SkScalarNearlyZero(p2)) {
713 return false;
714 }
715
716 const SkScalar zero = SkIntToScalar(0);
717 const SkScalar one = SkIntToScalar(1);
718
719 const SkScalar sx = in[SkMatrix::kMScaleX];
720 const SkScalar kx = in[SkMatrix::kMSkewX];
721 const SkScalar tx = in[SkMatrix::kMTransX];
722 const SkScalar ky = in[SkMatrix::kMSkewY];
723 const SkScalar sy = in[SkMatrix::kMScaleY];
724 const SkScalar ty = in[SkMatrix::kMTransY];
725 const SkScalar p0 = in[SkMatrix::kMPersp0];
726 const SkScalar p1 = in[SkMatrix::kMPersp1];
727
728 // Perspective matrix would be:
729 // 1 0 0
730 // 0 1 0
731 // p0 p1 p2
732 // But we need the inverse of persp.
733 perspectiveInverse->setAll(one, zero, zero,
734 zero, one, zero,
735 -p0/p2, -p1/p2, 1/p2);
736
737 affine->setAll(sx - p0 * tx / p2, kx - p1 * tx / p2, tx / p2,
738 ky - p0 * ty / p2, sy - p1 * ty / p2, ty / p2,
739 zero, zero, one);
740
741 return true;
742}
743
halcanarybc59ac62015-01-23 04:18:53 -0800744namespace {
745SkPDFObject* create_range_object() {
746 SkPDFArray* range = SkNEW(SkPDFArray);
747 range->reserve(6);
748 range->appendInt(0);
749 range->appendInt(1);
750 range->appendInt(0);
751 range->appendInt(1);
752 range->appendInt(0);
753 range->appendInt(1);
754 return range;
755}
756
757template <typename T> void unref(T* ptr) { ptr->unref();}
758} // namespace
759
760SK_DECLARE_STATIC_LAZY_PTR(SkPDFObject, rangeObject,
761 create_range_object, unref<SkPDFObject>);
762
763static SkPDFStream* make_ps_function(const SkString& psCode,
764 SkPDFArray* domain) {
765 SkAutoDataUnref funcData(
766 SkData::NewWithCopy(psCode.c_str(), psCode.size()));
767 SkPDFStream* result = SkNEW_ARGS(SkPDFStream, (funcData.get()));
768 result->insertInt("FunctionType", 4);
769 result->insert("Domain", domain);
770 result->insert("Range", rangeObject.get());
771 return result;
772}
773
halcanary530ea8e2015-01-23 06:17:35 -0800774SkPDFFunctionShader* SkPDFFunctionShader::Create(
halcanary792c80f2015-02-20 07:21:05 -0800775 SkPDFCanon* canon, SkAutoTDelete<SkPDFShader::State>* autoState) {
halcanarybc59ac62015-01-23 04:18:53 -0800776 const SkPDFShader::State& state = **autoState;
777
edisonn@google.com83d8eda2013-10-24 13:19:28 +0000778 SkString (*codeFunction)(const SkShader::GradientInfo& info,
779 const SkMatrix& perspectiveRemover) = NULL;
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000780 SkPoint transformPoints[2];
781
782 // Depending on the type of the gradient, we want to transform the
783 // coordinate space in different ways.
halcanarybc59ac62015-01-23 04:18:53 -0800784 const SkShader::GradientInfo* info = &state.fInfo;
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000785 transformPoints[0] = info->fPoint[0];
786 transformPoints[1] = info->fPoint[1];
halcanarybc59ac62015-01-23 04:18:53 -0800787 switch (state.fType) {
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000788 case SkShader::kLinear_GradientType:
789 codeFunction = &linearCode;
790 break;
791 case SkShader::kRadial_GradientType:
792 transformPoints[1] = transformPoints[0];
793 transformPoints[1].fX += info->fRadius[0];
794 codeFunction = &radialCode;
795 break;
796 case SkShader::kRadial2_GradientType: {
halcanarybc59ac62015-01-23 04:18:53 -0800797 // Bail out if the radii are the same.
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000798 if (info->fRadius[0] == info->fRadius[1]) {
halcanarybc59ac62015-01-23 04:18:53 -0800799 return NULL;
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000800 }
801 transformPoints[1] = transformPoints[0];
802 SkScalar dr = info->fRadius[1] - info->fRadius[0];
803 transformPoints[1].fX += dr;
804 codeFunction = &twoPointRadialCode;
805 break;
806 }
rileya@google.com62197282012-07-10 20:05:23 +0000807 case SkShader::kConical_GradientType: {
808 transformPoints[1] = transformPoints[0];
809 transformPoints[1].fX += SK_Scalar1;
810 codeFunction = &twoPointConicalCode;
811 break;
812 }
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000813 case SkShader::kSweep_GradientType:
814 transformPoints[1] = transformPoints[0];
vandebo@chromium.orgc39c8672012-04-17 21:46:18 +0000815 transformPoints[1].fX += SK_Scalar1;
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000816 codeFunction = &sweepCode;
817 break;
818 case SkShader::kColor_GradientType:
819 case SkShader::kNone_GradientType:
vandebo@chromium.org020798a2011-12-21 01:49:35 +0000820 default:
halcanarybc59ac62015-01-23 04:18:53 -0800821 return NULL;
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000822 }
823
824 // Move any scaling (assuming a unit gradient) or translation
825 // (and rotation for linear gradient), of the final gradient from
826 // info->fPoints to the matrix (updating bbox appropriately). Now
827 // the gradient can be drawn on on the unit segment.
828 SkMatrix mapperMatrix;
829 unitToPointsMatrix(transformPoints, &mapperMatrix);
edisonn@google.com83d8eda2013-10-24 13:19:28 +0000830
halcanarybc59ac62015-01-23 04:18:53 -0800831 SkMatrix finalMatrix = state.fCanvasTransform;
832 finalMatrix.preConcat(state.fShaderTransform);
commit-bot@chromium.orgace22692013-06-12 21:33:02 +0000833 finalMatrix.preConcat(mapperMatrix);
834
edisonn@google.com83d8eda2013-10-24 13:19:28 +0000835 // Preserves as much as posible in the final matrix, and only removes
836 // the perspective. The inverse of the perspective is stored in
837 // perspectiveInverseOnly matrix and has 3 useful numbers
838 // (p0, p1, p2), while everything else is either 0 or 1.
839 // In this way the shader will handle it eficiently, with minimal code.
840 SkMatrix perspectiveInverseOnly = SkMatrix::I();
841 if (finalMatrix.hasPerspective()) {
842 if (!split_perspective(finalMatrix,
843 &finalMatrix, &perspectiveInverseOnly)) {
halcanarybc59ac62015-01-23 04:18:53 -0800844 return NULL;
edisonn@google.com83d8eda2013-10-24 13:19:28 +0000845 }
846 }
847
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000848 SkRect bbox;
halcanarybc59ac62015-01-23 04:18:53 -0800849 bbox.set(state.fBBox);
fmalita7b4d4c72015-02-12 13:56:50 -0800850 if (!inverse_transform_bbox(finalMatrix, &bbox)) {
halcanarybc59ac62015-01-23 04:18:53 -0800851 return NULL;
vandebo@chromium.org386dfc02012-04-17 22:31:52 +0000852 }
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000853
vandebo@chromium.orgd96d17b2013-01-04 19:31:24 +0000854 SkAutoTUnref<SkPDFArray> domain(new SkPDFArray);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000855 domain->reserve(4);
reed@google.comc789cf12011-07-20 12:14:33 +0000856 domain->appendScalar(bbox.fLeft);
857 domain->appendScalar(bbox.fRight);
858 domain->appendScalar(bbox.fTop);
859 domain->appendScalar(bbox.fBottom);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000860
861 SkString functionCode;
halcanaryfb62b3d2015-01-21 09:59:14 -0800862 // The two point radial gradient further references
halcanarybc59ac62015-01-23 04:18:53 -0800863 // state.fInfo
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000864 // in translating from x, y coordinates to the t parameter. So, we have
865 // to transform the points and radii according to the calculated matrix.
halcanarybc59ac62015-01-23 04:18:53 -0800866 if (state.fType == SkShader::kRadial2_GradientType) {
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000867 SkShader::GradientInfo twoPointRadialInfo = *info;
868 SkMatrix inverseMapperMatrix;
vandebo@chromium.orgb0549902012-04-13 20:45:46 +0000869 if (!mapperMatrix.invert(&inverseMapperMatrix)) {
halcanarybc59ac62015-01-23 04:18:53 -0800870 return NULL;
vandebo@chromium.orgb0549902012-04-13 20:45:46 +0000871 }
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000872 inverseMapperMatrix.mapPoints(twoPointRadialInfo.fPoint, 2);
873 twoPointRadialInfo.fRadius[0] =
874 inverseMapperMatrix.mapRadius(info->fRadius[0]);
875 twoPointRadialInfo.fRadius[1] =
876 inverseMapperMatrix.mapRadius(info->fRadius[1]);
edisonn@google.com83d8eda2013-10-24 13:19:28 +0000877 functionCode = codeFunction(twoPointRadialInfo, perspectiveInverseOnly);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000878 } else {
edisonn@google.com83d8eda2013-10-24 13:19:28 +0000879 functionCode = codeFunction(*info, perspectiveInverseOnly);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000880 }
881
vandebo@chromium.orgd96d17b2013-01-04 19:31:24 +0000882 SkAutoTUnref<SkPDFDict> pdfShader(new SkPDFDict);
reed@google.comc789cf12011-07-20 12:14:33 +0000883 pdfShader->insertInt("ShadingType", 1);
884 pdfShader->insertName("ColorSpace", "DeviceRGB");
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000885 pdfShader->insert("Domain", domain.get());
vandebo@chromium.orgd96d17b2013-01-04 19:31:24 +0000886
halcanarybc59ac62015-01-23 04:18:53 -0800887 SkPDFStream* function = make_ps_function(functionCode, domain.get());
vandebo@chromium.orgd96d17b2013-01-04 19:31:24 +0000888 pdfShader->insert("Function", new SkPDFObjRef(function))->unref();
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000889
halcanarybc59ac62015-01-23 04:18:53 -0800890 SkAutoTUnref<SkPDFArray> matrixArray(
891 SkPDFUtils::MatrixToArray(finalMatrix));
892
893 SkPDFFunctionShader* pdfFunctionShader =
halcanary2e3f9d82015-02-27 12:41:03 -0800894 SkNEW_ARGS(SkPDFFunctionShader, (autoState->detach()));
halcanarybc59ac62015-01-23 04:18:53 -0800895
896 pdfFunctionShader->fResources.push(function);
897 // Pass ownership to resource list.
898
899 pdfFunctionShader->insertInt("PatternType", 2);
900 pdfFunctionShader->insert("Matrix", matrixArray.get());
901 pdfFunctionShader->insert("Shading", pdfShader.get());
902
halcanary792c80f2015-02-20 07:21:05 -0800903 canon->addFunctionShader(pdfFunctionShader);
halcanarybc59ac62015-01-23 04:18:53 -0800904 return pdfFunctionShader;
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000905}
906
halcanary530ea8e2015-01-23 06:17:35 -0800907SkPDFImageShader* SkPDFImageShader::Create(
halcanary792c80f2015-02-20 07:21:05 -0800908 SkPDFCanon* canon,
909 SkScalar dpi,
halcanarybc59ac62015-01-23 04:18:53 -0800910 SkAutoTDelete<SkPDFShader::State>* autoState) {
911 const SkPDFShader::State& state = **autoState;
912
913 state.fImage.lockPixels();
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000914
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +0000915 // The image shader pattern cell will be drawn into a separate device
916 // in pattern cell space (no scaling on the bitmap, though there may be
917 // translations so that all content is in the device, coordinates > 0).
918
919 // Map clip bounds to shader space to ensure the device is large enough
920 // to handle fake clamping.
halcanarybc59ac62015-01-23 04:18:53 -0800921 SkMatrix finalMatrix = state.fCanvasTransform;
922 finalMatrix.preConcat(state.fShaderTransform);
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +0000923 SkRect deviceBounds;
halcanarybc59ac62015-01-23 04:18:53 -0800924 deviceBounds.set(state.fBBox);
fmalita7b4d4c72015-02-12 13:56:50 -0800925 if (!inverse_transform_bbox(finalMatrix, &deviceBounds)) {
halcanarybc59ac62015-01-23 04:18:53 -0800926 return NULL;
vandebo@chromium.org386dfc02012-04-17 22:31:52 +0000927 }
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000928
halcanarybc59ac62015-01-23 04:18:53 -0800929 const SkBitmap* image = &state.fImage;
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +0000930 SkRect bitmapBounds;
931 image->getBounds(&bitmapBounds);
932
933 // For tiling modes, the bounds should be extended to include the bitmap,
934 // otherwise the bitmap gets clipped out and the shader is empty and awful.
935 // For clamp modes, we're only interested in the clip region, whether
936 // or not the main bitmap is in it.
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000937 SkShader::TileMode tileModes[2];
halcanarybc59ac62015-01-23 04:18:53 -0800938 tileModes[0] = state.fImageTileModes[0];
939 tileModes[1] = state.fImageTileModes[1];
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +0000940 if (tileModes[0] != SkShader::kClamp_TileMode ||
941 tileModes[1] != SkShader::kClamp_TileMode) {
942 deviceBounds.join(bitmapBounds);
943 }
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000944
reed@google.come1ca7052013-12-17 19:22:07 +0000945 SkISize size = SkISize::Make(SkScalarRoundToInt(deviceBounds.width()),
946 SkScalarRoundToInt(deviceBounds.height()));
halcanarya1f1ee92015-02-20 06:17:26 -0800947 SkAutoTUnref<SkPDFDevice> patternDevice(
halcanary792c80f2015-02-20 07:21:05 -0800948 SkPDFDevice::CreateUnflipped(size, dpi, canon));
halcanarya1f1ee92015-02-20 06:17:26 -0800949 SkCanvas canvas(patternDevice.get());
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +0000950
951 SkRect patternBBox;
952 image->getBounds(&patternBBox);
953
954 // Translate the canvas so that the bitmap origin is at (0, 0).
955 canvas.translate(-deviceBounds.left(), -deviceBounds.top());
956 patternBBox.offset(-deviceBounds.left(), -deviceBounds.top());
957 // Undo the translation in the final matrix
958 finalMatrix.preTranslate(deviceBounds.left(), deviceBounds.top());
959
960 // If the bitmap is out of bounds (i.e. clamp mode where we only see the
961 // stretched sides), canvas will clip this out and the extraneous data
962 // won't be saved to the PDF.
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000963 canvas.drawBitmap(*image, 0, 0);
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +0000964
965 SkScalar width = SkIntToScalar(image->width());
966 SkScalar height = SkIntToScalar(image->height());
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000967
968 // Tiling is implied. First we handle mirroring.
969 if (tileModes[0] == SkShader::kMirror_TileMode) {
970 SkMatrix xMirror;
971 xMirror.setScale(-1, 1);
972 xMirror.postTranslate(2 * width, 0);
Florin Malitac54d8db2014-12-10 12:02:16 -0500973 drawBitmapMatrix(&canvas, *image, xMirror);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000974 patternBBox.fRight += width;
975 }
976 if (tileModes[1] == SkShader::kMirror_TileMode) {
977 SkMatrix yMirror;
vandebo@chromium.org663515b2012-01-05 18:45:27 +0000978 yMirror.setScale(SK_Scalar1, -SK_Scalar1);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000979 yMirror.postTranslate(0, 2 * height);
Florin Malitac54d8db2014-12-10 12:02:16 -0500980 drawBitmapMatrix(&canvas, *image, yMirror);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000981 patternBBox.fBottom += height;
982 }
983 if (tileModes[0] == SkShader::kMirror_TileMode &&
984 tileModes[1] == SkShader::kMirror_TileMode) {
985 SkMatrix mirror;
986 mirror.setScale(-1, -1);
987 mirror.postTranslate(2 * width, 2 * height);
Florin Malitac54d8db2014-12-10 12:02:16 -0500988 drawBitmapMatrix(&canvas, *image, mirror);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +0000989 }
990
991 // Then handle Clamping, which requires expanding the pattern canvas to
992 // cover the entire surfaceBBox.
993
994 // If both x and y are in clamp mode, we start by filling in the corners.
995 // (Which are just a rectangles of the corner colors.)
996 if (tileModes[0] == SkShader::kClamp_TileMode &&
997 tileModes[1] == SkShader::kClamp_TileMode) {
998 SkPaint paint;
999 SkRect rect;
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +00001000 rect = SkRect::MakeLTRB(deviceBounds.left(), deviceBounds.top(), 0, 0);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001001 if (!rect.isEmpty()) {
1002 paint.setColor(image->getColor(0, 0));
1003 canvas.drawRect(rect, paint);
1004 }
1005
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +00001006 rect = SkRect::MakeLTRB(width, deviceBounds.top(),
1007 deviceBounds.right(), 0);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001008 if (!rect.isEmpty()) {
vandebo@chromium.org54ff85c2012-03-09 17:18:50 +00001009 paint.setColor(image->getColor(image->width() - 1, 0));
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001010 canvas.drawRect(rect, paint);
1011 }
1012
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +00001013 rect = SkRect::MakeLTRB(width, height,
1014 deviceBounds.right(), deviceBounds.bottom());
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001015 if (!rect.isEmpty()) {
vandebo@chromium.org54ff85c2012-03-09 17:18:50 +00001016 paint.setColor(image->getColor(image->width() - 1,
1017 image->height() - 1));
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001018 canvas.drawRect(rect, paint);
1019 }
1020
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +00001021 rect = SkRect::MakeLTRB(deviceBounds.left(), height,
1022 0, deviceBounds.bottom());
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001023 if (!rect.isEmpty()) {
vandebo@chromium.org54ff85c2012-03-09 17:18:50 +00001024 paint.setColor(image->getColor(0, image->height() - 1));
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001025 canvas.drawRect(rect, paint);
1026 }
1027 }
1028
1029 // Then expand the left, right, top, then bottom.
1030 if (tileModes[0] == SkShader::kClamp_TileMode) {
vandebo@chromium.org54ff85c2012-03-09 17:18:50 +00001031 SkIRect subset = SkIRect::MakeXYWH(0, 0, 1, image->height());
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +00001032 if (deviceBounds.left() < 0) {
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001033 SkBitmap left;
1034 SkAssertResult(image->extractSubset(&left, subset));
1035
1036 SkMatrix leftMatrix;
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +00001037 leftMatrix.setScale(-deviceBounds.left(), 1);
1038 leftMatrix.postTranslate(deviceBounds.left(), 0);
Florin Malitac54d8db2014-12-10 12:02:16 -05001039 drawBitmapMatrix(&canvas, left, leftMatrix);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001040
1041 if (tileModes[1] == SkShader::kMirror_TileMode) {
vandebo@chromium.org663515b2012-01-05 18:45:27 +00001042 leftMatrix.postScale(SK_Scalar1, -SK_Scalar1);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001043 leftMatrix.postTranslate(0, 2 * height);
Florin Malitac54d8db2014-12-10 12:02:16 -05001044 drawBitmapMatrix(&canvas, left, leftMatrix);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001045 }
vandebo@chromium.orgbe2048a2011-05-02 15:24:01 +00001046 patternBBox.fLeft = 0;
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001047 }
1048
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +00001049 if (deviceBounds.right() > width) {
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001050 SkBitmap right;
vandebo@chromium.org54ff85c2012-03-09 17:18:50 +00001051 subset.offset(image->width() - 1, 0);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001052 SkAssertResult(image->extractSubset(&right, subset));
1053
1054 SkMatrix rightMatrix;
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +00001055 rightMatrix.setScale(deviceBounds.right() - width, 1);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001056 rightMatrix.postTranslate(width, 0);
Florin Malitac54d8db2014-12-10 12:02:16 -05001057 drawBitmapMatrix(&canvas, right, rightMatrix);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001058
1059 if (tileModes[1] == SkShader::kMirror_TileMode) {
vandebo@chromium.org663515b2012-01-05 18:45:27 +00001060 rightMatrix.postScale(SK_Scalar1, -SK_Scalar1);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001061 rightMatrix.postTranslate(0, 2 * height);
Florin Malitac54d8db2014-12-10 12:02:16 -05001062 drawBitmapMatrix(&canvas, right, rightMatrix);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001063 }
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +00001064 patternBBox.fRight = deviceBounds.width();
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001065 }
1066 }
1067
1068 if (tileModes[1] == SkShader::kClamp_TileMode) {
vandebo@chromium.org54ff85c2012-03-09 17:18:50 +00001069 SkIRect subset = SkIRect::MakeXYWH(0, 0, image->width(), 1);
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +00001070 if (deviceBounds.top() < 0) {
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001071 SkBitmap top;
1072 SkAssertResult(image->extractSubset(&top, subset));
1073
1074 SkMatrix topMatrix;
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +00001075 topMatrix.setScale(SK_Scalar1, -deviceBounds.top());
1076 topMatrix.postTranslate(0, deviceBounds.top());
Florin Malitac54d8db2014-12-10 12:02:16 -05001077 drawBitmapMatrix(&canvas, top, topMatrix);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001078
1079 if (tileModes[0] == SkShader::kMirror_TileMode) {
1080 topMatrix.postScale(-1, 1);
1081 topMatrix.postTranslate(2 * width, 0);
Florin Malitac54d8db2014-12-10 12:02:16 -05001082 drawBitmapMatrix(&canvas, top, topMatrix);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001083 }
vandebo@chromium.orgbe2048a2011-05-02 15:24:01 +00001084 patternBBox.fTop = 0;
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001085 }
1086
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +00001087 if (deviceBounds.bottom() > height) {
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001088 SkBitmap bottom;
vandebo@chromium.org54ff85c2012-03-09 17:18:50 +00001089 subset.offset(0, image->height() - 1);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001090 SkAssertResult(image->extractSubset(&bottom, subset));
1091
1092 SkMatrix bottomMatrix;
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +00001093 bottomMatrix.setScale(SK_Scalar1, deviceBounds.bottom() - height);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001094 bottomMatrix.postTranslate(0, height);
Florin Malitac54d8db2014-12-10 12:02:16 -05001095 drawBitmapMatrix(&canvas, bottom, bottomMatrix);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001096
1097 if (tileModes[0] == SkShader::kMirror_TileMode) {
1098 bottomMatrix.postScale(-1, 1);
1099 bottomMatrix.postTranslate(2 * width, 0);
Florin Malitac54d8db2014-12-10 12:02:16 -05001100 drawBitmapMatrix(&canvas, bottom, bottomMatrix);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001101 }
commit-bot@chromium.orge324cc62013-08-21 23:10:45 +00001102 patternBBox.fBottom = deviceBounds.height();
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001103 }
1104 }
1105
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001106 // Put the canvas into the pattern stream (fContent).
halcanary334fcbc2015-02-24 12:56:16 -08001107 SkAutoTDelete<SkStreamAsset> content(patternDevice->content());
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001108
halcanarybc59ac62015-01-23 04:18:53 -08001109 SkPDFImageShader* imageShader =
halcanary2e3f9d82015-02-27 12:41:03 -08001110 SkNEW_ARGS(SkPDFImageShader, (autoState->detach()));
halcanarybc59ac62015-01-23 04:18:53 -08001111 imageShader->setData(content.get());
1112
halcanary792c80f2015-02-20 07:21:05 -08001113 populate_tiling_pattern_dict(imageShader, patternBBox,
1114 patternDevice->getResourceDict(), finalMatrix);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001115
halcanarybc59ac62015-01-23 04:18:53 -08001116 imageShader->fShaderState->fImage.unlockPixels();
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001117
halcanary792c80f2015-02-20 07:21:05 -08001118 canon->addImageShader(imageShader);
halcanarybc59ac62015-01-23 04:18:53 -08001119 return imageShader;
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001120}
1121
1122bool SkPDFShader::State::operator==(const SkPDFShader::State& b) const {
1123 if (fType != b.fType ||
1124 fCanvasTransform != b.fCanvasTransform ||
1125 fShaderTransform != b.fShaderTransform ||
1126 fBBox != b.fBBox) {
1127 return false;
1128 }
1129
1130 if (fType == SkShader::kNone_GradientType) {
1131 if (fPixelGeneration != b.fPixelGeneration ||
1132 fPixelGeneration == 0 ||
1133 fImageTileModes[0] != b.fImageTileModes[0] ||
1134 fImageTileModes[1] != b.fImageTileModes[1]) {
1135 return false;
1136 }
1137 } else {
1138 if (fInfo.fColorCount != b.fInfo.fColorCount ||
1139 memcmp(fInfo.fColors, b.fInfo.fColors,
1140 sizeof(SkColor) * fInfo.fColorCount) != 0 ||
1141 memcmp(fInfo.fColorOffsets, b.fInfo.fColorOffsets,
1142 sizeof(SkScalar) * fInfo.fColorCount) != 0 ||
1143 fInfo.fPoint[0] != b.fInfo.fPoint[0] ||
1144 fInfo.fTileMode != b.fInfo.fTileMode) {
1145 return false;
1146 }
1147
1148 switch (fType) {
1149 case SkShader::kLinear_GradientType:
1150 if (fInfo.fPoint[1] != b.fInfo.fPoint[1]) {
1151 return false;
1152 }
1153 break;
1154 case SkShader::kRadial_GradientType:
1155 if (fInfo.fRadius[0] != b.fInfo.fRadius[0]) {
1156 return false;
1157 }
1158 break;
1159 case SkShader::kRadial2_GradientType:
reed@google.com49085332012-06-11 18:54:07 +00001160 case SkShader::kConical_GradientType:
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001161 if (fInfo.fPoint[1] != b.fInfo.fPoint[1] ||
1162 fInfo.fRadius[0] != b.fInfo.fRadius[0] ||
1163 fInfo.fRadius[1] != b.fInfo.fRadius[1]) {
1164 return false;
1165 }
1166 break;
1167 case SkShader::kSweep_GradientType:
1168 case SkShader::kNone_GradientType:
1169 case SkShader::kColor_GradientType:
1170 break;
1171 }
1172 }
1173 return true;
1174}
1175
fmalitac3796c72015-01-13 08:06:11 -08001176SkPDFShader::State::State(const SkShader& shader, const SkMatrix& canvasTransform,
1177 const SkIRect& bbox, SkScalar rasterScale)
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001178 : fCanvasTransform(canvasTransform),
vandebo@chromium.orge1bc2742011-06-21 22:26:39 +00001179 fBBox(bbox),
1180 fPixelGeneration(0) {
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001181 fInfo.fColorCount = 0;
1182 fInfo.fColors = NULL;
1183 fInfo.fColorOffsets = NULL;
bsalomon@google.comf94b3a42012-10-31 18:09:01 +00001184 fShaderTransform = shader.getLocalMatrix();
vandebo@chromium.orge1bc2742011-06-21 22:26:39 +00001185 fImageTileModes[0] = fImageTileModes[1] = SkShader::kClamp_TileMode;
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001186
1187 fType = shader.asAGradient(&fInfo);
1188
1189 if (fType == SkShader::kNone_GradientType) {
1190 SkShader::BitmapType bitmapType;
1191 SkMatrix matrix;
rileya@google.com91f319c2012-07-25 17:18:31 +00001192 bitmapType = shader.asABitmap(&fImage, &matrix, fImageTileModes);
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001193 if (bitmapType != SkShader::kDefault_BitmapType) {
fmalitac3796c72015-01-13 08:06:11 -08001194 // Generic fallback for unsupported shaders:
1195 // * allocate a bbox-sized bitmap
1196 // * shade the whole area
1197 // * use the result as a bitmap shader
1198
fmalita7b4d4c72015-02-12 13:56:50 -08001199 // bbox is in device space. While that's exactly what we want for sizing our bitmap,
1200 // we need to map it into shader space for adjustments (to match
1201 // SkPDFImageShader::Create's behavior).
1202 SkRect shaderRect = SkRect::Make(bbox);
1203 if (!inverse_transform_bbox(canvasTransform, &shaderRect)) {
1204 fImage.reset();
1205 return;
1206 }
1207
fmalitac3796c72015-01-13 08:06:11 -08001208 // Clamp the bitmap size to about 1M pixels
1209 static const SkScalar kMaxBitmapArea = 1024 * 1024;
1210 SkScalar bitmapArea = rasterScale * bbox.width() * rasterScale * bbox.height();
1211 if (bitmapArea > kMaxBitmapArea) {
1212 rasterScale *= SkScalarSqrt(SkScalarDiv(kMaxBitmapArea, bitmapArea));
1213 }
1214
1215 SkISize size = SkISize::Make(SkScalarRoundToInt(rasterScale * bbox.width()),
1216 SkScalarRoundToInt(rasterScale * bbox.height()));
fmalita7b4d4c72015-02-12 13:56:50 -08001217 SkSize scale = SkSize::Make(SkIntToScalar(size.width()) / shaderRect.width(),
1218 SkIntToScalar(size.height()) / shaderRect.height());
fmalitac3796c72015-01-13 08:06:11 -08001219
1220 fImage.allocN32Pixels(size.width(), size.height());
1221 fImage.eraseColor(SK_ColorTRANSPARENT);
1222
1223 SkPaint p;
1224 p.setShader(const_cast<SkShader*>(&shader));
1225
1226 SkCanvas canvas(fImage);
1227 canvas.scale(scale.width(), scale.height());
fmalita7b4d4c72015-02-12 13:56:50 -08001228 canvas.translate(-shaderRect.x(), -shaderRect.y());
fmalitac3796c72015-01-13 08:06:11 -08001229 canvas.drawPaint(p);
1230
fmalita7b4d4c72015-02-12 13:56:50 -08001231 fShaderTransform.setTranslate(shaderRect.x(), shaderRect.y());
fmalitac3796c72015-01-13 08:06:11 -08001232 fShaderTransform.preScale(1 / scale.width(), 1 / scale.height());
1233 } else {
1234 SkASSERT(matrix.isIdentity());
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001235 }
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001236 fPixelGeneration = fImage.getGenerationID();
1237 } else {
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +00001238 AllocateGradientInfoStorage();
vandebo@chromium.orgda912d62011-03-08 18:31:02 +00001239 shader.asAGradient(&fInfo);
1240 }
1241}
commit-bot@chromium.org93a2e212013-07-23 23:16:03 +00001242
1243SkPDFShader::State::State(const SkPDFShader::State& other)
1244 : fType(other.fType),
1245 fCanvasTransform(other.fCanvasTransform),
1246 fShaderTransform(other.fShaderTransform),
1247 fBBox(other.fBBox)
1248{
1249 // Only gradients supported for now, since that is all that is used.
1250 // If needed, image state copy constructor can be added here later.
1251 SkASSERT(fType != SkShader::kNone_GradientType);
1252
1253 if (fType != SkShader::kNone_GradientType) {
1254 fInfo = other.fInfo;
1255
1256 AllocateGradientInfoStorage();
1257 for (int i = 0; i < fInfo.fColorCount; i++) {
1258 fInfo.fColors[i] = other.fInfo.fColors[i];
1259 fInfo.fColorOffsets[i] = other.fInfo.fColorOffsets[i];
1260 }
1261 }
1262}
1263
1264/**
1265 * Create a copy of this gradient state with alpha assigned to RGB luminousity.
1266 * Only valid for gradient states.
1267 */
1268SkPDFShader::State* SkPDFShader::State::CreateAlphaToLuminosityState() const {
1269 SkASSERT(fType != SkShader::kNone_GradientType);
1270
1271 SkPDFShader::State* newState = new SkPDFShader::State(*this);
1272
1273 for (int i = 0; i < fInfo.fColorCount; i++) {
1274 SkAlpha alpha = SkColorGetA(fInfo.fColors[i]);
1275 newState->fInfo.fColors[i] = SkColorSetARGB(255, alpha, alpha, alpha);
1276 }
1277
1278 return newState;
1279}
1280
1281/**
1282 * Create a copy of this gradient state with alpha set to fully opaque
1283 * Only valid for gradient states.
1284 */
1285SkPDFShader::State* SkPDFShader::State::CreateOpaqueState() const {
1286 SkASSERT(fType != SkShader::kNone_GradientType);
1287
1288 SkPDFShader::State* newState = new SkPDFShader::State(*this);
1289 for (int i = 0; i < fInfo.fColorCount; i++) {
1290 newState->fInfo.fColors[i] = SkColorSetA(fInfo.fColors[i],
1291 SK_AlphaOPAQUE);
1292 }
1293
1294 return newState;
1295}
1296
1297/**
1298 * Returns true if state is a gradient and the gradient has alpha.
1299 */
1300bool SkPDFShader::State::GradientHasAlpha() const {
1301 if (fType == SkShader::kNone_GradientType) {
1302 return false;
1303 }
1304
1305 for (int i = 0; i < fInfo.fColorCount; i++) {
1306 SkAlpha alpha = SkColorGetA(fInfo.fColors[i]);
1307 if (alpha != SK_AlphaOPAQUE) {
1308 return true;
1309 }
1310 }
1311 return false;
1312}
1313
1314void SkPDFShader::State::AllocateGradientInfoStorage() {
1315 fColorData.set(sk_malloc_throw(
1316 fInfo.fColorCount * (sizeof(SkColor) + sizeof(SkScalar))));
1317 fInfo.fColors = reinterpret_cast<SkColor*>(fColorData.get());
1318 fInfo.fColorOffsets =
1319 reinterpret_cast<SkScalar*>(fInfo.fColors + fInfo.fColorCount);
1320}