1 /*
2 * Copyright 2016 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/SkCanvas.h"
10 #include "include/core/SkColorSpace.h"
11 #include "include/core/SkImage.h"
12 #include "include/core/SkImageGenerator.h"
13 #include "include/core/SkImageInfo.h"
14 #include "include/core/SkPaint.h"
15 #include "include/core/SkRect.h"
16 #include "include/core/SkRefCnt.h"
17 #include "include/core/SkSize.h"
18 #include "include/core/SkSurface.h"
19 #include "include/core/SkTypes.h"
20 #include "include/gpu/ganesh/GrDirectContext.h"
21 #include "include/gpu/ganesh/SkSurfaceGanesh.h"
22 #include "tools/GpuToolUtils.h"
23 #include "tools/ToolUtils.h"
24
25 #if defined(SK_GRAPHITE)
26 #include "include/gpu/graphite/Surface.h"
27 #endif
28
29 namespace {
30
31 const SkISize kSize = SkISize::Make(100, 100);
32 const SkIRect kSubset = SkIRect::MakeLTRB(25, 25, 75, 75);
33 const SkRect kDest = SkRect::MakeXYWH(10, 10, 100, 100);
34
make_mask(const sk_sp<SkSurface> & surface)35 sk_sp<SkImage> make_mask(const sk_sp<SkSurface>& surface) {
36 ToolUtils::draw_checkerboard(surface->getCanvas(), 0x80808080, 0x00000000, 5);
37 return surface->makeImageSnapshot();
38 }
39
40 class MaskGenerator final : public SkImageGenerator {
41 public:
MaskGenerator(const SkImageInfo & info)42 MaskGenerator(const SkImageInfo& info) : INHERITED(info) {}
43
onGetPixels(const SkImageInfo & info,void * pixels,size_t rowBytes,const Options &)44 bool onGetPixels(const SkImageInfo& info, void* pixels,
45 size_t rowBytes, const Options&) override {
46 SkImageInfo surfaceInfo = info;
47 if (kAlpha_8_SkColorType == info.colorType()) {
48 surfaceInfo = surfaceInfo.makeColorSpace(nullptr);
49 }
50
51 make_mask(SkSurfaces::WrapPixels(surfaceInfo, pixels, rowBytes));
52 return true;
53 }
54
55 private:
56 using INHERITED = SkImageGenerator;
57 };
58
59 using MakerT = sk_sp<SkImage>(*)(SkCanvas*, const SkImageInfo&);
60 const MakerT makers[] = {
61 // SkImage_Raster
__anone42e12770202() 62 [](SkCanvas*, const SkImageInfo& info) -> sk_sp<SkImage> {
63 return make_mask(SkSurfaces::Raster(info));
64 },
65
66 // SkImage_Ganesh
__anone42e12770302() 67 [](SkCanvas* c, const SkImageInfo& info) -> sk_sp<SkImage> {
68 sk_sp<SkSurface> surface;
69 if (c->recordingContext()) {
70 surface =
71 SkSurfaces::RenderTarget(c->recordingContext(), skgpu::Budgeted::kNo, info);
72 } else {
73 #if defined(SK_GRAPHITE)
74 surface = SkSurfaces::RenderTarget(c->recorder(), info);
75 #endif
76 }
77 return make_mask(surface ? surface : SkSurfaces::Raster(info));
78 },
79
80 // SkImage_Lazy
__anone42e12770402() 81 [](SkCanvas*, const SkImageInfo& info) -> sk_sp<SkImage> {
82 return SkImages::DeferredFromGenerator(std::make_unique<MaskGenerator>(info));
83 },
84 };
85
86 } // namespace
87
88 // Checks whether subset SkImages preserve the original color type (A8 in this case).
89 DEF_SIMPLE_GM(imagemasksubset, canvas, 480, 480) {
90 SkPaint paint;
91 paint.setColor(0xff00ff00);
92
93 const SkImageInfo info = SkImageInfo::MakeA8(kSize.width(), kSize.height());
94
95 for (size_t i = 0; i < std::size(makers); ++i) {
96 sk_sp<SkImage> image = ToolUtils::MakeTextureImage(canvas, makers[i](canvas, info));
97 if (image) {
98 canvas->drawImageRect(image, SkRect::Make(kSubset), kDest, SkSamplingOptions(),
99 &paint, SkCanvas::kStrict_SrcRectConstraint);
100 sk_sp<SkImage> subset;
101
102 if (auto direct = GrAsDirectContext(canvas->recordingContext())) {
103 subset = image->makeSubset(direct, kSubset);
104 } else {
105 #if defined(SK_GRAPHITE)
106 subset = image->makeSubset(canvas->recorder(), kSubset, {});
107 #endif
108 }
109
110 canvas->drawImageRect(subset, kDest.makeOffset(kSize.width() * 1.5f, 0),
111 SkSamplingOptions(), &paint);
112 }
113 canvas->translate(0, kSize.height() * 1.5f);
114 }
115 }
116