commit 7bbe4c1696f2433d7228877d451b812e23c4adff
author dan sinclair <dsinclair@chromium.org> Tue Nov 21 12:11:32 2023 +0000
committer Dawn LUCI CQ <dawn-scoped@luci-project-accounts.iam.gserviceaccount.com> Tue Nov 21 12:11:32 2023 +0000
tree c94a073b36ba8d2594e491270e4866484c4bda9c
parent c57f10fb6117c7412408c16c5f9de153927dffde

[ir][msl] Add `construct` to MSL IR printer

This CL adds support for emitting constructors from the MSL IR
printer.

Bug: tint:1967
Change-Id: Id8df9a398e09f65448ecbf0cec9a5afa9a0fdd44
Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/161801
Reviewed-by: Ben Clayton <bclayton@google.com>
Kokoro: Kokoro <noreply+kokoro@google.com>
Commit-Queue: dan sinclair <dsinclair@chromium.org>

src/tint/lang/msl/writer/printer/printer.cc

diff --git a/src/tint/lang/msl/writer/printer/printer.cc b/src/tint/lang/msl/writer/printer/printer.cc
index c003776..3db90f7 100644
--- a/src/tint/lang/msl/writer/printer/printer.cc
+++ b/src/tint/lang/msl/writer/printer/printer.cc
@@ -36,6 +36,7 @@
 #include "src/tint/lang/core/fluent_types.h"
 #include "src/tint/lang/core/ir/binary.h"
 #include "src/tint/lang/core/ir/constant.h"
+#include "src/tint/lang/core/ir/construct.h"
 #include "src/tint/lang/core/ir/discard.h"
 #include "src/tint/lang/core/ir/exit_if.h"
 #include "src/tint/lang/core/ir/ice.h"
@@ -245,6 +246,7 @@
                 [&](core::ir::Unreachable*) { EmitUnreachable(); },  //
                 [&](core::ir::Var* v) { EmitVar(v); },               //
                 [&](core::ir::Discard*) { EmitDiscard(); },          //
+                [&](core::ir::Construct* c) { EmitConstruct(c); },   //
                 TINT_ICE_ON_NO_MATCH);
         }
     }
@@ -432,6 +434,57 @@
     /// Emit a discard instruction
     void EmitDiscard() { Line() << "discard_fragment();"; }
 
+    /// Emit a constructor
+    void EmitConstruct(core::ir::Construct* c) {
+        StringStream str;
+
+        Switch(
+            c->Result()->Type(),
+            [&](const core::type::Array*) {
+                EmitType(str, c->Result()->Type());
+                str << "{";
+                size_t i = 0;
+                for (auto* arg : c->Args()) {
+                    if (i > 0) {
+                        str << ", ";
+                    }
+                    str << Expr(arg);
+                    i++;
+                }
+                str << "}";
+            },
+            [&](const core::type::Struct* struct_ty) {
+                str << "{";
+                size_t i = 0;
+                for (auto* arg : c->Args()) {
+                    if (i > 0) {
+                        str << ", ";
+                    }
+                    // Emit field designators for structures to account for padding members.
+                    auto name = struct_ty->Members()[i]->Name().Name();
+                    str << "." << name << "=";
+                    str << Expr(arg);
+                    i++;
+                }
+                str << "}";
+            },
+            [&](Default) {
+                EmitType(str, c->Result()->Type());
+                str << "(";
+                size_t i = 0;
+                for (auto* arg : c->Args()) {
+                    if (i > 0) {
+                        str << ", ";
+                    }
+                    str << Expr(arg);
+                    i++;
+                }
+                str << ")";
+            });
+
+        Bind(c->Result(), str.str());
+    }
+
     /// Handles generating a address space
     /// @param out the output of the type stream
     /// @param sc the address space to generate