senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2011 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 "SkColorMatrixFilter.h" |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 10 | #include "SkGradientShader.h" |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 11 | #include "SkImage.h" |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 12 | |
| 13 | #define WIDTH 500 |
| 14 | #define HEIGHT 500 |
| 15 | |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 16 | static void set_color_matrix(SkPaint* paint, const SkColorMatrix& matrix) { |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 17 | paint->setColorFilter(SkColorFilter::MakeMatrixFilterRowMajor255(matrix.fMat)); |
reed@google.com | 7b6c193 | 2012-06-04 19:07:41 +0000 | [diff] [blame] | 18 | } |
| 19 | |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 20 | static void set_array(SkPaint* paint, const SkScalar array[]) { |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 21 | paint->setColorFilter(SkColorFilter::MakeMatrixFilterRowMajor255(array)); |
reed@google.com | 7b6c193 | 2012-06-04 19:07:41 +0000 | [diff] [blame] | 22 | } |
| 23 | |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 24 | class ColorMatrixGM : public skiagm::GM { |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 25 | public: |
| 26 | ColorMatrixGM() { |
caryclark | 65cdba6 | 2015-06-15 06:51:08 -0700 | [diff] [blame] | 27 | this->setBGColor(sk_tool_utils::color_to_565(0xFF808080)); |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 28 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 29 | |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 30 | protected: |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 31 | SkString onShortName() override { |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 32 | return SkString("colormatrix"); |
| 33 | } |
| 34 | |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 35 | SkISize onISize() override { |
tfarina | f539318 | 2014-06-09 23:59:03 -0700 | [diff] [blame] | 36 | return SkISize::Make(WIDTH, HEIGHT); |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 37 | } |
halcanary | 9d524f2 | 2016-03-29 09:03:52 -0700 | [diff] [blame] | 38 | |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 39 | void onOnceBeforeDraw() override { |
reed | 9ce9d67 | 2016-03-17 10:51:11 -0700 | [diff] [blame] | 40 | fSolidImg = CreateSolidBitmap(64, 64); |
| 41 | fTransparentImg = CreateTransparentBitmap(64, 64); |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 42 | } |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 43 | |
reed | 9ce9d67 | 2016-03-17 10:51:11 -0700 | [diff] [blame] | 44 | static sk_sp<SkImage> CreateSolidBitmap(int width, int height) { |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 45 | SkBitmap bm; |
reed@google.com | eb9a46c | 2014-01-25 16:46:20 +0000 | [diff] [blame] | 46 | bm.allocN32Pixels(width, height); |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 47 | SkCanvas canvas(bm); |
senorblanco@chromium.org | 50bdad8 | 2012-01-03 20:51:57 +0000 | [diff] [blame] | 48 | canvas.clear(0x0); |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 49 | for (int y = 0; y < height; ++y) { |
| 50 | for (int x = 0; x < width; ++x) { |
| 51 | SkPaint paint; |
| 52 | paint.setColor(SkColorSetARGB(255, x * 255 / width, y * 255 / height, 0)); |
tomhudson@google.com | 7558925 | 2012-04-10 17:42:21 +0000 | [diff] [blame] | 53 | canvas.drawRect(SkRect::MakeXYWH(SkIntToScalar(x), |
| 54 | SkIntToScalar(y), SK_Scalar1, SK_Scalar1), paint); |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 55 | } |
| 56 | } |
reed | 9ce9d67 | 2016-03-17 10:51:11 -0700 | [diff] [blame] | 57 | return SkImage::MakeFromBitmap(bm); |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 58 | } |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 59 | |
| 60 | // creates a bitmap with shades of transparent gray. |
reed | 9ce9d67 | 2016-03-17 10:51:11 -0700 | [diff] [blame] | 61 | static sk_sp<SkImage> CreateTransparentBitmap(int width, int height) { |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 62 | SkBitmap bm; |
reed@google.com | eb9a46c | 2014-01-25 16:46:20 +0000 | [diff] [blame] | 63 | bm.allocN32Pixels(width, height); |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 64 | SkCanvas canvas(bm); |
| 65 | canvas.clear(0x0); |
| 66 | |
| 67 | SkPoint pts[] = {{0, 0}, {SkIntToScalar(width), SkIntToScalar(height)}}; |
| 68 | SkColor colors[] = {0x00000000, 0xFFFFFFFF}; |
| 69 | SkPaint paint; |
reed | 2ad1aa6 | 2016-03-09 09:50:50 -0800 | [diff] [blame] | 70 | paint.setShader(SkGradientShader::MakeLinear(pts, colors, nullptr, 2, |
| 71 | SkShader::kClamp_TileMode)); |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 72 | canvas.drawRect(SkRect::MakeWH(SkIntToScalar(width), SkIntToScalar(height)), paint); |
reed | 9ce9d67 | 2016-03-17 10:51:11 -0700 | [diff] [blame] | 73 | return SkImage::MakeFromBitmap(bm); |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 74 | } |
| 75 | |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 76 | void onDraw(SkCanvas* canvas) override { |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 77 | SkPaint paint; |
| 78 | SkColorMatrix matrix; |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 79 | |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 80 | paint.setXfermodeMode(SkXfermode::kSrc_Mode); |
reed | 9ce9d67 | 2016-03-17 10:51:11 -0700 | [diff] [blame] | 81 | const SkImage* bmps[] = { fSolidImg.get(), fTransparentImg.get() }; |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 82 | |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 83 | for (size_t i = 0; i < SK_ARRAY_COUNT(bmps); ++i) { |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 84 | matrix.setIdentity(); |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 85 | set_color_matrix(&paint, matrix); |
| 86 | canvas->drawImage(bmps[i], 0, 0, &paint); |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 87 | |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 88 | matrix.setRotate(SkColorMatrix::kR_Axis, 90); |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 89 | set_color_matrix(&paint, matrix); |
| 90 | canvas->drawImage(bmps[i], 80, 0, &paint); |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 91 | |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 92 | matrix.setRotate(SkColorMatrix::kG_Axis, 90); |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 93 | set_color_matrix(&paint, matrix); |
| 94 | canvas->drawImage(bmps[i], 160, 0, &paint); |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 95 | |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 96 | matrix.setRotate(SkColorMatrix::kB_Axis, 90); |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 97 | set_color_matrix(&paint, matrix); |
| 98 | canvas->drawImage(bmps[i], 240, 0, &paint); |
| 99 | /////////////////////////////////////////////// |
commit-bot@chromium.org | 4b413c8 | 2013-11-25 19:44:07 +0000 | [diff] [blame] | 100 | matrix.setSaturation(0.0f); |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 101 | set_color_matrix(&paint, matrix); |
| 102 | canvas->drawImage(bmps[i], 0, 80, &paint); |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 103 | |
commit-bot@chromium.org | 4b413c8 | 2013-11-25 19:44:07 +0000 | [diff] [blame] | 104 | matrix.setSaturation(0.5f); |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 105 | set_color_matrix(&paint, matrix); |
| 106 | canvas->drawImage(bmps[i], 80, 80, &paint); |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 107 | |
commit-bot@chromium.org | 4b413c8 | 2013-11-25 19:44:07 +0000 | [diff] [blame] | 108 | matrix.setSaturation(1.0f); |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 109 | set_color_matrix(&paint, matrix); |
| 110 | canvas->drawImage(bmps[i], 160, 80, &paint); |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 111 | |
commit-bot@chromium.org | 4b413c8 | 2013-11-25 19:44:07 +0000 | [diff] [blame] | 112 | matrix.setSaturation(2.0f); |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 113 | set_color_matrix(&paint, matrix); |
| 114 | canvas->drawImage(bmps[i], 240, 80, &paint); |
| 115 | /////////////////////////////////////////////// |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 116 | matrix.setRGB2YUV(); |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 117 | set_color_matrix(&paint, matrix); |
| 118 | canvas->drawImage(bmps[i], 0, 160, &paint); |
senorblanco@chromium.org | b3a39b5 | 2012-01-05 18:28:56 +0000 | [diff] [blame] | 119 | |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 120 | matrix.setYUV2RGB(); |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 121 | set_color_matrix(&paint, matrix); |
| 122 | canvas->drawImage(bmps[i], 80, 160, &paint); |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 123 | |
| 124 | SkScalar s1 = SK_Scalar1; |
| 125 | SkScalar s255 = SkIntToScalar(255); |
| 126 | // Move red into alpha, set color to white |
| 127 | SkScalar data[20] = { |
| 128 | 0, 0, 0, 0, s255, |
| 129 | 0, 0, 0, 0, s255, |
| 130 | 0, 0, 0, 0, s255, |
| 131 | s1, 0, 0, 0, 0, |
| 132 | }; |
| 133 | |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 134 | set_array(&paint, data); |
| 135 | canvas->drawImage(bmps[i], 160, 160, &paint); |
| 136 | /////////////////////////////////////////////// |
bsalomon@google.com | 1282009 | 2012-10-16 17:50:48 +0000 | [diff] [blame] | 137 | canvas->translate(0, 240); |
| 138 | } |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 139 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame] | 140 | |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 141 | private: |
reed | 9ce9d67 | 2016-03-17 10:51:11 -0700 | [diff] [blame] | 142 | sk_sp<SkImage> fSolidImg; |
| 143 | sk_sp<SkImage> fTransparentImg; |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 144 | |
| 145 | typedef skiagm::GM INHERITED; |
senorblanco@chromium.org | e5ff3ce | 2011-12-20 20:58:18 +0000 | [diff] [blame] | 146 | }; |
reed | 5617900 | 2015-07-07 06:11:19 -0700 | [diff] [blame] | 147 | DEF_GM( return new ColorMatrixGM; ) |