[wgsl-writer] Emit array stride decoration. This CL adds array stride decorations to the WGSL writer. Bug: tint:179 Change-Id: I39d7625e1bdc876bbf9a83da7af76eb98bd09c28 Reviewed-on: https://dawn-review.googlesource.com/c/tint/+/26002 Reviewed-by: David Neto <dneto@google.com>
diff --git a/src/writer/wgsl/generator_impl.cc b/src/writer/wgsl/generator_impl.cc index 951c78d..dac8d3c 100644 --- a/src/writer/wgsl/generator_impl.cc +++ b/src/writer/wgsl/generator_impl.cc
@@ -356,6 +356,11 @@ out_ << alias->name(); } else if (type->IsArray()) { auto* ary = type->AsArray(); + + if (ary->has_array_stride()) { + out_ << "[[stride " << ary->array_stride() << "]] "; + } + out_ << "array<"; if (!EmitType(ary->type())) { return false;
diff --git a/src/writer/wgsl/generator_impl_type_test.cc b/src/writer/wgsl/generator_impl_type_test.cc index 1cbc223..48c8fc3 100644 --- a/src/writer/wgsl/generator_impl_type_test.cc +++ b/src/writer/wgsl/generator_impl_type_test.cc
@@ -55,6 +55,16 @@ EXPECT_EQ(g.result(), "array<bool, 4>"); } +TEST_F(WgslGeneratorImplTest, EmitType_Array_WithStride) { + ast::type::BoolType b; + ast::type::ArrayType a(&b, 4); + a.set_array_stride(16); + + GeneratorImpl g; + ASSERT_TRUE(g.EmitType(&a)) << g.error(); + EXPECT_EQ(g.result(), "[[stride 16]] array<bool, 4>"); +} + TEST_F(WgslGeneratorImplTest, EmitType_RuntimeArray) { ast::type::BoolType b; ast::type::ArrayType a(&b);