1 /*
2 * Copyright 2017 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/codec/SkCodec.h"
10 #include "include/core/SkBitmap.h"
11 #include "include/core/SkCanvas.h"
12 #include "include/core/SkColor.h"
13 #include "include/core/SkColorSpace.h"
14 #include "include/core/SkData.h"
15 #include "include/core/SkImage.h"
16 #include "include/core/SkImageInfo.h"
17 #include "include/core/SkRefCnt.h"
18 #include "include/core/SkSize.h"
19 #include "include/core/SkString.h"
20 #include "include/gpu/GrDirectContext.h"
21 #include "src/core/SkImagePriv.h"
22 #include "tools/Resources.h"
23
24 #include <initializer_list>
25 #include <memory>
26
make_raster_image(const char * path)27 sk_sp<SkImage> make_raster_image(const char* path) {
28 sk_sp<SkData> resourceData = GetResourceAsData(path);
29 std::unique_ptr<SkCodec> codec = SkCodec::MakeFromData(resourceData);
30
31 SkBitmap bitmap;
32 bitmap.allocPixels(codec->getInfo());
33
34 codec->getPixels(codec->getInfo(), bitmap.getPixels(), bitmap.rowBytes());
35 return SkImage::MakeFromBitmap(bitmap);
36 }
37
make_color_space(sk_sp<SkImage> orig,sk_sp<SkColorSpace> colorSpace,GrDirectContext * direct)38 sk_sp<SkImage> make_color_space(
39 sk_sp<SkImage> orig, sk_sp<SkColorSpace> colorSpace, GrDirectContext* direct) {
40 sk_sp<SkImage> xform = orig->makeColorSpace(colorSpace, direct);
41
42 // Assign an sRGB color space on the xformed image, so we can see the effects of the xform
43 // when we draw.
44 sk_sp<SkColorSpace> srgb = SkColorSpace::MakeSRGB();
45 if (colorSpace->gammaIsLinear()) {
46 srgb = SkColorSpace::MakeSRGBLinear();
47 }
48 return xform->reinterpretColorSpace(std::move(srgb));
49 }
50
51 DEF_SIMPLE_GM_CAN_FAIL(makecolorspace, canvas, errorMsg, 128 * 3, 128 * 4) {
52 sk_sp<SkColorSpace> wideGamut = SkColorSpace::MakeRGB(SkNamedTransferFn::kSRGB,
53 SkNamedGamut::kAdobeRGB);
54 sk_sp<SkColorSpace> wideGamutLinear = wideGamut->makeLinearGamma();
55
56 // Lazy images
57 sk_sp<SkImage> opaqueImage = GetResourceAsImage("images/mandrill_128.png");
58 sk_sp<SkImage> premulImage = GetResourceAsImage("images/color_wheel.png");
59 if (!opaqueImage || !premulImage) {
60 *errorMsg = "Failed to load images. Did you forget to set the resourcePath?";
61 return skiagm::DrawResult::kFail;
62 }
63 auto direct = GrAsDirectContext(canvas->recordingContext());
64 canvas->drawImage(opaqueImage, 0.0f, 0.0f);
65 canvas->drawImage(make_color_space(opaqueImage, wideGamut, direct), 128.0f, 0.0f);
66 canvas->drawImage(make_color_space(opaqueImage, wideGamutLinear, direct), 256.0f, 0.0f);
67 canvas->drawImage(premulImage, 0.0f, 128.0f);
68 canvas->drawImage(make_color_space(premulImage, wideGamut, direct), 128.0f, 128.0f);
69 canvas->drawImage(make_color_space(premulImage, wideGamutLinear, direct), 256.0f, 128.0f);
70 canvas->translate(0.0f, 256.0f);
71
72 // Raster images
73 opaqueImage = make_raster_image("images/mandrill_128.png");
74 premulImage = make_raster_image("images/color_wheel.png");
75 canvas->drawImage(opaqueImage, 0.0f, 0.0f);
76 canvas->drawImage(make_color_space(opaqueImage, wideGamut, direct), 128.0f, 0.0f);
77 canvas->drawImage(make_color_space(opaqueImage, wideGamutLinear, direct), 256.0f, 0.0f);
78 canvas->drawImage(premulImage, 0.0f, 128.0f);
79 canvas->drawImage(make_color_space(premulImage, wideGamut, direct), 128.0f, 128.0f);
80 canvas->drawImage(make_color_space(premulImage, wideGamutLinear, direct), 256.0f, 128.0f);
81 return skiagm::DrawResult::kOk;
82 }
83
84 DEF_SIMPLE_GM_BG(makecolortypeandspace, canvas, 128 * 3, 128 * 4, SK_ColorWHITE) {
85 sk_sp<SkImage> images[] = {
86 GetResourceAsImage("images/mandrill_128.png"),
87 GetResourceAsImage("images/color_wheel.png"),
88 };
89 auto rec2020 = SkColorSpace::MakeRGB(SkNamedTransferFn::kSRGB, SkNamedGamut::kRec2020);
90
91 // Use the lazy images on the first iteration, and concrete (raster/GPU) images on the second
92 auto direct = GrAsDirectContext(canvas->recordingContext());
93 for (bool lazy : {true, false}) {
94 for (int j = 0; j < 2; ++j) {
95 const SkImage* image = images[j].get();
96 if (!image) {
97 // Can happen on bots that abandon the GPU context
98 continue;
99 }
100
101 // Unmodified
102 canvas->drawImage(image, 0, 0);
103
104 // Change the color type/space of the image in a couple ways. In both cases, codec
105 // may fail, because we refude to decode transparent sources to opaque color types.
106 // Guard against that, to avoid cascading failures in DDL.
107
108 // 565 in a wide color space (should be visibly quantized). Fails with the color_wheel,
109 // because of the codec issues mentioned above.
110 auto image565 = image->makeColorTypeAndColorSpace(kRGB_565_SkColorType,
111 rec2020, direct);
112 if (!lazy || image565->makeRasterImage()) {
113 canvas->drawImage(image565, 128, 0);
114 }
115
116 // Grayscale in the original color space. This fails in even more cases, due to the
117 // above opaque issue, and because Ganesh doesn't support drawing to gray, at all.
118 auto imageGray = image->makeColorTypeAndColorSpace(kGray_8_SkColorType,
119 image->refColorSpace(),
120 direct);
121 if (!lazy || imageGray->makeRasterImage()) {
122 canvas->drawImage(imageGray, 256, 0);
123 }
124
125 images[j] = direct ? image->makeTextureImage(direct) : image->makeRasterImage();
126
127 canvas->translate(0, 128);
128 }
129 }
130 }
131
132 DEF_SIMPLE_GM_CAN_FAIL(reinterpretcolorspace, canvas, errorMsg, 128 * 3, 128 * 3) {
133 // We draw a 3x3 grid. The three rows are lazy (encoded), raster, and GPU (or another copy of
134 // raster so all configs look similar). In each row, we draw three variants:
135 // - The original image (should look normal).
136 // - The image, reinterpreted as being in the color-spin space. The pixel data isn't changed,
137 // so in untagged configs, this looks like the first column. In tagged configs, this has the
138 // the effect of rotating the colors (RGB -> GBR).
139 // - The image converted to the color-spin space, then reinterpreted as being sRGB. In all
140 // configs, this appears to be spun backwards (RGB -> BRG), and tests the composition of
141 // these two APIs.
142
143 // In all cases, every column should be identical. In tagged configs, the 'R' in the columns
144 // should be Red, Green, Blue.
145
146 sk_sp<SkColorSpace> srgb = SkColorSpace::MakeSRGB();
147 sk_sp<SkColorSpace> spin = srgb->makeColorSpin();
148 sk_sp<SkImage> image = GetResourceAsImage("images/color_wheel.png");
149 if (!image) {
150 *errorMsg = "Failed to load image. Did you forget to set the resourcePath?";
151 return skiagm::DrawResult::kFail;
152 }
153
154 // Lazy images
155 canvas->drawImage(image, 0.0f, 0.0f);
156 canvas->drawImage(image->reinterpretColorSpace(spin), 128.0f, 0.0f);
157 canvas->drawImage(image->makeColorSpace(spin, nullptr)->reinterpretColorSpace(srgb),
158 256.0f, 0.0f);
159
160 canvas->translate(0.0f, 128.0f);
161
162 // Raster images
163 image = image->makeRasterImage();
164 canvas->drawImage(image, 0.0f, 0.0f);
165 canvas->drawImage(image->reinterpretColorSpace(spin), 128.0f, 0.0f);
166 canvas->drawImage(image->makeColorSpace(spin, nullptr)->reinterpretColorSpace(srgb),
167 256.0f, 0.0f);
168
169 canvas->translate(0.0f, 128.0f);
170
171 // GPU images
172 auto direct = GrAsDirectContext(canvas->recordingContext());
173 if (auto gpuImage = image->makeTextureImage(direct)) {
174 image = gpuImage;
175 }
176
177 canvas->drawImage(image, 0.0f, 0.0f);
178 canvas->drawImage(image->reinterpretColorSpace(spin), 128.0f, 0.0f);
179 canvas->drawImage(image->makeColorSpace(spin, direct)->reinterpretColorSpace(srgb),
180 256.0f, 0.0f);
181
182 return skiagm::DrawResult::kOk;
183 }
184