| 1 | /* |
| 2 | * Copyright 2018 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 | /************************************************************************************************** |
| 9 | *** This file was autogenerated from GrRectBlurEffect.fp; do not modify. |
| 10 | **************************************************************************************************/ |
| 11 | #ifndef GrRectBlurEffect_DEFINED |
| 12 | #define GrRectBlurEffect_DEFINED |
| 13 | |
| 14 | #include "include/core/SkM44.h" |
| 15 | #include "include/core/SkTypes.h" |
| 16 | |
| 17 | #include <cmath> |
| 18 | #include "include/core/SkRect.h" |
| 19 | #include "include/core/SkScalar.h" |
| 20 | #include "include/gpu/GrRecordingContext.h" |
| 21 | #include "src/core/SkBlurMask.h" |
| 22 | #include "src/core/SkMathPriv.h" |
| 23 | #include "src/gpu/GrBitmapTextureMaker.h" |
| 24 | #include "src/gpu/GrProxyProvider.h" |
| 25 | #include "src/gpu/GrRecordingContextPriv.h" |
| 26 | #include "src/gpu/GrShaderCaps.h" |
| 27 | #include "src/gpu/effects/GrTextureEffect.h" |
| 28 | |
| 29 | #include "src/gpu/GrFragmentProcessor.h" |
| 30 | |
| 31 | class GrRectBlurEffect : public GrFragmentProcessor { |
| 32 | public: |
| 33 | static std::unique_ptr<GrFragmentProcessor> MakeIntegralFP(GrRecordingContext* context, |
| 34 | float sixSigma) { |
| 35 | // The texture we're producing represents the integral of a normal distribution over a |
| 36 | // six-sigma range centered at zero. We want enough resolution so that the linear |
| 37 | // interpolation done in texture lookup doesn't introduce noticeable artifacts. We |
| 38 | // conservatively choose to have 2 texels for each dst pixel. |
| 39 | int minWidth = 2 * sk_float_ceil2int(sixSigma); |
| 40 | // Bin by powers of 2 with a minimum so we get good profile reuse. |
| 41 | int width = std::max(SkNextPow2(minWidth), 32); |
| 42 | |
| 43 | static const GrUniqueKey::Domain kDomain = GrUniqueKey::GenerateDomain(); |
| 44 | GrUniqueKey key; |
| 45 | GrUniqueKey::Builder builder(&key, kDomain, 1, "Rect Blur Mask" ); |
| 46 | builder[0] = width; |
| 47 | builder.finish(); |
| 48 | |
| 49 | SkMatrix m = SkMatrix::Scale(width / sixSigma, 1.f); |
| 50 | |
| 51 | GrProxyProvider* proxyProvider = context->priv().proxyProvider(); |
| 52 | if (sk_sp<GrTextureProxy> proxy = proxyProvider->findOrCreateProxyByUniqueKey(key)) { |
| 53 | GrSwizzle swizzle = context->priv().caps()->getReadSwizzle(proxy->backendFormat(), |
| 54 | GrColorType::kAlpha_8); |
| 55 | GrSurfaceProxyView view{std::move(proxy), kTopLeft_GrSurfaceOrigin, swizzle}; |
| 56 | return GrTextureEffect::Make(std::move(view), kPremul_SkAlphaType, m, |
| 57 | GrSamplerState::Filter::kLinear); |
| 58 | } |
| 59 | |
| 60 | SkBitmap bitmap; |
| 61 | if (!bitmap.tryAllocPixels(SkImageInfo::MakeA8(width, 1))) { |
| 62 | return {}; |
| 63 | } |
| 64 | *bitmap.getAddr8(0, 0) = 255; |
| 65 | const float invWidth = 1.f / width; |
| 66 | for (int i = 1; i < width - 1; ++i) { |
| 67 | float x = (i + 0.5f) * invWidth; |
| 68 | x = (-6 * x + 3) * SK_ScalarRoot2Over2; |
| 69 | float integral = 0.5f * (std::erf(x) + 1.f); |
| 70 | *bitmap.getAddr8(i, 0) = SkToU8(sk_float_round2int(255.f * integral)); |
| 71 | } |
| 72 | *bitmap.getAddr8(width - 1, 0) = 0; |
| 73 | bitmap.setImmutable(); |
| 74 | |
| 75 | GrBitmapTextureMaker maker(context, bitmap, GrImageTexGenPolicy::kNew_Uncached_Budgeted); |
| 76 | auto view = maker.view(GrMipmapped::kNo); |
| 77 | if (!view) { |
| 78 | return {}; |
| 79 | } |
| 80 | SkASSERT(view.origin() == kTopLeft_GrSurfaceOrigin); |
| 81 | proxyProvider->assignUniqueKeyToProxy(key, view.asTextureProxy()); |
| 82 | return GrTextureEffect::Make(std::move(view), kPremul_SkAlphaType, m, |
| 83 | GrSamplerState::Filter::kLinear); |
| 84 | } |
| 85 | |
| 86 | static std::unique_ptr<GrFragmentProcessor> Make(std::unique_ptr<GrFragmentProcessor> inputFP, |
| 87 | GrRecordingContext* context, |
| 88 | const GrShaderCaps& caps, |
| 89 | const SkRect& rect, |
| 90 | float sigma) { |
| 91 | SkASSERT(rect.isSorted()); |
| 92 | if (!caps.floatIs32Bits()) { |
| 93 | // We promote the math that gets us into the Gaussian space to full float when the rect |
| 94 | // coords are large. If we don't have full float then fail. We could probably clip the |
| 95 | // rect to an outset device bounds instead. |
| 96 | if (SkScalarAbs(rect.fLeft) > 16000.f || SkScalarAbs(rect.fTop) > 16000.f || |
| 97 | SkScalarAbs(rect.fRight) > 16000.f || SkScalarAbs(rect.fBottom) > 16000.f) { |
| 98 | return nullptr; |
| 99 | } |
| 100 | } |
| 101 | |
| 102 | const float sixSigma = 6 * sigma; |
| 103 | std::unique_ptr<GrFragmentProcessor> integral = MakeIntegralFP(context, sixSigma); |
| 104 | if (!integral) { |
| 105 | return nullptr; |
| 106 | } |
| 107 | |
| 108 | // In the fast variant we think of the midpoint of the integral texture as aligning |
| 109 | // with the closest rect edge both in x and y. To simplify texture coord calculation we |
| 110 | // inset the rect so that the edge of the inset rect corresponds to t = 0 in the texture. |
| 111 | // It actually simplifies things a bit in the !isFast case, too. |
| 112 | float threeSigma = sixSigma / 2; |
| 113 | SkRect insetRect = {rect.fLeft + threeSigma, rect.fTop + threeSigma, |
| 114 | rect.fRight - threeSigma, rect.fBottom - threeSigma}; |
| 115 | |
| 116 | // In our fast variant we find the nearest horizontal and vertical edges and for each |
| 117 | // do a lookup in the integral texture for each and multiply them. When the rect is |
| 118 | // less than 6 sigma wide then things aren't so simple and we have to consider both the |
| 119 | // left and right edge of the rectangle (and similar in y). |
| 120 | bool isFast = insetRect.isSorted(); |
| 121 | return std::unique_ptr<GrFragmentProcessor>( |
| 122 | new GrRectBlurEffect(std::move(inputFP), insetRect, std::move(integral), isFast, |
| 123 | GrSamplerState::Filter::kLinear)); |
| 124 | } |
| 125 | GrRectBlurEffect(const GrRectBlurEffect& src); |
| 126 | std::unique_ptr<GrFragmentProcessor> clone() const override; |
| 127 | const char* name() const override { return "RectBlurEffect" ; } |
| 128 | SkRect rect; |
| 129 | bool isFast; |
| 130 | |
| 131 | private: |
| 132 | GrRectBlurEffect(std::unique_ptr<GrFragmentProcessor> inputFP, |
| 133 | SkRect rect, |
| 134 | std::unique_ptr<GrFragmentProcessor> integral, |
| 135 | bool isFast, |
| 136 | GrSamplerState samplerParams) |
| 137 | : INHERITED(kGrRectBlurEffect_ClassID, |
| 138 | (OptimizationFlags)(inputFP ? ProcessorOptimizationFlags(inputFP.get()) |
| 139 | : kAll_OptimizationFlags) & |
| 140 | kCompatibleWithCoverageAsAlpha_OptimizationFlag) |
| 141 | , rect(rect) |
| 142 | , isFast(isFast) { |
| 143 | this->registerChild(std::move(inputFP), SkSL::SampleUsage::PassThrough()); |
| 144 | SkASSERT(integral); |
| 145 | this->registerChild(std::move(integral), SkSL::SampleUsage::Explicit()); |
| 146 | } |
| 147 | GrGLSLFragmentProcessor* onCreateGLSLInstance() const override; |
| 148 | void onGetGLSLProcessorKey(const GrShaderCaps&, GrProcessorKeyBuilder*) const override; |
| 149 | bool onIsEqual(const GrFragmentProcessor&) const override; |
| 150 | #if GR_TEST_UTILS |
| 151 | SkString onDumpInfo() const override; |
| 152 | #endif |
| 153 | GR_DECLARE_FRAGMENT_PROCESSOR_TEST |
| 154 | typedef GrFragmentProcessor INHERITED; |
| 155 | }; |
| 156 | #endif |
| 157 | |