blob: d01bce2b30750c65faca85b6a4f25ae984e8874b [file] [log] [blame]
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +00001/*
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
Mike Kleinc0bd9f92019-04-23 12:05:21 -05008#include "gm/gm.h"
Ben Wagner7fde8e12019-05-01 17:28:53 -04009#include "include/core/SkBlendMode.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050010#include "include/core/SkCanvas.h"
Ben Wagner7fde8e12019-05-01 17:28:53 -040011#include "include/core/SkColor.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050012#include "include/core/SkColorFilter.h"
Ben Wagner7fde8e12019-05-01 17:28:53 -040013#include "include/core/SkMatrix.h"
14#include "include/core/SkPaint.h"
15#include "include/core/SkPoint.h"
16#include "include/core/SkRefCnt.h"
17#include "include/core/SkScalar.h"
18#include "include/core/SkShader.h"
19#include "include/core/SkSize.h"
20#include "include/core/SkString.h"
21#include "include/core/SkTileMode.h"
22#include "include/core/SkTypes.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050023#include "include/core/SkVertices.h"
24#include "include/effects/SkGradientShader.h"
Brian Osman3c358422020-03-23 10:44:12 -040025#include "include/effects/SkRuntimeEffect.h"
Ben Wagner7fde8e12019-05-01 17:28:53 -040026#include "include/private/SkTDArray.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050027#include "include/utils/SkRandom.h"
Brian Osman3c358422020-03-23 10:44:12 -040028#include "src/core/SkVerticesPriv.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050029#include "src/shaders/SkLocalMatrixShader.h"
Brian Osman3c358422020-03-23 10:44:12 -040030#include "src/utils/SkPatchUtils.h"
31#include "tools/Resources.h"
Brian Osmane41fa2d2020-03-20 15:45:49 -040032#include "tools/ToolUtils.h"
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +000033
Ben Wagner7fde8e12019-05-01 17:28:53 -040034#include <initializer_list>
35#include <utility>
36
Brian Salomon3f363692017-02-02 21:05:19 -050037static constexpr SkScalar kShaderSize = 40;
Florin Malitaf614ba22017-02-16 22:12:41 -050038static sk_sp<SkShader> make_shader1(SkScalar shaderScale) {
reed@google.com85e143c2013-12-30 15:51:25 +000039 const SkColor colors[] = {
40 SK_ColorRED, SK_ColorCYAN, SK_ColorGREEN, SK_ColorWHITE,
41 SK_ColorMAGENTA, SK_ColorBLUE, SK_ColorYELLOW,
42 };
Brian Salomon3f363692017-02-02 21:05:19 -050043 const SkPoint pts[] = {{kShaderSize / 4, 0}, {3 * kShaderSize / 4, kShaderSize}};
Mike Reed1f607332020-05-21 12:11:27 -040044 const SkMatrix localMatrix = SkMatrix::Scale(shaderScale, shaderScale);
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +000045
Florin Malitaf614ba22017-02-16 22:12:41 -050046 sk_sp<SkShader> grad = SkGradientShader::MakeLinear(pts, colors, nullptr,
47 SK_ARRAY_COUNT(colors),
Mike Reedfae8fce2019-04-03 10:27:45 -040048 SkTileMode::kMirror, 0,
Florin Malitaf614ba22017-02-16 22:12:41 -050049 &localMatrix);
50 // Throw in a couple of local matrix wrappers for good measure.
51 return shaderScale == 1
52 ? grad
53 : sk_make_sp<SkLocalMatrixShader>(
Mike Reed1f607332020-05-21 12:11:27 -040054 sk_make_sp<SkLocalMatrixShader>(std::move(grad), SkMatrix::Translate(-10, 0)),
55 SkMatrix::Translate(10, 0));
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +000056}
57
reed1a9b9642016-03-13 14:13:58 -070058static sk_sp<SkShader> make_shader2() {
Mike Reedc8bea7d2019-04-09 13:55:36 -040059 return SkShaders::Color(SK_ColorBLUE);
bsalomon093779c2016-03-08 11:53:31 -080060}
61
reedd053ce92016-03-22 10:17:23 -070062static sk_sp<SkColorFilter> make_color_filter() {
Mike Reedb286bc22019-04-08 16:23:20 -040063 return SkColorFilters::Blend(0xFFAABBCC, SkBlendMode::kDarken);
bsalomon093779c2016-03-08 11:53:31 -080064}
65
Brian Salomon3f363692017-02-02 21:05:19 -050066static constexpr SkScalar kMeshSize = 30;
67
68// start with the center of a 3x3 grid of vertices.
69static constexpr uint16_t kMeshFan[] = {
70 4,
71 0, 1, 2, 5, 8, 7, 6, 3, 0
72};
73
Brian Salomon199fb872017-02-06 09:41:10 -050074static const int kMeshIndexCnt = (int)SK_ARRAY_COUNT(kMeshFan);
Brian Salomon3f363692017-02-02 21:05:19 -050075static const int kMeshVertexCnt = 9;
76
77static void fill_mesh(SkPoint pts[kMeshVertexCnt], SkPoint texs[kMeshVertexCnt],
Florin Malitaf614ba22017-02-16 22:12:41 -050078 SkColor colors[kMeshVertexCnt], SkScalar shaderScale) {
Brian Salomon3f363692017-02-02 21:05:19 -050079 pts[0].set(0, 0);
80 pts[1].set(kMeshSize / 2, 3);
81 pts[2].set(kMeshSize, 0);
82 pts[3].set(3, kMeshSize / 2);
83 pts[4].set(kMeshSize / 2, kMeshSize / 2);
84 pts[5].set(kMeshSize - 3, kMeshSize / 2);
85 pts[6].set(0, kMeshSize);
86 pts[7].set(kMeshSize / 2, kMeshSize - 3);
87 pts[8].set(kMeshSize, kMeshSize);
88
Florin Malitaf614ba22017-02-16 22:12:41 -050089 const auto shaderSize = kShaderSize * shaderScale;
Brian Salomon3f363692017-02-02 21:05:19 -050090 texs[0].set(0, 0);
Florin Malitaf614ba22017-02-16 22:12:41 -050091 texs[1].set(shaderSize / 2, 0);
92 texs[2].set(shaderSize, 0);
93 texs[3].set(0, shaderSize / 2);
94 texs[4].set(shaderSize / 2, shaderSize / 2);
95 texs[5].set(shaderSize, shaderSize / 2);
96 texs[6].set(0, shaderSize);
97 texs[7].set(shaderSize / 2, shaderSize);
98 texs[8].set(shaderSize, shaderSize);
Brian Salomon3f363692017-02-02 21:05:19 -050099
100 SkRandom rand;
101 for (size_t i = 0; i < kMeshVertexCnt; ++i) {
102 colors[i] = rand.nextU() | 0xFF000000;
103 }
104}
105
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +0000106class VerticesGM : public skiagm::GM {
Brian Salomon3f363692017-02-02 21:05:19 -0500107 SkPoint fPts[kMeshVertexCnt];
108 SkPoint fTexs[kMeshVertexCnt];
109 SkColor fColors[kMeshVertexCnt];
reedd053ce92016-03-22 10:17:23 -0700110 sk_sp<SkShader> fShader1;
111 sk_sp<SkShader> fShader2;
112 sk_sp<SkColorFilter> fColorFilter;
Florin Malitaf614ba22017-02-16 22:12:41 -0500113 SkScalar fShaderScale;
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +0000114
115public:
Mike Reed887cdf12017-04-03 11:11:09 -0400116 VerticesGM(SkScalar shaderScale) : fShaderScale(shaderScale) {}
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +0000117
118protected:
commit-bot@chromium.orga90c6802014-04-30 13:20:45 +0000119
mtklein36352bf2015-03-25 18:17:31 -0700120 void onOnceBeforeDraw() override {
Florin Malitaf614ba22017-02-16 22:12:41 -0500121 fill_mesh(fPts, fTexs, fColors, fShaderScale);
122 fShader1 = make_shader1(fShaderScale);
reed1a9b9642016-03-13 14:13:58 -0700123 fShader2 = make_shader2();
reedd053ce92016-03-22 10:17:23 -0700124 fColorFilter = make_color_filter();
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +0000125 }
126
mtklein36352bf2015-03-25 18:17:31 -0700127 SkString onShortName() override {
reed@google.com60da8f32014-05-05 20:41:21 +0000128 SkString name("vertices");
Florin Malitaf614ba22017-02-16 22:12:41 -0500129 if (fShaderScale != 1) {
130 name.append("_scaled_shader");
131 }
reed@google.com60da8f32014-05-05 20:41:21 +0000132 return name;
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +0000133 }
134
mtklein36352bf2015-03-25 18:17:31 -0700135 SkISize onISize() override {
Brian Salomon32161802017-02-10 10:48:15 -0500136 return SkISize::Make(975, 1175);
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +0000137 }
138
mtklein36352bf2015-03-25 18:17:31 -0700139 void onDraw(SkCanvas* canvas) override {
Mike Reed7d954ad2016-10-28 15:42:34 -0400140 const SkBlendMode modes[] = {
141 SkBlendMode::kClear,
142 SkBlendMode::kSrc,
143 SkBlendMode::kDst,
144 SkBlendMode::kSrcOver,
145 SkBlendMode::kDstOver,
146 SkBlendMode::kSrcIn,
147 SkBlendMode::kDstIn,
148 SkBlendMode::kSrcOut,
149 SkBlendMode::kDstOut,
150 SkBlendMode::kSrcATop,
151 SkBlendMode::kDstATop,
152 SkBlendMode::kXor,
153 SkBlendMode::kPlus,
154 SkBlendMode::kModulate,
155 SkBlendMode::kScreen,
156 SkBlendMode::kOverlay,
157 SkBlendMode::kDarken,
158 SkBlendMode::kLighten,
159 SkBlendMode::kColorDodge,
160 SkBlendMode::kColorBurn,
161 SkBlendMode::kHardLight,
162 SkBlendMode::kSoftLight,
163 SkBlendMode::kDifference,
164 SkBlendMode::kExclusion,
165 SkBlendMode::kMultiply,
166 SkBlendMode::kHue,
167 SkBlendMode::kSaturation,
168 SkBlendMode::kColor,
169 SkBlendMode::kLuminosity,
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +0000170 };
skia.committer@gmail.com4c912862013-12-30 07:01:37 +0000171
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +0000172 SkPaint paint;
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +0000173
bsalomon093779c2016-03-08 11:53:31 -0800174 canvas->translate(4, 4);
175 int x = 0;
Brian Salomon199fb872017-02-06 09:41:10 -0500176 for (auto mode : modes) {
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +0000177 canvas->save();
Mike Reed9407e242019-02-15 16:13:57 -0500178 for (float alpha : {1.0f, 0.5f}) {
Brian Salomon199fb872017-02-06 09:41:10 -0500179 for (const auto& cf : {sk_sp<SkColorFilter>(nullptr), fColorFilter}) {
180 for (const auto& shader : {fShader1, fShader2}) {
181 static constexpr struct {
182 bool fHasColors;
183 bool fHasTexs;
184 } kAttrs[] = {{true, false}, {false, true}, {true, true}};
185 for (auto attrs : kAttrs) {
186 paint.setShader(shader);
187 paint.setColorFilter(cf);
Mike Reed9407e242019-02-15 16:13:57 -0500188 paint.setAlphaf(alpha);
Mike Reed5fa66452017-03-16 09:06:34 -0400189
190 const SkColor* colors = attrs.fHasColors ? fColors : nullptr;
191 const SkPoint* texs = attrs.fHasTexs ? fTexs : nullptr;
Mike Reed887cdf12017-04-03 11:11:09 -0400192 auto v = SkVertices::MakeCopy(SkVertices::kTriangleFan_VertexMode,
193 kMeshVertexCnt, fPts, texs, colors,
194 kMeshIndexCnt, kMeshFan);
195 canvas->drawVertices(v, mode, paint);
Brian Salomon199fb872017-02-06 09:41:10 -0500196 canvas->translate(40, 0);
197 ++x;
198 }
199 }
200 }
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +0000201 }
202 canvas->restore();
bsalomon093779c2016-03-08 11:53:31 -0800203 canvas->translate(0, 40);
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +0000204 }
205 }
206
mike@reedtribe.org0c87ea82013-12-30 04:07:34 +0000207private:
208 typedef skiagm::GM INHERITED;
209};
210
reed@google.com60da8f32014-05-05 20:41:21 +0000211/////////////////////////////////////////////////////////////////////////////////////
212
Mike Reed887cdf12017-04-03 11:11:09 -0400213DEF_GM(return new VerticesGM(1);)
214DEF_GM(return new VerticesGM(1 / kShaderSize);)
Brian Salomon3f363692017-02-02 21:05:19 -0500215
Mike Reed887cdf12017-04-03 11:11:09 -0400216static void draw_batching(SkCanvas* canvas) {
Mike Reed97eb4fe2017-03-14 12:04:16 -0400217 // Triangle fans can't batch so we convert to regular triangles,
218 static constexpr int kNumTris = kMeshIndexCnt - 2;
Mike Reed887cdf12017-04-03 11:11:09 -0400219 SkVertices::Builder builder(SkVertices::kTriangles_VertexMode, kMeshVertexCnt, 3 * kNumTris,
Mike Reedaa9e3322017-03-16 14:38:48 -0400220 SkVertices::kHasColors_BuilderFlag |
221 SkVertices::kHasTexCoords_BuilderFlag);
Mike Reed97eb4fe2017-03-14 12:04:16 -0400222
223 SkPoint* pts = builder.positions();
224 SkPoint* texs = builder.texCoords();
225 SkColor* colors = builder.colors();
226 fill_mesh(pts, texs, colors, 1);
Brian Salomon199fb872017-02-06 09:41:10 -0500227
Brian Salomon3f363692017-02-02 21:05:19 -0500228 SkTDArray<SkMatrix> matrices;
229 matrices.push()->reset();
230 matrices.push()->setTranslate(0, 40);
231 SkMatrix* m = matrices.push();
232 m->setRotate(45, kMeshSize / 2, kMeshSize / 2);
233 m->postScale(1.2f, .8f, kMeshSize / 2, kMeshSize / 2);
234 m->postTranslate(0, 80);
235
Florin Malitaf614ba22017-02-16 22:12:41 -0500236 auto shader = make_shader1(1);
Brian Salomon3f363692017-02-02 21:05:19 -0500237
Mike Reed97eb4fe2017-03-14 12:04:16 -0400238 uint16_t* indices = builder.indices();
Brian Salomon3f363692017-02-02 21:05:19 -0500239 for (size_t i = 0; i < kNumTris; ++i) {
240 indices[3 * i] = kMeshFan[0];
241 indices[3 * i + 1] = kMeshFan[i + 1];
242 indices[3 * i + 2] = kMeshFan[i + 2];
Mike Reed97eb4fe2017-03-14 12:04:16 -0400243
Brian Salomon3f363692017-02-02 21:05:19 -0500244 }
Brian Salomon199fb872017-02-06 09:41:10 -0500245
Brian Salomon199fb872017-02-06 09:41:10 -0500246 canvas->save();
Brian Salomon3f363692017-02-02 21:05:19 -0500247 canvas->translate(10, 10);
248 for (bool useShader : {false, true}) {
249 for (bool useTex : {false, true}) {
250 for (const auto& m : matrices) {
251 canvas->save();
252 canvas->concat(m);
253 SkPaint paint;
Brian Salomon3f363692017-02-02 21:05:19 -0500254 paint.setShader(useShader ? shader : nullptr);
Mike Reed5fa66452017-03-16 09:06:34 -0400255
256 const SkPoint* t = useTex ? texs : nullptr;
Mike Reed887cdf12017-04-03 11:11:09 -0400257 auto v = SkVertices::MakeCopy(SkVertices::kTriangles_VertexMode, kMeshVertexCnt,
258 pts, t, colors, kNumTris * 3, indices);
259 canvas->drawVertices(v, SkBlendMode::kModulate, paint);
Brian Salomon3f363692017-02-02 21:05:19 -0500260 canvas->restore();
261 }
262 canvas->translate(0, 120);
263 }
264 }
Brian Salomon199fb872017-02-06 09:41:10 -0500265 canvas->restore();
266}
267
268// This test exists to exercise batching in the gpu backend.
269DEF_SIMPLE_GM(vertices_batching, canvas, 100, 500) {
Mike Reed887cdf12017-04-03 11:11:09 -0400270 draw_batching(canvas);
Brian Salomon199fb872017-02-06 09:41:10 -0500271 canvas->translate(50, 0);
Mike Reed887cdf12017-04-03 11:11:09 -0400272 draw_batching(canvas);
Brian Salomon3f363692017-02-02 21:05:19 -0500273}
Mike Reedd32bdaf2020-03-06 12:52:52 -0500274
Brian Osmanffd11f4a2020-03-30 09:57:53 -0400275using AttrType = SkVertices::Attribute::Type;
Mike Reedd32bdaf2020-03-06 12:52:52 -0500276
Brian Osmanffd11f4a2020-03-30 09:57:53 -0400277DEF_SIMPLE_GM(vertices_data, canvas, 512, 256) {
278 for (auto attrType : {AttrType::kFloat4, AttrType::kByte4_unorm}) {
279 SkRect r = SkRect::MakeWH(256, 256);
280 int vcount = 4; // just a quad
281 int icount = 0;
282 SkVertices::Attribute attrs[] = { attrType };
283 SkVertices::Builder builder(SkVertices::kTriangleFan_VertexMode, vcount, icount, attrs, 1);
Mike Reedd32bdaf2020-03-06 12:52:52 -0500284
Brian Osmanffd11f4a2020-03-30 09:57:53 -0400285 r.toQuad(builder.positions());
Mike Reedd32bdaf2020-03-06 12:52:52 -0500286
Brian Osmanffd11f4a2020-03-30 09:57:53 -0400287 if (attrType == AttrType::kFloat4) {
288 SkV4* col = (SkV4*)builder.customData();
289 col[0] = {1, 0, 0, 1}; // red
290 col[1] = {0, 1, 0, 1}; // green
291 col[2] = {0, 0, 1, 1}; // blue
292 col[3] = {0.5, 0.5, 0.5, 1}; // gray
293 } else {
294 uint32_t* col = (uint32_t*)builder.customData();
295 col[0] = 0xFF0000FF;
296 col[1] = 0xFF00FF00;
297 col[2] = 0xFFFF0000;
298 col[3] = 0xFF7F7F7F;
Brian Osmanf11e3312020-03-24 14:57:38 -0400299 }
Brian Osmanffd11f4a2020-03-30 09:57:53 -0400300
301 SkPaint paint;
302 const char* gProg = R"(
303 varying float4 vtx_color;
304 void main(float2 p, inout half4 color) {
305 color = half4(vtx_color);
306 }
307 )";
308 auto[effect, errorText] = SkRuntimeEffect::Make(SkString(gProg));
309 paint.setShader(effect->makeShader(nullptr, nullptr, 0, nullptr, true));
310 canvas->drawVertices(builder.detach(), paint);
311 canvas->translate(r.width(), 0);
312 }
Mike Reedd32bdaf2020-03-06 12:52:52 -0500313}
Brian Osmane41fa2d2020-03-20 15:45:49 -0400314
315// Test case for skbug.com/10069. We need to draw the vertices twice (with different matrices) to
316// trigger the bug.
317DEF_SIMPLE_GM(vertices_perspective, canvas, 256, 256) {
318 SkPaint paint;
319 paint.setShader(ToolUtils::create_checkerboard_shader(SK_ColorBLACK, SK_ColorWHITE, 32));
320
321 SkRect r = SkRect::MakeWH(128, 128);
322
323 SkPoint pos[4];
324 r.toQuad(pos);
325 auto verts = SkVertices::MakeCopy(SkVertices::kTriangleFan_VertexMode, 4, pos, pos, nullptr);
326
327 SkMatrix persp;
328 persp.setPerspY(SK_Scalar1 / 100);
329
330 canvas->save();
331 canvas->concat(persp);
332 canvas->drawRect(r, paint);
333 canvas->restore();
334
335 canvas->save();
336 canvas->translate(r.width(), 0);
337 canvas->concat(persp);
338 canvas->drawRect(r, paint);
339 canvas->restore();
340
341 canvas->save();
342 canvas->translate(0, r.height());
343 canvas->concat(persp);
344 canvas->drawVertices(verts, paint);
345 canvas->restore();
346
347 canvas->save();
348 canvas->translate(r.width(), r.height());
349 canvas->concat(persp);
350 canvas->drawVertices(verts, paint);
351 canvas->restore();
352}
Brian Osman3c358422020-03-23 10:44:12 -0400353
354DEF_SIMPLE_GM(vertices_data_lerp, canvas, 256, 256) {
355 SkPoint pts[12] = {{0, 0}, {85, 0}, {171, 0}, {256, 0}, {256, 85}, {256, 171},
356 {256, 256}, {171, 256}, {85, 256}, {0, 256}, {0, 171}, {0, 85}};
357
358 auto patchVerts = SkPatchUtils::MakeVertices(pts, nullptr, nullptr, 12, 12);
Brian Osman8cbedf92020-03-31 10:38:31 -0400359 SkVerticesPriv pv(patchVerts->priv());
Brian Osman3c358422020-03-23 10:44:12 -0400360
Brian Osmanffd11f4a2020-03-30 09:57:53 -0400361 SkVertices::Attribute attrs[1] = { AttrType::kFloat };
362 SkVertices::Builder builder(pv.mode(), pv.vertexCount(), pv.indexCount(), attrs, 1);
Brian Osman3c358422020-03-23 10:44:12 -0400363
Brian Osman8cbedf92020-03-31 10:38:31 -0400364 memcpy(builder.positions(), pv.positions(), pv.vertexCount() * sizeof(SkPoint));
365 memcpy(builder.indices(), pv.indices(), pv.indexCount() * sizeof(uint16_t));
Brian Osman3c358422020-03-23 10:44:12 -0400366
367 SkRandom rnd;
Brian Osmanffd11f4a2020-03-30 09:57:53 -0400368 float* lerpData = (float*)builder.customData();
Brian Osman8cbedf92020-03-31 10:38:31 -0400369 for (int i = 0; i < pv.vertexCount(); ++i) {
Brian Osmanffd11f4a2020-03-30 09:57:53 -0400370 lerpData[i] = rnd.nextBool() ? 1.0f : 0.0f;
Brian Osman3c358422020-03-23 10:44:12 -0400371 }
372
373 auto verts = builder.detach();
374
375 SkPaint paint;
376 const char* gProg = R"(
Brian Osman28590d52020-03-23 16:59:08 -0400377 in shader c0;
378 in shader c1;
Brian Osman3c358422020-03-23 10:44:12 -0400379 varying float vtx_lerp;
380 void main(float2 p, inout half4 color) {
381 half4 col0 = sample(c0, p);
382 half4 col1 = sample(c1, p);
383 color = mix(col0, col1, half(vtx_lerp));
384 }
385 )";
386 auto [effect, errorText] = SkRuntimeEffect::Make(SkString(gProg));
Mike Reed1f607332020-05-21 12:11:27 -0400387 SkMatrix scale = SkMatrix::Scale(2, 2);
Brian Osman3c358422020-03-23 10:44:12 -0400388 sk_sp<SkShader> children[] = {
389 GetResourceAsImage("images/mandrill_256.png")->makeShader(),
390 GetResourceAsImage("images/color_wheel.png")->makeShader(scale),
391 };
392 paint.setShader(effect->makeShader(nullptr, children, 2, nullptr, false));
393
394 canvas->drawVertices(verts, paint);
395}
Brian Osman68219bf2020-04-07 16:04:24 -0400396
397static constexpr SkScalar kSin60 = 0.8660254f; // sqrt(3) / 2
398static constexpr SkPoint kHexVerts[] = {
399 { 0, 0 },
400 { 0, -1 },
401 { kSin60, -0.5f },
402 { kSin60, 0.5f },
403 { 0, 1 },
404 { -kSin60, 0.5f },
405 { -kSin60, -0.5f },
406 { 0, -1 },
407};
408
409static constexpr SkColor4f kColors[] = {
410 SkColors::kWhite,
411 SkColors::kRed,
412 SkColors::kYellow,
413 SkColors::kGreen,
414 SkColors::kCyan,
415 SkColors::kBlue,
416 SkColors::kMagenta,
417 SkColors::kRed,
418};
419
Brian Osmand1afef62020-04-09 16:24:23 -0400420using Attr = SkVertices::Attribute;
421
422DEF_SIMPLE_GM(vertices_custom_colors, canvas, 400, 200) {
Brian Osman68219bf2020-04-07 16:04:24 -0400423 ToolUtils::draw_checkerboard(canvas);
424
425 auto draw = [=](SkScalar cx, SkScalar cy, SkVertices::Builder& builder, const SkPaint& paint) {
426 memcpy(builder.positions(), kHexVerts, sizeof(kHexVerts));
427
428 canvas->save();
429 canvas->translate(cx, cy);
430 canvas->scale(45, 45);
431 canvas->drawVertices(builder.detach(), paint);
432 canvas->restore();
433 };
434
435 auto transColor = [](int i) {
436 return SkColor4f { kColors[i].fR, kColors[i].fG, kColors[i].fB, i % 2 ? 0.5f : 1.0f };
437 };
438
439 // Fixed function SkVertices, opaque
440 {
441 SkVertices::Builder builder(SkVertices::kTriangleFan_VertexMode, 8, 0,
442 SkVertices::kHasColors_BuilderFlag);
443 for (int i = 0; i < 8; ++i) {
444 builder.colors()[i] = kColors[i].toSkColor();
445 }
446 draw(50, 50, builder, SkPaint());
447 }
448
449 // Fixed function SkVertices, w/transparency
450 {
451 SkVertices::Builder builder(SkVertices::kTriangleFan_VertexMode, 8, 0,
452 SkVertices::kHasColors_BuilderFlag);
453 for (int i = 0; i < 8; ++i) {
454 builder.colors()[i] = transColor(i).toSkColor();
455 }
456 draw(50, 150, builder, SkPaint());
457 }
458
459 const char* gProg = R"(
460 varying half4 vtx_color;
461 void main(float2 p, inout half4 color) {
462 color = vtx_color;
463 }
464 )";
465 SkPaint skslPaint;
466 auto [effect, errorText] = SkRuntimeEffect::Make(SkString(gProg));
467 skslPaint.setShader(effect->makeShader(nullptr, nullptr, 0, nullptr, false));
468
Brian Osman68219bf2020-04-07 16:04:24 -0400469 Attr byteColorAttr(Attr::Type::kByte4_unorm, Attr::Usage::kColor);
470 Attr float4ColorAttr(Attr::Type::kFloat4, Attr::Usage::kColor);
471 Attr float3ColorAttr(Attr::Type::kFloat3, Attr::Usage::kColor);
472
473 // Custom vertices, byte colors, opaque
474 {
475 SkVertices::Builder builder(SkVertices::kTriangleFan_VertexMode, 8, 0, &byteColorAttr, 1);
476 for (int i = 0; i < 8; ++i) {
477 ((uint32_t*)builder.customData())[i] = kColors[i].toBytes_RGBA();
478 }
479 draw(150, 50, builder, skslPaint);
480 }
481
482 // Custom vertices, byte colors, w/transparency
483 {
484 SkVertices::Builder builder(SkVertices::kTriangleFan_VertexMode, 8, 0, &byteColorAttr, 1);
485 for (int i = 0; i < 8; ++i) {
486 ((uint32_t*)builder.customData())[i] = transColor(i).toBytes_RGBA();
487 }
488 draw(150, 150, builder, skslPaint);
489 }
490
491 // Custom vertices, float4 colors, opaque
492 {
493 SkVertices::Builder builder(SkVertices::kTriangleFan_VertexMode, 8, 0, &float4ColorAttr, 1);
494 for (int i = 0; i < 8; ++i) {
495 ((SkColor4f*)builder.customData())[i] = kColors[i];
496 }
497 draw(250, 50, builder, skslPaint);
498 }
499
500 // Custom vertices, float4 colors, w/transparency
501 {
502 SkVertices::Builder builder(SkVertices::kTriangleFan_VertexMode, 8, 0, &float4ColorAttr, 1);
503 SkColor4f* clr = (SkColor4f*)builder.customData();
504 for (int i = 0; i < 8; ++i) {
505 clr[i] = transColor(i);
506 }
507 draw(250, 150, builder, skslPaint);
508 }
509
510 // Custom vertices, float3 colors, opaque
511 {
512 SkVertices::Builder builder(SkVertices::kTriangleFan_VertexMode, 8, 0, &float3ColorAttr, 1);
513 for (int i = 0; i < 8; ++i) {
514 ((SkV3*)builder.customData())[i] = { kColors[i].fR, kColors[i].fG, kColors[i].fB };
515 }
516 draw(350, 50, builder, skslPaint);
517 }
Brian Osmand1afef62020-04-09 16:24:23 -0400518}
Brian Osman68219bf2020-04-07 16:04:24 -0400519
Brian Osman548de742020-04-24 12:02:25 -0400520static sk_sp<SkVertices> make_cone(Attr::Usage u, const char* markerName) {
521 Attr attr(Attr::Type::kFloat3, u, markerName);
Brian Osmand1afef62020-04-09 16:24:23 -0400522
523 constexpr int kPerimeterVerts = 64;
524 // +1 for the center, +1 to repeat the first perimeter point (so we draw a complete circle)
525 constexpr int kNumVerts = kPerimeterVerts + 2;
526
527 SkVertices::Builder builder(SkVertices::kTriangleFan_VertexMode, kNumVerts, /*index_count=*/ 0,
528 &attr, /*attr_count=*/ 1);
529
530 SkPoint* pos = builder.positions();
531 SkPoint3* vec = static_cast<SkPoint3*>(builder.customData());
532
533 pos[0] = { 0, 0 };
534 vec[0] = { 0, 0, 1 };
535
536 for (int i = 0; i < kPerimeterVerts + 1; ++i) {
537 SkScalar t = (i / SkIntToScalar(kPerimeterVerts)) * 2 * SK_ScalarPI;
538 SkScalar s = SkScalarSin(t),
539 c = SkScalarCos(t);
540 pos[i + 1] = { c, s };
541 vec[i + 1] = { c, s, 0 };
542 }
543
544 return builder.detach();
545}
546
Brian Osman2ad3dfb2020-05-15 15:47:25 -0400547DEF_SIMPLE_GM(vertices_custom_matrices, canvas, 400, 400) {
Brian Osmand1afef62020-04-09 16:24:23 -0400548 ToolUtils::draw_checkerboard(canvas);
549
Brian Osman548de742020-04-24 12:02:25 -0400550 const char* kViewSpace = "local_to_view";
551 const char* kWorldSpace = "local_to_world";
552 const char* kLocalSpace = "local_to_local";
Brian Osmand1afef62020-04-09 16:24:23 -0400553
554 auto draw = [=](SkScalar cx, SkScalar cy, sk_sp<SkVertices> vertices, const char* prog,
555 SkScalar squish = 1.0f) {
556 SkPaint paint;
557 auto [effect, errorText] = SkRuntimeEffect::Make(SkString(prog));
558 paint.setShader(effect->makeShader(nullptr, nullptr, 0, nullptr, false));
559
560 canvas->save();
561
562 // Device space: mesh is upright, translated to its "cell"
563 canvas->translate(cx, cy);
564
565 // View (camera) space: Mesh is upright, centered on origin, device scale
566 canvas->markCTM(kViewSpace);
567 canvas->rotate(90);
568
569 // World space: Mesh is sideways, centered on origin, device scale (possibly squished)
570 canvas->markCTM(kWorldSpace);
571 canvas->rotate(-90);
572 canvas->scale(45, 45 * squish);
573
574 // Local space: Mesh is upright, centered on origin, unit scale
575 canvas->markCTM(kLocalSpace);
576 canvas->drawVertices(vertices, paint);
577
578 canvas->restore();
579 };
580
581 const char* vectorProg = R"(
582 varying float3 vtx_vec;
583 void main(float2 p, inout half4 color) {
584 color.rgb = half3(vtx_vec) * 0.5 + 0.5;
585 })";
586
587 // raw, local vectors, normals, and positions should all look the same (no real transform)
Brian Osman548de742020-04-24 12:02:25 -0400588 draw(50, 50, make_cone(Attr::Usage::kRaw, nullptr), vectorProg);
Brian Osmand1afef62020-04-09 16:24:23 -0400589 draw(150, 50, make_cone(Attr::Usage::kVector, kLocalSpace), vectorProg);
590 draw(250, 50, make_cone(Attr::Usage::kNormalVector, kLocalSpace), vectorProg);
591 draw(350, 50, make_cone(Attr::Usage::kPosition, kLocalSpace), vectorProg);
592
593 // world-space vectors and normals are rotated 90 degrees, positions are centered but scaled up
594 draw(150, 150, make_cone(Attr::Usage::kVector, kWorldSpace), vectorProg);
595 draw(250, 150, make_cone(Attr::Usage::kNormalVector, kWorldSpace), vectorProg);
596 draw(350, 150, make_cone(Attr::Usage::kPosition, kWorldSpace), vectorProg);
597
598 // Squished vectors are "wrong", but normals are correct (because we use the inverse transpose)
599 // Positions remain scaled up (saturated), but otherwise correct
600 draw(150, 250, make_cone(Attr::Usage::kVector, kWorldSpace), vectorProg, 0.5f);
601 draw(250, 250, make_cone(Attr::Usage::kNormalVector, kWorldSpace), vectorProg, 0.5f);
602 draw(350, 250, make_cone(Attr::Usage::kPosition, kWorldSpace), vectorProg, 0.5f);
Brian Osman2ad3dfb2020-05-15 15:47:25 -0400603
604 draw( 50, 350, make_cone(Attr::Usage::kVector, nullptr), vectorProg, 0.5f);
605 draw(150, 350, make_cone(Attr::Usage::kNormalVector, nullptr), vectorProg, 0.5f);
606
607 // For canvas-space positions, color them according to their position relative to the center.
608 // We do this test twice, with and without saveLayer. That ensures that we get the canvas CTM,
609 // not just a local-to-device matrix, which exposes effect authors to an implementation detail.
610
611 const char* ctmPositionProg250 = R"(
612 varying float3 vtx_pos;
613 void main(float2 p, inout half4 color) {
614 color.rgb = (half3(vtx_pos) - half3(250, 350, 0)) / 50 + 0.5;
615 }
616 )";
617 draw(250, 350, make_cone(Attr::Usage::kPosition, nullptr), ctmPositionProg250, 0.5f);
618
619 const char* ctmPositionProg350 = R"(
620 varying float3 vtx_pos;
621 void main(float2 p, inout half4 color) {
622 color.rgb = (half3(vtx_pos) - half3(350, 350, 0)) / 50 + 0.5;
623 }
624 )";
625 canvas->saveLayer({ 300, 300, 400, 400 }, nullptr);
626 draw(350, 350, make_cone(Attr::Usage::kPosition, nullptr), ctmPositionProg350, 0.5f);
627 canvas->restore();
Brian Osman68219bf2020-04-07 16:04:24 -0400628}