ProgramBuilder: Migrate matrices to typ::TypePair

Used as a stepping stone to emitting the ast::Types instead.

Bug: tint:724
Change-Id: Ic81f0a00456e91ac089df32193c4323c2e779c29
Reviewed-on: https://dawn-review.googlesource.com/c/tint/+/48684
Commit-Queue: Ben Clayton <bclayton@google.com>
Reviewed-by: Antonio Maiorano <amaiorano@google.com>
diff --git a/src/program_builder.h b/src/program_builder.h
index 805dba3..1dec6d9 100644
--- a/src/program_builder.h
+++ b/src/program_builder.h
@@ -30,6 +30,7 @@
 #include "src/ast/i32.h"
 #include "src/ast/if_statement.h"
 #include "src/ast/loop_statement.h"
+#include "src/ast/matrix.h"
 #include "src/ast/member_accessor_expression.h"
 #include "src/ast/module.h"
 #include "src/ast/return_statement.h"
@@ -401,109 +402,118 @@
 
     /// @param type matrix subtype
     /// @return the tint AST type for a 2x3 matrix of `type`.
-    sem::Matrix* mat2x2(sem::Type* type) const {
-      return builder->create<sem::Matrix>(type, 2u, 2u);
+    typ::Matrix mat2x2(typ::Type type) const {
+      return {builder->create<ast::Matrix>(type, 2u, 2u),
+              builder->create<sem::Matrix>(type, 2u, 2u)};
     }
 
     /// @param type matrix subtype
     /// @return the tint AST type for a 2x3 matrix of `type`.
-    sem::Matrix* mat2x3(sem::Type* type) const {
-      return builder->create<sem::Matrix>(type, 3u, 2u);
+    typ::Matrix mat2x3(typ::Type type) const {
+      return {builder->create<ast::Matrix>(type, 3u, 2u),
+              builder->create<sem::Matrix>(type, 3u, 2u)};
     }
 
     /// @param type matrix subtype
     /// @return the tint AST type for a 2x4 matrix of `type`.
-    sem::Matrix* mat2x4(sem::Type* type) const {
-      return builder->create<sem::Matrix>(type, 4u, 2u);
+    typ::Matrix mat2x4(typ::Type type) const {
+      return {builder->create<ast::Matrix>(type, 4u, 2u),
+              builder->create<sem::Matrix>(type, 4u, 2u)};
     }
 
     /// @param type matrix subtype
     /// @return the tint AST type for a 3x2 matrix of `type`.
-    sem::Matrix* mat3x2(sem::Type* type) const {
-      return builder->create<sem::Matrix>(type, 2u, 3u);
+    typ::Matrix mat3x2(typ::Type type) const {
+      return {builder->create<ast::Matrix>(type, 2u, 3u),
+              builder->create<sem::Matrix>(type, 2u, 3u)};
     }
 
     /// @param type matrix subtype
     /// @return the tint AST type for a 3x3 matrix of `type`.
-    sem::Matrix* mat3x3(sem::Type* type) const {
-      return builder->create<sem::Matrix>(type, 3u, 3u);
+    typ::Matrix mat3x3(typ::Type type) const {
+      return {builder->create<ast::Matrix>(type, 3u, 3u),
+              builder->create<sem::Matrix>(type, 3u, 3u)};
     }
 
     /// @param type matrix subtype
     /// @return the tint AST type for a 3x4 matrix of `type`.
-    sem::Matrix* mat3x4(sem::Type* type) const {
-      return builder->create<sem::Matrix>(type, 4u, 3u);
+    typ::Matrix mat3x4(typ::Type type) const {
+      return {builder->create<ast::Matrix>(type, 4u, 3u),
+              builder->create<sem::Matrix>(type, 4u, 3u)};
     }
 
     /// @param type matrix subtype
     /// @return the tint AST type for a 4x2 matrix of `type`.
-    sem::Matrix* mat4x2(sem::Type* type) const {
-      return builder->create<sem::Matrix>(type, 2u, 4u);
+    typ::Matrix mat4x2(typ::Type type) const {
+      return {builder->create<ast::Matrix>(type, 2u, 4u),
+              builder->create<sem::Matrix>(type, 2u, 4u)};
     }
 
     /// @param type matrix subtype
     /// @return the tint AST type for a 4x3 matrix of `type`.
-    sem::Matrix* mat4x3(sem::Type* type) const {
-      return builder->create<sem::Matrix>(type, 3u, 4u);
+    typ::Matrix mat4x3(typ::Type type) const {
+      return {builder->create<ast::Matrix>(type, 3u, 4u),
+              builder->create<sem::Matrix>(type, 3u, 4u)};
     }
 
     /// @param type matrix subtype
     /// @return the tint AST type for a 4x4 matrix of `type`.
-    sem::Matrix* mat4x4(sem::Type* type) const {
-      return builder->create<sem::Matrix>(type, 4u, 4u);
+    typ::Matrix mat4x4(typ::Type type) const {
+      return {builder->create<ast::Matrix>(type, 4u, 4u),
+              builder->create<sem::Matrix>(type, 4u, 4u)};
     }
 
     /// @return the tint AST type for a 2x3 matrix of the C type `T`.
     template <typename T>
-    sem::Matrix* mat2x2() const {
+    typ::Matrix mat2x2() const {
       return mat2x2(Of<T>());
     }
 
     /// @return the tint AST type for a 2x3 matrix of the C type `T`.
     template <typename T>
-    sem::Matrix* mat2x3() const {
+    typ::Matrix mat2x3() const {
       return mat2x3(Of<T>());
     }
 
     /// @return the tint AST type for a 2x4 matrix of the C type `T`.
     template <typename T>
-    sem::Matrix* mat2x4() const {
+    typ::Matrix mat2x4() const {
       return mat2x4(Of<T>());
     }
 
     /// @return the tint AST type for a 3x2 matrix of the C type `T`.
     template <typename T>
-    sem::Matrix* mat3x2() const {
+    typ::Matrix mat3x2() const {
       return mat3x2(Of<T>());
     }
 
     /// @return the tint AST type for a 3x3 matrix of the C type `T`.
     template <typename T>
-    sem::Matrix* mat3x3() const {
+    typ::Matrix mat3x3() const {
       return mat3x3(Of<T>());
     }
 
     /// @return the tint AST type for a 3x4 matrix of the C type `T`.
     template <typename T>
-    sem::Matrix* mat3x4() const {
+    typ::Matrix mat3x4() const {
       return mat3x4(Of<T>());
     }
 
     /// @return the tint AST type for a 4x2 matrix of the C type `T`.
     template <typename T>
-    sem::Matrix* mat4x2() const {
+    typ::Matrix mat4x2() const {
       return mat4x2(Of<T>());
     }
 
     /// @return the tint AST type for a 4x3 matrix of the C type `T`.
     template <typename T>
-    sem::Matrix* mat4x3() const {
+    typ::Matrix mat4x3() const {
       return mat4x3(Of<T>());
     }
 
     /// @return the tint AST type for a 4x4 matrix of the C type `T`.
     template <typename T>
-    sem::Matrix* mat4x4() const {
+    typ::Matrix mat4x4() const {
       return mat4x4(Of<T>());
     }
 
diff --git a/src/typepair.h b/src/typepair.h
index 7638880..9a5ee20 100644
--- a/src/typepair.h
+++ b/src/typepair.h
@@ -25,6 +25,7 @@
 class Bool;
 class F32;
 class I32;
+class Matrix;
 class U32;
 class Vector;
 class Void;
@@ -34,6 +35,7 @@
 class Bool;
 class F32;
 class I32;
+class Matrix;
 class U32;
 class Vector;
 class Void;
@@ -85,6 +87,7 @@
 using Bool = TypePair<ast::Bool, sem::Bool>;
 using F32 = TypePair<ast::F32, sem::F32>;
 using I32 = TypePair<ast::I32, sem::I32>;
+using Matrix = TypePair<ast::Matrix, sem::Matrix>;
 using U32 = TypePair<ast::U32, sem::U32>;
 using Vector = TypePair<ast::Vector, sem::Vector>;
 using Void = TypePair<ast::Void, sem::Void>;
diff --git a/src/writer/hlsl/generator_impl_type_test.cc b/src/writer/hlsl/generator_impl_type_test.cc
index 5d8bfd6..f24dd4d 100644
--- a/src/writer/hlsl/generator_impl_type_test.cc
+++ b/src/writer/hlsl/generator_impl_type_test.cc
@@ -135,7 +135,7 @@
 }
 
 TEST_F(HlslGeneratorImplTest_Type, EmitType_Matrix) {
-  auto* mat2x3 = ty.mat2x3<f32>();
+  auto mat2x3 = ty.mat2x3<f32>();
 
   GeneratorImpl& gen = Build();
 
diff --git a/src/writer/msl/generator_impl_type_test.cc b/src/writer/msl/generator_impl_type_test.cc
index e489cba..33aefd0 100644
--- a/src/writer/msl/generator_impl_type_test.cc
+++ b/src/writer/msl/generator_impl_type_test.cc
@@ -154,7 +154,7 @@
 }
 
 TEST_F(MslGeneratorImplTest, EmitType_Matrix) {
-  auto* mat2x3 = ty.mat2x3<f32>();
+  auto mat2x3 = ty.mat2x3<f32>();
 
   GeneratorImpl& gen = Build();
 
diff --git a/src/writer/spirv/builder_type_test.cc b/src/writer/spirv/builder_type_test.cc
index 3fe6541..2f4a6d3 100644
--- a/src/writer/spirv/builder_type_test.cc
+++ b/src/writer/spirv/builder_type_test.cc
@@ -232,7 +232,7 @@
 }
 
 TEST_F(BuilderTest_Type, GenerateMatrix) {
-  auto* mat2x3 = ty.mat2x3<f32>();
+  auto mat2x3 = ty.mat2x3<f32>();
 
   spirv::Builder& b = Build();
 
@@ -248,7 +248,7 @@
 }
 
 TEST_F(BuilderTest_Type, ReturnsGeneratedMatrix) {
-  auto* mat = ty.mat4x3<i32>();
+  auto mat = ty.mat4x3<i32>();
   auto i32 = ty.i32();
 
   spirv::Builder& b = Build();
diff --git a/src/writer/wgsl/generator_impl_type_test.cc b/src/writer/wgsl/generator_impl_type_test.cc
index 296294b..c4f3e4a 100644
--- a/src/writer/wgsl/generator_impl_type_test.cc
+++ b/src/writer/wgsl/generator_impl_type_test.cc
@@ -140,7 +140,7 @@
 }
 
 TEST_F(WgslGeneratorImplTest, EmitType_Matrix) {
-  auto* mat2x3 = ty.mat2x3<f32>();
+  auto mat2x3 = ty.mat2x3<f32>();
   AST().AddConstructedType(ty.alias("make_type_reachable", mat2x3));
 
   GeneratorImpl& gen = Build();