Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 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 GrDualIntervalGradientColorizer.fp; do not modify. |
| 10 | **************************************************************************************************/ |
| 11 | #include "GrDualIntervalGradientColorizer.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 12 | |
John Stiles | 45f5b03 | 2020-07-27 17:31:29 -0400 | [diff] [blame] | 13 | #include "src/core/SkUtils.h" |
Greg Daniel | 456f9b5 | 2020-03-05 19:14:18 +0000 | [diff] [blame] | 14 | #include "src/gpu/GrTexture.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 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" |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 20 | class GrGLSLDualIntervalGradientColorizer : public GrGLSLFragmentProcessor { |
| 21 | public: |
| 22 | GrGLSLDualIntervalGradientColorizer() {} |
| 23 | void emitCode(EmitArgs& args) override { |
Mike Klein | d6ab77a | 2019-03-21 08:18:24 -0500 | [diff] [blame] | 24 | GrGLSLFPFragmentBuilder* fragBuilder = args.fFragBuilder; |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 25 | const GrDualIntervalGradientColorizer& _outer = |
| 26 | args.fFp.cast<GrDualIntervalGradientColorizer>(); |
| 27 | (void)_outer; |
Ethan Nicholas | bcd51e8 | 2019-04-09 10:40:41 -0400 | [diff] [blame] | 28 | auto scale01 = _outer.scale01; |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 29 | (void)scale01; |
Ethan Nicholas | bcd51e8 | 2019-04-09 10:40:41 -0400 | [diff] [blame] | 30 | auto bias01 = _outer.bias01; |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 31 | (void)bias01; |
Ethan Nicholas | bcd51e8 | 2019-04-09 10:40:41 -0400 | [diff] [blame] | 32 | auto scale23 = _outer.scale23; |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 33 | (void)scale23; |
Ethan Nicholas | bcd51e8 | 2019-04-09 10:40:41 -0400 | [diff] [blame] | 34 | auto bias23 = _outer.bias23; |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 35 | (void)bias23; |
Ethan Nicholas | bcd51e8 | 2019-04-09 10:40:41 -0400 | [diff] [blame] | 36 | auto threshold = _outer.threshold; |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 37 | (void)threshold; |
Ethan Nicholas | 16464c3 | 2020-04-06 13:53:05 -0400 | [diff] [blame] | 38 | scale01Var = args.fUniformHandler->addUniform(&_outer, kFragment_GrShaderFlag, |
| 39 | kFloat4_GrSLType, "scale01"); |
| 40 | bias01Var = args.fUniformHandler->addUniform(&_outer, kFragment_GrShaderFlag, |
| 41 | kFloat4_GrSLType, "bias01"); |
| 42 | scale23Var = args.fUniformHandler->addUniform(&_outer, kFragment_GrShaderFlag, |
| 43 | kFloat4_GrSLType, "scale23"); |
| 44 | bias23Var = args.fUniformHandler->addUniform(&_outer, kFragment_GrShaderFlag, |
| 45 | kFloat4_GrSLType, "bias23"); |
| 46 | thresholdVar = args.fUniformHandler->addUniform(&_outer, kFragment_GrShaderFlag, |
| 47 | kHalf_GrSLType, "threshold"); |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 48 | fragBuilder->codeAppendf( |
John Stiles | 6d3996c | 2020-07-31 12:33:31 -0400 | [diff] [blame] | 49 | R"SkSL(half t = half(%s.x); |
John Stiles | 5081942 | 2020-06-18 13:00:38 -0400 | [diff] [blame] | 50 | float4 scale, bias; |
| 51 | if (t < %s) { |
| 52 | scale = %s; |
| 53 | bias = %s; |
| 54 | } else { |
| 55 | scale = %s; |
| 56 | bias = %s; |
| 57 | } |
| 58 | %s = half4(float(t) * scale + bias); |
| 59 | )SkSL", |
John Stiles | 6d3996c | 2020-07-31 12:33:31 -0400 | [diff] [blame] | 60 | args.fSampleCoord, args.fUniformHandler->getUniformCStr(thresholdVar), |
Ethan Nicholas | bcd51e8 | 2019-04-09 10:40:41 -0400 | [diff] [blame] | 61 | args.fUniformHandler->getUniformCStr(scale01Var), |
| 62 | args.fUniformHandler->getUniformCStr(bias01Var), |
| 63 | args.fUniformHandler->getUniformCStr(scale23Var), |
| 64 | args.fUniformHandler->getUniformCStr(bias23Var), args.fOutputColor); |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 65 | } |
| 66 | |
| 67 | private: |
| 68 | void onSetData(const GrGLSLProgramDataManager& pdman, |
Mike Klein | d6ab77a | 2019-03-21 08:18:24 -0500 | [diff] [blame] | 69 | const GrFragmentProcessor& _proc) override { |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 70 | const GrDualIntervalGradientColorizer& _outer = |
| 71 | _proc.cast<GrDualIntervalGradientColorizer>(); |
| 72 | { |
Ethan Nicholas | bcd51e8 | 2019-04-09 10:40:41 -0400 | [diff] [blame] | 73 | const SkPMColor4f& scale01Value = _outer.scale01; |
| 74 | if (scale01Prev != scale01Value) { |
| 75 | scale01Prev = scale01Value; |
| 76 | pdman.set4fv(scale01Var, 1, scale01Value.vec()); |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 77 | } |
Ethan Nicholas | bcd51e8 | 2019-04-09 10:40:41 -0400 | [diff] [blame] | 78 | const SkPMColor4f& bias01Value = _outer.bias01; |
| 79 | if (bias01Prev != bias01Value) { |
| 80 | bias01Prev = bias01Value; |
| 81 | pdman.set4fv(bias01Var, 1, bias01Value.vec()); |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 82 | } |
Ethan Nicholas | bcd51e8 | 2019-04-09 10:40:41 -0400 | [diff] [blame] | 83 | const SkPMColor4f& scale23Value = _outer.scale23; |
| 84 | if (scale23Prev != scale23Value) { |
| 85 | scale23Prev = scale23Value; |
| 86 | pdman.set4fv(scale23Var, 1, scale23Value.vec()); |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 87 | } |
Ethan Nicholas | bcd51e8 | 2019-04-09 10:40:41 -0400 | [diff] [blame] | 88 | const SkPMColor4f& bias23Value = _outer.bias23; |
| 89 | if (bias23Prev != bias23Value) { |
| 90 | bias23Prev = bias23Value; |
| 91 | pdman.set4fv(bias23Var, 1, bias23Value.vec()); |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 92 | } |
Ethan Nicholas | bcd51e8 | 2019-04-09 10:40:41 -0400 | [diff] [blame] | 93 | float thresholdValue = _outer.threshold; |
| 94 | if (thresholdPrev != thresholdValue) { |
| 95 | thresholdPrev = thresholdValue; |
| 96 | pdman.set1f(thresholdVar, thresholdValue); |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 97 | } |
| 98 | } |
| 99 | } |
Ethan Nicholas | bcd51e8 | 2019-04-09 10:40:41 -0400 | [diff] [blame] | 100 | SkPMColor4f scale01Prev = {SK_FloatNaN, SK_FloatNaN, SK_FloatNaN, SK_FloatNaN}; |
| 101 | SkPMColor4f bias01Prev = {SK_FloatNaN, SK_FloatNaN, SK_FloatNaN, SK_FloatNaN}; |
| 102 | SkPMColor4f scale23Prev = {SK_FloatNaN, SK_FloatNaN, SK_FloatNaN, SK_FloatNaN}; |
| 103 | SkPMColor4f bias23Prev = {SK_FloatNaN, SK_FloatNaN, SK_FloatNaN, SK_FloatNaN}; |
| 104 | float thresholdPrev = SK_FloatNaN; |
| 105 | UniformHandle scale01Var; |
| 106 | UniformHandle bias01Var; |
| 107 | UniformHandle scale23Var; |
| 108 | UniformHandle bias23Var; |
| 109 | UniformHandle thresholdVar; |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 110 | }; |
| 111 | GrGLSLFragmentProcessor* GrDualIntervalGradientColorizer::onCreateGLSLInstance() const { |
| 112 | return new GrGLSLDualIntervalGradientColorizer(); |
| 113 | } |
Mike Klein | d6ab77a | 2019-03-21 08:18:24 -0500 | [diff] [blame] | 114 | void GrDualIntervalGradientColorizer::onGetGLSLProcessorKey(const GrShaderCaps& caps, |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 115 | GrProcessorKeyBuilder* b) const {} |
| 116 | bool GrDualIntervalGradientColorizer::onIsEqual(const GrFragmentProcessor& other) const { |
| 117 | const GrDualIntervalGradientColorizer& that = other.cast<GrDualIntervalGradientColorizer>(); |
| 118 | (void)that; |
Ethan Nicholas | bcd51e8 | 2019-04-09 10:40:41 -0400 | [diff] [blame] | 119 | if (scale01 != that.scale01) return false; |
| 120 | if (bias01 != that.bias01) return false; |
| 121 | if (scale23 != that.scale23) return false; |
| 122 | if (bias23 != that.bias23) return false; |
| 123 | if (threshold != that.threshold) return false; |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 124 | return true; |
| 125 | } |
| 126 | GrDualIntervalGradientColorizer::GrDualIntervalGradientColorizer( |
| 127 | const GrDualIntervalGradientColorizer& src) |
| 128 | : INHERITED(kGrDualIntervalGradientColorizer_ClassID, src.optimizationFlags()) |
Ethan Nicholas | bcd51e8 | 2019-04-09 10:40:41 -0400 | [diff] [blame] | 129 | , scale01(src.scale01) |
| 130 | , bias01(src.bias01) |
| 131 | , scale23(src.scale23) |
| 132 | , bias23(src.bias23) |
Brian Osman | 12c5d29 | 2020-07-13 16:11:35 -0400 | [diff] [blame] | 133 | , threshold(src.threshold) { |
| 134 | this->cloneAndRegisterAllChildProcessors(src); |
John Stiles | 6d3996c | 2020-07-31 12:33:31 -0400 | [diff] [blame] | 135 | this->setUsesSampleCoordsDirectly(); |
Brian Osman | 12c5d29 | 2020-07-13 16:11:35 -0400 | [diff] [blame] | 136 | } |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 137 | std::unique_ptr<GrFragmentProcessor> GrDualIntervalGradientColorizer::clone() const { |
John Stiles | fbd050b | 2020-08-03 13:21:46 -0400 | [diff] [blame^] | 138 | return std::make_unique<GrDualIntervalGradientColorizer>(*this); |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 139 | } |
| 140 | |
Brian Osman | 021ed51 | 2018-10-16 15:19:44 -0400 | [diff] [blame] | 141 | std::unique_ptr<GrFragmentProcessor> GrDualIntervalGradientColorizer::Make(const SkPMColor4f& c0, |
| 142 | const SkPMColor4f& c1, |
| 143 | const SkPMColor4f& c2, |
| 144 | const SkPMColor4f& c3, |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 145 | float threshold) { |
| 146 | // Derive scale and biases from the 4 colors and threshold |
Mike Klein | d6ab77a | 2019-03-21 08:18:24 -0500 | [diff] [blame] | 147 | auto vc0 = Sk4f::Load(c0.vec()); |
| 148 | auto vc1 = Sk4f::Load(c1.vec()); |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 149 | auto scale01 = (vc1 - vc0) / threshold; |
| 150 | // bias01 = c0 |
| 151 | |
Mike Klein | d6ab77a | 2019-03-21 08:18:24 -0500 | [diff] [blame] | 152 | auto vc2 = Sk4f::Load(c2.vec()); |
| 153 | auto vc3 = Sk4f::Load(c3.vec()); |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 154 | auto scale23 = (vc3 - vc2) / (1 - threshold); |
Mike Klein | d6ab77a | 2019-03-21 08:18:24 -0500 | [diff] [blame] | 155 | auto bias23 = vc2 - threshold * scale23; |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 156 | |
Mike Klein | d6ab77a | 2019-03-21 08:18:24 -0500 | [diff] [blame] | 157 | return std::unique_ptr<GrFragmentProcessor>(new GrDualIntervalGradientColorizer( |
| 158 | {scale01[0], scale01[1], scale01[2], scale01[3]}, c0, |
| 159 | {scale23[0], scale23[1], scale23[2], scale23[3]}, |
| 160 | {bias23[0], bias23[1], bias23[2], bias23[3]}, threshold)); |
Michael Ludwig | 0495f7a | 2018-09-12 15:23:33 -0400 | [diff] [blame] | 161 | } |