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 | |
| 8 | #ifndef GrMatrixConvolutionEffect_DEFINED |
| 9 | #define GrMatrixConvolutionEffect_DEFINED |
| 10 | |
| 11 | #include "GrSingleTextureEffect.h" |
egdaniel | 605dd0f | 2014-11-12 08:35:25 -0800 | [diff] [blame] | 12 | #include "GrInvariantOutput.h" |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 13 | #include "GrTextureDomain.h" |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 14 | |
| 15 | // A little bit less than the minimum # uniforms required by DX9SM2 (32). |
| 16 | // Allows for a 5x5 kernel (or 25x1, for that matter). |
| 17 | #define MAX_KERNEL_SIZE 25 |
| 18 | |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 19 | class GrMatrixConvolutionEffect : public GrSingleTextureEffect { |
| 20 | public: |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 21 | static GrFragmentProcessor* Create(GrTexture* texture, |
| 22 | const SkIRect& bounds, |
| 23 | const SkISize& kernelSize, |
| 24 | const SkScalar* kernel, |
| 25 | SkScalar gain, |
| 26 | SkScalar bias, |
| 27 | const SkIPoint& kernelOffset, |
| 28 | GrTextureDomain::Mode tileMode, |
| 29 | bool convolveAlpha) { |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 30 | return SkNEW_ARGS(GrMatrixConvolutionEffect, (texture, |
| 31 | bounds, |
| 32 | kernelSize, |
| 33 | kernel, |
| 34 | gain, |
| 35 | bias, |
| 36 | kernelOffset, |
| 37 | tileMode, |
| 38 | convolveAlpha)); |
| 39 | } |
joshualitt | 5acfea7 | 2014-08-11 13:55:34 -0700 | [diff] [blame] | 40 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 41 | static GrFragmentProcessor* CreateGaussian(GrTexture* texture, |
| 42 | const SkIRect& bounds, |
| 43 | const SkISize& kernelSize, |
| 44 | SkScalar gain, |
| 45 | SkScalar bias, |
| 46 | const SkIPoint& kernelOffset, |
| 47 | GrTextureDomain::Mode tileMode, |
| 48 | bool convolveAlpha, |
| 49 | SkScalar sigmaX, |
| 50 | SkScalar sigmaY); |
joshualitt | 5acfea7 | 2014-08-11 13:55:34 -0700 | [diff] [blame] | 51 | |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 52 | virtual ~GrMatrixConvolutionEffect(); |
| 53 | |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 54 | const SkIRect& bounds() const { return fBounds; } |
| 55 | const SkISize& kernelSize() const { return fKernelSize; } |
| 56 | const float* kernelOffset() const { return fKernelOffset; } |
| 57 | const float* kernel() const { return fKernel; } |
| 58 | float gain() const { return fGain; } |
| 59 | float bias() const { return fBias; } |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 60 | bool convolveAlpha() const { return fConvolveAlpha; } |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 61 | const GrTextureDomain& domain() const { return fDomain; } |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 62 | |
mtklein | 72c9faa | 2015-01-09 10:06:39 -0800 | [diff] [blame] | 63 | const char* name() const SK_OVERRIDE { return "MatrixConvolution"; } |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 64 | |
mtklein | 72c9faa | 2015-01-09 10:06:39 -0800 | [diff] [blame] | 65 | void getGLProcessorKey(const GrGLCaps&, GrProcessorKeyBuilder*) const SK_OVERRIDE; |
joshualitt | eb2a676 | 2014-12-04 11:35:33 -0800 | [diff] [blame] | 66 | |
mtklein | 72c9faa | 2015-01-09 10:06:39 -0800 | [diff] [blame] | 67 | GrGLFragmentProcessor* createGLInstance() const SK_OVERRIDE; |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 68 | |
| 69 | private: |
| 70 | GrMatrixConvolutionEffect(GrTexture*, |
| 71 | const SkIRect& bounds, |
| 72 | const SkISize& kernelSize, |
| 73 | const SkScalar* kernel, |
| 74 | SkScalar gain, |
| 75 | SkScalar bias, |
| 76 | const SkIPoint& kernelOffset, |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 77 | GrTextureDomain::Mode tileMode, |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 78 | bool convolveAlpha); |
| 79 | |
mtklein | 72c9faa | 2015-01-09 10:06:39 -0800 | [diff] [blame] | 80 | bool onIsEqual(const GrFragmentProcessor&) const SK_OVERRIDE; |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 81 | |
mtklein | 72c9faa | 2015-01-09 10:06:39 -0800 | [diff] [blame] | 82 | void onComputeInvariantOutput(GrInvariantOutput* inout) const SK_OVERRIDE { |
egdaniel | 1a8ecdf | 2014-10-03 06:24:12 -0700 | [diff] [blame] | 83 | // TODO: Try to do better? |
joshualitt | 56995b5 | 2014-12-11 15:44:02 -0800 | [diff] [blame] | 84 | inout->mulByUnknownFourComponents(); |
egdaniel | 1a8ecdf | 2014-10-03 06:24:12 -0700 | [diff] [blame] | 85 | } |
| 86 | |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 87 | SkIRect fBounds; |
| 88 | SkISize fKernelSize; |
joshualitt | 5acfea7 | 2014-08-11 13:55:34 -0700 | [diff] [blame] | 89 | float fKernel[MAX_KERNEL_SIZE]; |
joshualitt | 5ae5fc5 | 2014-07-29 12:59:27 -0700 | [diff] [blame] | 90 | float fGain; |
| 91 | float fBias; |
| 92 | float fKernelOffset[2]; |
| 93 | bool fConvolveAlpha; |
| 94 | GrTextureDomain fDomain; |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 95 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 96 | GR_DECLARE_FRAGMENT_PROCESSOR_TEST; |
joshualitt | ac97792 | 2014-07-22 09:52:11 -0700 | [diff] [blame] | 97 | |
| 98 | typedef GrSingleTextureEffect INHERITED; |
| 99 | }; |
| 100 | |
| 101 | #endif |