blob: 94eb72f5a08c730ea20d77fe75bc6f32150b2f21 [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
239 SkScalar dstX = SkIntToScalar(x);
240 SkScalar dstY = SkIntToScalar(y);
241 for (; count > 0; --count) {
242 SkPoint srcPt;
243 dstProc(fDstToIndex, dstX, dstY, &srcPt);
244 dstX += SK_Scalar1;
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000245
rileya@google.com589708b2012-07-26 20:04:23 +0000246 fRec.setup(srcPt.fX, srcPt.fY, 0, 0);
reed@google.com60040292013-02-04 18:21:23 +0000247 (*shadeProc)(&fRec, dstC, cache, toggle, 1);
reed@google.com60040292013-02-04 18:21:23 +0000248 toggle = next_dither_toggle(toggle);
rileya@google.com589708b2012-07-26 20:04:23 +0000249 }
250 }
251}
252
253bool SkTwoPointConicalGradient::setContext(const SkBitmap& device,
254 const SkPaint& paint,
255 const SkMatrix& matrix) {
256 if (!this->INHERITED::setContext(device, paint, matrix)) {
257 return false;
258 }
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000259
rileya@google.com589708b2012-07-26 20:04:23 +0000260 // we don't have a span16 proc
261 fFlags &= ~kHasSpan16_Flag;
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000262
rileya@google.com589708b2012-07-26 20:04:23 +0000263 // in general, we might discard based on computed-radius, so clear
264 // this flag (todo: sometimes we can detect that we never discard...)
265 fFlags &= ~kOpaqueAlpha_Flag;
266
267 return true;
268}
269
270SkShader::BitmapType SkTwoPointConicalGradient::asABitmap(
271 SkBitmap* bitmap, SkMatrix* matrix, SkShader::TileMode* xy) const {
272 SkPoint diff = fCenter2 - fCenter1;
rileya@google.com589708b2012-07-26 20:04:23 +0000273 SkScalar diffLen = 0;
274
275 if (bitmap) {
rileya@google.com1c6d64b2012-07-27 15:49:05 +0000276 this->getGradientTableBitmap(bitmap);
rileya@google.com589708b2012-07-26 20:04:23 +0000277 }
278 if (matrix) {
279 diffLen = diff.length();
280 }
281 if (matrix) {
282 if (diffLen) {
283 SkScalar invDiffLen = SkScalarInvert(diffLen);
284 // rotate to align circle centers with the x-axis
285 matrix->setSinCos(-SkScalarMul(invDiffLen, diff.fY),
286 SkScalarMul(invDiffLen, diff.fX));
287 } else {
288 matrix->reset();
289 }
290 matrix->preTranslate(-fCenter1.fX, -fCenter1.fY);
291 }
292 if (xy) {
293 xy[0] = fTileMode;
294 xy[1] = kClamp_TileMode;
295 }
296 return kTwoPointConical_BitmapType;
297}
298
299SkShader::GradientType SkTwoPointConicalGradient::asAGradient(
300 GradientInfo* info) const {
301 if (info) {
302 commonAsAGradient(info);
303 info->fPoint[0] = fCenter1;
304 info->fPoint[1] = fCenter2;
305 info->fRadius[0] = fRadius1;
306 info->fRadius[1] = fRadius2;
307 }
308 return kConical_GradientType;
309}
310
rileya@google.com589708b2012-07-26 20:04:23 +0000311SkTwoPointConicalGradient::SkTwoPointConicalGradient(
312 SkFlattenableReadBuffer& buffer)
313 : INHERITED(buffer),
314 fCenter1(buffer.readPoint()),
315 fCenter2(buffer.readPoint()),
316 fRadius1(buffer.readScalar()),
317 fRadius2(buffer.readScalar()) {
318 this->init();
319};
320
321void SkTwoPointConicalGradient::flatten(
322 SkFlattenableWriteBuffer& buffer) const {
323 this->INHERITED::flatten(buffer);
324 buffer.writePoint(fCenter1);
325 buffer.writePoint(fCenter2);
326 buffer.writeScalar(fRadius1);
327 buffer.writeScalar(fRadius2);
328}
329
rileya@google.comd7cc6512012-07-27 14:00:39 +0000330/////////////////////////////////////////////////////////////////////
331
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000332#if SK_SUPPORT_GPU
333
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000334#include "GrTBackendEffectFactory.h"
335
rileya@google.comd7cc6512012-07-27 14:00:39 +0000336// For brevity
337typedef GrGLUniformManager::UniformHandle UniformHandle;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000338
bsalomon@google.com0707c292012-10-25 21:45:42 +0000339class GrGLConical2Gradient : public GrGLGradientEffect {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000340public:
341
bsalomon@google.comc7818882013-03-20 19:19:53 +0000342 GrGLConical2Gradient(const GrBackendEffectFactory& factory, const GrDrawEffect&);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000343 virtual ~GrGLConical2Gradient() { }
344
bsalomon@google.comf78df332012-10-29 12:43:38 +0000345 virtual void emitCode(GrGLShaderBuilder*,
bsalomon@google.comc7818882013-03-20 19:19:53 +0000346 const GrDrawEffect&,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000347 EffectKey,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000348 const char* outputColor,
349 const char* inputColor,
bsalomon@google.com77af6802013-10-02 13:04:56 +0000350 const TransformedCoordsArray&,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000351 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
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000358 UniformHandle fParamUni;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000359
360 const char* fVSVaryingName;
361 const char* fFSVaryingName;
362
363 bool fIsDegenerate;
364
365 // @{
366 /// Values last uploaded as uniforms
367
bsalomon@google.com81712882012-11-01 17:12:34 +0000368 SkScalar fCachedCenter;
369 SkScalar fCachedRadius;
370 SkScalar fCachedDiffRadius;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000371
372 // @}
373
374private:
375
bsalomon@google.com0707c292012-10-25 21:45:42 +0000376 typedef GrGLGradientEffect INHERITED;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000377
378};
379
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000380/////////////////////////////////////////////////////////////////////
381
382class GrConical2Gradient : public GrGradientEffect {
383public:
384
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000385 static GrEffectRef* Create(GrContext* ctx,
386 const SkTwoPointConicalGradient& shader,
387 const SkMatrix& matrix,
388 SkShader::TileMode tm) {
bsalomon@google.com6340a412013-01-22 19:55:59 +0000389 AutoEffectUnref effect(SkNEW_ARGS(GrConical2Gradient, (ctx, shader, matrix, tm)));
bsalomon@google.coma1ebbe42013-01-16 15:51:47 +0000390 return CreateEffectRef(effect);
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000391 }
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000392
393 virtual ~GrConical2Gradient() { }
394
395 static const char* Name() { return "Two-Point Conical Gradient"; }
bsalomon@google.com396e61f2012-10-25 19:00:29 +0000396 virtual const GrBackendEffectFactory& getFactory() const SK_OVERRIDE {
397 return GrTBackendEffectFactory<GrConical2Gradient>::getInstance();
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000398 }
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000399
400 // The radial gradient parameters can collapse to a linear (instead of quadratic) equation.
401 bool isDegenerate() const { return SkScalarAbs(fDiffRadius) == SkScalarAbs(fCenterX1); }
bsalomon@google.com81712882012-11-01 17:12:34 +0000402 SkScalar center() const { return fCenterX1; }
403 SkScalar diffRadius() const { return fDiffRadius; }
404 SkScalar radius() const { return fRadius0; }
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000405
bsalomon@google.com422e81a2012-10-25 14:11:03 +0000406 typedef GrGLConical2Gradient GLEffect;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000407
408private:
bsalomon@google.com8a252f72013-01-22 20:35:13 +0000409 virtual bool onIsEqual(const GrEffect& sBase) const SK_OVERRIDE {
bsalomon@google.com6340a412013-01-22 19:55:59 +0000410 const GrConical2Gradient& s = CastEffect<GrConical2Gradient>(sBase);
bsalomon@google.com68b58c92013-01-17 16:50:08 +0000411 return (INHERITED::onIsEqual(sBase) &&
412 this->fCenterX1 == s.fCenterX1 &&
413 this->fRadius0 == s.fRadius0 &&
414 this->fDiffRadius == s.fDiffRadius);
415 }
416
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000417 GrConical2Gradient(GrContext* ctx,
418 const SkTwoPointConicalGradient& shader,
419 const SkMatrix& matrix,
420 SkShader::TileMode tm)
421 : INHERITED(ctx, shader, matrix, tm)
422 , fCenterX1(shader.getCenterX1())
423 , fRadius0(shader.getStartRadius())
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000424 , fDiffRadius(shader.getDiffRadius()) {
425 // We pass the linear part of the quadratic as a varying.
426 // float b = -2.0 * (fCenterX1 * x + fRadius0 * fDiffRadius * z)
427 fBTransform = this->getCoordTransform();
428 SkMatrix& bMatrix = *fBTransform.accessMatrix();
429 SkScalar r0dr = SkScalarMul(fRadius0, fDiffRadius);
430 bMatrix[SkMatrix::kMScaleX] = -2 * (SkScalarMul(fCenterX1, bMatrix[SkMatrix::kMScaleX]) +
431 SkScalarMul(r0dr, bMatrix[SkMatrix::kMPersp0]));
432 bMatrix[SkMatrix::kMSkewX] = -2 * (SkScalarMul(fCenterX1, bMatrix[SkMatrix::kMSkewX]) +
433 SkScalarMul(r0dr, bMatrix[SkMatrix::kMPersp1]));
434 bMatrix[SkMatrix::kMTransX] = -2 * (SkScalarMul(fCenterX1, bMatrix[SkMatrix::kMTransX]) +
435 SkScalarMul(r0dr, bMatrix[SkMatrix::kMPersp2]));
436 this->addCoordTransform(&fBTransform);
437 }
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000438
bsalomon@google.comf271cc72012-10-24 19:35:13 +0000439 GR_DECLARE_EFFECT_TEST;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000440
441 // @{
442 // Cache of values - these can change arbitrarily, EXCEPT
443 // we shouldn't change between degenerate and non-degenerate?!
444
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000445 GrCoordTransform fBTransform;
446 SkScalar fCenterX1;
447 SkScalar fRadius0;
448 SkScalar fDiffRadius;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000449
450 // @}
451
452 typedef GrGradientEffect INHERITED;
453};
454
bsalomon@google.comf271cc72012-10-24 19:35:13 +0000455GR_DEFINE_EFFECT_TEST(GrConical2Gradient);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000456
commit-bot@chromium.orge0e7cfe2013-09-09 20:09:12 +0000457GrEffectRef* GrConical2Gradient::TestCreate(SkRandom* random,
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000458 GrContext* context,
bsalomon@google.comc26d94f2013-03-25 18:19:00 +0000459 const GrDrawTargetCaps&,
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000460 GrTexture**) {
bsalomon@google.comd4726202012-08-03 14:34:46 +0000461 SkPoint center1 = {random->nextUScalar1(), random->nextUScalar1()};
462 SkScalar radius1 = random->nextUScalar1();
463 SkPoint center2;
464 SkScalar radius2;
465 do {
bsalomon@google.comfb883bf2012-12-11 15:32:04 +0000466 center2.set(random->nextUScalar1(), random->nextUScalar1());
bsalomon@google.comd4726202012-08-03 14:34:46 +0000467 radius2 = random->nextUScalar1 ();
468 // If the circles are identical the factory will give us an empty shader.
469 } while (radius1 == radius2 && center1 == center2);
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000470
bsalomon@google.comd4726202012-08-03 14:34:46 +0000471 SkColor colors[kMaxRandomGradientColors];
472 SkScalar stopsArray[kMaxRandomGradientColors];
473 SkScalar* stops = stopsArray;
474 SkShader::TileMode tm;
475 int colorCount = RandomGradientParams(random, colors, &stops, &tm);
476 SkAutoTUnref<SkShader> shader(SkGradientShader::CreateTwoPointConical(center1, radius1,
477 center2, radius2,
478 colors, stops, colorCount,
479 tm));
bsalomon@google.come197cbf2013-01-14 16:46:26 +0000480 SkPaint paint;
481 return shader->asNewEffect(context, paint);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000482}
483
484
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000485/////////////////////////////////////////////////////////////////////
486
bsalomon@google.com6340a412013-01-22 19:55:59 +0000487GrGLConical2Gradient::GrGLConical2Gradient(const GrBackendEffectFactory& factory,
bsalomon@google.comc7818882013-03-20 19:19:53 +0000488 const GrDrawEffect& drawEffect)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000489 : INHERITED(factory)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000490 , fVSVaryingName(NULL)
491 , fFSVaryingName(NULL)
bsalomon@google.com81712882012-11-01 17:12:34 +0000492 , fCachedCenter(SK_ScalarMax)
493 , fCachedRadius(-SK_ScalarMax)
494 , fCachedDiffRadius(-SK_ScalarMax) {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000495
bsalomon@google.comc7818882013-03-20 19:19:53 +0000496 const GrConical2Gradient& data = drawEffect.castEffect<GrConical2Gradient>();
rileya@google.comd7cc6512012-07-27 14:00:39 +0000497 fIsDegenerate = data.isDegenerate();
498}
499
bsalomon@google.comf78df332012-10-29 12:43:38 +0000500void GrGLConical2Gradient::emitCode(GrGLShaderBuilder* builder,
bsalomon@google.comc7818882013-03-20 19:19:53 +0000501 const GrDrawEffect&,
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000502 EffectKey key,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000503 const char* outputColor,
504 const char* inputColor,
bsalomon@google.com77af6802013-10-02 13:04:56 +0000505 const TransformedCoordsArray& coords,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000506 const TextureSamplerArray& samplers) {
bsalomon@google.com82d12232013-09-09 15:36:26 +0000507 this->emitUniforms(builder, key);
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000508 fParamUni = builder->addUniformArray(GrGLShaderBuilder::kFragment_Visibility,
509 kFloat_GrSLType, "Conical2FSParams", 6);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000510
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000511 SkString cName("c");
512 SkString ac4Name("ac4");
513 SkString dName("d");
514 SkString qName("q");
515 SkString r0Name("r0");
516 SkString r1Name("r1");
517 SkString tName("t");
518 SkString p0; // 4a
519 SkString p1; // 1/a
520 SkString p2; // distance between centers
521 SkString p3; // start radius
522 SkString p4; // start radius squared
523 SkString p5; // difference in radii (r1 - r0)
524
525 builder->getUniformVariable(fParamUni).appendArrayAccess(0, &p0);
526 builder->getUniformVariable(fParamUni).appendArrayAccess(1, &p1);
527 builder->getUniformVariable(fParamUni).appendArrayAccess(2, &p2);
528 builder->getUniformVariable(fParamUni).appendArrayAccess(3, &p3);
529 builder->getUniformVariable(fParamUni).appendArrayAccess(4, &p4);
530 builder->getUniformVariable(fParamUni).appendArrayAccess(5, &p5);
531
532 // We interpolate the linear component in coords[1].
533 SkASSERT(coords[0].type() == coords[1].type());
534 const char* coords2D;
535 SkString bVar;
536 if (kVec3f_GrSLType == coords[0].type()) {
537 builder->fsCodeAppendf("\tvec3 interpolants = vec3(%s.xy, %s.x) / %s.z;\n",
538 coords[0].c_str(), coords[1].c_str(), coords[0].c_str());
539 coords2D = "interpolants.xy";
540 bVar = "interpolants.z";
541 } else {
542 coords2D = coords[0].c_str();
543 bVar.printf("%s.x", coords[1].c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000544 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000545
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000546 // output will default to transparent black (we simply won't write anything
547 // else to it if invalid, instead of discarding or returning prematurely)
548 builder->fsCodeAppendf("\t%s = vec4(0.0,0.0,0.0,0.0);\n", outputColor);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000549
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000550 // c = (x^2)+(y^2) - params[4]
551 builder->fsCodeAppendf("\tfloat %s = dot(%s, %s) - %s;\n",
552 cName.c_str(), coords2D, coords2D, p4.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000553
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000554 // Non-degenerate case (quadratic)
555 if (!fIsDegenerate) {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000556
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000557 // ac4 = params[0] * c
558 builder->fsCodeAppendf("\tfloat %s = %s * %s;\n", ac4Name.c_str(), p0.c_str(),
559 cName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000560
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000561 // d = b^2 - ac4
562 builder->fsCodeAppendf("\tfloat %s = %s * %s - %s;\n", dName.c_str(),
563 bVar.c_str(), bVar.c_str(), ac4Name.c_str());
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000564
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000565 // only proceed if discriminant is >= 0
566 builder->fsCodeAppendf("\tif (%s >= 0.0) {\n", dName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000567
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000568 // intermediate value we'll use to compute the roots
569 // q = -0.5 * (b +/- sqrt(d))
570 builder->fsCodeAppendf("\t\tfloat %s = -0.5 * (%s + (%s < 0.0 ? -1.0 : 1.0)"
571 " * sqrt(%s));\n", qName.c_str(), bVar.c_str(),
572 bVar.c_str(), dName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000573
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000574 // compute both roots
575 // r0 = q * params[1]
576 builder->fsCodeAppendf("\t\tfloat %s = %s * %s;\n", r0Name.c_str(),
577 qName.c_str(), p1.c_str());
578 // r1 = c / q
579 builder->fsCodeAppendf("\t\tfloat %s = %s / %s;\n", r1Name.c_str(),
580 cName.c_str(), qName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000581
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000582 // Note: If there are two roots that both generate radius(t) > 0, the
583 // Canvas spec says to choose the larger t.
rileya@google.comd7cc6512012-07-27 14:00:39 +0000584
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000585 // so we'll look at the larger one first:
586 builder->fsCodeAppendf("\t\tfloat %s = max(%s, %s);\n", tName.c_str(),
587 r0Name.c_str(), r1Name.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000588
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000589 // if r(t) > 0, then we're done; t will be our x coordinate
590 builder->fsCodeAppendf("\t\tif (%s * %s + %s > 0.0) {\n", tName.c_str(),
591 p5.c_str(), p3.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000592
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000593 builder->fsCodeAppend("\t\t");
594 this->emitColor(builder, tName.c_str(), key, outputColor, inputColor, samplers);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000595
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000596 // otherwise, if r(t) for the larger root was <= 0, try the other root
597 builder->fsCodeAppend("\t\t} else {\n");
598 builder->fsCodeAppendf("\t\t\t%s = min(%s, %s);\n", tName.c_str(),
599 r0Name.c_str(), r1Name.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000600
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000601 // if r(t) > 0 for the smaller root, then t will be our x coordinate
602 builder->fsCodeAppendf("\t\t\tif (%s * %s + %s > 0.0) {\n",
603 tName.c_str(), p5.c_str(), p3.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000604
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000605 builder->fsCodeAppend("\t\t\t");
606 this->emitColor(builder, tName.c_str(), key, outputColor, inputColor, samplers);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000607
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000608 // end if (r(t) > 0) for smaller root
609 builder->fsCodeAppend("\t\t\t}\n");
610 // end if (r(t) > 0), else, for larger root
611 builder->fsCodeAppend("\t\t}\n");
612 // end if (discriminant >= 0)
613 builder->fsCodeAppend("\t}\n");
614 } else {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000615
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000616 // linear case: t = -c/b
617 builder->fsCodeAppendf("\tfloat %s = -(%s / %s);\n", tName.c_str(),
618 cName.c_str(), bVar.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000619
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000620 // if r(t) > 0, then t will be the x coordinate
621 builder->fsCodeAppendf("\tif (%s * %s + %s > 0.0) {\n", tName.c_str(),
622 p5.c_str(), p3.c_str());
623 builder->fsCodeAppend("\t");
624 this->emitColor(builder, tName.c_str(), key, outputColor, inputColor, samplers);
625 builder->fsCodeAppend("\t}\n");
rileya@google.comd7cc6512012-07-27 14:00:39 +0000626 }
627}
628
bsalomon@google.comc7818882013-03-20 19:19:53 +0000629void GrGLConical2Gradient::setData(const GrGLUniformManager& uman,
630 const GrDrawEffect& drawEffect) {
631 INHERITED::setData(uman, drawEffect);
632 const GrConical2Gradient& data = drawEffect.castEffect<GrConical2Gradient>();
commit-bot@chromium.org96ae6882013-08-14 12:09:00 +0000633 SkASSERT(data.isDegenerate() == fIsDegenerate);
bsalomon@google.com81712882012-11-01 17:12:34 +0000634 SkScalar centerX1 = data.center();
635 SkScalar radius0 = data.radius();
636 SkScalar diffRadius = data.diffRadius();
rileya@google.comd7cc6512012-07-27 14:00:39 +0000637
638 if (fCachedCenter != centerX1 ||
639 fCachedRadius != radius0 ||
640 fCachedDiffRadius != diffRadius) {
641
bsalomon@google.com81712882012-11-01 17:12:34 +0000642 SkScalar a = SkScalarMul(centerX1, centerX1) - diffRadius * diffRadius;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000643
644 // When we're in the degenerate (linear) case, the second
645 // value will be INF but the program doesn't read it. (We
646 // use the same 6 uniforms even though we don't need them
647 // all in the linear case just to keep the code complexity
648 // down).
649 float values[6] = {
bsalomon@google.com81712882012-11-01 17:12:34 +0000650 SkScalarToFloat(a * 4),
651 1.f / (SkScalarToFloat(a)),
652 SkScalarToFloat(centerX1),
653 SkScalarToFloat(radius0),
654 SkScalarToFloat(SkScalarMul(radius0, radius0)),
655 SkScalarToFloat(diffRadius)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000656 };
657
commit-bot@chromium.org5fd7d5c2013-10-04 01:20:09 +0000658 uman.set1fv(fParamUni, 0, 6, values);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000659 fCachedCenter = centerX1;
660 fCachedRadius = radius0;
661 fCachedDiffRadius = diffRadius;
662 }
663}
664
bsalomon@google.comc7818882013-03-20 19:19:53 +0000665GrGLEffect::EffectKey GrGLConical2Gradient::GenKey(const GrDrawEffect& drawEffect,
666 const GrGLCaps&) {
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000667 enum {
bsalomon@google.com82d12232013-09-09 15:36:26 +0000668 kIsDegenerate = 1 << kBaseKeyBitCnt,
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000669 };
670
bsalomon@google.com82d12232013-09-09 15:36:26 +0000671 EffectKey key = GenBaseGradientKey(drawEffect);
bsalomon@google.comc7818882013-03-20 19:19:53 +0000672 if (drawEffect.castEffect<GrConical2Gradient>().isDegenerate()) {
bsalomon@google.comd8b5fac2012-11-01 17:02:46 +0000673 key |= kIsDegenerate;
674 }
675 return key;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000676}
rileya@google.comd7cc6512012-07-27 14:00:39 +0000677
678/////////////////////////////////////////////////////////////////////
679
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000680GrEffectRef* SkTwoPointConicalGradient::asNewEffect(GrContext* context, const SkPaint&) const {
bsalomon@google.com00835cc2013-01-14 17:07:22 +0000681 SkASSERT(NULL != context);
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000682 SkASSERT(fPtsToUnit.isIdentity());
683 // invert the localM, translate to center1, rotate so center2 is on x axis.
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000684 SkMatrix matrix;
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000685 if (!this->getLocalMatrix().invert(&matrix)) {
humper@google.com84831ac2013-01-14 22:09:54 +0000686 return NULL;
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000687 }
688 matrix.postTranslate(-fCenter1.fX, -fCenter1.fY);
689
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000690 SkPoint diff = fCenter2 - fCenter1;
691 SkScalar diffLen = diff.length();
692 if (0 != diffLen) {
693 SkScalar invDiffLen = SkScalarInvert(diffLen);
bsalomon@google.comf94b3a42012-10-31 18:09:01 +0000694 SkMatrix rot;
695 rot.setSinCos(-SkScalarMul(invDiffLen, diff.fY),
696 SkScalarMul(invDiffLen, diff.fX));
697 matrix.postConcat(rot);
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000698 }
699
bsalomon@google.com0ac6af42013-01-16 15:16:18 +0000700 return GrConical2Gradient::Create(context, *this, matrix, fTileMode);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000701}
702
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000703#else
704
bsalomon@google.com5d2cd202013-01-16 15:31:06 +0000705GrEffectRef* SkTwoPointConicalGradient::asNewEffect(GrContext*, const SkPaint&) const {
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000706 SkDEBUGFAIL("Should not call in GPU-less build");
bsalomon@google.come197cbf2013-01-14 16:46:26 +0000707 return NULL;
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000708}
709
twiz@google.coma5e65ec2012-08-02 15:15:16 +0000710#endif
robertphillips@google.com76f9e932013-01-15 20:17:47 +0000711
712#ifdef SK_DEVELOPER
713void SkTwoPointConicalGradient::toString(SkString* str) const {
714 str->append("SkTwoPointConicalGradient: (");
715
716 str->append("center1: (");
717 str->appendScalar(fCenter1.fX);
718 str->append(", ");
719 str->appendScalar(fCenter1.fY);
720 str->append(") radius1: ");
721 str->appendScalar(fRadius1);
722 str->append(" ");
723
724 str->append("center2: (");
725 str->appendScalar(fCenter2.fX);
726 str->append(", ");
727 str->appendScalar(fCenter2.fY);
728 str->append(") radius2: ");
729 str->appendScalar(fRadius2);
730 str->append(" ");
731
732 this->INHERITED::toString(str);
733
734 str->append(")");
735}
736#endif