| /* |
| * Copyright 2012 Google Inc. |
| * |
| * Use of this source code is governed by a BSD-style license that can be |
| * found in the LICENSE file. |
| */ |
| |
| #include "src/gpu/effects/GrGaussianConvolutionFragmentProcessor.h" |
| |
| #include "include/gpu/GrTexture.h" |
| #include "src/gpu/GrTextureProxy.h" |
| #include "src/gpu/glsl/GrGLSLFragmentProcessor.h" |
| #include "src/gpu/glsl/GrGLSLFragmentShaderBuilder.h" |
| #include "src/gpu/glsl/GrGLSLProgramDataManager.h" |
| #include "src/gpu/glsl/GrGLSLUniformHandler.h" |
| |
| // For brevity |
| using UniformHandle = GrGLSLProgramDataManager::UniformHandle; |
| using Direction = GrGaussianConvolutionFragmentProcessor::Direction; |
| |
| class GrGLConvolutionEffect : public GrGLSLFragmentProcessor { |
| public: |
| void emitCode(EmitArgs&) override; |
| |
| static inline void GenKey(const GrProcessor&, const GrShaderCaps&, GrProcessorKeyBuilder*); |
| |
| protected: |
| void onSetData(const GrGLSLProgramDataManager&, const GrFragmentProcessor&) override; |
| |
| private: |
| UniformHandle fKernelUni; |
| UniformHandle fImageIncrementUni; |
| UniformHandle fBoundsUni; |
| |
| typedef GrGLSLFragmentProcessor INHERITED; |
| }; |
| |
| void GrGLConvolutionEffect::emitCode(EmitArgs& args) { |
| const GrGaussianConvolutionFragmentProcessor& ce = |
| args.fFp.cast<GrGaussianConvolutionFragmentProcessor>(); |
| |
| GrGLSLUniformHandler* uniformHandler = args.fUniformHandler; |
| fImageIncrementUni = uniformHandler->addUniform(kFragment_GrShaderFlag, kHalf2_GrSLType, |
| "ImageIncrement"); |
| if (ce.useBounds()) { |
| fBoundsUni = uniformHandler->addUniform(kFragment_GrShaderFlag, kHalf2_GrSLType, |
| "Bounds"); |
| } |
| |
| int width = ce.width(); |
| |
| int arrayCount = (width + 3) / 4; |
| SkASSERT(4 * arrayCount >= width); |
| |
| fKernelUni = uniformHandler->addUniformArray(kFragment_GrShaderFlag, kHalf4_GrSLType, |
| "Kernel", arrayCount); |
| |
| GrGLSLFPFragmentBuilder* fragBuilder = args.fFragBuilder; |
| SkString coords2D = fragBuilder->ensureCoords2D(args.fTransformedCoords[0].fVaryingPoint); |
| |
| fragBuilder->codeAppendf("%s = half4(0, 0, 0, 0);", args.fOutputColor); |
| |
| const GrShaderVar& kernel = uniformHandler->getUniformVariable(fKernelUni); |
| const char* imgInc = uniformHandler->getUniformCStr(fImageIncrementUni); |
| |
| fragBuilder->codeAppendf("float2 coord = %s - %d.0 * %s;", coords2D.c_str(), ce.radius(), imgInc); |
| fragBuilder->codeAppend("float2 coordSampled = half2(0, 0);"); |
| |
| // Manually unroll loop because some drivers don't; yields 20-30% speedup. |
| const char* kVecSuffix[4] = {".x", ".y", ".z", ".w"}; |
| for (int i = 0; i < width; i++) { |
| SkString index; |
| SkString kernelIndex; |
| index.appendS32(i / 4); |
| kernel.appendArrayAccess(index.c_str(), &kernelIndex); |
| kernelIndex.append(kVecSuffix[i & 0x3]); |
| |
| fragBuilder->codeAppend("coordSampled = coord;"); |
| if (ce.useBounds()) { |
| // We used to compute a bool indicating whether we're in bounds or not, cast it to a |
| // float, and then mul weight*texture_sample by the float. However, the Adreno 430 seems |
| // to have a bug that caused corruption. |
| const char* bounds = uniformHandler->getUniformCStr(fBoundsUni); |
| const char* component = ce.direction() == Direction::kY ? "y" : "x"; |
| |
| switch (ce.mode()) { |
| case GrTextureDomain::kClamp_Mode: { |
| fragBuilder->codeAppendf("coordSampled.%s = clamp(coord.%s, %s.x, %s.y);\n", |
| component, component, bounds, bounds); |
| break; |
| } |
| case GrTextureDomain::kRepeat_Mode: { |
| fragBuilder->codeAppendf("coordSampled.%s = " |
| "mod(coord.%s - %s.x, %s.y - %s.x) + %s.x;\n", |
| component, component, bounds, bounds, bounds, bounds); |
| break; |
| } |
| case GrTextureDomain::kDecal_Mode: { |
| fragBuilder->codeAppendf("if (coord.%s >= %s.x && coord.%s <= %s.y) {", |
| component, bounds, component, bounds); |
| break; |
| } |
| default: { |
| SK_ABORT("Unsupported operation."); |
| } |
| } |
| } |
| fragBuilder->codeAppendf("%s += ", args.fOutputColor); |
| fragBuilder->appendTextureLookup(args.fTexSamplers[0], "coordSampled"); |
| fragBuilder->codeAppendf(" * %s;\n", kernelIndex.c_str()); |
| if (GrTextureDomain::kDecal_Mode == ce.mode()) { |
| fragBuilder->codeAppend("}"); |
| } |
| fragBuilder->codeAppendf("coord += %s;\n", imgInc); |
| } |
| fragBuilder->codeAppendf("%s *= %s;\n", args.fOutputColor, args.fInputColor); |
| } |
| |
| void GrGLConvolutionEffect::onSetData(const GrGLSLProgramDataManager& pdman, |
| const GrFragmentProcessor& processor) { |
| const GrGaussianConvolutionFragmentProcessor& conv = |
| processor.cast<GrGaussianConvolutionFragmentProcessor>(); |
| GrSurfaceProxy* proxy = conv.textureSampler(0).proxy(); |
| GrTexture& texture = *proxy->peekTexture(); |
| |
| float imageIncrement[2] = {0}; |
| float ySign = proxy->origin() != kTopLeft_GrSurfaceOrigin ? 1.0f : -1.0f; |
| switch (conv.direction()) { |
| case Direction::kX: |
| imageIncrement[0] = 1.0f / texture.width(); |
| break; |
| case Direction::kY: |
| imageIncrement[1] = ySign / texture.height(); |
| break; |
| default: |
| SK_ABORT("Unknown filter direction."); |
| } |
| pdman.set2fv(fImageIncrementUni, 1, imageIncrement); |
| if (conv.useBounds()) { |
| float bounds[2] = {0}; |
| bounds[0] = conv.bounds()[0]; |
| bounds[1] = conv.bounds()[1]; |
| if (GrTextureDomain::kClamp_Mode == conv.mode()) { |
| bounds[0] += SK_ScalarHalf; |
| bounds[1] -= SK_ScalarHalf; |
| } |
| if (Direction::kX == conv.direction()) { |
| SkScalar inv = SkScalarInvert(SkIntToScalar(texture.width())); |
| bounds[0] *= inv; |
| bounds[1] *= inv; |
| } else { |
| SkScalar inv = SkScalarInvert(SkIntToScalar(texture.height())); |
| if (proxy->origin() != kTopLeft_GrSurfaceOrigin) { |
| float tmp = bounds[0]; |
| bounds[0] = 1.0f - (inv * bounds[1]); |
| bounds[1] = 1.0f - (inv * tmp); |
| } else { |
| bounds[0] *= inv; |
| bounds[1] *= inv; |
| } |
| } |
| |
| SkASSERT(bounds[0] <= bounds[1]); |
| pdman.set2f(fBoundsUni, bounds[0], bounds[1]); |
| } |
| int width = conv.width(); |
| |
| int arrayCount = (width + 3) / 4; |
| SkASSERT(4 * arrayCount >= width); |
| pdman.set4fv(fKernelUni, arrayCount, conv.kernel()); |
| } |
| |
| void GrGLConvolutionEffect::GenKey(const GrProcessor& processor, const GrShaderCaps&, |
| GrProcessorKeyBuilder* b) { |
| const GrGaussianConvolutionFragmentProcessor& conv = |
| processor.cast<GrGaussianConvolutionFragmentProcessor>(); |
| uint32_t key = conv.radius(); |
| key <<= 3; |
| if (conv.useBounds()) { |
| key |= Direction::kY == conv.direction() ? 0x4 : 0x0; |
| } |
| key |= static_cast<uint32_t>(conv.mode()); |
| b->add32(key); |
| } |
| |
| /////////////////////////////////////////////////////////////////////////////// |
| static void fill_in_1D_gaussian_kernel(float* kernel, int width, float gaussianSigma, int radius) { |
| const float twoSigmaSqrd = 2.0f * gaussianSigma * gaussianSigma; |
| if (SkScalarNearlyZero(twoSigmaSqrd, SK_ScalarNearlyZero)) { |
| for (int i = 0; i < width; ++i) { |
| kernel[i] = 0.0f; |
| } |
| return; |
| } |
| |
| const float denom = 1.0f / twoSigmaSqrd; |
| |
| float sum = 0.0f; |
| for (int i = 0; i < width; ++i) { |
| float x = static_cast<float>(i - radius); |
| // Note that the constant term (1/(sqrt(2*pi*sigma^2)) of the Gaussian |
| // is dropped here, since we renormalize the kernel below. |
| kernel[i] = sk_float_exp(-x * x * denom); |
| sum += kernel[i]; |
| } |
| // Normalize the kernel |
| float scale = 1.0f / sum; |
| for (int i = 0; i < width; ++i) { |
| kernel[i] *= scale; |
| } |
| } |
| |
| GrGaussianConvolutionFragmentProcessor::GrGaussianConvolutionFragmentProcessor( |
| sk_sp<GrTextureProxy> proxy, |
| SkAlphaType alphaType, |
| Direction direction, |
| int radius, |
| float gaussianSigma, |
| GrTextureDomain::Mode mode, |
| int bounds[2]) |
| : INHERITED(kGrGaussianConvolutionFragmentProcessor_ClassID, |
| ModulateForSamplerOptFlags(alphaType, mode == GrTextureDomain::kDecal_Mode)) |
| , fCoordTransform(proxy.get()) |
| , fTextureSampler(std::move(proxy)) |
| , fRadius(radius) |
| , fDirection(direction) |
| , fMode(mode) { |
| // Make sure the sampler's ctor uses the clamp wrap mode |
| SkASSERT(fTextureSampler.samplerState().wrapModeX() == GrSamplerState::WrapMode::kClamp && |
| fTextureSampler.samplerState().wrapModeY() == GrSamplerState::WrapMode::kClamp); |
| this->addCoordTransform(&fCoordTransform); |
| this->setTextureSamplerCnt(1); |
| SkASSERT(radius <= kMaxKernelRadius); |
| |
| fill_in_1D_gaussian_kernel(fKernel, this->width(), gaussianSigma, this->radius()); |
| |
| memcpy(fBounds, bounds, sizeof(fBounds)); |
| } |
| |
| GrGaussianConvolutionFragmentProcessor::GrGaussianConvolutionFragmentProcessor( |
| const GrGaussianConvolutionFragmentProcessor& that) |
| : INHERITED(kGrGaussianConvolutionFragmentProcessor_ClassID, that.optimizationFlags()) |
| , fCoordTransform(that.fCoordTransform) |
| , fTextureSampler(that.fTextureSampler) |
| , fRadius(that.fRadius) |
| , fDirection(that.fDirection) |
| , fMode(that.fMode) { |
| this->addCoordTransform(&fCoordTransform); |
| this->setTextureSamplerCnt(1); |
| memcpy(fKernel, that.fKernel, that.width() * sizeof(float)); |
| memcpy(fBounds, that.fBounds, sizeof(fBounds)); |
| } |
| |
| void GrGaussianConvolutionFragmentProcessor::onGetGLSLProcessorKey(const GrShaderCaps& caps, |
| GrProcessorKeyBuilder* b) const { |
| GrGLConvolutionEffect::GenKey(*this, caps, b); |
| } |
| |
| GrGLSLFragmentProcessor* GrGaussianConvolutionFragmentProcessor::onCreateGLSLInstance() const { |
| return new GrGLConvolutionEffect; |
| } |
| |
| bool GrGaussianConvolutionFragmentProcessor::onIsEqual(const GrFragmentProcessor& sBase) const { |
| const GrGaussianConvolutionFragmentProcessor& s = |
| sBase.cast<GrGaussianConvolutionFragmentProcessor>(); |
| return (this->radius() == s.radius() && this->direction() == s.direction() && |
| this->mode() == s.mode() && |
| 0 == memcmp(fBounds, s.fBounds, sizeof(fBounds)) && |
| 0 == memcmp(fKernel, s.fKernel, this->width() * sizeof(float))); |
| } |
| |
| /////////////////////////////////////////////////////////////////////////////// |
| |
| GR_DEFINE_FRAGMENT_PROCESSOR_TEST(GrGaussianConvolutionFragmentProcessor); |
| |
| #if GR_TEST_UTILS |
| std::unique_ptr<GrFragmentProcessor> GrGaussianConvolutionFragmentProcessor::TestCreate( |
| GrProcessorTestData* d) { |
| int texIdx = d->fRandom->nextBool() ? GrProcessorUnitTest::kSkiaPMTextureIdx |
| : GrProcessorUnitTest::kAlphaTextureIdx; |
| sk_sp<GrTextureProxy> proxy = d->textureProxy(texIdx); |
| |
| int bounds[2]; |
| int modeIdx = d->fRandom->nextRangeU(0, GrTextureDomain::kModeCount-1); |
| |
| Direction dir; |
| if (d->fRandom->nextBool()) { |
| dir = Direction::kX; |
| bounds[0] = d->fRandom->nextRangeU(0, proxy->width()-2); |
| bounds[1] = d->fRandom->nextRangeU(bounds[0]+1, proxy->width()-1); |
| } else { |
| dir = Direction::kY; |
| bounds[0] = d->fRandom->nextRangeU(0, proxy->height()-2); |
| bounds[1] = d->fRandom->nextRangeU(bounds[0]+1, proxy->height()-1); |
| } |
| |
| int radius = d->fRandom->nextRangeU(1, kMaxKernelRadius); |
| float sigma = radius / 3.f; |
| |
| auto alphaType = static_cast<SkAlphaType>( |
| d->fRandom->nextRangeU(kUnknown_SkAlphaType + 1, kLastEnum_SkAlphaType)); |
| return GrGaussianConvolutionFragmentProcessor::Make( |
| std::move(proxy), alphaType, dir, radius, sigma, |
| static_cast<GrTextureDomain::Mode>(modeIdx), bounds); |
| } |
| #endif |