Mike Klein | 8f5a7a6 | 2018-09-11 12:11:46 -0400 | [diff] [blame] | 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.h" |
| 9 | #include "SkColorSpace.h" |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 10 | #include "SkColorSpaceXformSteps.h" |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 11 | #include "SkGradientShader.h" |
Mike Klein | 8f5a7a6 | 2018-09-11 12:11:46 -0400 | [diff] [blame] | 12 | #include "SkString.h" |
| 13 | |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 14 | template <typename Fn> |
| 15 | static void mark(SkCanvas* canvas, Fn&& fn) { |
| 16 | SkPaint alpha; |
| 17 | alpha.setAlpha(0x50); |
| 18 | canvas->saveLayer(nullptr, &alpha); |
| 19 | canvas->translate(140,40); |
| 20 | canvas->scale(2,2); |
| 21 | fn(); |
| 22 | canvas->restore(); |
| 23 | } |
| 24 | |
| 25 | static void mark_good(SkCanvas* canvas) { |
| 26 | mark(canvas, [&]{ |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 27 | SkPaint paint; |
| 28 | |
| 29 | // A green circle. |
| 30 | paint.setColor(SkColorSetRGB(27, 158, 119)); |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 31 | canvas->drawCircle(0,0, 12, paint); |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 32 | |
| 33 | // Cut out a check mark. |
| 34 | paint.setBlendMode(SkBlendMode::kSrc); |
| 35 | paint.setColor(0x00000000); |
| 36 | paint.setStrokeWidth(2); |
| 37 | paint.setStyle(SkPaint::kStroke_Style); |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 38 | canvas->drawLine(-6, 0, |
| 39 | -1, 5, paint); |
| 40 | canvas->drawLine(-1, +5, |
| 41 | +7, -5, paint); |
| 42 | }); |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 43 | } |
| 44 | |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 45 | static void mark_bad(SkCanvas* canvas) { |
| 46 | mark(canvas, [&] { |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 47 | SkPaint paint; |
| 48 | |
| 49 | // A red circle. |
| 50 | paint.setColor(SkColorSetRGB(231, 41, 138)); |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 51 | canvas->drawCircle(0,0, 12, paint); |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 52 | |
| 53 | // Cut out an 'X'. |
| 54 | paint.setBlendMode(SkBlendMode::kSrc); |
| 55 | paint.setColor(0x00000000); |
| 56 | paint.setStrokeWidth(2); |
| 57 | paint.setStyle(SkPaint::kStroke_Style); |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 58 | canvas->drawLine(-5,-5, |
| 59 | +5,+5, paint); |
| 60 | canvas->drawLine(+5,-5, |
| 61 | -5,+5, paint); |
| 62 | }); |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 63 | } |
| 64 | |
| 65 | static bool nearly_equal(SkColor4f x, SkColor4f y) { |
| 66 | const float K = 0.01f; |
| 67 | return fabsf(x.fR - y.fR) < K |
| 68 | && fabsf(x.fG - y.fG) < K |
| 69 | && fabsf(x.fB - y.fB) < K |
| 70 | && fabsf(x.fA - y.fA) < K; |
| 71 | } |
| 72 | |
| 73 | static SkString fmt(SkColor4f c) { |
| 74 | return SkStringPrintf("%.2g %.2g %.2g %.2g", c.fR, c.fG, c.fB, c.fA); |
| 75 | } |
| 76 | |
| 77 | static SkColor4f transform(SkColor4f c, SkColorSpace* src, SkColorSpace* dst) { |
| 78 | SkColorSpaceXformSteps(src, kUnpremul_SkAlphaType, |
| 79 | dst, kUnpremul_SkAlphaType).apply(c.vec()); |
| 80 | return c; |
| 81 | } |
| 82 | |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 83 | static void compare_pixel(const char* label, |
| 84 | SkCanvas* canvas, int x, int y, |
| 85 | SkColor4f color, SkColorSpace* cs) { |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 86 | SkPaint text; |
| 87 | text.setAntiAlias(true); |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 88 | auto canvas_cs = canvas->imageInfo().refColorSpace(); |
Mike Klein | 8f5a7a6 | 2018-09-11 12:11:46 -0400 | [diff] [blame] | 89 | |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 90 | // I'm not really sure if this makes things easier or harder to follow, |
| 91 | // but we sniff the canvas to grab its current y-translate, so that (x,y) |
| 92 | // can be written in sort of chunk-relative terms. |
| 93 | const SkMatrix& m = canvas->getTotalMatrix(); |
| 94 | SkASSERT(m.isTranslate()); |
| 95 | SkScalar dy = m.getTranslateY(); |
| 96 | SkASSERT(dy == (int)dy); |
| 97 | y += (int)dy; |
Mike Klein | 8f5a7a6 | 2018-09-11 12:11:46 -0400 | [diff] [blame] | 98 | |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 99 | SkBitmap bm; |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 100 | bm.allocPixels(SkImageInfo::Make(1,1, kRGBA_F32_SkColorType, kUnpremul_SkAlphaType, canvas_cs)); |
| 101 | if (!canvas->readPixels(bm, x,y)) { |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 102 | mark_good(canvas); |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 103 | canvas->drawString("can't readPixels() on this canvas :(", 100,20, text); |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 104 | return; |
| 105 | } |
Mike Klein | 8f5a7a6 | 2018-09-11 12:11:46 -0400 | [diff] [blame] | 106 | |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 107 | SkColor4f pixel; |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 108 | memcpy(&pixel, bm.getAddr(0,0), sizeof(pixel)); |
Mike Klein | 8f5a7a6 | 2018-09-11 12:11:46 -0400 | [diff] [blame] | 109 | |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 110 | SkColor4f expected = transform(color,cs, canvas_cs.get()); |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 111 | if (canvas->imageInfo().colorType() < kRGBA_F16_SkColorType) { |
| 112 | // We can't expect normalized formats to hold values outside [0,1]. |
| 113 | expected = expected.pin(); |
| 114 | } |
| 115 | if (canvas->imageInfo().colorType() == kGray_8_SkColorType) { |
| 116 | // Drawing into Gray8 is known to be maybe-totally broken. |
| 117 | // TODO: update expectation here to be {lum,lum,lum,1} if we fix Gray8. |
| 118 | expected = SkColor4f{NAN, NAN, NAN, 1}; |
| 119 | } |
Mike Klein | 8f5a7a6 | 2018-09-11 12:11:46 -0400 | [diff] [blame] | 120 | |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 121 | if (nearly_equal(pixel, expected)) { |
| 122 | mark_good(canvas); |
Mike Klein | 8f5a7a6 | 2018-09-11 12:11:46 -0400 | [diff] [blame] | 123 | } else { |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 124 | mark_bad(canvas); |
| 125 | } |
| 126 | |
| 127 | struct { |
| 128 | const char* label; |
| 129 | SkColor4f color; |
| 130 | } lines[] = { |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 131 | {"Pixel:" , pixel }, |
| 132 | {"Expected:", expected}, |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 133 | }; |
| 134 | |
| 135 | SkAutoCanvasRestore saveRestore(canvas, true); |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 136 | canvas->drawString(label, 80,20, text); |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 137 | for (auto l : lines) { |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 138 | canvas->translate(0,20); |
Mike Klein | fd00764 | 2018-09-12 13:03:06 -0400 | [diff] [blame] | 139 | canvas->drawString(l.label, 80,20, text); |
| 140 | canvas->drawString(fmt(l.color).c_str(), 140,20, text); |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 141 | } |
| 142 | } |
| 143 | |
Mike Klein | 983886e | 2018-09-20 14:10:33 -0400 | [diff] [blame] | 144 | DEF_SIMPLE_GM(p3, canvas, 450, 1000) { |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 145 | auto p3 = SkColorSpace::MakeRGB(SkColorSpace::kSRGB_RenderTargetGamma, |
| 146 | SkColorSpace::kDCIP3_D65_Gamut); |
| 147 | |
| 148 | // Draw a P3 red rectangle and check the corner. |
| 149 | { |
| 150 | SkPaint paint; |
| 151 | paint.setColor4f({1,0,0,1}, p3.get()); |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 152 | |
Mike Klein | 62bd12f | 2018-09-17 12:39:09 -0400 | [diff] [blame] | 153 | canvas->drawRect({10,10,70,70}, paint); |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 154 | compare_pixel("drawRect P3 red ", |
| 155 | canvas, 10,10, |
| 156 | {1,0,0,1}, p3.get()); |
| 157 | } |
| 158 | |
| 159 | canvas->translate(0,80); |
| 160 | |
Mike Klein | 62bd12f | 2018-09-17 12:39:09 -0400 | [diff] [blame] | 161 | // Draw a P3 red bitmap, using a draw. |
| 162 | { |
| 163 | SkBitmap bm; |
| 164 | bm.allocPixels(SkImageInfo::Make(60,60, kRGBA_F16_SkColorType, kPremul_SkAlphaType, p3)); |
| 165 | |
| 166 | SkPaint paint; |
| 167 | paint.setColor4f({1,0,0,1}, p3.get()); |
| 168 | SkCanvas{bm}.drawPaint(paint); |
| 169 | |
| 170 | canvas->drawBitmap(bm, 10,10); |
| 171 | compare_pixel("drawBitmap P3 red, from drawPaint", |
| 172 | canvas, 10,10, |
| 173 | {1,0,0,1}, p3.get()); |
| 174 | } |
| 175 | |
| 176 | canvas->translate(0,80); |
| 177 | |
| 178 | // Draw a P3 red bitmap, using SkBitmap::eraseColor(). |
| 179 | { |
| 180 | SkBitmap bm; |
| 181 | bm.allocPixels(SkImageInfo::Make(60,60, kRGBA_F16_SkColorType, kPremul_SkAlphaType, p3)); |
| 182 | |
| 183 | bm.eraseColor(0xffff0000/*in P3*/); |
| 184 | |
| 185 | canvas->drawBitmap(bm, 10,10); |
| 186 | compare_pixel("drawBitmap P3 red, from SkBitmap::eraseColor()", |
| 187 | canvas, 10,10, |
| 188 | {1,0,0,1}, p3.get()); |
| 189 | } |
| 190 | |
| 191 | canvas->translate(0,80); |
| 192 | |
| 193 | // Draw a P3 red bitmap, using SkPixmap::erase(). |
| 194 | { |
| 195 | SkBitmap bm; |
| 196 | bm.allocPixels(SkImageInfo::Make(60,60, kRGBA_F16_SkColorType, kPremul_SkAlphaType, p3)); |
| 197 | |
| 198 | // At the moment only SkPixmap has an erase() that takes an SkColor4f. |
| 199 | SkPixmap pm; |
| 200 | SkAssertResult(bm.peekPixels(&pm)); |
| 201 | SkAssertResult(pm.erase({1,0,0,1} /*in p3*/)); |
| 202 | |
| 203 | canvas->drawBitmap(bm, 10,10); |
| 204 | compare_pixel("drawBitmap P3 red, from SkPixmap::erase", |
| 205 | canvas, 10,10, |
| 206 | {1,0,0,1}, p3.get()); |
| 207 | } |
| 208 | |
| 209 | canvas->translate(0,80); |
| 210 | |
Mike Klein | 983886e | 2018-09-20 14:10:33 -0400 | [diff] [blame] | 211 | // Draw a gradient from P3 red to P3 green interpolating in unpremul, and check the corners. |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 212 | { |
| 213 | |
| 214 | SkPoint points[] = {{10.5,10.5}, {69.5,69.5}}; |
| 215 | SkColor4f colors[] = {{1,0,0,1}, {0,1,0,1}}; |
| 216 | |
| 217 | SkPaint paint; |
| 218 | paint.setShader(SkGradientShader::MakeLinear(points, colors, p3, |
| 219 | nullptr, SK_ARRAY_COUNT(colors), |
| 220 | SkShader::kClamp_TileMode)); |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 221 | canvas->drawRect({10,10,70,70}, paint); |
Mike Klein | 5e31ec6 | 2018-09-12 18:04:49 -0400 | [diff] [blame] | 222 | canvas->save(); |
| 223 | compare_pixel("gradient P3 red", |
| 224 | canvas, 10,10, |
| 225 | {1,0,0,1}, p3.get()); |
| 226 | |
| 227 | canvas->translate(180, 0); |
| 228 | |
| 229 | compare_pixel("gradient P3 green", |
| 230 | canvas, 69,69, |
| 231 | {0,1,0,1}, p3.get()); |
| 232 | canvas->restore(); |
Mike Klein | 8f5a7a6 | 2018-09-11 12:11:46 -0400 | [diff] [blame] | 233 | } |
| 234 | |
Mike Klein | 9fb5a53 | 2018-09-13 15:23:38 -0400 | [diff] [blame] | 235 | canvas->translate(0,80); |
| 236 | |
Mike Klein | 983886e | 2018-09-20 14:10:33 -0400 | [diff] [blame] | 237 | // Draw a gradient from P3 red to P3 green interpolating in premul, and check the corners. |
| 238 | { |
| 239 | |
| 240 | SkPoint points[] = {{10.5,10.5}, {69.5,69.5}}; |
| 241 | SkColor4f colors[] = {{1,0,0,1}, {0,1,0,1}}; |
| 242 | |
| 243 | SkPaint paint; |
| 244 | paint.setShader( |
| 245 | SkGradientShader::MakeLinear(points, colors, p3, |
| 246 | nullptr, SK_ARRAY_COUNT(colors), |
| 247 | SkShader::kClamp_TileMode, |
| 248 | SkGradientShader::kInterpolateColorsInPremul_Flag, |
| 249 | nullptr/*local matrix*/)); |
| 250 | canvas->drawRect({10,10,70,70}, paint); |
| 251 | canvas->save(); |
| 252 | compare_pixel("gradient P3 red", |
| 253 | canvas, 10,10, |
| 254 | {1,0,0,1}, p3.get()); |
| 255 | |
| 256 | canvas->translate(180, 0); |
| 257 | |
| 258 | compare_pixel("gradient P3 green", |
| 259 | canvas, 69,69, |
| 260 | {0,1,0,1}, p3.get()); |
| 261 | canvas->restore(); |
| 262 | } |
| 263 | |
| 264 | canvas->translate(0,80); |
| 265 | |
Mike Klein | be56949 | 2018-09-14 09:34:21 -0400 | [diff] [blame] | 266 | // Draw an A8 image with a P3 red, scaled and not, as a shader or bitmap. |
Mike Klein | 9fb5a53 | 2018-09-13 15:23:38 -0400 | [diff] [blame] | 267 | { |
| 268 | uint8_t mask[256]; |
| 269 | for (int i = 0; i < 256; i++) { |
| 270 | mask[i] = 255-i; |
| 271 | } |
| 272 | SkBitmap bm; |
| 273 | bm.installPixels(SkImageInfo::MakeA8(16,16), mask, 16); |
| 274 | |
Mike Klein | be56949 | 2018-09-14 09:34:21 -0400 | [diff] [blame] | 275 | SkPaint as_bitmap; |
| 276 | as_bitmap.setColor4f({1,0,0,1}, p3.get()); |
| 277 | as_bitmap.setFilterQuality(kLow_SkFilterQuality); |
Mike Klein | 9fb5a53 | 2018-09-13 15:23:38 -0400 | [diff] [blame] | 278 | |
Mike Klein | be56949 | 2018-09-14 09:34:21 -0400 | [diff] [blame] | 279 | SkPaint as_shader; |
| 280 | as_shader.setColor4f({1,0,0,1}, p3.get()); |
| 281 | as_shader.setFilterQuality(kLow_SkFilterQuality); |
| 282 | as_shader.setShader(SkShader::MakeBitmapShader(bm, SkShader::kClamp_TileMode |
| 283 | , SkShader::kClamp_TileMode)); |
| 284 | |
| 285 | canvas->drawBitmap(bm, 10,10, &as_bitmap); |
| 286 | compare_pixel("A8 sprite bitmap P3 red", |
| 287 | canvas, 10,10, |
| 288 | {1,0,0,1}, p3.get()); |
| 289 | |
| 290 | canvas->translate(0, 80); |
| 291 | |
| 292 | canvas->save(); |
| 293 | canvas->translate(10,10); |
| 294 | canvas->drawRect({0,0,16,16}, as_shader); |
| 295 | canvas->restore(); |
| 296 | compare_pixel("A8 sprite shader P3 red", |
Mike Klein | 9fb5a53 | 2018-09-13 15:23:38 -0400 | [diff] [blame] | 297 | canvas, 10,10, |
| 298 | {1,0,0,1}, p3.get()); |
| 299 | |
| 300 | canvas->translate(0,80); |
| 301 | |
Mike Klein | be56949 | 2018-09-14 09:34:21 -0400 | [diff] [blame] | 302 | canvas->drawBitmapRect(bm, {10,10,70,70}, &as_bitmap); |
| 303 | compare_pixel("A8 scaled bitmap P3 red", |
| 304 | canvas, 10,10, |
| 305 | {1,0,0,1}, p3.get()); |
| 306 | |
| 307 | canvas->translate(0,80); |
| 308 | |
| 309 | canvas->save(); |
| 310 | canvas->translate(10,10); |
| 311 | canvas->scale(3.75,3.75); |
| 312 | canvas->drawRect({0,0,16,16}, as_shader); |
| 313 | canvas->restore(); |
| 314 | compare_pixel("A8 scaled shader P3 red", |
Mike Klein | 9fb5a53 | 2018-09-13 15:23:38 -0400 | [diff] [blame] | 315 | canvas, 10,10, |
| 316 | {1,0,0,1}, p3.get()); |
| 317 | } |
| 318 | |
Mike Klein | 8f5a7a6 | 2018-09-11 12:11:46 -0400 | [diff] [blame] | 319 | // TODO: draw P3 colors more ways |
| 320 | } |