blob: 2326ed19b033479776fb8a8856c0f9cb15fbf328 [file] [log] [blame]
/*
* Copyright 2023 Google LLC
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#ifndef GrPerlinNoise2Effect_DEFINED
#define GrPerlinNoise2Effect_DEFINED
#include "include/core/SkAlphaType.h"
#include "include/core/SkMatrix.h"
#include "include/core/SkPoint.h"
#include "include/core/SkSamplingOptions.h"
#include "include/core/SkString.h"
#include "include/private/SkSLSampleUsage.h"
#include "src/gpu/ganesh/GrCaps.h"
#include "src/gpu/ganesh/GrFragmentProcessor.h"
#include "src/gpu/ganesh/GrProcessorUnitTest.h"
#include "src/gpu/ganesh/GrSamplerState.h"
#include "src/gpu/ganesh/GrSurfaceProxyView.h"
#include "src/gpu/ganesh/effects/GrTextureEffect.h"
#include "src/gpu/ganesh/glsl/GrGLSLProgramDataManager.h"
#include "src/shaders/SkPerlinNoiseShaderImpl.h"
#include <memory>
#include <utility>
namespace skgpu {
class KeyBuilder;
}
struct GrShaderCaps;
enum class SkPerlinNoiseShaderType;
class GrPerlinNoise2Effect : public GrFragmentProcessor {
public:
static std::unique_ptr<GrFragmentProcessor> Make(
SkPerlinNoiseShaderType type,
int numOctaves,
bool stitchTiles,
std::unique_ptr<SkPerlinNoiseShader::PaintingData> paintingData,
GrSurfaceProxyView permutationsView,
GrSurfaceProxyView noiseView,
const GrCaps& caps) {
static constexpr GrSamplerState kRepeatXSampler = {GrSamplerState::WrapMode::kRepeat,
GrSamplerState::WrapMode::kClamp,
GrSamplerState::Filter::kNearest};
auto permutationsFP = GrTextureEffect::Make(std::move(permutationsView),
kPremul_SkAlphaType,
SkMatrix::I(),
kRepeatXSampler,
caps);
auto noiseFP = GrTextureEffect::Make(
std::move(noiseView), kPremul_SkAlphaType, SkMatrix::I(), kRepeatXSampler, caps);
return std::unique_ptr<GrFragmentProcessor>(
new GrPerlinNoise2Effect(type,
numOctaves,
stitchTiles,
std::move(paintingData),
std::move(permutationsFP),
std::move(noiseFP)));
}
const char* name() const override { return "PerlinNoise"; }
std::unique_ptr<GrFragmentProcessor> clone() const override {
return std::unique_ptr<GrFragmentProcessor>(new GrPerlinNoise2Effect(*this));
}
const SkPerlinNoiseShader::StitchData& stitchData() const {
return fPaintingData->fStitchDataInit;
}
SkPerlinNoiseShaderType type() const { return fType; }
bool stitchTiles() const { return fStitchTiles; }
const SkVector& baseFrequency() const { return fPaintingData->fBaseFrequency; }
int numOctaves() const { return fNumOctaves; }
private:
class Impl : public ProgramImpl {
public:
SkString emitHelper(EmitArgs& args);
void emitCode(EmitArgs&) override;
private:
void onSetData(const GrGLSLProgramDataManager&, const GrFragmentProcessor&) override;
GrGLSLProgramDataManager::UniformHandle fStitchDataUni;
GrGLSLProgramDataManager::UniformHandle fBaseFrequencyUni;
};
std::unique_ptr<ProgramImpl> onMakeProgramImpl() const override {
return std::make_unique<Impl>();
}
void onAddToKey(const GrShaderCaps& caps, skgpu::KeyBuilder* b) const override;
bool onIsEqual(const GrFragmentProcessor& sBase) const override {
const GrPerlinNoise2Effect& s = sBase.cast<GrPerlinNoise2Effect>();
return fType == s.fType &&
fPaintingData->fBaseFrequency == s.fPaintingData->fBaseFrequency &&
fNumOctaves == s.fNumOctaves && fStitchTiles == s.fStitchTiles &&
fPaintingData->fStitchDataInit == s.fPaintingData->fStitchDataInit;
}
GrPerlinNoise2Effect(SkPerlinNoiseShaderType type,
int numOctaves,
bool stitchTiles,
std::unique_ptr<SkPerlinNoiseShader::PaintingData> paintingData,
std::unique_ptr<GrFragmentProcessor> permutationsFP,
std::unique_ptr<GrFragmentProcessor> noiseFP)
: GrFragmentProcessor(kGrPerlinNoise2Effect_ClassID, kNone_OptimizationFlags)
, fType(type)
, fNumOctaves(numOctaves)
, fStitchTiles(stitchTiles)
, fPaintingData(std::move(paintingData)) {
this->registerChild(std::move(permutationsFP), SkSL::SampleUsage::Explicit());
this->registerChild(std::move(noiseFP), SkSL::SampleUsage::Explicit());
this->setUsesSampleCoordsDirectly();
}
GrPerlinNoise2Effect(const GrPerlinNoise2Effect& that)
: GrFragmentProcessor(that)
, fType(that.fType)
, fNumOctaves(that.fNumOctaves)
, fStitchTiles(that.fStitchTiles)
, fPaintingData(new SkPerlinNoiseShader::PaintingData(*that.fPaintingData)) {}
GR_DECLARE_FRAGMENT_PROCESSOR_TEST
SkPerlinNoiseShaderType fType;
int fNumOctaves;
bool fStitchTiles;
std::unique_ptr<SkPerlinNoiseShader::PaintingData> fPaintingData;
};
#endif