xref: /aosp_15_r20/external/skia/gm/displacement.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/SkBitmap.h"
10 #include "include/core/SkCanvas.h"
11 #include "include/core/SkColor.h"
12 #include "include/core/SkImage.h"
13 #include "include/core/SkImageFilter.h"
14 #include "include/core/SkPaint.h"
15 #include "include/core/SkRect.h"
16 #include "include/core/SkRefCnt.h"
17 #include "include/core/SkScalar.h"
18 #include "include/core/SkSize.h"
19 #include "include/core/SkString.h"
20 #include "include/effects/SkImageFilters.h"
21 #include "tools/ToolUtils.h"
22 #include "tools/fonts/FontToolUtils.h"
23 
24 #include <utility>
25 
26 namespace skiagm {
27 
28 class DisplacementMapGM : public GM {
29 public:
DisplacementMapGM()30     DisplacementMapGM() {
31         this->setBGColor(0xFF000000);
32     }
33 
34 protected:
getName() const35     SkString getName() const override { return SkString("displacement"); }
36 
onOnceBeforeDraw()37     void onOnceBeforeDraw() override {
38         fImage = ToolUtils::CreateStringImage(80, 80, 0xFF884422, 15, 55, 96, "g");
39 
40         SkColor c1 = ToolUtils::color_to_565(0xFF244484);
41         SkColor c2 = ToolUtils::color_to_565(0xFF804020);
42 
43         fCheckerboard = ToolUtils::create_checkerboard_image(80, 80, c1, c2, 8);
44         fSmall  = ToolUtils::create_checkerboard_image(64, 64, c1, c2, 8);
45         fLarge  = ToolUtils::create_checkerboard_image(96, 96, c1, c2, 8);
46         fLargeW = ToolUtils::create_checkerboard_image(96, 64, c1, c2, 8);
47         fLargeH = ToolUtils::create_checkerboard_image(64, 96, c1, c2, 8);
48     }
49 
getISize()50     SkISize getISize() override { return SkISize::Make(600, 500); }
51 
drawClippedBitmap(SkCanvas * canvas,int x,int y,const SkPaint & paint) const52     void drawClippedBitmap(SkCanvas* canvas, int x, int y, const SkPaint& paint) const {
53         canvas->save();
54         canvas->translate(SkIntToScalar(x), SkIntToScalar(y));
55         canvas->clipIRect(fImage->bounds());
56         canvas->drawImage(fImage, 0, 0, SkSamplingOptions(), &paint);
57         canvas->restore();
58     }
59 
onDraw(SkCanvas * canvas)60     void onDraw(SkCanvas* canvas) override {
61         canvas->clear(SK_ColorBLACK);
62         SkPaint paint;
63         sk_sp<SkImageFilter> displ(SkImageFilters::Image(fCheckerboard, SkFilterMode::kLinear));
64         paint.setImageFilter(SkImageFilters::DisplacementMap(
65                 SkColorChannel::kR, SkColorChannel::kG, 0.0f, displ, nullptr));
66         this->drawClippedBitmap(canvas, 0, 0, paint);
67         paint.setImageFilter(SkImageFilters::DisplacementMap(
68                 SkColorChannel::kB, SkColorChannel::kA, 16.0f, displ, nullptr));
69         this->drawClippedBitmap(canvas, 100, 0, paint);
70         paint.setImageFilter(SkImageFilters::DisplacementMap(
71                 SkColorChannel::kR, SkColorChannel::kB, 32.0f, displ, nullptr));
72         this->drawClippedBitmap(canvas, 200, 0, paint);
73         paint.setImageFilter(SkImageFilters::DisplacementMap(
74                 SkColorChannel::kG, SkColorChannel::kA, 48.0f, displ, nullptr));
75         this->drawClippedBitmap(canvas, 300, 0, paint);
76         paint.setImageFilter(SkImageFilters::DisplacementMap(
77                 SkColorChannel::kR, SkColorChannel::kA, 64.0f, displ, nullptr));
78         this->drawClippedBitmap(canvas, 400, 0, paint);
79 
80         paint.setImageFilter(SkImageFilters::DisplacementMap(
81                 SkColorChannel::kR, SkColorChannel::kG, 40.0f, displ, nullptr));
82         this->drawClippedBitmap(canvas, 0, 100, paint);
83         paint.setImageFilter(SkImageFilters::DisplacementMap(
84                 SkColorChannel::kB, SkColorChannel::kA, 40.0f, displ, nullptr));
85         this->drawClippedBitmap(canvas, 100, 100, paint);
86         paint.setImageFilter(SkImageFilters::DisplacementMap(
87                 SkColorChannel::kR, SkColorChannel::kB, 40.0f, displ, nullptr));
88         this->drawClippedBitmap(canvas, 200, 100, paint);
89         paint.setImageFilter(SkImageFilters::DisplacementMap(
90                 SkColorChannel::kG, SkColorChannel::kA, 40.0f, displ, nullptr));
91         this->drawClippedBitmap(canvas, 300, 100, paint);
92         paint.setImageFilter(SkImageFilters::DisplacementMap(
93                 SkColorChannel::kR, SkColorChannel::kA, 40.0f, displ, nullptr));
94         this->drawClippedBitmap(canvas, 400, 100, paint);
95 
96         SkIRect cropRect = SkIRect::MakeXYWH(30, 30, 40, 40);
97         paint.setImageFilter(SkImageFilters::DisplacementMap(
98                 SkColorChannel::kR, SkColorChannel::kG, 0.0f, displ, nullptr, &cropRect));
99         this->drawClippedBitmap(canvas, 0, 200, paint);
100         paint.setImageFilter(SkImageFilters::DisplacementMap(
101             SkColorChannel::kB, SkColorChannel::kA, 16.0f, displ, nullptr, &cropRect));
102         this->drawClippedBitmap(canvas, 100, 200, paint);
103         paint.setImageFilter(SkImageFilters::DisplacementMap(
104                 SkColorChannel::kR, SkColorChannel::kB, 32.0f, displ, nullptr, &cropRect));
105         this->drawClippedBitmap(canvas, 200, 200, paint);
106         paint.setImageFilter(SkImageFilters::DisplacementMap(
107                 SkColorChannel::kG, SkColorChannel::kA, 48.0f, displ, nullptr, &cropRect));
108         this->drawClippedBitmap(canvas, 300, 200, paint);
109         paint.setImageFilter(SkImageFilters::DisplacementMap(
110                 SkColorChannel::kR, SkColorChannel::kA, 64.0f, displ, nullptr, &cropRect));
111         this->drawClippedBitmap(canvas, 400, 200, paint);
112 
113         paint.setImageFilter(SkImageFilters::DisplacementMap(
114                 SkColorChannel::kR, SkColorChannel::kG, 40.0f, displ, nullptr, &cropRect));
115         this->drawClippedBitmap(canvas, 0, 300, paint);
116         paint.setImageFilter(SkImageFilters::DisplacementMap(
117                 SkColorChannel::kB, SkColorChannel::kA, 40.0f, displ, nullptr, &cropRect));
118         this->drawClippedBitmap(canvas, 100, 300, paint);
119         paint.setImageFilter(SkImageFilters::DisplacementMap(
120                 SkColorChannel::kR, SkColorChannel::kB, 40.0f, displ, nullptr, &cropRect));
121         this->drawClippedBitmap(canvas, 200, 300, paint);
122         paint.setImageFilter(SkImageFilters::DisplacementMap(
123                 SkColorChannel::kG, SkColorChannel::kA, 40.0f, displ, nullptr, &cropRect));
124         this->drawClippedBitmap(canvas, 300, 300, paint);
125         paint.setImageFilter(SkImageFilters::DisplacementMap(
126                 SkColorChannel::kR, SkColorChannel::kA, 40.0f, displ, nullptr, &cropRect));
127         this->drawClippedBitmap(canvas, 400, 300, paint);
128 
129         // Test for negative scale.
130         paint.setImageFilter(SkImageFilters::DisplacementMap(
131                 SkColorChannel::kG, SkColorChannel::kA, -40.0f, displ, nullptr));
132         this->drawClippedBitmap(canvas, 500, 0, paint);
133 
134         // Tests for images of different sizes
135         displ = SkImageFilters::Image(fSmall, SkFilterMode::kLinear);
136         paint.setImageFilter(SkImageFilters::DisplacementMap(
137                 SkColorChannel::kR, SkColorChannel::kG, 40.0f, std::move(displ), nullptr));
138         this->drawClippedBitmap(canvas, 0, 400, paint);
139         displ = SkImageFilters::Image(fLarge, SkFilterMode::kLinear);
140         paint.setImageFilter(SkImageFilters::DisplacementMap(
141                 SkColorChannel::kB, SkColorChannel::kA, 40.0f, std::move(displ), nullptr));
142         this->drawClippedBitmap(canvas, 100, 400, paint);
143         displ = SkImageFilters::Image(fLargeW, SkFilterMode::kLinear);
144         paint.setImageFilter(SkImageFilters::DisplacementMap(
145                 SkColorChannel::kR, SkColorChannel::kB, 40.0f, std::move(displ), nullptr));
146         this->drawClippedBitmap(canvas, 200, 400, paint);
147         displ = SkImageFilters::Image(fLargeH, SkFilterMode::kLinear);
148         paint.setImageFilter(SkImageFilters::DisplacementMap(
149                 SkColorChannel::kG, SkColorChannel::kA, 40.0f, std::move(displ), nullptr));
150         this->drawClippedBitmap(canvas, 300, 400, paint);
151 
152         // Test for no given displacement input. In this case, both displacement
153         // and color should use the same bitmap, given to SkCanvas::drawBitmap()
154         // as an input argument.
155         paint.setImageFilter(SkImageFilters::DisplacementMap(
156                 SkColorChannel::kG, SkColorChannel::kA, 40.0f, nullptr, nullptr));
157         this->drawClippedBitmap(canvas, 400, 400, paint);
158     }
159 
160 private:
161     sk_sp<SkImage> fImage;
162     sk_sp<SkImage> fCheckerboard, fSmall, fLarge, fLargeW, fLargeH;
163 
164     using INHERITED = GM;
165 };
166 
167 //////////////////////////////////////////////////////////////////////////////
168 
169 DEF_GM(return new DisplacementMapGM;)
170 }  // namespace skiagm
171