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