Files
c3c/src/compiler/types.c

2190 lines
55 KiB
C

// Copyright (c) 2019 Christoffer Lerno. All rights reserved.
// Use of this source code is governed by the GNU LGPLv3.0 license
// a copy of which can be found in the LICENSE file.
#include "compiler_internal.h"
static Type *flatten_raw_function_type(Type *type);
static struct
{
Type u0, u1, i8, i16, i32, i64, i128;
Type u8, u16, u32, u64, u128;
Type f16, f32, f64, f128;
Type usz, isz, uptr, iptr;
Type string;
Type voidstar, typeid, anyfault, member, typeinfo, untyped_list;
Type any, wildcard;
} t;
Type *type_bool = &t.u1;
Type *type_void = &t.u0;
Type *type_voidptr = &t.voidstar;
Type *type_float16 = &t.f16;
Type *type_float = &t.f32;
Type *type_double = &t.f64;
Type *type_f128 = &t.f128;
Type *type_typeid = &t.typeid;
Type *type_any = &t.any;
Type *type_typeinfo = &t.typeinfo;
Type *type_ichar = &t.i8;
Type *type_short = &t.i16;
Type *type_int = &t.i32;
Type *type_long = &t.i64;
Type *type_i128 = &t.i128;
Type *type_iptr = &t.iptr;
Type *type_isz = &t.isz;
Type *type_char = &t.u8;
Type *type_ushort = &t.u16;
Type *type_uint = &t.u32;
Type *type_ulong = &t.u64;
Type *type_u128 = &t.u128;
Type *type_uptr = &t.uptr;
Type *type_usz = &t.usz;
Type *type_anyfault = &t.anyfault;
Type *type_untypedlist = &t.untyped_list;
Type *type_wildcard = &t.wildcard;
Type *type_member = &t.member;
Type *type_chars = NULL;
Type *type_wildcard_optional = NULL;
Type *type_string = &t.string;
static unsigned size_subarray;
static AlignSize alignment_subarray;
static AlignSize max_alignment_vector;
#define PTR_OFFSET 0
#define INFERRED_ARRAY_OFFSET 1
#define FLEXIBLE_ARRAY_OFFSET 2
#define SUB_ARRAY_OFFSET 3
#define INFERRED_VECTOR_OFFSET 4
#define OPTIONAL_OFFSET 5
#define ARRAY_OFFSET 6
Type *type_cint;
Type *type_cuint;
void type_init_cint(void)
{
type_cint = type_int_signed_by_bitsize(platform_target.width_c_int);
type_cuint = type_int_unsigned_by_bitsize(platform_target.width_c_int);
}
Type *type_int_signed_by_bitsize(BitSize bitsize)
{
switch (bitsize)
{
case 8: return type_ichar;
case 16: return type_short;
case 32: return type_int;
case 64: return type_long;
case 128: return type_i128;
default: FATAL_ERROR("Illegal bitsize %d", bitsize);
}
}
Type *type_int_unsigned_by_bitsize(BitSize bit_size)
{
switch (bit_size)
{
case 8: return type_char;
case 16: return type_ushort;
case 32: return type_uint;
case 64: return type_ulong;
case 128: return type_u128;
default: FATAL_ERROR("Illegal bitsize %d", bit_size);
}
}
const char *type_quoted_error_string(Type *type)
{
if (type->canonical != type)
{
return str_printf("'%s' (%s)", type_to_error_string(type), type_to_error_string(type->canonical));
}
return str_printf("'%s'", type_to_error_string(type));
}
static void type_append_func_to_scratch(FunctionPrototype *prototype);
static void type_append_name_to_scratch(Type *type)
{
type = type->canonical;
switch (type->type_kind)
{
case TYPE_POISONED:
case TYPE_TYPEDEF:
UNREACHABLE;
case TYPE_FAULTTYPE:
case TYPE_ENUM:
case TYPE_STRUCT:
case TYPE_UNION:
case TYPE_DISTINCT:
case TYPE_BITSTRUCT:
scratch_buffer_append(type->decl->name);
break;
case TYPE_POINTER:
type_append_name_to_scratch(type->pointer);
scratch_buffer_append_char('*');
break;
case TYPE_OPTIONAL:
if (type->optional)
{
type_append_name_to_scratch(type->optional);
}
else
{
scratch_buffer_append("void");
}
scratch_buffer_append_char('!');
break;
case TYPE_SUBARRAY:
type_append_name_to_scratch(type->array.base);
scratch_buffer_append("[]");
break;
case TYPE_FLEXIBLE_ARRAY:
type_append_name_to_scratch(type->array.base);
scratch_buffer_append("[*]");
break;
case TYPE_VOID:
case TYPE_BOOL:
case ALL_INTS:
case ALL_FLOATS:
case TYPE_TYPEID:
case TYPE_ANYFAULT:
case TYPE_ANY:
case TYPE_VECTOR:
scratch_buffer_append(type->name);
break;
case TYPE_UNTYPED_LIST:
case TYPE_INFERRED_ARRAY:
case TYPE_INFERRED_VECTOR:
case TYPE_TYPEINFO:
case TYPE_MEMBER:
case TYPE_WILDCARD:
UNREACHABLE
break;
case TYPE_FUNC:
type_append_func_to_scratch(type->function.prototype);
break;
case TYPE_ARRAY:
type_append_name_to_scratch(type->array.base);
scratch_buffer_append_char('[');
scratch_buffer_append_signed_int(type->array.len);
scratch_buffer_append_char(']');
break;
}
}
static void type_append_func_to_scratch(FunctionPrototype *prototype)
{
type_append_name_to_scratch(prototype->rtype);
scratch_buffer_append_char('(');
unsigned elements = vec_size(prototype->param_types);
for (unsigned i = 0; i < elements; i++)
{
if (i > 0)
{
scratch_buffer_append_char(',');
}
type_append_name_to_scratch(prototype->param_types[i]);
}
if (prototype->raw_variadic && elements > 0)
{
scratch_buffer_append_char(',');
}
scratch_buffer_append_char(')');
}
const char *type_to_error_string(Type *type)
{
switch (type->type_kind)
{
case TYPE_POISONED:
return "poisoned";
case TYPE_VOID:
case TYPE_BOOL:
case ALL_INTS:
case ALL_FLOATS:
case TYPE_ANYFAULT:
case TYPE_UNTYPED_LIST:
case TYPE_ANY:
case TYPE_MEMBER:
case TYPE_WILDCARD:
return type->name;
case TYPE_ENUM:
case TYPE_FAULTTYPE:
case TYPE_TYPEDEF:
case TYPE_STRUCT:
case TYPE_UNION:
case TYPE_DISTINCT:
case TYPE_BITSTRUCT:
{
Decl *decl = type->decl;
if (!decl || !decl->unit || !decl->unit->module->generic_suffix) return type->name;
scratch_buffer_clear();
scratch_buffer_append(decl->name);
scratch_buffer_append(decl->unit->module->generic_suffix);
return scratch_buffer_copy();
}
case TYPE_FUNC:
scratch_buffer_clear();
scratch_buffer_append("fn ");
type_append_func_to_scratch(type->function.prototype);
return scratch_buffer_copy();
case TYPE_INFERRED_VECTOR:
return str_printf("%s[<*>]", type_to_error_string(type->array.base));
case TYPE_VECTOR:
return str_printf("%s[<%llu>]", type_to_error_string(type->array.base), (unsigned long long)type->array.len);
case TYPE_TYPEINFO:
return "typeinfo";
case TYPE_TYPEID:
return "typeid";
case TYPE_POINTER:
if (type->pointer->type_kind == TYPE_FUNC)
{
return type_to_error_string(type->pointer);
}
return str_printf("%s*", type_to_error_string(type->pointer));
case TYPE_OPTIONAL:
if (!type->optional) return "void!";
return str_printf("%s!", type_to_error_string(type->optional));
case TYPE_ARRAY:
return str_printf("%s[%llu]", type_to_error_string(type->array.base), (unsigned long long)type->array.len);
case TYPE_INFERRED_ARRAY:
case TYPE_FLEXIBLE_ARRAY:
return str_printf("%s[*]", type_to_error_string(type->array.base));
case TYPE_SUBARRAY:
return str_printf("%s[]", type_to_error_string(type->array.base));
}
UNREACHABLE
}
TypeSize type_size(Type *type)
{
RETRY:
switch (type->type_kind)
{
case TYPE_BITSTRUCT:
assert(type->decl->resolve_status == RESOLVE_DONE);
type = type->decl->bitstruct.base_type->type;
goto RETRY;
case TYPE_DISTINCT:
assert(type->decl->resolve_status == RESOLVE_DONE);
type = type->decl->distinct_decl.base_type;
goto RETRY;
case TYPE_VECTOR:
{
TypeSize width = type_size(type->array.base) * type->array.len;
if (width & (width - 1))
{
AlignSize alignment = next_highest_power_of_2((uint32_t) width);
width = aligned_offset((AlignSize)width, alignment);
}
return width;
}
case CT_TYPES:
UNREACHABLE;
case TYPE_FLEXIBLE_ARRAY:
return 0;
case TYPE_OPTIONAL:
type = type->optional;
goto RETRY;
case TYPE_TYPEDEF:
type = type->canonical;
goto RETRY;
case TYPE_FAULTTYPE:
type = type_iptr->canonical;
goto RETRY;
case TYPE_ENUM:
assert(type->decl->enums.type_info->resolve_status == RESOLVE_DONE);
type = type->decl->enums.type_info->type->canonical;
goto RETRY;
case TYPE_STRUCT:
case TYPE_UNION:
assert(type->decl->resolve_status == RESOLVE_DONE);
return type->decl->strukt.size;
case TYPE_VOID:
return 1;
case TYPE_BOOL:
case TYPE_TYPEID:
case ALL_INTS:
case ALL_FLOATS:
case TYPE_ANYFAULT:
case TYPE_ANY:
return type->builtin.bytesize;
case TYPE_FUNC:
case TYPE_POINTER:
return t.iptr.canonical->builtin.bytesize;
case TYPE_ARRAY:
return type_size(type->array.base) * type->array.len;
case TYPE_SUBARRAY:
return size_subarray;
}
UNREACHABLE
}
FunctionPrototype *type_get_resolved_prototype(Type *type)
{
assert(type->type_kind == TYPE_FUNC);
FunctionPrototype *prototype = type->function.prototype;
if (!prototype->is_resolved) c_abi_func_create(prototype);
return prototype;
}
bool type_flat_is_numlike(Type *type)
{
type = type_flatten(type);
if (type->type_kind == TYPE_VECTOR) type = type->array.base;
TypeKind kind = type->type_kind;
return kind >= TYPE_NUM_FIRST && kind <= TYPE_NUM_LAST;
}
bool type_flat_is_floatlike(Type *type)
{
type = type_flatten(type);
if (type->type_kind == TYPE_VECTOR) type = type->array.base;
TypeKind kind = type->type_kind;
return kind >= TYPE_FLOAT_FIRST && kind <= TYPE_FLOAT_LAST;
}
bool type_flat_is_intlike(Type *type)
{
type = type_flatten(type);
if (type->type_kind == TYPE_VECTOR) type = type->array.base;
TypeKind kind = type->type_kind;
return kind >= TYPE_INTEGER_FIRST && kind <= TYPE_INTEGER_LAST;
}
bool type_flat_is_boolintlike(Type *type)
{
type = type_flatten(type);
if (type->type_kind == TYPE_VECTOR) type = type->array.base;
TypeKind kind = type->type_kind;
return kind == TYPE_BOOL || (kind >= TYPE_INTEGER_FIRST && kind <= TYPE_INTEGER_LAST);
}
bool type_is_int128(Type *type)
{
TypeKind kind = type->canonical->type_kind;
return kind == TYPE_U128 || kind == TYPE_I128;
}
bool type_is_abi_aggregate(Type *type)
{
RETRY:
switch (type->type_kind)
{
case TYPE_OPTIONAL:
type = type->optional;
goto RETRY;
case TYPE_DISTINCT:
type = type->decl->distinct_decl.base_type;
goto RETRY;
case TYPE_TYPEDEF:
type = type->canonical;
goto RETRY;
case TYPE_BITSTRUCT:
case ALL_FLOATS:
case TYPE_VOID:
case ALL_INTS:
case TYPE_BOOL:
case TYPE_TYPEID:
case TYPE_POINTER:
case TYPE_ENUM:
case TYPE_FUNC:
case TYPE_VECTOR:
case TYPE_ANYFAULT:
case TYPE_FAULTTYPE:
return false;
case TYPE_STRUCT:
case TYPE_UNION:
case TYPE_SUBARRAY:
case TYPE_ARRAY:
case TYPE_ANY:
return true;
case CT_TYPES:
case TYPE_FLEXIBLE_ARRAY:
return false;
}
UNREACHABLE
}
Type *type_find_largest_union_element(Type *type)
{
assert(type->type_kind == TYPE_UNION);
ByteSize largest = 0;
Type *largest_type = NULL;
Decl **members = type->decl->strukt.members;
VECEACH(members, i)
{
if (type_size(type) > largest)
{
largest = type_size(type);
largest_type = type->canonical;
}
}
return largest_type;
}
bool type_is_ordered(Type *type)
{
RETRY:
switch (type->type_kind)
{
case ALL_FLOATS:
case ALL_INTS:
case TYPE_POINTER:
case TYPE_BOOL:
case TYPE_ENUM:
return true;
case TYPE_TYPEDEF:
type = type->canonical;
goto RETRY;
case TYPE_DISTINCT:
type = type->decl->distinct_decl.base_type;
goto RETRY;
default:
return false;
}
}
bool type_is_comparable(Type *type)
{
RETRY:
switch (type->type_kind)
{
case TYPE_INFERRED_VECTOR:
case TYPE_INFERRED_ARRAY:
case TYPE_POISONED:
UNREACHABLE
case TYPE_VOID:
case TYPE_UNION:
case TYPE_STRUCT:
case TYPE_BITSTRUCT:
case TYPE_FLEXIBLE_ARRAY:
case TYPE_OPTIONAL:
case TYPE_MEMBER:
return false;
case TYPE_TYPEDEF:
type = type->canonical;
goto RETRY;
case TYPE_SUBARRAY:
case TYPE_ARRAY:
// Arrays are comparable if elements are
type = type->array.base;
goto RETRY;
case TYPE_DISTINCT:
type = type->decl->distinct_decl.base_type;
goto RETRY;
case TYPE_BOOL:
case ALL_INTS:
case ALL_FLOATS:
case TYPE_ANY:
case TYPE_ANYFAULT:
case TYPE_TYPEID:
case TYPE_POINTER:
case TYPE_ENUM:
case TYPE_FUNC:
case TYPE_FAULTTYPE:
case TYPE_UNTYPED_LIST:
case TYPE_TYPEINFO:
case TYPE_VECTOR:
case TYPE_WILDCARD:
return true;
}
UNREACHABLE
}
void type_mangle_introspect_name_to_buffer(Type *type)
{
switch (type->type_kind)
{
case CT_TYPES:
UNREACHABLE
case TYPE_VOID:
case TYPE_BOOL:
case ALL_INTS:
case ALL_FLOATS:
case TYPE_ANY:
case TYPE_ANYFAULT:
case TYPE_TYPEID:
scratch_buffer_append(type->name);
return;
case TYPE_POINTER:
scratch_buffer_append("p$");
type_mangle_introspect_name_to_buffer(type->pointer);
return;
case TYPE_SUBARRAY:
scratch_buffer_append("sa$");
type_mangle_introspect_name_to_buffer(type->array.base);
return;
case TYPE_FLEXIBLE_ARRAY:
scratch_buffer_append("a0$");
type_mangle_introspect_name_to_buffer(type->array.base);
return;
case TYPE_OPTIONAL:
scratch_buffer_append("f$");
type_mangle_introspect_name_to_buffer(type->optional);
return;
case TYPE_VECTOR:
scratch_buffer_append_char('v');
scratch_buffer_append_unsigned_int(type->array.len);
scratch_buffer_append_char('$');
type_mangle_introspect_name_to_buffer(type->array.base);
return;
case TYPE_ARRAY:
scratch_buffer_append_char('a');
scratch_buffer_append_unsigned_int(type->array.len);
scratch_buffer_append_char('$');
type_mangle_introspect_name_to_buffer(type->array.base);
return;
case TYPE_FUNC:
type = type->function.prototype->raw_type;
if (type->function.module)
{
Module *module = type->function.module;
scratch_buffer_append(module->extname ? module->extname : module->name->module);
scratch_buffer_append_char('$');
scratch_buffer_append(type->name);
}
else
{
size_t len = strlen(type->name);
for (size_t i = 0; i < len; i++)
{
char c = type->name[i];
if (char_is_alphanum_(c))
{
scratch_buffer_append_char(c);
continue;
}
if (c == '$')
{
scratch_buffer_append("$$");
continue;
}
else
{
scratch_buffer_append_char('$');
}
}
}
return;
case TYPE_ENUM:
case TYPE_STRUCT:
case TYPE_UNION:
case TYPE_BITSTRUCT:
case TYPE_FAULTTYPE:
case TYPE_DISTINCT:
scratch_buffer_append(type->decl->extname);
return;
case TYPE_TYPEDEF:
type_mangle_introspect_name_to_buffer(type->canonical);
return;
}
UNREACHABLE
}
bool type_func_match(Type *fn_type, Type *rtype, unsigned arg_count, ...)
{
assert(type_is_func_ptr(fn_type));
Signature *sig = fn_type->pointer->function.signature;
if (rtype->canonical != typeinfotype(sig->rtype)->canonical) return false;
if (vec_size(sig->params) != arg_count) return false;
va_list ap;
va_start(ap, arg_count);
FOREACH_BEGIN(Decl* decl, sig->params)
Type *arg = va_arg(ap, Type*);
if (decl->type->canonical != arg->canonical)
{
va_end(ap);
return false;
}
FOREACH_END();
va_end(ap);
return true;
}
AlignSize type_abi_alignment(Type *type)
{
RETRY:
switch (type->type_kind)
{
case TYPE_POISONED:
case TYPE_TYPEINFO:
case TYPE_UNTYPED_LIST:
case TYPE_MEMBER:
case TYPE_WILDCARD:
UNREACHABLE;
case TYPE_BITSTRUCT:
type = type->decl->bitstruct.base_type->type;
goto RETRY;
case TYPE_INFERRED_VECTOR:
case TYPE_VECTOR:
{
ArraySize len = type->array.len;
if (!len) len = 1;
ByteSize width = type_size(type->array.base) * len;
AlignSize alignment = (AlignSize)(int32_t)width;
if (alignment & (alignment - 1))
{
alignment = (AlignSize)next_highest_power_of_2((uint32_t)alignment);
}
if (max_alignment_vector && alignment > max_alignment_vector) alignment = max_alignment_vector;
return alignment;
}
case TYPE_VOID:
return 1;
case TYPE_OPTIONAL:
type = type->optional;
goto RETRY;
case TYPE_DISTINCT:
type = type->decl->distinct_decl.base_type;
goto RETRY;
case TYPE_TYPEDEF:
type = type->canonical;
goto RETRY;
case TYPE_ENUM:
type = type->decl->enums.type_info->type->canonical;
goto RETRY;
case TYPE_FAULTTYPE:
return t.iptr.canonical->builtin.abi_alignment;
case TYPE_STRUCT:
case TYPE_UNION:
assert(type->decl->resolve_status == RESOLVE_DONE);
return type->decl->alignment;
case TYPE_BOOL:
case ALL_INTS:
case ALL_FLOATS:
case TYPE_ANY:
case TYPE_ANYFAULT:
return type->builtin.abi_alignment;
case TYPE_FUNC:
case TYPE_POINTER:
case TYPE_TYPEID:
return t.iptr.canonical->builtin.abi_alignment;
case TYPE_ARRAY:
case TYPE_INFERRED_ARRAY:
case TYPE_FLEXIBLE_ARRAY:
type = type->array.base;
goto RETRY;
case TYPE_SUBARRAY:
return alignment_subarray;
}
UNREACHABLE
}
static inline void create_type_cache(Type *type)
{
for (int i = 0; i < ARRAY_OFFSET; i++)
{
vec_add(type->type_cache, NULL);
}
}
static Type *type_generate_ptr(Type *ptr_type, bool canonical)
{
if (canonical) ptr_type = ptr_type->canonical;
if (!ptr_type->type_cache)
{
create_type_cache(ptr_type);
}
Type *ptr = ptr_type->type_cache[PTR_OFFSET];
if (ptr == NULL)
{
ptr = type_new(TYPE_POINTER, str_printf("%s*", ptr_type->name));
ptr->pointer = ptr_type;
ptr_type->type_cache[PTR_OFFSET] = ptr;
if (ptr_type == ptr_type->canonical)
{
ptr->canonical = ptr;
}
else
{
ptr->canonical = type_generate_ptr(ptr_type->canonical, true);
}
}
return ptr;
}
static Type *type_generate_optional(Type *optional_type, bool canonical)
{
if (canonical) optional_type = optional_type->canonical;
if (!optional_type->type_cache)
{
create_type_cache(optional_type);
}
Type *optional = optional_type->type_cache[OPTIONAL_OFFSET];
if (optional == NULL)
{
optional = type_new(TYPE_OPTIONAL, str_printf("%s!", optional_type->name));
optional->pointer = optional_type;
optional_type->type_cache[OPTIONAL_OFFSET] = optional;
if (optional_type == optional_type->canonical)
{
optional->canonical = optional;
}
else
{
optional->canonical = type_generate_optional(optional_type->canonical, true);
}
}
return optional;
}
static Type *type_generate_subarray(Type *arr_type, bool canonical)
{
if (canonical) arr_type = arr_type->canonical;
if (!arr_type->type_cache)
{
create_type_cache(arr_type);
}
Type *arr = arr_type->type_cache[SUB_ARRAY_OFFSET];
if (arr == NULL)
{
arr = type_new(TYPE_SUBARRAY, str_printf("%s[]", arr_type->name));
arr->array.base = arr_type;
arr_type->type_cache[SUB_ARRAY_OFFSET] = arr;
if (arr_type == arr_type->canonical)
{
arr->canonical = arr;
}
else
{
arr->canonical = type_generate_subarray(arr_type->canonical, true);
}
}
return arr;
}
static Type *type_generate_inferred_array(Type *arr_type, bool canonical)
{
if (canonical) arr_type = arr_type->canonical;
if (!arr_type->type_cache)
{
create_type_cache(arr_type);
}
Type *arr = arr_type->type_cache[INFERRED_ARRAY_OFFSET];
if (arr == NULL)
{
arr = type_new(TYPE_INFERRED_ARRAY, str_printf("%s[*]", arr_type->name));
arr->array.base = arr_type;
arr_type->type_cache[INFERRED_ARRAY_OFFSET] = arr;
if (arr_type == arr_type->canonical)
{
arr->canonical = arr;
}
else
{
arr->canonical = type_generate_inferred_array(arr_type->canonical, true);
}
}
return arr;
}
static Type *type_generate_inferred_vector(Type *arr_type, bool canonical)
{
if (canonical) arr_type = arr_type->canonical;
if (!arr_type->type_cache)
{
create_type_cache(arr_type);
}
Type *arr = arr_type->type_cache[INFERRED_VECTOR_OFFSET];
if (arr == NULL)
{
arr = type_new(TYPE_INFERRED_VECTOR, str_printf("%s[<*>]", arr_type->name));
arr->array.base = arr_type;
arr_type->type_cache[INFERRED_VECTOR_OFFSET] = arr;
if (arr_type == arr_type->canonical)
{
arr->canonical = arr;
}
else
{
arr->canonical = type_generate_inferred_vector(arr_type->canonical, true);
}
}
return arr;
}
static Type *type_generate_flexible_array(Type *arr_type, bool canonical)
{
if (canonical) arr_type = arr_type->canonical;
if (!arr_type->type_cache)
{
create_type_cache(arr_type);
}
Type *arr = arr_type->type_cache[FLEXIBLE_ARRAY_OFFSET];
if (arr == NULL)
{
arr = type_new(TYPE_FLEXIBLE_ARRAY, str_printf("%s[*]", arr_type->name));
arr->array.base = arr_type;
arr->array.len = 0;
arr_type->type_cache[FLEXIBLE_ARRAY_OFFSET] = arr;
if (arr_type == arr_type->canonical)
{
arr->canonical = arr;
}
else
{
arr->canonical = type_generate_flexible_array(arr_type->canonical, true);
}
}
return arr;
}
Type *type_get_ptr_recurse(Type *ptr_type)
{
if (ptr_type->type_kind == TYPE_OPTIONAL)
{
ptr_type = ptr_type->optional;
return type_get_optional(type_get_ptr(ptr_type));
}
return type_get_ptr(ptr_type);
}
Type *type_get_ptr(Type *ptr_type)
{
assert(!type_is_optional(ptr_type));
return type_generate_ptr(ptr_type, false);
}
Type *type_get_optional(Type *optional_type)
{
assert(!type_is_optional(optional_type));
return type_generate_optional(optional_type, false);
}
Type *type_get_subarray(Type *arr_type)
{
return type_generate_subarray(arr_type, false);
}
Type *type_get_inferred_array(Type *arr_type)
{
return type_generate_inferred_array(arr_type, false);
}
Type *type_get_inferred_vector(Type *arr_type)
{
return type_generate_inferred_vector(arr_type, false);
}
Type *type_get_flexible_array(Type *arr_type)
{
return type_generate_flexible_array(arr_type, false);
}
static inline bool array_structurally_equivalent_to_struct(Type *array, Type *type)
{
assert(array->type_kind == TYPE_ARRAY);
MemberIndex len = (MemberIndex)array->array.len;
if (!len) return type_size(type) == 0;
Type *base = array->array.base;
if (len == 1 && type_is_structurally_equivalent(base, type)) return true;
assert(type->type_kind != TYPE_UNION && "Does not work on unions");
if (!type_is_union_or_strukt(type)) return false;
Decl **members = type->decl->strukt.members;
// For structs / errors, all members must match.
MemberIndex offset = 0;
AlignSize align_size = type_abi_alignment(array);
Type *array_base = array->array.base;
VECEACH(members, i)
{
if (!type_is_structurally_equivalent(array_base, members[i]->type)) return false;
if (members[i]->offset != offset) return false;
offset += align_size;
}
return true;
}
bool type_is_structurally_equivalent(Type *type1, Type *type2)
{
type1 = type_flatten(type1);
type2 = type_flatten(type2);
if (type1 == type2) return true;
// If the other type is a union, we check against every member
// noting that there is only structural equivalence if it fills out the
if (type2->type_kind == TYPE_UNION)
{
Decl **members = type2->decl->strukt.members;
// If any member is structurally equivalent, then
// the cast is valid.
VECEACH(members, i)
{
if (type_is_structurally_equivalent(type1, members[i]->type)) return true;
}
// In this case we can't get a match.
return false;
}
if (type1->type_kind == TYPE_ARRAY)
{
return array_structurally_equivalent_to_struct(type1, type2);
}
if (type2->type_kind == TYPE_ARRAY)
{
return array_structurally_equivalent_to_struct(type2, type1);
}
if (!type_is_union_or_strukt(type1)) return false;
Decl **members = type1->decl->strukt.members;
if (type1->type_kind == TYPE_UNION)
{
// If any member is structurally equivalent, then
// the cast is valid.
VECEACH(members, i)
{
if (type_is_structurally_equivalent(members[i]->type, type2)) return true;
}
return false;
}
// The only thing we have left is to check against another structlike.
if (!type_is_union_or_strukt(type2)) return false;
Decl **other_members = type2->decl->strukt.members;
// For structs / errors, all members must match.
VECEACH(members, i)
{
if (!type_is_structurally_equivalent(members[i]->type, other_members[i]->type)) return false;
if (members[i]->offset != other_members[i]->offset) return false;
}
return true;
}
bool type_is_user_defined(Type *type)
{
switch (type->type_kind)
{
case TYPE_ENUM:
case TYPE_FUNC:
case TYPE_STRUCT:
case TYPE_UNION:
case TYPE_FAULTTYPE:
case TYPE_DISTINCT:
return true;
case TYPE_TYPEDEF:
return type->decl != NULL;
default:
return false;
}
}
Type *type_get_indexed_type(Type *type)
{
RETRY:
switch (type->type_kind)
{
case TYPE_POINTER:
return type->pointer;
case TYPE_ARRAY:
case TYPE_SUBARRAY:
case TYPE_INFERRED_ARRAY:
case TYPE_INFERRED_VECTOR:
case TYPE_FLEXIBLE_ARRAY:
case TYPE_VECTOR:
return type->array.base;
case TYPE_DISTINCT:
type = type->decl->distinct_decl.base_type;
goto RETRY;
case TYPE_OPTIONAL:
type = type->optional;
goto RETRY;
case TYPE_TYPEDEF:
type = type->canonical;
goto RETRY;
default:
return NULL;
}
}
static Type *type_create_array(Type *element_type, ArraySize len, bool vector, bool canonical)
{
if (canonical) element_type = element_type->canonical;
if (!element_type->type_cache)
{
create_type_cache(element_type);
}
int entries = (int)vec_size(element_type->type_cache);
for (int i = ARRAY_OFFSET; i < entries; i++)
{
Type *ptr_vec = element_type->type_cache[i];
if (vector)
{
if (ptr_vec->type_kind != TYPE_VECTOR) continue;
if (ptr_vec->array.len == len) return ptr_vec;
}
else
{
if (ptr_vec->type_kind == TYPE_VECTOR) continue;
if (ptr_vec->array.len == len) return ptr_vec;
}
}
Type *vec_arr;
if (vector)
{
vec_arr = type_new(TYPE_VECTOR, str_printf("%s[<%u>]", element_type->name, len));
vec_arr->array.base = element_type;
vec_arr->array.len = len;
}
else
{
vec_arr = type_new(TYPE_ARRAY, str_printf("%s[%u]", element_type->name, len));
vec_arr->array.base = element_type;
vec_arr->array.len = len;
}
if (element_type->canonical == element_type)
{
vec_arr->canonical = vec_arr;
}
else
{
vec_arr->canonical = type_create_array(element_type, len, vector, true);
}
vec_add(element_type->type_cache, vec_arr);
return vec_arr;
}
Type *type_get_array(Type *arr_type, ArraySize len)
{
return type_create_array(arr_type, len, false, false);
}
bool type_is_valid_for_vector(Type *type)
{
RETRY:
switch (type->type_kind)
{
case ALL_INTS:
case ALL_FLOATS:
case TYPE_BOOL:
return true;
case TYPE_TYPEDEF:
type = type->canonical;
goto RETRY;
default:
return false;
}
}
Type *type_get_vector_bool(Type *original_type)
{
Type *type = type_flatten(original_type);
ByteSize size = type_size(type->array.base);
return type_get_vector(type_int_signed_by_bitsize((unsigned)size * 8), (unsigned)original_type->array.len);
}
Type *type_get_vector(Type *vector_type, unsigned len)
{
return type_create_array(vector_type, len, true, false);
}
static void type_create(const char *name, Type *location, TypeKind kind, unsigned bitsize,
unsigned align, unsigned pref_align)
{
assert(align);
unsigned byte_size = (bitsize + 7) / 8;
*location = (Type) {
.type_kind = kind,
.builtin.bytesize = byte_size,
.builtin.bitsize = bitsize,
.builtin.abi_alignment = align,
.builtin.pref_alignment = pref_align ? pref_align : align,
.name = name,
.canonical = location,
};
location->name = name;
location->canonical = location;
global_context_add_type(location);
}
static void type_init(const char *name, Type *location, TypeKind kind, unsigned bitsize, AlignData align)
{
assert(align.align);
unsigned byte_size = (bitsize + 7) / 8;
*location = (Type) {
.type_kind = kind,
.builtin.bytesize = byte_size,
.builtin.bitsize = bitsize,
.builtin.abi_alignment = align.align / 8,
.builtin.pref_alignment = (align.pref_align ? align.pref_align : align.align) / 8,
.name = name,
.canonical = location,
};
location->name = name;
location->canonical = location;
global_context_add_type(location);
}
static void type_create_alias(const char *name, Type *location, Type *canonical)
{
*location = (Type) {
.type_kind = TYPE_TYPEDEF,
.name = name,
.canonical = canonical
};
global_context_add_type(location);
}
typedef struct
{
uint32_t key;
Type *value;
} FuncTypeEntry;
typedef struct
{
uint32_t count;
uint32_t capacity;
uint32_t max_load;
FuncTypeEntry *entries;
} FuncMap;
FuncMap map;
void type_func_prototype_init(uint32_t capacity)
{
assert(is_power_of_two(capacity) && capacity > 1);
map.entries = CALLOC(capacity * sizeof(FuncTypeEntry));
map.capacity = capacity;
map.max_load = (uint32_t)(TABLE_MAX_LOAD * capacity);
}
static uint32_t hash_function(Signature *sig)
{
uintptr_t hash = sig->variadic == VARIADIC_RAW ? 0 : 1;
hash = hash * 31 + (uintptr_t)flatten_raw_function_type(type_infoptr(sig->rtype)->type);
Decl **params = sig->params;
VECEACH(params, i)
{
Decl *param = params[i];
hash = hash * 31 + (uintptr_t)flatten_raw_function_type(param->type->canonical);
}
return (uint32_t)((hash >> 16) ^ hash);
}
static bool compare_func_param(Type *one, Type *other)
{
if (one == other) return true;
one = one->canonical;
other = other->canonical;
if (one == other) return true;
if (one->type_kind != other->type_kind) return false;
switch (one->type_kind)
{
case TYPE_POINTER:
return compare_func_param(one->pointer, other->pointer);
case TYPE_ARRAY:
if (one->array.len != other->array.len) return false;
FALLTHROUGH;
case TYPE_SUBARRAY:
case TYPE_FLEXIBLE_ARRAY:
return compare_func_param(one->array.base, other->array.base);
case TYPE_FUNC:
return one->function.prototype->raw_type == other->function.prototype->raw_type;
case TYPE_OPTIONAL:
return compare_func_param(one->optional, other->optional);
default:
return false;
}
}
static int compare_function(Signature *sig, FunctionPrototype *proto)
{
bool is_raw_variadic = sig->variadic == VARIADIC_RAW;
if (is_raw_variadic != proto->raw_variadic) return -1;
Decl **params = sig->params;
Type **other_params = proto->param_types;
unsigned param_count = vec_size(params);
if (param_count != vec_size(other_params)) return -1;
if (!compare_func_param(type_infoptr(sig->rtype)->type, proto->rtype)) return -1;
VECEACH(params, i)
{
Decl *param = params[i];
Type *other_param = other_params[i];
if (!compare_func_param(param->type, other_param->canonical)) return -1;
}
return 0;
}
static Type *flatten_raw_function_type(Type *type)
{
Type *other;
Type *current;
switch (type->type_kind)
{
case TYPE_TYPEDEF:
return flatten_raw_function_type(type->canonical);
case TYPE_FUNC:
return type->function.prototype->raw_type;
case TYPE_OPTIONAL:
current = type->optional;
other = flatten_raw_function_type(current);
return other == current ? type : type_get_optional(other);
case TYPE_POINTER:
current = type->pointer;
other = flatten_raw_function_type(current);
return other == current ? type : type_get_ptr(other);
case TYPE_ARRAY:
current = type->array.base;
other = flatten_raw_function_type(current);
return other == current ? type : type_get_array(other, type->array.len);
case TYPE_SUBARRAY:
current = type->array.base;
other = flatten_raw_function_type(current);
return other == current ? type : type_get_subarray(other);
case TYPE_FLEXIBLE_ARRAY:
current = type->array.base;
other = flatten_raw_function_type(current);
return other == current ? type : type_get_flexible_array(other);
case TYPE_INFERRED_ARRAY:
current = type->array.base;
other = flatten_raw_function_type(current);
return other == current ? type : type_get_inferred_array(other);
default:
return type;
}
}
static inline Type *func_create_new_func_proto(Signature *sig, CallABI abi, uint32_t hash, FuncTypeEntry *entry)
{
unsigned param_count = vec_size(sig->params);
FunctionPrototype *proto = CALLOCS(FunctionPrototype);
proto->raw_variadic = sig->variadic == VARIADIC_RAW;
proto->vararg_index = sig->vararg_index;
Type *rtype = type_infoptr(sig->rtype)->type;
proto->rtype = rtype;
if (type_is_optional(rtype))
{
proto->is_optional = true;
Type *real_return_type = rtype->optional;
proto->ret_by_ref_type = rtype->optional;
proto->ret_by_ref = real_return_type->type_kind != TYPE_VOID;
proto->abi_ret_type = type_anyfault;
}
else
{
proto->ret_by_ref_type = proto->abi_ret_type = rtype;
}
proto->call_abi = abi;
if (param_count)
{
Type **param_types = VECNEW(Type*, param_count);
Decl **param_copy = VECNEW(Decl*, param_count);
for (unsigned i = 0; i < param_count; i++)
{
Decl *decl = decl_copy(sig->params[i]);
decl->type = decl->type->canonical;
decl->var.type_info = NULL;
decl->var.init_expr = NULL;
decl->name = NULL;
vec_add(param_types, decl->type);
vec_add(param_copy, decl);
}
proto->param_types = param_types;
proto->param_copy = param_copy;
}
scratch_buffer_clear();
scratch_buffer_append("fn ");
type_append_name_to_scratch(proto->rtype);
scratch_buffer_append_char('(');
foreach(Type*, proto->param_types)
{
if (foreach_index != 0) scratch_buffer_append(", ");
type_append_name_to_scratch(val);
}
scratch_buffer_append_char(')');
Type *type = type_new(TYPE_FUNC, scratch_buffer_interned());
Signature *copy_sig = CALLOCS(Signature);
*copy_sig = *sig;
copy_sig->attrs = (CalleeAttributes) { .nodiscard = false };
copy_sig->params = proto->param_copy;
proto->raw_type = type;
type->function.prototype = proto;
type->function.module = NULL;
type->function.signature = copy_sig;
type->canonical = type;
entry->key = hash;
entry->value = type;
map.count++;
if (map.count >= map.max_load)
{
FuncTypeEntry *entries = map.entries;
uint32_t old_capacity = map.capacity;
uint32_t new_capacity = map.capacity = old_capacity << 2;
map.max_load = (uint32_t)(new_capacity * TABLE_MAX_LOAD);
FuncTypeEntry *new_map = CALLOC(new_capacity * sizeof(FuncTypeEntry));
uint32_t new_mask = new_capacity - 1;
for (uint32_t i = 0; i < old_capacity; i++)
{
uint32_t key = entries[i].key;
if (!key) continue;
uint32_t index = key & new_mask;
while (1)
{
entry = &new_map[index];
if (!entry->key)
{
entry->key = key;
entry->value = entries[i].value;
break;
}
index = (index + 1) & new_mask;
}
}
map.entries = new_map;
}
return type;
}
Type *type_get_func(Signature *signature, CallABI abi)
{
uint32_t hash = hash_function(signature);
uint32_t mask = map.capacity - 1;
uint32_t index = hash & mask;
FuncTypeEntry *entries = map.entries;
FuncTypeEntry *entry;
while (1)
{
entry = &entries[index];
if (!entry->key)
{
return func_create_new_func_proto(signature, abi, hash, entry);
}
if (entry->key == hash && compare_function(signature, entry->value->function.prototype) == 0)
{
return entry->value;
}
index = (index + 1) & mask;
}
}
static inline void type_init_int(const char *name, Type *type, TypeKind kind, BitSizes bits)
{
unsigned actual_bits = bits ? (unsigned int)(8 << (bits - 1)) : 1;
type_init(name, type, kind, actual_bits, platform_target.integers[bits]);
}
static inline void type_create_float(const char *name, Type *type, TypeKind kind, BitSizes bits)
{
unsigned actual_bits = bits ? (unsigned int)(8 << (bits - 1)) : 1;
type_init(name, type, kind, actual_bits, platform_target.floats[bits]);
}
void type_setup(PlatformTarget *target)
{
max_alignment_vector = (AlignSize)target->align_max_vector;
type_create_float("float16", &t.f16, TYPE_F16, BITS16);
type_create_float("float", &t.f32, TYPE_F32, BITS32);
type_create_float("double", &t.f64, TYPE_F64, BITS64);
type_create_float("float128", &t.f128, TYPE_F128, BITS128);
type_init_int("ichar", &t.i8, TYPE_I8, BITS8);
type_init_int("short", &t.i16, TYPE_I16, BITS16);
type_init_int("int", &t.i32, TYPE_I32, BITS32);
type_init_int("long", &t.i64, TYPE_I64, BITS64);
type_init_int("int128", &t.i128, TYPE_I128, BITS128);
type_init_int("bool", &t.u1, TYPE_BOOL, BITS8);
type_init_int("char", &t.u8, TYPE_U8, BITS8);
type_init_int("ushort", &t.u16, TYPE_U16, BITS16);
type_init_int("uint", &t.u32, TYPE_U32, BITS32);
type_init_int("ulong", &t.u64, TYPE_U64, BITS64);
type_init_int("uint128", &t.u128, TYPE_U128, BITS128);
type_init_int("void", &t.u0, TYPE_VOID, BITS8);
type_create("typeinfo", &t.typeinfo, TYPE_TYPEINFO, 1, 1, 1);
type_create("member_ref", &t.member, TYPE_MEMBER, 1, 1, 1);
type_create("untyped_list", &t.untyped_list, TYPE_UNTYPED_LIST, 1, 1, 1);
type_create("void", &t.wildcard, TYPE_WILDCARD, 1, 1, 1);
type_init("typeid", &t.typeid, TYPE_TYPEID, target->width_pointer, target->align_pointer);
type_init("void*", &t.voidstar, TYPE_POINTER, target->width_pointer, target->align_pointer);
create_type_cache(type_void);
type_void->type_cache[0] = &t.voidstar;
t.voidstar.pointer = type_void;
type_init("any", &t.any, TYPE_ANY, target->width_pointer * 2, target->align_pointer);
type_create_alias("usz", &t.usz, type_int_unsigned_by_bitsize(target->width_pointer));
type_create_alias("isz", &t.isz, type_int_signed_by_bitsize(target->width_pointer));
type_create_alias("uptr", &t.uptr, type_int_unsigned_by_bitsize(target->width_pointer));
type_create_alias("iptr", &t.iptr, type_int_signed_by_bitsize(target->width_pointer));
alignment_subarray = MAX(type_abi_alignment(&t.voidstar), type_abi_alignment(t.usz.canonical));
size_subarray = (unsigned)(alignment_subarray * 2);
type_init("anyfault", &t.anyfault, TYPE_ANYFAULT, target->width_pointer, target->align_pointer);
type_chars = type_get_subarray(type_char);
type_wildcard_optional = type_get_optional(type_wildcard);
Decl *string_decl = decl_new_with_type(symtab_preset("String", TOKEN_TYPE_IDENT), INVALID_SPAN, DECL_DISTINCT);
string_decl->extname = string_decl->name;
string_decl->is_substruct = true;
string_decl->distinct_decl.base_type = type_chars;
string_decl->resolve_status = RESOLVE_DONE;
type_string = string_decl->type;
global_context_add_type(string_decl->type);
global_context_add_decl(string_decl);
}
int type_kind_bitsize(TypeKind kind)
{
switch (kind)
{
case TYPE_I8:
case TYPE_U8:
return 8;
case TYPE_I16:
case TYPE_U16:
case TYPE_F16:
return 16;
case TYPE_I32:
case TYPE_U32:
case TYPE_F32:
return 32;
case TYPE_I64:
case TYPE_U64:
case TYPE_F64:
return 64;
case TYPE_I128:
case TYPE_U128:
case TYPE_F128:
return 128;
default:
UNREACHABLE;
}
}
bool type_is_scalar(Type *type)
{
RETRY:
switch (type->type_kind)
{
case CT_TYPES:
UNREACHABLE
case TYPE_VOID:
case TYPE_FUNC:
case TYPE_STRUCT:
case TYPE_UNION:
case TYPE_ARRAY:
case TYPE_SUBARRAY:
case TYPE_VECTOR:
case TYPE_ANY:
case TYPE_FLEXIBLE_ARRAY:
return false;
case TYPE_BOOL:
case ALL_INTS:
case ALL_FLOATS:
case TYPE_TYPEID:
case TYPE_POINTER:
case TYPE_ENUM:
case TYPE_FAULTTYPE:
case TYPE_ANYFAULT:
return true;
case TYPE_BITSTRUCT:
type = type->decl->bitstruct.base_type->type;
goto RETRY;
case TYPE_DISTINCT:
type = type->decl->distinct_decl.base_type;
goto RETRY;
case TYPE_OPTIONAL:
type = type->optional;
if (!type) return false;
goto RETRY;
case TYPE_TYPEDEF:
type = type->canonical;
goto RETRY;
}
UNREACHABLE
}
/**
* Check if a type is contained in another type.
*
* @param type canonical type
* @param possible_subtype canonical type
* @return true if it is a subtype
*/
bool type_is_subtype(Type *type, Type *possible_subtype)
{
assert(type == type->canonical && possible_subtype == possible_subtype->canonical);
while (1)
{
if (type == possible_subtype) return true;
if (type->type_kind != possible_subtype->type_kind) return false;
if (type->type_kind != TYPE_STRUCT) return false;
if (!possible_subtype->decl->strukt.members) return false;
Decl *first_element = possible_subtype->decl->strukt.members[0];
if (first_element->decl_kind != DECL_VAR) return false;
possible_subtype = first_element->type->canonical;
}
}
Type *type_from_token(TokenType type)
{
switch (type)
{
case TOKEN_ANY:
return type_any;
case TOKEN_ANYFAULT:
return type_anyfault;
case TOKEN_VOID:
return type_void;
case TOKEN_BOOL:
return type_bool;
case TOKEN_CHAR:
return type_char;
case TOKEN_FLOAT16:
return type_float16;
case TOKEN_DOUBLE:
return type_double;
case TOKEN_FLOAT:
return type_float;
case TOKEN_FLOAT128:
return type_f128;
case TOKEN_INT128:
return type_i128;
case TOKEN_ICHAR:
return type_ichar;
case TOKEN_INT:
return type_int;
case TOKEN_IPTR:
return type_iptr;
case TOKEN_ISZ:
return type_isz;
case TOKEN_LONG:
return type_long;
case TOKEN_SHORT:
return type_short;
case TOKEN_UINT128:
return type_u128;
case TOKEN_UINT:
return type_uint;
case TOKEN_ULONG:
return type_ulong;
case TOKEN_UPTR:
return type_uptr;
case TOKEN_USHORT:
return type_ushort;
case TOKEN_USZ:
return type_usz;
case TOKEN_TYPEID:
return type_typeid;
default:
UNREACHABLE
}
}
bool type_array_is_equivalent(Type *from, Type *to, bool is_explicit)
{
TypeKind to_kind = to->type_kind;
switch (from->type_kind)
{
case TYPE_INFERRED_ARRAY:
assert(to_kind != TYPE_INFERRED_ARRAY);
if (to_kind != TYPE_ARRAY) return false;
return type_array_element_is_equivalent(from->array.base, to->array.base, is_explicit);
case TYPE_ARRAY:
if (to_kind != TYPE_ARRAY && to_kind != TYPE_INFERRED_ARRAY) return false;
if (to->type_kind == TYPE_ARRAY && from->array.len != to->array.len) return false;
return type_array_element_is_equivalent(from->array.base, to->array.base, is_explicit);
case TYPE_INFERRED_VECTOR:
assert(to_kind != TYPE_INFERRED_VECTOR);
if (to->type_kind != TYPE_VECTOR) return false;
return type_array_element_is_equivalent(from->array.base, to->array.base, is_explicit);
case TYPE_VECTOR:
if (to_kind != TYPE_VECTOR && to_kind != TYPE_INFERRED_VECTOR) return false;
if (to->type_kind == TYPE_VECTOR && from->array.len != to->array.len) return false;
return type_array_element_is_equivalent(from->array.base, to->array.base, is_explicit);
default:
return false;
}
}
bool type_array_element_is_equivalent(Type *element1, Type *element2, bool is_explicit)
{
if (is_explicit)
{
element1 = type_flatten(element1);
element2 = type_flatten(element2);
}
else
{
element1 = element1->canonical;
element2 = element2->canonical;
}
if (element1 == element2) return true;
switch (element1->type_kind)
{
case TYPE_POINTER:
if (element2->type_kind != TYPE_POINTER) return false;
return type_is_pointer_equivalent(element1, element2, is_explicit);
case TYPE_STRUCT:
if (is_explicit) return type_is_structurally_equivalent(element1, element2);
return false;
case TYPE_VECTOR:
case TYPE_ARRAY:
case TYPE_INFERRED_ARRAY:
case TYPE_INFERRED_VECTOR:
return type_array_is_equivalent(element1, element2, is_explicit);
default:
return false;
}
}
bool type_is_pointer_equivalent(Type *pointer1, Type *pointer2, bool flatten_distinct)
{
RETRY:
if (flatten_distinct)
{
pointer1 = type_flatten(pointer1);
pointer2 = type_flatten(pointer2);
}
if (pointer1 == pointer2) return true;
if (pointer1 == type_voidptr || pointer2 == type_voidptr) return true;
Type *pointee1 = pointer1->pointer->canonical;
Type *pointee2 = pointer2->pointer->canonical;
if (flatten_distinct)
{
pointee1 = type_flatten(pointee1);
pointee2 = type_flatten(pointee2);
}
if (pointee1 == pointee2) return true;
if (type_is_subtype(pointee2, pointee1)) return true;
if (pointee1->type_kind != pointee2->type_kind)
{
if (type_is_any_arraylike(pointee1))
{
// Try array equivalence.
if (type_is_any_arraylike(pointee2))
{
if (type_array_is_equivalent(pointee1, pointee2, flatten_distinct)) return true;
}
// A possible int[4]* -> int* decay?
return type_is_pointer_equivalent(type_get_ptr(pointee1->array.base), pointer2, flatten_distinct);
}
// Not arraylike and no array decay. Failure.
return false;
}
if (pointee1->type_kind == TYPE_FUNC && pointee2->type_kind == TYPE_FUNC)
{
return pointee1->function.prototype->raw_type == pointee2->function.prototype->raw_type;
}
if (pointee1->type_kind == TYPE_POINTER)
{
pointer1 = pointee1;
pointer2 = pointee2;
goto RETRY;
}
return false;
}
bool type_may_have_method(Type *type)
{
DECL_TYPE_KIND_REAL(kind, type)
switch (kind)
{
case TYPE_DISTINCT:
case TYPE_UNION:
case TYPE_STRUCT:
case TYPE_ENUM:
case TYPE_FAULTTYPE:
case TYPE_BITSTRUCT:
case ALL_FLOATS:
case ALL_INTS:
case TYPE_ANY:
case TYPE_ANYFAULT:
case TYPE_TYPEID:
case TYPE_ARRAY:
case TYPE_SUBARRAY:
case TYPE_INFERRED_ARRAY:
case TYPE_INFERRED_VECTOR:
case TYPE_FLEXIBLE_ARRAY:
case TYPE_VECTOR:
case TYPE_BOOL:
return true;
case TYPE_TYPEDEF:
UNREACHABLE
case TYPE_POISONED:
case TYPE_VOID:
case TYPE_POINTER:
case TYPE_FUNC:
case TYPE_UNTYPED_LIST:
case TYPE_OPTIONAL:
case TYPE_TYPEINFO:
case TYPE_MEMBER:
case TYPE_WILDCARD:
return false;
}
UNREACHABLE
}
bool type_may_have_sub_elements(Type *type)
{
DECL_TYPE_KIND_REAL(kind, type)
switch (kind)
{
case TYPE_DISTINCT:
case TYPE_UNION:
case TYPE_STRUCT:
case TYPE_ENUM:
case TYPE_FAULTTYPE:
case TYPE_BITSTRUCT:
return true;
default:
return false;
}
}
Type *type_find_max_num_type(Type *num_type, Type *other_num)
{
TypeKind kind = num_type->type_kind;
TypeKind other_kind = other_num->type_kind;
assert(kind <= other_kind && "Expected ordering");
assert(kind != other_kind);
// 1. The only conversions need to happen if the other type is a number.
if (other_kind < TYPE_INTEGER_FIRST || other_kind > TYPE_FLOAT_LAST) return NULL;
// 2. First check the float case.
if (other_kind >= TYPE_FLOAT_FIRST && other_kind <= TYPE_FLOAT_LAST)
{
switch (other_kind)
{
case TYPE_F16:
case TYPE_F32:
case TYPE_F64:
case TYPE_F128:
// Pick the biggest, which will be in other_num due to ordering.
return other_num;
default:
UNREACHABLE
}
}
// Handle integer <=> integer conversions.
assert(type_kind_is_any_integer(other_kind) && type_is_integer(num_type));
// 4. Check the bit sizes.
unsigned other_bit_size = other_num->builtin.bitsize;
unsigned bit_size = num_type->builtin.bitsize;
// 5. The other type is unsigned
if (type_kind_is_unsigned(other_kind))
{
if (type_kind_is_signed(kind))
{
// 5a. Signed + Unsigned -> Signed
return bit_size >= other_bit_size ? num_type : type_int_signed_by_bitsize(other_bit_size);
}
// 5b. Unsigned + Unsigned -> return other_num which is the bigger due to ordering.
return other_num;
}
// 6. The other type is signed, then pick other_num which is bigger due to ordering.
return other_num;
}
/**
* max(Foo[:], Bar*) -> max(Foo*, Bar*)
* max(Foo[], Bar*) -> max(Foo*, Bar*)
* max(Foo[n]*, Bar*) -> max(Foo*, Bar*)
* max(void*, Foo*) -> void*
* max(Foo*, Bar*) -> max(Foo, Bar)*
* max(other, Foo*) -> NULL
*
* @param type
* @param other
* @return the max pointer type or NULL if none can be found.
*/
static inline Type *type_find_max_ptr_type(Type *type, Type *other)
{
// Subarray and vararray can implicitly convert to a pointer.
if (other->type_kind == TYPE_SUBARRAY)
{
Type *max_type = type_find_max_type(type->pointer, other->pointer);
if (!max_type) return NULL;
return type_get_ptr(max_type);
}
// Neither subarray, vararray or pointer? Then no max
if (other->type_kind != TYPE_POINTER) return NULL;
Type* other_pointer_type = other->pointer;
Type* pointer_type = type->pointer;
// Reorder if needed
if (other_pointer_type->type_kind < pointer_type->type_kind)
{
pointer_type = other_pointer_type;
other_pointer_type = type->pointer;
}
// void * is always max.
if (type_is_void(pointer_type)) return type_voidptr;
if (pointer_type->type_kind == TYPE_POINTER && other_pointer_type->type_kind == TYPE_ARRAY)
{
// Decay foo[n]* to foo*
other_pointer_type = type_get_ptr(other_pointer_type->array.base);
}
if (type_is_subtype(pointer_type->canonical, other_pointer_type->canonical))
{
return type;
}
if (type_is_subtype(other_pointer_type->canonical, pointer_type->canonical))
{
return other;
}
Type *max_type = type_find_max_type(pointer_type, other_pointer_type);
if (!max_type) return NULL;
return type_get_ptr(max_type);
}
Type *type_decay_array_pointer(Type *type)
{
assert(type->type_kind == TYPE_POINTER);
Type *ptr = type->pointer;
switch (ptr->type_kind)
{
case TYPE_ARRAY:
case TYPE_VECTOR:
return type_get_ptr(ptr->array.base->canonical);
default:
return type;
}
}
Type *type_find_max_type(Type *type, Type *other)
{
type = type->canonical;
other = other->canonical;
assert(!type_is_optional(type) && !type_is_optional(other));
if (type == other) return type;
if (type == type_wildcard) return other;
if (other == type_wildcard) return type;
// Lower inlined distinct types.
while (type->type_kind == TYPE_DISTINCT && type->decl->is_substruct) type = type->decl->distinct_decl.base_type;
while (other->type_kind == TYPE_DISTINCT && other->decl->is_substruct) other = other->decl->distinct_decl.base_type;
// We may now have a match.
if (type == other) return type;
// Sort types
if (type->type_kind > other->type_kind)
{
Type *temp = type;
type = other;
other = temp;
}
switch (type->type_kind)
{
case TYPE_INFERRED_ARRAY:
case TYPE_INFERRED_VECTOR:
case TYPE_POISONED:
case TYPE_OPTIONAL:
case TYPE_WILDCARD:
UNREACHABLE
case TYPE_VOID:
case TYPE_BOOL:
case TYPE_TYPEINFO:
case TYPE_ANY:
case TYPE_BITSTRUCT:
case TYPE_FLEXIBLE_ARRAY:
return NULL;
case ALL_INTS:
if (other->type_kind == TYPE_DISTINCT && type_underlying_is_numeric(other)) return other;
if (other->type_kind == TYPE_ENUM) return type_find_max_type(type, other->decl->enums.type_info->type->canonical);
if (other->type_kind == TYPE_VECTOR) return other;
return type_find_max_num_type(type, other);
case ALL_FLOATS:
if (other->type_kind == TYPE_DISTINCT && type_is_float(other->decl->distinct_decl.base_type)) return other;
if (other->type_kind == TYPE_VECTOR) return other;
return type_find_max_num_type(type, other);
case TYPE_POINTER:
if (type->pointer->type_kind == TYPE_ARRAY)
{
Type *array_base = type->pointer->array.base->canonical;
if (other->type_kind == TYPE_SUBARRAY &&
array_base == other->array.base->canonical)
{
return other;
}
if (other->type_kind == TYPE_POINTER)
{
Type *other_pointer = other->pointer;
if (other_pointer->type_kind == TYPE_ARRAY && other_pointer->array.base->canonical == array_base)
{
return type_get_subarray(array_base);
}
}
}
if (type->pointer->type_kind == TYPE_VECTOR)
{
Type *vector_base = type->pointer->array.base->canonical;
if (other->type_kind == TYPE_SUBARRAY && vector_base == other->array.base->canonical)
{
return other;
}
}
// We need to decay the pointer
type = type_decay_array_pointer(type);
// And possibly the other pointer as well
if (other->type_kind == TYPE_POINTER) other = type_decay_array_pointer(other);
return type_find_max_ptr_type(type, other);
case TYPE_ENUM:
// IMPROVE: should there be implicit conversion between one enum and the other in
// some way?
return NULL;
case TYPE_FAULTTYPE:
if (other->type_kind == TYPE_FAULTTYPE) return type_anyfault;
return NULL;
case TYPE_ANYFAULT:
return type_anyfault;
case TYPE_FUNC:
if (other->type_kind != TYPE_FUNC) return NULL;
other = other->function.prototype->raw_type;
type = other->function.prototype->raw_type;
return other == type ? type : NULL;
case TYPE_UNTYPED_LIST:
if (other->type_kind == TYPE_ARRAY) return other;
if (other->type_kind == TYPE_VECTOR) return other;
if (other->type_kind == TYPE_STRUCT) return other;
if (other->type_kind == TYPE_SUBARRAY) return other;
return NULL;
case TYPE_UNION:
case TYPE_STRUCT:
// union/struct + anything else => no
// even if the struct has an inline type, this should not
// be implicit
return NULL;
case TYPE_TYPEID:
case TYPE_MEMBER:
return NULL;
case TYPE_TYPEDEF:
UNREACHABLE
case TYPE_DISTINCT:
// distinct + any other type => no
return NULL;
case TYPE_ARRAY:
// array + [subarray, other array, vector] => no
return NULL;
case TYPE_SUBARRAY:
// subarray + [other subarray, vector] => no
return NULL;
case TYPE_VECTOR:
// No implicit conversion between vectors
return NULL;
}
UNREACHABLE
}
#define MAX_SEARCH_DEPTH 512
Type *type_find_common_ancestor(Type *left, Type *right)
{
if (left == right) return left;
left = left->canonical;
right = right->canonical;
if (left == right) return left;
if (left->type_kind != right->type_kind) return NULL;
if (left->type_kind == TYPE_POINTER)
{
Type *common = type_find_common_ancestor(left->pointer, right->pointer);
return common ? type_get_ptr(common) : NULL;
}
if (left->type_kind != TYPE_STRUCT) return NULL;
static Type *left_types[MAX_SEARCH_DEPTH];
int depth = 0;
while (depth < MAX_SEARCH_DEPTH)
{
if (!left->decl->strukt.members) break;
Decl *first_element = left->decl->strukt.members[0];
if (first_element->decl_kind != DECL_VAR) break;
if (first_element->type->canonical == right) return right;
left = first_element->type->canonical;
left_types[depth++] = left;
}
if (depth == MAX_SEARCH_DEPTH)
{
error_exit("Struct type depth %d exceeded.", MAX_SEARCH_DEPTH);
}
while (true)
{
if (!right->decl->strukt.members) return NULL;
Decl *first_element = right->decl->strukt.members[0];
if (first_element->decl_kind != DECL_VAR) return NULL;
right = first_element->type->canonical;
for (int i = 0; i < depth; i++)
{
if (right == left_types[i]) return right;
}
}
}
unsigned type_get_introspection_kind(TypeKind kind)
{
switch (kind)
{
case TYPE_POISONED:
return 0;
case TYPE_VOID:
return INTROSPECT_TYPE_VOID;
case TYPE_BOOL:
return INTROSPECT_TYPE_BOOL;
case TYPE_I8:
case TYPE_I16:
case TYPE_I32:
case TYPE_I64:
case TYPE_I128:
return INTROSPECT_TYPE_SIGNED_INT;
case TYPE_U8:
case TYPE_U16:
case TYPE_U32:
case TYPE_U64:
case TYPE_U128:
return INTROSPECT_TYPE_UNSIGNED_INT;
case TYPE_F16:
case TYPE_BF16:
case TYPE_F32:
case TYPE_F64:
case TYPE_F128:
return INTROSPECT_TYPE_FLOAT;
case TYPE_ANY:
return INTROSPECT_TYPE_ANY;
case TYPE_ANYFAULT:
return INTROSPECT_TYPE_ANYFAULT;
case TYPE_TYPEID:
return INTROSPECT_TYPE_TYPEID;
case TYPE_POINTER:
return INTROSPECT_TYPE_POINTER;
case TYPE_ENUM:
return INTROSPECT_TYPE_ENUM;
case TYPE_FUNC:
return INTROSPECT_TYPE_FUNC;
case TYPE_STRUCT:
return INTROSPECT_TYPE_STRUCT;
case TYPE_UNION:
return INTROSPECT_TYPE_UNION;
case TYPE_BITSTRUCT:
return INTROSPECT_TYPE_BITSTRUCT;
case TYPE_FAULTTYPE:
return INTROSPECT_TYPE_FAULT;
case TYPE_TYPEDEF:
UNREACHABLE
case TYPE_DISTINCT:
return INTROSPECT_TYPE_DISTINCT;
case TYPE_ARRAY:
case TYPE_INFERRED_ARRAY:
case TYPE_FLEXIBLE_ARRAY:
return INTROSPECT_TYPE_ARRAY;
case TYPE_SUBARRAY:
return INTROSPECT_TYPE_SUBARRAY;
case TYPE_VECTOR:
case TYPE_INFERRED_VECTOR:
return INTROSPECT_TYPE_VECTOR;
case TYPE_UNTYPED_LIST:
case TYPE_TYPEINFO:
case TYPE_OPTIONAL:
case TYPE_MEMBER:
case TYPE_WILDCARD:
UNREACHABLE
return 0;
}
UNREACHABLE
}
Module *type_base_module(Type *type)
{
RETRY:
switch (type->type_kind)
{
case TYPE_POISONED:
case TYPE_VOID:
case ALL_INTS:
case ALL_FLOATS:
case TYPE_BOOL:
case TYPE_ANY:
case TYPE_ANYFAULT:
case TYPE_TYPEID:
case TYPE_WILDCARD:
return NULL;
case TYPE_POINTER:
type = type->pointer;
goto RETRY;
case TYPE_FUNC:
return type->function.module;
case TYPE_ENUM:
case TYPE_STRUCT:
case TYPE_UNION:
case TYPE_BITSTRUCT:
case TYPE_FAULTTYPE:
case TYPE_DISTINCT:
return type->decl->unit ? type->decl->unit->module : NULL;
case TYPE_TYPEDEF:
type = type->canonical;
goto RETRY;
case TYPE_ARRAY:
case TYPE_SUBARRAY:
case TYPE_INFERRED_ARRAY:
case TYPE_FLEXIBLE_ARRAY:
case TYPE_VECTOR:
case TYPE_INFERRED_VECTOR:
type = type->array.base;
goto RETRY;
case TYPE_OPTIONAL:
type = type->optional;
goto RETRY;
case TYPE_UNTYPED_LIST:
case TYPE_TYPEINFO:
case TYPE_MEMBER:
UNREACHABLE
}
UNREACHABLE
}