epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2006 The Android Open Source Project |
| 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 | |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 8 | #include "include/core/SkColorFilter.h" |
| 9 | #include "include/core/SkString.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 10 | #include "include/private/SkColorData.h" |
Ben Wagner | 729a23f | 2019-05-17 16:29:34 -0400 | [diff] [blame] | 11 | #include "src/core/SkArenaAlloc.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 12 | #include "src/core/SkBlendModePriv.h" |
| 13 | #include "src/core/SkRasterPipeline.h" |
| 14 | #include "src/core/SkReadBuffer.h" |
Mike Reed | 190b6e5 | 2020-03-14 16:42:56 -0400 | [diff] [blame] | 15 | #include "src/core/SkVM.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 16 | #include "src/core/SkWriteBuffer.h" |
| 17 | #include "src/shaders/SkColorShader.h" |
| 18 | #include "src/shaders/SkComposeShader.h" |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 19 | |
Florin Malita | 9ab449a | 2019-07-09 13:22:50 -0400 | [diff] [blame] | 20 | namespace { |
| 21 | |
Florin Malita | 9ab449a | 2019-07-09 13:22:50 -0400 | [diff] [blame] | 22 | struct LocalMatrixStageRec final : public SkStageRec { |
| 23 | LocalMatrixStageRec(const SkStageRec& rec, const SkMatrix& lm) |
| 24 | : INHERITED(rec) { |
| 25 | if (!lm.isIdentity()) { |
| 26 | if (fLocalM) { |
| 27 | fStorage.setConcat(lm, *fLocalM); |
| 28 | fLocalM = fStorage.isIdentity() ? nullptr : &fStorage; |
| 29 | } else { |
| 30 | fLocalM = &lm; |
| 31 | } |
| 32 | } |
| 33 | } |
| 34 | |
| 35 | private: |
| 36 | SkMatrix fStorage; |
| 37 | |
| 38 | using INHERITED = SkStageRec; |
| 39 | }; |
| 40 | |
| 41 | } // namespace |
| 42 | |
Kevin Lubick | 988ce04 | 2020-03-25 13:13:20 -0400 | [diff] [blame] | 43 | sk_sp<SkShader> SkShaders::Blend(SkBlendMode mode, sk_sp<SkShader> dst, sk_sp<SkShader> src) { |
Florin Malita | 9ab449a | 2019-07-09 13:22:50 -0400 | [diff] [blame] | 44 | switch (mode) { |
| 45 | case SkBlendMode::kClear: return Color(0); |
Kevin Lubick | 988ce04 | 2020-03-25 13:13:20 -0400 | [diff] [blame] | 46 | case SkBlendMode::kDst: return dst; |
| 47 | case SkBlendMode::kSrc: return src; |
Florin Malita | 9ab449a | 2019-07-09 13:22:50 -0400 | [diff] [blame] | 48 | default: break; |
| 49 | } |
Kevin Lubick | 988ce04 | 2020-03-25 13:13:20 -0400 | [diff] [blame] | 50 | return sk_sp<SkShader>(new SkShader_Blend(mode, std::move(dst), std::move(src))); |
Florin Malita | 9ab449a | 2019-07-09 13:22:50 -0400 | [diff] [blame] | 51 | } |
| 52 | |
Kevin Lubick | 988ce04 | 2020-03-25 13:13:20 -0400 | [diff] [blame] | 53 | sk_sp<SkShader> SkShaders::Lerp(float weight, sk_sp<SkShader> dst, sk_sp<SkShader> src) { |
Mike Reed | 9b2c4e9 | 2019-04-02 09:54:14 -0400 | [diff] [blame] | 54 | if (SkScalarIsNaN(weight)) { |
Mike Reed | 7d954ad | 2016-10-28 15:42:34 -0400 | [diff] [blame] | 55 | return nullptr; |
| 56 | } |
Florin Malita | 9ab449a | 2019-07-09 13:22:50 -0400 | [diff] [blame] | 57 | if (dst == src || weight <= 0) { |
Kevin Lubick | 988ce04 | 2020-03-25 13:13:20 -0400 | [diff] [blame] | 58 | return dst; |
Mike Reed | 7d954ad | 2016-10-28 15:42:34 -0400 | [diff] [blame] | 59 | } |
Florin Malita | 9ab449a | 2019-07-09 13:22:50 -0400 | [diff] [blame] | 60 | if (weight >= 1) { |
Kevin Lubick | 988ce04 | 2020-03-25 13:13:20 -0400 | [diff] [blame] | 61 | return src; |
Mike Reed | 9b2c4e9 | 2019-04-02 09:54:14 -0400 | [diff] [blame] | 62 | } |
Kevin Lubick | 988ce04 | 2020-03-25 13:13:20 -0400 | [diff] [blame] | 63 | return sk_sp<SkShader>(new SkShader_Lerp(weight, std::move(dst), std::move(src))); |
Mike Reed | 7d954ad | 2016-10-28 15:42:34 -0400 | [diff] [blame] | 64 | } |
| 65 | |
reed@google.com | 82065d6 | 2011-02-07 15:30:46 +0000 | [diff] [blame] | 66 | /////////////////////////////////////////////////////////////////////////////// |
| 67 | |
Mike Reed | 9b2c4e9 | 2019-04-02 09:54:14 -0400 | [diff] [blame] | 68 | static bool append_shader_or_paint(const SkStageRec& rec, SkShader* shader) { |
| 69 | if (shader) { |
| 70 | if (!as_SB(shader)->appendStages(rec)) { |
| 71 | return false; |
| 72 | } |
| 73 | } else { |
| 74 | rec.fPipeline->append_constant_color(rec.fAlloc, rec.fPaint.getColor4f().premul().vec()); |
| 75 | } |
| 76 | return true; |
| 77 | } |
| 78 | |
| 79 | // Returns the output of e0, and leaves the output of e1 in r,g,b,a |
| 80 | static float* append_two_shaders(const SkStageRec& rec, SkShader* s0, SkShader* s1) { |
| 81 | struct Storage { |
| 82 | float fRes0[4 * SkRasterPipeline_kMaxStride]; |
| 83 | }; |
| 84 | auto storage = rec.fAlloc->make<Storage>(); |
| 85 | |
| 86 | if (!append_shader_or_paint(rec, s0)) { |
| 87 | return nullptr; |
| 88 | } |
| 89 | rec.fPipeline->append(SkRasterPipeline::store_src, storage->fRes0); |
| 90 | |
| 91 | if (!append_shader_or_paint(rec, s1)) { |
| 92 | return nullptr; |
| 93 | } |
| 94 | return storage->fRes0; |
| 95 | } |
| 96 | |
| 97 | /////////////////////////////////////////////////////////////////////////////// |
| 98 | |
| 99 | sk_sp<SkFlattenable> SkShader_Blend::CreateProc(SkReadBuffer& buffer) { |
Mike Reed | 01b2b83 | 2017-06-09 10:51:52 -0400 | [diff] [blame] | 100 | sk_sp<SkShader> dst(buffer.readShader()); |
| 101 | sk_sp<SkShader> src(buffer.readShader()); |
| 102 | unsigned mode = buffer.read32(); |
Mike Reed | 01b2b83 | 2017-06-09 10:51:52 -0400 | [diff] [blame] | 103 | |
| 104 | // check for valid mode before we cast to the enum type |
Mike Reed | c5166a9 | 2018-01-03 13:17:18 -0500 | [diff] [blame] | 105 | if (!buffer.validate(mode <= (unsigned)SkBlendMode::kLastMode)) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 106 | return nullptr; |
reed | 9fa60da | 2014-08-21 07:59:51 -0700 | [diff] [blame] | 107 | } |
Mike Reed | c8bea7d | 2019-04-09 13:55:36 -0400 | [diff] [blame] | 108 | return SkShaders::Blend(static_cast<SkBlendMode>(mode), std::move(dst), std::move(src)); |
reed | 9fa60da | 2014-08-21 07:59:51 -0700 | [diff] [blame] | 109 | } |
| 110 | |
Mike Reed | 9b2c4e9 | 2019-04-02 09:54:14 -0400 | [diff] [blame] | 111 | void SkShader_Blend::flatten(SkWriteBuffer& buffer) const { |
Mike Reed | 01b2b83 | 2017-06-09 10:51:52 -0400 | [diff] [blame] | 112 | buffer.writeFlattenable(fDst.get()); |
| 113 | buffer.writeFlattenable(fSrc.get()); |
Mike Reed | 7d954ad | 2016-10-28 15:42:34 -0400 | [diff] [blame] | 114 | buffer.write32((int)fMode); |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 115 | } |
| 116 | |
Florin Malita | 9ab449a | 2019-07-09 13:22:50 -0400 | [diff] [blame] | 117 | bool SkShader_Blend::onAppendStages(const SkStageRec& orig_rec) const { |
| 118 | const LocalMatrixStageRec rec(orig_rec, this->getLocalMatrix()); |
| 119 | |
Mike Reed | 9b2c4e9 | 2019-04-02 09:54:14 -0400 | [diff] [blame] | 120 | float* res0 = append_two_shaders(rec, fDst.get(), fSrc.get()); |
| 121 | if (!res0) { |
Mike Reed | 9959f72 | 2017-05-15 09:34:22 -0400 | [diff] [blame] | 122 | return false; |
| 123 | } |
Mike Reed | 5e398c2 | 2019-03-08 11:50:35 -0500 | [diff] [blame] | 124 | |
Mike Reed | 9b2c4e9 | 2019-04-02 09:54:14 -0400 | [diff] [blame] | 125 | rec.fPipeline->append(SkRasterPipeline::load_dst, res0); |
| 126 | SkBlendMode_AppendStages(fMode, rec.fPipeline); |
| 127 | return true; |
| 128 | } |
| 129 | |
Mike Klein | a68d933 | 2020-06-30 15:17:28 -0500 | [diff] [blame] | 130 | static skvm::Color program_or_paint(const sk_sp<SkShader>& sh, skvm::Builder* p, |
| 131 | skvm::Coord device, skvm::Coord local, skvm::Color paint, |
| 132 | const SkMatrixProvider& mats, const SkMatrix* localM, |
Mike Reed | 8046837 | 2021-03-19 14:05:19 -0400 | [diff] [blame] | 133 | const SkColorInfo& dst, |
Mike Klein | a68d933 | 2020-06-30 15:17:28 -0500 | [diff] [blame] | 134 | skvm::Uniforms* uniforms, SkArenaAlloc* alloc) { |
Mike Reed | 8046837 | 2021-03-19 14:05:19 -0400 | [diff] [blame] | 135 | return sh ? as_SB(sh)->program(p, device,local, paint, mats,localM, dst, uniforms,alloc) |
Mike Klein | a68d933 | 2020-06-30 15:17:28 -0500 | [diff] [blame] | 136 | : p->premul(paint); |
| 137 | } |
| 138 | |
Brian Osman | 5aaaeea | 2020-06-22 14:26:03 -0400 | [diff] [blame] | 139 | skvm::Color SkShader_Blend::onProgram(skvm::Builder* p, |
| 140 | skvm::Coord device, skvm::Coord local, skvm::Color paint, |
| 141 | const SkMatrixProvider& mats, const SkMatrix* localM, |
Mike Reed | 8046837 | 2021-03-19 14:05:19 -0400 | [diff] [blame] | 142 | const SkColorInfo& dst, |
Mike Klein | 276a785 | 2020-03-15 08:46:09 -0500 | [diff] [blame] | 143 | skvm::Uniforms* uniforms, SkArenaAlloc* alloc) const { |
Mike Klein | a434e0f | 2020-03-23 09:33:48 -0500 | [diff] [blame] | 144 | skvm::Color d,s; |
Mike Reed | 8046837 | 2021-03-19 14:05:19 -0400 | [diff] [blame] | 145 | if ((d = program_or_paint(fDst, p, device,local, paint, mats,localM, dst, uniforms,alloc)) && |
| 146 | (s = program_or_paint(fSrc, p, device,local, paint, mats,localM, dst, uniforms,alloc))) |
Mike Reed | 6352f00 | 2020-03-14 23:30:10 -0400 | [diff] [blame] | 147 | { |
Mike Klein | a434e0f | 2020-03-23 09:33:48 -0500 | [diff] [blame] | 148 | return p->blend(fMode, s,d); |
Mike Reed | 190b6e5 | 2020-03-14 16:42:56 -0400 | [diff] [blame] | 149 | } |
Mike Reed | 6352f00 | 2020-03-14 23:30:10 -0400 | [diff] [blame] | 150 | return {}; |
Mike Reed | 190b6e5 | 2020-03-14 16:42:56 -0400 | [diff] [blame] | 151 | } |
| 152 | |
| 153 | |
Mike Reed | 9b2c4e9 | 2019-04-02 09:54:14 -0400 | [diff] [blame] | 154 | sk_sp<SkFlattenable> SkShader_Lerp::CreateProc(SkReadBuffer& buffer) { |
| 155 | sk_sp<SkShader> dst(buffer.readShader()); |
| 156 | sk_sp<SkShader> src(buffer.readShader()); |
| 157 | float t = buffer.readScalar(); |
Mike Reed | c8bea7d | 2019-04-09 13:55:36 -0400 | [diff] [blame] | 158 | return buffer.isValid() ? SkShaders::Lerp(t, std::move(dst), std::move(src)) : nullptr; |
Mike Reed | 9b2c4e9 | 2019-04-02 09:54:14 -0400 | [diff] [blame] | 159 | } |
| 160 | |
| 161 | void SkShader_Lerp::flatten(SkWriteBuffer& buffer) const { |
| 162 | buffer.writeFlattenable(fDst.get()); |
| 163 | buffer.writeFlattenable(fSrc.get()); |
| 164 | buffer.writeScalar(fWeight); |
| 165 | } |
| 166 | |
Florin Malita | 9ab449a | 2019-07-09 13:22:50 -0400 | [diff] [blame] | 167 | bool SkShader_Lerp::onAppendStages(const SkStageRec& orig_rec) const { |
| 168 | const LocalMatrixStageRec rec(orig_rec, this->getLocalMatrix()); |
| 169 | |
Mike Reed | 9b2c4e9 | 2019-04-02 09:54:14 -0400 | [diff] [blame] | 170 | float* res0 = append_two_shaders(rec, fDst.get(), fSrc.get()); |
| 171 | if (!res0) { |
Mike Reed | 5e398c2 | 2019-03-08 11:50:35 -0500 | [diff] [blame] | 172 | return false; |
| 173 | } |
Mike Reed | 9959f72 | 2017-05-15 09:34:22 -0400 | [diff] [blame] | 174 | |
Mike Reed | 9b2c4e9 | 2019-04-02 09:54:14 -0400 | [diff] [blame] | 175 | rec.fPipeline->append(SkRasterPipeline::load_dst, res0); |
| 176 | rec.fPipeline->append(SkRasterPipeline::lerp_1_float, &fWeight); |
Mike Reed | 9959f72 | 2017-05-15 09:34:22 -0400 | [diff] [blame] | 177 | return true; |
| 178 | } |
commit-bot@chromium.org | 7959055 | 2014-05-13 18:14:45 +0000 | [diff] [blame] | 179 | |
Brian Osman | 5aaaeea | 2020-06-22 14:26:03 -0400 | [diff] [blame] | 180 | skvm::Color SkShader_Lerp::onProgram(skvm::Builder* p, |
| 181 | skvm::Coord device, skvm::Coord local, skvm::Color paint, |
| 182 | const SkMatrixProvider& mats, const SkMatrix* localM, |
Mike Reed | 8046837 | 2021-03-19 14:05:19 -0400 | [diff] [blame] | 183 | const SkColorInfo& dst, |
Mike Klein | 276a785 | 2020-03-15 08:46:09 -0500 | [diff] [blame] | 184 | skvm::Uniforms* uniforms, SkArenaAlloc* alloc) const { |
Mike Klein | a434e0f | 2020-03-23 09:33:48 -0500 | [diff] [blame] | 185 | skvm::Color d,s; |
Mike Reed | 8046837 | 2021-03-19 14:05:19 -0400 | [diff] [blame] | 186 | if ((d = program_or_paint(fDst, p, device,local, paint, mats,localM, dst, uniforms,alloc)) && |
| 187 | (s = program_or_paint(fSrc, p, device,local, paint, mats,localM, dst, uniforms,alloc))) |
Mike Reed | 6352f00 | 2020-03-14 23:30:10 -0400 | [diff] [blame] | 188 | { |
| 189 | auto t = p->uniformF(uniforms->pushF(fWeight)); |
| 190 | return { |
Mike Klein | a434e0f | 2020-03-23 09:33:48 -0500 | [diff] [blame] | 191 | p->lerp(d.r, s.r, t), |
| 192 | p->lerp(d.g, s.g, t), |
| 193 | p->lerp(d.b, s.b, t), |
| 194 | p->lerp(d.a, s.a, t), |
Mike Reed | 6352f00 | 2020-03-14 23:30:10 -0400 | [diff] [blame] | 195 | }; |
Mike Reed | 190b6e5 | 2020-03-14 16:42:56 -0400 | [diff] [blame] | 196 | } |
Mike Reed | 6352f00 | 2020-03-14 23:30:10 -0400 | [diff] [blame] | 197 | return {}; |
Mike Reed | 190b6e5 | 2020-03-14 16:42:56 -0400 | [diff] [blame] | 198 | } |
| 199 | |
wangyix | 73fa616 | 2015-09-01 09:45:08 -0700 | [diff] [blame] | 200 | #if SK_SUPPORT_GPU |
| 201 | |
Robert Phillips | b7bfbc2 | 2020-07-01 12:55:01 -0400 | [diff] [blame] | 202 | #include "include/gpu/GrRecordingContext.h" |
John Stiles | f743d4e | 2020-07-23 11:35:08 -0400 | [diff] [blame] | 203 | #include "src/gpu/effects/GrBlendFragmentProcessor.h" |
Brian Salomon | 5ca24f9 | 2021-03-23 17:04:41 -0400 | [diff] [blame] | 204 | #include "src/gpu/effects/GrSkSLFP.h" |
Mike Reed | c8bea7d | 2019-04-09 13:55:36 -0400 | [diff] [blame] | 205 | |
| 206 | static std::unique_ptr<GrFragmentProcessor> as_fp(const GrFPArgs& args, SkShader* shader) { |
Ethan Nicholas | ff4fb33 | 2019-04-09 16:28:19 -0400 | [diff] [blame] | 207 | return shader ? as_SB(shader)->asFragmentProcessor(args) : nullptr; |
Mike Reed | c8bea7d | 2019-04-09 13:55:36 -0400 | [diff] [blame] | 208 | } |
| 209 | |
Florin Malita | 9ab449a | 2019-07-09 13:22:50 -0400 | [diff] [blame] | 210 | std::unique_ptr<GrFragmentProcessor> SkShader_Blend::asFragmentProcessor( |
| 211 | const GrFPArgs& orig_args) const { |
| 212 | const GrFPArgs::WithPreLocalMatrix args(orig_args, this->getLocalMatrix()); |
Mike Reed | c8bea7d | 2019-04-09 13:55:36 -0400 | [diff] [blame] | 213 | auto fpA = as_fp(args, fDst.get()); |
| 214 | auto fpB = as_fp(args, fSrc.get()); |
John Stiles | f743d4e | 2020-07-23 11:35:08 -0400 | [diff] [blame] | 215 | return GrBlendFragmentProcessor::Make(std::move(fpB), std::move(fpA), fMode); |
wangyix | 73fa616 | 2015-09-01 09:45:08 -0700 | [diff] [blame] | 216 | } |
Mike Reed | 9b2c4e9 | 2019-04-02 09:54:14 -0400 | [diff] [blame] | 217 | |
Florin Malita | 9ab449a | 2019-07-09 13:22:50 -0400 | [diff] [blame] | 218 | std::unique_ptr<GrFragmentProcessor> SkShader_Lerp::asFragmentProcessor( |
| 219 | const GrFPArgs& orig_args) const { |
| 220 | const GrFPArgs::WithPreLocalMatrix args(orig_args, this->getLocalMatrix()); |
Mike Reed | c8bea7d | 2019-04-09 13:55:36 -0400 | [diff] [blame] | 221 | auto fpA = as_fp(args, fDst.get()); |
| 222 | auto fpB = as_fp(args, fSrc.get()); |
Brian Salomon | 5ca24f9 | 2021-03-23 17:04:41 -0400 | [diff] [blame] | 223 | |
| 224 | static constexpr char kCode[] = R"( |
| 225 | uniform shader a; |
| 226 | uniform shader b; |
| 227 | uniform half w; |
| 228 | |
| 229 | half4 main() { return mix(sample(a), sample(b), w); } |
| 230 | )"; |
| 231 | |
| 232 | auto builder = GrRuntimeFPBuilder::Make<kCode>(); |
| 233 | builder.uniform("w") = fWeight; |
| 234 | builder.child("a") = std::move(fpA); |
| 235 | builder.child("b") = std::move(fpB); |
Brian Osman | 8ed8081 | 2021-04-13 15:09:14 -0400 | [diff] [blame] | 236 | return builder.makeFP(); |
Mike Reed | c8bea7d | 2019-04-09 13:55:36 -0400 | [diff] [blame] | 237 | } |
wangyix | 73fa616 | 2015-09-01 09:45:08 -0700 | [diff] [blame] | 238 | #endif |