Mike Reed | 3fd3cc9 | 2019-06-20 12:40:30 -0400 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2019 Google Inc. |
| 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 | |
| 8 | #ifndef SkRTShader_DEFINED |
| 9 | #define SkRTShader_DEFINED |
| 10 | |
| 11 | #include "include/core/SkString.h" |
| 12 | #include "include/private/SkMutex.h" |
| 13 | #include "src/shaders/SkShaderBase.h" |
| 14 | #include "src/sksl/SkSLByteCode.h" |
| 15 | |
| 16 | #if SK_SUPPORT_GPU |
| 17 | #include "src/gpu/GrFragmentProcessor.h" |
| 18 | #endif |
| 19 | |
| 20 | class SkData; |
| 21 | class SkMatrix; |
| 22 | |
| 23 | class SkRTShader : public SkShaderBase { |
| 24 | public: |
Brian Osman | 7e16b58 | 2019-08-20 16:56:40 -0400 | [diff] [blame] | 25 | SkRTShader(int index, SkString sksl, sk_sp<SkData> inputs, const SkMatrix* localMatrix, |
| 26 | bool isOpaque); |
Mike Reed | 3fd3cc9 | 2019-06-20 12:40:30 -0400 | [diff] [blame] | 27 | |
Mike Reed | 3fd3cc9 | 2019-06-20 12:40:30 -0400 | [diff] [blame] | 28 | bool isOpaque() const override { return fIsOpaque; } |
| 29 | |
Mike Reed | 8c31f2b | 2019-07-16 16:50:14 -0400 | [diff] [blame] | 30 | #if SK_SUPPORT_GPU |
| 31 | std::unique_ptr<GrFragmentProcessor> asFragmentProcessor(const GrFPArgs&) const override; |
| 32 | #endif |
| 33 | |
Mike Reed | 3fd3cc9 | 2019-06-20 12:40:30 -0400 | [diff] [blame] | 34 | protected: |
| 35 | void flatten(SkWriteBuffer&) const override; |
| 36 | bool onAppendStages(const SkStageRec& rec) const override; |
| 37 | |
| 38 | private: |
| 39 | SK_FLATTENABLE_HOOKS(SkRTShader) |
| 40 | |
| 41 | SkString fSkSL; |
| 42 | sk_sp<SkData> fInputs; |
Mike Reed | 8c31f2b | 2019-07-16 16:50:14 -0400 | [diff] [blame] | 43 | const uint32_t fUniqueID; |
Mike Reed | 3fd3cc9 | 2019-06-20 12:40:30 -0400 | [diff] [blame] | 44 | const bool fIsOpaque; |
| 45 | |
| 46 | mutable SkMutex fByteCodeMutex; |
| 47 | mutable std::unique_ptr<SkSL::ByteCode> fByteCode; |
| 48 | |
| 49 | typedef SkShaderBase INHERITED; |
| 50 | }; |
| 51 | |
Stan Iliev | cdf1bc3 | 2019-12-02 11:35:11 -0500 | [diff] [blame] | 52 | class SK_API SkRuntimeShaderFactory { |
Brian Osman | 7e16b58 | 2019-08-20 16:56:40 -0400 | [diff] [blame] | 53 | public: |
| 54 | SkRuntimeShaderFactory(SkString sksl, bool isOpaque); |
| 55 | |
| 56 | sk_sp<SkShader> make(sk_sp<SkData> inputs, const SkMatrix* localMatrix); |
| 57 | |
| 58 | private: |
| 59 | int fIndex; |
| 60 | SkString fSkSL; |
| 61 | bool fIsOpaque; |
| 62 | }; |
| 63 | |
Mike Reed | 3fd3cc9 | 2019-06-20 12:40:30 -0400 | [diff] [blame] | 64 | #endif |