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 | |
Hal Canary | 6f6961e | 2017-01-31 13:50:44 -0500 | [diff] [blame] | 9 | #include "GrTextureProxy.h" |
egdaniel | 64c4728 | 2015-11-13 06:54:19 -0800 | [diff] [blame] | 10 | #include "glsl/GrGLSLFragmentProcessor.h" |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 11 | #include "glsl/GrGLSLFragmentShaderBuilder.h" |
egdaniel | 018fb62 | 2015-10-28 07:26:40 -0700 | [diff] [blame] | 12 | #include "glsl/GrGLSLProgramDataManager.h" |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 13 | #include "glsl/GrGLSLUniformHandler.h" |
Brian Salomon | 94efbf5 | 2016-11-29 13:43:05 -0500 | [diff] [blame] | 14 | #include "../private/GrGLSL.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: |
egdaniel | 018fb62 | 2015-10-28 07:26:40 -0700 | [diff] [blame] | 23 | void onSetData(const GrGLSLProgramDataManager&, const GrProcessor&) 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; |
cdalton | 5e58cee | 2016-02-11 12:49:47 -0800 | [diff] [blame] | 49 | fImageIncrementUni = uniformHandler->addUniform(kFragment_GrShaderFlag, |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 50 | kVec2f_GrSLType, kDefault_GrSLPrecision, |
| 51 | "ImageIncrement"); |
cdalton | 5e58cee | 2016-02-11 12:49:47 -0800 | [diff] [blame] | 52 | fKernelUni = uniformHandler->addUniformArray(kFragment_GrShaderFlag, |
jvanverth | 78d6eb0 | 2016-03-02 13:21:16 -0800 | [diff] [blame] | 53 | kVec4f_GrSLType, kDefault_GrSLPrecision, |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 54 | "Kernel", |
jvanverth | 78d6eb0 | 2016-03-02 13:21:16 -0800 | [diff] [blame] | 55 | arrayCount); |
cdalton | 5e58cee | 2016-02-11 12:49:47 -0800 | [diff] [blame] | 56 | fKernelOffsetUni = uniformHandler->addUniform(kFragment_GrShaderFlag, |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 57 | kVec2f_GrSLType, kDefault_GrSLPrecision, |
| 58 | "KernelOffset"); |
cdalton | 5e58cee | 2016-02-11 12:49:47 -0800 | [diff] [blame] | 59 | fGainUni = uniformHandler->addUniform(kFragment_GrShaderFlag, |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 60 | kFloat_GrSLType, kDefault_GrSLPrecision, "Gain"); |
cdalton | 5e58cee | 2016-02-11 12:49:47 -0800 | [diff] [blame] | 61 | fBiasUni = uniformHandler->addUniform(kFragment_GrShaderFlag, |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 62 | kFloat_GrSLType, kDefault_GrSLPrecision, "Bias"); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 63 | |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 64 | const char* kernelOffset = uniformHandler->getUniformCStr(fKernelOffsetUni); |
| 65 | const char* imgInc = uniformHandler->getUniformCStr(fImageIncrementUni); |
| 66 | const char* kernel = uniformHandler->getUniformCStr(fKernelUni); |
| 67 | const char* gain = uniformHandler->getUniformCStr(fGainUni); |
| 68 | const char* bias = uniformHandler->getUniformCStr(fBiasUni); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 69 | |
cdalton | 8528541 | 2016-02-18 12:37:07 -0800 | [diff] [blame] | 70 | GrGLSLFPFragmentBuilder* fragBuilder = args.fFragBuilder; |
bsalomon | 1a1aa93 | 2016-09-12 09:30:36 -0700 | [diff] [blame] | 71 | SkString coords2D = fragBuilder->ensureCoords2D(args.fTransformedCoords[0]); |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 72 | fragBuilder->codeAppend("vec4 sum = vec4(0, 0, 0, 0);"); |
| 73 | fragBuilder->codeAppendf("vec2 coord = %s - %s * %s;", coords2D.c_str(), kernelOffset, imgInc); |
| 74 | fragBuilder->codeAppend("vec4 c;"); |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 75 | |
jvanverth | 78d6eb0 | 2016-03-02 13:21:16 -0800 | [diff] [blame] | 76 | const char* kVecSuffix[4] = { ".x", ".y", ".z", ".w" }; |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 77 | for (int y = 0; y < kHeight; y++) { |
| 78 | for (int x = 0; x < kWidth; x++) { |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 79 | GrGLSLShaderBuilder::ShaderBlock block(fragBuilder); |
jvanverth | 78d6eb0 | 2016-03-02 13:21:16 -0800 | [diff] [blame] | 80 | int offset = y*kWidth + x; |
| 81 | |
halcanary | 9d524f2 | 2016-03-29 09:03:52 -0700 | [diff] [blame] | 82 | fragBuilder->codeAppendf("float k = %s[%d]%s;", kernel, offset / 4, |
jvanverth | 78d6eb0 | 2016-03-02 13:21:16 -0800 | [diff] [blame] | 83 | kVecSuffix[offset & 0x3]); |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 84 | SkString coord; |
| 85 | coord.printf("coord + vec2(%d, %d) * %s", x, y, imgInc); |
egdaniel | a2e3e0f | 2015-11-19 07:23:45 -0800 | [diff] [blame] | 86 | fDomain.sampleTexture(fragBuilder, |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 87 | uniformHandler, |
Brian Salomon | 1edc5b9 | 2016-11-29 13:43:46 -0500 | [diff] [blame] | 88 | args.fShaderCaps, |
egdaniel | a2e3e0f | 2015-11-19 07:23:45 -0800 | [diff] [blame] | 89 | domain, |
| 90 | "c", |
| 91 | coord, |
cdalton | 3f6f76f | 2016-04-11 12:18:09 -0700 | [diff] [blame] | 92 | args.fTexSamplers[0]); |
robertphillips | bf536af | 2016-02-04 06:11:53 -0800 | [diff] [blame] | 93 | if (!mce.convolveAlpha()) { |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 94 | fragBuilder->codeAppend("c.rgb /= c.a;"); |
| 95 | fragBuilder->codeAppend("c.rgb = clamp(c.rgb, 0.0, 1.0);"); |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 96 | } |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 97 | fragBuilder->codeAppend("sum += c * k;"); |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 98 | } |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 99 | } |
robertphillips | bf536af | 2016-02-04 06:11:53 -0800 | [diff] [blame] | 100 | if (mce.convolveAlpha()) { |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 101 | fragBuilder->codeAppendf("%s = sum * %s + %s;", args.fOutputColor, gain, bias); |
Brian Salomon | a12c153 | 2017-02-13 12:41:44 -0500 | [diff] [blame] | 102 | fragBuilder->codeAppendf("%s.a = clamp(%s.a, 0, 1);", args.fOutputColor, args.fOutputColor); |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 103 | fragBuilder->codeAppendf("%s.rgb = clamp(%s.rgb, 0.0, %s.a);", |
| 104 | args.fOutputColor, args.fOutputColor, args.fOutputColor); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 105 | } else { |
egdaniel | a2e3e0f | 2015-11-19 07:23:45 -0800 | [diff] [blame] | 106 | fDomain.sampleTexture(fragBuilder, |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 107 | uniformHandler, |
Brian Salomon | 1edc5b9 | 2016-11-29 13:43:46 -0500 | [diff] [blame] | 108 | args.fShaderCaps, |
egdaniel | a2e3e0f | 2015-11-19 07:23:45 -0800 | [diff] [blame] | 109 | domain, |
| 110 | "c", |
| 111 | coords2D, |
cdalton | 3f6f76f | 2016-04-11 12:18:09 -0700 | [diff] [blame] | 112 | args.fTexSamplers[0]); |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 113 | fragBuilder->codeAppendf("%s.a = c.a;", args.fOutputColor); |
Brian Salomon | a12c153 | 2017-02-13 12:41:44 -0500 | [diff] [blame] | 114 | fragBuilder->codeAppendf("%s.rgb = clamp(sum.rgb * %s + %s, 0, 1);", args.fOutputColor, gain, bias); |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 115 | fragBuilder->codeAppendf("%s.rgb *= %s.a;", args.fOutputColor, args.fOutputColor); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 116 | } |
joshualitt | 5acfea7 | 2014-08-11 13:55:34 -0700 | [diff] [blame] | 117 | |
| 118 | SkString modulate; |
wangyix | 7c157a9 | 2015-07-22 15:08:53 -0700 | [diff] [blame] | 119 | GrGLSLMulVarBy4f(&modulate, args.fOutputColor, args.fInputColor); |
egdaniel | 4ca2e60 | 2015-11-18 08:01:26 -0800 | [diff] [blame] | 120 | fragBuilder->codeAppend(modulate.c_str()); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 121 | } |
| 122 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 123 | void GrGLMatrixConvolutionEffect::GenKey(const GrProcessor& processor, |
Brian Salomon | 94efbf5 | 2016-11-29 13:43:05 -0500 | [diff] [blame] | 124 | const GrShaderCaps&, GrProcessorKeyBuilder* b) { |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 125 | const GrMatrixConvolutionEffect& m = processor.cast<GrMatrixConvolutionEffect>(); |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 126 | SkASSERT(m.kernelSize().width() <= 0x7FFF && m.kernelSize().height() <= 0xFFFF); |
| 127 | uint32_t key = m.kernelSize().width() << 16 | m.kernelSize().height(); |
caryclark | 952538e | 2016-02-26 05:01:42 -0800 | [diff] [blame] | 128 | key |= m.convolveAlpha() ? 1U << 31 : 0; |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 129 | b->add32(key); |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 130 | b->add32(GrTextureDomain::GLDomain::DomainKey(m.domain())); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 131 | } |
| 132 | |
egdaniel | 018fb62 | 2015-10-28 07:26:40 -0700 | [diff] [blame] | 133 | void GrGLMatrixConvolutionEffect::onSetData(const GrGLSLProgramDataManager& pdman, |
| 134 | const GrProcessor& processor) { |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 135 | const GrMatrixConvolutionEffect& conv = processor.cast<GrMatrixConvolutionEffect>(); |
Brian Salomon | db4183d | 2016-11-17 12:48:40 -0500 | [diff] [blame] | 136 | GrTexture* texture = conv.textureSampler(0).texture(); |
robertphillips | bf536af | 2016-02-04 06:11:53 -0800 | [diff] [blame] | 137 | |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 138 | float imageIncrement[2]; |
Brian Salomon | 0bbecb2 | 2016-11-17 11:38:22 -0500 | [diff] [blame] | 139 | float ySign = texture->origin() == kTopLeft_GrSurfaceOrigin ? 1.0f : -1.0f; |
| 140 | imageIncrement[0] = 1.0f / texture->width(); |
| 141 | imageIncrement[1] = ySign / texture->height(); |
kkinnunen | 7510b22 | 2014-07-30 00:04:16 -0700 | [diff] [blame] | 142 | pdman.set2fv(fImageIncrementUni, 1, imageIncrement); |
| 143 | pdman.set2fv(fKernelOffsetUni, 1, conv.kernelOffset()); |
jvanverth | 78d6eb0 | 2016-03-02 13:21:16 -0800 | [diff] [blame] | 144 | int kernelCount = conv.kernelSize().width() * conv.kernelSize().height(); |
| 145 | int arrayCount = (kernelCount + 3) / 4; |
| 146 | SkASSERT(4 * arrayCount >= kernelCount); |
| 147 | pdman.set4fv(fKernelUni, arrayCount, conv.kernel()); |
kkinnunen | 7510b22 | 2014-07-30 00:04:16 -0700 | [diff] [blame] | 148 | pdman.set1f(fGainUni, conv.gain()); |
| 149 | pdman.set1f(fBiasUni, conv.bias()); |
Robert Phillips | e98234f | 2017-01-09 14:23:59 -0500 | [diff] [blame] | 150 | fDomain.setData(pdman, conv.domain(), texture); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 151 | } |
| 152 | |
Robert Phillips | 296b1cc | 2017-03-15 10:42:12 -0400 | [diff] [blame] | 153 | GrMatrixConvolutionEffect::GrMatrixConvolutionEffect(GrResourceProvider* resourceProvider, |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 154 | sk_sp<GrTextureProxy> proxy, |
| 155 | const SkIRect& bounds, |
| 156 | const SkISize& kernelSize, |
| 157 | const SkScalar* kernel, |
| 158 | SkScalar gain, |
| 159 | SkScalar bias, |
| 160 | const SkIPoint& kernelOffset, |
| 161 | GrTextureDomain::Mode tileMode, |
| 162 | bool convolveAlpha) |
| 163 | // To advertise either the modulation or opaqueness optimizations we'd have to examine the |
| 164 | // parameters. |
Robert Phillips | 296b1cc | 2017-03-15 10:42:12 -0400 | [diff] [blame] | 165 | : INHERITED(resourceProvider, kNone_OptimizationFlags, proxy, nullptr, SkMatrix::I()) |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 166 | , fKernelSize(kernelSize) |
| 167 | , fGain(SkScalarToFloat(gain)) |
| 168 | , fBias(SkScalarToFloat(bias) / 255.0f) |
| 169 | , fConvolveAlpha(convolveAlpha) |
| 170 | , fDomain(proxy.get(), GrTextureDomain::MakeTexelDomainForMode(bounds, tileMode), tileMode) { |
| 171 | this->initClassID<GrMatrixConvolutionEffect>(); |
| 172 | for (int i = 0; i < kernelSize.width() * kernelSize.height(); i++) { |
| 173 | fKernel[i] = SkScalarToFloat(kernel[i]); |
| 174 | } |
| 175 | fKernelOffset[0] = static_cast<float>(kernelOffset.x()); |
| 176 | fKernelOffset[1] = static_cast<float>(kernelOffset.y()); |
| 177 | } |
| 178 | |
Brian Salomon | 94efbf5 | 2016-11-29 13:43:05 -0500 | [diff] [blame] | 179 | void GrMatrixConvolutionEffect::onGetGLSLProcessorKey(const GrShaderCaps& caps, |
egdaniel | 57d3b03 | 2015-11-13 11:57:27 -0800 | [diff] [blame] | 180 | GrProcessorKeyBuilder* b) const { |
joshualitt | eb2a676 | 2014-12-04 11:35:33 -0800 | [diff] [blame] | 181 | GrGLMatrixConvolutionEffect::GenKey(*this, caps, b); |
| 182 | } |
| 183 | |
egdaniel | 57d3b03 | 2015-11-13 11:57:27 -0800 | [diff] [blame] | 184 | GrGLSLFragmentProcessor* GrMatrixConvolutionEffect::onCreateGLSLInstance() const { |
robertphillips | bf536af | 2016-02-04 06:11:53 -0800 | [diff] [blame] | 185 | return new GrGLMatrixConvolutionEffect; |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 186 | } |
| 187 | |
bsalomon | 0e08fc1 | 2014-10-15 08:19:04 -0700 | [diff] [blame] | 188 | bool GrMatrixConvolutionEffect::onIsEqual(const GrFragmentProcessor& sBase) const { |
joshualitt | 49586be | 2014-09-16 08:21:41 -0700 | [diff] [blame] | 189 | const GrMatrixConvolutionEffect& s = sBase.cast<GrMatrixConvolutionEffect>(); |
bsalomon | 420d7e9 | 2014-10-16 09:18:09 -0700 | [diff] [blame] | 190 | return fKernelSize == s.kernelSize() && |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 191 | !memcmp(fKernel, s.kernel(), |
| 192 | fKernelSize.width() * fKernelSize.height() * sizeof(float)) && |
| 193 | fGain == s.gain() && |
| 194 | fBias == s.bias() && |
| 195 | fKernelOffset == s.kernelOffset() && |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 196 | fConvolveAlpha == s.convolveAlpha() && |
| 197 | fDomain == s.domain(); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 198 | } |
| 199 | |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 200 | static void fill_in_2D_gaussian_kernel(float* kernel, int width, int height, |
| 201 | SkScalar sigmaX, SkScalar sigmaY) { |
joshualitt | 5acfea7 | 2014-08-11 13:55:34 -0700 | [diff] [blame] | 202 | SkASSERT(width * height <= MAX_KERNEL_SIZE); |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 203 | const float sigmaXDenom = 1.0f / (2.0f * SkScalarToFloat(SkScalarSquare(sigmaX))); |
| 204 | const float sigmaYDenom = 1.0f / (2.0f * SkScalarToFloat(SkScalarSquare(sigmaY))); |
| 205 | const int xRadius = width / 2; |
| 206 | const int yRadius = height / 2; |
| 207 | |
joshualitt | 5acfea7 | 2014-08-11 13:55:34 -0700 | [diff] [blame] | 208 | float sum = 0.0f; |
joshualitt | 5acfea7 | 2014-08-11 13:55:34 -0700 | [diff] [blame] | 209 | for (int x = 0; x < width; x++) { |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 210 | float xTerm = static_cast<float>(x - xRadius); |
| 211 | xTerm = xTerm * xTerm * sigmaXDenom; |
| 212 | for (int y = 0; y < height; y++) { |
| 213 | float yTerm = static_cast<float>(y - yRadius); |
| 214 | float xyTerm = sk_float_exp(-(xTerm + yTerm * yTerm * sigmaYDenom)); |
| 215 | // Note that the constant term (1/(sqrt(2*pi*sigma^2)) of the Gaussian |
| 216 | // is dropped here, since we renormalize the kernel below. |
| 217 | kernel[y * width + x] = xyTerm; |
| 218 | sum += xyTerm; |
| 219 | } |
joshualitt | 5acfea7 | 2014-08-11 13:55:34 -0700 | [diff] [blame] | 220 | } |
| 221 | // Normalize the kernel |
| 222 | float scale = 1.0f / sum; |
| 223 | for (int i = 0; i < width * height; ++i) { |
| 224 | kernel[i] *= scale; |
| 225 | } |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 226 | } |
| 227 | |
| 228 | |
| 229 | // Static function to create a 2D convolution |
Robert Phillips | 296b1cc | 2017-03-15 10:42:12 -0400 | [diff] [blame] | 230 | sk_sp<GrFragmentProcessor> GrMatrixConvolutionEffect::MakeGaussian( |
| 231 | GrResourceProvider* resourceProvider, |
| 232 | sk_sp<GrTextureProxy> proxy, |
| 233 | const SkIRect& bounds, |
| 234 | const SkISize& kernelSize, |
| 235 | SkScalar gain, |
| 236 | SkScalar bias, |
| 237 | const SkIPoint& kernelOffset, |
| 238 | GrTextureDomain::Mode tileMode, |
| 239 | bool convolveAlpha, |
| 240 | SkScalar sigmaX, |
| 241 | SkScalar sigmaY) { |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 242 | float kernel[MAX_KERNEL_SIZE]; |
| 243 | |
| 244 | fill_in_2D_gaussian_kernel(kernel, kernelSize.width(), kernelSize.height(), sigmaX, sigmaY); |
| 245 | |
| 246 | return sk_sp<GrFragmentProcessor>( |
Robert Phillips | 296b1cc | 2017-03-15 10:42:12 -0400 | [diff] [blame] | 247 | new GrMatrixConvolutionEffect(resourceProvider, std::move(proxy), bounds, kernelSize, |
| 248 | kernel, gain, bias, kernelOffset, tileMode, convolveAlpha)); |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 249 | } |
| 250 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 251 | GR_DEFINE_FRAGMENT_PROCESSOR_TEST(GrMatrixConvolutionEffect); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 252 | |
Hal Canary | 6f6961e | 2017-01-31 13:50:44 -0500 | [diff] [blame] | 253 | #if GR_TEST_UTILS |
bungeman | 06ca8ec | 2016-06-09 08:01:03 -0700 | [diff] [blame] | 254 | sk_sp<GrFragmentProcessor> GrMatrixConvolutionEffect::TestCreate(GrProcessorTestData* d) { |
Robert Phillips | dbc8eeb | 2017-02-21 10:04:31 -0500 | [diff] [blame] | 255 | int texIdx = d->fRandom->nextBool() ? GrProcessorUnitTest::kSkiaPMTextureIdx |
| 256 | : GrProcessorUnitTest::kAlphaTextureIdx; |
| 257 | sk_sp<GrTextureProxy> proxy = d->textureProxy(texIdx); |
| 258 | |
joshualitt | 0067ff5 | 2015-07-08 14:26:19 -0700 | [diff] [blame] | 259 | int width = d->fRandom->nextRangeU(1, MAX_KERNEL_SIZE); |
| 260 | int height = d->fRandom->nextRangeU(1, MAX_KERNEL_SIZE / width); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 261 | SkISize kernelSize = SkISize::Make(width, height); |
Ben Wagner | 7ecc596 | 2016-11-02 17:07:33 -0400 | [diff] [blame] | 262 | std::unique_ptr<SkScalar[]> kernel(new SkScalar[width * height]); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 263 | for (int i = 0; i < width * height; i++) { |
joshualitt | 0067ff5 | 2015-07-08 14:26:19 -0700 | [diff] [blame] | 264 | kernel.get()[i] = d->fRandom->nextSScalar1(); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 265 | } |
joshualitt | 0067ff5 | 2015-07-08 14:26:19 -0700 | [diff] [blame] | 266 | SkScalar gain = d->fRandom->nextSScalar1(); |
| 267 | SkScalar bias = d->fRandom->nextSScalar1(); |
| 268 | SkIPoint kernelOffset = SkIPoint::Make(d->fRandom->nextRangeU(0, kernelSize.width()), |
| 269 | d->fRandom->nextRangeU(0, kernelSize.height())); |
Robert Phillips | dbc8eeb | 2017-02-21 10:04:31 -0500 | [diff] [blame] | 270 | SkIRect bounds = SkIRect::MakeXYWH(d->fRandom->nextRangeU(0, proxy->width()), |
| 271 | d->fRandom->nextRangeU(0, proxy->height()), |
| 272 | d->fRandom->nextRangeU(0, proxy->width()), |
| 273 | d->fRandom->nextRangeU(0, proxy->height())); |
joshualitt | 0067ff5 | 2015-07-08 14:26:19 -0700 | [diff] [blame] | 274 | GrTextureDomain::Mode tileMode = |
| 275 | static_cast<GrTextureDomain::Mode>(d->fRandom->nextRangeU(0, 2)); |
| 276 | bool convolveAlpha = d->fRandom->nextBool(); |
Robert Phillips | 296b1cc | 2017-03-15 10:42:12 -0400 | [diff] [blame] | 277 | return GrMatrixConvolutionEffect::Make(d->resourceProvider(), |
Robert Phillips | dbc8eeb | 2017-02-21 10:04:31 -0500 | [diff] [blame] | 278 | std::move(proxy), |
bungeman | 06ca8ec | 2016-06-09 08:01:03 -0700 | [diff] [blame] | 279 | bounds, |
| 280 | kernelSize, |
| 281 | kernel.get(), |
| 282 | gain, |
| 283 | bias, |
| 284 | kernelOffset, |
| 285 | tileMode, |
| 286 | convolveAlpha); |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 287 | } |
Hal Canary | 6f6961e | 2017-01-31 13:50:44 -0500 | [diff] [blame] | 288 | #endif |