xref: /aosp_15_r20/external/skia/modules/skottie/src/effects/CCTonerEffect.cpp (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1 /*
2  * Copyright 2021 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 #include "include/core/SkColor.h"
9 #include "include/core/SkRefCnt.h"
10 #include "include/core/SkScalar.h"
11 #include "modules/skottie/src/Adapter.h"
12 #include "modules/skottie/src/SkottiePriv.h"
13 #include "modules/skottie/src/SkottieValue.h"
14 #include "modules/skottie/src/effects/Effects.h"
15 #include "modules/sksg/include/SkSGColorFilter.h"
16 #include "modules/sksg/include/SkSGPaint.h"
17 #include "modules/sksg/include/SkSGRenderNode.h"
18 #include "src/base/SkVx.h"
19 #include "src/core/SkSwizzlePriv.h"
20 
21 #include <cstddef>
22 #include <utility>
23 #include <vector>
24 
25 namespace skjson {
26 class ArrayValue;
27 }
28 
29 namespace skottie::internal {
30 
31 namespace {
32 
33 class CCTonerAdapter final : public DiscardableAdapterBase<CCTonerAdapter,
34                                                            sksg::GradientColorFilter> {
35     public:
CCTonerAdapter(const skjson::ArrayValue & jprops,sk_sp<sksg::RenderNode> layer,const AnimationBuilder & abuilder,std::vector<sk_sp<sksg::Color>> colorNodes)36         CCTonerAdapter(const skjson::ArrayValue& jprops,
37                        sk_sp<sksg::RenderNode> layer,
38                        const AnimationBuilder& abuilder,
39                        std::vector<sk_sp<sksg::Color>> colorNodes)
40             : INHERITED(sksg::GradientColorFilter::Make(std::move(layer), colorNodes))
41             , fColorNodes(std::move(colorNodes))
42         {
43             enum : size_t {
44                 kTone_Index        = 0,
45                 kHiColor_Index     = 1,
46                 kBrightColor_Index = 2,
47                 kMidColor_Index    = 3,
48                 kDarkColor_Index   = 4,
49                 kShadowColor_Index = 5,
50                 kBlendAmount_Index = 6,
51             };
52 
53 
54             EffectBinder(jprops, abuilder, this)
55                 .bind(       kTone_Index, fTone)
56                 .bind(    kHiColor_Index, fHighlights)
57                 .bind(kBrightColor_Index, fBrights)
58                 .bind(   kMidColor_Index, fMidtones)
59                 .bind(  kDarkColor_Index, fDarktones)
60                 .bind(kShadowColor_Index, fShadows);
61 
62         }
63     private:
lerpColor(SkColor c0,SkColor c1,float t)64         static SkColor lerpColor(SkColor c0, SkColor c1, float t) {
65             const auto c0_4f = Sk4f_fromL32(c0),
66                        c1_4f = Sk4f_fromL32(c1),
67                        c_4f = c0_4f + (c1_4f - c0_4f) * t;
68 
69             return Sk4f_toL32(c_4f);
70         }
71 
onSync()72         void onSync() override {
73             switch (SkScalarRoundToInt(fTone)) {
74                 // duotone
75                 case 1: fColorNodes.at(0)->setColor(fShadows);
76                         fColorNodes.at(1)->setColor(lerpColor(fShadows, fHighlights, 0.25));
77                         fColorNodes.at(2)->setColor(lerpColor(fShadows, fHighlights, 0.5));
78                         fColorNodes.at(3)->setColor(lerpColor(fShadows, fHighlights, 0.75));
79                         fColorNodes.at(4)->setColor(fHighlights);
80                         break;
81                 // tritone
82                 case 2: fColorNodes.at(0)->setColor(fShadows);
83                         fColorNodes.at(1)->setColor(lerpColor(fShadows, fMidtones, 0.5));
84                         fColorNodes.at(2)->setColor(fMidtones);
85                         fColorNodes.at(3)->setColor(lerpColor(fMidtones, fHighlights, 0.5));
86                         fColorNodes.at(4)->setColor(fHighlights);
87                         break;
88                 // pentone
89                 case 3: fColorNodes.at(0)->setColor(fShadows);
90                         fColorNodes.at(1)->setColor(fDarktones);
91                         fColorNodes.at(2)->setColor(fMidtones);
92                         fColorNodes.at(3)->setColor(fBrights);
93                         fColorNodes.at(4)->setColor(fHighlights);
94                         break;
95                 // solid
96                 default: fColorNodes.at(0)->setColor(fMidtones);
97                         fColorNodes.at(1)->setColor(fMidtones);
98                         fColorNodes.at(2)->setColor(fMidtones);
99                         fColorNodes.at(3)->setColor(fMidtones);
100                         fColorNodes.at(4)->setColor(fMidtones);
101                         break;
102             }
103 
104             this->node()->setWeight((100-fBlend)/100);
105         }
106 
107 
108         const std::vector<sk_sp<sksg::Color>> fColorNodes;
109 
110         ScalarValue fTone = 0;
111         ColorValue  fHighlights,
112                     fBrights,
113                     fMidtones,
114                     fDarktones,
115                     fShadows;
116         ScalarValue fBlend = 0;
117 
118         using INHERITED = DiscardableAdapterBase<CCTonerAdapter, sksg::GradientColorFilter>;
119 };
120 
121 } // namespace
122 
attachCCTonerEffect(const skjson::ArrayValue & jprops,sk_sp<sksg::RenderNode> layer) const123 sk_sp<sksg::RenderNode> EffectBuilder::attachCCTonerEffect(const skjson::ArrayValue& jprops,
124                                                              sk_sp<sksg::RenderNode> layer) const {
125     std::vector<sk_sp<sksg::Color>> colorNodes = {
126         sksg::Color::Make(SK_ColorRED),
127         sksg::Color::Make(SK_ColorRED),
128         sksg::Color::Make(SK_ColorRED),
129         sksg::Color::Make(SK_ColorRED),
130         sksg::Color::Make(SK_ColorRED)
131     };
132     return fBuilder->attachDiscardableAdapter<CCTonerAdapter>(jprops,
133                                                               std::move(layer),
134                                                               *fBuilder,
135                                                               std::move(colorNodes));
136 }
137 
138 } // namespace skottie::internal
139