diff --git a/src/check_expr.cpp b/src/check_expr.cpp index c8778a222..37babf91a 100644 --- a/src/check_expr.cpp +++ b/src/check_expr.cpp @@ -3681,10 +3681,8 @@ bool check_identifier_exists(Scope *s, Ast *node, bool nested = false, Scope **o typedef bool (BuiltinTypeIsProc)(Type *t); -BuiltinTypeIsProc *builtin_type_is_procs[BuiltinProc__type_end - BuiltinProc__type_begin - 1] = { - nullptr, // BuiltinProc_type_base_type - nullptr, // BuiltinProc_type_core_type - nullptr, // BuiltinProc_type_elem_type +BuiltinTypeIsProc *builtin_type_is_procs[BuiltinProc__type_simple_boolean_end - BuiltinProc__type_simple_boolean_begin] = { + nullptr, // BuiltinProc__type_simple_boolean_begin is_type_boolean, is_type_integer, @@ -3725,10 +3723,10 @@ BuiltinTypeIsProc *builtin_type_is_procs[BuiltinProc__type_end - BuiltinProc__ty is_type_bit_set, is_type_simd_vector, - type_has_nil, + is_type_polymorphic_record_specialized, + is_type_polymorphic_record_unspecialized, - nullptr, // BuiltinProc_type_proc_parameter_count - nullptr, // BuiltinProc_type_proc_return_count + type_has_nil, }; @@ -5482,15 +5480,18 @@ bool check_builtin_procedure(CheckerContext *c, Operand *operand, Ast *call, i32 case BuiltinProc_type_is_bit_field_value: case BuiltinProc_type_is_bit_set: case BuiltinProc_type_is_simd_vector: + case BuiltinProc_type_is_specialized_polymorphic_record: + case BuiltinProc_type_is_unspecialized_polymorphic_record: case BuiltinProc_type_has_nil: - GB_ASSERT(BuiltinProc__type_begin < id && id < BuiltinProc__type_end); + GB_ASSERT(BuiltinProc__type_simple_boolean_begin < id && id < BuiltinProc__type_simple_boolean_end); + operand->value = exact_value_bool(false); if (operand->mode != Addressing_Type) { gbString str = expr_to_string(ce->args[0]); error(operand->expr, "Expected a type for '%.*s', got '%s'", LIT(builtin_name), str); gb_string_free(str); } else { - i32 i = id - (BuiltinProc__type_begin+1); + i32 i = id - cast(i32)BuiltinProc__type_simple_boolean_begin; auto procedure = builtin_type_is_procs[i]; GB_ASSERT_MSG(procedure != nullptr, "%.*s", LIT(builtin_name)); operand->value = exact_value_bool(procedure(operand->type)); @@ -5555,6 +5556,95 @@ bool check_builtin_procedure(CheckerContext *c, Operand *operand, Ast *call, i32 operand->mode = Addressing_Constant; operand->type = t_untyped_integer; break; + + case BuiltinProc_type_polymorphic_record_parameter_count: + operand->value = exact_value_i64(0); + if (operand->mode != Addressing_Type) { + error(operand->expr, "Expected a record type for '%.*s'", LIT(builtin_name)); + } else { + Type *bt = base_type(operand->type); + if (bt->kind == Type_Struct) { + if (bt->Struct.polymorphic_params != nullptr) { + operand->value = exact_value_i64(bt->Struct.polymorphic_params->Tuple.variables.count); + } + } else if (bt->kind == Type_Union) { + if (bt->Union.polymorphic_params != nullptr) { + operand->value = exact_value_i64(bt->Union.polymorphic_params->Tuple.variables.count); + } + } else { + error(operand->expr, "Expected a record type for '%.*s'", LIT(builtin_name)); + } + } + operand->mode = Addressing_Constant; + operand->type = t_untyped_integer; + break; + case BuiltinProc_type_polymorphic_record_parameter_value: + if (operand->mode != Addressing_Type) { + error(operand->expr, "Expected a record type for '%.*s'", LIT(builtin_name)); + return false; + } else if (!is_type_polymorphic_record_specialized(operand->type)) { + error(operand->expr, "Expected a specialized polymorphic record type for '%.*s'", LIT(builtin_name)); + return false; + } else { + Operand op = {}; + check_expr(c, &op, ce->args[1]); + if (op.mode != Addressing_Constant && !is_type_integer(op.type)) { + error(op.expr, "Execpted a constant integer for the index of record parameter value"); + return false; + } + + i64 index = exact_value_to_i64(op.value); + if (index < 0) { + error(op.expr, "Execpted a non-negative integer for the index of record parameter value, got %lld", cast(long long)index); + return false; + } + + Entity *param = nullptr; + i64 count = 0; + + Type *bt = base_type(operand->type); + if (bt->kind == Type_Struct) { + if (bt->Struct.polymorphic_params != nullptr) { + count = bt->Struct.polymorphic_params->Tuple.variables.count; + if (index < count) { + param = bt->Struct.polymorphic_params->Tuple.variables[cast(isize)index]; + } + } + } else if (bt->kind == Type_Union) { + if (bt->Union.polymorphic_params != nullptr) { + count = bt->Union.polymorphic_params->Tuple.variables.count; + if (index < count) { + param = bt->Union.polymorphic_params->Tuple.variables[cast(isize)index]; + } + } + } else { + error(operand->expr, "Expected a specialized polymorphic record type for '%.*s'", LIT(builtin_name)); + return false; + } + + if (index >= count) { + error(op.expr, "Index of record parameter value out of bounds, expected 0..<%lld, got %lld", cast(long long)count, cast(long long)index); + return false; + } + GB_ASSERT(param != nullptr); + switch (param->kind) { + case Entity_Constant: + operand->mode = Addressing_Constant; + operand->type = param->type; + operand->value = param->Constant.value; + break; + case Entity_TypeName: + operand->mode = Addressing_Type; + operand->type = param->type; + break; + default: + GB_PANIC("Unhandle polymorphic record type"); + break; + } + + } + + break; } return true; @@ -7085,6 +7175,7 @@ ExprKind check_call_expr(CheckerContext *c, Operand *operand, Ast *call, Type *t i32 id = operand->builtin_id; if (!check_builtin_procedure(c, operand, call, id, type_hint)) { operand->mode = Addressing_Invalid; + operand->type = t_invalid; } operand->expr = call; return builtin_procs[id].kind; diff --git a/src/checker_builtin_procs.hpp b/src/checker_builtin_procs.hpp index add087afe..4981fdedb 100644 --- a/src/checker_builtin_procs.hpp +++ b/src/checker_builtin_procs.hpp @@ -110,12 +110,14 @@ enum BuiltinProcId { // Constant type tests + BuiltinProc__type_begin, BuiltinProc_type_base_type, BuiltinProc_type_core_type, BuiltinProc_type_elem_type, +BuiltinProc__type_simple_boolean_begin, BuiltinProc_type_is_boolean, BuiltinProc_type_is_integer, BuiltinProc_type_is_rune, @@ -155,13 +157,21 @@ BuiltinProc__type_begin, BuiltinProc_type_is_bit_set, BuiltinProc_type_is_simd_vector, - BuiltinProc_type_is_specialization_of, + BuiltinProc_type_is_specialized_polymorphic_record, + BuiltinProc_type_is_unspecialized_polymorphic_record, BuiltinProc_type_has_nil, +BuiltinProc__type_simple_boolean_end, + + BuiltinProc_type_is_specialization_of, + BuiltinProc_type_proc_parameter_count, BuiltinProc_type_proc_return_count, + BuiltinProc_type_polymorphic_record_parameter_count, + BuiltinProc_type_polymorphic_record_parameter_value, + BuiltinProc__type_end, @@ -282,6 +292,7 @@ gb_global BuiltinProc builtin_procs[BuiltinProc_COUNT] = { {STR_LIT("type_core_type"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics}, {STR_LIT("type_elem_type"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics}, + {STR_LIT(""), 0, false, Expr_Stmt, BuiltinProcPkg_intrinsics}, {STR_LIT("type_is_boolean"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics}, {STR_LIT("type_is_integer"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics}, {STR_LIT("type_is_rune"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics}, @@ -321,11 +332,18 @@ gb_global BuiltinProc builtin_procs[BuiltinProc_COUNT] = { {STR_LIT("type_is_bit_set"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics}, {STR_LIT("type_is_simd_vector"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics}, - {STR_LIT("type_is_specialization_of"), 2, false, Expr_Expr, BuiltinProcPkg_intrinsics}, + {STR_LIT("type_is_specialized_polymorphic_record"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics}, + {STR_LIT("type_is_unspecialized_polymorphic_record"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics}, {STR_LIT("type_has_nil"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics}, + {STR_LIT(""), 0, false, Expr_Stmt, BuiltinProcPkg_intrinsics}, + {STR_LIT("type_is_specialization_of"), 2, false, Expr_Expr, BuiltinProcPkg_intrinsics}, {STR_LIT("type_proc_parameter_count"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics}, {STR_LIT("type_proc_return_count"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics}, + + {STR_LIT("type_polymorphic_record_parameter_count"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics}, + {STR_LIT("type_polymorphic_record_parameter_value"), 2, false, Expr_Expr, BuiltinProcPkg_intrinsics}, + {STR_LIT(""), 0, false, Expr_Stmt, BuiltinProcPkg_intrinsics}, }; diff --git a/src/types.cpp b/src/types.cpp index cf8e603ba..e120c77c0 100644 --- a/src/types.cpp +++ b/src/types.cpp @@ -1471,11 +1471,12 @@ Scope *polymorphic_record_parent_scope(Type *t) { } bool is_type_polymorphic_record_specialized(Type *t) { + Type *original_type = t; t = base_type(t); if (t->kind == Type_Struct) { - return t->Struct.is_polymorphic && t->Struct.is_poly_specialized; + return t->Struct.is_poly_specialized; } else if (t->kind == Type_Union) { - return t->Union.is_polymorphic && t->Union.is_poly_specialized; + return t->Union.is_poly_specialized; } return false; }