authorgravatar for shawn@git.icuShawn Landden <shawn@git.icu> 2019-05-06 11:41:15-05:00
committergravatar for shawn@git.icuShawn Landden <shawn@git.icu> 2019-05-06 11:52:47-05:00
logc19bdc2d370f3a72ef492e2f4e6264e2a0dd91ec
tree26c14f9b6b1b0edaee401842a7fe3af34a462ee8
parent7a41af2632e693cd63476576bbbb965126e45b9b

stage1: add @hasField() built-in

This was quite straight-forward Closes: #1439

5 files changed, 94 insertions(+), 0 deletions(-)

src/all_types.hpp+10
......@@ -1394,6 +1394,7 @@ enum BuiltinFnId {
13941394 BuiltinFnIdMemberName,
13951395 BuiltinFnIdField,
13961396 BuiltinFnIdTypeInfo,
1397 BuiltinFnIdHasField,
13971398 BuiltinFnIdTypeof,
13981399 BuiltinFnIdAddWithOverflow,
13991400 BuiltinFnIdSubWithOverflow,
......@@ -2251,6 +2252,7 @@ enum IrInstructionId {
22512252 IrInstructionIdByteOffsetOf,
22522253 IrInstructionIdBitOffsetOf,
22532254 IrInstructionIdTypeInfo,
2255 IrInstructionIdHasField,
22542256 IrInstructionIdTypeId,
22552257 IrInstructionIdSetEvalBranchQuota,
22562258 IrInstructionIdPtrType,
......@@ -3229,6 +3231,14 @@ struct IrInstructionTypeInfo {
32293231 IrInstruction *type_value;
32303232};
32313233
3234struct IrInstructionHasField {
3235 IrInstruction base;
3236
3237 IrInstruction *container_type;
3238 Buf *field_name_buffer;
3239 IrInstruction *field_name_expr;
3240};
3241
32323242struct IrInstructionTypeId {
32333243 IrInstruction base;
32343244
src/codegen.cpp+2
......@@ -5530,6 +5530,7 @@ static LLVMValueRef ir_render_instruction(CodeGen *g, IrExecutable *executable,
55305530 case IrInstructionIdByteOffsetOf:
55315531 case IrInstructionIdBitOffsetOf:
55325532 case IrInstructionIdTypeInfo:
5533 case IrInstructionIdHasField:
55335534 case IrInstructionIdTypeId:
55345535 case IrInstructionIdSetEvalBranchQuota:
55355536 case IrInstructionIdPtrType:
......@@ -7271,6 +7272,7 @@ static void define_builtin_fns(CodeGen *g) {
72717272 create_builtin_fn(g, BuiltinFnIdMemberName, "memberName", 2);
72727273 create_builtin_fn(g, BuiltinFnIdField, "field", 2);
72737274 create_builtin_fn(g, BuiltinFnIdTypeInfo, "typeInfo", 1);
7275 create_builtin_fn(g, BuiltinFnIdHasField, "hasField", 2);
72747276 create_builtin_fn(g, BuiltinFnIdTypeof, "typeOf", 1); // TODO rename to TypeOf
72757277 create_builtin_fn(g, BuiltinFnIdAddWithOverflow, "addWithOverflow", 4);
72767278 create_builtin_fn(g, BuiltinFnIdSubWithOverflow, "subWithOverflow", 4);
src/ir.cpp+70
......@@ -851,6 +851,10 @@ static constexpr IrInstructionId ir_instruction_id(IrInstructionTypeInfo *) {
851851 return IrInstructionIdTypeInfo;
852852}
853853
854static constexpr IrInstructionId ir_instruction_id(IrInstructionHasField *) {
855 return IrInstructionIdHasField;
856}
857
854858static constexpr IrInstructionId ir_instruction_id(IrInstructionTypeId *) {
855859 return IrInstructionIdTypeId;
856860}
......@@ -1280,6 +1284,20 @@ static IrInstruction *ir_build_field_ptr(IrBuilder *irb, Scope *scope, AstNode *
12801284 return &instruction->base;
12811285}
12821286
1287static IrInstruction *ir_build_has_field(IrBuilder *irb, Scope *scope, AstNode *source_node,
1288 IrInstruction *container_type, IrInstruction *field_name_expr)
1289{
1290 IrInstructionHasField *instruction = ir_build_instruction<IrInstructionHasField>(irb, scope, source_node);
1291 instruction->container_type = container_type;
1292 instruction->field_name_buffer = nullptr;
1293 instruction->field_name_expr = field_name_expr;
1294
1295 ir_ref_instruction(container_type, irb->current_basic_block);
1296 ir_ref_instruction(field_name_expr, irb->current_basic_block);
1297
1298 return &instruction->base;
1299}
1300
12831301static IrInstruction *ir_build_struct_field_ptr(IrBuilder *irb, Scope *scope, AstNode *source_node,
12841302 IrInstruction *struct_ptr, TypeStructField *field)
12851303{
......@@ -4598,6 +4616,21 @@ static IrInstruction *ir_gen_builtin_fn_call(IrBuilder *irb, Scope *scope, AstNo
45984616
45994617 return ir_build_load_ptr(irb, scope, node, ptr_instruction);
46004618 }
4619 case BuiltinFnIdHasField:
4620 {
4621 AstNode *arg0_node = node->data.fn_call_expr.params.at(0);
4622 IrInstruction *arg0_value = ir_gen_node(irb, arg0_node, scope);
4623 if (arg0_value == irb->codegen->invalid_instruction)
4624 return arg0_value;
4625
4626 AstNode *arg1_node = node->data.fn_call_expr.params.at(1);
4627 IrInstruction *arg1_value = ir_gen_node(irb, arg1_node, scope);
4628 if (arg1_value == irb->codegen->invalid_instruction)
4629 return arg1_value;
4630
4631 IrInstruction *type_info = ir_build_has_field(irb, scope, node, arg0_value, arg1_value);
4632 return ir_lval_wrap(irb, scope, type_info, lval);
4633 }
46014634 case BuiltinFnIdTypeInfo:
46024635 {
46034636 AstNode *arg0_node = node->data.fn_call_expr.params.at(0);
......@@ -20578,6 +20611,40 @@ static IrInstruction *ir_analyze_instruction_member_name(IrAnalyze *ira, IrInstr
2057820611 }
2057920612}
2058020613
20614static IrInstruction *ir_analyze_instruction_has_field(IrAnalyze *ira, IrInstructionHasField *instruction) {
20615 Error err;
20616 IrInstruction *container_type_value = instruction->container_type->child;
20617 ZigType *container_type = ir_resolve_type(ira, container_type_value);
20618 if (type_is_invalid(container_type))
20619 return ira->codegen->invalid_instruction;
20620
20621 if ((err = ensure_complete_type(ira->codegen, container_type)))
20622 return ira->codegen->invalid_instruction;
20623
20624 Buf *field_name = instruction->field_name_buffer;
20625 if (!field_name) {
20626 IrInstruction *field_name_expr = instruction->field_name_expr->child;
20627 field_name = ir_resolve_str(ira, field_name_expr);
20628 if (!field_name)
20629 return ira->codegen->invalid_instruction;
20630 }
20631
20632 bool result;
20633 if (container_type->id == ZigTypeIdStruct)
20634 result = (bool)find_struct_type_field(container_type, field_name);
20635 else if (container_type->id == ZigTypeIdEnum)
20636 result = (bool)find_enum_type_field(container_type, field_name);
20637 else if (container_type->id == ZigTypeIdUnion)
20638 result = (bool)find_union_type_field(container_type, field_name);
20639 else {
20640 ir_add_error(ira, container_type_value,
20641 buf_sprintf("type '%s' does not support @memberName", buf_ptr(&container_type->name)));
20642 return ira->codegen->invalid_instruction;
20643 }
20644 return ir_build_const_bool(&ira->new_irb,
20645 instruction->base.scope, instruction->base.source_node, result);
20646}
20647
2058120648static IrInstruction *ir_analyze_instruction_breakpoint(IrAnalyze *ira, IrInstructionBreakpoint *instruction) {
2058220649 IrInstruction *result = ir_build_breakpoint(&ira->new_irb,
2058320650 instruction->base.scope, instruction->base.source_node);
......@@ -23068,6 +23135,8 @@ static IrInstruction *ir_analyze_instruction_nocast(IrAnalyze *ira, IrInstructio
2306823135 return ir_analyze_instruction_bit_offset_of(ira, (IrInstructionBitOffsetOf *)instruction);
2306923136 case IrInstructionIdTypeInfo:
2307023137 return ir_analyze_instruction_type_info(ira, (IrInstructionTypeInfo *) instruction);
23138 case IrInstructionIdHasField:
23139 return ir_analyze_instruction_has_field(ira, (IrInstructionHasField *) instruction);
2307123140 case IrInstructionIdTypeId:
2307223141 return ir_analyze_instruction_type_id(ira, (IrInstructionTypeId *)instruction);
2307323142 case IrInstructionIdSetEvalBranchQuota:
......@@ -23355,6 +23424,7 @@ bool ir_has_side_effects(IrInstruction *instruction) {
2335523424 case IrInstructionIdByteOffsetOf:
2335623425 case IrInstructionIdBitOffsetOf:
2335723426 case IrInstructionIdTypeInfo:
23427 case IrInstructionIdHasField:
2335823428 case IrInstructionIdTypeId:
2335923429 case IrInstructionIdAlignCast:
2336023430 case IrInstructionIdOpaqueType:
src/ir_print.cpp+11
......@@ -1107,6 +1107,14 @@ static void ir_print_type_info(IrPrint *irp, IrInstructionTypeInfo *instruction)
11071107 fprintf(irp->f, ")");
11081108}
11091109
1110static void ir_print_has_field(IrPrint *irp, IrInstructionHasField *instruction) {
1111 fprintf(irp->f, "@hasField(");
1112 ir_print_other_instruction(irp, instruction->container_type);
1113 fprintf(irp->f, ",");
1114 ir_print_other_instruction(irp, instruction->field_name_expr);
1115 fprintf(irp->f, ")");
1116}
1117
11101118static void ir_print_type_id(IrPrint *irp, IrInstructionTypeId *instruction) {
11111119 fprintf(irp->f, "@typeId(");
11121120 ir_print_other_instruction(irp, instruction->type_value);
......@@ -1757,6 +1765,9 @@ static void ir_print_instruction(IrPrint *irp, IrInstruction *instruction) {
17571765 case IrInstructionIdTypeInfo:
17581766 ir_print_type_info(irp, (IrInstructionTypeInfo *)instruction);
17591767 break;
1768 case IrInstructionIdHasField:
1769 ir_print_has_field(irp, (IrInstructionHasField *)instruction);
1770 break;
17601771 case IrInstructionIdTypeId:
17611772 ir_print_type_id(irp, (IrInstructionTypeId *)instruction);
17621773 break;
test/stage1/behavior.zig+1
......@@ -94,4 +94,5 @@ comptime {
9494 _ = @import("behavior/void.zig");
9595 _ = @import("behavior/while.zig");
9696 _ = @import("behavior/widening.zig");
97 _ = @import("behavior/hasfield.zig");
9798}