blob: 4f7351003a30130932341ae6a93724f9a8393e7f [file] [log] [blame]
rileya@google.com589708b2012-07-26 20:04:23 +00001
2/*
3 * Copyright 2012 Google Inc.
4 *
5 * Use of this source code is governed by a BSD-style license that can be
6 * found in the LICENSE file.
7 */
8
9 #include "SkTwoPointRadialGradient.h"
10
11/* Two-point radial gradients are specified by two circles, each with a center
12 point and radius. The gradient can be considered to be a series of
13 concentric circles, with the color interpolated from the start circle
14 (at t=0) to the end circle (at t=1).
15
16 For each point (x, y) in the span, we want to find the
17 interpolated circle that intersects that point. The center
18 of the desired circle (Cx, Cy) falls at some distance t
19 along the line segment between the start point (Sx, Sy) and
20 end point (Ex, Ey):
21
22 Cx = (1 - t) * Sx + t * Ex (0 <= t <= 1)
23 Cy = (1 - t) * Sy + t * Ey
24
25 The radius of the desired circle (r) is also a linear interpolation t
26 between the start and end radii (Sr and Er):
27
28 r = (1 - t) * Sr + t * Er
29
30 But
31
32 (x - Cx)^2 + (y - Cy)^2 = r^2
33
34 so
35
36 (x - ((1 - t) * Sx + t * Ex))^2
37 + (y - ((1 - t) * Sy + t * Ey))^2
38 = ((1 - t) * Sr + t * Er)^2
39
40 Solving for t yields
41
42 [(Sx - Ex)^2 + (Sy - Ey)^2 - (Er - Sr)^2)] * t^2
43 + [2 * (Sx - Ex)(x - Sx) + 2 * (Sy - Ey)(y - Sy) - 2 * (Er - Sr) * Sr] * t
44 + [(x - Sx)^2 + (y - Sy)^2 - Sr^2] = 0
45
46 To simplify, let Dx = Sx - Ex, Dy = Sy - Ey, Dr = Er - Sr, dx = x - Sx, dy = y - Sy
47
48 [Dx^2 + Dy^2 - Dr^2)] * t^2
49 + 2 * [Dx * dx + Dy * dy - Dr * Sr] * t
50 + [dx^2 + dy^2 - Sr^2] = 0
51
52 A quadratic in t. The two roots of the quadratic reflect the two
53 possible circles on which the point may fall. Solving for t yields
54 the gradient value to use.
55
56 If a<0, the start circle is entirely contained in the
57 end circle, and one of the roots will be <0 or >1 (off the line
58 segment). If a>0, the start circle falls at least partially
59 outside the end circle (or vice versa), and the gradient
60 defines a "tube" where a point may be on one circle (on the
61 inside of the tube) or the other (outside of the tube). We choose
62 one arbitrarily.
63
64 In order to keep the math to within the limits of fixed point,
65 we divide the entire quadratic by Dr^2, and replace
66 (x - Sx)/Dr with x' and (y - Sy)/Dr with y', giving
67
68 [Dx^2 / Dr^2 + Dy^2 / Dr^2 - 1)] * t^2
69 + 2 * [x' * Dx / Dr + y' * Dy / Dr - Sr / Dr] * t
70 + [x'^2 + y'^2 - Sr^2/Dr^2] = 0
71
72 (x' and y' are computed by appending the subtract and scale to the
73 fDstToIndex matrix in the constructor).
74
75 Since the 'A' component of the quadratic is independent of x' and y', it
76 is precomputed in the constructor. Since the 'B' component is linear in
77 x' and y', if x and y are linear in the span, 'B' can be computed
78 incrementally with a simple delta (db below). If it is not (e.g.,
79 a perspective projection), it must be computed in the loop.
80
81*/
82
83namespace {
84
85inline SkFixed two_point_radial(SkScalar b, SkScalar fx, SkScalar fy,
86 SkScalar sr2d2, SkScalar foura,
87 SkScalar oneOverTwoA, bool posRoot) {
88 SkScalar c = SkScalarSquare(fx) + SkScalarSquare(fy) - sr2d2;
89 if (0 == foura) {
90 return SkScalarToFixed(SkScalarDiv(-c, b));
91 }
92
93 SkScalar discrim = SkScalarSquare(b) - SkScalarMul(foura, c);
94 if (discrim < 0) {
95 discrim = -discrim;
96 }
97 SkScalar rootDiscrim = SkScalarSqrt(discrim);
98 SkScalar result;
99 if (posRoot) {
100 result = SkScalarMul(-b + rootDiscrim, oneOverTwoA);
101 } else {
102 result = SkScalarMul(-b - rootDiscrim, oneOverTwoA);
103 }
104 return SkScalarToFixed(result);
105}
106
107typedef void (* TwoPointRadialShadeProc)(SkScalar fx, SkScalar dx,
108 SkScalar fy, SkScalar dy,
109 SkScalar b, SkScalar db,
110 SkScalar fSr2D2, SkScalar foura, SkScalar fOneOverTwoA, bool posRoot,
111 SkPMColor* SK_RESTRICT dstC, const SkPMColor* SK_RESTRICT cache,
112 int count);
113
114void shadeSpan_twopoint_clamp(SkScalar fx, SkScalar dx,
115 SkScalar fy, SkScalar dy,
116 SkScalar b, SkScalar db,
117 SkScalar fSr2D2, SkScalar foura, SkScalar fOneOverTwoA, bool posRoot,
118 SkPMColor* SK_RESTRICT dstC, const SkPMColor* SK_RESTRICT cache,
119 int count) {
120 for (; count > 0; --count) {
121 SkFixed t = two_point_radial(b, fx, fy, fSr2D2, foura,
122 fOneOverTwoA, posRoot);
123 SkFixed index = SkClampMax(t, 0xFFFF);
124 SkASSERT(index <= 0xFFFF);
125 *dstC++ = cache[index >> SkGradientShaderBase::kCache32Shift];
126 fx += dx;
127 fy += dy;
128 b += db;
129 }
130}
131void shadeSpan_twopoint_mirror(SkScalar fx, SkScalar dx,
132 SkScalar fy, SkScalar dy,
133 SkScalar b, SkScalar db,
134 SkScalar fSr2D2, SkScalar foura, SkScalar fOneOverTwoA, bool posRoot,
135 SkPMColor* SK_RESTRICT dstC, const SkPMColor* SK_RESTRICT cache,
136 int count) {
137 for (; count > 0; --count) {
138 SkFixed t = two_point_radial(b, fx, fy, fSr2D2, foura,
139 fOneOverTwoA, posRoot);
140 SkFixed index = mirror_tileproc(t);
141 SkASSERT(index <= 0xFFFF);
142 *dstC++ = cache[index >> SkGradientShaderBase::kCache32Shift];
143 fx += dx;
144 fy += dy;
145 b += db;
146 }
147}
148
149void shadeSpan_twopoint_repeat(SkScalar fx, SkScalar dx,
150 SkScalar fy, SkScalar dy,
151 SkScalar b, SkScalar db,
152 SkScalar fSr2D2, SkScalar foura, SkScalar fOneOverTwoA, bool posRoot,
153 SkPMColor* SK_RESTRICT dstC, const SkPMColor* SK_RESTRICT cache,
154 int count) {
155 for (; count > 0; --count) {
156 SkFixed t = two_point_radial(b, fx, fy, fSr2D2, foura,
157 fOneOverTwoA, posRoot);
158 SkFixed index = repeat_tileproc(t);
159 SkASSERT(index <= 0xFFFF);
160 *dstC++ = cache[index >> SkGradientShaderBase::kCache32Shift];
161 fx += dx;
162 fy += dy;
163 b += db;
164 }
165}
166}
167
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000168/////////////////////////////////////////////////////////////////////
169
rileya@google.com589708b2012-07-26 20:04:23 +0000170SkTwoPointRadialGradient::SkTwoPointRadialGradient(
171 const SkPoint& start, SkScalar startRadius,
172 const SkPoint& end, SkScalar endRadius,
173 const SkColor colors[], const SkScalar pos[],
174 int colorCount, SkShader::TileMode mode,
175 SkUnitMapper* mapper)
176 : SkGradientShaderBase(colors, pos, colorCount, mode, mapper),
177 fCenter1(start),
178 fCenter2(end),
179 fRadius1(startRadius),
180 fRadius2(endRadius) {
181 init();
182}
183
184SkShader::BitmapType SkTwoPointRadialGradient::asABitmap(
185 SkBitmap* bitmap,
186 SkMatrix* matrix,
187 SkShader::TileMode* xy) const {
188 if (bitmap) {
rileya@google.com1c6d64b2012-07-27 15:49:05 +0000189 this->getGradientTableBitmap(bitmap);
rileya@google.com589708b2012-07-26 20:04:23 +0000190 }
191 SkScalar diffL = 0; // just to avoid gcc warning
192 if (matrix) {
193 diffL = SkScalarSqrt(SkScalarSquare(fDiff.fX) +
194 SkScalarSquare(fDiff.fY));
195 }
196 if (matrix) {
197 if (diffL) {
198 SkScalar invDiffL = SkScalarInvert(diffL);
199 matrix->setSinCos(-SkScalarMul(invDiffL, fDiff.fY),
200 SkScalarMul(invDiffL, fDiff.fX));
201 } else {
202 matrix->reset();
203 }
204 matrix->preConcat(fPtsToUnit);
205 }
206 if (xy) {
207 xy[0] = fTileMode;
208 xy[1] = kClamp_TileMode;
209 }
210 return kTwoPointRadial_BitmapType;
211}
212
213SkShader::GradientType SkTwoPointRadialGradient::asAGradient(
214 SkShader::GradientInfo* info) const {
215 if (info) {
216 commonAsAGradient(info);
217 info->fPoint[0] = fCenter1;
218 info->fPoint[1] = fCenter2;
219 info->fRadius[0] = fRadius1;
220 info->fRadius[1] = fRadius2;
221 }
222 return kRadial2_GradientType;
223}
224
rileya@google.com589708b2012-07-26 20:04:23 +0000225void SkTwoPointRadialGradient::shadeSpan(int x, int y, SkPMColor* dstCParam,
226 int count) {
227 SkASSERT(count > 0);
228
229 SkPMColor* SK_RESTRICT dstC = dstCParam;
230
231 // Zero difference between radii: fill with transparent black.
232 if (fDiffRadius == 0) {
233 sk_bzero(dstC, count * sizeof(*dstC));
234 return;
235 }
236 SkMatrix::MapXYProc dstProc = fDstToIndexProc;
237 TileProc proc = fTileProc;
238 const SkPMColor* SK_RESTRICT cache = this->getCache32();
239
240 SkScalar foura = fA * 4;
241 bool posRoot = fDiffRadius < 0;
242 if (fDstToIndexClass != kPerspective_MatrixClass) {
243 SkPoint srcPt;
244 dstProc(fDstToIndex, SkIntToScalar(x) + SK_ScalarHalf,
245 SkIntToScalar(y) + SK_ScalarHalf, &srcPt);
246 SkScalar dx, fx = srcPt.fX;
247 SkScalar dy, fy = srcPt.fY;
248
249 if (fDstToIndexClass == kFixedStepInX_MatrixClass) {
250 SkFixed fixedX, fixedY;
251 (void)fDstToIndex.fixedStepInX(SkIntToScalar(y), &fixedX, &fixedY);
252 dx = SkFixedToScalar(fixedX);
253 dy = SkFixedToScalar(fixedY);
254 } else {
255 SkASSERT(fDstToIndexClass == kLinear_MatrixClass);
256 dx = fDstToIndex.getScaleX();
257 dy = fDstToIndex.getSkewY();
258 }
259 SkScalar b = (SkScalarMul(fDiff.fX, fx) +
260 SkScalarMul(fDiff.fY, fy) - fStartRadius) * 2;
261 SkScalar db = (SkScalarMul(fDiff.fX, dx) +
262 SkScalarMul(fDiff.fY, dy)) * 2;
263
264 TwoPointRadialShadeProc shadeProc = shadeSpan_twopoint_repeat;
265 if (SkShader::kClamp_TileMode == fTileMode) {
266 shadeProc = shadeSpan_twopoint_clamp;
267 } else if (SkShader::kMirror_TileMode == fTileMode) {
268 shadeProc = shadeSpan_twopoint_mirror;
269 } else {
270 SkASSERT(SkShader::kRepeat_TileMode == fTileMode);
271 }
272 (*shadeProc)(fx, dx, fy, dy, b, db,
273 fSr2D2, foura, fOneOverTwoA, posRoot,
274 dstC, cache, count);
275 } else { // perspective case
276 SkScalar dstX = SkIntToScalar(x);
277 SkScalar dstY = SkIntToScalar(y);
278 for (; count > 0; --count) {
279 SkPoint srcPt;
280 dstProc(fDstToIndex, dstX, dstY, &srcPt);
281 SkScalar fx = srcPt.fX;
282 SkScalar fy = srcPt.fY;
283 SkScalar b = (SkScalarMul(fDiff.fX, fx) +
284 SkScalarMul(fDiff.fY, fy) - fStartRadius) * 2;
285 SkFixed t = two_point_radial(b, fx, fy, fSr2D2, foura,
286 fOneOverTwoA, posRoot);
287 SkFixed index = proc(t);
288 SkASSERT(index <= 0xFFFF);
289 *dstC++ = cache[index >> SkGradientShaderBase::kCache32Shift];
290 dstX += SK_Scalar1;
291 }
292 }
293}
294
reed@google.coma641f3f2012-12-13 22:16:30 +0000295bool SkTwoPointRadialGradient::setContext( const SkBitmap& device,
296 const SkPaint& paint,
297 const SkMatrix& matrix){
298 // For now, we might have divided by zero, so detect that
299 if (0 == fDiffRadius) {
rileya@google.com589708b2012-07-26 20:04:23 +0000300 return false;
301 }
302
reed@google.coma641f3f2012-12-13 22:16:30 +0000303 if (!this->INHERITED::setContext(device, paint, matrix)) {
rileya@google.com589708b2012-07-26 20:04:23 +0000304 return false;
305 }
306
307 // we don't have a span16 proc
308 fFlags &= ~kHasSpan16_Flag;
309 return true;
310}
311
robertphillips@google.com76f9e932013-01-15 20:17:47 +0000312#ifdef SK_DEVELOPER
313void SkTwoPointRadialGradient::toString(SkString* str) const {
314 str->append("SkTwoPointRadialGradient: (");
315
316 str->append("center1: (");
317 str->appendScalar(fCenter1.fX);
318 str->append(", ");
319 str->appendScalar(fCenter1.fY);
320 str->append(") radius1: ");
321 str->appendScalar(fRadius1);
322 str->append(" ");
323
324 str->append("center2: (");
325 str->appendScalar(fCenter2.fX);
326 str->append(", ");
327 str->appendScalar(fCenter2.fY);
328 str->append(") radius2: ");
329 str->appendScalar(fRadius2);
330 str->append(" ");
331
332 this->INHERITED::toString(str);
333
334 str->append(")");
335}
336#endif
337
rileya@google.com589708b2012-07-26 20:04:23 +0000338SkTwoPointRadialGradient::SkTwoPointRadialGradient(
339 SkFlattenableReadBuffer& buffer)
340 : INHERITED(buffer),
341 fCenter1(buffer.readPoint()),
342 fCenter2(buffer.readPoint()),
343 fRadius1(buffer.readScalar()),
344 fRadius2(buffer.readScalar()) {
345 init();
346};
347
348void SkTwoPointRadialGradient::flatten(
349 SkFlattenableWriteBuffer& buffer) const {
350 this->INHERITED::flatten(buffer);
351 buffer.writePoint(fCenter1);
352 buffer.writePoint(fCenter2);
353 buffer.writeScalar(fRadius1);
354 buffer.writeScalar(fRadius2);
355}
356
357void SkTwoPointRadialGradient::init() {
358 fDiff = fCenter1 - fCenter2;
359 fDiffRadius = fRadius2 - fRadius1;
360 // hack to avoid zero-divide for now
361 SkScalar inv = fDiffRadius ? SkScalarInvert(fDiffRadius) : 0;
362 fDiff.fX = SkScalarMul(fDiff.fX, inv);
363 fDiff.fY = SkScalarMul(fDiff.fY, inv);
364 fStartRadius = SkScalarMul(fRadius1, inv);
365 fSr2D2 = SkScalarSquare(fStartRadius);
366 fA = SkScalarSquare(fDiff.fX) + SkScalarSquare(fDiff.fY) - SK_Scalar1;
367 fOneOverTwoA = fA ? SkScalarInvert(fA * 2) : 0;
368
369 fPtsToUnit.setTranslate(-fCenter1.fX, -fCenter1.fY);
370 fPtsToUnit.postScale(inv, inv);
371}
372
rileya@google.comd7cc6512012-07-27 14:00:39 +0000373/////////////////////////////////////////////////////////////////////
374
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000375#if SK_SUPPORT_GPU
376
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000377#include "GrTBackendEffectFactory.h"
378
rileya@google.comd7cc6512012-07-27 14:00:39 +0000379// For brevity
380typedef GrGLUniformManager::UniformHandle UniformHandle;
381static const UniformHandle kInvalidUniformHandle = GrGLUniformManager::kInvalidUniformHandle;
382
bsalomon@google.com0707c292012-10-25 21:45:42 +0000383class GrGLRadial2Gradient : public GrGLGradientEffect {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000384
385public:
386
bsalomon@google.com396e61f2012-10-25 19:00:29 +0000387 GrGLRadial2Gradient(const GrBackendEffectFactory& factory,
bsalomon@google.coma469c282012-10-24 18:28:34 +0000388 const GrEffect&);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000389 virtual ~GrGLRadial2Gradient() { }
390
bsalomon@google.comf78df332012-10-29 12:43:38 +0000391 virtual void emitCode(GrGLShaderBuilder*,
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000392 const GrEffectStage&,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000393 EffectKey,
394 const char* vertexCoords,
395 const char* outputColor,
396 const char* inputColor,
397 const TextureSamplerArray&) SK_OVERRIDE;
bsalomon@google.com28a15fb2012-10-26 17:53:18 +0000398 virtual void setData(const GrGLUniformManager&, const GrEffectStage&) SK_OVERRIDE;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000399
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000400 static EffectKey GenKey(const GrEffectStage&, const GrGLCaps& caps);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000401
402protected:
403
404 UniformHandle fVSParamUni;
405 UniformHandle fFSParamUni;
406
407 const char* fVSVaryingName;
408 const char* fFSVaryingName;
409
410 bool fIsDegenerate;
411
412 // @{
413 /// Values last uploaded as uniforms
414
bsalomon@google.com81712882012-11-01 17:12:34 +0000415 SkScalar fCachedCenter;
416 SkScalar fCachedRadius;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000417 bool fCachedPosRoot;
418
419 // @}
420
421private:
422
bsalomon@google.com0707c292012-10-25 21:45:42 +0000423 typedef GrGLGradientEffect INHERITED;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000424
425};
426
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000427/////////////////////////////////////////////////////////////////////
428
429class GrRadial2Gradient : public GrGradientEffect {
430public:
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000431 static GrEffectRef* Create(GrContext* ctx,
432 const SkTwoPointRadialGradient& shader,
433 const SkMatrix& matrix,
434 SkShader::TileMode tm) {
435 SkAutoTUnref<GrEffect> effect(SkNEW_ARGS(GrRadial2Gradient, (ctx, shader, matrix, tm)));
bsalomon@google.coma1ebbe42013-01-16 15:51:47 +0000436 return CreateEffectRef(effect);
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000437 }
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000438
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000439 virtual ~GrRadial2Gradient() { }
440
441 static const char* Name() { return "Two-Point Radial Gradient"; }
bsalomon@google.com396e61f2012-10-25 19:00:29 +0000442 virtual const GrBackendEffectFactory& getFactory() const SK_OVERRIDE {
443 return GrTBackendEffectFactory<GrRadial2Gradient>::getInstance();
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000444 }
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000445
446 // The radial gradient parameters can collapse to a linear (instead of quadratic) equation.
bsalomon@google.com81712882012-11-01 17:12:34 +0000447 bool isDegenerate() const { return SK_Scalar1 == fCenterX1; }
448 SkScalar center() const { return fCenterX1; }
449 SkScalar radius() const { return fRadius0; }
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000450 bool isPosRoot() const { return SkToBool(fPosRoot); }
451
bsalomon@google.com422e81a2012-10-25 14:11:03 +0000452 typedef GrGLRadial2Gradient GLEffect;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000453
454private:
bsalomon@google.com68b58c92013-01-17 16:50:08 +0000455 virtual bool onIsEqual(const GrEffect& sBase) const SK_OVERRIDE {
456 const GrRadial2Gradient& s = static_cast<const GrRadial2Gradient&>(sBase);
457 return (INHERITED::isEqual(sBase) &&
458 this->fCenterX1 == s.fCenterX1 &&
459 this->fRadius0 == s.fRadius0 &&
460 this->fPosRoot == s.fPosRoot);
461 }
462
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000463 GrRadial2Gradient(GrContext* ctx,
464 const SkTwoPointRadialGradient& shader,
465 const SkMatrix& matrix,
466 SkShader::TileMode tm)
467 : INHERITED(ctx, shader, matrix, tm)
468 , fCenterX1(shader.getCenterX1())
469 , fRadius0(shader.getStartRadius())
470 , fPosRoot(shader.getDiffRadius() < 0) { }
471
bsalomon@google.comf271cc72012-10-24 19:35:13 +0000472 GR_DECLARE_EFFECT_TEST;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000473
474 // @{
475 // Cache of values - these can change arbitrarily, EXCEPT
476 // we shouldn't change between degenerate and non-degenerate?!
477
bsalomon@google.com81712882012-11-01 17:12:34 +0000478 SkScalar fCenterX1;
479 SkScalar fRadius0;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000480 SkBool8 fPosRoot;
481
482 // @}
483
484 typedef GrGradientEffect INHERITED;
485};
486
487/////////////////////////////////////////////////////////////////////
488
bsalomon@google.comf271cc72012-10-24 19:35:13 +0000489GR_DEFINE_EFFECT_TEST(GrRadial2Gradient);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000490
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000491GrEffectRef* GrRadial2Gradient::TestCreate(SkRandom* random,
492 GrContext* context,
493 GrTexture**) {
bsalomon@google.comd4726202012-08-03 14:34:46 +0000494 SkPoint center1 = {random->nextUScalar1(), random->nextUScalar1()};
495 SkScalar radius1 = random->nextUScalar1();
496 SkPoint center2;
497 SkScalar radius2;
498 do {
bsalomon@google.comfb883bf2012-12-11 15:32:04 +0000499 center2.set(random->nextUScalar1(), random->nextUScalar1());
bsalomon@google.comd4726202012-08-03 14:34:46 +0000500 radius2 = random->nextUScalar1 ();
bsalomon@google.come197cbf2013-01-14 16:46:26 +0000501 // There is a bug in two point radial gradients with identical radii
bsalomon@google.comd4726202012-08-03 14:34:46 +0000502 } while (radius1 == radius2);
503
504 SkColor colors[kMaxRandomGradientColors];
505 SkScalar stopsArray[kMaxRandomGradientColors];
506 SkScalar* stops = stopsArray;
507 SkShader::TileMode tm;
508 int colorCount = RandomGradientParams(random, colors, &stops, &tm);
509 SkAutoTUnref<SkShader> shader(SkGradientShader::CreateTwoPointRadial(center1, radius1,
510 center2, radius2,
511 colors, stops, colorCount,
512 tm));
bsalomon@google.come197cbf2013-01-14 16:46:26 +0000513 SkPaint paint;
514 return shader->asNewEffect(context, paint);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000515}
516
517/////////////////////////////////////////////////////////////////////
518
rileya@google.comd7cc6512012-07-27 14:00:39 +0000519GrGLRadial2Gradient::GrGLRadial2Gradient(
bsalomon@google.com396e61f2012-10-25 19:00:29 +0000520 const GrBackendEffectFactory& factory,
bsalomon@google.coma469c282012-10-24 18:28:34 +0000521 const GrEffect& baseData)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000522 : INHERITED(factory)
523 , fVSParamUni(kInvalidUniformHandle)
524 , fFSParamUni(kInvalidUniformHandle)
525 , fVSVaryingName(NULL)
526 , fFSVaryingName(NULL)
bsalomon@google.com81712882012-11-01 17:12:34 +0000527 , fCachedCenter(SK_ScalarMax)
528 , fCachedRadius(-SK_ScalarMax)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000529 , fCachedPosRoot(0) {
530
531 const GrRadial2Gradient& data =
532 static_cast<const GrRadial2Gradient&>(baseData);
533 fIsDegenerate = data.isDegenerate();
534}
535
bsalomon@google.comf78df332012-10-29 12:43:38 +0000536void GrGLRadial2Gradient::emitCode(GrGLShaderBuilder* builder,
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000537 const GrEffectStage& stage,
538 EffectKey key,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000539 const char* vertexCoords,
540 const char* outputColor,
541 const char* inputColor,
542 const TextureSamplerArray& samplers) {
543
544 this->emitYCoordUniform(builder);
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000545 const char* fsCoords;
546 const char* vsCoordsVarying;
547 GrSLType coordsVaryingType;
548 this->setupMatrix(builder, key, vertexCoords, &fsCoords, &vsCoordsVarying, &coordsVaryingType);
549
rileya@google.comd7cc6512012-07-27 14:00:39 +0000550 // 2 copies of uniform array, 1 for each of vertex & fragment shader,
551 // to work around Xoom bug. Doesn't seem to cause performance decrease
552 // in test apps, but need to keep an eye on it.
553 fVSParamUni = builder->addUniformArray(GrGLShaderBuilder::kVertex_ShaderType,
554 kFloat_GrSLType, "Radial2VSParams", 6);
555 fFSParamUni = builder->addUniformArray(GrGLShaderBuilder::kFragment_ShaderType,
556 kFloat_GrSLType, "Radial2FSParams", 6);
557
558 // For radial gradients without perspective we can pass the linear
559 // part of the quadratic as a varying.
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000560 if (kVec2f_GrSLType == coordsVaryingType) {
561 builder->addVarying(kFloat_GrSLType, "Radial2BCoeff", &fVSVaryingName, &fFSVaryingName);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000562 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000563
bsalomon@google.comf78df332012-10-29 12:43:38 +0000564 // VS
565 {
566 SkString* code = &builder->fVSCode;
567 SkString p2;
568 SkString p3;
569 builder->getUniformVariable(fVSParamUni).appendArrayAccess(2, &p2);
570 builder->getUniformVariable(fVSParamUni).appendArrayAccess(3, &p3);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000571
bsalomon@google.comf78df332012-10-29 12:43:38 +0000572 // For radial gradients without perspective we can pass the linear
573 // part of the quadratic as a varying.
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000574 if (kVec2f_GrSLType == coordsVaryingType) {
bsalomon@google.comf78df332012-10-29 12:43:38 +0000575 // r2Var = 2 * (r2Parm[2] * varCoord.x - r2Param[3])
576 code->appendf("\t%s = 2.0 *(%s * %s.x - %s);\n",
577 fVSVaryingName, p2.c_str(),
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000578 vsCoordsVarying, p3.c_str());
bsalomon@google.comf78df332012-10-29 12:43:38 +0000579 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000580 }
581
bsalomon@google.comf78df332012-10-29 12:43:38 +0000582 // FS
583 {
584 SkString* code = &builder->fFSCode;
585 SkString cName("c");
586 SkString ac4Name("ac4");
587 SkString rootName("root");
588 SkString t;
589 SkString p0;
590 SkString p1;
591 SkString p2;
592 SkString p3;
593 SkString p4;
594 SkString p5;
595 builder->getUniformVariable(fFSParamUni).appendArrayAccess(0, &p0);
596 builder->getUniformVariable(fFSParamUni).appendArrayAccess(1, &p1);
597 builder->getUniformVariable(fFSParamUni).appendArrayAccess(2, &p2);
598 builder->getUniformVariable(fFSParamUni).appendArrayAccess(3, &p3);
599 builder->getUniformVariable(fFSParamUni).appendArrayAccess(4, &p4);
600 builder->getUniformVariable(fFSParamUni).appendArrayAccess(5, &p5);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000601
bsalomon@google.comf78df332012-10-29 12:43:38 +0000602 // If we we're able to interpolate the linear component,
603 // bVar is the varying; otherwise compute it
604 SkString bVar;
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000605 if (kVec2f_GrSLType == coordsVaryingType) {
bsalomon@google.comf78df332012-10-29 12:43:38 +0000606 bVar = fFSVaryingName;
607 } else {
608 bVar = "b";
609 code->appendf("\tfloat %s = 2.0 * (%s * %s.x - %s);\n",
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000610 bVar.c_str(), p2.c_str(), fsCoords, p3.c_str());
bsalomon@google.comf78df332012-10-29 12:43:38 +0000611 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000612
bsalomon@google.comf78df332012-10-29 12:43:38 +0000613 // c = (x^2)+(y^2) - params[4]
614 code->appendf("\tfloat %s = dot(%s, %s) - %s;\n",
615 cName.c_str(),
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000616 fsCoords,
617 fsCoords,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000618 p4.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000619
bsalomon@google.comf78df332012-10-29 12:43:38 +0000620 // If we aren't degenerate, emit some extra code, and accept a slightly
621 // more complex coord.
622 if (!fIsDegenerate) {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000623
bsalomon@google.comf78df332012-10-29 12:43:38 +0000624 // ac4 = 4.0 * params[0] * c
625 code->appendf("\tfloat %s = %s * 4.0 * %s;\n",
626 ac4Name.c_str(), p0.c_str(),
627 cName.c_str());
628
629 // root = sqrt(b^2-4ac)
630 // (abs to avoid exception due to fp precision)
631 code->appendf("\tfloat %s = sqrt(abs(%s*%s - %s));\n",
632 rootName.c_str(), bVar.c_str(), bVar.c_str(),
633 ac4Name.c_str());
634
635 // t is: (-b + params[5] * sqrt(b^2-4ac)) * params[1]
636 t.printf("(-%s + %s * %s) * %s", bVar.c_str(), p5.c_str(),
637 rootName.c_str(), p1.c_str());
638 } else {
639 // t is: -c/b
640 t.printf("-%s / %s", cName.c_str(), bVar.c_str());
641 }
642
643 this->emitColorLookup(builder, t.c_str(), outputColor, inputColor, samplers[0]);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000644 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000645}
646
bsalomon@google.com28a15fb2012-10-26 17:53:18 +0000647void GrGLRadial2Gradient::setData(const GrGLUniformManager& uman, const GrEffectStage& stage) {
648 INHERITED::setData(uman, stage);
649 const GrRadial2Gradient& data = static_cast<const GrRadial2Gradient&>(*stage.getEffect());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000650 GrAssert(data.isDegenerate() == fIsDegenerate);
bsalomon@google.com81712882012-11-01 17:12:34 +0000651 SkScalar centerX1 = data.center();
652 SkScalar radius0 = data.radius();
rileya@google.comd7cc6512012-07-27 14:00:39 +0000653 if (fCachedCenter != centerX1 ||
654 fCachedRadius != radius0 ||
655 fCachedPosRoot != data.isPosRoot()) {
656
bsalomon@google.com81712882012-11-01 17:12:34 +0000657 SkScalar a = SkScalarMul(centerX1, centerX1) - SK_Scalar1;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000658
659 // When we're in the degenerate (linear) case, the second
660 // value will be INF but the program doesn't read it. (We
661 // use the same 6 uniforms even though we don't need them
662 // all in the linear case just to keep the code complexity
663 // down).
664 float values[6] = {
bsalomon@google.com81712882012-11-01 17:12:34 +0000665 SkScalarToFloat(a),
666 1 / (2.f * SkScalarToFloat(a)),
667 SkScalarToFloat(centerX1),
668 SkScalarToFloat(radius0),
669 SkScalarToFloat(SkScalarMul(radius0, radius0)),
rileya@google.comd7cc6512012-07-27 14:00:39 +0000670 data.isPosRoot() ? 1.f : -1.f
671 };
672
673 uman.set1fv(fVSParamUni, 0, 6, values);
674 uman.set1fv(fFSParamUni, 0, 6, values);
675 fCachedCenter = centerX1;
676 fCachedRadius = radius0;
677 fCachedPosRoot = data.isPosRoot();
678 }
679}
680
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000681GrGLEffect::EffectKey GrGLRadial2Gradient::GenKey(const GrEffectStage& s, const GrGLCaps&) {
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000682 enum {
683 kIsDegenerate = 1 << kMatrixKeyBitCnt,
684 };
685
686 EffectKey key = GenMatrixKey(s);
687 if (static_cast<const GrRadial2Gradient&>(*s.getEffect()).isDegenerate()) {
688 key |= kIsDegenerate;
689 }
690 return key;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000691}
rileya@google.comd7cc6512012-07-27 14:00:39 +0000692
693/////////////////////////////////////////////////////////////////////
694
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000695GrEffectRef* SkTwoPointRadialGradient::asNewEffect(GrContext* context, const SkPaint&) const {
bsalomon@google.com00835cc2013-01-14 17:07:22 +0000696 SkASSERT(NULL != context);
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000697 // invert the localM, translate to center1 (fPtsToUni), rotate so center2 is on x axis.
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000698 SkMatrix matrix;
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000699 if (!this->getLocalMatrix().invert(&matrix)) {
humper@google.com84831ac2013-01-14 22:09:54 +0000700 return NULL;
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000701 }
702 matrix.postConcat(fPtsToUnit);
703
704 SkScalar diffLen = fDiff.length();
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000705 if (0 != diffLen) {
706 SkScalar invDiffLen = SkScalarInvert(diffLen);
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000707 SkMatrix rot;
708 rot.setSinCos(-SkScalarMul(invDiffLen, fDiff.fY),
709 SkScalarMul(invDiffLen, fDiff.fX));
710 matrix.postConcat(rot);
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000711 }
712
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000713 return GrRadial2Gradient::Create(context, *this, matrix, fTileMode);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000714}
715
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000716#else
717
bsalomon@google.com5d2cd202013-01-16 15:31:06 +0000718GrEffectRef* SkTwoPointRadialGradient::asNewEffect(GrContext*, const SkPaint&) const {
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000719 SkDEBUGFAIL("Should not call in GPU-less build");
bsalomon@google.come197cbf2013-01-14 16:46:26 +0000720 return NULL;
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000721}
722
723#endif