mirror/zig
1
0
Fork 0
mirror of https://github.com/ziglang/zig.git synced 2026-01-20 14:25:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Sema: implement inline switch capture at comptime

Browse Source 
Closes #15157
This commit is contained in:
Veikka Tuominen 2023-04-04 13:39:06 +03:00
parent 080136e4ad
commit 82a6acca93
2 changed files with 32 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/Sema.zig
View File

@ -9963,6 +9963,26 @@ fn zirSwitchCapture(
const field_index = @intCast(u32, operand_ty.unionTagFieldIndex(item_val, sema.mod).?);
const union_obj = operand_ty.cast(Type.Payload.Union).?.data;
const field_ty = union_obj.fields.values()[field_index].ty;
if (try sema.resolveDefinedValue(block, sema.src, operand_ptr)) |union_val| {
if (is_ref) {
const ptr_field_ty = try Type.ptr(sema.arena, sema.mod, .{
.pointee_type = field_ty,
.mutable = operand_ptr_ty.ptrIsMutable(),
.@"volatile" = operand_ptr_ty.isVolatilePtr(),
.@"addrspace" = operand_ptr_ty.ptrAddressSpace(),
});
return sema.addConstant(
ptr_field_ty,
try Value.Tag.field_ptr.create(sema.arena, .{
.container_ptr = union_val,
.container_ty = operand_ty,
.field_index = field_index,
}),
);
}
const tag_and_val = union_val.castTag(.@"union").?.data;
return sema.addConstant(field_ty, tag_and_val.val);
}
if (is_ref) {
const ptr_field_ty = try Type.ptr(sema.arena, sema.mod, .{
.pointee_type = field_ty,

12
test/behavior/switch.zig
View File

@ -694,3 +694,15 @@ test "switch item sizeof" {
try S.doTheTest();
comptime try S.doTheTest();
}
test "comptime inline switch" {
const U = union(enum) { a: type, b: type };
const value = comptime blk: {
var u: U = .{ .a = u32 };
break :blk switch (u) {
inline .a, .b => |v| v,
};
};
try expectEqual(u32, value);
}