1 /* 2 * Copyright 2023 Google LLC 3 * 4 * Use of this source code is governed by a BSD-style license that can be 5 * found in the LICENSE file. 6 */ 7 #ifndef SkWorkingColorSpaceShader_DEFINED 8 #define SkWorkingColorSpaceShader_DEFINED 9 10 #include "include/core/SkColorSpace.h" 11 #include "include/core/SkFlattenable.h" 12 #include "include/core/SkRefCnt.h" 13 #include "include/core/SkShader.h" 14 #include "include/private/base/SkAssert.h" 15 #include "src/shaders/SkShaderBase.h" 16 17 #include <utility> 18 19 class SkReadBuffer; 20 class SkWriteBuffer; 21 struct SkStageRec; 22 23 class SkWorkingColorSpaceShader final : public SkShaderBase { 24 public: Make(sk_sp<SkShader> shader,sk_sp<SkColorSpace> workingSpace)25 static sk_sp<SkShader> Make(sk_sp<SkShader> shader, sk_sp<SkColorSpace> workingSpace) { 26 if (!shader) { 27 return nullptr; 28 } else if (!workingSpace) { 29 return shader; 30 } else { 31 return sk_sp(new SkWorkingColorSpaceShader(std::move(shader), std::move(workingSpace))); 32 } 33 } 34 type()35 ShaderType type() const override { return ShaderType::kWorkingColorSpace; } 36 shader()37 sk_sp<SkShader> shader() const { return fShader; } workingSpace()38 sk_sp<SkColorSpace> workingSpace() const { return fWorkingSpace; } 39 40 private: SkWorkingColorSpaceShader(sk_sp<SkShader> shader,sk_sp<SkColorSpace> workingSpace)41 SkWorkingColorSpaceShader(sk_sp<SkShader> shader, sk_sp<SkColorSpace> workingSpace) 42 : fShader(std::move(shader)), fWorkingSpace(std::move(workingSpace)) { 43 SkASSERT(fShader); 44 SkASSERT(fWorkingSpace); 45 } 46 47 bool appendStages(const SkStageRec& rec, const SkShaders::MatrixRec&) const override; 48 49 friend void ::SkRegisterWorkingColorSpaceShaderFlattenable(); 50 SK_FLATTENABLE_HOOKS(SkWorkingColorSpaceShader) 51 52 void flatten(SkWriteBuffer& buffer) const override; 53 54 sk_sp<SkShader> fShader; 55 sk_sp<SkColorSpace> fWorkingSpace; 56 }; 57 58 #endif 59