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 | 9fb5a53 | 2018-09-13 15:23:38 -0400 | [diff] [blame^] | 144 | DEF_SIMPLE_GM(p3, canvas, 450, 400) { |
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()); |
| 152 | canvas->drawRect({10,10,70,70}, SkPaint{}); |
| 153 | canvas->drawRect({10,10,70,70}, paint); |
| 154 | |
| 155 | compare_pixel("drawRect P3 red ", |
| 156 | canvas, 10,10, |
| 157 | {1,0,0,1}, p3.get()); |
| 158 | } |
| 159 | |
| 160 | canvas->translate(0,80); |
| 161 | |
| 162 | // Draw a gradient from P3 red to P3 green, and check the corners. |
| 163 | { |
| 164 | |
| 165 | SkPoint points[] = {{10.5,10.5}, {69.5,69.5}}; |
| 166 | SkColor4f colors[] = {{1,0,0,1}, {0,1,0,1}}; |
| 167 | |
| 168 | SkPaint paint; |
| 169 | paint.setShader(SkGradientShader::MakeLinear(points, colors, p3, |
| 170 | nullptr, SK_ARRAY_COUNT(colors), |
| 171 | SkShader::kClamp_TileMode)); |
| 172 | canvas->drawRect({10,10,70,70}, SkPaint{}); |
| 173 | canvas->drawRect({10,10,70,70}, paint); |
| 174 | |
| 175 | canvas->save(); |
| 176 | compare_pixel("gradient P3 red", |
| 177 | canvas, 10,10, |
| 178 | {1,0,0,1}, p3.get()); |
| 179 | |
| 180 | canvas->translate(180, 0); |
| 181 | |
| 182 | compare_pixel("gradient P3 green", |
| 183 | canvas, 69,69, |
| 184 | {0,1,0,1}, p3.get()); |
| 185 | canvas->restore(); |
Mike Klein | 8f5a7a6 | 2018-09-11 12:11:46 -0400 | [diff] [blame] | 186 | } |
| 187 | |
Mike Klein | 9fb5a53 | 2018-09-13 15:23:38 -0400 | [diff] [blame^] | 188 | canvas->translate(0,80); |
| 189 | |
| 190 | // Draw an A8 image with the paint color set to P3 red, hitting a sprite case. |
| 191 | // and |
| 192 | // Draw a scaled A8 image with the paint color set to P3 red, hitting a non-sprite case. |
| 193 | { |
| 194 | uint8_t mask[256]; |
| 195 | for (int i = 0; i < 256; i++) { |
| 196 | mask[i] = 255-i; |
| 197 | } |
| 198 | SkBitmap bm; |
| 199 | bm.installPixels(SkImageInfo::MakeA8(16,16), mask, 16); |
| 200 | |
| 201 | SkPaint paint; |
| 202 | paint.setFilterQuality(kLow_SkFilterQuality); |
| 203 | paint.setColor4f({1,0,0,1}, p3.get()); |
| 204 | |
| 205 | canvas->drawBitmap(bm, 10,10, &paint); |
| 206 | compare_pixel("A8 sprite P3 red", |
| 207 | canvas, 10,10, |
| 208 | {1,0,0,1}, p3.get()); |
| 209 | |
| 210 | canvas->translate(0,80); |
| 211 | |
| 212 | canvas->drawBitmapRect(bm, {10,10,70,70}, &paint); |
| 213 | compare_pixel("A8 image P3 red", |
| 214 | canvas, 10,10, |
| 215 | {1,0,0,1}, p3.get()); |
| 216 | } |
| 217 | |
Mike Klein | 8f5a7a6 | 2018-09-11 12:11:46 -0400 | [diff] [blame] | 218 | // TODO: draw P3 colors more ways |
| 219 | } |