blob: fa19eac9f274e89c6eac3409e78e854c54baacc5 [file] [log] [blame]
/*
* Copyright 2021 Google LLC
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include "include/core/SkCanvas.h"
#include "include/core/SkMesh.h"
#include "src/core/SkZip.h"
#include "tests/Test.h"
using Attribute = SkMeshSpecification::Attribute;
using Varying = SkMeshSpecification::Varying;
static const char* attr_type_str(const Attribute::Type type) {
switch (type) {
case Attribute::Type::kFloat: return "float";
case Attribute::Type::kFloat2: return "float2";
case Attribute::Type::kFloat3: return "float3";
case Attribute::Type::kFloat4: return "float4";
case Attribute::Type::kUByte4_unorm: return "ubyte4_unorm";
}
SkUNREACHABLE;
}
static const char* var_type_str(const Varying::Type type) {
switch (type) {
case Varying::Type::kFloat: return "float";
case Varying::Type::kFloat2: return "float2";
case Varying::Type::kFloat3: return "float3";
case Varying::Type::kFloat4: return "float4";
case Varying::Type::kHalf: return "half";
case Varying::Type::kHalf2: return "half2";
case Varying::Type::kHalf3: return "half3";
case Varying::Type::kHalf4: return "half4";
}
SkUNREACHABLE;
}
static SkString make_description(SkSpan<const Attribute> attributes,
size_t stride,
SkSpan<const Varying> varyings,
const SkString& vs,
const SkString& fs) {
static constexpr size_t kMax = 10;
SkString result;
result.appendf("Attributes (count=%zu, stride=%zu):\n", attributes.size(), stride);
for (size_t i = 0; i < std::min(kMax, attributes.size()); ++i) {
const auto& a = attributes[i];
result.appendf(" {%-10s, %3zu, \"%s\"}\n", attr_type_str(a.type), a.offset, a.name.c_str());
}
if (kMax < attributes.size()) {
result.append(" ...\n");
}
result.appendf("Varyings (count=%zu):\n", varyings.size());
for (size_t i = 0; i < std::min(kMax, varyings.size()); ++i) {
const auto& v = varyings[i];
result.appendf(" {%5s, \"%s\"}\n", var_type_str(v.type), v.name.c_str());
}
if (kMax < varyings.size()) {
result.append(" ...\n");
}
result.appendf("\n--VS--\n%s\n------\n", vs.c_str());
result.appendf("\n--FS--\n%s\n------\n", fs.c_str());
return result;
}
static bool check_for_failure(skiatest::Reporter* r,
SkSpan<const Attribute> attributes,
size_t stride,
SkSpan<const Varying> varyings,
const SkString& vs,
const SkString& fs) {
auto [spec, error] = SkMeshSpecification::Make(attributes, stride, varyings, vs, fs);
SkString description;
if (!spec) {
return true;
}
ERRORF(r,
"Expected to fail but succeeded:\n%s",
make_description(attributes, stride, varyings, vs, fs).c_str());
return false;
}
static bool check_for_success(skiatest::Reporter* r,
SkSpan<const Attribute> attributes,
size_t stride,
SkSpan<const Varying> varyings,
const SkString& vs,
const SkString& fs,
sk_sp<SkMeshSpecification>* spec = nullptr) {
auto [s, error] = SkMeshSpecification::Make(attributes, stride, varyings, vs, fs);
if (s) {
REPORTER_ASSERT(r, error.isEmpty());
if (spec) {
*spec = std::move(s);
}
return true;
}
ERRORF(r,
"Expected to succeed but failed:\n%sError:\n%s",
make_description(attributes, stride, varyings, vs, fs).c_str(),
error.c_str());
return false;
}
// Simple valid strings to make specifications
static const SkString kValidVS
{"float2 main(Attributes attrs, out Varyings v) { return float2(10); }"};
// There are multiple valid VS signatures.
static const SkString kValidFSes[] {
SkString{"void main(Varyings varyings) {}"},
SkString{"float2 main(Varyings varyings) { return float2(10); }"},
SkString{"void main(Varyings varyings, out half4 color) { color = half4(.2); }"},
SkString{R"(
float2 main(Varyings varyings, out half4 color) {
color = half4(.2);
return float2(10);
}
)"},
};
// Simple valid attributes, stride, and varyings to make specifications
static const Attribute kValidAttrs[] = {
{Attribute::Type::kFloat4, 0, SkString{"pos"}},
};
static constexpr size_t kValidStride = 4*4;
static const Varying kValidVaryings[] = {
{Varying::Type::kFloat2, SkString{"uv"}},
};
static void test_good(skiatest::Reporter* r) {
for (const auto& validFS : kValidFSes) {
if (!check_for_success(r,
kValidAttrs,
kValidStride,
kValidVaryings,
kValidVS,
validFS)) {
return;
}
}
}
static void test_bad_sig(skiatest::Reporter* r) {
static constexpr const char* kVSBody = "{ return float2(10); }";
static constexpr const char* kInvalidVSSigs[] {
"float3 main(Attributes attrs, out Varyings v)", // bad return
"float2 main(inout Attributes attrs, out Varyings v)", // inout Attributes
"float2 main(Attributes attrs, inout Varyings v)", // inout Varyings
"float2 main(Attributes attrs)", // no Varyings
"float2 main(out Varyings)", // no Attributes
"float2 main(out Varyings, in Attributes)", // wrong param order
"float2 main(Attributes attrs, out Varyings v, float2)" // extra arg
};
static constexpr const char* kNoColorFSBody = "{ return float2(10); }";
static constexpr const char* kInvalidNoColorFSSigs[] {
"half2 main(Varyings v)", // bad return
"float2 main(in Attributes v)", // wrong param type
"float2 main(out Varyings attrs)", // out Varyings
"float2 main()", // no args
"float2 main(Attributes attrs, float)" // extra arg
};
static constexpr const char* kColorFSBody = "{ color = half4(.2); return float2(10); }";
static constexpr const char* kInvalidColorFSSigs[] {
"half2 main(Varyings v, out half4 color)", // bad return
"float2 main(in Attributes v, out half4 color)", // wrong first param type
"float2 main(in Varyings v, out half3 color)", // wrong second param type
"float2 main(out Varyings attrs, out half4 color)", // out Varyings
"float2 main(Varyings attrs, half4 color)", // in color
"float2 main(Attributes attrs, out half4 color, float)" // extra arg
};
for (const char* vsSig : kInvalidVSSigs) {
SkString invalidVS;
invalidVS.appendf("%s %s", vsSig, kVSBody);
for (const auto& validFS : kValidFSes) {
if (!check_for_failure(r,
kValidAttrs,
kValidStride,
kValidVaryings,
invalidVS,
validFS)) {
return;
}
}
}
for (const char* noColorFSSig : kInvalidNoColorFSSigs) {
SkString invalidFS;
invalidFS.appendf("%s %s", noColorFSSig, kNoColorFSBody);
if (!check_for_failure(r,
kValidAttrs,
kValidStride,
kValidVaryings,
kValidVS,
invalidFS)) {
return;
}
}
for (const char* colorFSSig : kInvalidColorFSSigs) {
SkString invalidFS;
invalidFS.appendf("%s %s", colorFSSig, kColorFSBody);
if (!check_for_failure(r,
kValidAttrs,
kValidStride,
kValidVaryings,
kValidVS,
invalidFS)) {
return;
}
}
}
// We allow the optional out color from the FS to either be float4 or half4
static void test_float4_color(skiatest::Reporter* r) {
static const SkString kFloat4FS {
R"(
float2 main(Varyings varyings, out float4 color) {
color = float4(.2); return float2(10);
}
)"
};
check_for_success(r,
kValidAttrs,
kValidStride,
kValidVaryings,
kValidVS,
kFloat4FS);
}
// We don't allow child effects in custom meshes currently.
static void test_bad_globals(skiatest::Reporter* r) {
static constexpr const char* kBadGlobals[] {
"uniform shader myshader;"
};
for (const auto& global : kBadGlobals) {
SkString badVS = kValidVS;
badVS.prepend(global);
if (!check_for_failure(r,
kValidAttrs,
kValidStride,
kValidVaryings,
badVS,
kValidFSes[0])) {
return;
}
}
for (const auto& global : kBadGlobals) {
SkString badFS = kValidFSes[0];
badFS.prepend(global);
if (!check_for_failure(r,
kValidAttrs,
kValidStride,
kValidVaryings,
kValidVS,
badFS)) {
return;
}
}
}
static void test_good_uniforms(skiatest::Reporter* r) {
using Uniform = SkMeshSpecification::Uniform;
using Type = Uniform::Type;
using Flags = Uniform::Flags;
constexpr Flags kVS = Uniform::kVertex_Flag;
constexpr Flags kFS = Uniform::kFragment_Flag;
constexpr Flags kColor = Uniform::kColor_Flag;
auto make_uni = [](Type type, const char* name, size_t offset, uint32_t flags, int count = 0) {
if (count) {
return Uniform{SkString(name), offset, type, count, flags | Uniform::kArray_Flag};
} else {
SkASSERT(!(flags & Uniform::kArray_Flag));
return Uniform{SkString(name), offset, type, 1, flags};
}
};
// Each test case is a set of VS and FS uniform declarations followed and the expected output
// of SkMeshSpecification::uniforms().
struct {
const std::vector<const char*> vsUniformDecls;
const std::vector<const char*> fsUniformDecls;
const std::vector<SkMeshSpecification::Uniform> expectations;
} static kTestCases[] {
// A single VS uniform.
{
{
"uniform float x;"
},
{},
{
make_uni(Type::kFloat, "x", 0, kVS)
}
},
// A single FS uniform.
{
{},
{
"uniform float2 v;"
},
{
make_uni(Type::kFloat2, "v", 0, kFS)
}
},
// A single uniform in both that uses color layout.
{
{
"layout(color) uniform float4 color;",
},
{
"layout(color) uniform float4 color;",
},
{
make_uni(Type::kFloat4, "color", 0, kVS|kFS|kColor)
}
},
// A shared uniform after an unshared vertex uniform
{
{
"layout(color) uniform float4 color;",
" uniform float x[5];",
},
{
"uniform float x[5];",
},
{
make_uni(Type::kFloat4, "color", 0, kVS|kColor, 0),
make_uni(Type::kFloat , "x" , 16, kVS|kFS , 5)
}
},
// A shared uniform before an unshared vertex uniform
{
{
"uniform half x[2];",
"uniform int y;",
},
{
"uniform half x[2];",
},
{
make_uni(Type::kFloat, "x", 0, kVS|kFS, 2),
make_uni(Type::kInt, "y", 8, kVS , 0)
}
},
// A shared uniform after an unshared fragment uniform
{
{
"uniform float3x3 m;",
},
{
"uniform int2 i2;",
"uniform float3x3 m;",
},
{
make_uni(Type::kFloat3x3, "m" , 0, kVS|kFS),
make_uni(Type::kInt2 , "i2", 36, kFS )
}
},
// A shared uniform before an unshared fragment uniform
{
{
"uniform half4x4 m[4];",
},
{
"uniform half4x4 m[4];",
"uniform int3 i3[1];",
},
{
make_uni(Type::kFloat4x4, "m", 0, kVS|kFS, 4),
make_uni(Type::kInt3, "i3", 256, kFS , 1)
}
},
// Complex case with 2 shared uniforms that are declared in the opposite order.
{
{
"uniform float x;"
"uniform half4x4 m[4];", // shared
"uniform int2 i2[2];"
"uniform float3 v[8];" // shared
"uniform int3 i3;"
},
{
"uniform float y;"
"uniform float3 v[8];" // shared
"uniform int4 i4[2];"
"uniform half4x4 m[4];", // shared
"uniform int i;"
},
{
make_uni(Type::kFloat, "x" , 0, kVS , 0),
make_uni(Type::kFloat4x4, "m" , 4, kVS|kFS, 4),
make_uni(Type::kInt2, "i2", 260, kVS , 2),
make_uni(Type::kFloat3, "v" , 276, kVS|kFS, 8),
make_uni(Type::kInt3, "i3", 372, kVS , 0),
make_uni(Type::kFloat, "y" , 384, kFS , 0),
make_uni(Type::kInt4, "i4", 388, kFS , 2),
make_uni(Type::kInt, "i" , 420, kFS , 0),
}
},
};
for (const auto& c : kTestCases) {
SkString vs = kValidVS;
SkString unis;
for (const auto u : c.vsUniformDecls) {
unis.append(u);
}
vs.prepend(unis);
SkString fs = kValidFSes[0];
unis = {};
for (const auto u : c.fsUniformDecls) {
unis.append(u);
}
fs.prepend(unis);
auto attrs = SkSpan(kValidAttrs);
auto varys = SkSpan(kValidVaryings);
sk_sp<SkMeshSpecification> spec;
if (!check_for_success(r, attrs, kValidStride, varys, vs, fs, &spec)) {
return;
}
SkString desc = make_description(attrs, kValidStride, varys, vs, fs);
auto uniforms = spec->uniforms();
if (uniforms.size() != c.expectations.size()) {
ERRORF(r,
"Expected %zu uniforms but actually %zu:\n%s",
c.expectations.size(),
uniforms.size(),
desc.c_str());
return;
}
for (const auto& [actual, expected] : SkMakeZip(uniforms, c.expectations)) {
if (actual.name != expected.name) {
ERRORF(r,
"Actual uniform name (%s) does not match expected name (%s)",
actual.name.c_str(),
expected.name.c_str());
return;
}
const char* name = actual.name.c_str();
if (actual.type != expected.type) {
ERRORF(r,
"Uniform %s: Actual type (%d) does not match expected type (%d)",
name,
static_cast<int>(actual.type),
static_cast<int>(expected.type));
return;
}
if (actual.count != expected.count) {
ERRORF(r,
"Uniform %s: Actual count (%d) does not match expected count (%d)",
name,
actual.count,
expected.count);
return;
}
if (actual.flags != expected.flags) {
ERRORF(r,
"Uniform %s: Actual flags (0x%04x) do not match expected flags (0x%04x)",
name,
actual.flags,
expected.flags);
return;
}
if (actual.offset != expected.offset) {
ERRORF(r,
"Uniform %s: Actual offset (%zu) does not match expected offset (%zu)",
name,
actual.offset,
expected.offset);
return;
}
}
}
}
static void test_bad_uniforms(skiatest::Reporter* r) {
// We assume general uniform declarations are broadly tested generically in SkSL. Here we are
// concerned with agreement between VS and FS declarations, which is a unique aspect of
// SkMeshSpecification.
// Each test case is a fs and vs uniform declaration with the same name but some other
// difference that should make them incompatible.
static std::tuple<const char*, const char*> kTestCases[]{
// different types
{"uniform float x;", "uniform int x;"},
// array vs non-array
{"uniform float2x2 m[1];", "uniform float2x2 m;"},
// array count mismatch
{"uniform int3 i[1];", "uniform int3 i[2];"},
// layout difference
{"layout(color) uniform float4 color;", "uniform float4 color;"},
};
for (bool reverse : {false, true}) {
for (auto [u1, u2] : kTestCases) {
if (reverse) {
using std::swap;
swap(u1, u2);
}
SkString vs = kValidVS;
vs.prepend(u1);
SkString fs = kValidFSes[0];
fs.prepend(u2);
auto attrs = SkSpan(kValidAttrs);
auto varys = SkSpan(kValidVaryings);
if (!check_for_failure(r, attrs, kValidStride, varys, vs, fs)) {
return;
}
}
}
}
static void test_no_main(skiatest::Reporter* r) {
static const SkString kHelper{"float2 swiz(float2 x) { return z.yx; }"};
// Empty VS
if (!check_for_failure(r,
kValidAttrs,
kValidStride,
kValidVaryings,
SkString{},
kValidFSes[0])) {
return;
}
// VS with helper function but no main
if (!check_for_failure(r,
kValidAttrs,
kValidStride,
kValidVaryings,
kHelper,
kValidFSes[0])) {
return;
}
// Empty FS
if (!check_for_failure(r,
kValidAttrs,
kValidStride,
kValidVaryings,
kValidVS,
SkString{})) {
return;
}
// VS with helper function but no main
if (!check_for_failure(r,
kValidAttrs,
kValidStride,
kValidVaryings,
kValidVS,
kHelper)) {
return;
}
}
static void test_zero_attrs(skiatest::Reporter* r) {
// We require at least one attribute
check_for_failure(r,
SkSpan<Attribute>(),
kValidStride,
kValidVaryings,
kValidVS,
kValidFSes[0]);
}
static void test_zero_varyings(skiatest::Reporter* r) {
// Varyings are not required.
check_for_success(r,
kValidAttrs,
kValidStride,
SkSpan<Varying>(),
kValidVS,
kValidFSes[0]);
}
static void test_bad_strides(skiatest::Reporter* r) {
// Zero stride
if (!check_for_failure(r,
kValidAttrs,
0,
kValidVaryings,
kValidVS,
kValidFSes[0])) {
return;
}
// Unaligned
if (!check_for_failure(r,
kValidAttrs,
kValidStride + 1,
kValidVaryings,
kValidVS,
kValidFSes[0])) {
return;
}
// Too large
if (!check_for_failure(r,
kValidAttrs,
1 << 20,
kValidVaryings,
kValidVS,
kValidFSes[0])) {
return;
}
}
static void test_bad_offsets(skiatest::Reporter* r) {
{ // offset isn't aligned
static const Attribute kAttributes[] {
{Attribute::Type::kFloat4, 1, SkString{"var"}},
};
if (!check_for_failure(r,
kAttributes,
32,
kValidVaryings,
kValidVS,
kValidFSes[0])) {
return;
}
}
{ // straddles stride boundary
static const Attribute kAttributes[] {
{Attribute::Type::kFloat4, 0, SkString{"var"}},
{Attribute::Type::kFloat2, 16, SkString{"var"}},
};
if (!check_for_failure(r,
kAttributes,
20,
kValidVaryings,
kValidVS,
kValidFSes[0])) {
return;
}
}
{ // straddles stride boundary with attempt to overflow
static const Attribute kAttributes[] {
{Attribute::Type::kFloat, std::numeric_limits<size_t>::max() - 3, SkString{"var"}},
};
if (!check_for_failure(r,
kAttributes,
4,
kValidVaryings,
kValidVS,
kValidFSes[0])) {
return;
}
}
}
static void test_too_many_attrs(skiatest::Reporter* r) {
static constexpr size_t kN = 500;
std::vector<Attribute> attrs;
attrs.reserve(kN);
for (size_t i = 0; i < kN; ++i) {
attrs.push_back({Attribute::Type::kFloat4, 0, SkStringPrintf("attr%zu", i)});
}
check_for_failure(r,
attrs,
4*4,
kValidVaryings,
kValidVS,
kValidFSes[0]);
}
static void test_too_many_varyings(skiatest::Reporter* r) {
static constexpr size_t kN = 500;
std::vector<Varying> varyings;
varyings.reserve(kN);
for (size_t i = 0; i < kN; ++i) {
varyings.push_back({Varying::Type::kFloat4, SkStringPrintf("varying%zu", i)});
}
check_for_failure(r,
kValidAttrs,
kValidStride,
SkSpan(varyings),
kValidVS,
kValidFSes[0]);
}
static void test_duplicate_attribute_names(skiatest::Reporter* r) {
static const Attribute kAttributes[] {
{Attribute::Type::kFloat4, 0, SkString{"var"}},
{Attribute::Type::kFloat2, 16, SkString{"var"}}
};
check_for_failure(r,
kAttributes,
24,
kValidVaryings,
kValidVS,
kValidFSes[0]);
}
static void test_duplicate_varying_names(skiatest::Reporter* r) {
static const Varying kVaryings[] {
{Varying::Type::kFloat4, SkString{"var"}},
{Varying::Type::kFloat3, SkString{"var"}}
};
check_for_failure(r,
kValidAttrs,
kValidStride,
kVaryings,
kValidVS,
kValidFSes[0]);
}
static constexpr const char* kSneakyName = "name; float3 sneaky";
static void test_sneaky_attribute_name(skiatest::Reporter* r) {
static const Attribute kAttributes[] {
{Attribute::Type::kFloat4, 0, SkString{kSneakyName}},
};
check_for_failure(r,
kAttributes,
16,
kValidVaryings,
kValidVS,
kValidFSes[0]);
}
static void test_sneaky_varying_name(skiatest::Reporter* r) {
static const Varying kVaryings[] {
{Varying::Type::kFloat4, SkString{kSneakyName}},
};
check_for_failure(r,
kValidAttrs,
kValidStride,
kVaryings,
kValidVS,
kValidFSes[0]);
}
static void test_empty_attribute_name(skiatest::Reporter* r) {
static const Attribute kAttributes[] {
{Attribute::Type::kFloat4, 0, SkString{}},
};
check_for_failure(r,
kAttributes,
16,
kValidVaryings,
kValidVS,
kValidFSes[0]);
}
static void test_empty_varying_name(skiatest::Reporter* r) {
static const Varying kVaryings[] {
{Varying::Type::kFloat4, SkString{}},
};
check_for_failure(r,
kValidAttrs,
kValidStride,
kVaryings,
kValidVS,
kValidFSes[0]);
}
DEF_TEST(MeshSpec, reporter) {
struct X {};
test_good(reporter);
test_bad_sig(reporter);
test_float4_color(reporter);
test_bad_globals(reporter);
test_good_uniforms(reporter);
test_bad_uniforms(reporter);
test_no_main(reporter);
test_zero_attrs(reporter);
test_zero_varyings(reporter);
test_bad_strides(reporter);
test_bad_offsets(reporter);
test_too_many_attrs(reporter);
test_too_many_varyings(reporter);
// skbug.com/12712
if ((false)) {
test_duplicate_attribute_names(reporter);
test_duplicate_varying_names(reporter);
}
test_sneaky_attribute_name(reporter);
test_sneaky_varying_name(reporter);
test_empty_attribute_name(reporter);
test_empty_varying_name(reporter);
}