authorgravatar for andrew@ziglang.orgAndrew Kelley <andrew@ziglang.org> 2021-09-13 20:08:44-07:00
committergravatar for andrew@ziglang.orgAndrew Kelley <andrew@ziglang.org> 2021-09-13 20:11:35-07:00
log5529febab056c870f7b2a123b0645b5e3b1146c9
treea0e71a36908a9ab4e83e50dd5b96812561dade34
parentf011f13933b72f4d63a5f635c7646b68beee726e

stage2: implement Value.copy for structs and unions

The stage2_os hack inside `@import("builtin")` is no longer needed.

4 files changed, 70 insertions(+), 46 deletions(-)

lib/std/start.zig+1-1
......@@ -92,7 +92,7 @@ fn _start2() callconv(.Naked) noreturn {
9292}
9393
9494fn exit2(code: usize) noreturn {
95 switch (builtin.stage2_os) {
95 switch (native_os) {
9696 .linux => switch (builtin.stage2_arch) {
9797 .x86_64 => {
9898 asm volatile ("syscall"
src/Compilation.zig-3
......@@ -3671,8 +3671,6 @@ pub fn generateBuiltinZigSource(comp: *Compilation, allocator: *Allocator) Alloc
36713671 \\pub const zig_is_stage2 = {};
36723672 \\/// Temporary until self-hosted supports the `cpu.arch` value.
36733673 \\pub const stage2_arch: std.Target.Cpu.Arch = .{};
3674 \\/// Temporary until self-hosted supports the `os.tag` value.
3675 \\pub const stage2_os: std.Target.Os.Tag = .{};
36763674 \\
36773675 \\pub const output_mode = std.builtin.OutputMode.{};
36783676 \\pub const link_mode = std.builtin.LinkMode.{};
......@@ -3688,7 +3686,6 @@ pub fn generateBuiltinZigSource(comp: *Compilation, allocator: *Allocator) Alloc
36883686 build_options.version,
36893687 !use_stage1,
36903688 std.zig.fmtId(@tagName(target.cpu.arch)),
3691 std.zig.fmtId(@tagName(target.os.tag)),
36923689 std.zig.fmtId(@tagName(comp.bin_file.options.output_mode)),
36933690 std.zig.fmtId(@tagName(comp.bin_file.options.link_mode)),
36943691 comp.bin_file.options.is_test,
src/Sema.zig+3-3
......@@ -6313,7 +6313,7 @@ fn zirTypeInfo(sema: *Sema, block: *Scope.Block, inst: Zir.Inst.Index) CompileEr
63136313 sema.arena,
63146314 @enumToInt(@typeInfo(std.builtin.TypeInfo).Union.tag_type.?.Fn),
63156315 ),
6316 .val = try Value.Tag.@"struct".create(sema.arena, field_values.ptr),
6316 .val = try Value.Tag.@"struct".create(sema.arena, field_values),
63176317 }),
63186318 );
63196319 },
......@@ -6335,7 +6335,7 @@ fn zirTypeInfo(sema: *Sema, block: *Scope.Block, inst: Zir.Inst.Index) CompileEr
63356335 sema.arena,
63366336 @enumToInt(@typeInfo(std.builtin.TypeInfo).Union.tag_type.?.Int),
63376337 ),
6338 .val = try Value.Tag.@"struct".create(sema.arena, field_values.ptr),
6338 .val = try Value.Tag.@"struct".create(sema.arena, field_values),
63396339 }),
63406340 );
63416341 },
......@@ -6943,7 +6943,7 @@ fn zirStructInit(sema: *Sema, block: *Scope.Block, inst: Zir.Inst.Index, is_ref:
69436943 for (field_inits) |field_init, i| {
69446944 values[i] = (sema.resolveMaybeUndefVal(block, src, field_init) catch unreachable).?;
69456945 }
6946 return sema.addConstant(resolved_ty, try Value.Tag.@"struct".create(sema.arena, values.ptr));
6946 return sema.addConstant(resolved_ty, try Value.Tag.@"struct".create(sema.arena, values));
69476947 }
69486948
69496949 return mod.fail(&block.base, src, "TODO: Sema.zirStructInit for runtime-known struct values", .{});
src/value.zig+66-39
......@@ -344,7 +344,9 @@ pub const Value = extern union {
344344 return null;
345345 }
346346
347 pub fn copy(self: Value, allocator: *Allocator) error{OutOfMemory}!Value {
347 /// It's intentional that this function is not passed a corresponding Type, so that
348 /// a Value can be copied from a Sema to a Decl prior to resolving struct/union field types.
349 pub fn copy(self: Value, arena: *Allocator) error{OutOfMemory}!Value {
348350 if (self.tag_if_small_enough < Tag.no_payload_count) {
349351 return Value{ .tag_if_small_enough = self.tag_if_small_enough };
350352 } else switch (self.ptr_otherwise.tag) {
......@@ -421,37 +423,37 @@ pub const Value = extern union {
421423
422424 .ty => {
423425 const payload = self.castTag(.ty).?;
424 const new_payload = try allocator.create(Payload.Ty);
426 const new_payload = try arena.create(Payload.Ty);
425427 new_payload.* = .{
426428 .base = payload.base,
427 .data = try payload.data.copy(allocator),
429 .data = try payload.data.copy(arena),
428430 };
429431 return Value{ .ptr_otherwise = &new_payload.base };
430432 },
431 .int_type => return self.copyPayloadShallow(allocator, Payload.IntType),
432 .int_u64 => return self.copyPayloadShallow(allocator, Payload.U64),
433 .int_i64 => return self.copyPayloadShallow(allocator, Payload.I64),
433 .int_type => return self.copyPayloadShallow(arena, Payload.IntType),
434 .int_u64 => return self.copyPayloadShallow(arena, Payload.U64),
435 .int_i64 => return self.copyPayloadShallow(arena, Payload.I64),
434436 .int_big_positive, .int_big_negative => {
435437 const old_payload = self.cast(Payload.BigInt).?;
436 const new_payload = try allocator.create(Payload.BigInt);
438 const new_payload = try arena.create(Payload.BigInt);
437439 new_payload.* = .{
438440 .base = .{ .tag = self.ptr_otherwise.tag },
439 .data = try allocator.dupe(std.math.big.Limb, old_payload.data),
441 .data = try arena.dupe(std.math.big.Limb, old_payload.data),
440442 };
441443 return Value{ .ptr_otherwise = &new_payload.base };
442444 },
443 .function => return self.copyPayloadShallow(allocator, Payload.Function),
444 .extern_fn => return self.copyPayloadShallow(allocator, Payload.Decl),
445 .variable => return self.copyPayloadShallow(allocator, Payload.Variable),
446 .decl_ref => return self.copyPayloadShallow(allocator, Payload.Decl),
447 .decl_ref_mut => return self.copyPayloadShallow(allocator, Payload.DeclRefMut),
445 .function => return self.copyPayloadShallow(arena, Payload.Function),
446 .extern_fn => return self.copyPayloadShallow(arena, Payload.Decl),
447 .variable => return self.copyPayloadShallow(arena, Payload.Variable),
448 .decl_ref => return self.copyPayloadShallow(arena, Payload.Decl),
449 .decl_ref_mut => return self.copyPayloadShallow(arena, Payload.DeclRefMut),
448450 .elem_ptr => {
449451 const payload = self.castTag(.elem_ptr).?;
450 const new_payload = try allocator.create(Payload.ElemPtr);
452 const new_payload = try arena.create(Payload.ElemPtr);
451453 new_payload.* = .{
452454 .base = payload.base,
453455 .data = .{
454 .array_ptr = try payload.data.array_ptr.copy(allocator),
456 .array_ptr = try payload.data.array_ptr.copy(arena),
455457 .index = payload.data.index,
456458 },
457459 };
......@@ -459,17 +461,17 @@ pub const Value = extern union {
459461 },
460462 .field_ptr => {
461463 const payload = self.castTag(.field_ptr).?;
462 const new_payload = try allocator.create(Payload.FieldPtr);
464 const new_payload = try arena.create(Payload.FieldPtr);
463465 new_payload.* = .{
464466 .base = payload.base,
465467 .data = .{
466 .container_ptr = try payload.data.container_ptr.copy(allocator),
468 .container_ptr = try payload.data.container_ptr.copy(arena),
467469 .field_index = payload.data.field_index,
468470 },
469471 };
470472 return Value{ .ptr_otherwise = &new_payload.base };
471473 },
472 .bytes => return self.copyPayloadShallow(allocator, Payload.Bytes),
474 .bytes => return self.copyPayloadShallow(arena, Payload.Bytes),
473475 .repeated,
474476 .eu_payload,
475477 .eu_payload_ptr,
......@@ -477,61 +479,85 @@ pub const Value = extern union {
477479 .opt_payload_ptr,
478480 => {
479481 const payload = self.cast(Payload.SubValue).?;
480 const new_payload = try allocator.create(Payload.SubValue);
482 const new_payload = try arena.create(Payload.SubValue);
481483 new_payload.* = .{
482484 .base = payload.base,
483 .data = try payload.data.copy(allocator),
485 .data = try payload.data.copy(arena),
484486 };
485487 return Value{ .ptr_otherwise = &new_payload.base };
486488 },
487489 .array => {
488490 const payload = self.castTag(.array).?;
489 const new_payload = try allocator.create(Payload.Array);
491 const new_payload = try arena.create(Payload.Array);
490492 new_payload.* = .{
491493 .base = payload.base,
492 .data = try allocator.alloc(Value, payload.data.len),
494 .data = try arena.alloc(Value, payload.data.len),
493495 };
494496 std.mem.copy(Value, new_payload.data, payload.data);
495497 return Value{ .ptr_otherwise = &new_payload.base };
496498 },
497499 .slice => {
498500 const payload = self.castTag(.slice).?;
499 const new_payload = try allocator.create(Payload.Slice);
501 const new_payload = try arena.create(Payload.Slice);
500502 new_payload.* = .{
501503 .base = payload.base,
502504 .data = .{
503 .ptr = try payload.data.ptr.copy(allocator),
504 .len = try payload.data.len.copy(allocator),
505 .ptr = try payload.data.ptr.copy(arena),
506 .len = try payload.data.len.copy(arena),
505507 },
506508 };
507509 return Value{ .ptr_otherwise = &new_payload.base };
508510 },
509 .float_16 => return self.copyPayloadShallow(allocator, Payload.Float_16),
510 .float_32 => return self.copyPayloadShallow(allocator, Payload.Float_32),
511 .float_64 => return self.copyPayloadShallow(allocator, Payload.Float_64),
512 .float_128 => return self.copyPayloadShallow(allocator, Payload.Float_128),
511 .float_16 => return self.copyPayloadShallow(arena, Payload.Float_16),
512 .float_32 => return self.copyPayloadShallow(arena, Payload.Float_32),
513 .float_64 => return self.copyPayloadShallow(arena, Payload.Float_64),
514 .float_128 => return self.copyPayloadShallow(arena, Payload.Float_128),
513515 .enum_literal => {
514516 const payload = self.castTag(.enum_literal).?;
515 const new_payload = try allocator.create(Payload.Bytes);
517 const new_payload = try arena.create(Payload.Bytes);
516518 new_payload.* = .{
517519 .base = payload.base,
518 .data = try allocator.dupe(u8, payload.data),
520 .data = try arena.dupe(u8, payload.data),
521 };
522 return Value{ .ptr_otherwise = &new_payload.base };
523 },
524 .enum_field_index => return self.copyPayloadShallow(arena, Payload.U32),
525 .@"error" => return self.copyPayloadShallow(arena, Payload.Error),
526
527 .@"struct" => {
528 const old_field_values = self.castTag(.@"struct").?.data;
529 const new_payload = try arena.create(Payload.Struct);
530 new_payload.* = .{
531 .base = .{ .tag = .@"struct" },
532 .data = try arena.alloc(Value, old_field_values.len),
533 };
534 for (old_field_values) |old_field_val, i| {
535 new_payload.data[i] = try old_field_val.copy(arena);
536 }
537 return Value{ .ptr_otherwise = &new_payload.base };
538 },
539
540 .@"union" => {
541 const tag_and_val = self.castTag(.@"union").?.data;
542 const new_payload = try arena.create(Payload.Union);
543 new_payload.* = .{
544 .base = .{ .tag = .@"union" },
545 .data = .{
546 .tag = try tag_and_val.tag.copy(arena),
547 .val = try tag_and_val.val.copy(arena),
548 },
519549 };
520550 return Value{ .ptr_otherwise = &new_payload.base };
521551 },
522 .enum_field_index => return self.copyPayloadShallow(allocator, Payload.U32),
523 .@"error" => return self.copyPayloadShallow(allocator, Payload.Error),
524 .@"struct" => @panic("TODO can't copy struct value without knowing the type"),
525 .@"union" => @panic("TODO can't copy union value without knowing the type"),
526552
527553 .inferred_alloc => unreachable,
528554 .inferred_alloc_comptime => unreachable,
529555 }
530556 }
531557
532 fn copyPayloadShallow(self: Value, allocator: *Allocator, comptime T: type) error{OutOfMemory}!Value {
558 fn copyPayloadShallow(self: Value, arena: *Allocator, comptime T: type) error{OutOfMemory}!Value {
533559 const payload = self.cast(T).?;
534 const new_payload = try allocator.create(T);
560 const new_payload = try arena.create(T);
535561 new_payload.* = payload.*;
536562 return Value{ .ptr_otherwise = &new_payload.base };
537563 }
......@@ -1932,8 +1958,9 @@ pub const Value = extern union {
19321958 pub const base_tag = Tag.@"struct";
19331959
19341960 base: Payload = .{ .tag = base_tag },
1935 /// Field values. The number and type are according to the struct type.
1936 data: [*]Value,
1961 /// Field values. The types are according to the struct type.
1962 /// The length is provided here so that copying a Value does not depend on the Type.
1963 data: []Value,
19371964 };
19381965
19391966 pub const Union = struct {