blob: 434339cde761c3a71c33800f8b6bee69fa8cd3ae [file] [log] [blame]
rileya@google.com589708b2012-07-26 20:04:23 +00001/*
2 * Copyright 2012 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7
8#include "SkTwoPointConicalGradient.h"
9
10static int valid_divide(float numer, float denom, float* ratio) {
11 SkASSERT(ratio);
12 if (0 == denom) {
13 return 0;
14 }
15 *ratio = numer / denom;
16 return 1;
17}
18
19// Return the number of distinct real roots, and write them into roots[] in
20// ascending order
21static int find_quad_roots(float A, float B, float C, float roots[2]) {
22 SkASSERT(roots);
rmistry@google.comfbfcd562012-08-23 18:09:54 +000023
rileya@google.com589708b2012-07-26 20:04:23 +000024 if (A == 0) {
25 return valid_divide(-C, B, roots);
26 }
rmistry@google.comfbfcd562012-08-23 18:09:54 +000027
rileya@google.com589708b2012-07-26 20:04:23 +000028 float R = B*B - 4*A*C;
29 if (R < 0) {
30 return 0;
31 }
32 R = sk_float_sqrt(R);
33
34#if 1
35 float Q = B;
36 if (Q < 0) {
37 Q -= R;
38 } else {
39 Q += R;
40 }
41#else
42 // on 10.6 this was much slower than the above branch :(
43 float Q = B + copysignf(R, B);
44#endif
45 Q *= -0.5f;
46 if (0 == Q) {
47 roots[0] = 0;
48 return 1;
49 }
50
51 float r0 = Q / A;
52 float r1 = C / Q;
53 roots[0] = r0 < r1 ? r0 : r1;
54 roots[1] = r0 > r1 ? r0 : r1;
55 return 2;
56}
57
58static float lerp(float x, float dx, float t) {
59 return x + t * dx;
60}
61
62static float sqr(float x) { return x * x; }
63
64void TwoPtRadial::init(const SkPoint& center0, SkScalar rad0,
65 const SkPoint& center1, SkScalar rad1) {
66 fCenterX = SkScalarToFloat(center0.fX);
67 fCenterY = SkScalarToFloat(center0.fY);
68 fDCenterX = SkScalarToFloat(center1.fX) - fCenterX;
69 fDCenterY = SkScalarToFloat(center1.fY) - fCenterY;
70 fRadius = SkScalarToFloat(rad0);
71 fDRadius = SkScalarToFloat(rad1) - fRadius;
72
73 fA = sqr(fDCenterX) + sqr(fDCenterY) - sqr(fDRadius);
74 fRadius2 = sqr(fRadius);
75 fRDR = fRadius * fDRadius;
76}
77
78void TwoPtRadial::setup(SkScalar fx, SkScalar fy, SkScalar dfx, SkScalar dfy) {
79 fRelX = SkScalarToFloat(fx) - fCenterX;
80 fRelY = SkScalarToFloat(fy) - fCenterY;
81 fIncX = SkScalarToFloat(dfx);
82 fIncY = SkScalarToFloat(dfy);
83 fB = -2 * (fDCenterX * fRelX + fDCenterY * fRelY + fRDR);
84 fDB = -2 * (fDCenterX * fIncX + fDCenterY * fIncY);
85}
86
87SkFixed TwoPtRadial::nextT() {
88 float roots[2];
rmistry@google.comfbfcd562012-08-23 18:09:54 +000089
rileya@google.com589708b2012-07-26 20:04:23 +000090 float C = sqr(fRelX) + sqr(fRelY) - fRadius2;
91 int countRoots = find_quad_roots(fA, fB, C, roots);
92
93 fRelX += fIncX;
94 fRelY += fIncY;
95 fB += fDB;
96
97 if (0 == countRoots) {
98 return kDontDrawT;
99 }
100
101 // Prefer the bigger t value if both give a radius(t) > 0
102 // find_quad_roots returns the values sorted, so we start with the last
103 float t = roots[countRoots - 1];
104 float r = lerp(fRadius, fDRadius, t);
105 if (r <= 0) {
106 t = roots[0]; // might be the same as roots[countRoots-1]
107 r = lerp(fRadius, fDRadius, t);
108 if (r <= 0) {
109 return kDontDrawT;
110 }
111 }
112 return SkFloatToFixed(t);
113}
114
reed@google.com60040292013-02-04 18:21:23 +0000115typedef void (*TwoPointConicalProc)(TwoPtRadial* rec, SkPMColor* dstC,
116 const SkPMColor* cache, int toggle, int count);
rileya@google.com589708b2012-07-26 20:04:23 +0000117
118static void twopoint_clamp(TwoPtRadial* rec, SkPMColor* SK_RESTRICT dstC,
reed@google.com60040292013-02-04 18:21:23 +0000119 const SkPMColor* SK_RESTRICT cache, int toggle,
120 int count) {
rileya@google.com589708b2012-07-26 20:04:23 +0000121 for (; count > 0; --count) {
122 SkFixed t = rec->nextT();
123 if (TwoPtRadial::DontDrawT(t)) {
124 *dstC++ = 0;
125 } else {
126 SkFixed index = SkClampMax(t, 0xFFFF);
127 SkASSERT(index <= 0xFFFF);
reed@google.com60040292013-02-04 18:21:23 +0000128 *dstC++ = cache[toggle +
129 (index >> SkGradientShaderBase::kCache32Shift)];
rileya@google.com589708b2012-07-26 20:04:23 +0000130 }
reed@google.com60040292013-02-04 18:21:23 +0000131 toggle = next_dither_toggle(toggle);
rileya@google.com589708b2012-07-26 20:04:23 +0000132 }
133}
134
135static void twopoint_repeat(TwoPtRadial* rec, SkPMColor* SK_RESTRICT dstC,
reed@google.com60040292013-02-04 18:21:23 +0000136 const SkPMColor* SK_RESTRICT cache, int toggle,
137 int count) {
rileya@google.com589708b2012-07-26 20:04:23 +0000138 for (; count > 0; --count) {
139 SkFixed t = rec->nextT();
140 if (TwoPtRadial::DontDrawT(t)) {
141 *dstC++ = 0;
142 } else {
143 SkFixed index = repeat_tileproc(t);
144 SkASSERT(index <= 0xFFFF);
reed@google.com60040292013-02-04 18:21:23 +0000145 *dstC++ = cache[toggle +
146 (index >> SkGradientShaderBase::kCache32Shift)];
rileya@google.com589708b2012-07-26 20:04:23 +0000147 }
reed@google.com60040292013-02-04 18:21:23 +0000148 toggle = next_dither_toggle(toggle);
rileya@google.com589708b2012-07-26 20:04:23 +0000149 }
150}
151
152static void twopoint_mirror(TwoPtRadial* rec, SkPMColor* SK_RESTRICT dstC,
reed@google.com60040292013-02-04 18:21:23 +0000153 const SkPMColor* SK_RESTRICT cache, int toggle,
154 int count) {
rileya@google.com589708b2012-07-26 20:04:23 +0000155 for (; count > 0; --count) {
156 SkFixed t = rec->nextT();
157 if (TwoPtRadial::DontDrawT(t)) {
158 *dstC++ = 0;
159 } else {
160 SkFixed index = mirror_tileproc(t);
161 SkASSERT(index <= 0xFFFF);
reed@google.com60040292013-02-04 18:21:23 +0000162 *dstC++ = cache[toggle +
163 (index >> SkGradientShaderBase::kCache32Shift)];
rileya@google.com589708b2012-07-26 20:04:23 +0000164 }
reed@google.com60040292013-02-04 18:21:23 +0000165 toggle = next_dither_toggle(toggle);
rileya@google.com589708b2012-07-26 20:04:23 +0000166 }
167}
168
169void SkTwoPointConicalGradient::init() {
170 fRec.init(fCenter1, fRadius1, fCenter2, fRadius2);
171 fPtsToUnit.reset();
172}
173
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000174/////////////////////////////////////////////////////////////////////
175
rileya@google.com589708b2012-07-26 20:04:23 +0000176SkTwoPointConicalGradient::SkTwoPointConicalGradient(
reed@google.com3d3a8602013-05-24 14:58:44 +0000177 const SkPoint& start, SkScalar startRadius,
178 const SkPoint& end, SkScalar endRadius,
179 const Descriptor& desc)
reed@google.com437d6eb2013-05-23 19:03:05 +0000180 : SkGradientShaderBase(desc),
rileya@google.com589708b2012-07-26 20:04:23 +0000181 fCenter1(start),
182 fCenter2(end),
183 fRadius1(startRadius),
184 fRadius2(endRadius) {
185 // this is degenerate, and should be caught by our caller
186 SkASSERT(fCenter1 != fCenter2 || fRadius1 != fRadius2);
187 this->init();
188}
189
commit-bot@chromium.org3fbab822013-03-20 00:49:57 +0000190bool SkTwoPointConicalGradient::isOpaque() const {
robertphillips@google.comcb6d97c2013-07-09 13:50:09 +0000191 // Because areas outside the cone are left untouched, we cannot treat the
192 // shader as opaque even if the gradient itself is opaque.
193 // TODO(junov): Compute whether the cone fills the plane crbug.com/222380
194 return false;
commit-bot@chromium.org3fbab822013-03-20 00:49:57 +0000195}
196
rileya@google.com589708b2012-07-26 20:04:23 +0000197void SkTwoPointConicalGradient::shadeSpan(int x, int y, SkPMColor* dstCParam,
198 int count) {
reed@google.com60040292013-02-04 18:21:23 +0000199 int toggle = init_dither_toggle(x, y);
reed@google.com60040292013-02-04 18:21:23 +0000200
rileya@google.com589708b2012-07-26 20:04:23 +0000201 SkASSERT(count > 0);
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000202
rileya@google.com589708b2012-07-26 20:04:23 +0000203 SkPMColor* SK_RESTRICT dstC = dstCParam;
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000204
rileya@google.com589708b2012-07-26 20:04:23 +0000205 SkMatrix::MapXYProc dstProc = fDstToIndexProc;
bsalomon@google.com100abf42012-09-05 17:40:04 +0000206
rileya@google.com589708b2012-07-26 20:04:23 +0000207 const SkPMColor* SK_RESTRICT cache = this->getCache32();
208
reed@google.com60040292013-02-04 18:21:23 +0000209 TwoPointConicalProc shadeProc = twopoint_repeat;
rileya@google.com589708b2012-07-26 20:04:23 +0000210 if (SkShader::kClamp_TileMode == fTileMode) {
211 shadeProc = twopoint_clamp;
212 } else if (SkShader::kMirror_TileMode == fTileMode) {
213 shadeProc = twopoint_mirror;
214 } else {
215 SkASSERT(SkShader::kRepeat_TileMode == fTileMode);
216 }
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000217
rileya@google.com589708b2012-07-26 20:04:23 +0000218 if (fDstToIndexClass != kPerspective_MatrixClass) {
219 SkPoint srcPt;
220 dstProc(fDstToIndex, SkIntToScalar(x) + SK_ScalarHalf,
221 SkIntToScalar(y) + SK_ScalarHalf, &srcPt);
222 SkScalar dx, fx = srcPt.fX;
223 SkScalar dy, fy = srcPt.fY;
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000224
rileya@google.com589708b2012-07-26 20:04:23 +0000225 if (fDstToIndexClass == kFixedStepInX_MatrixClass) {
226 SkFixed fixedX, fixedY;
227 (void)fDstToIndex.fixedStepInX(SkIntToScalar(y), &fixedX, &fixedY);
228 dx = SkFixedToScalar(fixedX);
229 dy = SkFixedToScalar(fixedY);
230 } else {
231 SkASSERT(fDstToIndexClass == kLinear_MatrixClass);
232 dx = fDstToIndex.getScaleX();
233 dy = fDstToIndex.getSkewY();
234 }
235
236 fRec.setup(fx, fy, dx, dy);
reed@google.com60040292013-02-04 18:21:23 +0000237 (*shadeProc)(&fRec, dstC, cache, toggle, count);
rileya@google.com589708b2012-07-26 20:04:23 +0000238 } else { // perspective case
mike@reedtribe.org139a2352013-11-14 20:15:51 +0000239 SkScalar dstX = SkIntToScalar(x) + SK_ScalarHalf;
240 SkScalar dstY = SkIntToScalar(y) + SK_ScalarHalf;
rileya@google.com589708b2012-07-26 20:04:23 +0000241 for (; count > 0; --count) {
242 SkPoint srcPt;
243 dstProc(fDstToIndex, dstX, dstY, &srcPt);
rileya@google.com589708b2012-07-26 20:04:23 +0000244 fRec.setup(srcPt.fX, srcPt.fY, 0, 0);
reed@google.com60040292013-02-04 18:21:23 +0000245 (*shadeProc)(&fRec, dstC, cache, toggle, 1);
mike@reedtribe.org139a2352013-11-14 20:15:51 +0000246
247 dstX += SK_Scalar1;
reed@google.com60040292013-02-04 18:21:23 +0000248 toggle = next_dither_toggle(toggle);
mike@reedtribe.org139a2352013-11-14 20:15:51 +0000249 dstC += 1;
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(
commit-bot@chromium.org8b0e8ac2014-01-30 18:58:24 +0000313 SkReadBuffer& buffer)
rileya@google.com589708b2012-07-26 20:04:23 +0000314 : 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(
commit-bot@chromium.org8b0e8ac2014-01-30 18:58:24 +0000323 SkWriteBuffer& buffer) const {
rileya@google.com589708b2012-07-26 20:04:23 +0000324 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,
bsalomon@google.com77af6802013-10-02 13:04:56 +0000351 const TransformedCoordsArray&,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000352 const TextureSamplerArray&) SK_OVERRIDE;
bsalomon@google.comc7818882013-03-20 19:19:53 +0000353 virtual void setData(const GrGLUniformManager&, const GrDrawEffect&) SK_OVERRIDE;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000354
bsalomon@google.comc7818882013-03-20 19:19:53 +0000355 static EffectKey GenKey(const GrDrawEffect&, const GrGLCaps& caps);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000356
357protected:
358
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000359 UniformHandle fParamUni;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000360
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())
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000425 , fDiffRadius(shader.getDiffRadius()) {
426 // We pass the linear part of the quadratic as a varying.
427 // float b = -2.0 * (fCenterX1 * x + fRadius0 * fDiffRadius * z)
428 fBTransform = this->getCoordTransform();
429 SkMatrix& bMatrix = *fBTransform.accessMatrix();
430 SkScalar r0dr = SkScalarMul(fRadius0, fDiffRadius);
431 bMatrix[SkMatrix::kMScaleX] = -2 * (SkScalarMul(fCenterX1, bMatrix[SkMatrix::kMScaleX]) +
432 SkScalarMul(r0dr, bMatrix[SkMatrix::kMPersp0]));
433 bMatrix[SkMatrix::kMSkewX] = -2 * (SkScalarMul(fCenterX1, bMatrix[SkMatrix::kMSkewX]) +
434 SkScalarMul(r0dr, bMatrix[SkMatrix::kMPersp1]));
435 bMatrix[SkMatrix::kMTransX] = -2 * (SkScalarMul(fCenterX1, bMatrix[SkMatrix::kMTransX]) +
436 SkScalarMul(r0dr, bMatrix[SkMatrix::kMPersp2]));
437 this->addCoordTransform(&fBTransform);
438 }
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000439
bsalomon@google.comf271cc72012-10-24 19:35:13 +0000440 GR_DECLARE_EFFECT_TEST;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000441
442 // @{
443 // Cache of values - these can change arbitrarily, EXCEPT
444 // we shouldn't change between degenerate and non-degenerate?!
445
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000446 GrCoordTransform fBTransform;
447 SkScalar fCenterX1;
448 SkScalar fRadius0;
449 SkScalar fDiffRadius;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000450
451 // @}
452
453 typedef GrGradientEffect INHERITED;
454};
455
bsalomon@google.comf271cc72012-10-24 19:35:13 +0000456GR_DEFINE_EFFECT_TEST(GrConical2Gradient);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000457
commit-bot@chromium.orge0e7cfe2013-09-09 20:09:12 +0000458GrEffectRef* GrConical2Gradient::TestCreate(SkRandom* random,
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000459 GrContext* context,
bsalomon@google.comc26d94f2013-03-25 18:19:00 +0000460 const GrDrawTargetCaps&,
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000461 GrTexture**) {
bsalomon@google.comd4726202012-08-03 14:34:46 +0000462 SkPoint center1 = {random->nextUScalar1(), random->nextUScalar1()};
463 SkScalar radius1 = random->nextUScalar1();
464 SkPoint center2;
465 SkScalar radius2;
466 do {
bsalomon@google.comfb883bf2012-12-11 15:32:04 +0000467 center2.set(random->nextUScalar1(), random->nextUScalar1());
bsalomon@google.comd4726202012-08-03 14:34:46 +0000468 radius2 = random->nextUScalar1 ();
469 // If the circles are identical the factory will give us an empty shader.
470 } while (radius1 == radius2 && center1 == center2);
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000471
bsalomon@google.comd4726202012-08-03 14:34:46 +0000472 SkColor colors[kMaxRandomGradientColors];
473 SkScalar stopsArray[kMaxRandomGradientColors];
474 SkScalar* stops = stopsArray;
475 SkShader::TileMode tm;
476 int colorCount = RandomGradientParams(random, colors, &stops, &tm);
477 SkAutoTUnref<SkShader> shader(SkGradientShader::CreateTwoPointConical(center1, radius1,
478 center2, radius2,
479 colors, stops, colorCount,
480 tm));
bsalomon@google.come197cbf2013-01-14 16:46:26 +0000481 SkPaint paint;
482 return shader->asNewEffect(context, paint);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000483}
484
485
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000486/////////////////////////////////////////////////////////////////////
487
bsalomon@google.com6340a412013-01-22 19:55:59 +0000488GrGLConical2Gradient::GrGLConical2Gradient(const GrBackendEffectFactory& factory,
bsalomon@google.comc7818882013-03-20 19:19:53 +0000489 const GrDrawEffect& drawEffect)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000490 : INHERITED(factory)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000491 , fVSVaryingName(NULL)
492 , fFSVaryingName(NULL)
bsalomon@google.com81712882012-11-01 17:12:34 +0000493 , fCachedCenter(SK_ScalarMax)
494 , fCachedRadius(-SK_ScalarMax)
495 , fCachedDiffRadius(-SK_ScalarMax) {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000496
bsalomon@google.comc7818882013-03-20 19:19:53 +0000497 const GrConical2Gradient& data = drawEffect.castEffect<GrConical2Gradient>();
rileya@google.comd7cc6512012-07-27 14:00:39 +0000498 fIsDegenerate = data.isDegenerate();
499}
500
bsalomon@google.comf78df332012-10-29 12:43:38 +0000501void GrGLConical2Gradient::emitCode(GrGLShaderBuilder* builder,
bsalomon@google.comc7818882013-03-20 19:19:53 +0000502 const GrDrawEffect&,
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000503 EffectKey key,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000504 const char* outputColor,
505 const char* inputColor,
bsalomon@google.com77af6802013-10-02 13:04:56 +0000506 const TransformedCoordsArray& coords,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000507 const TextureSamplerArray& samplers) {
bsalomon@google.com82d12232013-09-09 15:36:26 +0000508 this->emitUniforms(builder, key);
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000509 fParamUni = builder->addUniformArray(GrGLShaderBuilder::kFragment_Visibility,
510 kFloat_GrSLType, "Conical2FSParams", 6);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000511
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000512 SkString cName("c");
513 SkString ac4Name("ac4");
514 SkString dName("d");
515 SkString qName("q");
516 SkString r0Name("r0");
517 SkString r1Name("r1");
518 SkString tName("t");
519 SkString p0; // 4a
520 SkString p1; // 1/a
521 SkString p2; // distance between centers
522 SkString p3; // start radius
523 SkString p4; // start radius squared
524 SkString p5; // difference in radii (r1 - r0)
525
526 builder->getUniformVariable(fParamUni).appendArrayAccess(0, &p0);
527 builder->getUniformVariable(fParamUni).appendArrayAccess(1, &p1);
528 builder->getUniformVariable(fParamUni).appendArrayAccess(2, &p2);
529 builder->getUniformVariable(fParamUni).appendArrayAccess(3, &p3);
530 builder->getUniformVariable(fParamUni).appendArrayAccess(4, &p4);
531 builder->getUniformVariable(fParamUni).appendArrayAccess(5, &p5);
532
533 // We interpolate the linear component in coords[1].
534 SkASSERT(coords[0].type() == coords[1].type());
535 const char* coords2D;
536 SkString bVar;
537 if (kVec3f_GrSLType == coords[0].type()) {
538 builder->fsCodeAppendf("\tvec3 interpolants = vec3(%s.xy, %s.x) / %s.z;\n",
539 coords[0].c_str(), coords[1].c_str(), coords[0].c_str());
540 coords2D = "interpolants.xy";
541 bVar = "interpolants.z";
542 } else {
543 coords2D = coords[0].c_str();
544 bVar.printf("%s.x", coords[1].c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000545 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000546
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000547 // output will default to transparent black (we simply won't write anything
548 // else to it if invalid, instead of discarding or returning prematurely)
549 builder->fsCodeAppendf("\t%s = vec4(0.0,0.0,0.0,0.0);\n", outputColor);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000550
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000551 // c = (x^2)+(y^2) - params[4]
552 builder->fsCodeAppendf("\tfloat %s = dot(%s, %s) - %s;\n",
553 cName.c_str(), coords2D, coords2D, p4.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000554
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000555 // Non-degenerate case (quadratic)
556 if (!fIsDegenerate) {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000557
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000558 // ac4 = params[0] * c
559 builder->fsCodeAppendf("\tfloat %s = %s * %s;\n", ac4Name.c_str(), p0.c_str(),
560 cName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000561
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000562 // d = b^2 - ac4
563 builder->fsCodeAppendf("\tfloat %s = %s * %s - %s;\n", dName.c_str(),
564 bVar.c_str(), bVar.c_str(), ac4Name.c_str());
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000565
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000566 // only proceed if discriminant is >= 0
567 builder->fsCodeAppendf("\tif (%s >= 0.0) {\n", dName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000568
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000569 // intermediate value we'll use to compute the roots
570 // q = -0.5 * (b +/- sqrt(d))
571 builder->fsCodeAppendf("\t\tfloat %s = -0.5 * (%s + (%s < 0.0 ? -1.0 : 1.0)"
572 " * sqrt(%s));\n", qName.c_str(), bVar.c_str(),
573 bVar.c_str(), dName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000574
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000575 // compute both roots
576 // r0 = q * params[1]
577 builder->fsCodeAppendf("\t\tfloat %s = %s * %s;\n", r0Name.c_str(),
578 qName.c_str(), p1.c_str());
579 // r1 = c / q
580 builder->fsCodeAppendf("\t\tfloat %s = %s / %s;\n", r1Name.c_str(),
581 cName.c_str(), qName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000582
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000583 // Note: If there are two roots that both generate radius(t) > 0, the
584 // Canvas spec says to choose the larger t.
rileya@google.comd7cc6512012-07-27 14:00:39 +0000585
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000586 // so we'll look at the larger one first:
587 builder->fsCodeAppendf("\t\tfloat %s = max(%s, %s);\n", tName.c_str(),
588 r0Name.c_str(), r1Name.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000589
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000590 // if r(t) > 0, then we're done; t will be our x coordinate
591 builder->fsCodeAppendf("\t\tif (%s * %s + %s > 0.0) {\n", tName.c_str(),
592 p5.c_str(), p3.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000593
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000594 builder->fsCodeAppend("\t\t");
595 this->emitColor(builder, tName.c_str(), key, outputColor, inputColor, samplers);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000596
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000597 // otherwise, if r(t) for the larger root was <= 0, try the other root
598 builder->fsCodeAppend("\t\t} else {\n");
599 builder->fsCodeAppendf("\t\t\t%s = min(%s, %s);\n", tName.c_str(),
600 r0Name.c_str(), r1Name.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000601
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000602 // if r(t) > 0 for the smaller root, then t will be our x coordinate
603 builder->fsCodeAppendf("\t\t\tif (%s * %s + %s > 0.0) {\n",
604 tName.c_str(), p5.c_str(), p3.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000605
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000606 builder->fsCodeAppend("\t\t\t");
607 this->emitColor(builder, tName.c_str(), key, outputColor, inputColor, samplers);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000608
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000609 // end if (r(t) > 0) for smaller root
610 builder->fsCodeAppend("\t\t\t}\n");
611 // end if (r(t) > 0), else, for larger root
612 builder->fsCodeAppend("\t\t}\n");
613 // end if (discriminant >= 0)
614 builder->fsCodeAppend("\t}\n");
615 } else {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000616
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000617 // linear case: t = -c/b
618 builder->fsCodeAppendf("\tfloat %s = -(%s / %s);\n", tName.c_str(),
619 cName.c_str(), bVar.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000620
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000621 // if r(t) > 0, then t will be the x coordinate
622 builder->fsCodeAppendf("\tif (%s * %s + %s > 0.0) {\n", tName.c_str(),
623 p5.c_str(), p3.c_str());
624 builder->fsCodeAppend("\t");
625 this->emitColor(builder, tName.c_str(), key, outputColor, inputColor, samplers);
626 builder->fsCodeAppend("\t}\n");
rileya@google.comd7cc6512012-07-27 14:00:39 +0000627 }
628}
629
bsalomon@google.comc7818882013-03-20 19:19:53 +0000630void GrGLConical2Gradient::setData(const GrGLUniformManager& uman,
631 const GrDrawEffect& drawEffect) {
632 INHERITED::setData(uman, drawEffect);
633 const GrConical2Gradient& data = drawEffect.castEffect<GrConical2Gradient>();
commit-bot@chromium.org96ae6882013-08-14 12:09:00 +0000634 SkASSERT(data.isDegenerate() == fIsDegenerate);
bsalomon@google.com81712882012-11-01 17:12:34 +0000635 SkScalar centerX1 = data.center();
636 SkScalar radius0 = data.radius();
637 SkScalar diffRadius = data.diffRadius();
rileya@google.comd7cc6512012-07-27 14:00:39 +0000638
639 if (fCachedCenter != centerX1 ||
640 fCachedRadius != radius0 ||
641 fCachedDiffRadius != diffRadius) {
642
bsalomon@google.com81712882012-11-01 17:12:34 +0000643 SkScalar a = SkScalarMul(centerX1, centerX1) - diffRadius * diffRadius;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000644
645 // When we're in the degenerate (linear) case, the second
646 // value will be INF but the program doesn't read it. (We
647 // use the same 6 uniforms even though we don't need them
648 // all in the linear case just to keep the code complexity
649 // down).
650 float values[6] = {
bsalomon@google.com81712882012-11-01 17:12:34 +0000651 SkScalarToFloat(a * 4),
652 1.f / (SkScalarToFloat(a)),
653 SkScalarToFloat(centerX1),
654 SkScalarToFloat(radius0),
655 SkScalarToFloat(SkScalarMul(radius0, radius0)),
656 SkScalarToFloat(diffRadius)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000657 };
658
commit-bot@chromium.orgd3baf202013-11-07 22:06:08 +0000659 uman.set1fv(fParamUni, 6, values);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000660 fCachedCenter = centerX1;
661 fCachedRadius = radius0;
662 fCachedDiffRadius = diffRadius;
663 }
664}
665
bsalomon@google.comc7818882013-03-20 19:19:53 +0000666GrGLEffect::EffectKey GrGLConical2Gradient::GenKey(const GrDrawEffect& drawEffect,
667 const GrGLCaps&) {
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000668 enum {
bsalomon@google.com82d12232013-09-09 15:36:26 +0000669 kIsDegenerate = 1 << kBaseKeyBitCnt,
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000670 };
671
bsalomon@google.com82d12232013-09-09 15:36:26 +0000672 EffectKey key = GenBaseGradientKey(drawEffect);
bsalomon@google.comc7818882013-03-20 19:19:53 +0000673 if (drawEffect.castEffect<GrConical2Gradient>().isDegenerate()) {
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000674 key |= kIsDegenerate;
675 }
676 return key;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000677}
rileya@google.comd7cc6512012-07-27 14:00:39 +0000678
679/////////////////////////////////////////////////////////////////////
680
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000681GrEffectRef* SkTwoPointConicalGradient::asNewEffect(GrContext* context, const SkPaint&) const {
bsalomon@google.com00835cc2013-01-14 17:07:22 +0000682 SkASSERT(NULL != context);
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000683 SkASSERT(fPtsToUnit.isIdentity());
684 // invert the localM, translate to center1, rotate so center2 is on x axis.
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000685 SkMatrix matrix;
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000686 if (!this->getLocalMatrix().invert(&matrix)) {
humper@google.com84831ac2013-01-14 22:09:54 +0000687 return NULL;
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000688 }
689 matrix.postTranslate(-fCenter1.fX, -fCenter1.fY);
690
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000691 SkPoint diff = fCenter2 - fCenter1;
692 SkScalar diffLen = diff.length();
693 if (0 != diffLen) {
694 SkScalar invDiffLen = SkScalarInvert(diffLen);
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000695 SkMatrix rot;
696 rot.setSinCos(-SkScalarMul(invDiffLen, diff.fY),
697 SkScalarMul(invDiffLen, diff.fX));
698 matrix.postConcat(rot);
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000699 }
700
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000701 return GrConical2Gradient::Create(context, *this, matrix, fTileMode);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000702}
703
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000704#else
705
bsalomon@google.com5d2cd202013-01-16 15:31:06 +0000706GrEffectRef* SkTwoPointConicalGradient::asNewEffect(GrContext*, const SkPaint&) const {
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000707 SkDEBUGFAIL("Should not call in GPU-less build");
bsalomon@google.come197cbf2013-01-14 16:46:26 +0000708 return NULL;
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000709}
710
twiz@google.coma5e65ec2012-08-02 15:15:16 +0000711#endif
robertphillips@google.com76f9e932013-01-15 20:17:47 +0000712
commit-bot@chromium.org0f10f7b2014-03-13 18:02:17 +0000713#ifndef SK_IGNORE_TO_STRING
robertphillips@google.com76f9e932013-01-15 20:17:47 +0000714void SkTwoPointConicalGradient::toString(SkString* str) const {
715 str->append("SkTwoPointConicalGradient: (");
716
717 str->append("center1: (");
718 str->appendScalar(fCenter1.fX);
719 str->append(", ");
720 str->appendScalar(fCenter1.fY);
721 str->append(") radius1: ");
722 str->appendScalar(fRadius1);
723 str->append(" ");
724
725 str->append("center2: (");
726 str->appendScalar(fCenter2.fX);
727 str->append(", ");
728 str->appendScalar(fCenter2.fY);
729 str->append(") radius2: ");
730 str->appendScalar(fRadius2);
731 str->append(" ");
732
733 this->INHERITED::toString(str);
734
735 str->append(")");
736}
737#endif