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/GrDirectContext.h"
21 #include "tools/ToolUtils.h"
22
23 namespace {
24
25 const SkISize kSize = SkISize::Make(100, 100);
26 const SkIRect kSubset = SkIRect::MakeLTRB(25, 25, 75, 75);
27 const SkRect kDest = SkRect::MakeXYWH(10, 10, 100, 100);
28
make_mask(const sk_sp<SkSurface> & surface)29 sk_sp<SkImage> make_mask(const sk_sp<SkSurface>& surface) {
30 ToolUtils::draw_checkerboard(surface->getCanvas(), 0x80808080, 0x00000000, 5);
31 return surface->makeImageSnapshot();
32 }
33
34 class MaskGenerator final : public SkImageGenerator {
35 public:
MaskGenerator(const SkImageInfo & info)36 MaskGenerator(const SkImageInfo& info) : INHERITED(info) {}
37
onGetPixels(const SkImageInfo & info,void * pixels,size_t rowBytes,const Options &)38 bool onGetPixels(const SkImageInfo& info, void* pixels, size_t rowBytes, const Options&)
39 override {
40 SkImageInfo surfaceInfo = info;
41 if (kAlpha_8_SkColorType == info.colorType()) {
42 surfaceInfo = surfaceInfo.makeColorSpace(nullptr);
43 }
44
45 make_mask(SkSurface::MakeRasterDirect(surfaceInfo, pixels, rowBytes));
46 return true;
47 }
48
49 private:
50 using INHERITED = SkImageGenerator;
51 };
52
53 using MakerT = sk_sp<SkImage>(*)(SkCanvas*, const SkImageInfo&);
54 const MakerT makers[] = {
55 // SkImage_Raster
__anonb9b960a80202() 56 [](SkCanvas*, const SkImageInfo& info) -> sk_sp<SkImage> {
57 return make_mask(SkSurface::MakeRaster(info));
58 },
59
60 // SkImage_Gpu
__anonb9b960a80302() 61 [](SkCanvas* c, const SkImageInfo& info) -> sk_sp<SkImage> {
62 sk_sp<SkSurface> surface;
63 surface = SkSurface::MakeRenderTarget(c->recordingContext(), SkBudgeted::kNo, info);
64 return make_mask(surface ? surface : SkSurface::MakeRaster(info));
65 },
66
67 // SkImage_Lazy
__anonb9b960a80402() 68 [](SkCanvas*, const SkImageInfo& info) -> sk_sp<SkImage> {
69 return SkImage::MakeFromGenerator(std::make_unique<MaskGenerator>(info));
70 },
71 };
72
73 } // namespace
74
75 // Checks whether subset SkImages preserve the original color type (A8 in this case).
76 DEF_SIMPLE_GM(imagemasksubset, canvas, 480, 480) {
77 SkPaint paint;
78 paint.setColor(0xff00ff00);
79
80 const SkImageInfo info = SkImageInfo::MakeA8(kSize.width(), kSize.height());
81
82 for (size_t i = 0; i < SK_ARRAY_COUNT(makers); ++i) {
83 sk_sp<SkImage> image = makers[i](canvas, info);
84 if (image) {
85 canvas->drawImageRect(image, SkRect::Make(kSubset), kDest, SkSamplingOptions(),
86 &paint, SkCanvas::kStrict_SrcRectConstraint);
87 auto direct = GrAsDirectContext(canvas->recordingContext());
88 sk_sp<SkImage> subset = image->makeSubset(kSubset, direct);
89 canvas->drawImageRect(subset, kDest.makeOffset(kSize.width() * 1.5f, 0),
90 SkSamplingOptions(), &paint);
91 }
92 canvas->translate(0, kSize.height() * 1.5f);
93 }
94 }
95