zig

commit 84039a57e4684e8df10e657bb76c6acb3fb89238 (tree)
parent 7aaea20e7e5e98bc5b2c4c8ddcf8f6aaa4a9c55d
Author: joachimschmidt557 <joachim.schmidt557@outlook.com>
Date:   Fri, 30 Jul 2021 22:48:20 +0200

stage2 codegen: Implement genTypedValue for enums

Diffstat:
M
src/codegen.zig
 | 
23
+++++++++++++++++++++++
M
test/stage2/arm.zig
 | 
22
++++++++++++++++++++++

2 files changed, 45 insertions(+), 0 deletions(-)
diff --git a/src/codegen.zig b/src/codegen.zig
@@ -4740,6 +4740,29 @@ fn Function(comptime arch: std.Target.Cpu.Arch) type {
                     }
                     return self.fail("TODO non pointer optionals", .{});
                 },
+                .Enum => {
+                    if (typed_value.val.castTag(.enum_field_index)) |field_index| {
+                        switch (typed_value.ty.tag()) {
+                            .enum_simple => {
+                                return MCValue{ .immediate = field_index.data };
+                            },
+                            .enum_full, .enum_nonexhaustive => {
+                                const enum_full = typed_value.ty.cast(Type.Payload.EnumFull).?.data;
+                                if (enum_full.values.count() != 0) {
+                                    const tag_val = enum_full.values.keys()[field_index.data];
+                                    return self.genTypedValue(.{ .ty = enum_full.tag_ty, .val = tag_val });
+                                } else {
+                                    return MCValue{ .immediate = field_index.data };
+                                }
+                            },
+                            else => unreachable,
+                        }
+                    } else {
+                        var int_tag_buffer: Type.Payload.Bits = undefined;
+                        const int_tag_ty = typed_value.ty.intTagType(&int_tag_buffer);
+                        return self.genTypedValue(.{ .ty = int_tag_ty, .val = typed_value.val });
+                    }
+                },
                 .ErrorSet => {
                     switch (typed_value.val.tag()) {
                         .@"error" => {
diff --git a/test/stage2/arm.zig b/test/stage2/arm.zig
@@ -300,6 +300,28 @@ pub fn addCases(ctx: *TestContext) !void {
     }
 
     {
+        var case = ctx.exe("enums", linux_arm);
+        case.addCompareOutput(
+            \\const Number = enum { one, two, three };
+            \\
+            \\pub fn main() void {
+            \\    var x: Number = .one;
+            \\    var y = Number.two;
+            \\    var z = @intToEnum(Number, 2);
+            \\    assert(@enumToInt(x) == 0);
+            \\    assert(@enumToInt(y) == 1);
+            \\    assert(@enumToInt(z) == 2);
+            \\}
+            \\
+            \\fn assert(ok: bool) void {
+            \\    if (!ok) unreachable; // assertion failure
+            \\}
+        ,
+            "",
+        );
+    }
+
+    {
         var case = ctx.exe("recursive fibonacci", linux_arm);
         case.addCompareOutput(
             \\pub fn main() void {