blob: e4b186621ee9037896fcf6f2f521dceed806d2ab [file] [log] [blame]
/*
* Copyright 2014 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#ifndef SkLocalMatrixShader_DEFINED
#define SkLocalMatrixShader_DEFINED
#include "src/core/SkReadBuffer.h"
#include "src/core/SkWriteBuffer.h"
#include "src/shaders/SkShaderBase.h"
class GrFragmentProcessor;
class SkArenaAlloc;
class SkLocalMatrixShader final : public SkShaderBase {
public:
template <typename T, typename... Args>
static std::enable_if_t<std::is_base_of_v<SkShader, T>, sk_sp<SkShader>>
MakeWrapped(const SkMatrix* localMatrix, Args&&... args) {
auto t = sk_make_sp<T>(std::forward<Args>(args)...);
if (!localMatrix || localMatrix->isIdentity()) {
return std::move(t);
}
return sk_make_sp<SkLocalMatrixShader>(sk_sp<SkShader>(std::move(t)), *localMatrix);
}
SkLocalMatrixShader(sk_sp<SkShader> wrapped, const SkMatrix& localMatrix)
: fLocalMatrix(localMatrix), fWrappedShader(std::move(wrapped)) {}
GradientType asGradient(GradientInfo* info, SkMatrix* localMatrix) const override;
#if defined(SK_GANESH)
std::unique_ptr<GrFragmentProcessor> asFragmentProcessor(const GrFPArgs&,
const MatrixRec&) const override;
#endif
#if defined(SK_GRAPHITE)
void addToKey(const skgpu::graphite::KeyContext&,
skgpu::graphite::PaintParamsKeyBuilder*,
skgpu::graphite::PipelineDataGatherer*) const override;
#endif
sk_sp<SkShader> makeAsALocalMatrixShader(SkMatrix* localMatrix) const override {
if (localMatrix) {
*localMatrix = fLocalMatrix;
}
return fWrappedShader;
}
protected:
void flatten(SkWriteBuffer&) const override;
#ifdef SK_ENABLE_LEGACY_SHADERCONTEXT
Context* onMakeContext(const ContextRec&, SkArenaAlloc*) const override;
#endif
SkImage* onIsAImage(SkMatrix* matrix, SkTileMode* mode) const override;
bool appendStages(const SkStageRec&, const MatrixRec&) const override;
skvm::Color program(skvm::Builder*,
skvm::Coord device,
skvm::Coord local,
skvm::Color paint,
const MatrixRec&,
const SkColorInfo& dst,
skvm::Uniforms* uniforms,
SkArenaAlloc*) const override;
private:
SK_FLATTENABLE_HOOKS(SkLocalMatrixShader)
SkMatrix fLocalMatrix;
sk_sp<SkShader> fWrappedShader;
using INHERITED = SkShaderBase;
};
#endif