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/gm.h"
9 #include "gm/verifiers/gmverifier.h"
10 #include "include/core/SkBitmap.h"
11 #include "include/core/SkBlendMode.h"
12 #include "include/core/SkCanvas.h"
13 #include "include/core/SkFilterQuality.h"
14 #include "include/core/SkFont.h"
15 #include "include/core/SkFontTypes.h"
16 #include "include/core/SkMatrix.h"
17 #include "include/core/SkPaint.h"
18 #include "include/core/SkRect.h"
19 #include "include/core/SkShader.h"
20 #include "include/core/SkTileMode.h"
21 #include "include/core/SkTypeface.h"
22 #include "include/gpu/GrContext.h"
23 #include "src/core/SkTraceEvent.h"
24 #include "tools/ToolUtils.h"
25 
26 #include <stdarg.h>
27 
28 class GrRenderTargetContext;
29 
30 using namespace skiagm;
31 
32 constexpr char GM::kErrorMsg_DrawSkippedGpuOnly[];
33 
draw_failure_message(SkCanvas * canvas,const char format[],...)34 static void draw_failure_message(SkCanvas* canvas, const char format[], ...)  {
35     SkString failureMsg;
36 
37     va_list argp;
38     va_start(argp, format);
39     failureMsg.appendVAList(format, argp);
40     va_end(argp);
41 
42     constexpr SkScalar kOffset = 5.0f;
43     canvas->drawColor(SkColorSetRGB(200,0,0));
44     SkFont font;
45     SkRect bounds;
46     font.measureText(failureMsg.c_str(), failureMsg.size(), SkTextEncoding::kUTF8, &bounds);
47     SkPaint textPaint(SkColors::kWhite);
48     canvas->drawString(failureMsg, kOffset, bounds.height() + kOffset, font, textPaint);
49 }
50 
draw_gpu_only_message(SkCanvas * canvas)51 static void draw_gpu_only_message(SkCanvas* canvas) {
52     SkBitmap bmp;
53     bmp.allocN32Pixels(128, 64);
54     SkCanvas bmpCanvas(bmp);
55     bmpCanvas.drawColor(SK_ColorWHITE);
56     SkFont  font(ToolUtils::create_portable_typeface(), 20);
57     SkPaint paint(SkColors::kRed);
58     bmpCanvas.drawString("GPU Only", 20, 40, font, paint);
59     SkMatrix localM;
60     localM.setRotate(35.f);
61     localM.postTranslate(10.f, 0.f);
62     paint.setShader(bmp.makeShader(SkTileMode::kMirror, SkTileMode::kMirror, &localM));
63     paint.setFilterQuality(kMedium_SkFilterQuality);
64     canvas->drawPaint(paint);
65 }
66 
GM(SkColor bgColor)67 GM::GM(SkColor bgColor) {
68     fMode = kGM_Mode;
69     fBGColor = bgColor;
70     fHaveCalledOnceBeforeDraw = false;
71 }
72 
~GM()73 GM::~GM() {}
74 
draw(SkCanvas * canvas,SkString * errorMsg)75 DrawResult GM::draw(SkCanvas* canvas, SkString* errorMsg) {
76     TRACE_EVENT1("GM", TRACE_FUNC, "name", TRACE_STR_COPY(this->getName()));
77     this->drawBackground(canvas);
78     return this->drawContent(canvas, errorMsg);
79 }
80 
drawContent(SkCanvas * canvas,SkString * errorMsg)81 DrawResult GM::drawContent(SkCanvas* canvas, SkString* errorMsg) {
82     TRACE_EVENT0("GM", TRACE_FUNC);
83     if (!fHaveCalledOnceBeforeDraw) {
84         fHaveCalledOnceBeforeDraw = true;
85         this->onOnceBeforeDraw();
86     }
87     SkAutoCanvasRestore acr(canvas, true);
88     DrawResult drawResult = this->onDraw(canvas, errorMsg);
89     if (DrawResult::kOk != drawResult) {
90         if (DrawResult::kFail == drawResult) {
91             draw_failure_message(canvas, "DRAW FAILED: %s", errorMsg->c_str());
92         } else if (SkString(kErrorMsg_DrawSkippedGpuOnly) == *errorMsg) {
93             draw_gpu_only_message(canvas);
94         } else {
95             draw_failure_message(canvas, "DRAW SKIPPED: %s", errorMsg->c_str());
96         }
97     }
98     return drawResult;
99 }
100 
drawBackground(SkCanvas * canvas)101 void GM::drawBackground(SkCanvas* canvas) {
102     TRACE_EVENT0("GM", TRACE_FUNC);
103     if (!fHaveCalledOnceBeforeDraw) {
104         fHaveCalledOnceBeforeDraw = true;
105         this->onOnceBeforeDraw();
106     }
107     SkAutoCanvasRestore acr(canvas, true);
108     canvas->drawColor(fBGColor, SkBlendMode::kSrc);
109 }
110 
onDraw(SkCanvas * canvas,SkString * errorMsg)111 DrawResult GM::onDraw(SkCanvas* canvas, SkString* errorMsg) {
112     this->onDraw(canvas);
113     return DrawResult::kOk;
114 }
onDraw(SkCanvas *)115 void GM::onDraw(SkCanvas*) { SK_ABORT("Not implemented."); }
116 
117 
onISize()118 SkISize SimpleGM::onISize() { return fSize; }
onShortName()119 SkString SimpleGM::onShortName() { return fName; }
onDraw(SkCanvas * canvas,SkString * errorMsg)120 DrawResult SimpleGM::onDraw(SkCanvas* canvas, SkString* errorMsg) {
121     return fDrawProc(canvas, errorMsg);
122 }
123 
onISize()124 SkISize SimpleGpuGM::onISize() { return fSize; }
onShortName()125 SkString SimpleGpuGM::onShortName() { return fName; }
onDraw(GrContext * ctx,GrRenderTargetContext * rtc,SkCanvas * canvas,SkString * errorMsg)126 DrawResult SimpleGpuGM::onDraw(GrContext* ctx, GrRenderTargetContext* rtc, SkCanvas* canvas,
127                                SkString* errorMsg) {
128     return fDrawProc(ctx, rtc, canvas, errorMsg);
129 }
130 
getName()131 const char* GM::getName() {
132     if (fShortName.size() == 0) {
133         fShortName = this->onShortName();
134     }
135     return fShortName.c_str();
136 }
137 
setBGColor(SkColor color)138 void GM::setBGColor(SkColor color) {
139     fBGColor = color;
140 }
141 
animate(double nanos)142 bool GM::animate(double nanos) { return this->onAnimate(nanos); }
143 
runAsBench() const144 bool GM::runAsBench() const { return false; }
modifyGrContextOptions(GrContextOptions * options)145 void GM::modifyGrContextOptions(GrContextOptions* options) {}
146 
getVerifiers() const147 std::unique_ptr<verifiers::VerifierList> GM::getVerifiers() const {
148     // No verifiers by default.
149     return nullptr;
150 }
151 
onOnceBeforeDraw()152 void GM::onOnceBeforeDraw() {}
153 
onAnimate(double)154 bool GM::onAnimate(double /*nanos*/) { return false; }
155 
onChar(SkUnichar uni)156 bool GM::onChar(SkUnichar uni) { return false; }
157 
onGetControls(SkMetaData *)158 bool GM::onGetControls(SkMetaData*) { return false; }
159 
onSetControls(const SkMetaData &)160 void GM::onSetControls(const SkMetaData&) {}
161 
162 /////////////////////////////////////////////////////////////////////////////////////////////
163 
drawSizeBounds(SkCanvas * canvas,SkColor color)164 void GM::drawSizeBounds(SkCanvas* canvas, SkColor color) {
165     canvas->drawRect(SkRect::Make(this->getISize()), SkPaint(SkColor4f::FromColor(color)));
166 }
167 
168 // need to explicitly declare this, or we get some weird infinite loop llist
169 template GMRegistry* GMRegistry::gHead;
170 
onDraw(GrContext * ctx,GrRenderTargetContext * rtc,SkCanvas * canvas,SkString * errorMsg)171 DrawResult GpuGM::onDraw(GrContext* ctx, GrRenderTargetContext* rtc, SkCanvas* canvas,
172                           SkString* errorMsg) {
173     this->onDraw(ctx, rtc, canvas);
174     return DrawResult::kOk;
175 }
onDraw(GrContext *,GrRenderTargetContext *,SkCanvas *)176 void GpuGM::onDraw(GrContext*, GrRenderTargetContext*, SkCanvas*) {
177     SK_ABORT("Not implemented.");
178 }
179 
onDraw(SkCanvas * canvas,SkString * errorMsg)180 DrawResult GpuGM::onDraw(SkCanvas* canvas, SkString* errorMsg) {
181     GrContext* ctx = canvas->getGrContext();
182     GrRenderTargetContext* rtc = canvas->internal_private_accessTopLayerRenderTargetContext();
183     if (!ctx || !rtc) {
184         *errorMsg = kErrorMsg_DrawSkippedGpuOnly;
185         return DrawResult::kSkip;
186     }
187     if (ctx->abandoned()) {
188         *errorMsg = "GrContext abandoned.";
189         return DrawResult::kSkip;
190     }
191     return this->onDraw(ctx, rtc, canvas, errorMsg);
192 }
193 
194 template <typename Fn>
mark(SkCanvas * canvas,SkScalar x,SkScalar y,Fn && fn)195 static void mark(SkCanvas* canvas, SkScalar x, SkScalar y, Fn&& fn) {
196     SkPaint alpha;
197     alpha.setAlpha(0x50);
198     canvas->saveLayer(nullptr, &alpha);
199         canvas->translate(x,y);
200         canvas->scale(2,2);
201         fn();
202     canvas->restore();
203 }
204 
MarkGMGood(SkCanvas * canvas,SkScalar x,SkScalar y)205 void MarkGMGood(SkCanvas* canvas, SkScalar x, SkScalar y) {
206     mark(canvas, x,y, [&]{
207         // A green circle.
208         canvas->drawCircle(0, 0, 12, SkPaint(SkColor4f::FromColor(SkColorSetRGB(27, 158, 119))));
209 
210         // Cut out a check mark.
211         SkPaint paint(SkColors::kTransparent);
212         paint.setBlendMode(SkBlendMode::kSrc);
213         paint.setStrokeWidth(2);
214         paint.setStyle(SkPaint::kStroke_Style);
215         canvas->drawLine(-6, 0,
216                          -1, 5, paint);
217         canvas->drawLine(-1, +5,
218                          +7, -5, paint);
219     });
220 }
221 
MarkGMBad(SkCanvas * canvas,SkScalar x,SkScalar y)222 void MarkGMBad(SkCanvas* canvas, SkScalar x, SkScalar y) {
223     mark(canvas, x,y, [&] {
224         // A red circle.
225         canvas->drawCircle(0,0, 12, SkPaint(SkColor4f::FromColor(SkColorSetRGB(231, 41, 138))));
226 
227         // Cut out an 'X'.
228         SkPaint paint(SkColors::kTransparent);
229         paint.setBlendMode(SkBlendMode::kSrc);
230         paint.setStrokeWidth(2);
231         paint.setStyle(SkPaint::kStroke_Style);
232         canvas->drawLine(-5,-5,
233                          +5,+5, paint);
234         canvas->drawLine(+5,-5,
235                          -5,+5, paint);
236     });
237 }
238