1 /*
2 * Copyright 2018 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/SkImage.h"
15 #include "include/core/SkImageInfo.h"
16 #include "include/core/SkMatrix.h"
17 #include "include/core/SkPaint.h"
18 #include "include/core/SkPoint.h"
19 #include "include/core/SkRect.h"
20 #include "include/core/SkRefCnt.h"
21 #include "include/core/SkScalar.h"
22 #include "include/core/SkShader.h"
23 #include "include/core/SkSize.h"
24 #include "include/core/SkString.h"
25 #include "include/core/SkSurface.h"
26 #include "include/core/SkTileMode.h"
27 #include "include/core/SkTypes.h"
28 #include "include/effects/SkGradientShader.h"
29 #include "include/gpu/ganesh/GrDirectContext.h"
30 #include "tools/GpuToolUtils.h"
31 #include "tools/ToolUtils.h"
32
33 #include <algorithm>
34 #include <initializer_list>
35 #include <utility>
36
37 // Makes a set of m x n tiled images to be drawn with SkCanvas::experimental_drawImageSetV1().
make_image_tiles(int tileW,int tileH,int m,int n,const SkColor colors[4],SkCanvas::ImageSetEntry set[],const SkColor bgColor=SK_ColorLTGRAY)38 static void make_image_tiles(int tileW, int tileH, int m, int n, const SkColor colors[4],
39 SkCanvas::ImageSetEntry set[], const SkColor bgColor=SK_ColorLTGRAY) {
40 const int w = tileW * m;
41 const int h = tileH * n;
42 auto surf = SkSurfaces::Raster(
43 SkImageInfo::Make(w, h, kRGBA_8888_SkColorType, kPremul_SkAlphaType));
44 surf->getCanvas()->clear(bgColor);
45
46 static constexpr SkScalar kStripeW = 10;
47 static constexpr SkScalar kStripeSpacing = 30;
48 SkPaint paint;
49
50 SkPoint pts1[] = {{0.f, 0.f}, {(SkScalar)w, (SkScalar)h}};
51 auto grad = SkGradientShader::MakeLinear(pts1, colors, nullptr, 2, SkTileMode::kClamp);
52 paint.setShader(std::move(grad));
53 paint.setAntiAlias(true);
54 paint.setStyle(SkPaint::kStroke_Style);
55 paint.setStrokeWidth(kStripeW);
56 SkPoint stripePts[] = {{-w - kStripeW, -kStripeW}, {kStripeW, h + kStripeW}};
57 while (stripePts[0].fX <= w) {
58 surf->getCanvas()->drawPoints(SkCanvas::kLines_PointMode, 2, stripePts, paint);
59 stripePts[0].fX += kStripeSpacing;
60 stripePts[1].fX += kStripeSpacing;
61 }
62
63 SkPoint pts2[] = {{0.f, (SkScalar)h}, {(SkScalar)w, 0.f}};
64 grad = SkGradientShader::MakeLinear(pts2, colors + 2, nullptr, 2, SkTileMode::kClamp);
65 paint.setShader(std::move(grad));
66 paint.setBlendMode(SkBlendMode::kMultiply);
67 stripePts[0] = {-w - kStripeW, h + kStripeW};
68 stripePts[1] = {kStripeW, -kStripeW};
69 while (stripePts[0].fX <= w) {
70 surf->getCanvas()->drawPoints(SkCanvas::kLines_PointMode, 2, stripePts, paint);
71 stripePts[0].fX += kStripeSpacing;
72 stripePts[1].fX += kStripeSpacing;
73 }
74 auto fullImage = surf->makeImageSnapshot();
75 for (int y = 0; y < n; ++y) {
76 for (int x = 0; x < m; ++x) {
77 // Images will have 1 pixel of overlap at interior seams for filtering continuity.
78 SkIRect subset = SkIRect::MakeXYWH(x * tileW - 1, y * tileH - 1, tileW + 2, tileH + 2);
79 set[y * m + x].fAAFlags = SkCanvas::kNone_QuadAAFlags;
80 if (x == 0) {
81 subset.fLeft = 0;
82 set[y * m + x].fAAFlags |= SkCanvas::kLeft_QuadAAFlag;
83 }
84 if (x == m - 1) {
85 subset.fRight = w;
86 set[y * m + x].fAAFlags |= SkCanvas::kRight_QuadAAFlag;
87 }
88 if (y == 0) {
89 subset.fTop = 0;
90 set[y * m + x].fAAFlags |= SkCanvas::kTop_QuadAAFlag;
91 }
92 if (y == n - 1) {
93 subset.fBottom = h;
94 set[y * m + x].fAAFlags |= SkCanvas::kBottom_QuadAAFlag;
95 }
96 set[y * m + x].fImage = fullImage->makeSubset(nullptr, subset);
97 set[y * m + x].fSrcRect =
98 SkRect::MakeXYWH(x == 0 ? 0 : 1, y == 0 ? 0 : 1, tileW, tileH);
99 set[y * m + x].fDstRect = SkRect::MakeXYWH(x * tileW, y * tileH, tileW, tileH);
100 set[y * m + x].fAlpha = 1.f;
101 SkASSERT(set[y * m + x].fImage);
102 }
103 }
104 }
105
106 namespace skiagm {
107
108 class DrawImageSetGM : public GM {
109 private:
getName() const110 SkString getName() const override { return SkString("draw_image_set"); }
getISize()111 SkISize getISize() override { return {1000, 725}; }
onOnceBeforeDraw()112 void onOnceBeforeDraw() override {
113 static constexpr SkColor kColors[] = {SK_ColorCYAN, SK_ColorBLACK,
114 SK_ColorMAGENTA, SK_ColorBLACK};
115 make_image_tiles(kTileW, kTileH, kM, kN, kColors, fSet);
116 }
117
onDraw(SkCanvas * canvas)118 void onDraw(SkCanvas* canvas) override {
119 SkScalar d = SkVector{kM * kTileW, kN * kTileH}.length();
120 SkMatrix matrices[4];
121 // rotation
122 matrices[0].setRotate(30);
123 matrices[0].postTranslate(d / 3, 0);
124 // perespective
125 SkPoint src[4];
126 SkRect::MakeWH(kM * kTileW, kN * kTileH).toQuad(src);
127 SkPoint dst[4] = {{0, 0},
128 {kM * kTileW + 10.f, -5.f},
129 {kM * kTileW - 28.f, kN * kTileH + 40.f},
130 {45.f, kN * kTileH - 25.f}};
131 SkAssertResult(matrices[1].setPolyToPoly(src, dst, 4));
132 matrices[1].postTranslate(d, 50.f);
133 // skew
134 matrices[2].setRotate(-60.f);
135 matrices[2].postSkew(0.5f, -1.15f);
136 matrices[2].postScale(0.6f, 1.05f);
137 matrices[2].postTranslate(d, 2.6f * d);
138 // perspective + mirror in x.
139 dst[1] = {-.25 * kM * kTileW, 0};
140 dst[0] = {5.f / 4.f * kM * kTileW, 0};
141 dst[3] = {2.f / 3.f * kM * kTileW, 1 / 2.f * kN * kTileH};
142 dst[2] = {1.f / 3.f * kM * kTileW, 1 / 2.f * kN * kTileH - 0.1f * kTileH};
143 SkAssertResult(matrices[3].setPolyToPoly(src, dst, 4));
144 matrices[3].postTranslate(100.f, d);
145 for (auto fm : {SkFilterMode::kNearest, SkFilterMode::kLinear}) {
146 SkPaint setPaint;
147 setPaint.setBlendMode(SkBlendMode::kSrcOver);
148 SkSamplingOptions sampling(fm);
149
150 for (size_t m = 0; m < std::size(matrices); ++m) {
151 // Draw grid of red lines at interior tile boundaries.
152 static constexpr SkScalar kLineOutset = 10.f;
153 SkPaint paint;
154 paint.setAntiAlias(true);
155 paint.setColor(SK_ColorRED);
156 paint.setStyle(SkPaint::kStroke_Style);
157 paint.setStrokeWidth(0.f);
158 for (int x = 1; x < kM; ++x) {
159 SkPoint pts[] = {{x * kTileW, 0}, {x * kTileW, kN * kTileH}};
160 matrices[m].mapPoints(pts, 2);
161 SkVector v = pts[1] - pts[0];
162 v.setLength(v.length() + kLineOutset);
163 canvas->drawLine(pts[1] - v, pts[0] + v, paint);
164 }
165 for (int y = 1; y < kN; ++y) {
166 SkPoint pts[] = {{0, y * kTileH}, {kTileW * kM, y * kTileH}};
167 matrices[m].mapPoints(pts, 2);
168 SkVector v = pts[1] - pts[0];
169 v.setLength(v.length() + kLineOutset);
170 canvas->drawLine(pts[1] - v, pts[0] + v, paint);
171 }
172 canvas->save();
173 canvas->concat(matrices[m]);
174 canvas->experimental_DrawEdgeAAImageSet(fSet, kM * kN, nullptr, nullptr, sampling,
175 &setPaint,
176 SkCanvas::kFast_SrcRectConstraint);
177 canvas->restore();
178 }
179 // A more exotic case with an unusual blend mode, mixed aa flags set, and alpha,
180 // subsets the image. And another with all the above plus a color filter.
181 SkCanvas::ImageSetEntry entry;
182 entry.fSrcRect = SkRect::MakeWH(kTileW, kTileH).makeInset(kTileW / 4.f, kTileH / 4.f);
183 entry.fDstRect = SkRect::MakeWH(1.5 * kTileW, 1.5 * kTileH).makeOffset(d / 4, 2 * d);
184 entry.fImage = fSet[0].fImage;
185 entry.fAlpha = 0.7f;
186 entry.fAAFlags = SkCanvas::kLeft_QuadAAFlag | SkCanvas::kTop_QuadAAFlag;
187 canvas->save();
188 canvas->rotate(3.f);
189
190 setPaint.setBlendMode(SkBlendMode::kExclusion);
191 canvas->experimental_DrawEdgeAAImageSet(&entry, 1, nullptr, nullptr, sampling,
192 &setPaint, SkCanvas::kFast_SrcRectConstraint);
193 canvas->translate(entry.fDstRect.width() + 8.f, 0);
194 SkPaint cfPaint = setPaint;
195 cfPaint.setColorFilter(SkColorFilters::LinearToSRGBGamma());
196 canvas->experimental_DrawEdgeAAImageSet(&entry, 1, nullptr, nullptr, sampling,
197 &cfPaint, SkCanvas::kFast_SrcRectConstraint);
198 canvas->restore();
199 canvas->translate(2 * d, 0);
200 }
201 }
202 inline static constexpr int kM = 4;
203 inline static constexpr int kN = 3;
204 inline static constexpr SkScalar kTileW = 30;
205 inline static constexpr SkScalar kTileH = 60;
206 SkCanvas::ImageSetEntry fSet[kM * kN];
207 };
208
209 // This GM exercises rect-stays-rect type matrices to test that filtering and antialiasing are not
210 // incorrectly disabled.
211 class DrawImageSetRectToRectGM : public GM {
212 private:
getName() const213 SkString getName() const override { return SkString("draw_image_set_rect_to_rect"); }
getISize()214 SkISize getISize() override { return {1250, 850}; }
onOnceBeforeDraw()215 void onOnceBeforeDraw() override {
216 static constexpr SkColor kColors[] = {SK_ColorBLUE, SK_ColorWHITE,
217 SK_ColorRED, SK_ColorWHITE};
218 make_image_tiles(kTileW, kTileH, kM, kN, kColors, fSet);
219 }
220
onDraw(SkCanvas * canvas)221 void onDraw(SkCanvas* canvas) override {
222 ToolUtils::draw_checkerboard(canvas, SK_ColorBLACK, SK_ColorWHITE, 50);
223 static constexpr SkScalar kW = kM * kTileW;
224 static constexpr SkScalar kH = kN * kTileH;
225 SkMatrix matrices[5];
226 // Identity
227 matrices[0].reset();
228 // 90 degree rotation
229 matrices[1].setRotate(90, kW / 2.f, kH / 2.f);
230 // Scaling
231 matrices[2].setScale(2.f, 0.5f);
232 // Mirror in x and y
233 matrices[3].setScale(-1.f, -1.f);
234 matrices[3].postTranslate(kW, kH);
235 // Mirror in y, rotate, and scale.
236 matrices[4].setScale(1.f, -1.f);
237 matrices[4].postTranslate(0, kH);
238 matrices[4].postRotate(90, kW / 2.f, kH / 2.f);
239 matrices[4].postScale(2.f, 0.5f);
240
241 SkPaint paint;
242 paint.setBlendMode(SkBlendMode::kSrcOver);
243
244 static constexpr SkScalar kTranslate = std::max(kW, kH) * 2.f + 10.f;
245 canvas->translate(5.f, 5.f);
246 canvas->save();
247 for (SkScalar frac : {0.f, 0.5f}) {
248 canvas->save();
249 canvas->translate(frac, frac);
250 for (size_t m = 0; m < std::size(matrices); ++m) {
251 canvas->save();
252 canvas->concat(matrices[m]);
253 canvas->experimental_DrawEdgeAAImageSet(fSet, kM * kN, nullptr, nullptr,
254 SkSamplingOptions(SkFilterMode::kLinear),
255 &paint, SkCanvas::kFast_SrcRectConstraint);
256 canvas->restore();
257 canvas->translate(kTranslate, 0);
258 }
259 canvas->restore();
260 canvas->restore();
261 canvas->translate(0, kTranslate);
262 canvas->save();
263 }
264 for (SkVector scale : {SkVector{2.f, 0.5f}, SkVector{0.5, 2.f}}) {
265 SkCanvas::ImageSetEntry scaledSet[kM * kN];
266 std::copy_n(fSet, kM * kN, scaledSet);
267 for (int i = 0; i < kM * kN; ++i) {
268 scaledSet[i].fDstRect.fLeft *= scale.fX;
269 scaledSet[i].fDstRect.fTop *= scale.fY;
270 scaledSet[i].fDstRect.fRight *= scale.fX;
271 scaledSet[i].fDstRect.fBottom *= scale.fY;
272 scaledSet[i].fAlpha = 0 == (i % 3) ? 0.4f : 1.f;
273 }
274 for (size_t m = 0; m < std::size(matrices); ++m) {
275 canvas->save();
276 canvas->concat(matrices[m]);
277 canvas->experimental_DrawEdgeAAImageSet(scaledSet, kM * kN, nullptr, nullptr,
278 SkSamplingOptions(SkFilterMode::kLinear),
279 &paint, SkCanvas::kFast_SrcRectConstraint);
280 canvas->restore();
281 canvas->translate(kTranslate, 0);
282 }
283 canvas->restore();
284 canvas->translate(0, kTranslate);
285 canvas->save();
286 }
287 }
288 inline static constexpr int kM = 2;
289 inline static constexpr int kN = 2;
290 inline static constexpr int kTileW = 40;
291 inline static constexpr int kTileH = 50;
292 SkCanvas::ImageSetEntry fSet[kM * kN];
293 };
294
295 // This GM exercises alpha-only and color textures being combined correctly with the paint's color.
296 class DrawImageSetAlphaOnlyGM : public GM {
297 private:
getName() const298 SkString getName() const override { return SkString("draw_image_set_alpha_only"); }
getISize()299 SkISize getISize() override { return {kM * kTileW, 2 * kN * kTileH}; }
300
onGpuSetup(SkCanvas * canvas,SkString *,GraphiteTestContext *)301 DrawResult onGpuSetup(SkCanvas* canvas, SkString*, GraphiteTestContext*) override {
302 auto direct = GrAsDirectContext(canvas->recordingContext());
303 #if defined(SK_GRAPHITE)
304 auto recorder = canvas->recorder();
305 #endif
306 static constexpr SkColor kColors[] = {SK_ColorBLUE, SK_ColorTRANSPARENT,
307 SK_ColorRED, SK_ColorTRANSPARENT};
308 static constexpr SkColor kBGColor = SkColorSetARGB(128, 128, 128, 128);
309 make_image_tiles(kTileW, kTileH, kM, kN, kColors, fSet, kBGColor);
310
311 // Modify the alpha of the entries, decreasing by column, and convert even rows to
312 // alpha-only textures.
313 sk_sp<SkColorSpace> alphaSpace = SkColorSpace::MakeSRGB();
314 for (int y = 0; y < kN; ++y) {
315 for (int x = 0; x < kM; ++x) {
316 int i = y * kM + x;
317 fSet[i].fAlpha = (kM - x) / (float) kM;
318 if (y % 2 == 0) {
319 #if defined(SK_GRAPHITE)
320 if (recorder) {
321 fSet[i].fImage = fSet[i].fImage->makeColorTypeAndColorSpace(
322 recorder, kAlpha_8_SkColorType, alphaSpace, {});
323 } else
324 #endif
325 {
326 fSet[i].fImage = fSet[i].fImage->makeColorTypeAndColorSpace(
327 direct, kAlpha_8_SkColorType, alphaSpace);
328 }
329 }
330 }
331 }
332 return skiagm::DrawResult::kOk;
333 }
334
onDraw(SkCanvas * canvas)335 void onDraw(SkCanvas* canvas) override {
336 ToolUtils::draw_checkerboard(canvas, SK_ColorGRAY, SK_ColorDKGRAY, 25);
337
338 SkPaint paint;
339 paint.setBlendMode(SkBlendMode::kSrcOver);
340 paint.setColor4f({0.2f, 0.8f, 0.4f, 1.f}); // colorizes even rows, no effect on odd rows
341
342 // Top rows use experimental edge set API
343 canvas->experimental_DrawEdgeAAImageSet(fSet, kM * kN, nullptr, nullptr,
344 SkSamplingOptions(SkFilterMode::kLinear), &paint,
345 SkCanvas::kFast_SrcRectConstraint);
346
347 canvas->translate(0.f, kN * kTileH);
348
349 // Bottom rows draw each image from the set using the regular API
350 for (int y = 0; y < kN; ++y) {
351 for (int x = 0; x < kM; ++x) {
352 int i = y * kM + x;
353 SkPaint entryPaint = paint;
354 entryPaint.setAlphaf(fSet[i].fAlpha * paint.getAlphaf());
355 sk_sp<SkImage> orig = sk_ref_sp(const_cast<SkImage*>(fSet[i].fImage.get()));
356 canvas->drawImageRect(ToolUtils::MakeTextureImage(canvas, std::move(orig)),
357 fSet[i].fSrcRect, fSet[i].fDstRect,
358 SkSamplingOptions(), &entryPaint,
359 SkCanvas::kFast_SrcRectConstraint);
360 }
361 }
362 }
363
364 inline static constexpr int kM = 4;
365 inline static constexpr int kN = 4;
366 inline static constexpr int kTileW = 50;
367 inline static constexpr int kTileH = 50;
368 SkCanvas::ImageSetEntry fSet[kM * kN];
369 };
370
371 DEF_GM(return new DrawImageSetGM();)
372 DEF_GM(return new DrawImageSetRectToRectGM();)
373 DEF_GM(return new DrawImageSetAlphaOnlyGM();)
374
375 } // namespace skiagm
376