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