SPIR-V: Function parameter generation

2 files changed, 16 insertions, 2 deletions

diff --git a/src/codegen/spirv.zig b/src/codegen/spirv.zig
index db4ddbd657..a6b8216774 100644
--- a/src/codegen/spirv.zig
+++ b/src/codegen/spirv.zig
@@ -57,6 +57,7 @@ pub const DeclGen = struct {
     module: *Module,
     spv: *SPIRVModule,
 
+    args: std.ArrayList(u32),
     types: TypeMap,
 
     decl: *Decl,
@@ -213,7 +214,17 @@ pub const DeclGen = struct {
                 prototype_id,
             });
 
-            // TODO: Parameters
+            const params = tv.ty.fnParamLen();
+            var i: usize = 0;
+
+            try self.args.ensureCapacity(params);
+            while (i < params) : (i += 1) {
+                const param_type_id = self.types.get(tv.ty.fnParamType(i)).?;
+                const arg_result_id = self.spv.allocResultId();
+                try writeInstruction(&self.spv.fn_decls, .OpFunctionParameter, &[_]u32{ param_type_id, arg_result_id });
+                self.args.appendAssumeCapacity(arg_result_id);
+            }
+
             // TODO: Body
 
             try writeInstruction(&self.spv.fn_decls, .OpFunctionEnd, &[_]u32{});
diff --git a/src/link/SpirV.zig b/src/link/SpirV.zig
index 8d5feafa5a..8bc0d9fe9f 100644
--- a/src/link/SpirV.zig
+++ b/src/link/SpirV.zig
@@ -140,23 +140,26 @@ pub fn flushModule(self: *SpirV, comp: *Compilation) !void {
     // Now, actually generate the code for all declarations.
     {
         // We are just going to re-use this same DeclGen for every Decl, and we are just going to
-        // change the decl. Otherwise, we would have to keep a separate `types`, and re-construct this
+        // change the decl. Otherwise, we would have to keep a separate `args` and `types`, and re-construct this
         // structure every time.
         var decl_gen = codegen.DeclGen{
             .module = module,
             .spv = &spv,
+            .args = std.ArrayList(u32).init(self.base.allocator),
             .types = codegen.TypeMap.init(self.base.allocator),
             .decl = undefined,
             .error_msg = undefined,
         };
 
         defer decl_gen.types.deinit();
+        defer decl_gen.args.deinit();
 
         for (module.decl_table.items()) |entry| {
             const decl = entry.value;
             if (decl.typed_value != .most_recent)
                 continue;
 
+            decl_gen.args.items.len = 0;
             decl_gen.decl = decl;
             decl_gen.error_msg = null;