1 /*
2 * Copyright 2013 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
10 #include "sk_tool_utils.h"
11 #include "DecodeFile.h"
12 #include "Resources.h"
13 #include "SampleCode.h"
14 #include "SkBlurMask.h"
15 #include "SkBlurDrawLooper.h"
16 #include "SkCanvas.h"
17 #include "SkColorPriv.h"
18 #include "SkOSFile.h"
19 #include "SkOSPath.h"
20 #include "SkStream.h"
21 #include "SkString.h"
22 #include "SkSystemEventTypes.h"
23 #include "SkTypes.h"
24 #include "SkUtils.h"
25 #include "SkView.h"
26
27 /**
28 * Interprets c as an unpremultiplied color, and returns the
29 * premultiplied equivalent.
30 */
premultiply_unpmcolor(SkPMColor c)31 static SkPMColor premultiply_unpmcolor(SkPMColor c) {
32 U8CPU a = SkGetPackedA32(c);
33 U8CPU r = SkGetPackedR32(c);
34 U8CPU g = SkGetPackedG32(c);
35 U8CPU b = SkGetPackedB32(c);
36 return SkPreMultiplyARGB(a, r, g, b);
37 }
38
39 class UnpremulView : public SampleView {
40 public:
UnpremulView(SkString res)41 UnpremulView(SkString res)
42 : fResPath(res)
43 , fPremul(true)
44 , fDecodeSucceeded(false) {
45 this->nextImage();
46 }
47
48 protected:
49 // overrides from SkEventSink
onQuery(SkEvent * evt)50 bool onQuery(SkEvent* evt) override {
51 if (SampleCode::TitleQ(*evt)) {
52 SampleCode::TitleR(evt, "unpremul");
53 return true;
54 }
55 SkUnichar uni;
56 if (SampleCode::CharQ(*evt, &uni)) {
57 char utf8[kMaxBytesInUTF8Sequence];
58 size_t size = SkUTF8_FromUnichar(uni, utf8);
59 // Only consider events for single char keys
60 if (1 == size) {
61 switch (utf8[0]) {
62 case fNextImageChar:
63 this->nextImage();
64 return true;
65 case fTogglePremulChar:
66 this->togglePremul();
67 return true;
68 default:
69 break;
70 }
71 }
72 }
73 return this->INHERITED::onQuery(evt);
74 }
75
onDrawBackground(SkCanvas * canvas)76 void onDrawBackground(SkCanvas* canvas) override {
77 sk_tool_utils::draw_checkerboard(canvas, 0xFFCCCCCC, 0xFFFFFFFF, 12);
78 }
79
onDrawContent(SkCanvas * canvas)80 void onDrawContent(SkCanvas* canvas) override {
81 SkPaint paint;
82 paint.setAntiAlias(true);
83 paint.setTextSize(SkIntToScalar(24));
84 auto looper(
85 SkBlurDrawLooper::Make(SK_ColorBLUE, SkBlurMask::ConvertRadiusToSigma(SkIntToScalar(2)),
86 0, 0));
87 paint.setLooper(looper);
88 SkScalar height = paint.getFontMetrics(nullptr);
89 if (!fDecodeSucceeded) {
90 SkString failure;
91 if (fResPath.size() == 0) {
92 failure.printf("resource path is required!");
93 } else {
94 failure.printf("Failed to decode %s", fCurrFile.c_str());
95 }
96 canvas->drawText(failure.c_str(), failure.size(), 0, height, paint);
97 return;
98 }
99
100 // Name, size of the file, and whether or not it is premultiplied.
101 SkString header(SkOSPath::Basename(fCurrFile.c_str()));
102 header.appendf(" [%dx%d] %s", fBitmap.width(), fBitmap.height(),
103 (fPremul ? "premultiplied" : "unpremultiplied"));
104 canvas->drawText(header.c_str(), header.size(), 0, height, paint);
105 canvas->translate(0, height);
106
107 // Help messages
108 header.printf("Press '%c' to move to the next image.'", fNextImageChar);
109 canvas->drawText(header.c_str(), header.size(), 0, height, paint);
110 canvas->translate(0, height);
111
112 header.printf("Press '%c' to toggle premultiplied decode.", fTogglePremulChar);
113 canvas->drawText(header.c_str(), header.size(), 0, height, paint);
114
115 // Now draw the image itself.
116 canvas->translate(height * 2, height * 2);
117 if (!fPremul) {
118 // A premultiplied bitmap cannot currently be drawn.
119 SkAutoLockPixels alp(fBitmap);
120 // Copy it to a bitmap which can be drawn, converting
121 // to premultiplied:
122 SkBitmap bm;
123 bm.allocN32Pixels(fBitmap.width(), fBitmap.height());
124 for (int i = 0; i < fBitmap.width(); ++i) {
125 for (int j = 0; j < fBitmap.height(); ++j) {
126 *bm.getAddr32(i, j) = premultiply_unpmcolor(*fBitmap.getAddr32(i, j));
127 }
128 }
129 canvas->drawBitmap(bm, 0, 0);
130 } else {
131 canvas->drawBitmap(fBitmap, 0, 0);
132 }
133 }
134
135 private:
136 const SkString fResPath;
137 SkString fCurrFile;
138 bool fPremul;
139 bool fDecodeSucceeded;
140 SkBitmap fBitmap;
141 SkOSFile::Iter fFileIter;
142
143 static const char fNextImageChar = 'j';
144 static const char fTogglePremulChar = 'h';
145
nextImage()146 void nextImage() {
147 if (fResPath.size() == 0) {
148 return;
149 }
150 SkString basename;
151 if (!fFileIter.next(&basename)) {
152 fFileIter.reset(fResPath.c_str());
153 if (!fFileIter.next(&basename)) {
154 // Perhaps this should draw some error message?
155 return;
156 }
157 }
158 fCurrFile = SkOSPath::Join(fResPath.c_str(), basename.c_str());
159 this->decodeCurrFile();
160 }
161
decodeCurrFile()162 void decodeCurrFile() {
163 if (fCurrFile.size() == 0) {
164 fDecodeSucceeded = false;
165 return;
166 }
167 fDecodeSucceeded = decode_file(fCurrFile.c_str(), &fBitmap, kN32_SkColorType, !fPremul);
168 this->inval(nullptr);
169 }
170
togglePremul()171 void togglePremul() {
172 fPremul = !fPremul;
173 this->decodeCurrFile();
174 }
175
176 typedef SampleView INHERITED;
177 };
178
179 //////////////////////////////////////////////////////////////////////////////
180
MyFactory()181 static SkView* MyFactory() {
182 return new UnpremulView(GetResourcePath());
183 }
184 static SkViewRegister reg(MyFactory);
185