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