From 260a7966ee2a97b8308ff44b04d312bcbc952357 Mon Sep 17 00:00:00 2001
From: Robin Voetter <robin@voetter.nl>
Date: Wed, 1 Jul 2020 17:44:23 +0200
Subject: [PATCH] Categorize commands under dispatch type

---
 generator/render.zig | 56 ++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 52 insertions(+), 4 deletions(-)

diff --git a/generator/render.zig b/generator/render.zig
index c043c6c..e7a7eae 100644
--- a/generator/render.zig
+++ b/generator/render.zig
@@ -144,6 +144,12 @@ fn Renderer(comptime WriterType: type) type {
             },
         };
 
+        const CommandDispatchType = enum {
+            base,
+            instance,
+            device,
+        };
+
         writer: WriterType,
         allocator: *Allocator,
         registry: *const reg.Registry,
@@ -248,6 +254,38 @@ fn Renderer(comptime WriterType: type) type {
             return .other;
         }
 
+        fn classifyCommandDispatch(self: Self, name: []const u8, command: reg.Command) CommandDispatchType {
+            const device_handles = std.ComptimeStringMap(void, .{
+                .{"VkDevice", {}},
+                .{"VkCommandBuffer", {}},
+                .{"VkQueue", {}},
+            });
+
+            const override_functions = std.ComptimeStringMap(CommandDispatchType, .{
+                .{"vkGetInstanceProcAddr", .base},
+                .{"vkCreateInstance", .base},
+                .{"vkEnumerateInstanceLayerProperties", .base},
+                .{"vkEnumerateInstanceExtensionProperties", .base},
+                .{"vkEnumerateInstanceVersion", .base},
+                .{"vkGetDeviceProcAddr", .instance},
+            });
+
+            if (override_functions.get(name)) |dispatch_type| {
+                return dispatch_type;
+            }
+
+            switch (command.params[0].param_type) {
+                .name => |first_param_type_name| {
+                    if (device_handles.get(first_param_type_name)) |_| {
+                        return .device;
+                    }
+                },
+                else => {},
+            }
+
+            return .instance;
+        }
+
         fn render(self: *Self) !void {
             try self.writer.writeAll(preamble);
 
@@ -704,15 +742,25 @@ fn Renderer(comptime WriterType: type) type {
         }
 
         fn renderWrappers(self: *Self) !void {
-            try self.writer.writeAll(
-                \\pub fn Wrapper(comptime Self: type) type {
-                \\    return struct {
+            try self.renderWrappersOfDispatchType("BaseWrapper", .base);
+            try self.renderWrappersOfDispatchType("InstanceWrapper", .instance);
+            try self.renderWrappersOfDispatchType("DeviceWrapper", .device);
+        }
+
+        fn renderWrappersOfDispatchType(self: *Self, name: []const u8, dispatch_type: CommandDispatchType) !void {
+            try self.writer.print(
+                \\pub fn {}(comptime Self: type) type {{
+                \\    return struct {{
                 \\
+                , .{name}
             );
 
             for (self.registry.decls) |decl| {
                 if (decl.decl_type == .command) {
-                    try self.renderWrapper(decl.name, decl.decl_type.command);
+                    const command = decl.decl_type.command;
+                    if (self.classifyCommandDispatch(decl.name, command) == dispatch_type) {
+                        try self.renderWrapper(decl.name, decl.decl_type.command);
+                    }
                 }
             }