blob: 8380a3f5b94d5dccb71865777eebf28961c48555 [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)));
436 return CreateEffectPtr(effect);
437 }
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 }
bsalomon@google.coma469c282012-10-24 18:28:34 +0000445 virtual bool isEqual(const GrEffect& sBase) const SK_OVERRIDE {
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000446 const GrRadial2Gradient& s = static_cast<const GrRadial2Gradient&>(sBase);
447 return (INHERITED::isEqual(sBase) &&
448 this->fCenterX1 == s.fCenterX1 &&
449 this->fRadius0 == s.fRadius0 &&
450 this->fPosRoot == s.fPosRoot);
451 }
452
453 // The radial gradient parameters can collapse to a linear (instead of quadratic) equation.
bsalomon@google.com81712882012-11-01 17:12:34 +0000454 bool isDegenerate() const { return SK_Scalar1 == fCenterX1; }
455 SkScalar center() const { return fCenterX1; }
456 SkScalar radius() const { return fRadius0; }
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000457 bool isPosRoot() const { return SkToBool(fPosRoot); }
458
bsalomon@google.com422e81a2012-10-25 14:11:03 +0000459 typedef GrGLRadial2Gradient GLEffect;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000460
461private:
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000462 GrRadial2Gradient(GrContext* ctx,
463 const SkTwoPointRadialGradient& shader,
464 const SkMatrix& matrix,
465 SkShader::TileMode tm)
466 : INHERITED(ctx, shader, matrix, tm)
467 , fCenterX1(shader.getCenterX1())
468 , fRadius0(shader.getStartRadius())
469 , fPosRoot(shader.getDiffRadius() < 0) { }
470
bsalomon@google.comf271cc72012-10-24 19:35:13 +0000471 GR_DECLARE_EFFECT_TEST;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000472
473 // @{
474 // Cache of values - these can change arbitrarily, EXCEPT
475 // we shouldn't change between degenerate and non-degenerate?!
476
bsalomon@google.com81712882012-11-01 17:12:34 +0000477 SkScalar fCenterX1;
478 SkScalar fRadius0;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000479 SkBool8 fPosRoot;
480
481 // @}
482
483 typedef GrGradientEffect INHERITED;
484};
485
486/////////////////////////////////////////////////////////////////////
487
bsalomon@google.comf271cc72012-10-24 19:35:13 +0000488GR_DEFINE_EFFECT_TEST(GrRadial2Gradient);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000489
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000490GrEffectRef* GrRadial2Gradient::TestCreate(SkRandom* random,
491 GrContext* context,
492 GrTexture**) {
bsalomon@google.comd4726202012-08-03 14:34:46 +0000493 SkPoint center1 = {random->nextUScalar1(), random->nextUScalar1()};
494 SkScalar radius1 = random->nextUScalar1();
495 SkPoint center2;
496 SkScalar radius2;
497 do {
bsalomon@google.comfb883bf2012-12-11 15:32:04 +0000498 center2.set(random->nextUScalar1(), random->nextUScalar1());
bsalomon@google.comd4726202012-08-03 14:34:46 +0000499 radius2 = random->nextUScalar1 ();
bsalomon@google.come197cbf2013-01-14 16:46:26 +0000500 // There is a bug in two point radial gradients with identical radii
bsalomon@google.comd4726202012-08-03 14:34:46 +0000501 } while (radius1 == radius2);
502
503 SkColor colors[kMaxRandomGradientColors];
504 SkScalar stopsArray[kMaxRandomGradientColors];
505 SkScalar* stops = stopsArray;
506 SkShader::TileMode tm;
507 int colorCount = RandomGradientParams(random, colors, &stops, &tm);
508 SkAutoTUnref<SkShader> shader(SkGradientShader::CreateTwoPointRadial(center1, radius1,
509 center2, radius2,
510 colors, stops, colorCount,
511 tm));
bsalomon@google.come197cbf2013-01-14 16:46:26 +0000512 SkPaint paint;
513 return shader->asNewEffect(context, paint);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000514}
515
516/////////////////////////////////////////////////////////////////////
517
rileya@google.comd7cc6512012-07-27 14:00:39 +0000518GrGLRadial2Gradient::GrGLRadial2Gradient(
bsalomon@google.com396e61f2012-10-25 19:00:29 +0000519 const GrBackendEffectFactory& factory,
bsalomon@google.coma469c282012-10-24 18:28:34 +0000520 const GrEffect& baseData)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000521 : INHERITED(factory)
522 , fVSParamUni(kInvalidUniformHandle)
523 , fFSParamUni(kInvalidUniformHandle)
524 , fVSVaryingName(NULL)
525 , fFSVaryingName(NULL)
bsalomon@google.com81712882012-11-01 17:12:34 +0000526 , fCachedCenter(SK_ScalarMax)
527 , fCachedRadius(-SK_ScalarMax)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000528 , fCachedPosRoot(0) {
529
530 const GrRadial2Gradient& data =
531 static_cast<const GrRadial2Gradient&>(baseData);
532 fIsDegenerate = data.isDegenerate();
533}
534
bsalomon@google.comf78df332012-10-29 12:43:38 +0000535void GrGLRadial2Gradient::emitCode(GrGLShaderBuilder* builder,
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000536 const GrEffectStage& stage,
537 EffectKey key,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000538 const char* vertexCoords,
539 const char* outputColor,
540 const char* inputColor,
541 const TextureSamplerArray& samplers) {
542
543 this->emitYCoordUniform(builder);
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000544 const char* fsCoords;
545 const char* vsCoordsVarying;
546 GrSLType coordsVaryingType;
547 this->setupMatrix(builder, key, vertexCoords, &fsCoords, &vsCoordsVarying, &coordsVaryingType);
548
rileya@google.comd7cc6512012-07-27 14:00:39 +0000549 // 2 copies of uniform array, 1 for each of vertex & fragment shader,
550 // to work around Xoom bug. Doesn't seem to cause performance decrease
551 // in test apps, but need to keep an eye on it.
552 fVSParamUni = builder->addUniformArray(GrGLShaderBuilder::kVertex_ShaderType,
553 kFloat_GrSLType, "Radial2VSParams", 6);
554 fFSParamUni = builder->addUniformArray(GrGLShaderBuilder::kFragment_ShaderType,
555 kFloat_GrSLType, "Radial2FSParams", 6);
556
557 // For radial gradients without perspective we can pass the linear
558 // part of the quadratic as a varying.
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000559 if (kVec2f_GrSLType == coordsVaryingType) {
560 builder->addVarying(kFloat_GrSLType, "Radial2BCoeff", &fVSVaryingName, &fFSVaryingName);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000561 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000562
bsalomon@google.comf78df332012-10-29 12:43:38 +0000563 // VS
564 {
565 SkString* code = &builder->fVSCode;
566 SkString p2;
567 SkString p3;
568 builder->getUniformVariable(fVSParamUni).appendArrayAccess(2, &p2);
569 builder->getUniformVariable(fVSParamUni).appendArrayAccess(3, &p3);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000570
bsalomon@google.comf78df332012-10-29 12:43:38 +0000571 // For radial gradients without perspective we can pass the linear
572 // part of the quadratic as a varying.
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000573 if (kVec2f_GrSLType == coordsVaryingType) {
bsalomon@google.comf78df332012-10-29 12:43:38 +0000574 // r2Var = 2 * (r2Parm[2] * varCoord.x - r2Param[3])
575 code->appendf("\t%s = 2.0 *(%s * %s.x - %s);\n",
576 fVSVaryingName, p2.c_str(),
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000577 vsCoordsVarying, p3.c_str());
bsalomon@google.comf78df332012-10-29 12:43:38 +0000578 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000579 }
580
bsalomon@google.comf78df332012-10-29 12:43:38 +0000581 // FS
582 {
583 SkString* code = &builder->fFSCode;
584 SkString cName("c");
585 SkString ac4Name("ac4");
586 SkString rootName("root");
587 SkString t;
588 SkString p0;
589 SkString p1;
590 SkString p2;
591 SkString p3;
592 SkString p4;
593 SkString p5;
594 builder->getUniformVariable(fFSParamUni).appendArrayAccess(0, &p0);
595 builder->getUniformVariable(fFSParamUni).appendArrayAccess(1, &p1);
596 builder->getUniformVariable(fFSParamUni).appendArrayAccess(2, &p2);
597 builder->getUniformVariable(fFSParamUni).appendArrayAccess(3, &p3);
598 builder->getUniformVariable(fFSParamUni).appendArrayAccess(4, &p4);
599 builder->getUniformVariable(fFSParamUni).appendArrayAccess(5, &p5);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000600
bsalomon@google.comf78df332012-10-29 12:43:38 +0000601 // If we we're able to interpolate the linear component,
602 // bVar is the varying; otherwise compute it
603 SkString bVar;
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000604 if (kVec2f_GrSLType == coordsVaryingType) {
bsalomon@google.comf78df332012-10-29 12:43:38 +0000605 bVar = fFSVaryingName;
606 } else {
607 bVar = "b";
608 code->appendf("\tfloat %s = 2.0 * (%s * %s.x - %s);\n",
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000609 bVar.c_str(), p2.c_str(), fsCoords, p3.c_str());
bsalomon@google.comf78df332012-10-29 12:43:38 +0000610 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000611
bsalomon@google.comf78df332012-10-29 12:43:38 +0000612 // c = (x^2)+(y^2) - params[4]
613 code->appendf("\tfloat %s = dot(%s, %s) - %s;\n",
614 cName.c_str(),
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000615 fsCoords,
616 fsCoords,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000617 p4.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000618
bsalomon@google.comf78df332012-10-29 12:43:38 +0000619 // If we aren't degenerate, emit some extra code, and accept a slightly
620 // more complex coord.
621 if (!fIsDegenerate) {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000622
bsalomon@google.comf78df332012-10-29 12:43:38 +0000623 // ac4 = 4.0 * params[0] * c
624 code->appendf("\tfloat %s = %s * 4.0 * %s;\n",
625 ac4Name.c_str(), p0.c_str(),
626 cName.c_str());
627
628 // root = sqrt(b^2-4ac)
629 // (abs to avoid exception due to fp precision)
630 code->appendf("\tfloat %s = sqrt(abs(%s*%s - %s));\n",
631 rootName.c_str(), bVar.c_str(), bVar.c_str(),
632 ac4Name.c_str());
633
634 // t is: (-b + params[5] * sqrt(b^2-4ac)) * params[1]
635 t.printf("(-%s + %s * %s) * %s", bVar.c_str(), p5.c_str(),
636 rootName.c_str(), p1.c_str());
637 } else {
638 // t is: -c/b
639 t.printf("-%s / %s", cName.c_str(), bVar.c_str());
640 }
641
642 this->emitColorLookup(builder, t.c_str(), outputColor, inputColor, samplers[0]);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000643 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000644}
645
bsalomon@google.com28a15fb2012-10-26 17:53:18 +0000646void GrGLRadial2Gradient::setData(const GrGLUniformManager& uman, const GrEffectStage& stage) {
647 INHERITED::setData(uman, stage);
648 const GrRadial2Gradient& data = static_cast<const GrRadial2Gradient&>(*stage.getEffect());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000649 GrAssert(data.isDegenerate() == fIsDegenerate);
bsalomon@google.com81712882012-11-01 17:12:34 +0000650 SkScalar centerX1 = data.center();
651 SkScalar radius0 = data.radius();
rileya@google.comd7cc6512012-07-27 14:00:39 +0000652 if (fCachedCenter != centerX1 ||
653 fCachedRadius != radius0 ||
654 fCachedPosRoot != data.isPosRoot()) {
655
bsalomon@google.com81712882012-11-01 17:12:34 +0000656 SkScalar a = SkScalarMul(centerX1, centerX1) - SK_Scalar1;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000657
658 // When we're in the degenerate (linear) case, the second
659 // value will be INF but the program doesn't read it. (We
660 // use the same 6 uniforms even though we don't need them
661 // all in the linear case just to keep the code complexity
662 // down).
663 float values[6] = {
bsalomon@google.com81712882012-11-01 17:12:34 +0000664 SkScalarToFloat(a),
665 1 / (2.f * SkScalarToFloat(a)),
666 SkScalarToFloat(centerX1),
667 SkScalarToFloat(radius0),
668 SkScalarToFloat(SkScalarMul(radius0, radius0)),
rileya@google.comd7cc6512012-07-27 14:00:39 +0000669 data.isPosRoot() ? 1.f : -1.f
670 };
671
672 uman.set1fv(fVSParamUni, 0, 6, values);
673 uman.set1fv(fFSParamUni, 0, 6, values);
674 fCachedCenter = centerX1;
675 fCachedRadius = radius0;
676 fCachedPosRoot = data.isPosRoot();
677 }
678}
679
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000680GrGLEffect::EffectKey GrGLRadial2Gradient::GenKey(const GrEffectStage& s, const GrGLCaps&) {
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000681 enum {
682 kIsDegenerate = 1 << kMatrixKeyBitCnt,
683 };
684
685 EffectKey key = GenMatrixKey(s);
686 if (static_cast<const GrRadial2Gradient&>(*s.getEffect()).isDegenerate()) {
687 key |= kIsDegenerate;
688 }
689 return key;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000690}
rileya@google.comd7cc6512012-07-27 14:00:39 +0000691
692/////////////////////////////////////////////////////////////////////
693
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000694GrEffectRef* SkTwoPointRadialGradient::asNewEffect(GrContext* context, const SkPaint&) const {
bsalomon@google.com00835cc2013-01-14 17:07:22 +0000695 SkASSERT(NULL != context);
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000696 // invert the localM, translate to center1 (fPtsToUni), rotate so center2 is on x axis.
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000697 SkMatrix matrix;
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000698 if (!this->getLocalMatrix().invert(&matrix)) {
humper@google.com84831ac2013-01-14 22:09:54 +0000699 return NULL;
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000700 }
701 matrix.postConcat(fPtsToUnit);
702
703 SkScalar diffLen = fDiff.length();
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000704 if (0 != diffLen) {
705 SkScalar invDiffLen = SkScalarInvert(diffLen);
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000706 SkMatrix rot;
707 rot.setSinCos(-SkScalarMul(invDiffLen, fDiff.fY),
708 SkScalarMul(invDiffLen, fDiff.fX));
709 matrix.postConcat(rot);
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000710 }
711
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000712 return GrRadial2Gradient::Create(context, *this, matrix, fTileMode);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000713}
714
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000715#else
716
bsalomon@google.come197cbf2013-01-14 16:46:26 +0000717GrEffect* SkTwoPointRadialGradient::asNewEffect(GrContext* context, const SkPaint&) const {
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000718 SkDEBUGFAIL("Should not call in GPU-less build");
bsalomon@google.come197cbf2013-01-14 16:46:26 +0000719 return NULL;
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000720}
721
722#endif