123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472 |
- /*
- * Copyright 2018 Google Inc.
- *
- * Use of this source code is governed by a BSD-style license that can be
- * found in the LICENSE file.
- */
- #include "gm/gm.h"
- #include "include/core/SkBitmap.h"
- #include "include/core/SkCanvas.h"
- #include "include/core/SkColor.h"
- #include "include/core/SkColorSpace.h"
- #include "include/core/SkFilterQuality.h"
- #include "include/core/SkFont.h"
- #include "include/core/SkImageInfo.h"
- #include "include/core/SkMatrix.h"
- #include "include/core/SkPaint.h"
- #include "include/core/SkPathEffect.h"
- #include "include/core/SkPixmap.h"
- #include "include/core/SkPoint.h"
- #include "include/core/SkRefCnt.h"
- #include "include/core/SkScalar.h"
- #include "include/core/SkShader.h"
- #include "include/core/SkString.h"
- #include "include/core/SkTileMode.h"
- #include "include/core/SkTypes.h"
- #include "include/effects/SkDashPathEffect.h"
- #include "include/effects/SkGradientShader.h"
- #include "src/core/SkColorSpaceXformSteps.h"
- #include <math.h>
- #include <string.h>
- static bool nearly_equal(SkColor4f x, SkColor4f y) {
- const float K = 0.01f;
- return fabsf(x.fR - y.fR) < K
- && fabsf(x.fG - y.fG) < K
- && fabsf(x.fB - y.fB) < K
- && fabsf(x.fA - y.fA) < K;
- }
- static SkString fmt(SkColor4f c) {
- return SkStringPrintf("%.2g %.2g %.2g %.2g", c.fR, c.fG, c.fB, c.fA);
- }
- static SkColor4f transform(SkColor4f c, SkColorSpace* src, SkColorSpace* dst) {
- SkColorSpaceXformSteps(src, kUnpremul_SkAlphaType,
- dst, kUnpremul_SkAlphaType).apply(c.vec());
- return c;
- }
- static void compare_pixel(const char* label,
- SkCanvas* canvas, int x, int y,
- SkColor4f color, SkColorSpace* cs) {
- SkPaint paint;
- SkFont font;
- auto canvas_cs = canvas->imageInfo().refColorSpace();
- // I'm not really sure if this makes things easier or harder to follow,
- // but we sniff the canvas to grab its current y-translate, so that (x,y)
- // can be written in sort of chunk-relative terms.
- const SkMatrix& m = canvas->getTotalMatrix();
- SkASSERT(m.isTranslate());
- SkScalar dy = m.getTranslateY();
- SkASSERT(dy == (int)dy);
- y += (int)dy;
- SkBitmap bm;
- bm.allocPixels(SkImageInfo::Make(1,1, kRGBA_F32_SkColorType, kUnpremul_SkAlphaType, canvas_cs));
- if (!canvas->readPixels(bm, x,y)) {
- MarkGMGood(canvas, 140,40);
- canvas->drawString("can't readPixels() on this canvas :(", 100,20, font, paint);
- return;
- }
- SkColor4f pixel;
- memcpy(&pixel, bm.getAddr(0,0), sizeof(pixel));
- SkColor4f expected = transform(color,cs, canvas_cs.get());
- if (canvas->imageInfo().colorType() < kRGBA_F16_SkColorType) {
- // We can't expect normalized formats to hold values outside [0,1].
- for (int i = 0; i < 4; ++i) {
- expected[i] = SkTPin(expected[i], 0.0f, 1.0f);
- }
- }
- if (canvas->imageInfo().colorType() == kGray_8_SkColorType) {
- // Drawing into Gray8 is known to be maybe-totally broken.
- // TODO: update expectation here to be {lum,lum,lum,1} if we fix Gray8.
- expected = SkColor4f{NAN, NAN, NAN, 1};
- }
- if (nearly_equal(pixel, expected)) {
- MarkGMGood(canvas, 140,40);
- } else {
- MarkGMBad(canvas, 140,40);
- }
- struct {
- const char* label;
- SkColor4f color;
- } lines[] = {
- {"Pixel:" , pixel },
- {"Expected:", expected},
- };
- SkAutoCanvasRestore saveRestore(canvas, true);
- canvas->drawString(label, 80,20, font, paint);
- for (auto l : lines) {
- canvas->translate(0,20);
- canvas->drawString(l.label, 80,20, font, paint);
- canvas->drawString(fmt(l.color).c_str(), 140,20, font, paint);
- }
- }
- DEF_SIMPLE_GM(p3, canvas, 450, 1300) {
- auto p3 = SkColorSpace::MakeRGB(SkNamedTransferFn::kSRGB, SkNamedGamut::kDCIP3);
- auto srgb = SkColorSpace::MakeSRGB();
- auto p3_to_srgb = [&](SkColor4f c) {
- SkPaint p;
- p.setColor4f(c, p3.get());
- return p.getColor4f();
- };
- // Draw a P3 red rectangle and check the corner.
- {
- SkPaint paint;
- paint.setColor4f({1,0,0,1}, p3.get());
- canvas->drawRect({10,10,70,70}, paint);
- compare_pixel("drawRect P3 red ",
- canvas, 10,10,
- {1,0,0,1}, p3.get());
- }
- canvas->translate(0,80);
- // Draw a P3 red bitmap, using a draw.
- {
- SkBitmap bm;
- bm.allocPixels(SkImageInfo::Make(60,60, kRGBA_F16_SkColorType, kPremul_SkAlphaType, p3));
- SkPaint paint;
- paint.setColor4f({1,0,0,1}, p3.get());
- SkCanvas{bm}.drawPaint(paint);
- canvas->drawBitmap(bm, 10,10);
- compare_pixel("drawBitmap P3 red, from drawPaint",
- canvas, 10,10,
- {1,0,0,1}, p3.get());
- }
- canvas->translate(0,80);
- // Draw a P3 red bitmap, using SkBitmap::eraseColor().
- {
- SkBitmap bm;
- bm.allocPixels(SkImageInfo::Make(60,60, kRGBA_F16_SkColorType, kPremul_SkAlphaType, p3));
- bm.eraseColor(0xffff0000/*in P3*/);
- canvas->drawBitmap(bm, 10,10);
- compare_pixel("drawBitmap P3 red, from SkBitmap::eraseColor()",
- canvas, 10,10,
- {1,0,0,1}, p3.get());
- }
- canvas->translate(0,80);
- // Draw a P3 red bitmap, using SkPixmap::erase().
- {
- SkBitmap bm;
- bm.allocPixels(SkImageInfo::Make(60,60, kRGBA_F16_SkColorType, kPremul_SkAlphaType, p3));
- // At the moment only SkPixmap has an erase() that takes an SkColor4f.
- SkPixmap pm;
- SkAssertResult(bm.peekPixels(&pm));
- SkAssertResult(pm.erase({1,0,0,1} /*in p3*/));
- canvas->drawBitmap(bm, 10,10);
- compare_pixel("drawBitmap P3 red, from SkPixmap::erase",
- canvas, 10,10,
- {1,0,0,1}, p3.get());
- }
- canvas->translate(0,80);
- // Draw a P3 red bitmap wrapped in a shader, using SkPixmap::erase().
- {
- SkBitmap bm;
- bm.allocPixels(SkImageInfo::Make(60,60, kRGBA_F16_SkColorType, kPremul_SkAlphaType, p3));
- // At the moment only SkPixmap has an erase() that takes an SkColor4f.
- SkPixmap pm;
- SkAssertResult(bm.peekPixels(&pm));
- SkAssertResult(pm.erase({1,0,0,1} /*in p3*/));
- SkPaint paint;
- paint.setShader(bm.makeShader(SkTileMode::kRepeat, SkTileMode::kRepeat));
- canvas->drawRect({10,10,70,70}, paint);
- compare_pixel("drawBitmapAsShader P3 red, from SkPixmap::erase",
- canvas, 10,10,
- {1,0,0,1}, p3.get());
- }
- canvas->translate(0,80);
- // TODO(mtklein): sample and check the middle points of these gradients too.
- // Draw a gradient from P3 red to P3 green interpolating in unpremul P3, checking the corners.
- {
- SkPoint points[] = {{10.5,10.5}, {69.5,69.5}};
- SkColor4f colors[] = {{1,0,0,1}, {0,1,0,1}};
- SkPaint paint;
- paint.setShader(SkGradientShader::MakeLinear(points, colors, p3,
- nullptr, SK_ARRAY_COUNT(colors),
- SkTileMode::kClamp));
- canvas->drawRect({10,10,70,70}, paint);
- canvas->save();
- compare_pixel("UPM P3 gradient, P3 red",
- canvas, 10,10,
- {1,0,0,1}, p3.get());
- canvas->translate(180, 0);
- compare_pixel("UPM P3 gradient, P3 green",
- canvas, 69,69,
- {0,1,0,1}, p3.get());
- canvas->restore();
- }
- canvas->translate(0,80);
- // Draw a gradient from P3 red to P3 green interpolating in premul P3, checking the corners.
- {
- SkPoint points[] = {{10.5,10.5}, {69.5,69.5}};
- SkColor4f colors[] = {{1,0,0,1}, {0,1,0,1}};
- SkPaint paint;
- paint.setShader(
- SkGradientShader::MakeLinear(points, colors, p3,
- nullptr, SK_ARRAY_COUNT(colors),
- SkTileMode::kClamp,
- SkGradientShader::kInterpolateColorsInPremul_Flag,
- nullptr/*local matrix*/));
- canvas->drawRect({10,10,70,70}, paint);
- canvas->save();
- compare_pixel("PM P3 gradient, P3 red",
- canvas, 10,10,
- {1,0,0,1}, p3.get());
- canvas->translate(180, 0);
- compare_pixel("PM P3 gradient, P3 green",
- canvas, 69,69,
- {0,1,0,1}, p3.get());
- canvas->restore();
- }
- canvas->translate(0,80);
- // Draw a gradient from P3 red to P3 green interpolating in unpremul sRGB, checking the corners.
- {
- SkPoint points[] = {{10.5,10.5}, {69.5,69.5}};
- SkColor4f colors[] = {p3_to_srgb({1,0,0,1}), p3_to_srgb({0,1,0,1})};
- SkPaint paint;
- paint.setShader(SkGradientShader::MakeLinear(points, colors, srgb,
- nullptr, SK_ARRAY_COUNT(colors),
- SkTileMode::kClamp));
- canvas->drawRect({10,10,70,70}, paint);
- canvas->save();
- compare_pixel("UPM sRGB gradient, P3 red",
- canvas, 10,10,
- {1,0,0,1}, p3.get());
- canvas->translate(180, 0);
- compare_pixel("UPM sRGB gradient, P3 green",
- canvas, 69,69,
- {0,1,0,1}, p3.get());
- canvas->restore();
- }
- canvas->translate(0,80);
- // Draw a gradient from P3 red to P3 green interpolating in premul sRGB, checking the corners.
- {
- SkPoint points[] = {{10.5,10.5}, {69.5,69.5}};
- SkColor4f colors[] = {p3_to_srgb({1,0,0,1}), p3_to_srgb({0,1,0,1})};
- SkPaint paint;
- paint.setShader(
- SkGradientShader::MakeLinear(points, colors, srgb,
- nullptr, SK_ARRAY_COUNT(colors),
- SkTileMode::kClamp,
- SkGradientShader::kInterpolateColorsInPremul_Flag,
- nullptr/*local matrix*/));
- canvas->drawRect({10,10,70,70}, paint);
- canvas->save();
- compare_pixel("PM sRGB gradient, P3 red",
- canvas, 10,10,
- {1,0,0,1}, p3.get());
- canvas->translate(180, 0);
- compare_pixel("PM sRGB gradient, P3 green",
- canvas, 69,69,
- {0,1,0,1}, p3.get());
- canvas->restore();
- }
- canvas->translate(0,80);
- // Leon's blue -> green -> red gradient, interpolating in premul.
- {
- SkPoint points[] = {{10.5,10.5}, {10.5,69.5}};
- SkColor4f colors[] = { {0,0,1,1}, {0,1,0,1}, {1,0,0,1} };
- SkPaint paint;
- paint.setShader(
- SkGradientShader::MakeLinear(points, colors, p3,
- nullptr, SK_ARRAY_COUNT(colors),
- SkTileMode::kClamp,
- SkGradientShader::kInterpolateColorsInPremul_Flag,
- nullptr/*local matrix*/));
- canvas->drawRect({10,10,70,70}, paint);
- canvas->save();
- compare_pixel("Leon's gradient, P3 blue",
- canvas, 10,10,
- {0,0,1,1}, p3.get());
- canvas->translate(180, 0);
- compare_pixel("Leon's gradient, P3 red",
- canvas, 10,69,
- {1,0,0,1}, p3.get());
- canvas->restore();
- }
- canvas->translate(0,80);
- // Draw an A8 image with a P3 red, scaled and not, as a shader or bitmap.
- {
- uint8_t mask[256];
- for (int i = 0; i < 256; i++) {
- mask[i] = 255-i;
- }
- SkBitmap bm;
- bm.installPixels(SkImageInfo::MakeA8(16,16), mask, 16);
- SkPaint as_bitmap;
- as_bitmap.setColor4f({1,0,0,1}, p3.get());
- as_bitmap.setFilterQuality(kLow_SkFilterQuality);
- SkPaint as_shader;
- as_shader.setColor4f({1,0,0,1}, p3.get());
- as_shader.setFilterQuality(kLow_SkFilterQuality);
- as_shader.setShader(bm.makeShader());
- canvas->drawBitmap(bm, 10,10, &as_bitmap);
- compare_pixel("A8 sprite bitmap P3 red",
- canvas, 10,10,
- {1,0,0,1}, p3.get());
- canvas->translate(0, 80);
- canvas->save();
- canvas->translate(10,10);
- canvas->drawRect({0,0,16,16}, as_shader);
- canvas->restore();
- compare_pixel("A8 sprite shader P3 red",
- canvas, 10,10,
- {1,0,0,1}, p3.get());
- canvas->translate(0,80);
- canvas->drawBitmapRect(bm, {10,10,70,70}, &as_bitmap);
- compare_pixel("A8 scaled bitmap P3 red",
- canvas, 10,10,
- {1,0,0,1}, p3.get());
- canvas->translate(0,80);
- canvas->save();
- canvas->translate(10,10);
- canvas->scale(3.75,3.75);
- canvas->drawRect({0,0,16,16}, as_shader);
- canvas->restore();
- compare_pixel("A8 scaled shader P3 red",
- canvas, 10,10,
- {1,0,0,1}, p3.get());
- }
- // TODO: draw P3 colors more ways
- }
- DEF_SIMPLE_GM(p3_ovals, canvas, 450, 320) {
- auto p3 = SkColorSpace::MakeRGB(SkNamedTransferFn::kSRGB, SkNamedGamut::kDCIP3);
- // Test cases that exercise each Op in GrOvalOpFactory.cpp
- // Draw a circle and check the center (CircleOp)
- {
- SkPaint paint;
- paint.setAntiAlias(true);
- paint.setColor4f({ 1,0,0,1 }, p3.get());
- canvas->drawCircle(40, 40, 30, paint);
- compare_pixel("drawCircle P3 red ",
- canvas, 40, 40,
- { 1,0,0,1 }, p3.get());
- }
- canvas->translate(0, 80);
- // Draw an oval and check the center (EllipseOp)
- {
- SkPaint paint;
- paint.setAntiAlias(true);
- paint.setColor4f({ 1,0,0,1 }, p3.get());
- canvas->drawOval({ 20,10,60,70 }, paint);
- compare_pixel("drawOval P3 red ",
- canvas, 40, 40,
- { 1,0,0,1 }, p3.get());
- }
- canvas->translate(0, 80);
- // Draw a butt-capped dashed circle and check the top of the stroke (ButtCappedDashedCircleOp)
- {
- SkPaint paint;
- paint.setAntiAlias(true);
- paint.setColor4f({ 1,0,0,1 }, p3.get());
- paint.setStyle(SkPaint::kStroke_Style);
- float intervals[] = { 70, 10 };
- paint.setPathEffect(SkDashPathEffect::Make(intervals, 2, 0));
- paint.setStrokeWidth(10);
- canvas->drawCircle(40, 40, 30, paint);
- compare_pixel("drawDashedCircle P3 red ",
- canvas, 40, 10,
- { 1,0,0,1 }, p3.get());
- }
- canvas->translate(0, 80);
- // Draw an oval with rotation and check the center (DIEllipseOp)
- {
- SkPaint paint;
- paint.setAntiAlias(true);
- paint.setColor4f({ 1,0,0,1 }, p3.get());
- canvas->save();
- canvas->translate(40, 40);
- canvas->rotate(45);
- canvas->drawOval({ -20,-30,20,30 }, paint);
- canvas->restore();
- compare_pixel("drawRotatedOval P3 red ",
- canvas, 40, 40,
- { 1,0,0,1 }, p3.get());
- }
- canvas->translate(0, 80);
- }
|