blob: 3de22b76d00803991ee87a9d8ec1ea4f038cb72d [file] [log] [blame]
* Copyright 2012 Google Inc.
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
#ifndef SkGradientShaderPriv_DEFINED
#define SkGradientShaderPriv_DEFINED
#include "include/effects/SkGradientShader.h"
#include "include/core/SkMatrix.h"
#include "include/private/SkTArray.h"
#include "include/private/SkTemplates.h"
#include "src/core/SkArenaAlloc.h"
#include "src/core/SkVM.h"
#include "src/shaders/SkShaderBase.h"
class SkColorSpace;
class SkRasterPipeline;
class SkReadBuffer;
class SkWriteBuffer;
class SkGradientShaderBase : public SkShaderBase {
struct Descriptor {
const SkMatrix* fLocalMatrix;
const SkColor4f* fColors;
sk_sp<SkColorSpace> fColorSpace;
const SkScalar* fPos;
int fCount;
SkTileMode fTileMode;
uint32_t fGradFlags;
void flatten(SkWriteBuffer&) const;
class DescriptorScope : public Descriptor {
DescriptorScope() {}
bool unflatten(SkReadBuffer&);
// fColors and fPos always point into local memory, so they can be safely mutated
SkColor4f* mutableColors() { return const_cast<SkColor4f*>(fColors); }
SkScalar* mutablePos() { return const_cast<SkScalar*>(fPos); }
SkSTArray<16, SkColor4f, true> fColorStorage;
SkSTArray<16, SkScalar , true> fPosStorage;
SkMatrix fLocalMatrixStorage;
SkGradientShaderBase(const Descriptor& desc, const SkMatrix& ptsToUnit);
~SkGradientShaderBase() override;
bool isOpaque() const override;
uint32_t getGradFlags() const { return fGradFlags; }
const SkMatrix& getGradientMatrix() const { return fPtsToUnit; }
class GradientShaderBase4fContext;
SkGradientShaderBase(SkReadBuffer& );
void flatten(SkWriteBuffer&) const override;
void commonAsAGradient(GradientInfo*) const;
bool onAsLuminanceColor(SkColor*) const override;
bool onAppendStages(const SkStageRec&) const override;
skvm::Color onProgram(skvm::Builder*, skvm::Coord device, skvm::Coord local, skvm::Color paint,
const SkMatrixProvider&, const SkMatrix* localM, const SkColorInfo& dstCS,
skvm::Uniforms* uniforms, SkArenaAlloc* alloc) const override;
virtual void appendGradientStages(SkArenaAlloc* alloc, SkRasterPipeline* tPipeline,
SkRasterPipeline* postPipeline) const = 0;
// Produce t from (x,y), modifying mask if it should be anything other than ~0.
virtual skvm::F32 transformT(skvm::Builder*, skvm::Uniforms*,
skvm::Coord coord, skvm::I32* mask) const = 0;
template <typename T, typename... Args>
static Context* CheckedMakeContext(SkArenaAlloc* alloc, Args&&... args) {
auto* ctx = alloc->make<T>(std::forward<Args>(args)...);
if (!ctx->isValid()) {
return nullptr;
return ctx;
const SkMatrix fPtsToUnit;
SkTileMode fTileMode;
uint8_t fGradFlags;
SkScalar getPos(int i) const {
SkASSERT(i < fColorCount);
return fOrigPos ? fOrigPos[i] : SkIntToScalar(i) / (fColorCount - 1);
SkColor getLegacyColor(int i) const {
SkASSERT(i < fColorCount);
return fOrigColors4f[i].toSkColor();
bool colorsCanConvertToSkColor() const {
bool canConvert = true;
for (int i = 0; i < fColorCount; ++i) {
canConvert &= fOrigColors4f[i].fitsInBytes();
return canConvert;
SkColor4f* fOrigColors4f; // original colors, as floats
SkScalar* fOrigPos; // original positions
int fColorCount;
sk_sp<SkColorSpace> fColorSpace; // color space of gradient stops
bool colorsAreOpaque() const { return fColorsAreOpaque; }
SkTileMode getTileMode() const { return fTileMode; }
// Reserve inline space for up to 4 stops.
inline static constexpr size_t kInlineStopCount = 4;
inline static constexpr size_t kInlineStorageSize = (sizeof(SkColor4f) + sizeof(SkScalar))
* kInlineStopCount;
SkAutoSTMalloc<kInlineStorageSize, uint8_t> fStorage;
bool fColorsAreOpaque;
using INHERITED = SkShaderBase;
struct SkColor4fXformer {
SkColor4fXformer(const SkColor4f* colors, int colorCount, SkColorSpace* src, SkColorSpace* dst);
const SkColor4f* fColors;
SkSTArray<4, SkColor4f, true> fStorage;