tomhudson@google.com | d8f856c | 2012-05-10 12:13:36 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2012 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 GrConvolutionEffect_DEFINED |
| 9 | #define GrConvolutionEffect_DEFINED |
| 10 | |
bsalomon@google.com | b505a12 | 2012-05-31 18:40:36 +0000 | [diff] [blame] | 11 | #include "Gr1DKernelEffect.h" |
tomhudson@google.com | d8f856c | 2012-05-10 12:13:36 +0000 | [diff] [blame] | 12 | |
bsalomon@google.com | ae4f96a | 2012-05-18 19:54:48 +0000 | [diff] [blame] | 13 | class GrGLConvolutionEffect; |
| 14 | |
bsalomon@google.com | b505a12 | 2012-05-31 18:40:36 +0000 | [diff] [blame] | 15 | /** |
rmistry@google.com | fbfcd56 | 2012-08-23 18:09:54 +0000 | [diff] [blame] | 16 | * A convolution effect. The kernel is specified as an array of 2 * half-width |
bsalomon@google.com | b505a12 | 2012-05-31 18:40:36 +0000 | [diff] [blame] | 17 | * + 1 weights. Each texel is multiplied by it's weight and summed to determine |
| 18 | * the output color. The output color is modulated by the input color. |
| 19 | */ |
| 20 | class GrConvolutionEffect : public Gr1DKernelEffect { |
tomhudson@google.com | d8f856c | 2012-05-10 12:13:36 +0000 | [diff] [blame] | 21 | |
| 22 | public: |
| 23 | |
tomhudson@google.com | fde2c0a | 2012-07-16 12:23:32 +0000 | [diff] [blame] | 24 | /// Convolve with an arbitrary user-specified kernel |
bsalomon@google.com | 0ac6af4 | 2013-01-16 15:16:18 +0000 | [diff] [blame] | 25 | static GrEffectRef* Create(GrTexture* tex, Direction dir, int halfWidth, const float* kernel) { |
bsalomon@google.com | 6340a41 | 2013-01-22 19:55:59 +0000 | [diff] [blame] | 26 | AutoEffectUnref effect(SkNEW_ARGS(GrConvolutionEffect, (tex, |
| 27 | dir, |
| 28 | halfWidth, |
| 29 | kernel))); |
bsalomon@google.com | a1ebbe4 | 2013-01-16 15:51:47 +0000 | [diff] [blame] | 30 | return CreateEffectRef(effect); |
bsalomon@google.com | 0ac6af4 | 2013-01-16 15:16:18 +0000 | [diff] [blame] | 31 | } |
tomhudson@google.com | fde2c0a | 2012-07-16 12:23:32 +0000 | [diff] [blame] | 32 | |
bsalomon@google.com | b4a55b7 | 2012-11-02 20:45:37 +0000 | [diff] [blame] | 33 | /// Convolve with a Gaussian kernel |
bsalomon@google.com | c5fae9e | 2013-01-24 19:00:02 +0000 | [diff] [blame] | 34 | static GrEffectRef* CreateGaussian(GrTexture* tex, |
bsalomon@google.com | 6340a41 | 2013-01-22 19:55:59 +0000 | [diff] [blame] | 35 | Direction dir, |
| 36 | int halfWidth, |
| 37 | float gaussianSigma) { |
| 38 | AutoEffectUnref effect(SkNEW_ARGS(GrConvolutionEffect, (tex, |
| 39 | dir, |
| 40 | halfWidth, |
| 41 | gaussianSigma))); |
bsalomon@google.com | a1ebbe4 | 2013-01-16 15:51:47 +0000 | [diff] [blame] | 42 | return CreateEffectRef(effect); |
bsalomon@google.com | 0ac6af4 | 2013-01-16 15:16:18 +0000 | [diff] [blame] | 43 | } |
| 44 | |
tomhudson@google.com | d8f856c | 2012-05-10 12:13:36 +0000 | [diff] [blame] | 45 | virtual ~GrConvolutionEffect(); |
| 46 | |
bsalomon@google.com | ae4f96a | 2012-05-18 19:54:48 +0000 | [diff] [blame] | 47 | const float* kernel() const { return fKernel; } |
bsalomon@google.com | 289efe0 | 2012-05-21 20:57:59 +0000 | [diff] [blame] | 48 | |
bsalomon@google.com | ae4f96a | 2012-05-18 19:54:48 +0000 | [diff] [blame] | 49 | static const char* Name() { return "Convolution"; } |
| 50 | |
bsalomon@google.com | 422e81a | 2012-10-25 14:11:03 +0000 | [diff] [blame] | 51 | typedef GrGLConvolutionEffect GLEffect; |
bsalomon@google.com | 289efe0 | 2012-05-21 20:57:59 +0000 | [diff] [blame] | 52 | |
bsalomon@google.com | 396e61f | 2012-10-25 19:00:29 +0000 | [diff] [blame] | 53 | virtual const GrBackendEffectFactory& getFactory() const SK_OVERRIDE; |
bsalomon@google.com | 68b58c9 | 2013-01-17 16:50:08 +0000 | [diff] [blame] | 54 | |
sugoi@google.com | e0e385c | 2013-03-11 18:50:03 +0000 | [diff] [blame] | 55 | virtual void getConstantColorComponents(GrColor*, uint32_t* validFlags) const { |
bsalomon@google.com | 68b58c9 | 2013-01-17 16:50:08 +0000 | [diff] [blame] | 56 | // If the texture was opaque we could know that the output color if we knew the sum of the |
| 57 | // kernel values. |
| 58 | *validFlags = 0; |
| 59 | } |
bsalomon@google.com | b505a12 | 2012-05-31 18:40:36 +0000 | [diff] [blame] | 60 | |
| 61 | enum { |
| 62 | // This was decided based on the min allowed value for the max texture |
| 63 | // samples per fragment program run in DX9SM2 (32). A sigma param of 4.0 |
| 64 | // on a blur filter gives a kernel width of 25 while a sigma of 5.0 |
| 65 | // would exceed a 32 wide kernel. |
| 66 | kMaxKernelRadius = 12, |
| 67 | // With a C++11 we could have a constexpr version of WidthFromRadius() |
| 68 | // and not have to duplicate this calculation. |
| 69 | kMaxKernelWidth = 2 * kMaxKernelRadius + 1, |
| 70 | }; |
tomhudson@google.com | d8f856c | 2012-05-10 12:13:36 +0000 | [diff] [blame] | 71 | |
| 72 | protected: |
| 73 | |
bsalomon@google.com | b505a12 | 2012-05-31 18:40:36 +0000 | [diff] [blame] | 74 | float fKernel[kMaxKernelWidth]; |
tomhudson@google.com | d8f856c | 2012-05-10 12:13:36 +0000 | [diff] [blame] | 75 | |
| 76 | private: |
bsalomon@google.com | 0ac6af4 | 2013-01-16 15:16:18 +0000 | [diff] [blame] | 77 | GrConvolutionEffect(GrTexture*, Direction, |
| 78 | int halfWidth, const float* kernel); |
| 79 | |
| 80 | /// Convolve with a Gaussian kernel |
| 81 | GrConvolutionEffect(GrTexture*, Direction, |
| 82 | int halfWidth, |
| 83 | float gaussianSigma); |
| 84 | |
bsalomon@google.com | 8a252f7 | 2013-01-22 20:35:13 +0000 | [diff] [blame] | 85 | virtual bool onIsEqual(const GrEffect&) const SK_OVERRIDE; |
bsalomon@google.com | 68b58c9 | 2013-01-17 16:50:08 +0000 | [diff] [blame] | 86 | |
bsalomon@google.com | f271cc7 | 2012-10-24 19:35:13 +0000 | [diff] [blame] | 87 | GR_DECLARE_EFFECT_TEST; |
tomhudson@google.com | d8f856c | 2012-05-10 12:13:36 +0000 | [diff] [blame] | 88 | |
bsalomon@google.com | b505a12 | 2012-05-31 18:40:36 +0000 | [diff] [blame] | 89 | typedef Gr1DKernelEffect INHERITED; |
tomhudson@google.com | d8f856c | 2012-05-10 12:13:36 +0000 | [diff] [blame] | 90 | }; |
| 91 | |
| 92 | #endif |