sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2014 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 "SkColorCubeFilter.h" |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 10 | #include "SkData.h" |
| 11 | #include "SkGradientShader.h" |
scroggo | 565901d | 2015-12-10 10:44:13 -0800 | [diff] [blame] | 12 | #include "SkTemplates.h" |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 13 | |
| 14 | namespace skiagm { |
| 15 | |
reed | 2ad1aa6 | 2016-03-09 09:50:50 -0800 | [diff] [blame] | 16 | static sk_sp<SkShader> MakeLinear() { |
mtklein | dbfd7ab | 2016-09-01 11:24:54 -0700 | [diff] [blame] | 17 | constexpr SkPoint pts[2] = { |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 18 | { 0, 0 }, |
| 19 | { SkIntToScalar(80), SkIntToScalar(80) } |
| 20 | }; |
mtklein | dbfd7ab | 2016-09-01 11:24:54 -0700 | [diff] [blame] | 21 | constexpr SkColor colors[] = { SK_ColorYELLOW, SK_ColorBLUE }; |
reed | 2ad1aa6 | 2016-03-09 09:50:50 -0800 | [diff] [blame] | 22 | return SkGradientShader::MakeLinear(pts, colors, nullptr, 2, SkShader::kRepeat_TileMode, 0, |
| 23 | &SkMatrix::I()); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 24 | } |
| 25 | |
| 26 | class ColorCubeGM : public GM { |
| 27 | public: |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 28 | ColorCubeGM() : fInitialized(false) { |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 29 | this->setBGColor(0xFF000000); |
| 30 | } |
| 31 | |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 32 | protected: |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 33 | SkString onShortName() override { |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 34 | return SkString("colorcube"); |
| 35 | } |
| 36 | |
| 37 | void make_3Dluts() { |
| 38 | make_3Dlut(&f3DLut4, 4, true, false, false); |
| 39 | make_3Dlut(&f3DLut8, 8, false, true, false); |
| 40 | make_3Dlut(&f3DLut16, 16, false, true, true); |
| 41 | make_3Dlut(&f3DLut32, 32, true, true, false); |
| 42 | make_3Dlut(&f3DLut64, 64, true, false, true); |
| 43 | } |
| 44 | |
| 45 | void make_bitmap() { |
| 46 | fBitmap.allocN32Pixels(80, 80); |
| 47 | SkCanvas canvas(fBitmap); |
| 48 | canvas.clear(0x00000000); |
| 49 | SkPaint paint; |
| 50 | paint.setAntiAlias(true); |
reed | 2ad1aa6 | 2016-03-09 09:50:50 -0800 | [diff] [blame] | 51 | paint.setShader(MakeLinear()); |
| 52 | canvas.drawRect(SkRect::MakeWH(80, 80), paint); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 53 | } |
| 54 | |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 55 | void make_3Dlut(sk_sp<SkData>* data, int size, bool invR, bool invG, bool invB) { |
| 56 | *data = SkData::MakeUninitialized(sizeof(SkColor) * size * size * size); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 57 | SkColor* pixels = (SkColor*)((*data)->writable_data()); |
scroggo | 565901d | 2015-12-10 10:44:13 -0800 | [diff] [blame] | 58 | SkAutoTMalloc<uint8_t> lutMemory(size); |
| 59 | SkAutoTMalloc<uint8_t> invLutMemory(size); |
| 60 | uint8_t* lut = lutMemory.get(); |
| 61 | uint8_t* invLut = invLutMemory.get(); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 62 | const int maxIndex = size - 1; |
| 63 | for (int i = 0; i < size; i++) { |
| 64 | lut[i] = (i * 255) / maxIndex; |
| 65 | invLut[i] = ((maxIndex - i) * 255) / maxIndex; |
| 66 | } |
| 67 | for (int r = 0; r < size; ++r) { |
| 68 | for (int g = 0; g < size; ++g) { |
| 69 | for (int b = 0; b < size; ++b) { |
caryclark | 1259601 | 2015-07-29 05:27:47 -0700 | [diff] [blame] | 70 | pixels[(size * ((size * b) + g)) + r] = sk_tool_utils::color_to_565( |
| 71 | SkColorSetARGB(0xFF, |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 72 | invR ? invLut[r] : lut[r], |
| 73 | invG ? invLut[g] : lut[g], |
caryclark | 1259601 | 2015-07-29 05:27:47 -0700 | [diff] [blame] | 74 | invB ? invLut[b] : lut[b])); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 75 | } |
| 76 | } |
| 77 | } |
| 78 | } |
| 79 | |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 80 | SkISize onISize() override { |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 81 | return SkISize::Make(500, 100); |
| 82 | } |
| 83 | |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 84 | void onDraw(SkCanvas* canvas) override { |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 85 | if (!fInitialized) { |
| 86 | this->make_bitmap(); |
| 87 | this->make_3Dluts(); |
| 88 | fInitialized = true; |
| 89 | } |
| 90 | canvas->clear(0x00000000); |
| 91 | SkPaint paint; |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 92 | paint.setColorFilter(SkColorCubeFilter::Make(f3DLut4, 4)); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 93 | canvas->drawBitmap(fBitmap, 10, 10, &paint); |
| 94 | |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 95 | paint.setColorFilter(SkColorCubeFilter::Make(f3DLut8, 8)); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 96 | canvas->drawBitmap(fBitmap, 110, 10, &paint); |
| 97 | |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 98 | paint.setColorFilter(SkColorCubeFilter::Make(f3DLut16, 16)); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 99 | canvas->drawBitmap(fBitmap, 210, 10, &paint); |
| 100 | |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 101 | paint.setColorFilter(SkColorCubeFilter::Make(f3DLut32, 32)); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 102 | canvas->drawBitmap(fBitmap, 310, 10, &paint); |
| 103 | |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 104 | paint.setColorFilter(SkColorCubeFilter::Make(f3DLut64, 64)); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 105 | canvas->drawBitmap(fBitmap, 410, 10, &paint); |
| 106 | } |
| 107 | |
| 108 | private: |
| 109 | typedef GM INHERITED; |
| 110 | bool fInitialized; |
| 111 | SkBitmap fBitmap; |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 112 | sk_sp<SkData> f3DLut4; |
| 113 | sk_sp<SkData> f3DLut8; |
| 114 | sk_sp<SkData> f3DLut16; |
| 115 | sk_sp<SkData> f3DLut32; |
| 116 | sk_sp<SkData> f3DLut64; |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 117 | }; |
| 118 | |
| 119 | ////////////////////////////////////////////////////////////////////////////// |
| 120 | |
| 121 | static GM* MyFactory(void*) { return new ColorCubeGM; } |
| 122 | static GMRegistry reg(MyFactory); |
| 123 | |
| 124 | } |