| /* |
| * Copyright 2019 Google LLC |
| * |
| * Use of this source code is governed by a BSD-style license that can be |
| * found in the LICENSE file. |
| */ |
| |
| #include "include/core/SkColorFilter.h" |
| #include "include/core/SkData.h" |
| #include "include/core/SkSurface.h" |
| #include "include/effects/SkRuntimeEffect.h" |
| #include "include/private/SkMutex.h" |
| #include "src/core/SkBlenderBase.h" |
| #include "src/core/SkCanvasPriv.h" |
| #include "src/core/SkColorFilterBase.h" |
| #include "src/core/SkColorSpacePriv.h" |
| #include "src/core/SkColorSpaceXformSteps.h" |
| #include "src/core/SkLRUCache.h" |
| #include "src/core/SkMatrixProvider.h" |
| #include "src/core/SkOpts.h" |
| #include "src/core/SkRasterPipeline.h" |
| #include "src/core/SkReadBuffer.h" |
| #include "src/core/SkRuntimeEffectPriv.h" |
| #include "src/core/SkUtils.h" |
| #include "src/core/SkVM.h" |
| #include "src/core/SkWriteBuffer.h" |
| #include "src/sksl/SkSLAnalysis.h" |
| #include "src/sksl/SkSLCompiler.h" |
| #include "src/sksl/SkSLUtil.h" |
| #include "src/sksl/codegen/SkSLVMCodeGenerator.h" |
| #include "src/sksl/ir/SkSLFunctionDefinition.h" |
| #include "src/sksl/ir/SkSLVarDeclarations.h" |
| |
| #if SK_SUPPORT_GPU |
| #include "include/gpu/GrRecordingContext.h" |
| #include "src/gpu/GrColorInfo.h" |
| #include "src/gpu/GrFPArgs.h" |
| #include "src/gpu/GrImageInfo.h" |
| #include "src/gpu/GrSurfaceFillContext.h" |
| #include "src/gpu/effects/GrMatrixEffect.h" |
| #include "src/gpu/effects/GrSkSLFP.h" |
| #include "src/image/SkImage_Gpu.h" |
| #endif |
| |
| #include <algorithm> |
| |
| #ifdef SK_ENABLE_SKSL |
| |
| namespace SkSL { |
| class SharedCompiler { |
| public: |
| SharedCompiler() : fLock(compiler_mutex()) { |
| if (!gImpl) { |
| gImpl = new Impl(); |
| } |
| } |
| |
| SkSL::Compiler* operator->() const { return gImpl->fCompiler; } |
| |
| private: |
| SkAutoMutexExclusive fLock; |
| |
| static SkMutex& compiler_mutex() { |
| static SkMutex& mutex = *(new SkMutex); |
| return mutex; |
| } |
| |
| struct Impl { |
| Impl() { |
| // These caps are configured to apply *no* workarounds. This avoids changes that are |
| // unnecessary (GLSL intrinsic rewrites), or possibly incorrect (adding do-while loops). |
| // We may apply other "neutral" transformations to the user's SkSL, including inlining. |
| // Anything determined by the device caps is deferred to the GPU backend. The processor |
| // set produces the final program (including our re-emitted SkSL), and the backend's |
| // compiler resolves any necessary workarounds. |
| fCaps = ShaderCapsFactory::Standalone(); |
| fCaps->fBuiltinFMASupport = true; |
| fCaps->fBuiltinDeterminantSupport = true; |
| // Don't inline if it would require a do loop, some devices don't support them. |
| fCaps->fCanUseDoLoops = false; |
| |
| // SkSL created by the GPU backend is typically parsed, converted to a backend format, |
| // and the IR is immediately discarded. In that situation, it makes sense to use node |
| // pools to accelerate the IR allocations. Here, SkRuntimeEffect instances are often |
| // long-lived (especially those created internally for runtime FPs). In this situation, |
| // we're willing to pay for a slightly longer compile so that we don't waste huge |
| // amounts of memory. |
| fCaps->fUseNodePools = false; |
| |
| fCompiler = new SkSL::Compiler(fCaps.get()); |
| } |
| |
| SkSL::ShaderCapsPointer fCaps; |
| SkSL::Compiler* fCompiler; |
| }; |
| |
| static Impl* gImpl; |
| }; |
| |
| SharedCompiler::Impl* SharedCompiler::gImpl = nullptr; |
| |
| } // namespace SkSL |
| |
| static bool init_uniform_type(const SkSL::Context& ctx, |
| const SkSL::Type* type, |
| SkRuntimeEffect::Uniform* v) { |
| using Type = SkRuntimeEffect::Uniform::Type; |
| if (*type == *ctx.fTypes.fFloat) { v->type = Type::kFloat; return true; } |
| if (*type == *ctx.fTypes.fHalf) { v->type = Type::kFloat; return true; } |
| if (*type == *ctx.fTypes.fFloat2) { v->type = Type::kFloat2; return true; } |
| if (*type == *ctx.fTypes.fHalf2) { v->type = Type::kFloat2; return true; } |
| if (*type == *ctx.fTypes.fFloat3) { v->type = Type::kFloat3; return true; } |
| if (*type == *ctx.fTypes.fHalf3) { v->type = Type::kFloat3; return true; } |
| if (*type == *ctx.fTypes.fFloat4) { v->type = Type::kFloat4; return true; } |
| if (*type == *ctx.fTypes.fHalf4) { v->type = Type::kFloat4; return true; } |
| if (*type == *ctx.fTypes.fFloat2x2) { v->type = Type::kFloat2x2; return true; } |
| if (*type == *ctx.fTypes.fHalf2x2) { v->type = Type::kFloat2x2; return true; } |
| if (*type == *ctx.fTypes.fFloat3x3) { v->type = Type::kFloat3x3; return true; } |
| if (*type == *ctx.fTypes.fHalf3x3) { v->type = Type::kFloat3x3; return true; } |
| if (*type == *ctx.fTypes.fFloat4x4) { v->type = Type::kFloat4x4; return true; } |
| if (*type == *ctx.fTypes.fHalf4x4) { v->type = Type::kFloat4x4; return true; } |
| |
| if (*type == *ctx.fTypes.fInt) { v->type = Type::kInt; return true; } |
| if (*type == *ctx.fTypes.fInt2) { v->type = Type::kInt2; return true; } |
| if (*type == *ctx.fTypes.fInt3) { v->type = Type::kInt3; return true; } |
| if (*type == *ctx.fTypes.fInt4) { v->type = Type::kInt4; return true; } |
| |
| return false; |
| } |
| |
| static SkRuntimeEffect::Child::Type child_type(const SkSL::Type& type) { |
| switch (type.typeKind()) { |
| case SkSL::Type::TypeKind::kColorFilter: return SkRuntimeEffect::Child::Type::kColorFilter; |
| case SkSL::Type::TypeKind::kShader: return SkRuntimeEffect::Child::Type::kShader; |
| default: SkUNREACHABLE; |
| } |
| } |
| |
| static bool verify_child_effects(const std::vector<SkRuntimeEffect::Child>& reflected, |
| SkSpan<SkRuntimeEffect::ChildPtr> effectPtrs) { |
| // Verify that the number of passed-in child-effect pointers matches the SkSL code. |
| if (reflected.size() != effectPtrs.size()) { |
| return false; |
| } |
| // Verify that each child object's type matches its declared type in the SkSL. |
| for (size_t i = 0; i < effectPtrs.size(); ++i) { |
| switch (reflected[i].type) { |
| case SkRuntimeEffect::Child::Type::kShader: |
| if (effectPtrs[i].colorFilter) { |
| return false; |
| } |
| continue; |
| case SkRuntimeEffect::Child::Type::kColorFilter: |
| if (effectPtrs[i].shader) { |
| return false; |
| } |
| continue; |
| } |
| SkDEBUGFAILF("unrecognized SkSL child type %d", (int)reflected[i].type); |
| } |
| return true; |
| } |
| |
| // TODO: Many errors aren't caught until we process the generated Program here. Catching those |
| // in the IR generator would provide better errors messages (with locations). |
| #define RETURN_FAILURE(...) return Result{nullptr, SkStringPrintf(__VA_ARGS__)} |
| |
| SkRuntimeEffect::Result SkRuntimeEffect::MakeFromSource(SkString sksl, |
| const Options& options, |
| SkSL::ProgramKind kind) { |
| std::unique_ptr<SkSL::Program> program; |
| { |
| // We keep this SharedCompiler in a separate scope to make sure it's destroyed before |
| // calling the Make overload at the end, which creates its own (non-reentrant) |
| // SharedCompiler instance |
| SkSL::SharedCompiler compiler; |
| SkSL::Program::Settings settings; |
| settings.fInlineThreshold = 0; |
| settings.fForceNoInline = options.forceNoInline; |
| settings.fEnforceES2Restrictions = options.enforceES2Restrictions; |
| settings.fAllowNarrowingConversions = true; |
| program = compiler->convertProgram(kind, SkSL::String(sksl.c_str(), sksl.size()), settings); |
| |
| if (!program) { |
| RETURN_FAILURE("%s", compiler->errorText().c_str()); |
| } |
| } |
| return MakeInternal(std::move(program), options, kind); |
| } |
| |
| SkRuntimeEffect::Result SkRuntimeEffect::MakeFromDSL(std::unique_ptr<SkSL::Program> program, |
| const Options& options, |
| SkSL::ProgramKind kind) { |
| // This factory is used for all DSL runtime effects, which don't have anything stored in the |
| // program's source. Populate it so that we can compute fHash, and serialize these effects. |
| program->fSource = std::make_unique<SkSL::String>(program->description()); |
| return MakeInternal(std::move(program), options, kind); |
| } |
| |
| SkRuntimeEffect::Result SkRuntimeEffect::MakeInternal(std::unique_ptr<SkSL::Program> program, |
| const Options& options, |
| SkSL::ProgramKind kind) { |
| SkSL::SharedCompiler compiler; |
| |
| // Find 'main', then locate the sample coords parameter. (It might not be present.) |
| const SkSL::FunctionDefinition* main = SkSL::Program_GetFunction(*program, "main"); |
| if (!main) { |
| RETURN_FAILURE("missing 'main' function"); |
| } |
| const auto& mainParams = main->declaration().parameters(); |
| auto iter = std::find_if(mainParams.begin(), mainParams.end(), [](const SkSL::Variable* p) { |
| return p->modifiers().fLayout.fBuiltin == SK_MAIN_COORDS_BUILTIN; |
| }); |
| const SkSL::ProgramUsage::VariableCounts sampleCoordsUsage = |
| iter != mainParams.end() ? program->usage()->get(**iter) |
| : SkSL::ProgramUsage::VariableCounts{}; |
| |
| uint32_t flags = 0; |
| switch (kind) { |
| case SkSL::ProgramKind::kRuntimeColorFilter: flags |= kAllowColorFilter_Flag; break; |
| case SkSL::ProgramKind::kRuntimeShader: flags |= kAllowShader_Flag; break; |
| case SkSL::ProgramKind::kRuntimeBlender: flags |= kAllowBlender_Flag; break; |
| default: SkUNREACHABLE; |
| } |
| |
| if (sampleCoordsUsage.fRead || sampleCoordsUsage.fWrite) { |
| flags |= kUsesSampleCoords_Flag; |
| } |
| |
| // TODO(skia:12202): When we can layer modules, implement this restriction by moving the |
| // declaration of sk_FragCoord to a private module. |
| if (!options.allowFragCoord && SkSL::Analysis::ReferencesFragCoords(*program)) { |
| RETURN_FAILURE("unknown identifier 'sk_FragCoord'"); |
| } |
| |
| // Color filters and blends are not allowed to depend on position (local or device) in any way. |
| // The signature of main, and the declarations in sksl_rt_colorfilter/sksl_rt_blend should |
| // guarantee this. |
| if (flags & (kAllowColorFilter_Flag | kAllowBlender_Flag)) { |
| SkASSERT(!(flags & kUsesSampleCoords_Flag)); |
| SkASSERT(!SkSL::Analysis::ReferencesFragCoords(*program)); |
| } |
| |
| if (SkSL::Analysis::CallsSampleOutsideMain(*program)) { |
| flags |= kSamplesOutsideMain_Flag; |
| } |
| |
| size_t offset = 0; |
| std::vector<Uniform> uniforms; |
| std::vector<Child> children; |
| std::vector<SkSL::SampleUsage> sampleUsages; |
| int elidedSampleCoords = 0; |
| const SkSL::Context& ctx(compiler->context()); |
| |
| // Go through program elements, pulling out information that we need |
| for (const SkSL::ProgramElement* elem : program->elements()) { |
| // Variables (uniform, etc.) |
| if (elem->is<SkSL::GlobalVarDeclaration>()) { |
| const SkSL::GlobalVarDeclaration& global = elem->as<SkSL::GlobalVarDeclaration>(); |
| const SkSL::VarDeclaration& varDecl = global.declaration()->as<SkSL::VarDeclaration>(); |
| |
| const SkSL::Variable& var = varDecl.var(); |
| const SkSL::Type& varType = var.type(); |
| |
| // Child effects that can be sampled ('shader' or 'colorFilter') |
| if (varType.isEffectChild()) { |
| // Runtime blends currently don't support child effects. |
| if (kind == SkSL::ProgramKind::kRuntimeBlender) { |
| RETURN_FAILURE("'%s' is not allowed in runtime blend", |
| varType.displayName().c_str()); |
| } |
| |
| Child c; |
| c.name = SkString(var.name()); |
| c.type = child_type(varType); |
| c.index = children.size(); |
| children.push_back(c); |
| sampleUsages.push_back(SkSL::Analysis::GetSampleUsage( |
| *program, var, sampleCoordsUsage.fWrite != 0, &elidedSampleCoords)); |
| } |
| // 'uniform' variables |
| else if (var.modifiers().fFlags & SkSL::Modifiers::kUniform_Flag) { |
| Uniform uni; |
| uni.name = SkString(var.name()); |
| uni.flags = 0; |
| uni.count = 1; |
| |
| const SkSL::Type* type = &var.type(); |
| if (type->isArray()) { |
| uni.flags |= Uniform::kArray_Flag; |
| uni.count = type->columns(); |
| type = &type->componentType(); |
| } |
| |
| if (!init_uniform_type(ctx, type, &uni)) { |
| RETURN_FAILURE("Invalid uniform type: '%s'", type->displayName().c_str()); |
| } |
| |
| if (var.modifiers().fLayout.fFlags & SkSL::Layout::Flag::kSRGBUnpremul_Flag) { |
| uni.flags |= Uniform::kSRGBUnpremul_Flag; |
| } |
| |
| uni.offset = offset; |
| offset += uni.sizeInBytes(); |
| SkASSERT(SkIsAlign4(offset)); |
| |
| uniforms.push_back(uni); |
| } |
| } |
| } |
| |
| // If the sample coords are never written to, then we will have converted sample calls that use |
| // them unmodified into "passthrough" sampling. If all references to the sample coords were of |
| // that form, then we don't actually "use" sample coords. We unset the flag to prevent creating |
| // an extra (unused) varying holding the coords. |
| if (elidedSampleCoords == sampleCoordsUsage.fRead && sampleCoordsUsage.fWrite == 0) { |
| flags &= ~kUsesSampleCoords_Flag; |
| } |
| |
| #undef RETURN_FAILURE |
| |
| sk_sp<SkRuntimeEffect> effect(new SkRuntimeEffect(std::move(program), |
| options, |
| *main, |
| std::move(uniforms), |
| std::move(children), |
| std::move(sampleUsages), |
| flags)); |
| return Result{std::move(effect), SkString()}; |
| } |
| |
| SkRuntimeEffect::Result SkRuntimeEffect::MakeForColorFilter(SkString sksl, const Options& options) { |
| auto result = MakeFromSource(std::move(sksl), options, SkSL::ProgramKind::kRuntimeColorFilter); |
| SkASSERT(!result.effect || result.effect->allowColorFilter()); |
| return result; |
| } |
| |
| SkRuntimeEffect::Result SkRuntimeEffect::MakeForShader(SkString sksl, const Options& options) { |
| auto result = MakeFromSource(std::move(sksl), options, SkSL::ProgramKind::kRuntimeShader); |
| SkASSERT(!result.effect || result.effect->allowShader()); |
| return result; |
| } |
| |
| SkRuntimeEffect::Result SkRuntimeEffect::MakeForBlender(SkString sksl, const Options& options) { |
| auto result = MakeFromSource(std::move(sksl), options, SkSL::ProgramKind::kRuntimeBlender); |
| SkASSERT(!result.effect || result.effect->allowBlender()); |
| return result; |
| } |
| |
| SkRuntimeEffect::Result SkRuntimeEffect::MakeForColorFilter(std::unique_ptr<SkSL::Program> program, |
| const Options& options) { |
| auto result = MakeFromDSL(std::move(program), options, SkSL::ProgramKind::kRuntimeColorFilter); |
| SkASSERT(!result.effect || result.effect->allowColorFilter()); |
| return result; |
| } |
| |
| SkRuntimeEffect::Result SkRuntimeEffect::MakeForShader(std::unique_ptr<SkSL::Program> program, |
| const Options& options) { |
| auto result = MakeFromDSL(std::move(program), options, SkSL::ProgramKind::kRuntimeShader); |
| SkASSERT(!result.effect || result.effect->allowShader()); |
| return result; |
| } |
| |
| SkRuntimeEffect::Result SkRuntimeEffect::MakeForBlender(std::unique_ptr<SkSL::Program> program, |
| const Options& options) { |
| auto result = MakeFromDSL(std::move(program), options, SkSL::ProgramKind::kRuntimeBlender); |
| SkASSERT(!result.effect || result.effect->allowBlender()); |
| return result; |
| } |
| |
| SkRuntimeEffect::Result SkRuntimeEffect::MakeForColorFilter( |
| std::unique_ptr<SkSL::Program> program) { |
| return MakeForColorFilter(std::move(program), Options{}); |
| } |
| |
| SkRuntimeEffect::Result SkRuntimeEffect::MakeForShader(std::unique_ptr<SkSL::Program> program) { |
| return MakeForShader(std::move(program), Options{}); |
| } |
| |
| SkRuntimeEffect::Result SkRuntimeEffect::MakeForBlender(std::unique_ptr<SkSL::Program> program) { |
| return MakeForBlender(std::move(program), Options{}); |
| } |
| |
| sk_sp<SkRuntimeEffect> SkMakeCachedRuntimeEffect(SkRuntimeEffect::Result (*make)(SkString sksl), |
| SkString sksl) { |
| SK_BEGIN_REQUIRE_DENSE |
| struct Key { |
| uint32_t skslHashA; |
| uint32_t skslHashB; |
| |
| bool operator==(const Key& that) const { |
| return this->skslHashA == that.skslHashA |
| && this->skslHashB == that.skslHashB; |
| } |
| |
| explicit Key(const SkString& sksl) |
| : skslHashA(SkOpts::hash(sksl.c_str(), sksl.size(), 0)) |
| , skslHashB(SkOpts::hash(sksl.c_str(), sksl.size(), 1)) {} |
| }; |
| SK_END_REQUIRE_DENSE |
| |
| static auto* mutex = new SkMutex; |
| static auto* cache = new SkLRUCache<Key, sk_sp<SkRuntimeEffect>>(11/*totally arbitrary*/); |
| |
| Key key(sksl); |
| { |
| SkAutoMutexExclusive _(*mutex); |
| if (sk_sp<SkRuntimeEffect>* found = cache->find(key)) { |
| return *found; |
| } |
| } |
| |
| auto [effect, err] = make(std::move(sksl)); |
| if (!effect) { |
| return nullptr; |
| } |
| SkASSERT(err.isEmpty()); |
| |
| { |
| SkAutoMutexExclusive _(*mutex); |
| cache->insert_or_update(key, effect); |
| } |
| return effect; |
| } |
| |
| size_t SkRuntimeEffect::Uniform::sizeInBytes() const { |
| static_assert(sizeof(int) == sizeof(float)); |
| auto element_size = [](Type type) -> size_t { |
| switch (type) { |
| case Type::kFloat: return sizeof(float); |
| case Type::kFloat2: return sizeof(float) * 2; |
| case Type::kFloat3: return sizeof(float) * 3; |
| case Type::kFloat4: return sizeof(float) * 4; |
| |
| case Type::kFloat2x2: return sizeof(float) * 4; |
| case Type::kFloat3x3: return sizeof(float) * 9; |
| case Type::kFloat4x4: return sizeof(float) * 16; |
| |
| case Type::kInt: return sizeof(int); |
| case Type::kInt2: return sizeof(int) * 2; |
| case Type::kInt3: return sizeof(int) * 3; |
| case Type::kInt4: return sizeof(int) * 4; |
| default: SkUNREACHABLE; |
| } |
| }; |
| return element_size(this->type) * this->count; |
| } |
| |
| SkRuntimeEffect::SkRuntimeEffect(std::unique_ptr<SkSL::Program> baseProgram, |
| const Options& options, |
| const SkSL::FunctionDefinition& main, |
| std::vector<Uniform>&& uniforms, |
| std::vector<Child>&& children, |
| std::vector<SkSL::SampleUsage>&& sampleUsages, |
| uint32_t flags) |
| : fHash(SkOpts::hash_fn(baseProgram->fSource->c_str(), baseProgram->fSource->size(), 0)) |
| , fBaseProgram(std::move(baseProgram)) |
| , fMain(main) |
| , fUniforms(std::move(uniforms)) |
| , fChildren(std::move(children)) |
| , fSampleUsages(std::move(sampleUsages)) |
| , fFlags(flags) { |
| SkASSERT(fBaseProgram); |
| SkASSERT(fChildren.size() == fSampleUsages.size()); |
| |
| // Everything from SkRuntimeEffect::Options which could influence the compiled result needs to |
| // be accounted for in `fHash`. If you've added a new field to Options and caused the static- |
| // assert below to trigger, please incorporate your field into `fHash` and update KnownOptions |
| // to match the layout of Options. |
| struct KnownOptions { bool forceNoInline, enforceES2Restrictions, allowFragCoord; }; |
| static_assert(sizeof(Options) == sizeof(KnownOptions)); |
| fHash = SkOpts::hash_fn(&options.forceNoInline, |
| sizeof(options.forceNoInline), fHash); |
| fHash = SkOpts::hash_fn(&options.enforceES2Restrictions, |
| sizeof(options.enforceES2Restrictions), fHash); |
| fHash = SkOpts::hash_fn(&options.allowFragCoord, |
| sizeof(options.allowFragCoord), fHash); |
| |
| fFilterColorProgram = SkFilterColorProgram::Make(this); |
| } |
| |
| SkRuntimeEffect::~SkRuntimeEffect() = default; |
| |
| const std::string& SkRuntimeEffect::source() const { |
| return *fBaseProgram->fSource; |
| } |
| |
| size_t SkRuntimeEffect::uniformSize() const { |
| return fUniforms.empty() ? 0 |
| : SkAlign4(fUniforms.back().offset + fUniforms.back().sizeInBytes()); |
| } |
| |
| const SkRuntimeEffect::Uniform* SkRuntimeEffect::findUniform(const char* name) const { |
| SkASSERT(name); |
| size_t len = strlen(name); |
| auto iter = std::find_if(fUniforms.begin(), fUniforms.end(), [name, len](const Uniform& u) { |
| return u.name.equals(name, len); |
| }); |
| return iter == fUniforms.end() ? nullptr : &(*iter); |
| } |
| |
| const SkRuntimeEffect::Child* SkRuntimeEffect::findChild(const char* name) const { |
| SkASSERT(name); |
| size_t len = strlen(name); |
| auto iter = std::find_if(fChildren.begin(), fChildren.end(), [name, len](const Child& c) { |
| return c.name.equals(name, len); |
| }); |
| return iter == fChildren.end() ? nullptr : &(*iter); |
| } |
| |
| std::unique_ptr<SkFilterColorProgram> SkFilterColorProgram::Make(const SkRuntimeEffect* effect) { |
| // Our per-effect program technique is only possible (and necessary) for color filters |
| if (!effect->allowColorFilter()) { |
| return nullptr; |
| } |
| |
| // We allocate a uniform color for the input color, and one for each call to sample(). When we |
| // encounter a sample call, we record the index of the child being sampled, as well as the color |
| // being passed. In most cases, we can record enough information to perfectly re-create that |
| // call when we're later running the program. (We support calls that pass the original input |
| // color, an immediate color, or the results of a previous sample call). If the color is none |
| // of those, we are unable to use this per-effect program, and callers will need to fall back |
| // to another (slower) implementation. |
| |
| // We also require that any children are *also* color filters (not shaders). In theory we could |
| // detect the coords being passed to shader children, and replicate those calls, but that's |
| // very complicated, and has diminishing returns. (eg, for table lookup color filters). |
| if (!std::all_of(effect->fChildren.begin(), |
| effect->fChildren.end(), |
| [](const SkRuntimeEffect::Child& c) { |
| return c.type == SkRuntimeEffect::Child::Type::kColorFilter; |
| })) { |
| return nullptr; |
| } |
| |
| skvm::Builder p; |
| |
| // For SkSL uniforms, we reserve space and allocate skvm Uniform ids for each one. When we run |
| // the program, these ids will be loads from the *first* arg ptr, the uniform data of the |
| // specific color filter instance. |
| skvm::Uniforms skslUniforms{p.uniform(), 0}; |
| const size_t uniformCount = effect->uniformSize() / 4; |
| std::vector<skvm::Val> uniform; |
| uniform.reserve(uniformCount); |
| for (size_t i = 0; i < uniformCount; i++) { |
| uniform.push_back(p.uniform32(skslUniforms.push(/*placeholder*/ 0)).id); |
| } |
| |
| auto is_simple_uniform = [&](skvm::Color c, int* baseOffset) { |
| skvm::Uniform ur, ug, ub, ua; |
| if (!p.allUniform(c.r.id, &ur, c.g.id, &ug, c.b.id, &ub, c.a.id, &ua)) { |
| return false; |
| } |
| skvm::Ptr uniPtr = skslUniforms.base; |
| if (ur.ptr != uniPtr || ug.ptr != uniPtr || ub.ptr != uniPtr || ua.ptr != uniPtr) { |
| return false; |
| } |
| *baseOffset = ur.offset; |
| return ug.offset == ur.offset + 4 && |
| ub.offset == ur.offset + 8 && |
| ua.offset == ur.offset + 12; |
| }; |
| |
| // We reserve a uniform color for each call to sample(). While processing the SkSL, we record |
| // the index of the child being sampled, and the color being filtered (in a SampleCall struct). |
| // When we run this program later, we use the SampleCall to evaluate the correct child, and |
| // populate these uniform values. These Uniform ids are loads from the *second* arg ptr. |
| skvm::Uniforms childColorUniforms{p.uniform(), 0}; |
| skvm::Color inputColor = p.uniformColor(/*placeholder*/ SkColors::kWhite, &childColorUniforms); |
| std::vector<SkFilterColorProgram::SampleCall> sampleCalls; |
| std::vector<skvm::Color> childColors; |
| auto ids_equal = [](skvm::Color x, skvm::Color y) { |
| return x.r.id == y.r.id && x.g.id == y.g.id && x.b.id == y.b.id && x.a.id == y.a.id; |
| }; |
| bool allSampleCallsSupported = true; |
| auto sampleChild = [&](int ix, skvm::Coord, skvm::Color c) { |
| skvm::Color result = p.uniformColor(/*placeholder*/ SkColors::kWhite, &childColorUniforms); |
| SkFilterColorProgram::SampleCall call; |
| call.fChild = ix; |
| if (ids_equal(c, inputColor)) { |
| call.fKind = SkFilterColorProgram::SampleCall::Kind::kInputColor; |
| } else if (p.allImm(c.r.id, &call.fImm.fR, |
| c.g.id, &call.fImm.fG, |
| c.b.id, &call.fImm.fB, |
| c.a.id, &call.fImm.fA)) { |
| call.fKind = SkFilterColorProgram::SampleCall::Kind::kImmediate; |
| } else if (auto it = std::find_if(childColors.begin(), |
| childColors.end(), |
| [&](skvm::Color x) { return ids_equal(x, c); }); |
| it != childColors.end()) { |
| call.fKind = SkFilterColorProgram::SampleCall::Kind::kPrevious; |
| call.fPrevious = SkTo<int>(it - childColors.begin()); |
| } else if (is_simple_uniform(c, &call.fOffset)) { |
| call.fKind = SkFilterColorProgram::SampleCall::Kind::kUniform; |
| } else { |
| allSampleCallsSupported = false; |
| } |
| sampleCalls.push_back(call); |
| childColors.push_back(result); |
| return result; |
| }; |
| |
| // Emit the skvm instructions for the SkSL |
| skvm::Coord zeroCoord = {p.splat(0.0f), p.splat(0.0f)}; |
| skvm::Color result = SkSL::ProgramToSkVM(*effect->fBaseProgram, |
| effect->fMain, |
| &p, |
| SkMakeSpan(uniform), |
| /*device=*/zeroCoord, |
| /*local=*/zeroCoord, |
| inputColor, |
| inputColor, |
| sampleChild); |
| |
| // Then store the result to the *third* arg ptr |
| p.store({skvm::PixelFormat::FLOAT, 32, 32, 32, 32, 0, 32, 64, 96}, p.arg(16), result); |
| |
| if (!allSampleCallsSupported) { |
| return nullptr; |
| } |
| |
| // This is conservative. If a filter gets the input color by sampling a null child, we'll |
| // return an (acceptable) false negative. All internal runtime color filters should work. |
| bool alphaUnchanged = (inputColor.a.id == result.a.id); |
| |
| // We'll use this program to filter one color at a time, don't bother with jit |
| return std::unique_ptr<SkFilterColorProgram>( |
| new SkFilterColorProgram(p.done(/*debug_name=*/nullptr, /*allow_jit=*/false), |
| std::move(sampleCalls), |
| alphaUnchanged)); |
| } |
| |
| SkFilterColorProgram::SkFilterColorProgram(skvm::Program program, |
| std::vector<SampleCall> sampleCalls, |
| bool alphaUnchanged) |
| : fProgram(std::move(program)) |
| , fSampleCalls(std::move(sampleCalls)) |
| , fAlphaUnchanged(alphaUnchanged) {} |
| |
| SkPMColor4f SkFilterColorProgram::eval( |
| const SkPMColor4f& inColor, |
| const void* uniformData, |
| std::function<SkPMColor4f(int, SkPMColor4f)> evalChild) const { |
| // Our program defines sampling any child as returning a uniform color. Assemble a buffer |
| // containing those colors. The first entry is always the input color. Subsequent entries |
| // are for each sample call, based on the information in fSampleCalls. For any null children, |
| // the sample result is just the passed-in color. |
| SkSTArray<4, SkPMColor4f, true> childColors; |
| childColors.push_back(inColor); |
| for (const auto& s : fSampleCalls) { |
| SkPMColor4f passedColor = inColor; |
| switch (s.fKind) { |
| case SampleCall::Kind::kInputColor: break; |
| case SampleCall::Kind::kImmediate: passedColor = s.fImm; break; |
| case SampleCall::Kind::kPrevious: passedColor = childColors[s.fPrevious + 1]; break; |
| case SampleCall::Kind::kUniform: |
| passedColor = *SkTAddOffset<const SkPMColor4f>(uniformData, s.fOffset); |
| break; |
| } |
| childColors.push_back(evalChild(s.fChild, passedColor)); |
| } |
| |
| SkPMColor4f result; |
| fProgram.eval(1, uniformData, childColors.begin(), result.vec()); |
| return result; |
| } |
| |
| const SkFilterColorProgram* SkRuntimeEffect::getFilterColorProgram() { |
| return fFilterColorProgram.get(); |
| } |
| |
| /////////////////////////////////////////////////////////////////////////////////////////////////// |
| |
| static sk_sp<SkData> get_xformed_uniforms(const SkRuntimeEffect* effect, |
| sk_sp<SkData> baseUniforms, |
| const SkColorSpace* dstCS) { |
| using Flags = SkRuntimeEffect::Uniform::Flags; |
| using Type = SkRuntimeEffect::Uniform::Type; |
| SkColorSpaceXformSteps steps(sk_srgb_singleton(), kUnpremul_SkAlphaType, |
| dstCS, kUnpremul_SkAlphaType); |
| |
| sk_sp<SkData> uniforms = nullptr; |
| auto writableData = [&]() { |
| if (!uniforms) { |
| uniforms = SkData::MakeWithCopy(baseUniforms->data(), baseUniforms->size()); |
| } |
| return uniforms->writable_data(); |
| }; |
| |
| for (const auto& v : effect->uniforms()) { |
| if (v.flags & Flags::kSRGBUnpremul_Flag) { |
| SkASSERT(v.type == Type::kFloat3 || v.type == Type::kFloat4); |
| if (steps.flags.mask()) { |
| float* color = SkTAddOffset<float>(writableData(), v.offset); |
| if (v.type == Type::kFloat4) { |
| // RGBA, easy case |
| for (int i = 0; i < v.count; ++i) { |
| steps.apply(color); |
| color += 4; |
| } |
| } else { |
| // RGB, need to pad out to include alpha. Technically, this isn't necessary, |
| // because steps shouldn't include unpremul or premul, and thus shouldn't |
| // read or write the fourth element. But let's be safe. |
| float rgba[4]; |
| for (int i = 0; i < v.count; ++i) { |
| memcpy(rgba, color, 3 * sizeof(float)); |
| rgba[3] = 1.0f; |
| steps.apply(rgba); |
| memcpy(color, rgba, 3 * sizeof(float)); |
| color += 3; |
| } |
| } |
| } |
| } |
| } |
| return uniforms ? uniforms : baseUniforms; |
| } |
| |
| #if SK_SUPPORT_GPU |
| static GrFPResult make_effect_fp(sk_sp<SkRuntimeEffect> effect, |
| const char* name, |
| sk_sp<SkData> uniforms, |
| std::unique_ptr<GrFragmentProcessor> inputFP, |
| SkSpan<const SkRuntimeEffect::ChildPtr> children, |
| const GrFPArgs& childArgs) { |
| SkSTArray<8, std::unique_ptr<GrFragmentProcessor>> childFPs; |
| for (const auto& child : children) { |
| if (child.shader) { |
| auto childFP = as_SB(child.shader)->asFragmentProcessor(childArgs); |
| if (!childFP) { |
| return GrFPFailure(std::move(inputFP)); |
| } |
| childFPs.push_back(std::move(childFP)); |
| } else if (child.colorFilter) { |
| auto [success, childFP] = as_CFB(child.colorFilter) |
| ->asFragmentProcessor(/*inputFP=*/nullptr, |
| childArgs.fContext, |
| *childArgs.fDstColorInfo); |
| if (!success) { |
| return GrFPFailure(std::move(inputFP)); |
| } |
| childFPs.push_back(std::move(childFP)); |
| } else { |
| childFPs.push_back(nullptr); |
| } |
| } |
| auto fp = GrSkSLFP::MakeWithData(std::move(effect), |
| name, |
| std::move(inputFP), |
| /*destColorFP=*/nullptr, |
| std::move(uniforms), |
| SkMakeSpan(childFPs)); |
| SkASSERT(fp); |
| return GrFPSuccess(std::move(fp)); |
| } |
| #endif |
| |
| class SkRuntimeColorFilter : public SkColorFilterBase { |
| public: |
| SkRuntimeColorFilter(sk_sp<SkRuntimeEffect> effect, |
| sk_sp<SkData> uniforms, |
| SkSpan<SkRuntimeEffect::ChildPtr> children) |
| : fEffect(std::move(effect)) |
| , fUniforms(std::move(uniforms)) |
| , fChildren(children.begin(), children.end()) {} |
| |
| #if SK_SUPPORT_GPU |
| GrFPResult asFragmentProcessor(std::unique_ptr<GrFragmentProcessor> inputFP, |
| GrRecordingContext* context, |
| const GrColorInfo& colorInfo) const override { |
| sk_sp<SkData> uniforms = |
| get_xformed_uniforms(fEffect.get(), fUniforms, colorInfo.colorSpace()); |
| SkASSERT(uniforms); |
| |
| GrFPArgs childArgs(context, SkSimpleMatrixProvider(SkMatrix::I()), &colorInfo); |
| return make_effect_fp(fEffect, |
| "runtime_color_filter", |
| std::move(uniforms), |
| std::move(inputFP), |
| SkMakeSpan(fChildren), |
| childArgs); |
| } |
| #endif |
| |
| bool onAppendStages(const SkStageRec& rec, bool shaderIsOpaque) const override { |
| return false; |
| } |
| |
| skvm::Color onProgram(skvm::Builder* p, skvm::Color c, |
| const SkColorInfo& dst, |
| skvm::Uniforms* uniforms, SkArenaAlloc* alloc) const override { |
| sk_sp<SkData> inputs = get_xformed_uniforms(fEffect.get(), fUniforms, dst.colorSpace()); |
| SkASSERT(inputs); |
| |
| // There should be no way for the color filter to use device coords, but we need to supply |
| // something. (Uninitialized values can trigger asserts in skvm::Builder). |
| skvm::Coord zeroCoord = { p->splat(0.0f), p->splat(0.0f) }; |
| |
| auto sampleChild = [&](int ix, skvm::Coord coord, skvm::Color color) { |
| if (fChildren[ix].shader) { |
| SkSimpleMatrixProvider mats{SkMatrix::I()}; |
| return as_SB(fChildren[ix].shader) |
| ->program(p, coord, coord, color, mats, nullptr, dst, uniforms, alloc); |
| } else if (fChildren[ix].colorFilter) { |
| return as_CFB(fChildren[ix].colorFilter)->program(p, color, dst, uniforms, alloc); |
| } else { |
| return color; |
| } |
| }; |
| |
| const size_t uniformCount = fEffect->uniformSize() / 4; |
| std::vector<skvm::Val> uniform; |
| uniform.reserve(uniformCount); |
| for (size_t i = 0; i < uniformCount; i++) { |
| int bits; |
| memcpy(&bits, (const char*)inputs->data() + 4*i, 4); |
| uniform.push_back(p->uniform32(uniforms->push(bits)).id); |
| } |
| |
| return SkSL::ProgramToSkVM(*fEffect->fBaseProgram, fEffect->fMain, p, SkMakeSpan(uniform), |
| /*device=*/zeroCoord, /*local=*/zeroCoord, c, c, sampleChild); |
| } |
| |
| SkPMColor4f onFilterColor4f(const SkPMColor4f& color, SkColorSpace* dstCS) const override { |
| // Get the generic program for filtering a single color |
| const SkFilterColorProgram* program = fEffect->getFilterColorProgram(); |
| if (!program) { |
| // We were unable to build a cached (per-effect) program. Use the base-class fallback, |
| // which builds a program for the specific filter instance. |
| return SkColorFilterBase::onFilterColor4f(color, dstCS); |
| } |
| |
| // Get our specific uniform values |
| sk_sp<SkData> inputs = get_xformed_uniforms(fEffect.get(), fUniforms, dstCS); |
| SkASSERT(inputs); |
| |
| auto evalChild = [&](int index, SkPMColor4f inColor) { |
| const auto& child = fChildren[index]; |
| |
| // Guaranteed by initFilterColorInfo |
| SkASSERT(!child.shader); |
| return child.colorFilter ? as_CFB(child.colorFilter)->onFilterColor4f(inColor, dstCS) |
| : inColor; |
| }; |
| |
| return program->eval(color, inputs->data(), evalChild); |
| } |
| |
| bool onIsAlphaUnchanged() const override { |
| return fEffect->getFilterColorProgram() && |
| fEffect->getFilterColorProgram()->isAlphaUnchanged(); |
| } |
| |
| void flatten(SkWriteBuffer& buffer) const override { |
| buffer.writeString(fEffect->source().c_str()); |
| buffer.writeDataAsByteArray(fUniforms.get()); |
| buffer.write32(fChildren.size()); |
| for (const auto& child : fChildren) { |
| buffer.writeFlattenable(child.shader ? (const SkFlattenable*)child.shader.get() |
| : child.colorFilter.get()); |
| } |
| } |
| |
| SK_FLATTENABLE_HOOKS(SkRuntimeColorFilter) |
| |
| private: |
| sk_sp<SkRuntimeEffect> fEffect; |
| sk_sp<SkData> fUniforms; |
| std::vector<SkRuntimeEffect::ChildPtr> fChildren; |
| }; |
| |
| sk_sp<SkFlattenable> SkRuntimeColorFilter::CreateProc(SkReadBuffer& buffer) { |
| SkString sksl; |
| buffer.readString(&sksl); |
| sk_sp<SkData> uniforms = buffer.readByteArrayAsData(); |
| |
| auto effect = SkMakeCachedRuntimeEffect(SkRuntimeEffect::MakeForColorFilter, std::move(sksl)); |
| if (!buffer.validate(effect != nullptr)) { |
| return nullptr; |
| } |
| |
| size_t childCount = buffer.read32(); |
| if (!buffer.validate(childCount == effect->children().count())) { |
| return nullptr; |
| } |
| |
| SkSTArray<4, SkRuntimeEffect::ChildPtr> children(childCount); |
| for (const auto& child : effect->children()) { |
| if (child.type == SkRuntimeEffect::Child::Type::kShader) { |
| children.emplace_back(buffer.readShader()); |
| } else { |
| SkASSERT(child.type == SkRuntimeEffect::Child::Type::kColorFilter); |
| children.emplace_back(buffer.readColorFilter()); |
| } |
| } |
| if (!buffer.isValid()) { |
| return nullptr; |
| } |
| |
| return effect->makeColorFilter(std::move(uniforms), SkMakeSpan(children)); |
| } |
| |
| /////////////////////////////////////////////////////////////////////////////////////////////////// |
| |
| class SkRTShader : public SkShaderBase { |
| public: |
| SkRTShader(sk_sp<SkRuntimeEffect> effect, |
| sk_sp<SkData> uniforms, |
| const SkMatrix* localMatrix, |
| SkSpan<SkRuntimeEffect::ChildPtr> children, |
| bool isOpaque) |
| : SkShaderBase(localMatrix) |
| , fEffect(std::move(effect)) |
| , fIsOpaque(isOpaque) |
| , fUniforms(std::move(uniforms)) |
| , fChildren(children.begin(), children.end()) {} |
| |
| bool isOpaque() const override { return fIsOpaque; } |
| |
| #if SK_SUPPORT_GPU |
| std::unique_ptr<GrFragmentProcessor> asFragmentProcessor(const GrFPArgs& args) const override { |
| SkMatrix matrix; |
| if (!this->totalLocalMatrix(args.fPreLocalMatrix)->invert(&matrix)) { |
| return nullptr; |
| } |
| |
| sk_sp<SkData> uniforms = |
| get_xformed_uniforms(fEffect.get(), fUniforms, args.fDstColorInfo->colorSpace()); |
| SkASSERT(uniforms); |
| |
| // If we sample children with explicit colors, this may not be true. |
| // TODO: Determine this via analysis? |
| GrFPArgs childArgs = args; |
| childArgs.fInputColorIsOpaque = false; |
| |
| auto [success, fp] = make_effect_fp(fEffect, |
| "runtime_shader", |
| std::move(uniforms), |
| /*inputFP=*/nullptr, |
| SkMakeSpan(fChildren), |
| childArgs); |
| if (!success) { |
| return nullptr; |
| } |
| |
| // If the shader was created with isOpaque = true, we *force* that result here. |
| // CPU does the same thing (in SkShaderBase::program). |
| if (fIsOpaque) { |
| fp = GrFragmentProcessor::SwizzleOutput(std::move(fp), GrSwizzle::RGB1()); |
| } |
| fp = GrMatrixEffect::Make(matrix, std::move(fp)); |
| // Three cases of GrClampType to think about: |
| // kAuto - Normalized fixed-point. If fIsOpaque, then A is 1 (above), and the format's |
| // range ensures RGB must be no larger. If !fIsOpaque, we clamp here. |
| // kManual - Normalized floating point. Whether or not we set A above, the format's range |
| // means we need to clamp RGB. |
| // kNone - Unclamped floating point. No clamping is done, ever. |
| GrClampType clampType = GrColorTypeClampType(args.fDstColorInfo->colorType()); |
| if (clampType == GrClampType::kManual || (clampType == GrClampType::kAuto && !fIsOpaque)) { |
| return GrFragmentProcessor::ClampPremulOutput(std::move(fp)); |
| } else { |
| return std::move(fp); |
| } |
| } |
| #endif |
| |
| bool onAppendStages(const SkStageRec& rec) const override { |
| return false; |
| } |
| |
| skvm::Color onProgram(skvm::Builder* p, |
| skvm::Coord device, skvm::Coord local, skvm::Color paint, |
| const SkMatrixProvider& matrices, const SkMatrix* localM, |
| const SkColorInfo& dst, |
| skvm::Uniforms* uniforms, SkArenaAlloc* alloc) const override { |
| sk_sp<SkData> inputs = get_xformed_uniforms(fEffect.get(), fUniforms, dst.colorSpace()); |
| SkASSERT(inputs); |
| |
| SkMatrix inv; |
| if (!this->computeTotalInverse(matrices.localToDevice(), localM, &inv)) { |
| return {}; |
| } |
| local = SkShaderBase::ApplyMatrix(p,inv,local,uniforms); |
| |
| auto sampleChild = [&](int ix, skvm::Coord coord, skvm::Color color) { |
| if (fChildren[ix].shader) { |
| SkOverrideDeviceMatrixProvider mats{matrices, SkMatrix::I()}; |
| return as_SB(fChildren[ix].shader) |
| ->program(p, device, coord, color, mats, nullptr, dst, uniforms, alloc); |
| } else if (fChildren[ix].colorFilter) { |
| return as_CFB(fChildren[ix].colorFilter)->program(p, color, dst, uniforms, alloc); |
| } else { |
| return color; |
| } |
| }; |
| |
| const size_t uniformCount = fEffect->uniformSize() / 4; |
| std::vector<skvm::Val> uniform; |
| uniform.reserve(uniformCount); |
| for (size_t i = 0; i < uniformCount; i++) { |
| int bits; |
| memcpy(&bits, (const char*)inputs->data() + 4*i, 4); |
| uniform.push_back(p->uniform32(uniforms->push(bits)).id); |
| } |
| |
| return SkSL::ProgramToSkVM(*fEffect->fBaseProgram, fEffect->fMain, p, SkMakeSpan(uniform), |
| device, local, paint, paint, sampleChild); |
| } |
| |
| void flatten(SkWriteBuffer& buffer) const override { |
| uint32_t flags = 0; |
| if (fIsOpaque) { |
| flags |= kIsOpaque_Flag; |
| } |
| if (!this->getLocalMatrix().isIdentity()) { |
| flags |= kHasLocalMatrix_Flag; |
| } |
| |
| buffer.writeString(fEffect->source().c_str()); |
| buffer.writeDataAsByteArray(fUniforms.get()); |
| buffer.write32(flags); |
| if (flags & kHasLocalMatrix_Flag) { |
| buffer.writeMatrix(this->getLocalMatrix()); |
| } |
| buffer.write32(fChildren.size()); |
| for (const auto& child : fChildren) { |
| buffer.writeFlattenable(child.shader ? (const SkFlattenable*)child.shader.get() |
| : child.colorFilter.get()); |
| } |
| } |
| |
| SkRuntimeEffect* asRuntimeEffect() const override { return fEffect.get(); } |
| |
| SK_FLATTENABLE_HOOKS(SkRTShader) |
| |
| private: |
| enum Flags { |
| kIsOpaque_Flag = 1 << 0, |
| kHasLocalMatrix_Flag = 1 << 1, |
| }; |
| |
| sk_sp<SkRuntimeEffect> fEffect; |
| bool fIsOpaque; |
| |
| sk_sp<SkData> fUniforms; |
| std::vector<SkRuntimeEffect::ChildPtr> fChildren; |
| }; |
| |
| sk_sp<SkFlattenable> SkRTShader::CreateProc(SkReadBuffer& buffer) { |
| SkString sksl; |
| buffer.readString(&sksl); |
| sk_sp<SkData> uniforms = buffer.readByteArrayAsData(); |
| uint32_t flags = buffer.read32(); |
| |
| bool isOpaque = SkToBool(flags & kIsOpaque_Flag); |
| SkMatrix localM, *localMPtr = nullptr; |
| if (flags & kHasLocalMatrix_Flag) { |
| buffer.readMatrix(&localM); |
| localMPtr = &localM; |
| } |
| |
| auto effect = SkMakeCachedRuntimeEffect(SkRuntimeEffect::MakeForShader, std::move(sksl)); |
| if (!buffer.validate(effect != nullptr)) { |
| return nullptr; |
| } |
| |
| size_t childCount = buffer.read32(); |
| if (!buffer.validate(childCount == effect->children().count())) { |
| return nullptr; |
| } |
| |
| SkSTArray<4, SkRuntimeEffect::ChildPtr> children(childCount); |
| for (const auto& child : effect->children()) { |
| if (child.type == SkRuntimeEffect::Child::Type::kShader) { |
| children.emplace_back(buffer.readShader()); |
| } else { |
| SkASSERT(child.type == SkRuntimeEffect::Child::Type::kColorFilter); |
| children.emplace_back(buffer.readColorFilter()); |
| } |
| } |
| if (!buffer.isValid()) { |
| return nullptr; |
| } |
| |
| return effect->makeShader(std::move(uniforms), SkMakeSpan(children), localMPtr, isOpaque); |
| } |
| |
| /////////////////////////////////////////////////////////////////////////////////////////////////// |
| |
| class SkRuntimeBlender : public SkBlenderBase { |
| public: |
| SkRuntimeBlender(sk_sp<SkRuntimeEffect> effect, sk_sp<SkData> uniforms) |
| : fEffect(std::move(effect)) |
| , fUniforms(std::move(uniforms)) {} |
| |
| SkRuntimeEffect* asRuntimeEffect() const override { return fEffect.get(); } |
| |
| skvm::Color onProgram(skvm::Builder* p, skvm::Color src, skvm::Color dst, |
| const SkColorInfo& colorInfo, skvm::Uniforms* uniforms, |
| SkArenaAlloc* alloc) const override { |
| sk_sp<SkData> inputs = get_xformed_uniforms(fEffect.get(), fUniforms, |
| colorInfo.colorSpace()); |
| SkASSERT(inputs); |
| |
| const size_t uniformCount = fEffect->uniformSize() / 4; |
| std::vector<skvm::Val> uniform; |
| uniform.reserve(uniformCount); |
| for (size_t i = 0; i < uniformCount; i++) { |
| int bits; |
| memcpy(&bits, (const char*)inputs->data() + 4*i, 4); |
| uniform.push_back(p->uniform32(uniforms->push(bits)).id); |
| } |
| |
| // Emit the blend function as an SkVM program. |
| skvm::Coord zeroCoord = {p->splat(0.0f), p->splat(0.0f)}; |
| return SkSL::ProgramToSkVM(*fEffect->fBaseProgram, fEffect->fMain, p, SkMakeSpan(uniform), |
| /*device=*/zeroCoord, /*local=*/zeroCoord, |
| src, dst, /*sampleChild=*/nullptr); |
| } |
| |
| #if SK_SUPPORT_GPU |
| std::unique_ptr<GrFragmentProcessor> asFragmentProcessor( |
| std::unique_ptr<GrFragmentProcessor> srcFP, |
| std::unique_ptr<GrFragmentProcessor> dstFP, |
| const GrFPArgs& args) const override { |
| sk_sp<SkData> uniforms = get_xformed_uniforms(fEffect.get(), fUniforms, |
| args.fDstColorInfo->colorSpace()); |
| SkASSERT(uniforms); |
| |
| return GrSkSLFP::MakeWithData(fEffect, |
| "runtime_blender", |
| std::move(srcFP), |
| std::move(dstFP), |
| std::move(uniforms), |
| /*childFPs=*/{}); |
| } |
| #endif |
| |
| void flatten(SkWriteBuffer& buffer) const override { |
| buffer.writeString(fEffect->source().c_str()); |
| buffer.writeDataAsByteArray(fUniforms.get()); |
| } |
| |
| SK_FLATTENABLE_HOOKS(SkRuntimeBlender) |
| |
| private: |
| using INHERITED = SkBlenderBase; |
| |
| sk_sp<SkRuntimeEffect> fEffect; |
| sk_sp<SkData> fUniforms; |
| }; |
| |
| sk_sp<SkFlattenable> SkRuntimeBlender::CreateProc(SkReadBuffer& buffer) { |
| SkString sksl; |
| buffer.readString(&sksl); |
| sk_sp<SkData> uniforms = buffer.readByteArrayAsData(); |
| |
| auto effect = SkMakeCachedRuntimeEffect(SkRuntimeEffect::MakeForBlender, std::move(sksl)); |
| if (!buffer.validate(effect != nullptr)) { |
| return nullptr; |
| } |
| |
| return effect->makeBlender(std::move(uniforms)); |
| } |
| |
| /////////////////////////////////////////////////////////////////////////////////////////////////// |
| |
| sk_sp<SkShader> SkRuntimeEffect::makeShader(sk_sp<SkData> uniforms, |
| sk_sp<SkShader> childShaders[], |
| size_t childCount, |
| const SkMatrix* localMatrix, |
| bool isOpaque) const { |
| SkSTArray<4, ChildPtr> children(childCount); |
| for (size_t i = 0; i < childCount; ++i) { |
| children.emplace_back(childShaders[i]); |
| } |
| return this->makeShader(std::move(uniforms), SkMakeSpan(children), localMatrix, isOpaque); |
| } |
| |
| sk_sp<SkShader> SkRuntimeEffect::makeShader(sk_sp<SkData> uniforms, |
| SkSpan<ChildPtr> children, |
| const SkMatrix* localMatrix, |
| bool isOpaque) const { |
| if (!this->allowShader()) { |
| return nullptr; |
| } |
| if (!verify_child_effects(fChildren, children)) { |
| return nullptr; |
| } |
| if (!uniforms) { |
| uniforms = SkData::MakeEmpty(); |
| } |
| if (uniforms->size() != this->uniformSize()) { |
| return nullptr; |
| } |
| return sk_sp<SkShader>(new SkRTShader(sk_ref_sp(this), std::move(uniforms), localMatrix, |
| children, isOpaque)); |
| } |
| |
| sk_sp<SkImage> SkRuntimeEffect::makeImage(GrRecordingContext* recordingContext, |
| sk_sp<SkData> uniforms, |
| SkSpan<ChildPtr> children, |
| const SkMatrix* localMatrix, |
| SkImageInfo resultInfo, |
| bool mipmapped) const { |
| if (recordingContext) { |
| #if SK_SUPPORT_GPU |
| if (!recordingContext->priv().caps()->mipmapSupport()) { |
| mipmapped = false; |
| } |
| auto fillContext = GrSurfaceFillContext::Make(recordingContext, |
| resultInfo, |
| SkBackingFit::kExact, |
| /*sample count*/ 1, |
| GrMipmapped(mipmapped)); |
| if (!fillContext) { |
| return nullptr; |
| } |
| uniforms = get_xformed_uniforms(this, std::move(uniforms), resultInfo.colorSpace()); |
| SkASSERT(uniforms); |
| |
| SkSimpleMatrixProvider matrixProvider(SkMatrix::I()); |
| GrColorInfo colorInfo(resultInfo.colorInfo()); |
| GrFPArgs args(recordingContext, matrixProvider, &colorInfo); |
| SkSTArray<8, std::unique_ptr<GrFragmentProcessor>> childFPs; |
| for (size_t i = 0; i < children.size(); ++i) { |
| // TODO: add support for other types of child effects |
| if (!children[i].shader) { |
| return nullptr; |
| } |
| childFPs.push_back(as_SB(children[i].shader)->asFragmentProcessor(args)); |
| } |
| auto fp = GrSkSLFP::MakeWithData(sk_ref_sp(this), |
| "runtime_image", |
| /*inputFP=*/nullptr, |
| /*destColorFP=*/nullptr, |
| std::move(uniforms), |
| SkMakeSpan(childFPs)); |
| |
| if (localMatrix) { |
| SkMatrix invLM; |
| if (!localMatrix->invert(&invLM)) { |
| return nullptr; |
| } |
| fillContext->fillWithFP(invLM, std::move(fp)); |
| } else { |
| fillContext->fillWithFP(std::move(fp)); |
| } |
| return sk_sp<SkImage>(new SkImage_Gpu(sk_ref_sp(recordingContext), |
| kNeedNewImageUniqueID, |
| fillContext->readSurfaceView(), |
| resultInfo.colorInfo())); |
| #else |
| return nullptr; |
| #endif |
| } |
| if (resultInfo.alphaType() == kUnpremul_SkAlphaType) { |
| // We don't have a good way of supporting this right now. In this case the runtime effect |
| // will produce a unpremul value. The shader generated from it is assumed to produce |
| // premul and RGB get pinned to A. Moreover, after the blend in premul the new dst is |
| // unpremul'ed, producing a double unpremul result. |
| return nullptr; |
| } |
| auto surf = SkSurface::MakeRaster(resultInfo); |
| if (!surf) { |
| return nullptr; |
| } |
| SkCanvas* canvas = surf->getCanvas(); |
| SkTLazy<SkCanvas> tempCanvas; |
| auto shader = this->makeShader(std::move(uniforms), children, localMatrix, false); |
| if (!shader) { |
| return nullptr; |
| } |
| SkPaint paint; |
| paint.setShader(std::move(shader)); |
| paint.setBlendMode(SkBlendMode::kSrc); |
| canvas->drawPaint(paint); |
| // TODO: Specify snapshot should have mip levels if mipmapped is true. |
| return surf->makeImageSnapshot(); |
| } |
| |
| sk_sp<SkColorFilter> SkRuntimeEffect::makeColorFilter(sk_sp<SkData> uniforms, |
| sk_sp<SkColorFilter> childColorFilters[], |
| size_t childCount) const { |
| SkSTArray<4, ChildPtr> children(childCount); |
| for (size_t i = 0; i < childCount; ++i) { |
| children.emplace_back(childColorFilters[i]); |
| } |
| return this->makeColorFilter(std::move(uniforms), SkMakeSpan(children)); |
| } |
| |
| sk_sp<SkColorFilter> SkRuntimeEffect::makeColorFilter(sk_sp<SkData> uniforms, |
| SkSpan<ChildPtr> children) const { |
| if (!this->allowColorFilter()) { |
| return nullptr; |
| } |
| if (!verify_child_effects(fChildren, children)) { |
| return nullptr; |
| } |
| if (!uniforms) { |
| uniforms = SkData::MakeEmpty(); |
| } |
| if (uniforms->size() != this->uniformSize()) { |
| return nullptr; |
| } |
| return sk_sp<SkColorFilter>(new SkRuntimeColorFilter(sk_ref_sp(this), std::move(uniforms), |
| children)); |
| } |
| |
| sk_sp<SkColorFilter> SkRuntimeEffect::makeColorFilter(sk_sp<SkData> uniforms) const { |
| return this->makeColorFilter(std::move(uniforms), /*children=*/{}); |
| } |
| |
| sk_sp<SkBlender> SkRuntimeEffect::makeBlender(sk_sp<SkData> uniforms) const { |
| if (!this->allowBlender()) { |
| return nullptr; |
| } |
| if (!uniforms) { |
| uniforms = SkData::MakeEmpty(); |
| } |
| if (uniforms->size() != this->uniformSize() || !fChildren.empty()) { |
| return nullptr; |
| } |
| return sk_sp<SkBlender>(new SkRuntimeBlender(sk_ref_sp(this), std::move(uniforms))); |
| } |
| |
| /////////////////////////////////////////////////////////////////////////////////////////////////// |
| |
| void SkRuntimeEffect::RegisterFlattenables() { |
| SK_REGISTER_FLATTENABLE(SkRuntimeColorFilter); |
| SK_REGISTER_FLATTENABLE(SkRTShader); |
| SK_REGISTER_FLATTENABLE(SkRuntimeBlender); |
| } |
| |
| SkRuntimeShaderBuilder::SkRuntimeShaderBuilder(sk_sp<SkRuntimeEffect> effect) |
| : INHERITED(std::move(effect)) {} |
| |
| SkRuntimeShaderBuilder::~SkRuntimeShaderBuilder() = default; |
| |
| sk_sp<SkImage> SkRuntimeShaderBuilder::makeImage(GrRecordingContext* recordingContext, |
| const SkMatrix* localMatrix, |
| SkImageInfo resultInfo, |
| bool mipmapped) { |
| return this->effect()->makeImage(recordingContext, |
| this->uniforms(), |
| SkMakeSpan(this->children(), this->numChildren()), |
| localMatrix, |
| resultInfo, |
| mipmapped); |
| } |
| |
| sk_sp<SkShader> SkRuntimeShaderBuilder::makeShader(const SkMatrix* localMatrix, bool isOpaque) { |
| return this->effect()->makeShader(this->uniforms(), |
| SkMakeSpan(this->children(), this->numChildren()), |
| localMatrix, |
| isOpaque); |
| } |
| |
| SkRuntimeBlendBuilder::SkRuntimeBlendBuilder(sk_sp<SkRuntimeEffect> effect) |
| : INHERITED(std::move(effect)) {} |
| |
| SkRuntimeBlendBuilder::~SkRuntimeBlendBuilder() = default; |
| |
| sk_sp<SkBlender> SkRuntimeBlendBuilder::makeBlender() { |
| return this->effect()->makeBlender(this->uniforms()); |
| } |
| |
| #endif // SK_ENABLE_SKSL |