xref: /aosp_15_r20/external/skia/gm/lumafilter.cpp (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1 /*
2  * Copyright 2013 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 "gm/gm.h"
9 #include "include/core/SkBlendMode.h"
10 #include "include/core/SkCanvas.h"
11 #include "include/core/SkColor.h"
12 #include "include/core/SkColorFilter.h"
13 #include "include/core/SkColorSpace.h"
14 #include "include/core/SkFont.h"
15 #include "include/core/SkFontTypes.h"
16 #include "include/core/SkPaint.h"
17 #include "include/core/SkPoint.h"
18 #include "include/core/SkRect.h"
19 #include "include/core/SkRefCnt.h"
20 #include "include/core/SkScalar.h"
21 #include "include/core/SkShader.h"
22 #include "include/core/SkSize.h"
23 #include "include/core/SkString.h"
24 #include "include/core/SkTileMode.h"
25 #include "include/core/SkTypeface.h"
26 #include "include/core/SkTypes.h"
27 #include "include/effects/SkGradientShader.h"
28 #include "include/effects/SkLumaColorFilter.h"
29 #include "include/effects/SkRuntimeEffect.h"
30 #include "src/core/SkColorFilterPriv.h"
31 #include "tools/DecodeUtils.h"
32 #include "tools/Resources.h"
33 #include "tools/ToolUtils.h"
34 #include "tools/fonts/FontToolUtils.h"
35 
36 #include <string.h>
37 
38 static SkScalar kSize   = 80;
39 static SkScalar kInset  = 10;
40 static SkColor  kColor1 = SkColorSetARGB(0xff, 0xff, 0xff, 0);
41 static SkColor  kColor2 = SkColorSetARGB(0xff, 0x82, 0xff, 0);
42 
draw_label(SkCanvas * canvas,const char * label,const SkPoint & offset)43 static void draw_label(SkCanvas* canvas, const char* label,
44                        const SkPoint& offset) {
45     SkFont font = ToolUtils::DefaultPortableFont();
46     font.setEdging(SkFont::Edging::kAlias);
47 
48     size_t len = strlen(label);
49 
50     SkScalar width = font.measureText(label, len, SkTextEncoding::kUTF8);
51     canvas->drawSimpleText(label, len, SkTextEncoding::kUTF8, offset.x() - width / 2, offset.y(),
52                            font, SkPaint());
53 }
54 
draw_scene(SkCanvas * canvas,const sk_sp<SkColorFilter> & filter,SkBlendMode mode,const sk_sp<SkShader> & s1,const sk_sp<SkShader> & s2)55 static void draw_scene(SkCanvas* canvas, const sk_sp<SkColorFilter>& filter, SkBlendMode mode,
56                        const sk_sp<SkShader>& s1, const sk_sp<SkShader>& s2) {
57     SkPaint paint;
58     paint.setAntiAlias(true);
59     SkRect r, c, bounds = SkRect::MakeWH(kSize, kSize);
60 
61     c = bounds;
62     c.fRight = bounds.centerX();
63     paint.setARGB(0x20, 0, 0, 0xff);
64     canvas->drawRect(bounds, paint);
65 
66     canvas->saveLayer(&bounds, nullptr);
67 
68     r = bounds;
69     r.inset(kInset, 0);
70     paint.setShader(s1);
71     paint.setColor(s1 ? SK_ColorBLACK : SkColorSetA(kColor1, 0x80));
72     canvas->drawOval(r, paint);
73     if (!s1) {
74         canvas->save();
75         canvas->clipRect(c);
76         paint.setColor(kColor1);
77         canvas->drawOval(r, paint);
78         canvas->restore();
79     }
80 
81     SkPaint xferPaint;
82     xferPaint.setBlendMode(mode);
83     canvas->saveLayer(&bounds, &xferPaint);
84 
85     r = bounds;
86     r.inset(0, kInset);
87     paint.setShader(s2);
88     paint.setColor(s2 ? SK_ColorBLACK : SkColorSetA(kColor2, 0x80));
89     paint.setColorFilter(filter);
90     canvas->drawOval(r, paint);
91     if (!s2) {
92         canvas->save();
93         canvas->clipRect(c);
94         paint.setColor(kColor2);
95         canvas->drawOval(r, paint);
96         canvas->restore();
97     }
98 
99     canvas->restore();
100     canvas->restore();
101 }
102 
103 class LumaFilterGM : public skiagm::GM {
104 protected:
onOnceBeforeDraw()105     void onOnceBeforeDraw() override {
106         SkColor  g1Colors[] = { kColor1, SkColorSetA(kColor1, 0x20) };
107         SkColor  g2Colors[] = { kColor2, SkColorSetA(kColor2, 0x20) };
108         SkPoint  g1Points[] = { { 0, 0 }, { 0,     100 } };
109         SkPoint  g2Points[] = { { 0, 0 }, { kSize, 0   } };
110         SkScalar pos[] = { 0.2f, 1.0f };
111 
112         fFilter = SkLumaColorFilter::Make();
113         fGr1 = SkGradientShader::MakeLinear(g1Points, g1Colors, pos, std::size(g1Colors),
114                                             SkTileMode::kClamp);
115         fGr2 = SkGradientShader::MakeLinear(g2Points, g2Colors, pos, std::size(g2Colors),
116                                             SkTileMode::kClamp);
117     }
118 
getName() const119     SkString getName() const override { return SkString("lumafilter"); }
120 
getISize()121     SkISize getISize() override { return SkISize::Make(600, 420); }
122 
onDraw(SkCanvas * canvas)123     void onDraw(SkCanvas* canvas) override {
124         SkBlendMode modes[] = {
125             SkBlendMode::kSrcOver,
126             SkBlendMode::kDstOver,
127             SkBlendMode::kSrcATop,
128             SkBlendMode::kDstATop,
129             SkBlendMode::kSrcIn,
130             SkBlendMode::kDstIn,
131         };
132         struct {
133             const sk_sp<SkShader>& fShader1;
134             const sk_sp<SkShader>& fShader2;
135         } shaders[] = {
136             { nullptr, nullptr },
137             { nullptr, fGr2 },
138             { fGr1, nullptr },
139             { fGr1, fGr2 },
140         };
141 
142         SkScalar gridStep = kSize + 2 * kInset;
143         for (size_t i = 0; i < std::size(modes); ++i) {
144             draw_label(canvas, SkBlendMode_Name(modes[i]),
145                        SkPoint::Make(gridStep * (0.5f + i), 20));
146         }
147 
148         for (size_t i = 0; i < std::size(shaders); ++i) {
149             canvas->save();
150             canvas->translate(kInset, gridStep * i + 30);
151             for (size_t m = 0; m < std::size(modes); ++m) {
152                 draw_scene(canvas, fFilter, modes[m], shaders[i].fShader1,
153                            shaders[i].fShader2);
154                 canvas->translate(gridStep, 0);
155             }
156             canvas->restore();
157         }
158     }
159 
160 private:
161     sk_sp<SkColorFilter>    fFilter;
162     sk_sp<SkShader>         fGr1, fGr2;
163 
164     using INHERITED = skiagm::GM;
165 };
166 
167 //////////////////////////////////////////////////////////////////////////////
168 
169 DEF_GM(return new LumaFilterGM;)
170 
171 DEF_SIMPLE_GM(AlternateLuma, canvas, 384,128) {
172     sk_sp<SkImage> img = ToolUtils::GetResourceAsImage("images/mandrill_128.png");
173     if (!img) {
174         return;
175     }
176 
177     // Normal luma colorfilter on the left.
178     SkPaint paint;
179     paint.setColorFilter(SkLumaColorFilter::Make());
180     canvas->drawImage(img, 0,0, SkFilterMode::kNearest, &paint);
181     canvas->translate(128,0);
182 
183     // Original image in the middle for reference.
184     canvas->drawImage(img, 0,0);
185     canvas->translate(128,0);
186 
187     // Here, RGB holds CIE XYZ. Splatting the G (Y) channel should result in (near) greyscale.
188     auto [effect, err] = SkRuntimeEffect::MakeForColorFilter(SkString{
189             "half4 main(half4 inColor) { return inColor.ggga; }"});
190     SkASSERT(effect && err.isEmpty());
191 
192     sk_sp<SkColorFilter> filter = effect->makeColorFilter(SkData::MakeEmpty());
193     SkASSERT(filter);
194 
195     SkAlphaType unpremul = kUnpremul_SkAlphaType;
196     paint.setColorFilter(SkColorFilterPriv::WithWorkingFormat(std::move(filter),
197                                                               &SkNamedTransferFn::kLinear,
198                                                               &SkNamedGamut::kXYZ,
199                                                               &unpremul));
200     canvas->drawImage(img, 0,0, SkFilterMode::kNearest, &paint);
201 }
202