1 /*
2  * Copyright 2018 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 /**************************************************************************************************
9  *** This file was autogenerated from GrAARectEffect.fp; do not modify.
10  **************************************************************************************************/
11 #include "GrAARectEffect.h"
12 
13 #include "src/core/SkUtils.h"
14 #include "src/gpu/GrTexture.h"
15 #include "src/gpu/glsl/GrGLSLFragmentProcessor.h"
16 #include "src/gpu/glsl/GrGLSLFragmentShaderBuilder.h"
17 #include "src/gpu/glsl/GrGLSLProgramBuilder.h"
18 #include "src/sksl/SkSLCPP.h"
19 #include "src/sksl/SkSLUtil.h"
20 class GrGLSLAARectEffect : public GrGLSLFragmentProcessor {
21 public:
GrGLSLAARectEffect()22     GrGLSLAARectEffect() {}
emitCode(EmitArgs & args)23     void emitCode(EmitArgs& args) override {
24         GrGLSLFPFragmentBuilder* fragBuilder = args.fFragBuilder;
25         const GrAARectEffect& _outer = args.fFp.cast<GrAARectEffect>();
26         (void)_outer;
27         auto edgeType = _outer.edgeType;
28         (void)edgeType;
29         auto rect = _outer.rect;
30         (void)rect;
31         prevRect = float4(-1.0);
32         rectUniformVar = args.fUniformHandler->addUniform(
33                 &_outer, kFragment_GrShaderFlag, kFloat4_GrSLType, "rectUniform");
34         fragBuilder->codeAppendf(
35                 R"SkSL(float4 prevRect = float4(%f, %f, %f, %f);
36 half coverage;
37 @switch (%d) {
38     case 0:
39     case 2:
40         coverage = half(all(greaterThan(float4(sk_FragCoord.xy, %s.zw), float4(%s.xy, sk_FragCoord.xy))) ? 1 : 0);
41         break;
42     default:
43         half4 dists4 = clamp(half4(1.0, 1.0, -1.0, -1.0) * half4(sk_FragCoord.xyxy - %s), 0.0, 1.0);
44         half2 dists2 = (dists4.xy + dists4.zw) - 1.0;
45         coverage = dists2.x * dists2.y;
46 }
47 @if (%d == 2 || %d == 3) {
48     coverage = 1.0 - coverage;
49 })SkSL",
50                 prevRect.left(),
51                 prevRect.top(),
52                 prevRect.right(),
53                 prevRect.bottom(),
54                 (int)_outer.edgeType,
55                 args.fUniformHandler->getUniformCStr(rectUniformVar),
56                 args.fUniformHandler->getUniformCStr(rectUniformVar),
57                 args.fUniformHandler->getUniformCStr(rectUniformVar),
58                 (int)_outer.edgeType,
59                 (int)_outer.edgeType);
60         SkString _sample0 = this->invokeChild(0, args);
61         fragBuilder->codeAppendf(
62                 R"SkSL(
63 return %s * coverage;
64 )SkSL",
65                 _sample0.c_str());
66     }
67 
68 private:
onSetData(const GrGLSLProgramDataManager & pdman,const GrFragmentProcessor & _proc)69     void onSetData(const GrGLSLProgramDataManager& pdman,
70                    const GrFragmentProcessor& _proc) override {
71         const GrAARectEffect& _outer = _proc.cast<GrAARectEffect>();
72         auto edgeType = _outer.edgeType;
73         (void)edgeType;
74         auto rect = _outer.rect;
75         (void)rect;
76         UniformHandle& rectUniform = rectUniformVar;
77         (void)rectUniform;
78 
79         SkASSERT(rect.isSorted());
80         // The AA math in the shader evaluates to 0 at the uploaded coordinates, so outset by 0.5
81         // to interpolate from 0 at a half pixel inset and 1 at a half pixel outset of rect.
82         const SkRect& newRect =
83                 GrProcessorEdgeTypeIsAA(edgeType) ? rect.makeOutset(.5f, .5f) : rect;
84         if (newRect != prevRect) {
85             pdman.set4f(rectUniform, newRect.fLeft, newRect.fTop, newRect.fRight, newRect.fBottom);
86             prevRect = newRect;
87         }
88     }
89     SkRect prevRect = float4(0);
90     UniformHandle rectUniformVar;
91 };
onMakeProgramImpl() const92 std::unique_ptr<GrGLSLFragmentProcessor> GrAARectEffect::onMakeProgramImpl() const {
93     return std::make_unique<GrGLSLAARectEffect>();
94 }
onGetGLSLProcessorKey(const GrShaderCaps & caps,GrProcessorKeyBuilder * b) const95 void GrAARectEffect::onGetGLSLProcessorKey(const GrShaderCaps& caps,
96                                            GrProcessorKeyBuilder* b) const {
97     b->addBits(2, (uint32_t)edgeType, "edgeType");
98 }
onIsEqual(const GrFragmentProcessor & other) const99 bool GrAARectEffect::onIsEqual(const GrFragmentProcessor& other) const {
100     const GrAARectEffect& that = other.cast<GrAARectEffect>();
101     (void)that;
102     if (edgeType != that.edgeType) return false;
103     if (rect != that.rect) return false;
104     return true;
105 }
GrAARectEffect(const GrAARectEffect & src)106 GrAARectEffect::GrAARectEffect(const GrAARectEffect& src)
107         : INHERITED(kGrAARectEffect_ClassID, src.optimizationFlags())
108         , edgeType(src.edgeType)
109         , rect(src.rect) {
110     this->cloneAndRegisterAllChildProcessors(src);
111 }
clone() const112 std::unique_ptr<GrFragmentProcessor> GrAARectEffect::clone() const {
113     return std::make_unique<GrAARectEffect>(*this);
114 }
115 #if GR_TEST_UTILS
onDumpInfo() const116 SkString GrAARectEffect::onDumpInfo() const {
117     return SkStringPrintf("(edgeType=%d, rect=float4(%f, %f, %f, %f))",
118                           (int)edgeType,
119                           rect.left(),
120                           rect.top(),
121                           rect.right(),
122                           rect.bottom());
123 }
124 #endif
125 GR_DEFINE_FRAGMENT_PROCESSOR_TEST(GrAARectEffect);
126 #if GR_TEST_UTILS
TestCreate(GrProcessorTestData * d)127 std::unique_ptr<GrFragmentProcessor> GrAARectEffect::TestCreate(GrProcessorTestData* d) {
128     SkScalar l = d->fRandom->nextSScalar1();
129     SkScalar t = d->fRandom->nextSScalar1();
130     SkScalar r = d->fRandom->nextSScalar1();
131     SkScalar b = d->fRandom->nextSScalar1();
132     SkRect rect = SkRect::MakeLTRB(l, t, r, b);
133     rect.sort();
134     GrClipEdgeType edgeType =
135             static_cast<GrClipEdgeType>(d->fRandom->nextULessThan(kGrClipEdgeTypeCnt));
136 
137     return GrAARectEffect::Make(d->inputFP(), edgeType, rect);
138 }
139 #endif
140