joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2014 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 | #include "GrMatrixConvolutionEffect.h" |
Robert Phillips | 296b1cc | 2017-03-15 10:42:12 -0400 | [diff] [blame] | 8 | |
Robert Phillips | 646e429 | 2017-06-13 12:44:56 -0400 | [diff] [blame] | 9 | #include "GrTexture.h" |
Hal Canary | 6f6961e | 2017-01-31 13:50:44 -0500 | [diff] [blame] | 10 | #include "GrTextureProxy.h" |
egdaniel | 64c4728 | 2015-11-13 06:54:19 -0800 | [diff] [blame] | 11 | #include "glsl/GrGLSLFragmentProcessor.h" |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 12 | #include "glsl/GrGLSLFragmentShaderBuilder.h" |
egdaniel | 018fb62 | 2015-10-28 07:26:40 -0700 | [diff] [blame] | 13 | #include "glsl/GrGLSLProgramDataManager.h" |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 14 | #include "glsl/GrGLSLUniformHandler.h" |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 15 | |
egdaniel | 64c4728 | 2015-11-13 06:54:19 -0800 | [diff] [blame] | 16 | class GrGLMatrixConvolutionEffect : public GrGLSLFragmentProcessor { |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 17 | public: |
robertphillips | 9cdb992 | 2016-02-03 12:25:40 -0800 | [diff] [blame] | 18 | void emitCode(EmitArgs&) override; |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 19 | |
Brian Salomon | 94efbf5 | 2016-11-29 13:43:05 -0500 | [diff] [blame] | 20 | static inline void GenKey(const GrProcessor&, const GrShaderCaps&, GrProcessorKeyBuilder*); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 21 | |
wangyix | b1daa86 | 2015-08-18 11:29:31 -0700 | [diff] [blame] | 22 | protected: |
Brian Salomon | ab015ef | 2017-04-04 10:15:51 -0400 | [diff] [blame] | 23 | void onSetData(const GrGLSLProgramDataManager&, const GrFragmentProcessor&) override; |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 24 | |
| 25 | private: |
egdaniel | 018fb62 | 2015-10-28 07:26:40 -0700 | [diff] [blame] | 26 | typedef GrGLSLProgramDataManager::UniformHandle UniformHandle; |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 27 | |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 28 | UniformHandle fKernelUni; |
| 29 | UniformHandle fImageIncrementUni; |
| 30 | UniformHandle fKernelOffsetUni; |
| 31 | UniformHandle fGainUni; |
| 32 | UniformHandle fBiasUni; |
| 33 | GrTextureDomain::GLDomain fDomain; |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 34 | |
egdaniel | 64c4728 | 2015-11-13 06:54:19 -0800 | [diff] [blame] | 35 | typedef GrGLSLFragmentProcessor INHERITED; |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 36 | }; |
| 37 | |
wangyix | 7c157a9 | 2015-07-22 15:08:53 -0700 | [diff] [blame] | 38 | void GrGLMatrixConvolutionEffect::emitCode(EmitArgs& args) { |
robertphillips | bf536af | 2016-02-04 06:11:53 -0800 | [diff] [blame] | 39 | const GrMatrixConvolutionEffect& mce = args.fFp.cast<GrMatrixConvolutionEffect>(); |
| 40 | const GrTextureDomain& domain = mce.domain(); |
| 41 | |
| 42 | int kWidth = mce.kernelSize().width(); |
| 43 | int kHeight = mce.kernelSize().height(); |
| 44 | |
egdaniel | e506ee6 | 2016-03-02 14:29:10 -0800 | [diff] [blame] | 45 | int arrayCount = (kWidth * kHeight + 3) / 4; |
| 46 | SkASSERT(4 * arrayCount >= kWidth * kHeight); |
jvanverth | 78d6eb0 | 2016-03-02 13:21:16 -0800 | [diff] [blame] | 47 | |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 48 | GrGLSLUniformHandler* uniformHandler = args.fUniformHandler; |
Ethan Nicholas | f7b8820 | 2017-09-18 14:10:39 -0400 | [diff] [blame] | 49 | fImageIncrementUni = uniformHandler->addUniform(kFragment_GrShaderFlag, kHalf2_GrSLType, |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 50 | "ImageIncrement"); |
Ethan Nicholas | f7b8820 | 2017-09-18 14:10:39 -0400 | [diff] [blame] | 51 | fKernelUni = uniformHandler->addUniformArray(kFragment_GrShaderFlag, kHalf4_GrSLType, |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 52 | "Kernel", |
jvanverth | 78d6eb0 | 2016-03-02 13:21:16 -0800 | [diff] [blame] | 53 | arrayCount); |
Ethan Nicholas | f7b8820 | 2017-09-18 14:10:39 -0400 | [diff] [blame] | 54 | fKernelOffsetUni = uniformHandler->addUniform(kFragment_GrShaderFlag, kHalf2_GrSLType, |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 55 | "KernelOffset"); |
Ethan Nicholas | f7b8820 | 2017-09-18 14:10:39 -0400 | [diff] [blame] | 56 | fGainUni = uniformHandler->addUniform(kFragment_GrShaderFlag, kHalf_GrSLType, "Gain"); |
| 57 | fBiasUni = uniformHandler->addUniform(kFragment_GrShaderFlag, kHalf_GrSLType, "Bias"); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 58 | |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 59 | const char* kernelOffset = uniformHandler->getUniformCStr(fKernelOffsetUni); |
| 60 | const char* imgInc = uniformHandler->getUniformCStr(fImageIncrementUni); |
| 61 | const char* kernel = uniformHandler->getUniformCStr(fKernelUni); |
| 62 | const char* gain = uniformHandler->getUniformCStr(fGainUni); |
| 63 | const char* bias = uniformHandler->getUniformCStr(fBiasUni); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 64 | |
cdalton | 8528541 | 2016-02-18 12:37:07 -0800 | [diff] [blame] | 65 | GrGLSLFPFragmentBuilder* fragBuilder = args.fFragBuilder; |
bsalomon | 1a1aa93 | 2016-09-12 09:30:36 -0700 | [diff] [blame] | 66 | SkString coords2D = fragBuilder->ensureCoords2D(args.fTransformedCoords[0]); |
Ethan Nicholas | f7b8820 | 2017-09-18 14:10:39 -0400 | [diff] [blame] | 67 | fragBuilder->codeAppend("half4 sum = half4(0, 0, 0, 0);"); |
Ethan Nicholas | 8aa4569 | 2017-09-20 11:24:15 -0400 | [diff] [blame] | 68 | fragBuilder->codeAppendf("float2 coord = %s - %s * %s;", coords2D.c_str(), kernelOffset, imgInc); |
Ethan Nicholas | f7b8820 | 2017-09-18 14:10:39 -0400 | [diff] [blame] | 69 | fragBuilder->codeAppend("half4 c;"); |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 70 | |
jvanverth | 78d6eb0 | 2016-03-02 13:21:16 -0800 | [diff] [blame] | 71 | const char* kVecSuffix[4] = { ".x", ".y", ".z", ".w" }; |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 72 | for (int y = 0; y < kHeight; y++) { |
| 73 | for (int x = 0; x < kWidth; x++) { |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 74 | GrGLSLShaderBuilder::ShaderBlock block(fragBuilder); |
jvanverth | 78d6eb0 | 2016-03-02 13:21:16 -0800 | [diff] [blame] | 75 | int offset = y*kWidth + x; |
| 76 | |
Ethan Nicholas | f7b8820 | 2017-09-18 14:10:39 -0400 | [diff] [blame] | 77 | fragBuilder->codeAppendf("half k = %s[%d]%s;", kernel, offset / 4, |
jvanverth | 78d6eb0 | 2016-03-02 13:21:16 -0800 | [diff] [blame] | 78 | kVecSuffix[offset & 0x3]); |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 79 | SkString coord; |
Ethan Nicholas | f7b8820 | 2017-09-18 14:10:39 -0400 | [diff] [blame] | 80 | coord.printf("coord + half2(%d, %d) * %s", x, y, imgInc); |
egdaniel | a2e3e0f | 2015-11-19 07:23:45 -0800 | [diff] [blame] | 81 | fDomain.sampleTexture(fragBuilder, |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 82 | uniformHandler, |
Brian Salomon | 1edc5b9 | 2016-11-29 13:43:46 -0500 | [diff] [blame] | 83 | args.fShaderCaps, |
egdaniel | a2e3e0f | 2015-11-19 07:23:45 -0800 | [diff] [blame] | 84 | domain, |
| 85 | "c", |
| 86 | coord, |
cdalton | 3f6f76f | 2016-04-11 12:18:09 -0700 | [diff] [blame] | 87 | args.fTexSamplers[0]); |
robertphillips | bf536af | 2016-02-04 06:11:53 -0800 | [diff] [blame] | 88 | if (!mce.convolveAlpha()) { |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 89 | fragBuilder->codeAppend("c.rgb /= c.a;"); |
Ethan Nicholas | 12fb9cf | 2018-08-03 16:16:57 -0400 | [diff] [blame] | 90 | fragBuilder->codeAppend("c.rgb = saturate(c.rgb);"); |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 91 | } |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 92 | fragBuilder->codeAppend("sum += c * k;"); |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 93 | } |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 94 | } |
robertphillips | bf536af | 2016-02-04 06:11:53 -0800 | [diff] [blame] | 95 | if (mce.convolveAlpha()) { |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 96 | fragBuilder->codeAppendf("%s = sum * %s + %s;", args.fOutputColor, gain, bias); |
Ethan Nicholas | 12fb9cf | 2018-08-03 16:16:57 -0400 | [diff] [blame] | 97 | fragBuilder->codeAppendf("%s.a = saturate(%s.a);", args.fOutputColor, args.fOutputColor); |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 98 | fragBuilder->codeAppendf("%s.rgb = clamp(%s.rgb, 0.0, %s.a);", |
| 99 | args.fOutputColor, args.fOutputColor, args.fOutputColor); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 100 | } else { |
egdaniel | a2e3e0f | 2015-11-19 07:23:45 -0800 | [diff] [blame] | 101 | fDomain.sampleTexture(fragBuilder, |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 102 | uniformHandler, |
Brian Salomon | 1edc5b9 | 2016-11-29 13:43:46 -0500 | [diff] [blame] | 103 | args.fShaderCaps, |
egdaniel | a2e3e0f | 2015-11-19 07:23:45 -0800 | [diff] [blame] | 104 | domain, |
| 105 | "c", |
| 106 | coords2D, |
cdalton | 3f6f76f | 2016-04-11 12:18:09 -0700 | [diff] [blame] | 107 | args.fTexSamplers[0]); |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 108 | fragBuilder->codeAppendf("%s.a = c.a;", args.fOutputColor); |
Ethan Nicholas | 12fb9cf | 2018-08-03 16:16:57 -0400 | [diff] [blame] | 109 | fragBuilder->codeAppendf("%s.rgb = saturate(sum.rgb * %s + %s);", args.fOutputColor, gain, bias); |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 110 | fragBuilder->codeAppendf("%s.rgb *= %s.a;", args.fOutputColor, args.fOutputColor); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 111 | } |
Ethan Nicholas | 2983f40 | 2017-05-08 09:36:08 -0400 | [diff] [blame] | 112 | fragBuilder->codeAppendf("%s *= %s;\n", args.fOutputColor, args.fInputColor); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 113 | } |
| 114 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 115 | void GrGLMatrixConvolutionEffect::GenKey(const GrProcessor& processor, |
Brian Salomon | 94efbf5 | 2016-11-29 13:43:05 -0500 | [diff] [blame] | 116 | const GrShaderCaps&, GrProcessorKeyBuilder* b) { |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 117 | const GrMatrixConvolutionEffect& m = processor.cast<GrMatrixConvolutionEffect>(); |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 118 | SkASSERT(m.kernelSize().width() <= 0x7FFF && m.kernelSize().height() <= 0xFFFF); |
| 119 | uint32_t key = m.kernelSize().width() << 16 | m.kernelSize().height(); |
caryclark | 952538e | 2016-02-26 05:01:42 -0800 | [diff] [blame] | 120 | key |= m.convolveAlpha() ? 1U << 31 : 0; |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 121 | b->add32(key); |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 122 | b->add32(GrTextureDomain::GLDomain::DomainKey(m.domain())); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 123 | } |
| 124 | |
egdaniel | 018fb62 | 2015-10-28 07:26:40 -0700 | [diff] [blame] | 125 | void GrGLMatrixConvolutionEffect::onSetData(const GrGLSLProgramDataManager& pdman, |
Brian Salomon | ab015ef | 2017-04-04 10:15:51 -0400 | [diff] [blame] | 126 | const GrFragmentProcessor& processor) { |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 127 | const GrMatrixConvolutionEffect& conv = processor.cast<GrMatrixConvolutionEffect>(); |
Brian Salomon | 7da2fc7 | 2018-12-10 13:40:07 -0500 | [diff] [blame] | 128 | GrTextureProxy* proxy = conv.textureSampler(0).proxy(); |
Brian Salomon | fd98c2c | 2018-07-31 17:25:29 -0400 | [diff] [blame] | 129 | GrTexture* texture = proxy->peekTexture(); |
robertphillips | bf536af | 2016-02-04 06:11:53 -0800 | [diff] [blame] | 130 | |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 131 | float imageIncrement[2]; |
Robert Phillips | c686ce3 | 2017-07-21 14:12:29 -0400 | [diff] [blame] | 132 | float ySign = proxy->origin() == kTopLeft_GrSurfaceOrigin ? 1.0f : -1.0f; |
Brian Salomon | 0bbecb2 | 2016-11-17 11:38:22 -0500 | [diff] [blame] | 133 | imageIncrement[0] = 1.0f / texture->width(); |
| 134 | imageIncrement[1] = ySign / texture->height(); |
kkinnunen | 7510b22 | 2014-07-30 00:04:16 -0700 | [diff] [blame] | 135 | pdman.set2fv(fImageIncrementUni, 1, imageIncrement); |
| 136 | pdman.set2fv(fKernelOffsetUni, 1, conv.kernelOffset()); |
jvanverth | 78d6eb0 | 2016-03-02 13:21:16 -0800 | [diff] [blame] | 137 | int kernelCount = conv.kernelSize().width() * conv.kernelSize().height(); |
| 138 | int arrayCount = (kernelCount + 3) / 4; |
| 139 | SkASSERT(4 * arrayCount >= kernelCount); |
| 140 | pdman.set4fv(fKernelUni, arrayCount, conv.kernel()); |
kkinnunen | 7510b22 | 2014-07-30 00:04:16 -0700 | [diff] [blame] | 141 | pdman.set1f(fGainUni, conv.gain()); |
| 142 | pdman.set1f(fBiasUni, conv.bias()); |
Michael Ludwig | be315a2 | 2018-12-17 09:50:51 -0500 | [diff] [blame^] | 143 | fDomain.setData(pdman, conv.domain(), proxy, conv.textureSampler(0).samplerState()); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 144 | } |
| 145 | |
Robert Phillips | 5140f9a | 2018-05-11 16:11:45 -0400 | [diff] [blame] | 146 | GrMatrixConvolutionEffect::GrMatrixConvolutionEffect(sk_sp<GrTextureProxy> srcProxy, |
| 147 | const SkIRect& srcBounds, |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 148 | const SkISize& kernelSize, |
| 149 | const SkScalar* kernel, |
| 150 | SkScalar gain, |
| 151 | SkScalar bias, |
| 152 | const SkIPoint& kernelOffset, |
| 153 | GrTextureDomain::Mode tileMode, |
| 154 | bool convolveAlpha) |
Brian Salomon | 6cd51b5 | 2017-07-26 19:07:15 -0400 | [diff] [blame] | 155 | // To advertise either the modulation or opaqueness optimizations we'd have to examine the |
| 156 | // parameters. |
Ethan Nicholas | abff956 | 2017-10-09 10:54:08 -0400 | [diff] [blame] | 157 | : INHERITED(kGrMatrixConvolutionEffect_ClassID, kNone_OptimizationFlags) |
Robert Phillips | 5140f9a | 2018-05-11 16:11:45 -0400 | [diff] [blame] | 158 | , fCoordTransform(srcProxy.get()) |
Michael Ludwig | be315a2 | 2018-12-17 09:50:51 -0500 | [diff] [blame^] | 159 | , fDomain(srcProxy.get(), GrTextureDomain::MakeTexelDomain(srcBounds, tileMode), |
| 160 | tileMode, tileMode) |
Robert Phillips | 5140f9a | 2018-05-11 16:11:45 -0400 | [diff] [blame] | 161 | , fTextureSampler(std::move(srcProxy)) |
Brian Salomon | 6cd51b5 | 2017-07-26 19:07:15 -0400 | [diff] [blame] | 162 | , fKernelSize(kernelSize) |
| 163 | , fGain(SkScalarToFloat(gain)) |
| 164 | , fBias(SkScalarToFloat(bias) / 255.0f) |
| 165 | , fConvolveAlpha(convolveAlpha) { |
Brian Salomon | 6cd51b5 | 2017-07-26 19:07:15 -0400 | [diff] [blame] | 166 | this->addCoordTransform(&fCoordTransform); |
Brian Salomon | f7dcd76 | 2018-07-30 14:48:15 -0400 | [diff] [blame] | 167 | this->setTextureSamplerCnt(1); |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 168 | for (int i = 0; i < kernelSize.width() * kernelSize.height(); i++) { |
| 169 | fKernel[i] = SkScalarToFloat(kernel[i]); |
| 170 | } |
| 171 | fKernelOffset[0] = static_cast<float>(kernelOffset.x()); |
| 172 | fKernelOffset[1] = static_cast<float>(kernelOffset.y()); |
| 173 | } |
| 174 | |
Brian Salomon | 3f6f965 | 2017-07-28 07:34:05 -0400 | [diff] [blame] | 175 | GrMatrixConvolutionEffect::GrMatrixConvolutionEffect(const GrMatrixConvolutionEffect& that) |
Ethan Nicholas | abff956 | 2017-10-09 10:54:08 -0400 | [diff] [blame] | 176 | : INHERITED(kGrMatrixConvolutionEffect_ClassID, kNone_OptimizationFlags) |
Brian Salomon | 3f6f965 | 2017-07-28 07:34:05 -0400 | [diff] [blame] | 177 | , fCoordTransform(that.fCoordTransform) |
| 178 | , fDomain(that.fDomain) |
| 179 | , fTextureSampler(that.fTextureSampler) |
| 180 | , fKernelSize(that.fKernelSize) |
| 181 | , fGain(that.fGain) |
| 182 | , fBias(that.fBias) |
| 183 | , fConvolveAlpha(that.fConvolveAlpha) { |
Brian Salomon | 3f6f965 | 2017-07-28 07:34:05 -0400 | [diff] [blame] | 184 | this->addCoordTransform(&fCoordTransform); |
Brian Salomon | f7dcd76 | 2018-07-30 14:48:15 -0400 | [diff] [blame] | 185 | this->setTextureSamplerCnt(1); |
Brian Salomon | 3f6f965 | 2017-07-28 07:34:05 -0400 | [diff] [blame] | 186 | memcpy(fKernel, that.fKernel, sizeof(float) * fKernelSize.width() * fKernelSize.height()); |
| 187 | memcpy(fKernelOffset, that.fKernelOffset, sizeof(fKernelOffset)); |
| 188 | } |
| 189 | |
Brian Salomon | aff329b | 2017-08-11 09:40:37 -0400 | [diff] [blame] | 190 | std::unique_ptr<GrFragmentProcessor> GrMatrixConvolutionEffect::clone() const { |
| 191 | return std::unique_ptr<GrFragmentProcessor>(new GrMatrixConvolutionEffect(*this)); |
Brian Salomon | 3f6f965 | 2017-07-28 07:34:05 -0400 | [diff] [blame] | 192 | } |
| 193 | |
Brian Salomon | 94efbf5 | 2016-11-29 13:43:05 -0500 | [diff] [blame] | 194 | void GrMatrixConvolutionEffect::onGetGLSLProcessorKey(const GrShaderCaps& caps, |
egdaniel | 57d3b03 | 2015-11-13 11:57:27 -0800 | [diff] [blame] | 195 | GrProcessorKeyBuilder* b) const { |
joshualitt | eb2a676 | 2014-12-04 11:35:33 -0800 | [diff] [blame] | 196 | GrGLMatrixConvolutionEffect::GenKey(*this, caps, b); |
| 197 | } |
| 198 | |
egdaniel | 57d3b03 | 2015-11-13 11:57:27 -0800 | [diff] [blame] | 199 | GrGLSLFragmentProcessor* GrMatrixConvolutionEffect::onCreateGLSLInstance() const { |
robertphillips | bf536af | 2016-02-04 06:11:53 -0800 | [diff] [blame] | 200 | return new GrGLMatrixConvolutionEffect; |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 201 | } |
| 202 | |
bsalomon | 0e08fc1 | 2014-10-15 08:19:04 -0700 | [diff] [blame] | 203 | bool GrMatrixConvolutionEffect::onIsEqual(const GrFragmentProcessor& sBase) const { |
joshualitt | 49586be | 2014-09-16 08:21:41 -0700 | [diff] [blame] | 204 | const GrMatrixConvolutionEffect& s = sBase.cast<GrMatrixConvolutionEffect>(); |
bsalomon | 420d7e9 | 2014-10-16 09:18:09 -0700 | [diff] [blame] | 205 | return fKernelSize == s.kernelSize() && |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 206 | !memcmp(fKernel, s.kernel(), |
| 207 | fKernelSize.width() * fKernelSize.height() * sizeof(float)) && |
| 208 | fGain == s.gain() && |
| 209 | fBias == s.bias() && |
Brian Salomon | ce06e26 | 2017-08-01 16:23:40 -0400 | [diff] [blame] | 210 | !memcmp(fKernelOffset, s.kernelOffset(), sizeof(fKernelOffset)) && |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 211 | fConvolveAlpha == s.convolveAlpha() && |
| 212 | fDomain == s.domain(); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 213 | } |
| 214 | |
Robert Phillips | 4e962c6 | 2018-06-20 07:46:19 -0400 | [diff] [blame] | 215 | static void fill_in_1D_gaussian_kernel_with_stride(float* kernel, int size, int stride, |
| 216 | float twoSigmaSqrd) { |
| 217 | SkASSERT(!SkScalarNearlyZero(twoSigmaSqrd, SK_ScalarNearlyZero)); |
| 218 | |
| 219 | const float sigmaDenom = 1.0f / twoSigmaSqrd; |
| 220 | const int radius = size / 2; |
| 221 | |
| 222 | float sum = 0.0f; |
| 223 | for (int i = 0; i < size; ++i) { |
| 224 | float term = static_cast<float>(i - radius); |
| 225 | // Note that the constant term (1/(sqrt(2*pi*sigma^2)) of the Gaussian |
| 226 | // is dropped here, since we renormalize the kernel below. |
| 227 | kernel[i * stride] = sk_float_exp(-term * term * sigmaDenom); |
| 228 | sum += kernel[i * stride]; |
| 229 | } |
| 230 | // Normalize the kernel |
| 231 | float scale = 1.0f / sum; |
| 232 | for (int i = 0; i < size; ++i) { |
| 233 | kernel[i * stride] *= scale; |
| 234 | } |
| 235 | } |
| 236 | |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 237 | static void fill_in_2D_gaussian_kernel(float* kernel, int width, int height, |
| 238 | SkScalar sigmaX, SkScalar sigmaY) { |
joshualitt | 5acfea7 | 2014-08-11 13:55:34 -0700 | [diff] [blame] | 239 | SkASSERT(width * height <= MAX_KERNEL_SIZE); |
Greg Daniel | 4eda8d9 | 2018-04-03 14:03:15 -0400 | [diff] [blame] | 240 | const float twoSigmaSqrdX = 2.0f * SkScalarToFloat(SkScalarSquare(sigmaX)); |
| 241 | const float twoSigmaSqrdY = 2.0f * SkScalarToFloat(SkScalarSquare(sigmaY)); |
| 242 | |
Robert Phillips | 4e962c6 | 2018-06-20 07:46:19 -0400 | [diff] [blame] | 243 | // TODO: in all of these degenerate cases we're uploading (and using) a whole lot of zeros. |
Greg Daniel | 4eda8d9 | 2018-04-03 14:03:15 -0400 | [diff] [blame] | 244 | if (SkScalarNearlyZero(twoSigmaSqrdX, SK_ScalarNearlyZero) || |
| 245 | SkScalarNearlyZero(twoSigmaSqrdY, SK_ScalarNearlyZero)) { |
Robert Phillips | 4e962c6 | 2018-06-20 07:46:19 -0400 | [diff] [blame] | 246 | // In this case the 2D Gaussian degenerates to a 1D Gaussian (in X or Y) or a point |
| 247 | SkASSERT(3 == width || 3 == height); |
| 248 | memset(kernel, 0, width*height*sizeof(float)); |
| 249 | |
| 250 | if (SkScalarNearlyZero(twoSigmaSqrdX, SK_ScalarNearlyZero) && |
| 251 | SkScalarNearlyZero(twoSigmaSqrdY, SK_ScalarNearlyZero)) { |
| 252 | // A point |
| 253 | SkASSERT(3 == width && 3 == height); |
| 254 | kernel[4] = 1.0f; |
| 255 | } else if (SkScalarNearlyZero(twoSigmaSqrdX, SK_ScalarNearlyZero)) { |
| 256 | // A 1D Gaussian in Y |
| 257 | SkASSERT(3 == width); |
| 258 | // Down the middle column of the kernel with a stride of width |
| 259 | fill_in_1D_gaussian_kernel_with_stride(&kernel[1], height, width, twoSigmaSqrdY); |
| 260 | } else { |
| 261 | // A 1D Gaussian in X |
| 262 | SkASSERT(SkScalarNearlyZero(twoSigmaSqrdY, SK_ScalarNearlyZero)); |
| 263 | SkASSERT(3 == height); |
| 264 | // Down the middle row of the kernel with a stride of 1 |
| 265 | fill_in_1D_gaussian_kernel_with_stride(&kernel[width], width, 1, twoSigmaSqrdX); |
Greg Daniel | 4eda8d9 | 2018-04-03 14:03:15 -0400 | [diff] [blame] | 266 | } |
| 267 | return; |
| 268 | } |
| 269 | |
| 270 | const float sigmaXDenom = 1.0f / twoSigmaSqrdX; |
| 271 | const float sigmaYDenom = 1.0f / twoSigmaSqrdY; |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 272 | const int xRadius = width / 2; |
| 273 | const int yRadius = height / 2; |
| 274 | |
joshualitt | 5acfea7 | 2014-08-11 13:55:34 -0700 | [diff] [blame] | 275 | float sum = 0.0f; |
joshualitt | 5acfea7 | 2014-08-11 13:55:34 -0700 | [diff] [blame] | 276 | for (int x = 0; x < width; x++) { |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 277 | float xTerm = static_cast<float>(x - xRadius); |
| 278 | xTerm = xTerm * xTerm * sigmaXDenom; |
| 279 | for (int y = 0; y < height; y++) { |
| 280 | float yTerm = static_cast<float>(y - yRadius); |
| 281 | float xyTerm = sk_float_exp(-(xTerm + yTerm * yTerm * sigmaYDenom)); |
| 282 | // Note that the constant term (1/(sqrt(2*pi*sigma^2)) of the Gaussian |
| 283 | // is dropped here, since we renormalize the kernel below. |
| 284 | kernel[y * width + x] = xyTerm; |
| 285 | sum += xyTerm; |
| 286 | } |
joshualitt | 5acfea7 | 2014-08-11 13:55:34 -0700 | [diff] [blame] | 287 | } |
| 288 | // Normalize the kernel |
| 289 | float scale = 1.0f / sum; |
| 290 | for (int i = 0; i < width * height; ++i) { |
| 291 | kernel[i] *= scale; |
| 292 | } |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 293 | } |
| 294 | |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 295 | // Static function to create a 2D convolution |
Brian Salomon | aff329b | 2017-08-11 09:40:37 -0400 | [diff] [blame] | 296 | std::unique_ptr<GrFragmentProcessor> GrMatrixConvolutionEffect::MakeGaussian( |
Robert Phillips | 5140f9a | 2018-05-11 16:11:45 -0400 | [diff] [blame] | 297 | sk_sp<GrTextureProxy> srcProxy, |
| 298 | const SkIRect& srcBounds, |
Brian Salomon | aff329b | 2017-08-11 09:40:37 -0400 | [diff] [blame] | 299 | const SkISize& kernelSize, |
| 300 | SkScalar gain, |
| 301 | SkScalar bias, |
| 302 | const SkIPoint& kernelOffset, |
| 303 | GrTextureDomain::Mode tileMode, |
| 304 | bool convolveAlpha, |
| 305 | SkScalar sigmaX, |
| 306 | SkScalar sigmaY) { |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 307 | float kernel[MAX_KERNEL_SIZE]; |
| 308 | |
| 309 | fill_in_2D_gaussian_kernel(kernel, kernelSize.width(), kernelSize.height(), sigmaX, sigmaY); |
| 310 | |
Brian Salomon | aff329b | 2017-08-11 09:40:37 -0400 | [diff] [blame] | 311 | return std::unique_ptr<GrFragmentProcessor>( |
Robert Phillips | 5140f9a | 2018-05-11 16:11:45 -0400 | [diff] [blame] | 312 | new GrMatrixConvolutionEffect(std::move(srcProxy), srcBounds, kernelSize, kernel, |
| 313 | gain, bias, kernelOffset, tileMode, convolveAlpha)); |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 314 | } |
| 315 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 316 | GR_DEFINE_FRAGMENT_PROCESSOR_TEST(GrMatrixConvolutionEffect); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 317 | |
Hal Canary | 6f6961e | 2017-01-31 13:50:44 -0500 | [diff] [blame] | 318 | #if GR_TEST_UTILS |
Brian Salomon | aff329b | 2017-08-11 09:40:37 -0400 | [diff] [blame] | 319 | std::unique_ptr<GrFragmentProcessor> GrMatrixConvolutionEffect::TestCreate(GrProcessorTestData* d) { |
Robert Phillips | dbc8eeb | 2017-02-21 10:04:31 -0500 | [diff] [blame] | 320 | int texIdx = d->fRandom->nextBool() ? GrProcessorUnitTest::kSkiaPMTextureIdx |
| 321 | : GrProcessorUnitTest::kAlphaTextureIdx; |
| 322 | sk_sp<GrTextureProxy> proxy = d->textureProxy(texIdx); |
| 323 | |
joshualitt | 0067ff5 | 2015-07-08 14:26:19 -0700 | [diff] [blame] | 324 | int width = d->fRandom->nextRangeU(1, MAX_KERNEL_SIZE); |
| 325 | int height = d->fRandom->nextRangeU(1, MAX_KERNEL_SIZE / width); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 326 | SkISize kernelSize = SkISize::Make(width, height); |
Ben Wagner | 7ecc596 | 2016-11-02 17:07:33 -0400 | [diff] [blame] | 327 | std::unique_ptr<SkScalar[]> kernel(new SkScalar[width * height]); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 328 | for (int i = 0; i < width * height; i++) { |
joshualitt | 0067ff5 | 2015-07-08 14:26:19 -0700 | [diff] [blame] | 329 | kernel.get()[i] = d->fRandom->nextSScalar1(); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 330 | } |
joshualitt | 0067ff5 | 2015-07-08 14:26:19 -0700 | [diff] [blame] | 331 | SkScalar gain = d->fRandom->nextSScalar1(); |
| 332 | SkScalar bias = d->fRandom->nextSScalar1(); |
| 333 | SkIPoint kernelOffset = SkIPoint::Make(d->fRandom->nextRangeU(0, kernelSize.width()), |
| 334 | d->fRandom->nextRangeU(0, kernelSize.height())); |
Robert Phillips | dbc8eeb | 2017-02-21 10:04:31 -0500 | [diff] [blame] | 335 | SkIRect bounds = SkIRect::MakeXYWH(d->fRandom->nextRangeU(0, proxy->width()), |
| 336 | d->fRandom->nextRangeU(0, proxy->height()), |
| 337 | d->fRandom->nextRangeU(0, proxy->width()), |
| 338 | d->fRandom->nextRangeU(0, proxy->height())); |
joshualitt | 0067ff5 | 2015-07-08 14:26:19 -0700 | [diff] [blame] | 339 | GrTextureDomain::Mode tileMode = |
| 340 | static_cast<GrTextureDomain::Mode>(d->fRandom->nextRangeU(0, 2)); |
| 341 | bool convolveAlpha = d->fRandom->nextBool(); |
Robert Phillips | fbcef6e | 2017-06-15 12:07:18 -0400 | [diff] [blame] | 342 | return GrMatrixConvolutionEffect::Make(std::move(proxy), |
bungeman | 06ca8ec | 2016-06-09 08:01:03 -0700 | [diff] [blame] | 343 | bounds, |
| 344 | kernelSize, |
| 345 | kernel.get(), |
| 346 | gain, |
| 347 | bias, |
| 348 | kernelOffset, |
| 349 | tileMode, |
| 350 | convolveAlpha); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 351 | } |
Hal Canary | 6f6961e | 2017-01-31 13:50:44 -0500 | [diff] [blame] | 352 | #endif |