blob: acd0eeedfb4b4de8927bb647948c241d4b2c8d57 [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
116typedef void (*TwoPointRadialProc)(TwoPtRadial* rec, SkPMColor* dstC,
117 const SkPMColor* cache, int count);
118
119static void twopoint_clamp(TwoPtRadial* rec, SkPMColor* SK_RESTRICT dstC,
120 const SkPMColor* SK_RESTRICT cache, int count) {
121 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);
128 *dstC++ = cache[index >> SkGradientShaderBase::kCache32Shift];
129 }
130 }
131}
132
133static void twopoint_repeat(TwoPtRadial* rec, SkPMColor* SK_RESTRICT dstC,
134 const SkPMColor* SK_RESTRICT cache, int count) {
135 for (; count > 0; --count) {
136 SkFixed t = rec->nextT();
137 if (TwoPtRadial::DontDrawT(t)) {
138 *dstC++ = 0;
139 } else {
140 SkFixed index = repeat_tileproc(t);
141 SkASSERT(index <= 0xFFFF);
142 *dstC++ = cache[index >> SkGradientShaderBase::kCache32Shift];
143 }
144 }
145}
146
147static void twopoint_mirror(TwoPtRadial* rec, SkPMColor* SK_RESTRICT dstC,
148 const SkPMColor* SK_RESTRICT cache, int count) {
149 for (; count > 0; --count) {
150 SkFixed t = rec->nextT();
151 if (TwoPtRadial::DontDrawT(t)) {
152 *dstC++ = 0;
153 } else {
154 SkFixed index = mirror_tileproc(t);
155 SkASSERT(index <= 0xFFFF);
156 *dstC++ = cache[index >> SkGradientShaderBase::kCache32Shift];
157 }
158 }
159}
160
161void SkTwoPointConicalGradient::init() {
162 fRec.init(fCenter1, fRadius1, fCenter2, fRadius2);
163 fPtsToUnit.reset();
164}
165
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000166/////////////////////////////////////////////////////////////////////
167
rileya@google.com589708b2012-07-26 20:04:23 +0000168SkTwoPointConicalGradient::SkTwoPointConicalGradient(
169 const SkPoint& start, SkScalar startRadius,
170 const SkPoint& end, SkScalar endRadius,
171 const SkColor colors[], const SkScalar pos[],
172 int colorCount, SkShader::TileMode mode,
173 SkUnitMapper* mapper)
174 : SkGradientShaderBase(colors, pos, colorCount, mode, mapper),
175 fCenter1(start),
176 fCenter2(end),
177 fRadius1(startRadius),
178 fRadius2(endRadius) {
179 // this is degenerate, and should be caught by our caller
180 SkASSERT(fCenter1 != fCenter2 || fRadius1 != fRadius2);
181 this->init();
182}
183
184void SkTwoPointConicalGradient::shadeSpan(int x, int y, SkPMColor* dstCParam,
185 int count) {
186 SkASSERT(count > 0);
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000187
rileya@google.com589708b2012-07-26 20:04:23 +0000188 SkPMColor* SK_RESTRICT dstC = dstCParam;
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000189
rileya@google.com589708b2012-07-26 20:04:23 +0000190 SkMatrix::MapXYProc dstProc = fDstToIndexProc;
bsalomon@google.com100abf42012-09-05 17:40:04 +0000191
rileya@google.com589708b2012-07-26 20:04:23 +0000192 const SkPMColor* SK_RESTRICT cache = this->getCache32();
193
194 TwoPointRadialProc shadeProc = twopoint_repeat;
195 if (SkShader::kClamp_TileMode == fTileMode) {
196 shadeProc = twopoint_clamp;
197 } else if (SkShader::kMirror_TileMode == fTileMode) {
198 shadeProc = twopoint_mirror;
199 } else {
200 SkASSERT(SkShader::kRepeat_TileMode == fTileMode);
201 }
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000202
rileya@google.com589708b2012-07-26 20:04:23 +0000203 if (fDstToIndexClass != kPerspective_MatrixClass) {
204 SkPoint srcPt;
205 dstProc(fDstToIndex, SkIntToScalar(x) + SK_ScalarHalf,
206 SkIntToScalar(y) + SK_ScalarHalf, &srcPt);
207 SkScalar dx, fx = srcPt.fX;
208 SkScalar dy, fy = srcPt.fY;
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000209
rileya@google.com589708b2012-07-26 20:04:23 +0000210 if (fDstToIndexClass == kFixedStepInX_MatrixClass) {
211 SkFixed fixedX, fixedY;
212 (void)fDstToIndex.fixedStepInX(SkIntToScalar(y), &fixedX, &fixedY);
213 dx = SkFixedToScalar(fixedX);
214 dy = SkFixedToScalar(fixedY);
215 } else {
216 SkASSERT(fDstToIndexClass == kLinear_MatrixClass);
217 dx = fDstToIndex.getScaleX();
218 dy = fDstToIndex.getSkewY();
219 }
220
221 fRec.setup(fx, fy, dx, dy);
222 (*shadeProc)(&fRec, dstC, cache, count);
223 } else { // perspective case
224 SkScalar dstX = SkIntToScalar(x);
225 SkScalar dstY = SkIntToScalar(y);
226 for (; count > 0; --count) {
227 SkPoint srcPt;
228 dstProc(fDstToIndex, dstX, dstY, &srcPt);
229 dstX += SK_Scalar1;
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000230
rileya@google.com589708b2012-07-26 20:04:23 +0000231 fRec.setup(srcPt.fX, srcPt.fY, 0, 0);
232 (*shadeProc)(&fRec, dstC, cache, 1);
233 }
234 }
235}
236
237bool SkTwoPointConicalGradient::setContext(const SkBitmap& device,
238 const SkPaint& paint,
239 const SkMatrix& matrix) {
240 if (!this->INHERITED::setContext(device, paint, matrix)) {
241 return false;
242 }
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000243
rileya@google.com589708b2012-07-26 20:04:23 +0000244 // we don't have a span16 proc
245 fFlags &= ~kHasSpan16_Flag;
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000246
rileya@google.com589708b2012-07-26 20:04:23 +0000247 // in general, we might discard based on computed-radius, so clear
248 // this flag (todo: sometimes we can detect that we never discard...)
249 fFlags &= ~kOpaqueAlpha_Flag;
250
251 return true;
252}
253
254SkShader::BitmapType SkTwoPointConicalGradient::asABitmap(
255 SkBitmap* bitmap, SkMatrix* matrix, SkShader::TileMode* xy) const {
256 SkPoint diff = fCenter2 - fCenter1;
rileya@google.com589708b2012-07-26 20:04:23 +0000257 SkScalar diffLen = 0;
258
259 if (bitmap) {
rileya@google.com1c6d64b2012-07-27 15:49:05 +0000260 this->getGradientTableBitmap(bitmap);
rileya@google.com589708b2012-07-26 20:04:23 +0000261 }
262 if (matrix) {
263 diffLen = diff.length();
264 }
265 if (matrix) {
266 if (diffLen) {
267 SkScalar invDiffLen = SkScalarInvert(diffLen);
268 // rotate to align circle centers with the x-axis
269 matrix->setSinCos(-SkScalarMul(invDiffLen, diff.fY),
270 SkScalarMul(invDiffLen, diff.fX));
271 } else {
272 matrix->reset();
273 }
274 matrix->preTranslate(-fCenter1.fX, -fCenter1.fY);
275 }
276 if (xy) {
277 xy[0] = fTileMode;
278 xy[1] = kClamp_TileMode;
279 }
280 return kTwoPointConical_BitmapType;
281}
282
283SkShader::GradientType SkTwoPointConicalGradient::asAGradient(
284 GradientInfo* info) const {
285 if (info) {
286 commonAsAGradient(info);
287 info->fPoint[0] = fCenter1;
288 info->fPoint[1] = fCenter2;
289 info->fRadius[0] = fRadius1;
290 info->fRadius[1] = fRadius2;
291 }
292 return kConical_GradientType;
293}
294
rileya@google.com589708b2012-07-26 20:04:23 +0000295SkTwoPointConicalGradient::SkTwoPointConicalGradient(
296 SkFlattenableReadBuffer& buffer)
297 : INHERITED(buffer),
298 fCenter1(buffer.readPoint()),
299 fCenter2(buffer.readPoint()),
300 fRadius1(buffer.readScalar()),
301 fRadius2(buffer.readScalar()) {
302 this->init();
303};
304
305void SkTwoPointConicalGradient::flatten(
306 SkFlattenableWriteBuffer& buffer) const {
307 this->INHERITED::flatten(buffer);
308 buffer.writePoint(fCenter1);
309 buffer.writePoint(fCenter2);
310 buffer.writeScalar(fRadius1);
311 buffer.writeScalar(fRadius2);
312}
313
rileya@google.comd7cc6512012-07-27 14:00:39 +0000314/////////////////////////////////////////////////////////////////////
315
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000316#if SK_SUPPORT_GPU
317
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000318#include "GrTBackendEffectFactory.h"
319
rileya@google.comd7cc6512012-07-27 14:00:39 +0000320// For brevity
321typedef GrGLUniformManager::UniformHandle UniformHandle;
322static const UniformHandle kInvalidUniformHandle = GrGLUniformManager::kInvalidUniformHandle;
323
bsalomon@google.com0707c292012-10-25 21:45:42 +0000324class GrGLConical2Gradient : public GrGLGradientEffect {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000325public:
326
bsalomon@google.com396e61f2012-10-25 19:00:29 +0000327 GrGLConical2Gradient(const GrBackendEffectFactory& factory,
bsalomon@google.coma469c282012-10-24 18:28:34 +0000328 const GrEffect&);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000329 virtual ~GrGLConical2Gradient() { }
330
bsalomon@google.comf78df332012-10-29 12:43:38 +0000331 virtual void emitCode(GrGLShaderBuilder*,
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000332 const GrEffectStage&,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000333 EffectKey,
334 const char* vertexCoords,
335 const char* outputColor,
336 const char* inputColor,
337 const TextureSamplerArray&) SK_OVERRIDE;
bsalomon@google.com28a15fb2012-10-26 17:53:18 +0000338 virtual void setData(const GrGLUniformManager&, const GrEffectStage&) SK_OVERRIDE;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000339
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000340 static EffectKey GenKey(const GrEffectStage&, const GrGLCaps& caps);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000341
342protected:
343
344 UniformHandle fVSParamUni;
345 UniformHandle fFSParamUni;
346
347 const char* fVSVaryingName;
348 const char* fFSVaryingName;
349
350 bool fIsDegenerate;
351
352 // @{
353 /// Values last uploaded as uniforms
354
355 GrScalar fCachedCenter;
356 GrScalar fCachedRadius;
357 GrScalar fCachedDiffRadius;
358
359 // @}
360
361private:
362
bsalomon@google.com0707c292012-10-25 21:45:42 +0000363 typedef GrGLGradientEffect INHERITED;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000364
365};
366
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000367/////////////////////////////////////////////////////////////////////
368
369class GrConical2Gradient : public GrGradientEffect {
370public:
371
bsalomon@google.com1ce49fc2012-09-18 14:14:49 +0000372 GrConical2Gradient(GrContext* ctx,
373 const SkTwoPointConicalGradient& shader,
374 SkShader::TileMode tm)
375 : INHERITED(ctx, shader, tm)
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000376 , fCenterX1(shader.getCenterX1())
377 , fRadius0(shader.getStartRadius())
378 , fDiffRadius(shader.getDiffRadius()) { }
379
380 virtual ~GrConical2Gradient() { }
381
382 static const char* Name() { return "Two-Point Conical Gradient"; }
bsalomon@google.com396e61f2012-10-25 19:00:29 +0000383 virtual const GrBackendEffectFactory& getFactory() const SK_OVERRIDE {
384 return GrTBackendEffectFactory<GrConical2Gradient>::getInstance();
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000385 }
bsalomon@google.coma469c282012-10-24 18:28:34 +0000386 virtual bool isEqual(const GrEffect& sBase) const SK_OVERRIDE {
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000387 const GrConical2Gradient& s = static_cast<const GrConical2Gradient&>(sBase);
388 return (INHERITED::isEqual(sBase) &&
389 this->fCenterX1 == s.fCenterX1 &&
390 this->fRadius0 == s.fRadius0 &&
391 this->fDiffRadius == s.fDiffRadius);
392 }
393
394 // The radial gradient parameters can collapse to a linear (instead of quadratic) equation.
395 bool isDegenerate() const { return SkScalarAbs(fDiffRadius) == SkScalarAbs(fCenterX1); }
396 GrScalar center() const { return fCenterX1; }
397 GrScalar diffRadius() const { return fDiffRadius; }
398 GrScalar radius() const { return fRadius0; }
399
bsalomon@google.com422e81a2012-10-25 14:11:03 +0000400 typedef GrGLConical2Gradient GLEffect;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000401
402private:
bsalomon@google.comf271cc72012-10-24 19:35:13 +0000403 GR_DECLARE_EFFECT_TEST;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000404
405 // @{
406 // Cache of values - these can change arbitrarily, EXCEPT
407 // we shouldn't change between degenerate and non-degenerate?!
408
409 GrScalar fCenterX1;
410 GrScalar fRadius0;
411 GrScalar fDiffRadius;
412
413 // @}
414
415 typedef GrGradientEffect INHERITED;
416};
417
bsalomon@google.comf271cc72012-10-24 19:35:13 +0000418GR_DEFINE_EFFECT_TEST(GrConical2Gradient);
bsalomon@google.comd4726202012-08-03 14:34:46 +0000419
bsalomon@google.coma469c282012-10-24 18:28:34 +0000420GrEffect* GrConical2Gradient::TestCreate(SkRandom* random,
421 GrContext* context,
422 GrTexture**) {
bsalomon@google.comd4726202012-08-03 14:34:46 +0000423 SkPoint center1 = {random->nextUScalar1(), random->nextUScalar1()};
424 SkScalar radius1 = random->nextUScalar1();
425 SkPoint center2;
426 SkScalar radius2;
427 do {
428 center1.set(random->nextUScalar1(), random->nextUScalar1());
429 radius2 = random->nextUScalar1 ();
430 // If the circles are identical the factory will give us an empty shader.
431 } while (radius1 == radius2 && center1 == center2);
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000432
bsalomon@google.comd4726202012-08-03 14:34:46 +0000433 SkColor colors[kMaxRandomGradientColors];
434 SkScalar stopsArray[kMaxRandomGradientColors];
435 SkScalar* stops = stopsArray;
436 SkShader::TileMode tm;
437 int colorCount = RandomGradientParams(random, colors, &stops, &tm);
438 SkAutoTUnref<SkShader> shader(SkGradientShader::CreateTwoPointConical(center1, radius1,
439 center2, radius2,
440 colors, stops, colorCount,
441 tm));
bsalomon@google.com08283af2012-10-26 13:01:20 +0000442 GrEffectStage stage;
443 shader->asNewEffect(context, &stage);
444 GrAssert(NULL != stage.getEffect());
bsalomon@google.com8ea78d82012-10-24 20:11:30 +0000445 // const_cast and ref is a hack! Will remove when asNewEffect returns GrEffect*
bsalomon@google.com08283af2012-10-26 13:01:20 +0000446 stage.getEffect()->ref();
447 return const_cast<GrEffect*>(stage.getEffect());
bsalomon@google.comd4726202012-08-03 14:34:46 +0000448}
449
450
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000451/////////////////////////////////////////////////////////////////////
452
rileya@google.comd7cc6512012-07-27 14:00:39 +0000453GrGLConical2Gradient::GrGLConical2Gradient(
bsalomon@google.com396e61f2012-10-25 19:00:29 +0000454 const GrBackendEffectFactory& factory,
bsalomon@google.coma469c282012-10-24 18:28:34 +0000455 const GrEffect& baseData)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000456 : INHERITED(factory)
457 , fVSParamUni(kInvalidUniformHandle)
458 , fFSParamUni(kInvalidUniformHandle)
459 , fVSVaryingName(NULL)
460 , fFSVaryingName(NULL)
461 , fCachedCenter(GR_ScalarMax)
462 , fCachedRadius(-GR_ScalarMax)
463 , fCachedDiffRadius(-GR_ScalarMax) {
464
465 const GrConical2Gradient& data =
466 static_cast<const GrConical2Gradient&>(baseData);
467 fIsDegenerate = data.isDegenerate();
468}
469
bsalomon@google.comf78df332012-10-29 12:43:38 +0000470void GrGLConical2Gradient::emitCode(GrGLShaderBuilder* builder,
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000471 const GrEffectStage&,
bsalomon@google.comf78df332012-10-29 12:43:38 +0000472 EffectKey,
473 const char* vertexCoords,
474 const char* outputColor,
475 const char* inputColor,
476 const TextureSamplerArray& samplers) {
477 this->emitYCoordUniform(builder);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000478 // 2 copies of uniform array, 1 for each of vertex & fragment shader,
479 // to work around Xoom bug. Doesn't seem to cause performance decrease
480 // in test apps, but need to keep an eye on it.
481 fVSParamUni = builder->addUniformArray(GrGLShaderBuilder::kVertex_ShaderType,
482 kFloat_GrSLType, "Conical2VSParams", 6);
483 fFSParamUni = builder->addUniformArray(GrGLShaderBuilder::kFragment_ShaderType,
484 kFloat_GrSLType, "Conical2FSParams", 6);
485
486 // For radial gradients without perspective we can pass the linear
487 // part of the quadratic as a varying.
bsalomon@google.com34bcb9f2012-08-28 18:20:18 +0000488 if (!builder->defaultTextureMatrixIsPerspective()) {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000489 builder->addVarying(kFloat_GrSLType, "Conical2BCoeff",
490 &fVSVaryingName, &fFSVaryingName);
491 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000492
bsalomon@google.comf78df332012-10-29 12:43:38 +0000493 // VS
494 {
495 SkString* code = &builder->fVSCode;
496 SkString p2; // distance between centers
497 SkString p3; // start radius
498 SkString p5; // difference in radii (r1 - r0)
499 builder->getUniformVariable(fVSParamUni).appendArrayAccess(2, &p2);
500 builder->getUniformVariable(fVSParamUni).appendArrayAccess(3, &p3);
501 builder->getUniformVariable(fVSParamUni).appendArrayAccess(5, &p5);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000502
bsalomon@google.comf78df332012-10-29 12:43:38 +0000503 // For radial gradients without perspective we can pass the linear
504 // part of the quadratic as a varying.
505 if (!builder->defaultTextureMatrixIsPerspective()) {
506 // r2Var = -2 * (r2Parm[2] * varCoord.x - r2Param[3] * r2Param[5])
507 code->appendf("\t%s = -2.0 * (%s * %s.x + %s * %s);\n",
508 fVSVaryingName, p2.c_str(),
509 vertexCoords, p3.c_str(), p5.c_str());
510 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000511 }
512
bsalomon@google.comf78df332012-10-29 12:43:38 +0000513 // FS
514 {
515 SkString* code = &builder->fFSCode;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000516
bsalomon@google.comf78df332012-10-29 12:43:38 +0000517 SkString cName("c");
518 SkString ac4Name("ac4");
519 SkString dName("d");
520 SkString qName("q");
521 SkString r0Name("r0");
522 SkString r1Name("r1");
523 SkString tName("t");
524 SkString p0; // 4a
525 SkString p1; // 1/a
526 SkString p2; // distance between centers
527 SkString p3; // start radius
528 SkString p4; // start radius squared
529 SkString p5; // difference in radii (r1 - r0)
rileya@google.comd7cc6512012-07-27 14:00:39 +0000530
bsalomon@google.comf78df332012-10-29 12:43:38 +0000531 builder->getUniformVariable(fFSParamUni).appendArrayAccess(0, &p0);
532 builder->getUniformVariable(fFSParamUni).appendArrayAccess(1, &p1);
533 builder->getUniformVariable(fFSParamUni).appendArrayAccess(2, &p2);
534 builder->getUniformVariable(fFSParamUni).appendArrayAccess(3, &p3);
535 builder->getUniformVariable(fFSParamUni).appendArrayAccess(4, &p4);
536 builder->getUniformVariable(fFSParamUni).appendArrayAccess(5, &p5);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000537
bsalomon@google.comf78df332012-10-29 12:43:38 +0000538 // If we we're able to interpolate the linear component,
539 // bVar is the varying; otherwise compute it
540 SkString bVar;
541 if (!builder->defaultTextureMatrixIsPerspective()) {
542 bVar = fFSVaryingName;
543 } else {
544 bVar = "b";
545 code->appendf("\tfloat %s = -2.0 * (%s * %s.x + %s * %s);\n",
546 bVar.c_str(), p2.c_str(), builder->defaultTexCoordsName(),
547 p3.c_str(), p5.c_str());
548 }
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000549
bsalomon@google.comf78df332012-10-29 12:43:38 +0000550 // output will default to transparent black (we simply won't write anything
551 // else to it if invalid, instead of discarding or returning prematurely)
552 code->appendf("\t%s = vec4(0.0,0.0,0.0,0.0);\n", outputColor);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000553
bsalomon@google.comf78df332012-10-29 12:43:38 +0000554 // c = (x^2)+(y^2) - params[4]
555 code->appendf("\tfloat %s = dot(%s, %s) - %s;\n", cName.c_str(),
556 builder->defaultTexCoordsName(), builder->defaultTexCoordsName(),
557 p4.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000558
bsalomon@google.comf78df332012-10-29 12:43:38 +0000559 // Non-degenerate case (quadratic)
560 if (!fIsDegenerate) {
rileya@google.comd7cc6512012-07-27 14:00:39 +0000561
bsalomon@google.comf78df332012-10-29 12:43:38 +0000562 // ac4 = params[0] * c
563 code->appendf("\tfloat %s = %s * %s;\n", ac4Name.c_str(), p0.c_str(),
564 cName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000565
bsalomon@google.comf78df332012-10-29 12:43:38 +0000566 // d = b^2 - ac4
567 code->appendf("\tfloat %s = %s * %s - %s;\n", dName.c_str(),
568 bVar.c_str(), bVar.c_str(), ac4Name.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000569
bsalomon@google.comf78df332012-10-29 12:43:38 +0000570 // only proceed if discriminant is >= 0
571 code->appendf("\tif (%s >= 0.0) {\n", dName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000572
bsalomon@google.comf78df332012-10-29 12:43:38 +0000573 // intermediate value we'll use to compute the roots
574 // q = -0.5 * (b +/- sqrt(d))
575 code->appendf("\t\tfloat %s = -0.5 * (%s + (%s < 0.0 ? -1.0 : 1.0)"
576 " * sqrt(%s));\n", qName.c_str(), bVar.c_str(),
577 bVar.c_str(), dName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000578
bsalomon@google.comf78df332012-10-29 12:43:38 +0000579 // compute both roots
580 // r0 = q * params[1]
581 code->appendf("\t\tfloat %s = %s * %s;\n", r0Name.c_str(),
582 qName.c_str(), p1.c_str());
583 // r1 = c / q
584 code->appendf("\t\tfloat %s = %s / %s;\n", r1Name.c_str(),
585 cName.c_str(), qName.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000586
bsalomon@google.comf78df332012-10-29 12:43:38 +0000587 // Note: If there are two roots that both generate radius(t) > 0, the
588 // Canvas spec says to choose the larger t.
rileya@google.comd7cc6512012-07-27 14:00:39 +0000589
bsalomon@google.comf78df332012-10-29 12:43:38 +0000590 // so we'll look at the larger one first:
591 code->appendf("\t\tfloat %s = max(%s, %s);\n", tName.c_str(),
592 r0Name.c_str(), r1Name.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000593
bsalomon@google.comf78df332012-10-29 12:43:38 +0000594 // if r(t) > 0, then we're done; t will be our x coordinate
595 code->appendf("\t\tif (%s * %s + %s > 0.0) {\n", tName.c_str(),
596 p5.c_str(), p3.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000597
bsalomon@google.comf78df332012-10-29 12:43:38 +0000598 code->appendf("\t\t");
599 this->emitColorLookup(builder, tName.c_str(), outputColor, inputColor, samplers[0]);
rileya@google.comd7cc6512012-07-27 14:00:39 +0000600
bsalomon@google.comf78df332012-10-29 12:43:38 +0000601 // otherwise, if r(t) for the larger root was <= 0, try the other root
602 code->appendf("\t\t} else {\n");
603 code->appendf("\t\t\t%s = min(%s, %s);\n", tName.c_str(),
604 r0Name.c_str(), r1Name.c_str());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000605
bsalomon@google.comf78df332012-10-29 12:43:38 +0000606 // if r(t) > 0 for the smaller root, then t will be our x coordinate
607 code->appendf("\t\t\tif (%s * %s + %s > 0.0) {\n",
608 tName.c_str(), p5.c_str(), p3.c_str());
609
610 code->appendf("\t\t\t");
611 this->emitColorLookup(builder, tName.c_str(), outputColor, inputColor, samplers[0]);
612
613 // end if (r(t) > 0) for smaller root
614 code->appendf("\t\t\t}\n");
615 // end if (r(t) > 0), else, for larger root
616 code->appendf("\t\t}\n");
617 // end if (discriminant >= 0)
618 code->appendf("\t}\n");
619 } else {
620
621 // linear case: t = -c/b
622 code->appendf("\tfloat %s = -(%s / %s);\n", tName.c_str(),
623 cName.c_str(), bVar.c_str());
624
625 // if r(t) > 0, then t will be the x coordinate
626 code->appendf("\tif (%s * %s + %s > 0.0) {\n", tName.c_str(),
627 p5.c_str(), p3.c_str());
628 code->appendf("\t");
629 this->emitColorLookup(builder, tName.c_str(), outputColor, inputColor, samplers[0]);
630 code->appendf("\t}\n");
631 }
rileya@google.comd7cc6512012-07-27 14:00:39 +0000632 }
633}
634
bsalomon@google.com28a15fb2012-10-26 17:53:18 +0000635void GrGLConical2Gradient::setData(const GrGLUniformManager& uman, const GrEffectStage& stage) {
636 INHERITED::setData(uman, stage);
637 const GrConical2Gradient& data = static_cast<const GrConical2Gradient&>(*stage.getEffect());
rileya@google.comd7cc6512012-07-27 14:00:39 +0000638 GrAssert(data.isDegenerate() == fIsDegenerate);
639 GrScalar centerX1 = data.center();
640 GrScalar radius0 = data.radius();
641 GrScalar diffRadius = data.diffRadius();
642
643 if (fCachedCenter != centerX1 ||
644 fCachedRadius != radius0 ||
645 fCachedDiffRadius != diffRadius) {
646
647 GrScalar a = GrMul(centerX1, centerX1) - diffRadius * diffRadius;
648
649 // When we're in the degenerate (linear) case, the second
650 // value will be INF but the program doesn't read it. (We
651 // use the same 6 uniforms even though we don't need them
652 // all in the linear case just to keep the code complexity
653 // down).
654 float values[6] = {
655 GrScalarToFloat(a * 4),
656 1.f / (GrScalarToFloat(a)),
657 GrScalarToFloat(centerX1),
658 GrScalarToFloat(radius0),
659 GrScalarToFloat(SkScalarMul(radius0, radius0)),
660 GrScalarToFloat(diffRadius)
661 };
662
663 uman.set1fv(fVSParamUni, 0, 6, values);
664 uman.set1fv(fFSParamUni, 0, 6, values);
665 fCachedCenter = centerX1;
666 fCachedRadius = radius0;
667 fCachedDiffRadius = diffRadius;
668 }
669}
670
bsalomon@google.com2eaaefd2012-10-29 19:51:22 +0000671GrGLEffect::EffectKey GrGLConical2Gradient::GenKey(const GrEffectStage& s, const GrGLCaps&) {
672 return (static_cast<const GrConical2Gradient&>(*s.getEffect()).isDegenerate());
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000673}
rileya@google.comd7cc6512012-07-27 14:00:39 +0000674
675/////////////////////////////////////////////////////////////////////
676
bsalomon@google.com8ea78d82012-10-24 20:11:30 +0000677bool SkTwoPointConicalGradient::asNewEffect(GrContext* context,
bsalomon@google.com08283af2012-10-26 13:01:20 +0000678 GrEffectStage* stage) const {
679 SkASSERT(NULL != context && NULL != stage);
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000680
681 SkMatrix matrix;
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000682 SkPoint diff = fCenter2 - fCenter1;
683 SkScalar diffLen = diff.length();
684 if (0 != diffLen) {
685 SkScalar invDiffLen = SkScalarInvert(diffLen);
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000686 matrix.setSinCos(-SkScalarMul(invDiffLen, diff.fY),
687 SkScalarMul(invDiffLen, diff.fX));
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000688 } else {
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000689 matrix.reset();
rileya@google.com98e8b6d2012-07-31 20:38:06 +0000690 }
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000691 matrix.preTranslate(-fCenter1.fX, -fCenter1.fY);
692
693 SkMatrix localM;
694 if (this->getLocalMatrix(&localM)) {
695 if (!localM.invert(&localM)) {
696 return false;
697 }
698 matrix.preConcat(localM);
699 }
700
bsalomon@google.com08283af2012-10-26 13:01:20 +0000701 stage->setEffect(SkNEW_ARGS(GrConical2Gradient, (context, *this, fTileMode)), matrix)->unref();
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000702
703 return true;
rileya@google.comd7cc6512012-07-27 14:00:39 +0000704}
705
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000706#else
707
bsalomon@google.com08283af2012-10-26 13:01:20 +0000708bool SkTwoPointConicalGradient::asNewEffect(GrContext*, GrEffectStage*) const {
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000709 SkDEBUGFAIL("Should not call in GPU-less build");
bsalomon@google.comdfdb7e52012-10-16 15:19:45 +0000710 return false;
bsalomon@google.comcf8fb1f2012-08-02 14:03:32 +0000711}
712
twiz@google.coma5e65ec2012-08-02 15:15:16 +0000713#endif