Make all GrFragmentProcessors GL independent.
[platform/upstream/libSkiaSharp.git] / src / gpu / effects / GrBezierEffect.cpp
1 /*
2  * Copyright 2013 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 "GrBezierEffect.h"
9
10 #include "gl/GrGLGeometryProcessor.h"
11 #include "glsl/GrGLSLFragmentProcessor.h"
12 #include "glsl/GrGLSLProgramBuilder.h"
13 #include "glsl/GrGLSLProgramDataManager.h"
14 #include "glsl/GrGLSLUtil.h"
15
16 class GrGLConicEffect : public GrGLGeometryProcessor {
17 public:
18     GrGLConicEffect(const GrGeometryProcessor&);
19
20     void onEmitCode(EmitArgs&, GrGPArgs*) override;
21
22     static inline void GenKey(const GrGeometryProcessor&,
23                               const GrGLSLCaps&,
24                               GrProcessorKeyBuilder*);
25
26     void setData(const GrGLSLProgramDataManager& pdman,
27                  const GrPrimitiveProcessor& primProc) override {
28         const GrConicEffect& ce = primProc.cast<GrConicEffect>();
29
30         if (!ce.viewMatrix().isIdentity() && !fViewMatrix.cheapEqualTo(ce.viewMatrix())) {
31             fViewMatrix = ce.viewMatrix();
32             float viewMatrix[3 * 3];
33             GrGLSLGetMatrix<3>(viewMatrix, fViewMatrix);
34             pdman.setMatrix3f(fViewMatrixUniform, viewMatrix);
35         }
36
37         if (ce.color() != fColor) {
38             float c[4];
39             GrColorToRGBAFloat(ce.color(), c);
40             pdman.set4fv(fColorUniform, 1, c);
41             fColor = ce.color();
42         }
43
44         if (ce.coverageScale() != 0xff && ce.coverageScale() != fCoverageScale) {
45             pdman.set1f(fCoverageScaleUniform, GrNormalizeByteToFloat(ce.coverageScale()));
46             fCoverageScale = ce.coverageScale();
47         }
48     }
49
50     void setTransformData(const GrPrimitiveProcessor& primProc,
51                           const GrGLSLProgramDataManager& pdman,
52                           int index,
53                           const SkTArray<const GrCoordTransform*, true>& transforms) override {
54         this->setTransformDataHelper<GrConicEffect>(primProc, pdman, index, transforms);
55     }
56
57 private:
58     SkMatrix fViewMatrix;
59     GrColor fColor;
60     uint8_t fCoverageScale;
61     GrPrimitiveEdgeType fEdgeType;
62     UniformHandle fColorUniform;
63     UniformHandle fCoverageScaleUniform;
64     UniformHandle fViewMatrixUniform;
65
66     typedef GrGLGeometryProcessor INHERITED;
67 };
68
69 GrGLConicEffect::GrGLConicEffect(const GrGeometryProcessor& processor)
70     : fViewMatrix(SkMatrix::InvalidMatrix()), fColor(GrColor_ILLEGAL), fCoverageScale(0xff) {
71     const GrConicEffect& ce = processor.cast<GrConicEffect>();
72     fEdgeType = ce.getEdgeType();
73 }
74
75 void GrGLConicEffect::onEmitCode(EmitArgs& args, GrGPArgs* gpArgs) {
76     GrGLSLGPBuilder* pb = args.fPB;
77     GrGLSLVertexBuilder* vsBuilder = args.fPB->getVertexShaderBuilder();
78     const GrConicEffect& gp = args.fGP.cast<GrConicEffect>();
79
80     // emit attributes
81     vsBuilder->emitAttributes(gp);
82
83     GrGLSLVertToFrag v(kVec4f_GrSLType);
84     args.fPB->addVarying("ConicCoeffs", &v);
85     vsBuilder->codeAppendf("%s = %s;", v.vsOut(), gp.inConicCoeffs()->fName);
86
87     // Setup pass through color
88     if (!gp.colorIgnored()) {
89         this->setupUniformColor(args.fPB, args.fOutputColor, &fColorUniform);
90     }
91
92     // Setup position
93     this->setupPosition(pb, gpArgs, gp.inPosition()->fName, gp.viewMatrix(), &fViewMatrixUniform);
94
95     // emit transforms with position
96     this->emitTransforms(pb, gpArgs->fPositionVar, gp.inPosition()->fName, gp.localMatrix(),
97                          args.fTransformsIn, args.fTransformsOut);
98
99     GrGLSLFragmentBuilder* fsBuilder = args.fPB->getFragmentShaderBuilder();
100     fsBuilder->codeAppend("float edgeAlpha;");
101
102     switch (fEdgeType) {
103         case kHairlineAA_GrProcessorEdgeType: {
104             SkAssertResult(fsBuilder->enableFeature(
105                     GrGLSLFragmentShaderBuilder::kStandardDerivatives_GLSLFeature));
106             fsBuilder->codeAppendf("vec3 dklmdx = dFdx(%s.xyz);", v.fsIn());
107             fsBuilder->codeAppendf("vec3 dklmdy = dFdy(%s.xyz);", v.fsIn());
108             fsBuilder->codeAppendf("float dfdx ="
109                                    "2.0 * %s.x * dklmdx.x - %s.y * dklmdx.z - %s.z * dklmdx.y;",
110                                    v.fsIn(), v.fsIn(), v.fsIn());
111             fsBuilder->codeAppendf("float dfdy ="
112                                    "2.0 * %s.x * dklmdy.x - %s.y * dklmdy.z - %s.z * dklmdy.y;",
113                                    v.fsIn(), v.fsIn(), v.fsIn());
114             fsBuilder->codeAppend("vec2 gF = vec2(dfdx, dfdy);");
115             fsBuilder->codeAppend("float gFM = sqrt(dot(gF, gF));");
116             fsBuilder->codeAppendf("float func = %s.x*%s.x - %s.y*%s.z;", v.fsIn(), v.fsIn(),
117                                    v.fsIn(), v.fsIn());
118             fsBuilder->codeAppend("func = abs(func);");
119             fsBuilder->codeAppend("edgeAlpha = func / gFM;");
120             fsBuilder->codeAppend("edgeAlpha = max(1.0 - edgeAlpha, 0.0);");
121             // Add line below for smooth cubic ramp
122             // fsBuilder->codeAppend("edgeAlpha = edgeAlpha*edgeAlpha*(3.0-2.0*edgeAlpha);");
123             break;
124         }
125         case kFillAA_GrProcessorEdgeType: {
126             SkAssertResult(fsBuilder->enableFeature(
127                     GrGLSLFragmentShaderBuilder::kStandardDerivatives_GLSLFeature));
128             fsBuilder->codeAppendf("vec3 dklmdx = dFdx(%s.xyz);", v.fsIn());
129             fsBuilder->codeAppendf("vec3 dklmdy = dFdy(%s.xyz);", v.fsIn());
130             fsBuilder->codeAppendf("float dfdx ="
131                                    "2.0 * %s.x * dklmdx.x - %s.y * dklmdx.z - %s.z * dklmdx.y;",
132                                    v.fsIn(), v.fsIn(), v.fsIn());
133             fsBuilder->codeAppendf("float dfdy ="
134                                    "2.0 * %s.x * dklmdy.x - %s.y * dklmdy.z - %s.z * dklmdy.y;",
135                                    v.fsIn(), v.fsIn(), v.fsIn());
136             fsBuilder->codeAppend("vec2 gF = vec2(dfdx, dfdy);");
137             fsBuilder->codeAppend("float gFM = sqrt(dot(gF, gF));");
138             fsBuilder->codeAppendf("float func = %s.x * %s.x - %s.y * %s.z;", v.fsIn(), v.fsIn(),
139                                    v.fsIn(), v.fsIn());
140             fsBuilder->codeAppend("edgeAlpha = func / gFM;");
141             fsBuilder->codeAppend("edgeAlpha = clamp(1.0 - edgeAlpha, 0.0, 1.0);");
142             // Add line below for smooth cubic ramp
143             // fsBuilder->codeAppend("edgeAlpha = edgeAlpha*edgeAlpha*(3.0-2.0*edgeAlpha);");
144             break;
145         }
146         case kFillBW_GrProcessorEdgeType: {
147             fsBuilder->codeAppendf("edgeAlpha = %s.x * %s.x - %s.y * %s.z;", v.fsIn(), v.fsIn(),
148                                    v.fsIn(), v.fsIn());
149             fsBuilder->codeAppend("edgeAlpha = float(edgeAlpha < 0.0);");
150             break;
151         }
152         default:
153             SkFAIL("Shouldn't get here");
154     }
155
156     // TODO should we really be doing this?
157     if (gp.coverageScale() != 0xff) {
158         const char* coverageScale;
159         fCoverageScaleUniform = pb->addUniform(GrGLSLProgramBuilder::kFragment_Visibility,
160                                                kFloat_GrSLType,
161                                                kDefault_GrSLPrecision,
162                                                "Coverage",
163                                                &coverageScale);
164         fsBuilder->codeAppendf("%s = vec4(%s * edgeAlpha);", args.fOutputCoverage, coverageScale);
165     } else {
166         fsBuilder->codeAppendf("%s = vec4(edgeAlpha);", args.fOutputCoverage);
167     }
168 }
169
170 void GrGLConicEffect::GenKey(const GrGeometryProcessor& gp,
171                              const GrGLSLCaps&,
172                              GrProcessorKeyBuilder* b) {
173     const GrConicEffect& ce = gp.cast<GrConicEffect>();
174     uint32_t key = ce.isAntiAliased() ? (ce.isFilled() ? 0x0 : 0x1) : 0x2;
175     key |= GrColor_ILLEGAL != ce.color() ? 0x4 : 0x0;
176     key |= 0xff != ce.coverageScale() ? 0x8 : 0x0;
177     key |= ce.usesLocalCoords() && ce.localMatrix().hasPerspective() ? 0x10 : 0x0;
178     key |= ComputePosKey(ce.viewMatrix()) << 5;
179     b->add32(key);
180 }
181
182 //////////////////////////////////////////////////////////////////////////////
183
184 GrConicEffect::~GrConicEffect() {}
185
186 void GrConicEffect::getGLProcessorKey(const GrGLSLCaps& caps,
187                                       GrProcessorKeyBuilder* b) const {
188     GrGLConicEffect::GenKey(*this, caps, b);
189 }
190
191 GrGLPrimitiveProcessor* GrConicEffect::createGLInstance(const GrGLSLCaps&) const {
192     return new GrGLConicEffect(*this);
193 }
194
195 GrConicEffect::GrConicEffect(GrColor color, const SkMatrix& viewMatrix, uint8_t coverage,
196                              GrPrimitiveEdgeType edgeType, const SkMatrix& localMatrix,
197                              bool usesLocalCoords)
198     : fColor(color)
199     , fViewMatrix(viewMatrix)
200     , fLocalMatrix(viewMatrix)
201     , fUsesLocalCoords(usesLocalCoords)
202     , fCoverageScale(coverage)
203     , fEdgeType(edgeType) {
204     this->initClassID<GrConicEffect>();
205     fInPosition = &this->addVertexAttrib(Attribute("inPosition", kVec2f_GrVertexAttribType,
206                                                    kHigh_GrSLPrecision));
207     fInConicCoeffs = &this->addVertexAttrib(Attribute("inConicCoeffs",
208                                                       kVec4f_GrVertexAttribType));
209 }
210
211 //////////////////////////////////////////////////////////////////////////////
212
213 GR_DEFINE_GEOMETRY_PROCESSOR_TEST(GrConicEffect);
214
215 const GrGeometryProcessor* GrConicEffect::TestCreate(GrProcessorTestData* d) {
216     GrGeometryProcessor* gp;
217     do {
218         GrPrimitiveEdgeType edgeType =
219                 static_cast<GrPrimitiveEdgeType>(
220                         d->fRandom->nextULessThan(kGrProcessorEdgeTypeCnt));
221         gp = GrConicEffect::Create(GrRandomColor(d->fRandom), GrTest::TestMatrix(d->fRandom),
222                                    edgeType, *d->fCaps,
223                                    GrTest::TestMatrix(d->fRandom), d->fRandom->nextBool());
224     } while (nullptr == gp);
225     return gp;
226 }
227
228 //////////////////////////////////////////////////////////////////////////////
229 // Quad
230 //////////////////////////////////////////////////////////////////////////////
231
232 class GrGLQuadEffect : public GrGLGeometryProcessor {
233 public:
234     GrGLQuadEffect(const GrGeometryProcessor&);
235
236     void onEmitCode(EmitArgs&, GrGPArgs*) override;
237
238     static inline void GenKey(const GrGeometryProcessor&,
239                               const GrGLSLCaps&,
240                               GrProcessorKeyBuilder*);
241
242     void setData(const GrGLSLProgramDataManager& pdman,
243                  const GrPrimitiveProcessor& primProc) override {
244         const GrQuadEffect& qe = primProc.cast<GrQuadEffect>();
245
246         if (!qe.viewMatrix().isIdentity() && !fViewMatrix.cheapEqualTo(qe.viewMatrix())) {
247             fViewMatrix = qe.viewMatrix();
248             float viewMatrix[3 * 3];
249             GrGLSLGetMatrix<3>(viewMatrix, fViewMatrix);
250             pdman.setMatrix3f(fViewMatrixUniform, viewMatrix);
251         }
252
253         if (qe.color() != fColor) {
254             float c[4];
255             GrColorToRGBAFloat(qe.color(), c);
256             pdman.set4fv(fColorUniform, 1, c);
257             fColor = qe.color();
258         }
259
260         if (qe.coverageScale() != 0xff && qe.coverageScale() != fCoverageScale) {
261             pdman.set1f(fCoverageScaleUniform, GrNormalizeByteToFloat(qe.coverageScale()));
262             fCoverageScale = qe.coverageScale();
263         }
264     }
265
266     void setTransformData(const GrPrimitiveProcessor& primProc,
267                           const GrGLSLProgramDataManager& pdman,
268                           int index,
269                           const SkTArray<const GrCoordTransform*, true>& transforms) override {
270         this->setTransformDataHelper<GrQuadEffect>(primProc, pdman, index, transforms);
271     }
272
273 private:
274     SkMatrix fViewMatrix;
275     GrColor fColor;
276     uint8_t fCoverageScale;
277     GrPrimitiveEdgeType fEdgeType;
278     UniformHandle fColorUniform;
279     UniformHandle fCoverageScaleUniform;
280     UniformHandle fViewMatrixUniform;
281
282     typedef GrGLGeometryProcessor INHERITED;
283 };
284
285 GrGLQuadEffect::GrGLQuadEffect(const GrGeometryProcessor& processor)
286     : fViewMatrix(SkMatrix::InvalidMatrix()), fColor(GrColor_ILLEGAL), fCoverageScale(0xff) {
287     const GrQuadEffect& ce = processor.cast<GrQuadEffect>();
288     fEdgeType = ce.getEdgeType();
289 }
290
291 void GrGLQuadEffect::onEmitCode(EmitArgs& args, GrGPArgs* gpArgs) {
292     GrGLSLGPBuilder* pb = args.fPB;
293     GrGLSLVertexBuilder* vsBuilder = args.fPB->getVertexShaderBuilder();
294     const GrQuadEffect& gp = args.fGP.cast<GrQuadEffect>();
295
296     // emit attributes
297     vsBuilder->emitAttributes(gp);
298
299     GrGLSLVertToFrag v(kVec4f_GrSLType);
300     args.fPB->addVarying("HairQuadEdge", &v);
301     vsBuilder->codeAppendf("%s = %s;", v.vsOut(), gp.inHairQuadEdge()->fName);
302
303     // Setup pass through color
304     if (!gp.colorIgnored()) {
305         this->setupUniformColor(args.fPB, args.fOutputColor, &fColorUniform);
306     }
307
308     // Setup position
309     this->setupPosition(pb, gpArgs, gp.inPosition()->fName, gp.viewMatrix(), &fViewMatrixUniform);
310
311     // emit transforms with position
312     this->emitTransforms(pb, gpArgs->fPositionVar, gp.inPosition()->fName, gp.localMatrix(),
313                          args.fTransformsIn, args.fTransformsOut);
314
315     GrGLSLFragmentBuilder* fsBuilder = args.fPB->getFragmentShaderBuilder();
316     fsBuilder->codeAppendf("float edgeAlpha;");
317
318     switch (fEdgeType) {
319         case kHairlineAA_GrProcessorEdgeType: {
320             SkAssertResult(fsBuilder->enableFeature(
321                     GrGLSLFragmentShaderBuilder::kStandardDerivatives_GLSLFeature));
322             fsBuilder->codeAppendf("vec2 duvdx = dFdx(%s.xy);", v.fsIn());
323             fsBuilder->codeAppendf("vec2 duvdy = dFdy(%s.xy);", v.fsIn());
324             fsBuilder->codeAppendf("vec2 gF = vec2(2.0 * %s.x * duvdx.x - duvdx.y,"
325                                    "               2.0 * %s.x * duvdy.x - duvdy.y);",
326                                    v.fsIn(), v.fsIn());
327             fsBuilder->codeAppendf("edgeAlpha = (%s.x * %s.x - %s.y);", v.fsIn(), v.fsIn(), v.fsIn());
328             fsBuilder->codeAppend("edgeAlpha = sqrt(edgeAlpha * edgeAlpha / dot(gF, gF));");
329             fsBuilder->codeAppend("edgeAlpha = max(1.0 - edgeAlpha, 0.0);");
330             // Add line below for smooth cubic ramp
331             // fsBuilder->codeAppend("edgeAlpha = edgeAlpha*edgeAlpha*(3.0-2.0*edgeAlpha);");
332             break;
333         }
334         case kFillAA_GrProcessorEdgeType: {
335             SkAssertResult(fsBuilder->enableFeature(
336                     GrGLSLFragmentShaderBuilder::kStandardDerivatives_GLSLFeature));
337             fsBuilder->codeAppendf("vec2 duvdx = dFdx(%s.xy);", v.fsIn());
338             fsBuilder->codeAppendf("vec2 duvdy = dFdy(%s.xy);", v.fsIn());
339             fsBuilder->codeAppendf("vec2 gF = vec2(2.0 * %s.x * duvdx.x - duvdx.y,"
340                                    "               2.0 * %s.x * duvdy.x - duvdy.y);",
341                                    v.fsIn(), v.fsIn());
342             fsBuilder->codeAppendf("edgeAlpha = (%s.x * %s.x - %s.y);", v.fsIn(), v.fsIn(), v.fsIn());
343             fsBuilder->codeAppend("edgeAlpha = edgeAlpha / sqrt(dot(gF, gF));");
344             fsBuilder->codeAppend("edgeAlpha = clamp(1.0 - edgeAlpha, 0.0, 1.0);");
345             // Add line below for smooth cubic ramp
346             // fsBuilder->codeAppend("edgeAlpha = edgeAlpha*edgeAlpha*(3.0-2.0*edgeAlpha);");
347             break;
348         }
349         case kFillBW_GrProcessorEdgeType: {
350             fsBuilder->codeAppendf("edgeAlpha = (%s.x * %s.x - %s.y);", v.fsIn(), v.fsIn(), v.fsIn());
351             fsBuilder->codeAppend("edgeAlpha = float(edgeAlpha < 0.0);");
352             break;
353         }
354         default:
355             SkFAIL("Shouldn't get here");
356     }
357
358     if (0xff != gp.coverageScale()) {
359         const char* coverageScale;
360         fCoverageScaleUniform = pb->addUniform(GrGLSLProgramBuilder::kFragment_Visibility,
361                                           kFloat_GrSLType,
362                                           kDefault_GrSLPrecision,
363                                           "Coverage",
364                                           &coverageScale);
365         fsBuilder->codeAppendf("%s = vec4(%s * edgeAlpha);", args.fOutputCoverage, coverageScale);
366     } else {
367         fsBuilder->codeAppendf("%s = vec4(edgeAlpha);", args.fOutputCoverage);
368     }
369 }
370
371 void GrGLQuadEffect::GenKey(const GrGeometryProcessor& gp,
372                             const GrGLSLCaps&,
373                             GrProcessorKeyBuilder* b) {
374     const GrQuadEffect& ce = gp.cast<GrQuadEffect>();
375     uint32_t key = ce.isAntiAliased() ? (ce.isFilled() ? 0x0 : 0x1) : 0x2;
376     key |= ce.color() != GrColor_ILLEGAL ? 0x4 : 0x0;
377     key |= ce.coverageScale() != 0xff ? 0x8 : 0x0;
378     key |= ce.usesLocalCoords() && ce.localMatrix().hasPerspective() ? 0x10 : 0x0;
379     key |= ComputePosKey(ce.viewMatrix()) << 5;
380     b->add32(key);
381 }
382
383 //////////////////////////////////////////////////////////////////////////////
384
385 GrQuadEffect::~GrQuadEffect() {}
386
387 void GrQuadEffect::getGLProcessorKey(const GrGLSLCaps& caps,
388                                      GrProcessorKeyBuilder* b) const {
389     GrGLQuadEffect::GenKey(*this, caps, b);
390 }
391
392 GrGLPrimitiveProcessor* GrQuadEffect::createGLInstance(const GrGLSLCaps&) const {
393     return new GrGLQuadEffect(*this);
394 }
395
396 GrQuadEffect::GrQuadEffect(GrColor color, const SkMatrix& viewMatrix, uint8_t coverage,
397                            GrPrimitiveEdgeType edgeType, const SkMatrix& localMatrix,
398                            bool usesLocalCoords)
399     : fColor(color)
400     , fViewMatrix(viewMatrix)
401     , fLocalMatrix(localMatrix)
402     , fUsesLocalCoords(usesLocalCoords)
403     , fCoverageScale(coverage)
404     , fEdgeType(edgeType) {
405     this->initClassID<GrQuadEffect>();
406     fInPosition = &this->addVertexAttrib(Attribute("inPosition", kVec2f_GrVertexAttribType,
407                                                    kHigh_GrSLPrecision));
408     fInHairQuadEdge = &this->addVertexAttrib(Attribute("inHairQuadEdge",
409                                                         kVec4f_GrVertexAttribType));
410 }
411
412 //////////////////////////////////////////////////////////////////////////////
413
414 GR_DEFINE_GEOMETRY_PROCESSOR_TEST(GrQuadEffect);
415
416 const GrGeometryProcessor* GrQuadEffect::TestCreate(GrProcessorTestData* d) {
417     GrGeometryProcessor* gp;
418     do {
419         GrPrimitiveEdgeType edgeType = static_cast<GrPrimitiveEdgeType>(
420                 d->fRandom->nextULessThan(kGrProcessorEdgeTypeCnt));
421         gp = GrQuadEffect::Create(GrRandomColor(d->fRandom),
422                                   GrTest::TestMatrix(d->fRandom),
423                                   edgeType, *d->fCaps,
424                                   GrTest::TestMatrix(d->fRandom),
425                                   d->fRandom->nextBool());
426     } while (nullptr == gp);
427     return gp;
428 }
429
430 //////////////////////////////////////////////////////////////////////////////
431 // Cubic
432 //////////////////////////////////////////////////////////////////////////////
433
434 class GrGLCubicEffect : public GrGLGeometryProcessor {
435 public:
436     GrGLCubicEffect(const GrGeometryProcessor&);
437
438     void onEmitCode(EmitArgs&, GrGPArgs*) override;
439
440     static inline void GenKey(const GrGeometryProcessor&,
441                               const GrGLSLCaps&,
442                               GrProcessorKeyBuilder*);
443
444     void setData(const GrGLSLProgramDataManager& pdman,
445                  const GrPrimitiveProcessor& primProc) override {
446         const GrCubicEffect& ce = primProc.cast<GrCubicEffect>();
447
448         if (!ce.viewMatrix().isIdentity() && !fViewMatrix.cheapEqualTo(ce.viewMatrix())) {
449             fViewMatrix = ce.viewMatrix();
450             float viewMatrix[3 * 3];
451             GrGLSLGetMatrix<3>(viewMatrix, fViewMatrix);
452             pdman.setMatrix3f(fViewMatrixUniform, viewMatrix);
453         }
454
455         if (ce.color() != fColor) {
456             float c[4];
457             GrColorToRGBAFloat(ce.color(), c);
458             pdman.set4fv(fColorUniform, 1, c);
459             fColor = ce.color();
460         }
461     }
462
463 private:
464     SkMatrix fViewMatrix;
465     GrColor fColor;
466     GrPrimitiveEdgeType fEdgeType;
467     UniformHandle fColorUniform;
468     UniformHandle fViewMatrixUniform;
469
470     typedef GrGLGeometryProcessor INHERITED;
471 };
472
473 GrGLCubicEffect::GrGLCubicEffect(const GrGeometryProcessor& processor)
474     : fViewMatrix(SkMatrix::InvalidMatrix()), fColor(GrColor_ILLEGAL) {
475     const GrCubicEffect& ce = processor.cast<GrCubicEffect>();
476     fEdgeType = ce.getEdgeType();
477 }
478
479 void GrGLCubicEffect::onEmitCode(EmitArgs& args, GrGPArgs* gpArgs) {
480     GrGLSLVertexBuilder* vsBuilder = args.fPB->getVertexShaderBuilder();
481     const GrCubicEffect& gp = args.fGP.cast<GrCubicEffect>();
482
483     // emit attributes
484     vsBuilder->emitAttributes(gp);
485
486     GrGLSLVertToFrag v(kVec4f_GrSLType);
487     args.fPB->addVarying("CubicCoeffs", &v, kHigh_GrSLPrecision);
488     vsBuilder->codeAppendf("%s = %s;", v.vsOut(), gp.inCubicCoeffs()->fName);
489
490     // Setup pass through color
491     if (!gp.colorIgnored()) {
492         this->setupUniformColor(args.fPB, args.fOutputColor, &fColorUniform);
493     }
494
495     // Setup position
496     this->setupPosition(args.fPB, gpArgs, gp.inPosition()->fName, gp.viewMatrix(),
497                         &fViewMatrixUniform);
498
499     // emit transforms with position
500     this->emitTransforms(args.fPB, gpArgs->fPositionVar, gp.inPosition()->fName, args.fTransformsIn,
501                          args.fTransformsOut);
502
503     GrGLSLFragmentBuilder* fsBuilder = args.fPB->getFragmentShaderBuilder();
504
505     GrGLSLShaderVar edgeAlpha("edgeAlpha", kFloat_GrSLType, 0, kHigh_GrSLPrecision);
506     GrGLSLShaderVar dklmdx("dklmdx", kVec3f_GrSLType, 0, kHigh_GrSLPrecision);
507     GrGLSLShaderVar dklmdy("dklmdy", kVec3f_GrSLType, 0, kHigh_GrSLPrecision);
508     GrGLSLShaderVar dfdx("dfdx", kFloat_GrSLType, 0, kHigh_GrSLPrecision);
509     GrGLSLShaderVar dfdy("dfdy", kFloat_GrSLType, 0, kHigh_GrSLPrecision);
510     GrGLSLShaderVar gF("gF", kVec2f_GrSLType, 0, kHigh_GrSLPrecision);
511     GrGLSLShaderVar gFM("gFM", kFloat_GrSLType, 0, kHigh_GrSLPrecision);
512     GrGLSLShaderVar func("func", kFloat_GrSLType, 0, kHigh_GrSLPrecision);
513
514     fsBuilder->declAppend(edgeAlpha);
515     fsBuilder->declAppend(dklmdx);
516     fsBuilder->declAppend(dklmdy);
517     fsBuilder->declAppend(dfdx);
518     fsBuilder->declAppend(dfdy);
519     fsBuilder->declAppend(gF);
520     fsBuilder->declAppend(gFM);
521     fsBuilder->declAppend(func);
522
523     switch (fEdgeType) {
524         case kHairlineAA_GrProcessorEdgeType: {
525             SkAssertResult(fsBuilder->enableFeature(
526                     GrGLSLFragmentShaderBuilder::kStandardDerivatives_GLSLFeature));
527             fsBuilder->codeAppendf("%s = dFdx(%s.xyz);", dklmdx.c_str(), v.fsIn());
528             fsBuilder->codeAppendf("%s = dFdy(%s.xyz);", dklmdy.c_str(), v.fsIn());
529             fsBuilder->codeAppendf("%s = 3.0 * %s.x * %s.x * %s.x - %s.y * %s.z - %s.z * %s.y;",
530                                    dfdx.c_str(), v.fsIn(), v.fsIn(), dklmdx.c_str(), v.fsIn(),
531                                    dklmdx.c_str(), v.fsIn(), dklmdx.c_str());
532             fsBuilder->codeAppendf("%s = 3.0 * %s.x * %s.x * %s.x - %s.y * %s.z - %s.z * %s.y;",
533                                    dfdy.c_str(), v.fsIn(), v.fsIn(), dklmdy.c_str(), v.fsIn(),
534                                    dklmdy.c_str(), v.fsIn(), dklmdy.c_str());
535             fsBuilder->codeAppendf("%s = vec2(%s, %s);", gF.c_str(), dfdx.c_str(), dfdy.c_str());
536             fsBuilder->codeAppendf("%s = sqrt(dot(%s, %s));", gFM.c_str(), gF.c_str(), gF.c_str());
537             fsBuilder->codeAppendf("%s = %s.x * %s.x * %s.x - %s.y * %s.z;",
538                                    func.c_str(), v.fsIn(), v.fsIn(), v.fsIn(), v.fsIn(), v.fsIn());
539             fsBuilder->codeAppendf("%s = abs(%s);", func.c_str(), func.c_str());
540             fsBuilder->codeAppendf("%s = %s / %s;",
541                                    edgeAlpha.c_str(), func.c_str(), gFM.c_str());
542             fsBuilder->codeAppendf("%s = max(1.0 - %s, 0.0);",
543                                    edgeAlpha.c_str(), edgeAlpha.c_str());
544             // Add line below for smooth cubic ramp
545             // fsBuilder->codeAppendf("%s = %s * %s * (3.0 - 2.0 * %s);",
546             //                        edgeAlpha.c_str(), edgeAlpha.c_str(), edgeAlpha.c_str(),
547             //                        edgeAlpha.c_str());
548             break;
549         }
550         case kFillAA_GrProcessorEdgeType: {
551             SkAssertResult(fsBuilder->enableFeature(
552                     GrGLSLFragmentShaderBuilder::kStandardDerivatives_GLSLFeature));
553             fsBuilder->codeAppendf("%s = dFdx(%s.xyz);", dklmdx.c_str(), v.fsIn());
554             fsBuilder->codeAppendf("%s = dFdy(%s.xyz);", dklmdy.c_str(), v.fsIn());
555             fsBuilder->codeAppendf("%s ="
556                                    "3.0 * %s.x * %s.x * %s.x - %s.y * %s.z - %s.z * %s.y;",
557                                    dfdx.c_str(), v.fsIn(), v.fsIn(), dklmdx.c_str(), v.fsIn(),
558                                    dklmdx.c_str(), v.fsIn(), dklmdx.c_str());
559             fsBuilder->codeAppendf("%s = 3.0 * %s.x * %s.x * %s.x - %s.y * %s.z - %s.z * %s.y;",
560                                    dfdy.c_str(), v.fsIn(), v.fsIn(), dklmdy.c_str(), v.fsIn(),
561                                    dklmdy.c_str(), v.fsIn(), dklmdy.c_str());
562             fsBuilder->codeAppendf("%s = vec2(%s, %s);", gF.c_str(), dfdx.c_str(), dfdy.c_str());
563             fsBuilder->codeAppendf("%s = sqrt(dot(%s, %s));", gFM.c_str(), gF.c_str(), gF.c_str());
564             fsBuilder->codeAppendf("%s = %s.x * %s.x * %s.x - %s.y * %s.z;",
565                                    func.c_str(), v.fsIn(), v.fsIn(), v.fsIn(), v.fsIn(), v.fsIn());
566             fsBuilder->codeAppendf("%s = %s / %s;",
567                                    edgeAlpha.c_str(), func.c_str(), gFM.c_str());
568             fsBuilder->codeAppendf("%s = clamp(1.0 - %s, 0.0, 1.0);",
569                                    edgeAlpha.c_str(), edgeAlpha.c_str());
570             // Add line below for smooth cubic ramp
571             // fsBuilder->codeAppendf("%s = %s * %s * (3.0 - 2.0 * %s);",
572             //                        edgeAlpha.c_str(), edgeAlpha.c_str(), edgeAlpha.c_str(),
573             //                        edgeAlpha.c_str());
574             break;
575         }
576         case kFillBW_GrProcessorEdgeType: {
577             fsBuilder->codeAppendf("%s = %s.x * %s.x * %s.x - %s.y * %s.z;",
578                                    edgeAlpha.c_str(), v.fsIn(), v.fsIn(), v.fsIn(), v.fsIn(), v.fsIn());
579             fsBuilder->codeAppendf("%s = float(%s < 0.0);", edgeAlpha.c_str(), edgeAlpha.c_str());
580             break;
581         }
582         default:
583             SkFAIL("Shouldn't get here");
584     }
585
586
587     fsBuilder->codeAppendf("%s = vec4(%s);", args.fOutputCoverage, edgeAlpha.c_str());
588 }
589
590 void GrGLCubicEffect::GenKey(const GrGeometryProcessor& gp,
591                              const GrGLSLCaps&,
592                              GrProcessorKeyBuilder* b) {
593     const GrCubicEffect& ce = gp.cast<GrCubicEffect>();
594     uint32_t key = ce.isAntiAliased() ? (ce.isFilled() ? 0x0 : 0x1) : 0x2;
595     key |= ce.color() != GrColor_ILLEGAL ? 0x4 : 0x8;
596     key |= ComputePosKey(ce.viewMatrix()) << 5;
597     b->add32(key);
598 }
599
600 //////////////////////////////////////////////////////////////////////////////
601
602 GrCubicEffect::~GrCubicEffect() {}
603
604 void GrCubicEffect::getGLProcessorKey(const GrGLSLCaps& caps, GrProcessorKeyBuilder* b) const {
605     GrGLCubicEffect::GenKey(*this, caps, b);
606 }
607
608 GrGLPrimitiveProcessor* GrCubicEffect::createGLInstance(const GrGLSLCaps&) const {
609     return new GrGLCubicEffect(*this);
610 }
611
612 GrCubicEffect::GrCubicEffect(GrColor color, const SkMatrix& viewMatrix,
613                              GrPrimitiveEdgeType edgeType)
614     : fColor(color)
615     , fViewMatrix(viewMatrix)
616     , fEdgeType(edgeType) {
617     this->initClassID<GrCubicEffect>();
618     fInPosition = &this->addVertexAttrib(Attribute("inPosition", kVec2f_GrVertexAttribType,
619                                                    kHigh_GrSLPrecision));
620     fInCubicCoeffs = &this->addVertexAttrib(Attribute("inCubicCoeffs",
621                                                         kVec4f_GrVertexAttribType));
622 }
623
624 //////////////////////////////////////////////////////////////////////////////
625
626 GR_DEFINE_GEOMETRY_PROCESSOR_TEST(GrCubicEffect);
627
628 const GrGeometryProcessor* GrCubicEffect::TestCreate(GrProcessorTestData* d) {
629     GrGeometryProcessor* gp;
630     do {
631         GrPrimitiveEdgeType edgeType =
632                 static_cast<GrPrimitiveEdgeType>(
633                         d->fRandom->nextULessThan(kGrProcessorEdgeTypeCnt));
634         gp = GrCubicEffect::Create(GrRandomColor(d->fRandom),
635                                    GrTest::TestMatrix(d->fRandom), edgeType, *d->fCaps);
636     } while (nullptr == gp);
637     return gp;
638 }
639