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 GrSweepGradientLayout.fp; do not modify.
10  **************************************************************************************************/
11 #include "GrSweepGradientLayout.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 GrGLSLSweepGradientLayout : public GrGLSLFragmentProcessor {
21 public:
GrGLSLSweepGradientLayout()22     GrGLSLSweepGradientLayout() {}
emitCode(EmitArgs & args)23     void emitCode(EmitArgs& args) override {
24         GrGLSLFPFragmentBuilder* fragBuilder = args.fFragBuilder;
25         const GrSweepGradientLayout& _outer = args.fFp.cast<GrSweepGradientLayout>();
26         (void)_outer;
27         auto bias = _outer.bias;
28         (void)bias;
29         auto scale = _outer.scale;
30         (void)scale;
31         biasVar = args.fUniformHandler->addUniform(&_outer, kFragment_GrShaderFlag, kHalf_GrSLType,
32                                                    "bias");
33         scaleVar = args.fUniformHandler->addUniform(&_outer, kFragment_GrShaderFlag, kHalf_GrSLType,
34                                                     "scale");
35         fragBuilder->codeAppendf(
36                 R"SkSL(half angle = sk_Caps.atan2ImplementedAsAtanYOverX ? half(2.0 * atan(-%s.y, length(%s) - %s.x)) : half(atan(-%s.y, -%s.x));
37 half t = ((angle * 0.15915493667125702 + 0.5) + %s) * %s;
38 return half4(t, 1.0, 0.0, 0.0);
39 )SkSL",
40                 args.fSampleCoord, args.fSampleCoord, args.fSampleCoord, args.fSampleCoord,
41                 args.fSampleCoord, args.fUniformHandler->getUniformCStr(biasVar),
42                 args.fUniformHandler->getUniformCStr(scaleVar));
43     }
44 
45 private:
onSetData(const GrGLSLProgramDataManager & pdman,const GrFragmentProcessor & _proc)46     void onSetData(const GrGLSLProgramDataManager& pdman,
47                    const GrFragmentProcessor& _proc) override {
48         const GrSweepGradientLayout& _outer = _proc.cast<GrSweepGradientLayout>();
49         {
50             float biasValue = _outer.bias;
51             if (biasPrev != biasValue) {
52                 biasPrev = biasValue;
53                 pdman.set1f(biasVar, biasValue);
54             }
55             float scaleValue = _outer.scale;
56             if (scalePrev != scaleValue) {
57                 scalePrev = scaleValue;
58                 pdman.set1f(scaleVar, scaleValue);
59             }
60         }
61     }
62     float biasPrev = SK_FloatNaN;
63     float scalePrev = SK_FloatNaN;
64     UniformHandle biasVar;
65     UniformHandle scaleVar;
66 };
onCreateGLSLInstance() const67 GrGLSLFragmentProcessor* GrSweepGradientLayout::onCreateGLSLInstance() const {
68     return new GrGLSLSweepGradientLayout();
69 }
onGetGLSLProcessorKey(const GrShaderCaps & caps,GrProcessorKeyBuilder * b) const70 void GrSweepGradientLayout::onGetGLSLProcessorKey(const GrShaderCaps& caps,
71                                                   GrProcessorKeyBuilder* b) const {}
onIsEqual(const GrFragmentProcessor & other) const72 bool GrSweepGradientLayout::onIsEqual(const GrFragmentProcessor& other) const {
73     const GrSweepGradientLayout& that = other.cast<GrSweepGradientLayout>();
74     (void)that;
75     if (bias != that.bias) return false;
76     if (scale != that.scale) return false;
77     return true;
78 }
usesExplicitReturn() const79 bool GrSweepGradientLayout::usesExplicitReturn() const { return true; }
GrSweepGradientLayout(const GrSweepGradientLayout & src)80 GrSweepGradientLayout::GrSweepGradientLayout(const GrSweepGradientLayout& src)
81         : INHERITED(kGrSweepGradientLayout_ClassID, src.optimizationFlags())
82         , bias(src.bias)
83         , scale(src.scale) {
84     this->cloneAndRegisterAllChildProcessors(src);
85     this->setUsesSampleCoordsDirectly();
86 }
clone() const87 std::unique_ptr<GrFragmentProcessor> GrSweepGradientLayout::clone() const {
88     return std::make_unique<GrSweepGradientLayout>(*this);
89 }
90 #if GR_TEST_UTILS
onDumpInfo() const91 SkString GrSweepGradientLayout::onDumpInfo() const {
92     return SkStringPrintf("(bias=%f, scale=%f)", bias, scale);
93 }
94 #endif
95 GR_DEFINE_FRAGMENT_PROCESSOR_TEST(GrSweepGradientLayout);
96 #if GR_TEST_UTILS
TestCreate(GrProcessorTestData * d)97 std::unique_ptr<GrFragmentProcessor> GrSweepGradientLayout::TestCreate(GrProcessorTestData* d) {
98     SkScalar scale = GrGradientShader::RandomParams::kGradientScale;
99     SkPoint center = {d->fRandom->nextRangeScalar(0.0f, scale),
100                       d->fRandom->nextRangeScalar(0.0f, scale)};
101 
102     GrGradientShader::RandomParams params(d->fRandom);
103     auto shader = params.fUseColors4f
104                           ? SkGradientShader::MakeSweep(center.fX, center.fY, params.fColors4f,
105                                                         params.fColorSpace, params.fStops,
106                                                         params.fColorCount)
107                           : SkGradientShader::MakeSweep(center.fX, center.fY, params.fColors,
108                                                         params.fStops, params.fColorCount);
109     GrTest::TestAsFPArgs asFPArgs(d);
110     std::unique_ptr<GrFragmentProcessor> fp = as_SB(shader)->asFragmentProcessor(asFPArgs.args());
111     SkASSERT_RELEASE(fp);
112     return fp;
113 }
114 #endif
115 
Make(const SkSweepGradient & grad,const GrFPArgs & args)116 std::unique_ptr<GrFragmentProcessor> GrSweepGradientLayout::Make(const SkSweepGradient& grad,
117                                                                  const GrFPArgs& args) {
118     SkMatrix matrix;
119     if (!grad.totalLocalMatrix(args.fPreLocalMatrix)->invert(&matrix)) {
120         return nullptr;
121     }
122     matrix.postConcat(grad.getGradientMatrix());
123     return GrMatrixEffect::Make(
124             matrix, std::unique_ptr<GrFragmentProcessor>(
125                             new GrSweepGradientLayout(grad.getTBias(), grad.getTScale())));
126 }
127