authorgravatar for git@vexu.euVeikka Tuominen <git@vexu.eu> 2020-08-28 14:35:13+03:00
committergravatar for git@vexu.euVeikka Tuominen <git@vexu.eu> 2020-09-03 15:05:43+03:00
logff7c6e1e3cea86e130e15a720c729a05763b5f08
treeab2343a0c5b94a7a8e28cdf963be40f8ea98b622
parent9a59cdcd41f5a05d70a02d89178afaf8789791c6
signature Commit is signed but in an unrecognized format.

stage2: astgen orelse

1 files changed, 43 insertions(+), 25 deletions(-)

src-self-hosted/astgen.zig+43-25
......@@ -277,10 +277,10 @@ pub fn expr(mod: *Module, scope: *Scope, rl: ResultLoc, node: *ast.Node) InnerEr
277277 .ArrayAccess => return arrayAccess(mod, scope, rl, node.castTag(.ArrayAccess).?),
278278 .Catch => return catchExpr(mod, scope, rl, node.castTag(.Catch).?),
279279 .Comptime => return comptimeKeyword(mod, scope, rl, node.castTag(.Comptime).?),
280 .OrElse => return orelseExpr(mod, scope, rl, node.castTag(.OrElse).?),
280281
281282 .Defer => return mod.failNode(scope, node, "TODO implement astgen.expr for .Defer", .{}),
282283 .Range => return mod.failNode(scope, node, "TODO implement astgen.expr for .Range", .{}),
283 .OrElse => return mod.failNode(scope, node, "TODO implement astgen.expr for .OrElse", .{}),
284284 .Await => return mod.failNode(scope, node, "TODO implement astgen.expr for .Await", .{}),
285285 .Resume => return mod.failNode(scope, node, "TODO implement astgen.expr for .Resume", .{}),
286286 .Try => return mod.failNode(scope, node, "TODO implement astgen.expr for .Try", .{}),
......@@ -790,13 +790,31 @@ fn errorType(mod: *Module, scope: *Scope, node: *ast.Node.OneToken) InnerError!*
790790}
791791
792792fn catchExpr(mod: *Module, scope: *Scope, rl: ResultLoc, node: *ast.Node.Catch) InnerError!*zir.Inst {
793 return orelseCatchExpr(mod, scope, rl, node.lhs, node.op_token, .iserr, .unwrap_err_unsafe, node.rhs, node.payload);
794}
795
796fn orelseExpr(mod: *Module, scope: *Scope, rl: ResultLoc, node: *ast.Node.SimpleInfixOp) InnerError!*zir.Inst {
797 return orelseCatchExpr(mod, scope, rl, node.lhs, node.op_token, .isnull, .unwrap_optional_unsafe, node.rhs, null);
798}
799
800fn orelseCatchExpr(
801 mod: *Module,
802 scope: *Scope,
803 rl: ResultLoc,
804 lhs: *ast.Node,
805 op_token: ast.TokenIndex,
806 cond_op: zir.Inst.Tag,
807 unwrap_op: zir.Inst.Tag,
808 rhs: *ast.Node,
809 payload_node: ?*ast.Node,
810) InnerError!*zir.Inst {
793811 const tree = scope.tree();
794 const src = tree.token_locs[node.op_token].start;
812 const src = tree.token_locs[op_token].start;
795813
796 const err_union_ptr = try expr(mod, scope, .ref, node.lhs);
797 // TODO we could avoid an unnecessary copy if .iserr took a pointer
798 const err_union = try addZIRUnOp(mod, scope, src, .deref, err_union_ptr);
799 const cond = try addZIRUnOp(mod, scope, src, .iserr, err_union);
814 const operand_ptr = try expr(mod, scope, .ref, lhs);
815 // TODO we could avoid an unnecessary copy if .iserr, .isnull took a pointer
816 const err_union = try addZIRUnOp(mod, scope, src, .deref, operand_ptr);
817 const cond = try addZIRUnOp(mod, scope, src, cond_op, err_union);
800818
801819 var block_scope: Scope.GenZIR = .{
802820 .parent = scope,
......@@ -825,55 +843,55 @@ fn catchExpr(mod: *Module, scope: *Scope, rl: ResultLoc, node: *ast.Node.Catch)
825843 .inferred_ptr, .bitcasted_ptr, .block_ptr => .{ .block_ptr = block },
826844 };
827845
828 var err_scope: Scope.GenZIR = .{
846 var then_scope: Scope.GenZIR = .{
829847 .parent = scope,
830848 .decl = block_scope.decl,
831849 .arena = block_scope.arena,
832850 .instructions = .{},
833851 };
834 defer err_scope.instructions.deinit(mod.gpa);
852 defer then_scope.instructions.deinit(mod.gpa);
835853
836854 var err_val_scope: Scope.LocalVal = undefined;
837 const err_sub_scope = blk: {
838 const payload = node.payload orelse
839 break :blk &err_scope.base;
855 const then_sub_scope = blk: {
856 const payload = payload_node orelse
857 break :blk &then_scope.base;
840858
841859 const err_name = tree.tokenSlice(payload.castTag(.Payload).?.error_symbol.firstToken());
842860 if (mem.eql(u8, err_name, "_"))
843 break :blk &err_scope.base;
861 break :blk &then_scope.base;
844862
845 const unwrapped_err_ptr = try addZIRUnOp(mod, &err_scope.base, src, .unwrap_err_code, err_union_ptr);
863 const unwrapped_err_ptr = try addZIRUnOp(mod, &then_scope.base, src, .unwrap_err_code, operand_ptr);
846864 err_val_scope = .{
847 .parent = &err_scope.base,
848 .gen_zir = &err_scope,
865 .parent = &then_scope.base,
866 .gen_zir = &then_scope,
849867 .name = err_name,
850 .inst = try addZIRUnOp(mod, &err_scope.base, src, .deref, unwrapped_err_ptr),
868 .inst = try addZIRUnOp(mod, &then_scope.base, src, .deref, unwrapped_err_ptr),
851869 };
852870 break :blk &err_val_scope.base;
853871 };
854872
855 _ = try addZIRInst(mod, &err_scope.base, src, zir.Inst.Break, .{
873 _ = try addZIRInst(mod, &then_scope.base, src, zir.Inst.Break, .{
856874 .block = block,
857 .operand = try expr(mod, err_sub_scope, branch_rl, node.rhs),
875 .operand = try expr(mod, then_sub_scope, branch_rl, rhs),
858876 }, .{});
859877
860 var not_err_scope: Scope.GenZIR = .{
878 var else_scope: Scope.GenZIR = .{
861879 .parent = scope,
862880 .decl = block_scope.decl,
863881 .arena = block_scope.arena,
864882 .instructions = .{},
865883 };
866 defer not_err_scope.instructions.deinit(mod.gpa);
884 defer else_scope.instructions.deinit(mod.gpa);
867885
868 const unwrapped_payload = try addZIRUnOp(mod, &not_err_scope.base, src, .unwrap_err_unsafe, err_union_ptr);
869 _ = try addZIRInst(mod, &not_err_scope.base, src, zir.Inst.Break, .{
886 const unwrapped_payload = try addZIRUnOp(mod, &else_scope.base, src, unwrap_op, operand_ptr);
887 _ = try addZIRInst(mod, &else_scope.base, src, zir.Inst.Break, .{
870888 .block = block,
871889 .operand = unwrapped_payload,
872890 }, .{});
873891
874 condbr.positionals.then_body = .{ .instructions = try err_scope.arena.dupe(*zir.Inst, err_scope.instructions.items) };
875 condbr.positionals.else_body = .{ .instructions = try not_err_scope.arena.dupe(*zir.Inst, not_err_scope.instructions.items) };
876 return rlWrap(mod, scope, rl, &block.base);
892 condbr.positionals.then_body = .{ .instructions = try then_scope.arena.dupe(*zir.Inst, then_scope.instructions.items) };
893 condbr.positionals.else_body = .{ .instructions = try else_scope.arena.dupe(*zir.Inst, else_scope.instructions.items) };
894 return rlWrapPtr(mod, scope, rl, &block.base);
877895}
878896
879897/// Return whether the identifier names of two tokens are equal. Resolves @"" tokens without allocating.