blob: ee6e25a88eb5a0bbddfa15494651d0a154fa394b [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 "SkTwoPointConicalGradient.h"
10
11static int valid_divide(float numer, float denom, float* ratio) {
12 SkASSERT(ratio);
13 if (0 == denom) {
14 return 0;
15 }
16 *ratio = numer / denom;
17 return 1;
18}
19
20// Return the number of distinct real roots, and write them into roots[] in
21// ascending order
22static int find_quad_roots(float A, float B, float C, float roots[2]) {
23 SkASSERT(roots);
rmistry@google.comfbfcd562012-08-23 18:09:54 +000024
rileya@google.com589708b2012-07-26 20:04:23 +000025 if (A == 0) {
26 return valid_divide(-C, B, roots);
27 }
rmistry@google.comfbfcd562012-08-23 18:09:54 +000028
rileya@google.com589708b2012-07-26 20:04:23 +000029 float R = B*B - 4*A*C;
30 if (R < 0) {
31 return 0;
32 }
33 R = sk_float_sqrt(R);
34
35#if 1
36 float Q = B;
37 if (Q < 0) {
38 Q -= R;
39 } else {
40 Q += R;
41 }
42#else
43 // on 10.6 this was much slower than the above branch :(
44 float Q = B + copysignf(R, B);
45#endif
46 Q *= -0.5f;
47 if (0 == Q) {
48 roots[0] = 0;
49 return 1;
50 }
51
52 float r0 = Q / A;
53 float r1 = C / Q;
54 roots[0] = r0 < r1 ? r0 : r1;
55 roots[1] = r0 > r1 ? r0 : r1;
56 return 2;
57}
58
59static float lerp(float x, float dx, float t) {
60 return x + t * dx;
61}
62
63static float sqr(float x) { return x * x; }
64
65void TwoPtRadial::init(const SkPoint& center0, SkScalar rad0,
66 const SkPoint& center1, SkScalar rad1) {
67 fCenterX = SkScalarToFloat(center0.fX);
68 fCenterY = SkScalarToFloat(center0.fY);
69 fDCenterX = SkScalarToFloat(center1.fX) - fCenterX;
70 fDCenterY = SkScalarToFloat(center1.fY) - fCenterY;
71 fRadius = SkScalarToFloat(rad0);
72 fDRadius = SkScalarToFloat(rad1) - fRadius;
73
74 fA = sqr(fDCenterX) + sqr(fDCenterY) - sqr(fDRadius);
75 fRadius2 = sqr(fRadius);
76 fRDR = fRadius * fDRadius;
77}
78
79void TwoPtRadial::setup(SkScalar fx, SkScalar fy, SkScalar dfx, SkScalar dfy) {
80 fRelX = SkScalarToFloat(fx) - fCenterX;
81 fRelY = SkScalarToFloat(fy) - fCenterY;
82 fIncX = SkScalarToFloat(dfx);
83 fIncY = SkScalarToFloat(dfy);
84 fB = -2 * (fDCenterX * fRelX + fDCenterY * fRelY + fRDR);
85 fDB = -2 * (fDCenterX * fIncX + fDCenterY * fIncY);
86}
87
88SkFixed TwoPtRadial::nextT() {
89 float roots[2];
rmistry@google.comfbfcd562012-08-23 18:09:54 +000090
rileya@google.com589708b2012-07-26 20:04:23 +000091 float C = sqr(fRelX) + sqr(fRelY) - fRadius2;
92 int countRoots = find_quad_roots(fA, fB, C, roots);
93
94 fRelX += fIncX;
95 fRelY += fIncY;
96 fB += fDB;
97
98 if (0 == countRoots) {
99 return kDontDrawT;
100 }
101
102 // Prefer the bigger t value if both give a radius(t) > 0
103 // find_quad_roots returns the values sorted, so we start with the last
104 float t = roots[countRoots - 1];
105 float r = lerp(fRadius, fDRadius, t);
106 if (r <= 0) {
107 t = roots[0]; // might be the same as roots[countRoots-1]
108 r = lerp(fRadius, fDRadius, t);
109 if (r <= 0) {
110 return kDontDrawT;
111 }
112 }
113 return SkFloatToFixed(t);
114}
115
reed@google.com60040292013-02-04 18:21:23 +0000116typedef void (*TwoPointConicalProc)(TwoPtRadial* rec, SkPMColor* dstC,
117 const SkPMColor* cache, int toggle, int count);
rileya@google.com589708b2012-07-26 20:04:23 +0000118
119static void twopoint_clamp(TwoPtRadial* rec, SkPMColor* SK_RESTRICT dstC,
reed@google.com60040292013-02-04 18:21:23 +0000120 const SkPMColor* SK_RESTRICT cache, int toggle,
121 int count) {
rileya@google.com589708b2012-07-26 20:04:23 +0000122 for (; count > 0; --count) {
123 SkFixed t = rec->nextT();
124 if (TwoPtRadial::DontDrawT(t)) {
125 *dstC++ = 0;
126 } else {
127 SkFixed index = SkClampMax(t, 0xFFFF);
128 SkASSERT(index <= 0xFFFF);
reed@google.com60040292013-02-04 18:21:23 +0000129 *dstC++ = cache[toggle +
130 (index >> SkGradientShaderBase::kCache32Shift)];
rileya@google.com589708b2012-07-26 20:04:23 +0000131 }
reed@google.com60040292013-02-04 18:21:23 +0000132 toggle = next_dither_toggle(toggle);
rileya@google.com589708b2012-07-26 20:04:23 +0000133 }
134}
135
136static void twopoint_repeat(TwoPtRadial* rec, SkPMColor* SK_RESTRICT dstC,
reed@google.com60040292013-02-04 18:21:23 +0000137 const SkPMColor* SK_RESTRICT cache, int toggle,
138 int count) {
rileya@google.com589708b2012-07-26 20:04:23 +0000139 for (; count > 0; --count) {
140 SkFixed t = rec->nextT();
141 if (TwoPtRadial::DontDrawT(t)) {
142 *dstC++ = 0;
143 } else {
144 SkFixed index = repeat_tileproc(t);
145 SkASSERT(index <= 0xFFFF);
reed@google.com60040292013-02-04 18:21:23 +0000146 *dstC++ = cache[toggle +
147 (index >> SkGradientShaderBase::kCache32Shift)];
rileya@google.com589708b2012-07-26 20:04:23 +0000148 }
reed@google.com60040292013-02-04 18:21:23 +0000149 toggle = next_dither_toggle(toggle);
rileya@google.com589708b2012-07-26 20:04:23 +0000150 }
151}
152
153static void twopoint_mirror(TwoPtRadial* rec, SkPMColor* SK_RESTRICT dstC,
reed@google.com60040292013-02-04 18:21:23 +0000154 const SkPMColor* SK_RESTRICT cache, int toggle,
155 int count) {
rileya@google.com589708b2012-07-26 20:04:23 +0000156 for (; count > 0; --count) {
157 SkFixed t = rec->nextT();
158 if (TwoPtRadial::DontDrawT(t)) {
159 *dstC++ = 0;
160 } else {
161 SkFixed index = mirror_tileproc(t);
162 SkASSERT(index <= 0xFFFF);
reed@google.com60040292013-02-04 18:21:23 +0000163 *dstC++ = cache[toggle +
164 (index >> SkGradientShaderBase::kCache32Shift)];
rileya@google.com589708b2012-07-26 20:04:23 +0000165 }
reed@google.com60040292013-02-04 18:21:23 +0000166 toggle = next_dither_toggle(toggle);
rileya@google.com589708b2012-07-26 20:04:23 +0000167 }
168}
169
170void SkTwoPointConicalGradient::init() {
171 fRec.init(fCenter1, fRadius1, fCenter2, fRadius2);
172 fPtsToUnit.reset();
173}
174
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000175/////////////////////////////////////////////////////////////////////
176
rileya@google.com589708b2012-07-26 20:04:23 +0000177SkTwoPointConicalGradient::SkTwoPointConicalGradient(
reed@google.com3d3a8602013-05-24 14:58:44 +0000178 const SkPoint& start, SkScalar startRadius,
179 const SkPoint& end, SkScalar endRadius,
180 const Descriptor& desc)
reed@google.com437d6eb2013-05-23 19:03:05 +0000181 : SkGradientShaderBase(desc),
rileya@google.com589708b2012-07-26 20:04:23 +0000182 fCenter1(start),
183 fCenter2(end),
184 fRadius1(startRadius),
185 fRadius2(endRadius) {
186 // this is degenerate, and should be caught by our caller
187 SkASSERT(fCenter1 != fCenter2 || fRadius1 != fRadius2);
188 this->init();
189}
190
commit-bot@chromium.org3fbab822013-03-20 00:49:57 +0000191bool SkTwoPointConicalGradient::isOpaque() const {
robertphillips@google.comcb6d97c2013-07-09 13:50:09 +0000192 // Because areas outside the cone are left untouched, we cannot treat the
193 // shader as opaque even if the gradient itself is opaque.
194 // TODO(junov): Compute whether the cone fills the plane crbug.com/222380
195 return false;
commit-bot@chromium.org3fbab822013-03-20 00:49:57 +0000196}
197
rileya@google.com589708b2012-07-26 20:04:23 +0000198void SkTwoPointConicalGradient::shadeSpan(int x, int y, SkPMColor* dstCParam,
199 int count) {
reed@google.com60040292013-02-04 18:21:23 +0000200 int toggle = init_dither_toggle(x, y);
reed@google.com60040292013-02-04 18:21:23 +0000201
rileya@google.com589708b2012-07-26 20:04:23 +0000202 SkASSERT(count > 0);
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000203
rileya@google.com589708b2012-07-26 20:04:23 +0000204 SkPMColor* SK_RESTRICT dstC = dstCParam;
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000205
rileya@google.com589708b2012-07-26 20:04:23 +0000206 SkMatrix::MapXYProc dstProc = fDstToIndexProc;
bsalomon@google.com100abf42012-09-05 17:40:04 +0000207
rileya@google.com589708b2012-07-26 20:04:23 +0000208 const SkPMColor* SK_RESTRICT cache = this->getCache32();
209
reed@google.com60040292013-02-04 18:21:23 +0000210 TwoPointConicalProc shadeProc = twopoint_repeat;
rileya@google.com589708b2012-07-26 20:04:23 +0000211 if (SkShader::kClamp_TileMode == fTileMode) {
212 shadeProc = twopoint_clamp;
213 } else if (SkShader::kMirror_TileMode == fTileMode) {
214 shadeProc = twopoint_mirror;
215 } else {
216 SkASSERT(SkShader::kRepeat_TileMode == fTileMode);
217 }
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000218
rileya@google.com589708b2012-07-26 20:04:23 +0000219 if (fDstToIndexClass != kPerspective_MatrixClass) {
220 SkPoint srcPt;
221 dstProc(fDstToIndex, SkIntToScalar(x) + SK_ScalarHalf,
222 SkIntToScalar(y) + SK_ScalarHalf, &srcPt);
223 SkScalar dx, fx = srcPt.fX;
224 SkScalar dy, fy = srcPt.fY;
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000225
rileya@google.com589708b2012-07-26 20:04:23 +0000226 if (fDstToIndexClass == kFixedStepInX_MatrixClass) {
227 SkFixed fixedX, fixedY;
228 (void)fDstToIndex.fixedStepInX(SkIntToScalar(y), &fixedX, &fixedY);
229 dx = SkFixedToScalar(fixedX);
230 dy = SkFixedToScalar(fixedY);
231 } else {
232 SkASSERT(fDstToIndexClass == kLinear_MatrixClass);
233 dx = fDstToIndex.getScaleX();
234 dy = fDstToIndex.getSkewY();
235 }
236
237 fRec.setup(fx, fy, dx, dy);
reed@google.com60040292013-02-04 18:21:23 +0000238 (*shadeProc)(&fRec, dstC, cache, toggle, count);
rileya@google.com589708b2012-07-26 20:04:23 +0000239 } else { // perspective case
240 SkScalar dstX = SkIntToScalar(x);
241 SkScalar dstY = SkIntToScalar(y);
242 for (; count > 0; --count) {
243 SkPoint srcPt;
244 dstProc(fDstToIndex, dstX, dstY, &srcPt);
245 dstX += SK_Scalar1;
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000246
rileya@google.com589708b2012-07-26 20:04:23 +0000247 fRec.setup(srcPt.fX, srcPt.fY, 0, 0);
reed@google.com60040292013-02-04 18:21:23 +0000248 (*shadeProc)(&fRec, dstC, cache, toggle, 1);
reed@google.com60040292013-02-04 18:21:23 +0000249 toggle = next_dither_toggle(toggle);
rileya@google.com589708b2012-07-26 20:04:23 +0000250 }
251 }
252}
253
254bool SkTwoPointConicalGradient::setContext(const SkBitmap& device,
255 const SkPaint& paint,
256 const SkMatrix& matrix) {
257 if (!this->INHERITED::setContext(device, paint, matrix)) {
258 return false;
259 }
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000260
rileya@google.com589708b2012-07-26 20:04:23 +0000261 // we don't have a span16 proc
262 fFlags &= ~kHasSpan16_Flag;
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000263
rileya@google.com589708b2012-07-26 20:04:23 +0000264 // in general, we might discard based on computed-radius, so clear
265 // this flag (todo: sometimes we can detect that we never discard...)
266 fFlags &= ~kOpaqueAlpha_Flag;
267
268 return true;
269}
270
271SkShader::BitmapType SkTwoPointConicalGradient::asABitmap(
272 SkBitmap* bitmap, SkMatrix* matrix, SkShader::TileMode* xy) const {
273 SkPoint diff = fCenter2 - fCenter1;
rileya@google.com589708b2012-07-26 20:04:23 +0000274 SkScalar diffLen = 0;
275
276 if (bitmap) {
rileya@google.com1c6d64b2012-07-27 15:49:05 +0000277 this->getGradientTableBitmap(bitmap);
rileya@google.com589708b2012-07-26 20:04:23 +0000278 }
279 if (matrix) {
280 diffLen = diff.length();
281 }
282 if (matrix) {
283 if (diffLen) {
284 SkScalar invDiffLen = SkScalarInvert(diffLen);
285 // rotate to align circle centers with the x-axis
286 matrix->setSinCos(-SkScalarMul(invDiffLen, diff.fY),
287 SkScalarMul(invDiffLen, diff.fX));
288 } else {
289 matrix->reset();
290 }
291 matrix->preTranslate(-fCenter1.fX, -fCenter1.fY);
292 }
293 if (xy) {
294 xy[0] = fTileMode;
295 xy[1] = kClamp_TileMode;
296 }
297 return kTwoPointConical_BitmapType;
298}
299
300SkShader::GradientType SkTwoPointConicalGradient::asAGradient(
301 GradientInfo* info) const {
302 if (info) {
303 commonAsAGradient(info);
304 info->fPoint[0] = fCenter1;
305 info->fPoint[1] = fCenter2;
306 info->fRadius[0] = fRadius1;
307 info->fRadius[1] = fRadius2;
308 }
309 return kConical_GradientType;
310}
311
rileya@google.com589708b2012-07-26 20:04:23 +0000312SkTwoPointConicalGradient::SkTwoPointConicalGradient(
313 SkFlattenableReadBuffer& buffer)
314 : INHERITED(buffer),
315 fCenter1(buffer.readPoint()),
316 fCenter2(buffer.readPoint()),
317 fRadius1(buffer.readScalar()),
318 fRadius2(buffer.readScalar()) {
319 this->init();
320};
321
322void SkTwoPointConicalGradient::flatten(
323 SkFlattenableWriteBuffer& buffer) const {
324 this->INHERITED::flatten(buffer);
325 buffer.writePoint(fCenter1);
326 buffer.writePoint(fCenter2);
327 buffer.writeScalar(fRadius1);
328 buffer.writeScalar(fRadius2);
329}
330
rileya@google.comd7cc6512012-07-27 14:00:39 +0000331/////////////////////////////////////////////////////////////////////
332
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000333#if SK_SUPPORT_GPU
334
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000335#include "GrTBackendEffectFactory.h"
336
rileya@google.comd7cc6512012-07-27 14:00:39 +0000337// For brevity
338typedef GrGLUniformManager::UniformHandle UniformHandle;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000339
bsalomon@google.com0707c292012-10-25 21:45:42 +0000340class GrGLConical2Gradient : public GrGLGradientEffect {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000341public:
342
bsalomon@google.comc7818882013-03-20 19:19:53 +0000343 GrGLConical2Gradient(const GrBackendEffectFactory& factory, const GrDrawEffect&);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000344 virtual ~GrGLConical2Gradient() { }
345
bsalomon@google.comf78df332012-10-29 12:43:38 +0000346 virtual void emitCode(GrGLShaderBuilder*,
bsalomon@google.comc7818882013-03-20 19:19:53 +0000347 const GrDrawEffect&,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000348 EffectKey,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000349 const char* outputColor,
350 const char* inputColor,
351 const TextureSamplerArray&) SK_OVERRIDE;
bsalomon@google.comc7818882013-03-20 19:19:53 +0000352 virtual void setData(const GrGLUniformManager&, const GrDrawEffect&) SK_OVERRIDE;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000353
bsalomon@google.comc7818882013-03-20 19:19:53 +0000354 static EffectKey GenKey(const GrDrawEffect&, const GrGLCaps& caps);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000355
356protected:
357
358 UniformHandle fVSParamUni;
359 UniformHandle fFSParamUni;
360
361 const char* fVSVaryingName;
362 const char* fFSVaryingName;
363
364 bool fIsDegenerate;
365
366 // @{
367 /// Values last uploaded as uniforms
368
bsalomon@google.com81712882012-11-01 17:12:34 +0000369 SkScalar fCachedCenter;
370 SkScalar fCachedRadius;
371 SkScalar fCachedDiffRadius;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000372
373 // @}
374
375private:
376
bsalomon@google.com0707c292012-10-25 21:45:42 +0000377 typedef GrGLGradientEffect INHERITED;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000378
379};
380
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000381/////////////////////////////////////////////////////////////////////
382
383class GrConical2Gradient : public GrGradientEffect {
384public:
385
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000386 static GrEffectRef* Create(GrContext* ctx,
387 const SkTwoPointConicalGradient& shader,
388 const SkMatrix& matrix,
389 SkShader::TileMode tm) {
bsalomon@google.com6340a412013-01-22 19:55:59 +0000390 AutoEffectUnref effect(SkNEW_ARGS(GrConical2Gradient, (ctx, shader, matrix, tm)));
bsalomon@google.coma1ebbe42013-01-16 15:51:47 +0000391 return CreateEffectRef(effect);
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000392 }
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000393
394 virtual ~GrConical2Gradient() { }
395
396 static const char* Name() { return "Two-Point Conical Gradient"; }
bsalomon@google.com396e61f2012-10-25 19:00:29 +0000397 virtual const GrBackendEffectFactory& getFactory() const SK_OVERRIDE {
398 return GrTBackendEffectFactory<GrConical2Gradient>::getInstance();
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000399 }
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000400
401 // The radial gradient parameters can collapse to a linear (instead of quadratic) equation.
402 bool isDegenerate() const { return SkScalarAbs(fDiffRadius) == SkScalarAbs(fCenterX1); }
bsalomon@google.com81712882012-11-01 17:12:34 +0000403 SkScalar center() const { return fCenterX1; }
404 SkScalar diffRadius() const { return fDiffRadius; }
405 SkScalar radius() const { return fRadius0; }
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000406
bsalomon@google.com422e81a2012-10-25 14:11:03 +0000407 typedef GrGLConical2Gradient GLEffect;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000408
409private:
bsalomon@google.com8a252f72013-01-22 20:35:13 +0000410 virtual bool onIsEqual(const GrEffect& sBase) const SK_OVERRIDE {
bsalomon@google.com6340a412013-01-22 19:55:59 +0000411 const GrConical2Gradient& s = CastEffect<GrConical2Gradient>(sBase);
bsalomon@google.com68b58c92013-01-17 16:50:08 +0000412 return (INHERITED::onIsEqual(sBase) &&
413 this->fCenterX1 == s.fCenterX1 &&
414 this->fRadius0 == s.fRadius0 &&
415 this->fDiffRadius == s.fDiffRadius);
416 }
417
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000418 GrConical2Gradient(GrContext* ctx,
419 const SkTwoPointConicalGradient& shader,
420 const SkMatrix& matrix,
421 SkShader::TileMode tm)
422 : INHERITED(ctx, shader, matrix, tm)
423 , fCenterX1(shader.getCenterX1())
424 , fRadius0(shader.getStartRadius())
425 , fDiffRadius(shader.getDiffRadius()) { }
426
bsalomon@google.comf271cc72012-10-24 19:35:13 +0000427 GR_DECLARE_EFFECT_TEST;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000428
429 // @{
430 // Cache of values - these can change arbitrarily, EXCEPT
431 // we shouldn't change between degenerate and non-degenerate?!
432
bsalomon@google.com81712882012-11-01 17:12:34 +0000433 SkScalar fCenterX1;
434 SkScalar fRadius0;
435 SkScalar fDiffRadius;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000436
437 // @}
438
439 typedef GrGradientEffect INHERITED;
440};
441
bsalomon@google.comf271cc72012-10-24 19:35:13 +0000442GR_DEFINE_EFFECT_TEST(GrConical2Gradient);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000443
bsalomon@google.com73a96942013-02-13 16:31:19 +0000444GrEffectRef* GrConical2Gradient::TestCreate(SkMWCRandom* random,
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000445 GrContext* context,
bsalomon@google.comc26d94f2013-03-25 18:19:00 +0000446 const GrDrawTargetCaps&,
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000447 GrTexture**) {
bsalomon@google.comd4726202012-08-03 14:34:46 +0000448 SkPoint center1 = {random->nextUScalar1(), random->nextUScalar1()};
449 SkScalar radius1 = random->nextUScalar1();
450 SkPoint center2;
451 SkScalar radius2;
452 do {
bsalomon@google.comfb883bf2012-12-11 15:32:04 +0000453 center2.set(random->nextUScalar1(), random->nextUScalar1());
bsalomon@google.comd4726202012-08-03 14:34:46 +0000454 radius2 = random->nextUScalar1 ();
455 // If the circles are identical the factory will give us an empty shader.
456 } while (radius1 == radius2 && center1 == center2);
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000457
bsalomon@google.comd4726202012-08-03 14:34:46 +0000458 SkColor colors[kMaxRandomGradientColors];
459 SkScalar stopsArray[kMaxRandomGradientColors];
460 SkScalar* stops = stopsArray;
461 SkShader::TileMode tm;
462 int colorCount = RandomGradientParams(random, colors, &stops, &tm);
463 SkAutoTUnref<SkShader> shader(SkGradientShader::CreateTwoPointConical(center1, radius1,
464 center2, radius2,
465 colors, stops, colorCount,
466 tm));
bsalomon@google.come197cbf2013-01-14 16:46:26 +0000467 SkPaint paint;
468 return shader->asNewEffect(context, paint);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000469}
470
471
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000472/////////////////////////////////////////////////////////////////////
473
bsalomon@google.com6340a412013-01-22 19:55:59 +0000474GrGLConical2Gradient::GrGLConical2Gradient(const GrBackendEffectFactory& factory,
bsalomon@google.comc7818882013-03-20 19:19:53 +0000475 const GrDrawEffect& drawEffect)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000476 : INHERITED(factory)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000477 , fVSVaryingName(NULL)
478 , fFSVaryingName(NULL)
bsalomon@google.com81712882012-11-01 17:12:34 +0000479 , fCachedCenter(SK_ScalarMax)
480 , fCachedRadius(-SK_ScalarMax)
481 , fCachedDiffRadius(-SK_ScalarMax) {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000482
bsalomon@google.comc7818882013-03-20 19:19:53 +0000483 const GrConical2Gradient& data = drawEffect.castEffect<GrConical2Gradient>();
rileya@google.comd7cc6512012-07-27 14:00:39 +0000484 fIsDegenerate = data.isDegenerate();
485}
486
bsalomon@google.comf78df332012-10-29 12:43:38 +0000487void GrGLConical2Gradient::emitCode(GrGLShaderBuilder* builder,
bsalomon@google.comc7818882013-03-20 19:19:53 +0000488 const GrDrawEffect&,
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000489 EffectKey key,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000490 const char* outputColor,
491 const char* inputColor,
492 const TextureSamplerArray& samplers) {
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000493 const char* fsCoords;
494 const char* vsCoordsVarying;
495 GrSLType coordsVaryingType;
bsalomon@google.comc7818882013-03-20 19:19:53 +0000496 this->setupMatrix(builder, key, &fsCoords, &vsCoordsVarying, &coordsVaryingType);
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000497
bsalomon@google.comf78df332012-10-29 12:43:38 +0000498 this->emitYCoordUniform(builder);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000499 // 2 copies of uniform array, 1 for each of vertex & fragment shader,
500 // to work around Xoom bug. Doesn't seem to cause performance decrease
501 // in test apps, but need to keep an eye on it.
502 fVSParamUni = builder->addUniformArray(GrGLShaderBuilder::kVertex_ShaderType,
503 kFloat_GrSLType, "Conical2VSParams", 6);
504 fFSParamUni = builder->addUniformArray(GrGLShaderBuilder::kFragment_ShaderType,
505 kFloat_GrSLType, "Conical2FSParams", 6);
506
507 // For radial gradients without perspective we can pass the linear
508 // part of the quadratic as a varying.
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000509 if (kVec2f_GrSLType == coordsVaryingType) {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000510 builder->addVarying(kFloat_GrSLType, "Conical2BCoeff",
511 &fVSVaryingName, &fFSVaryingName);
512 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000513
bsalomon@google.comf78df332012-10-29 12:43:38 +0000514 // VS
515 {
bsalomon@google.comf78df332012-10-29 12:43:38 +0000516 SkString p2; // distance between centers
517 SkString p3; // start radius
518 SkString p5; // difference in radii (r1 - r0)
519 builder->getUniformVariable(fVSParamUni).appendArrayAccess(2, &p2);
520 builder->getUniformVariable(fVSParamUni).appendArrayAccess(3, &p3);
521 builder->getUniformVariable(fVSParamUni).appendArrayAccess(5, &p5);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000522
bsalomon@google.comf78df332012-10-29 12:43:38 +0000523 // For radial gradients without perspective we can pass the linear
524 // part of the quadratic as a varying.
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000525 if (kVec2f_GrSLType == coordsVaryingType) {
bsalomon@google.comf78df332012-10-29 12:43:38 +0000526 // r2Var = -2 * (r2Parm[2] * varCoord.x - r2Param[3] * r2Param[5])
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000527 builder->vsCodeAppendf("\t%s = -2.0 * (%s * %s.x + %s * %s);\n",
528 fVSVaryingName, p2.c_str(),
529 vsCoordsVarying, p3.c_str(), p5.c_str());
bsalomon@google.comf78df332012-10-29 12:43:38 +0000530 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000531 }
532
bsalomon@google.comf78df332012-10-29 12:43:38 +0000533 // FS
534 {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000535
bsalomon@google.comf78df332012-10-29 12:43:38 +0000536 SkString cName("c");
537 SkString ac4Name("ac4");
538 SkString dName("d");
539 SkString qName("q");
540 SkString r0Name("r0");
541 SkString r1Name("r1");
542 SkString tName("t");
543 SkString p0; // 4a
544 SkString p1; // 1/a
545 SkString p2; // distance between centers
546 SkString p3; // start radius
547 SkString p4; // start radius squared
548 SkString p5; // difference in radii (r1 - r0)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000549
bsalomon@google.comf78df332012-10-29 12:43:38 +0000550 builder->getUniformVariable(fFSParamUni).appendArrayAccess(0, &p0);
551 builder->getUniformVariable(fFSParamUni).appendArrayAccess(1, &p1);
552 builder->getUniformVariable(fFSParamUni).appendArrayAccess(2, &p2);
553 builder->getUniformVariable(fFSParamUni).appendArrayAccess(3, &p3);
554 builder->getUniformVariable(fFSParamUni).appendArrayAccess(4, &p4);
555 builder->getUniformVariable(fFSParamUni).appendArrayAccess(5, &p5);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000556
bsalomon@google.comf78df332012-10-29 12:43:38 +0000557 // If we we're able to interpolate the linear component,
558 // bVar is the varying; otherwise compute it
559 SkString bVar;
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000560 if (kVec2f_GrSLType == coordsVaryingType) {
bsalomon@google.comf78df332012-10-29 12:43:38 +0000561 bVar = fFSVaryingName;
562 } else {
563 bVar = "b";
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000564 builder->fsCodeAppendf("\tfloat %s = -2.0 * (%s * %s.x + %s * %s);\n",
565 bVar.c_str(), p2.c_str(), fsCoords,
566 p3.c_str(), p5.c_str());
bsalomon@google.comf78df332012-10-29 12:43:38 +0000567 }
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000568
bsalomon@google.comf78df332012-10-29 12:43:38 +0000569 // output will default to transparent black (we simply won't write anything
570 // else to it if invalid, instead of discarding or returning prematurely)
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000571 builder->fsCodeAppendf("\t%s = vec4(0.0,0.0,0.0,0.0);\n", outputColor);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000572
bsalomon@google.comf78df332012-10-29 12:43:38 +0000573 // c = (x^2)+(y^2) - params[4]
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000574 builder->fsCodeAppendf("\tfloat %s = dot(%s, %s) - %s;\n", cName.c_str(),
575 fsCoords, fsCoords,
576 p4.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000577
bsalomon@google.comf78df332012-10-29 12:43:38 +0000578 // Non-degenerate case (quadratic)
579 if (!fIsDegenerate) {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000580
bsalomon@google.comf78df332012-10-29 12:43:38 +0000581 // ac4 = params[0] * c
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000582 builder->fsCodeAppendf("\tfloat %s = %s * %s;\n", ac4Name.c_str(), p0.c_str(),
583 cName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000584
bsalomon@google.comf78df332012-10-29 12:43:38 +0000585 // d = b^2 - ac4
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000586 builder->fsCodeAppendf("\tfloat %s = %s * %s - %s;\n", dName.c_str(),
587 bVar.c_str(), bVar.c_str(), ac4Name.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000588
bsalomon@google.comf78df332012-10-29 12:43:38 +0000589 // only proceed if discriminant is >= 0
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000590 builder->fsCodeAppendf("\tif (%s >= 0.0) {\n", dName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000591
bsalomon@google.comf78df332012-10-29 12:43:38 +0000592 // intermediate value we'll use to compute the roots
593 // q = -0.5 * (b +/- sqrt(d))
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000594 builder->fsCodeAppendf("\t\tfloat %s = -0.5 * (%s + (%s < 0.0 ? -1.0 : 1.0)"
595 " * sqrt(%s));\n", qName.c_str(), bVar.c_str(),
596 bVar.c_str(), dName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000597
bsalomon@google.comf78df332012-10-29 12:43:38 +0000598 // compute both roots
599 // r0 = q * params[1]
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000600 builder->fsCodeAppendf("\t\tfloat %s = %s * %s;\n", r0Name.c_str(),
601 qName.c_str(), p1.c_str());
bsalomon@google.comf78df332012-10-29 12:43:38 +0000602 // r1 = c / q
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000603 builder->fsCodeAppendf("\t\tfloat %s = %s / %s;\n", r1Name.c_str(),
604 cName.c_str(), qName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000605
bsalomon@google.comf78df332012-10-29 12:43:38 +0000606 // Note: If there are two roots that both generate radius(t) > 0, the
607 // Canvas spec says to choose the larger t.
rileya@google.comd7cc6512012-07-27 14:00:39 +0000608
bsalomon@google.comf78df332012-10-29 12:43:38 +0000609 // so we'll look at the larger one first:
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000610 builder->fsCodeAppendf("\t\tfloat %s = max(%s, %s);\n", tName.c_str(),
611 r0Name.c_str(), r1Name.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000612
bsalomon@google.comf78df332012-10-29 12:43:38 +0000613 // if r(t) > 0, then we're done; t will be our x coordinate
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000614 builder->fsCodeAppendf("\t\tif (%s * %s + %s > 0.0) {\n", tName.c_str(),
615 p5.c_str(), p3.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000616
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000617 builder->fsCodeAppend("\t\t");
bsalomon@google.comf78df332012-10-29 12:43:38 +0000618 this->emitColorLookup(builder, tName.c_str(), outputColor, inputColor, samplers[0]);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000619
bsalomon@google.comf78df332012-10-29 12:43:38 +0000620 // otherwise, if r(t) for the larger root was <= 0, try the other root
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000621 builder->fsCodeAppend("\t\t} else {\n");
622 builder->fsCodeAppendf("\t\t\t%s = min(%s, %s);\n", tName.c_str(),
623 r0Name.c_str(), r1Name.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000624
bsalomon@google.comf78df332012-10-29 12:43:38 +0000625 // if r(t) > 0 for the smaller root, then t will be our x coordinate
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000626 builder->fsCodeAppendf("\t\t\tif (%s * %s + %s > 0.0) {\n",
627 tName.c_str(), p5.c_str(), p3.c_str());
bsalomon@google.comf78df332012-10-29 12:43:38 +0000628
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000629 builder->fsCodeAppend("\t\t\t");
bsalomon@google.comf78df332012-10-29 12:43:38 +0000630 this->emitColorLookup(builder, tName.c_str(), outputColor, inputColor, samplers[0]);
631
632 // end if (r(t) > 0) for smaller root
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000633 builder->fsCodeAppend("\t\t\t}\n");
bsalomon@google.comf78df332012-10-29 12:43:38 +0000634 // end if (r(t) > 0), else, for larger root
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000635 builder->fsCodeAppend("\t\t}\n");
bsalomon@google.comf78df332012-10-29 12:43:38 +0000636 // end if (discriminant >= 0)
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000637 builder->fsCodeAppend("\t}\n");
bsalomon@google.comf78df332012-10-29 12:43:38 +0000638 } else {
639
640 // linear case: t = -c/b
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000641 builder->fsCodeAppendf("\tfloat %s = -(%s / %s);\n", tName.c_str(),
642 cName.c_str(), bVar.c_str());
bsalomon@google.comf78df332012-10-29 12:43:38 +0000643
644 // if r(t) > 0, then t will be the x coordinate
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000645 builder->fsCodeAppendf("\tif (%s * %s + %s > 0.0) {\n", tName.c_str(),
646 p5.c_str(), p3.c_str());
647 builder->fsCodeAppend("\t");
bsalomon@google.comf78df332012-10-29 12:43:38 +0000648 this->emitColorLookup(builder, tName.c_str(), outputColor, inputColor, samplers[0]);
bsalomon@google.comf910d3b2013-03-07 17:06:57 +0000649 builder->fsCodeAppend("\t}\n");
bsalomon@google.comf78df332012-10-29 12:43:38 +0000650 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000651 }
652}
653
bsalomon@google.comc7818882013-03-20 19:19:53 +0000654void GrGLConical2Gradient::setData(const GrGLUniformManager& uman,
655 const GrDrawEffect& drawEffect) {
656 INHERITED::setData(uman, drawEffect);
657 const GrConical2Gradient& data = drawEffect.castEffect<GrConical2Gradient>();
commit-bot@chromium.org96ae6882013-08-14 12:09:00 +0000658 SkASSERT(data.isDegenerate() == fIsDegenerate);
bsalomon@google.com81712882012-11-01 17:12:34 +0000659 SkScalar centerX1 = data.center();
660 SkScalar radius0 = data.radius();
661 SkScalar diffRadius = data.diffRadius();
rileya@google.comd7cc6512012-07-27 14:00:39 +0000662
663 if (fCachedCenter != centerX1 ||
664 fCachedRadius != radius0 ||
665 fCachedDiffRadius != diffRadius) {
666
bsalomon@google.com81712882012-11-01 17:12:34 +0000667 SkScalar a = SkScalarMul(centerX1, centerX1) - diffRadius * diffRadius;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000668
669 // When we're in the degenerate (linear) case, the second
670 // value will be INF but the program doesn't read it. (We
671 // use the same 6 uniforms even though we don't need them
672 // all in the linear case just to keep the code complexity
673 // down).
674 float values[6] = {
bsalomon@google.com81712882012-11-01 17:12:34 +0000675 SkScalarToFloat(a * 4),
676 1.f / (SkScalarToFloat(a)),
677 SkScalarToFloat(centerX1),
678 SkScalarToFloat(radius0),
679 SkScalarToFloat(SkScalarMul(radius0, radius0)),
680 SkScalarToFloat(diffRadius)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000681 };
682
683 uman.set1fv(fVSParamUni, 0, 6, values);
684 uman.set1fv(fFSParamUni, 0, 6, values);
685 fCachedCenter = centerX1;
686 fCachedRadius = radius0;
687 fCachedDiffRadius = diffRadius;
688 }
689}
690
bsalomon@google.comc7818882013-03-20 19:19:53 +0000691GrGLEffect::EffectKey GrGLConical2Gradient::GenKey(const GrDrawEffect& drawEffect,
692 const GrGLCaps&) {
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000693 enum {
694 kIsDegenerate = 1 << kMatrixKeyBitCnt,
695 };
696
bsalomon@google.comc7818882013-03-20 19:19:53 +0000697 EffectKey key = GenMatrixKey(drawEffect);
698 if (drawEffect.castEffect<GrConical2Gradient>().isDegenerate()) {
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000699 key |= kIsDegenerate;
700 }
701 return key;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000702}
rileya@google.comd7cc6512012-07-27 14:00:39 +0000703
704/////////////////////////////////////////////////////////////////////
705
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000706GrEffectRef* SkTwoPointConicalGradient::asNewEffect(GrContext* context, const SkPaint&) const {
bsalomon@google.com00835cc2013-01-14 17:07:22 +0000707 SkASSERT(NULL != context);
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000708 SkASSERT(fPtsToUnit.isIdentity());
709 // invert the localM, translate to center1, rotate so center2 is on x axis.
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000710 SkMatrix matrix;
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000711 if (!this->getLocalMatrix().invert(&matrix)) {
humper@google.com84831ac2013-01-14 22:09:54 +0000712 return NULL;
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000713 }
714 matrix.postTranslate(-fCenter1.fX, -fCenter1.fY);
715
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000716 SkPoint diff = fCenter2 - fCenter1;
717 SkScalar diffLen = diff.length();
718 if (0 != diffLen) {
719 SkScalar invDiffLen = SkScalarInvert(diffLen);
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000720 SkMatrix rot;
721 rot.setSinCos(-SkScalarMul(invDiffLen, diff.fY),
722 SkScalarMul(invDiffLen, diff.fX));
723 matrix.postConcat(rot);
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000724 }
725
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000726 return GrConical2Gradient::Create(context, *this, matrix, fTileMode);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000727}
728
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000729#else
730
bsalomon@google.com5d2cd202013-01-16 15:31:06 +0000731GrEffectRef* SkTwoPointConicalGradient::asNewEffect(GrContext*, const SkPaint&) const {
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000732 SkDEBUGFAIL("Should not call in GPU-less build");
bsalomon@google.come197cbf2013-01-14 16:46:26 +0000733 return NULL;
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000734}
735
twiz@google.coma5e65ec2012-08-02 15:15:16 +0000736#endif
robertphillips@google.com76f9e932013-01-15 20:17:47 +0000737
738#ifdef SK_DEVELOPER
739void SkTwoPointConicalGradient::toString(SkString* str) const {
740 str->append("SkTwoPointConicalGradient: (");
741
742 str->append("center1: (");
743 str->appendScalar(fCenter1.fX);
744 str->append(", ");
745 str->appendScalar(fCenter1.fY);
746 str->append(") radius1: ");
747 str->appendScalar(fRadius1);
748 str->append(" ");
749
750 str->append("center2: (");
751 str->appendScalar(fCenter2.fX);
752 str->append(", ");
753 str->appendScalar(fCenter2.fY);
754 str->append(") radius2: ");
755 str->appendScalar(fRadius2);
756 str->append(" ");
757
758 this->INHERITED::toString(str);
759
760 str->append(")");
761}
762#endif