7419 lines
352 KiB
Zig
7419 lines
352 KiB
Zig
const std = @import("std");
|
|
const ast = @import("ast.zig");
|
|
const Node = ast.Node;
|
|
const Span = ast.Span;
|
|
const llvm = @import("llvm_api.zig");
|
|
const c = llvm.c;
|
|
const types = @import("types.zig");
|
|
const Type = types.Type;
|
|
const Builtins = @import("builtins.zig").Builtins;
|
|
const Parser = @import("parser.zig").Parser;
|
|
const errors = @import("errors.zig");
|
|
const sema = @import("sema.zig");
|
|
const comptime_mod = @import("comptime.zig");
|
|
const unescape = @import("unescape.zig");
|
|
|
|
pub const TargetConfig = struct {
|
|
/// Target triple (e.g. "aarch64-apple-darwin"). Null = host default.
|
|
triple: ?[*:0]const u8 = null,
|
|
/// CPU name (e.g. "generic", "apple-m1"). Null = "generic".
|
|
cpu: ?[*:0]const u8 = null,
|
|
/// CPU features string (e.g. "+avx2"). Null = "".
|
|
features: ?[*:0]const u8 = null,
|
|
/// Optimization level.
|
|
opt_level: OptLevel = .default,
|
|
/// Library search paths (-L flags).
|
|
lib_paths: []const []const u8 = &.{},
|
|
/// Output path override.
|
|
output_path: ?[]const u8 = null,
|
|
/// Linker command (null = "cc" on Unix, "link.exe" on Windows).
|
|
linker: ?[]const u8 = null,
|
|
/// Sysroot for cross-compilation (passed as --sysroot to linker).
|
|
sysroot: ?[]const u8 = null,
|
|
|
|
pub const OptLevel = enum {
|
|
none,
|
|
less,
|
|
default,
|
|
aggressive,
|
|
|
|
pub fn toLLVM(self: OptLevel) c.LLVMCodeGenOptLevel {
|
|
return switch (self) {
|
|
.none => c.LLVMCodeGenLevelNone,
|
|
.less => c.LLVMCodeGenLevelLess,
|
|
.default => c.LLVMCodeGenLevelDefault,
|
|
.aggressive => c.LLVMCodeGenLevelAggressive,
|
|
};
|
|
}
|
|
};
|
|
|
|
/// Check if target triple indicates aarch64/arm64 (runtime check, not comptime).
|
|
pub fn isAarch64(self: TargetConfig) bool {
|
|
return self.tripleHasPrefix("aarch64", "arm64");
|
|
}
|
|
|
|
/// Check if target triple indicates x86_64/x86-64.
|
|
pub fn isX86_64(self: TargetConfig) bool {
|
|
return self.tripleHasPrefix("x86_64", "x86-64");
|
|
}
|
|
|
|
/// Check if target triple indicates Windows (contains "windows" or "win32").
|
|
pub fn isWindows(self: TargetConfig) bool {
|
|
return self.tripleContains("windows") or self.tripleContains("win32");
|
|
}
|
|
|
|
fn tripleHasPrefix(self: TargetConfig, prefix1: []const u8, prefix2: []const u8) bool {
|
|
if (self.triple) |t| {
|
|
const span = std.mem.span(t);
|
|
return std.mem.startsWith(u8, span, prefix1) or std.mem.startsWith(u8, span, prefix2);
|
|
}
|
|
const dt = c.LLVMGetDefaultTargetTriple();
|
|
defer c.LLVMDisposeMessage(dt);
|
|
const span = std.mem.span(dt);
|
|
return std.mem.startsWith(u8, span, prefix1) or std.mem.startsWith(u8, span, prefix2);
|
|
}
|
|
|
|
fn tripleContains(self: TargetConfig, needle: []const u8) bool {
|
|
if (self.triple) |t| {
|
|
return std.mem.indexOf(u8, std.mem.span(t), needle) != null;
|
|
}
|
|
const dt = c.LLVMGetDefaultTargetTriple();
|
|
defer c.LLVMDisposeMessage(dt);
|
|
return std.mem.indexOf(u8, std.mem.span(dt), needle) != null;
|
|
}
|
|
|
|
pub fn getCpu(self: TargetConfig) [*:0]const u8 {
|
|
return self.cpu orelse "generic";
|
|
}
|
|
|
|
pub fn getFeatures(self: TargetConfig) [*:0]const u8 {
|
|
return self.features orelse "";
|
|
}
|
|
|
|
pub fn getLinker(self: TargetConfig) []const u8 {
|
|
return self.linker orelse "cc";
|
|
}
|
|
};
|
|
|
|
fn baseName(name: []const u8) []const u8 {
|
|
return if (std.mem.lastIndexOfScalar(u8, name, '.')) |idx| name[idx + 1 ..] else name;
|
|
}
|
|
|
|
pub const CodeGen = struct {
|
|
context: c.LLVMContextRef,
|
|
module: c.LLVMModuleRef,
|
|
builder: c.LLVMBuilderRef,
|
|
allocator: std.mem.Allocator,
|
|
// ORC ThreadSafeContext — wraps the LLVMContext for JIT compatibility
|
|
ts_context: c.LLVMOrcThreadSafeContextRef = null,
|
|
// Whether we still own the module (false after JIT takes ownership)
|
|
module_owned: bool = true,
|
|
// Cached target machine (created in init, reused by emitToFile)
|
|
target_machine: c.LLVMTargetMachineRef = null,
|
|
|
|
// Symbol table: maps variable names to their alloca pointers
|
|
named_values: std.StringHashMap(NamedValue),
|
|
// Unified type registry: single lookup for all named types (structs, enums, unions, aliases)
|
|
type_registry: std.StringHashMap(TypeRegistryEntry),
|
|
// Flags enum registry: tracks which enum names are flags
|
|
flags_enum_types: std.StringHashMap(void),
|
|
// Enum variant values: maps enum name → resolved i64 values per variant
|
|
enum_variant_values: std.StringHashMap([]const i64),
|
|
// Enum backing types: maps enum name → LLVM type for the backing integer (default i64)
|
|
enum_backing_types: std.StringHashMap(c.LLVMTypeRef),
|
|
// Built-in functions (printf, etc.)
|
|
builtins: ?Builtins,
|
|
// Current function being generated (for alloca insertion)
|
|
current_function: c.LLVMValueRef,
|
|
// Return type of the current function being generated
|
|
current_return_type: Type = .void_type,
|
|
// Scope stack: each entry records shadowed names and deferred expressions for one scope
|
|
scope_stack: std.ArrayList(Scope),
|
|
// Compile-time globals: maps name to global variable info for #run results
|
|
comptime_globals: std.StringHashMap(ComptimeGlobal),
|
|
// Top-level #run expressions for side effects only
|
|
comptime_side_effects: std.ArrayList(*Node),
|
|
// Generic function templates: maps name to AST for deferred monomorphization
|
|
generic_templates: std.StringHashMap(ast.FnDecl),
|
|
// Instantiated generic functions: maps mangled name to LLVM function
|
|
generic_instances: std.StringHashMap(c.LLVMValueRef),
|
|
// Active type parameter bindings during generic instantiation (null when not instantiating)
|
|
type_param_bindings: ?std.StringHashMap(Type) = null,
|
|
// Active value parameter bindings during generic struct instantiation
|
|
value_param_bindings: ?std.StringHashMap(i64) = null,
|
|
// Active comptime param AST nodes during generic function instantiation (for #insert substitution)
|
|
comptime_param_nodes: ?std.StringHashMap(*Node) = null,
|
|
// Generic struct templates: maps name to AST for deferred instantiation
|
|
generic_struct_templates: std.StringHashMap(ast.StructDecl),
|
|
// Known namespace names (for import resolution)
|
|
namespaces: std.StringHashMap(void),
|
|
// Functions declared with #builtin (only available when imported)
|
|
builtin_functions: std.StringHashMap(void),
|
|
// Function signatures: maps function name to display signature (e.g. "test :: () -> s32")
|
|
fn_signatures: std.StringHashMap([]const u8),
|
|
// Active namespace during body generation of imported modules
|
|
current_namespace: ?[]const u8 = null,
|
|
// Diagnostics list (optional, for structured error reporting)
|
|
diagnostics: ?*errors.DiagnosticList = null,
|
|
// Current source span (set at genExpr/genStmt/genExprAsType entry)
|
|
current_span: Span = .{ .start = 0, .end = 0 },
|
|
// Loop context: break/continue target basic blocks (null when not in a loop)
|
|
loop_break_bb: c.LLVMBasicBlockRef = null,
|
|
loop_continue_bb: c.LLVMBasicBlockRef = null,
|
|
// Sema result (optional, for type-aware comptime evaluation)
|
|
sema_result: ?*const sema.SemaResult = null,
|
|
// Root declarations from the AST (for VM on-demand function compilation)
|
|
root_decls: []const *Node = &.{},
|
|
// Cached LLVM struct type for string slices {ptr, i32}
|
|
string_struct_type: c.LLVMTypeRef = null,
|
|
// Cached LLVM struct type for Any {i32 tag, i64 value}
|
|
any_struct_type: c.LLVMTypeRef = null,
|
|
// Dynamic type ID assignment for Any tags (named types get unique IDs starting from 7)
|
|
any_type_id_map: std.StringHashMap(u64),
|
|
next_any_type_id: u64 = 7,
|
|
// Cache of auto-generated to_string functions for complex types
|
|
// Variadic function info: maps function name to variadic metadata
|
|
variadic_functions: std.StringHashMap(VariadicInfo),
|
|
// Maps function name to resolved sx parameter types (for accurate type conversion at call sites)
|
|
fn_param_types: std.StringHashMap([]const Type),
|
|
// Enriched Any type entries: maps type name to tag + category + sx type
|
|
any_type_entries: std.StringHashMap(AnyTypeEntry),
|
|
// Current match arm type entries (set during category match arm body generation)
|
|
current_match_tags: ?[]const u64 = null,
|
|
// Functions deferred to compile after all types are registered (e.g. any_to_string)
|
|
deferred_fn_bodies: std.ArrayList(DeferredFn),
|
|
// Libraries to link against (from #library directives)
|
|
foreign_libraries: std.ArrayList([]const u8),
|
|
// Set of foreign function names (for ABI lowering at call sites)
|
|
foreign_fns: std.StringHashMap(void),
|
|
// Global mutable variables (from top-level var_decl, e.g. function pointers loaded at runtime)
|
|
global_mutable_vars: std.StringHashMap(NamedValue),
|
|
// Declared return types for non-generic functions (preserves signedness lost by LLVM round-trip)
|
|
function_return_types: std.StringHashMap(Type),
|
|
// Target configuration (triple, cpu, opt level, lib paths, linker)
|
|
target_config: TargetConfig = .{},
|
|
// Cached primitive LLVM types (initialized once in init(), avoids repeated FFI calls)
|
|
cached_i1: c.LLVMTypeRef = null,
|
|
cached_i8: c.LLVMTypeRef = null,
|
|
cached_i16: c.LLVMTypeRef = null,
|
|
cached_i32: c.LLVMTypeRef = null,
|
|
cached_i64: c.LLVMTypeRef = null,
|
|
cached_f32: c.LLVMTypeRef = null,
|
|
cached_f64: c.LLVMTypeRef = null,
|
|
cached_ptr: c.LLVMTypeRef = null,
|
|
cached_void: c.LLVMTypeRef = null,
|
|
|
|
const DeferredFn = struct {
|
|
fd: ast.FnDecl,
|
|
name: []const u8, // qualified name (may differ from fd.name for namespaced functions)
|
|
namespace: ?[]const u8 = null,
|
|
};
|
|
|
|
const TypeCategory = enum {
|
|
struct_cat,
|
|
enum_cat,
|
|
vector_cat,
|
|
array_cat,
|
|
slice_cat,
|
|
pointer_cat,
|
|
};
|
|
|
|
const AnyTypeEntry = struct {
|
|
tag_id: u64,
|
|
category: TypeCategory,
|
|
sx_type: Type,
|
|
};
|
|
|
|
const VariadicInfo = struct {
|
|
fixed_param_count: u32, // number of non-variadic params
|
|
element_type_name: []const u8, // element type of the variadic slice (e.g. "s32")
|
|
};
|
|
|
|
// GenericTemplate and GenericStructTemplate used to be single-field wrappers;
|
|
// now the hashmaps store ast.FnDecl / ast.StructDecl directly.
|
|
|
|
const ComptimeGlobal = struct {
|
|
global: c.LLVMValueRef, // LLVM global variable
|
|
ty: Type, // sx type
|
|
expr: *Node, // the inner expression to JIT-evaluate
|
|
is_resolved: bool = false, // true if initializer already set (no JIT needed)
|
|
};
|
|
|
|
const StructInfo = struct {
|
|
field_names: []const []const u8,
|
|
field_types: []const Type,
|
|
field_defaults: []const ?*Node,
|
|
llvm_type: c.LLVMTypeRef,
|
|
display_name: ?[]const u8 = null, // pretty name for generic instances
|
|
type_param_names: []const []const u8 = &.{}, // original type param names (e.g. ["T"])
|
|
type_param_types: []const Type = &.{}, // resolved types (e.g. [s32])
|
|
template_name: ?[]const u8 = null, // original template name (e.g. "List")
|
|
};
|
|
|
|
const TaggedEnumInfo = struct {
|
|
variant_names: []const []const u8,
|
|
variant_types: []const Type, // void_type for void variants
|
|
llvm_type: c.LLVMTypeRef, // layout struct or { tag, [max_payload_size x i8] }
|
|
max_payload_size: u64,
|
|
payload_field_index: c_uint = 1, // struct field index of the payload array
|
|
};
|
|
|
|
const PromotedField = struct {
|
|
struct_name: []const u8, // the anonymous struct type name
|
|
field_index: usize, // field index within that struct
|
|
field_type: Type, // type of the promoted field
|
|
};
|
|
|
|
const UnionInfo = struct {
|
|
field_names: []const []const u8,
|
|
field_types: []const Type,
|
|
llvm_type: c.LLVMTypeRef, // [max_size x i8]
|
|
total_size: u64,
|
|
promoted_fields: std.StringHashMap(PromotedField),
|
|
};
|
|
|
|
const TypeRegistryEntry = union(enum) {
|
|
struct_info: StructInfo,
|
|
tagged_enum: TaggedEnumInfo,
|
|
union_info: UnionInfo,
|
|
plain_enum: []const []const u8,
|
|
alias: []const u8,
|
|
};
|
|
|
|
// Scope stack entry: records what a name mapped to before being shadowed
|
|
const ScopeEntry = struct {
|
|
name: []const u8,
|
|
prev: ?NamedValue, // null = name didn't exist before this scope
|
|
};
|
|
|
|
const Scope = struct {
|
|
saves: std.ArrayList(ScopeEntry),
|
|
defers: std.ArrayList(*Node),
|
|
};
|
|
|
|
const NamedValue = struct {
|
|
ptr: c.LLVMValueRef, // alloca pointer
|
|
ty: Type, // sx type
|
|
is_const: bool = false,
|
|
};
|
|
|
|
/// Unified value lookup result — avoids sequential hash lookups at hot paths.
|
|
const ValueLookup = union(enum) {
|
|
local: NamedValue,
|
|
comptime_global: *ComptimeGlobal,
|
|
global_mutable: NamedValue,
|
|
|
|
fn ty(self: ValueLookup) Type {
|
|
return switch (self) {
|
|
.local, .global_mutable => |nv| nv.ty,
|
|
.comptime_global => |ct| ct.ty,
|
|
};
|
|
}
|
|
|
|
fn ptr(self: ValueLookup) c.LLVMValueRef {
|
|
return switch (self) {
|
|
.local, .global_mutable => |nv| nv.ptr,
|
|
.comptime_global => |ct| ct.global,
|
|
};
|
|
}
|
|
|
|
fn asNamedValue(self: ValueLookup) ?NamedValue {
|
|
return switch (self) {
|
|
.local, .global_mutable => |nv| nv,
|
|
.comptime_global => null,
|
|
};
|
|
}
|
|
};
|
|
|
|
pub fn init(allocator: std.mem.Allocator, module_name: [*:0]const u8, target_config: TargetConfig) CodeGen {
|
|
// Create context via ORC ThreadSafeContext for JIT compatibility
|
|
const ts_ctx = c.LLVMOrcCreateNewThreadSafeContext();
|
|
const ctx = c.LLVMOrcThreadSafeContextGetContext(ts_ctx);
|
|
const module = c.LLVMModuleCreateWithNameInContext(module_name, ctx);
|
|
const builder = c.LLVMCreateBuilderInContext(ctx);
|
|
|
|
// Initialize LLVM targets — native-only when targeting host, all for cross-compilation
|
|
if (target_config.triple == null) {
|
|
llvm.initNativeTarget();
|
|
} else {
|
|
llvm.initAllTargets();
|
|
}
|
|
|
|
const triple_owned = target_config.triple == null;
|
|
const triple = target_config.triple orelse c.LLVMGetDefaultTargetTriple();
|
|
defer if (triple_owned) c.LLVMDisposeMessage(@constCast(triple));
|
|
|
|
c.LLVMSetTarget(module, triple);
|
|
|
|
var target: c.LLVMTargetRef = null;
|
|
var err_msg: [*c]u8 = null;
|
|
var tm: c.LLVMTargetMachineRef = null;
|
|
if (c.LLVMGetTargetFromTriple(triple, &target, &err_msg) == 0) {
|
|
tm = c.LLVMCreateTargetMachine(
|
|
target,
|
|
triple,
|
|
target_config.getCpu(),
|
|
target_config.getFeatures(),
|
|
target_config.opt_level.toLLVM(),
|
|
c.LLVMRelocPIC,
|
|
c.LLVMCodeModelDefault,
|
|
);
|
|
const dl = c.LLVMCreateTargetDataLayout(tm);
|
|
c.LLVMSetModuleDataLayout(module, dl);
|
|
c.LLVMDisposeTargetData(dl);
|
|
} else {
|
|
if (err_msg != null) c.LLVMDisposeMessage(err_msg);
|
|
}
|
|
return .{
|
|
.context = ctx,
|
|
.module = module,
|
|
.builder = builder,
|
|
.allocator = allocator,
|
|
.ts_context = ts_ctx,
|
|
.target_machine = tm,
|
|
.named_values = std.StringHashMap(NamedValue).init(allocator),
|
|
.type_registry = std.StringHashMap(TypeRegistryEntry).init(allocator),
|
|
.flags_enum_types = std.StringHashMap(void).init(allocator),
|
|
.enum_variant_values = std.StringHashMap([]const i64).init(allocator),
|
|
.enum_backing_types = std.StringHashMap(c.LLVMTypeRef).init(allocator),
|
|
.builtins = null,
|
|
.current_function = null,
|
|
.scope_stack = std.ArrayList(Scope).empty,
|
|
.comptime_globals = std.StringHashMap(ComptimeGlobal).init(allocator),
|
|
.comptime_side_effects = std.ArrayList(*Node).empty,
|
|
.generic_templates = std.StringHashMap(ast.FnDecl).init(allocator),
|
|
.generic_instances = std.StringHashMap(c.LLVMValueRef).init(allocator),
|
|
.generic_struct_templates = std.StringHashMap(ast.StructDecl).init(allocator),
|
|
.namespaces = std.StringHashMap(void).init(allocator),
|
|
.builtin_functions = std.StringHashMap(void).init(allocator),
|
|
.fn_signatures = std.StringHashMap([]const u8).init(allocator),
|
|
.variadic_functions = std.StringHashMap(VariadicInfo).init(allocator),
|
|
.fn_param_types = std.StringHashMap([]const Type).init(allocator),
|
|
.any_type_id_map = std.StringHashMap(u64).init(allocator),
|
|
.any_type_entries = std.StringHashMap(AnyTypeEntry).init(allocator),
|
|
.deferred_fn_bodies = std.ArrayList(DeferredFn).empty,
|
|
.foreign_libraries = std.ArrayList([]const u8).empty,
|
|
.foreign_fns = std.StringHashMap(void).init(allocator),
|
|
.global_mutable_vars = std.StringHashMap(NamedValue).init(allocator),
|
|
.function_return_types = std.StringHashMap(Type).init(allocator),
|
|
.target_config = target_config,
|
|
.cached_i1 = c.LLVMInt1TypeInContext(ctx),
|
|
.cached_i8 = c.LLVMInt8TypeInContext(ctx),
|
|
.cached_i16 = c.LLVMInt16TypeInContext(ctx),
|
|
.cached_i32 = c.LLVMInt32TypeInContext(ctx),
|
|
.cached_i64 = c.LLVMInt64TypeInContext(ctx),
|
|
.cached_f32 = c.LLVMFloatTypeInContext(ctx),
|
|
.cached_f64 = c.LLVMDoubleTypeInContext(ctx),
|
|
.cached_ptr = c.LLVMPointerTypeInContext(ctx, 0),
|
|
.cached_void = c.LLVMVoidTypeInContext(ctx),
|
|
};
|
|
}
|
|
|
|
pub fn deinit(self: *CodeGen) void {
|
|
self.named_values.deinit();
|
|
self.type_registry.deinit();
|
|
self.comptime_globals.deinit();
|
|
self.enum_backing_types.deinit();
|
|
self.generic_templates.deinit();
|
|
self.generic_instances.deinit();
|
|
self.generic_struct_templates.deinit();
|
|
self.namespaces.deinit();
|
|
self.builtin_functions.deinit();
|
|
self.fn_signatures.deinit();
|
|
self.variadic_functions.deinit();
|
|
self.any_type_id_map.deinit();
|
|
self.any_type_entries.deinit();
|
|
self.deferred_fn_bodies.deinit(self.allocator);
|
|
self.foreign_libraries.deinit(self.allocator);
|
|
self.foreign_fns.deinit();
|
|
c.LLVMDisposeBuilder(self.builder);
|
|
if (self.target_machine) |tm| c.LLVMDisposeTargetMachine(tm);
|
|
if (self.module_owned) {
|
|
c.LLVMDisposeModule(self.module);
|
|
}
|
|
if (self.ts_context) |ts_ctx| {
|
|
c.LLVMOrcDisposeThreadSafeContext(ts_ctx);
|
|
} else {
|
|
c.LLVMContextDispose(self.context);
|
|
}
|
|
}
|
|
|
|
fn getStructInfo(self: *CodeGen, name: []const u8) !StructInfo {
|
|
return self.lookupStructInfo(name) orelse
|
|
return self.emitErrorFmt("unknown struct type '{s}'", .{name});
|
|
}
|
|
|
|
fn getTaggedEnumInfo(self: *CodeGen, name: []const u8) !TaggedEnumInfo {
|
|
return self.lookupTaggedEnumInfo(name) orelse
|
|
return self.emitErrorFmt("unknown enum type '{s}'", .{name});
|
|
}
|
|
|
|
fn lookupType(self: *CodeGen, name: []const u8, comptime tag: std.meta.Tag(TypeRegistryEntry)) ?switch (tag) {
|
|
.struct_info => StructInfo,
|
|
.tagged_enum => TaggedEnumInfo,
|
|
.union_info => UnionInfo,
|
|
.plain_enum => []const []const u8,
|
|
.alias => []const u8,
|
|
} {
|
|
if (self.type_registry.get(name)) |e| {
|
|
if (e == tag) return @field(e, @tagName(tag));
|
|
}
|
|
return null;
|
|
}
|
|
|
|
fn lookupStructInfo(self: *CodeGen, name: []const u8) ?StructInfo { return self.lookupType(name, .struct_info); }
|
|
fn lookupTaggedEnumInfo(self: *CodeGen, name: []const u8) ?TaggedEnumInfo { return self.lookupType(name, .tagged_enum); }
|
|
fn lookupUnionInfo(self: *CodeGen, name: []const u8) ?UnionInfo { return self.lookupType(name, .union_info); }
|
|
fn lookupEnumVariants(self: *CodeGen, name: []const u8) ?[]const []const u8 { return self.lookupType(name, .plain_enum); }
|
|
fn lookupAlias(self: *CodeGen, name: []const u8) ?[]const u8 { return self.lookupType(name, .alias); }
|
|
|
|
fn isRegisteredType(self: *CodeGen, name: []const u8) bool {
|
|
return self.type_registry.contains(name);
|
|
}
|
|
|
|
fn resolveElementType(self: *CodeGen, name: []const u8, comptime kind: []const u8) !Type {
|
|
return Type.fromName(name) orelse
|
|
return self.emitErrorFmt("unknown " ++ kind ++ " element type '{s}'", .{name});
|
|
}
|
|
|
|
fn emitError(self: *CodeGen, msg: []const u8) error{CodeGenError} {
|
|
if (self.diagnostics) |diags| diags.add(.err, msg, self.current_span);
|
|
return error.CodeGenError;
|
|
}
|
|
|
|
fn emitErrorFmt(self: *CodeGen, comptime fmt: []const u8, args: anytype) error{CodeGenError} {
|
|
if (self.diagnostics) |diags| diags.addFmt(.err, self.current_span, fmt, args);
|
|
return error.CodeGenError;
|
|
}
|
|
|
|
fn requireBuiltins(self: *CodeGen) !Builtins {
|
|
return self.builtins orelse return self.emitError("builtins not available");
|
|
}
|
|
|
|
/// Unified value lookup: checks locals, comptime globals, then global mutables.
|
|
fn lookupValue(self: *CodeGen, name: []const u8) ?ValueLookup {
|
|
if (self.named_values.get(name)) |nv| return .{ .local = nv };
|
|
if (self.comptime_globals.getPtr(name)) |ct| return .{ .comptime_global = ct };
|
|
if (self.global_mutable_vars.get(name)) |gm| return .{ .global_mutable = gm };
|
|
return null;
|
|
}
|
|
|
|
/// Build an alloca in the entry block of the current function so that
|
|
/// stack space is reserved once, not on every loop iteration.
|
|
fn buildEntryBlockAlloca(self: *CodeGen, ty: c.LLVMTypeRef, name: [*:0]const u8) c.LLVMValueRef {
|
|
const entry_bb = c.LLVMGetEntryBasicBlock(self.current_function);
|
|
const first_instr = c.LLVMGetFirstInstruction(entry_bb);
|
|
const tmp_builder = c.LLVMCreateBuilderInContext(self.context);
|
|
defer c.LLVMDisposeBuilder(tmp_builder);
|
|
if (first_instr != null) {
|
|
c.LLVMPositionBuilderBefore(tmp_builder, first_instr);
|
|
} else {
|
|
c.LLVMPositionBuilderAtEnd(tmp_builder, entry_bb);
|
|
}
|
|
return c.LLVMBuildAlloca(tmp_builder, ty, name);
|
|
}
|
|
|
|
/// Convert a Zig slice to a null-terminated C string using a caller-provided stack buffer.
|
|
/// Returns the stack-based result when it fits, or falls back to allocator.dupeZ.
|
|
fn nameToCStr(self: *CodeGen, name: []const u8, buf: *[256]u8) [*:0]const u8 {
|
|
if (name.len < 256) {
|
|
@memcpy(buf[0..name.len], name);
|
|
buf[name.len] = 0;
|
|
return @ptrCast(buf[0..name.len :0]);
|
|
}
|
|
const duped = self.allocator.dupeZ(u8, name) catch unreachable;
|
|
return duped.ptr;
|
|
}
|
|
|
|
fn buildNamedAlloca(self: *CodeGen, ty: c.LLVMTypeRef, name: []const u8) !c.LLVMValueRef {
|
|
var buf: [256]u8 = undefined;
|
|
return self.buildEntryBlockAlloca(ty, self.nameToCStr(name, &buf));
|
|
}
|
|
|
|
pub fn typeToLLVM(self: *CodeGen, ty: Type) c.LLVMTypeRef {
|
|
return switch (ty) {
|
|
.signed, .unsigned => |w| switch (w) {
|
|
1 => self.cached_i1.?,
|
|
8 => self.cached_i8.?,
|
|
16 => self.cached_i16.?,
|
|
32 => self.cached_i32.?,
|
|
64 => self.cached_i64.?,
|
|
else => c.LLVMIntTypeInContext(self.context, w),
|
|
},
|
|
.f32 => self.f32Type(),
|
|
.f64 => self.f64Type(),
|
|
.void_type => self.voidType(),
|
|
.boolean => self.i1Type(),
|
|
.string_type, .slice_type => self.getStringStructType(), // slices use same {ptr, i32} layout
|
|
.enum_type => |name| self.getEnumLLVMType(name),
|
|
.struct_type => |name| if (self.lookupStructInfo(name)) |info| info.llvm_type else unreachable,
|
|
.union_type => |name| if (self.lookupTaggedEnumInfo(name)) |info| info.llvm_type else if (self.lookupUnionInfo(name)) |info| info.llvm_type else unreachable,
|
|
.array_type => |info| {
|
|
const elem_ty = Type.fromName(info.element_name) orelse unreachable;
|
|
return c.LLVMArrayType2(self.typeToLLVM(elem_ty), info.length);
|
|
},
|
|
.vector_type => |info| {
|
|
const elem_ty = Type.fromName(info.element_name) orelse unreachable;
|
|
return c.LLVMVectorType(self.typeToLLVM(elem_ty), info.length);
|
|
},
|
|
.pointer_type, .many_pointer_type, .function_type => self.ptrType(),
|
|
.any_type => self.getAnyStructType(),
|
|
.meta_type => self.ptrType(),
|
|
};
|
|
}
|
|
|
|
fn getEnumLLVMType(self: *CodeGen, enum_name: []const u8) c.LLVMTypeRef {
|
|
if (self.enum_backing_types.get(enum_name)) |llvm_ty| return llvm_ty;
|
|
return self.i64Type();
|
|
}
|
|
|
|
fn getAnyStructType(self: *CodeGen) c.LLVMTypeRef {
|
|
if (self.any_struct_type) |t| return t;
|
|
var field_types = [_]c.LLVMTypeRef{
|
|
self.i64Type(), // type tag
|
|
self.i64Type(), // value (fits all primitives)
|
|
};
|
|
self.any_struct_type = c.LLVMStructTypeInContext(self.context, &field_types, 2, 0);
|
|
return self.any_struct_type.?;
|
|
}
|
|
|
|
/// Type tag constants for Any type (builtins: 0-6, Type: 10, named types: 7+ dynamic)
|
|
const ANY_TAG_VOID: u64 = 0;
|
|
const ANY_TAG_BOOL: u64 = 1;
|
|
const ANY_TAG_S32: u64 = 2;
|
|
const ANY_TAG_S64: u64 = 3;
|
|
const ANY_TAG_F32: u64 = 4;
|
|
const ANY_TAG_F64: u64 = 5;
|
|
const ANY_TAG_STRING: u64 = 6;
|
|
const ANY_TAG_TYPE: u64 = 10;
|
|
|
|
/// Get or assign a unique type ID for a named type (struct, enum, union, vector, array).
|
|
/// IDs start at 7 and are assigned dynamically per compilation.
|
|
/// Also populates `any_type_entries` with category and type info.
|
|
fn getAnyTypeId(self: *CodeGen, name: []const u8, sx_type: Type) !u64 {
|
|
const gop = try self.any_type_id_map.getOrPut(name);
|
|
if (!gop.found_existing) {
|
|
gop.value_ptr.* = self.next_any_type_id;
|
|
self.next_any_type_id += 1;
|
|
// Skip over reserved slot 10 (ANY_TAG_TYPE)
|
|
if (self.next_any_type_id == ANY_TAG_TYPE) self.next_any_type_id += 1;
|
|
|
|
// Determine category from the sx type
|
|
const category: TypeCategory = switch (sx_type) {
|
|
.struct_type => .struct_cat,
|
|
.enum_type => .enum_cat,
|
|
.union_type => .enum_cat,
|
|
.vector_type => .vector_cat,
|
|
.array_type => .array_cat,
|
|
.slice_type => .slice_cat,
|
|
.pointer_type, .many_pointer_type => .pointer_cat,
|
|
else => .struct_cat, // fallback
|
|
};
|
|
try self.any_type_entries.put(name, .{
|
|
.tag_id = gop.value_ptr.*,
|
|
.category = category,
|
|
.sx_type = sx_type,
|
|
});
|
|
}
|
|
return gop.value_ptr.*;
|
|
}
|
|
|
|
/// Check if a function should have its body compilation deferred until after all types are registered.
|
|
/// Functions with non-variadic Any parameters (like any_to_string) use type-based match expressions
|
|
/// that need all types registered before compilation.
|
|
fn shouldDeferFnBody(fd: ast.FnDecl) bool {
|
|
for (fd.params) |param| {
|
|
if (!param.is_variadic and param.type_expr.data == .type_expr and
|
|
std.mem.eql(u8, param.type_expr.data.type_expr.name, "Any"))
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/// Pre-register a type in the Any type system so category matching (case slice:, case array:, etc.)
|
|
/// works in any_to_string even before buildAnyValue is called for this type.
|
|
fn preRegisterAnyType(self: *CodeGen, sx_type: Type) !void {
|
|
switch (sx_type) {
|
|
.struct_type => |name| {
|
|
_ = try self.getAnyTypeId(name, sx_type);
|
|
// Recursively register struct field types
|
|
if (self.lookupStructInfo(name)) |info| {
|
|
for (info.field_types) |ft| {
|
|
try self.preRegisterAnyType(ft);
|
|
}
|
|
}
|
|
},
|
|
.enum_type => |name| _ = try self.getAnyTypeId(name, sx_type),
|
|
.union_type => |name| _ = try self.getAnyTypeId(name, sx_type),
|
|
.vector_type => |info| _ = try self.getAnyTypeId(try std.fmt.allocPrint(self.allocator, "vec[{d}]{s}", .{ info.length, info.element_name }), sx_type),
|
|
.array_type => |info| _ = try self.getAnyTypeId(try std.fmt.allocPrint(self.allocator, "[{d}]{s}", .{ info.length, info.element_name }), sx_type),
|
|
.slice_type => |info| _ = try self.getAnyTypeId(try std.fmt.allocPrint(self.allocator, "[]{s}", .{info.element_name}), sx_type),
|
|
.pointer_type => |info| _ = try self.getAnyTypeId(try std.fmt.allocPrint(self.allocator, "*{s}", .{info.pointee_name}), sx_type),
|
|
.many_pointer_type => |info| _ = try self.getAnyTypeId(try std.fmt.allocPrint(self.allocator, "[*]{s}", .{info.element_name}), sx_type),
|
|
else => {},
|
|
}
|
|
}
|
|
|
|
/// Build an Any value { tag: i32, value: i64 } from a typed LLVM value.
|
|
/// Small values (ints, floats, bools, enums) are stored inline in the i64.
|
|
/// Complex values (strings, structs, unions) are stored via pointer (alloca + ptr-to-int).
|
|
fn buildAnyValue(self: *CodeGen, val: c.LLVMValueRef, in_ty: Type) !c.LLVMValueRef {
|
|
const any_ty = self.getAnyStructType();
|
|
const i64_ty = self.i64Type();
|
|
const undef = self.getUndef(any_ty);
|
|
|
|
// []u8 boxes as string (same repr, same Any tag)
|
|
const ty: Type = if (in_ty.isSlice() and std.mem.eql(u8, in_ty.slice_type.element_name, "u8"))
|
|
.string_type
|
|
else
|
|
in_ty;
|
|
|
|
// Determine tag
|
|
const tag: u64 = switch (ty) {
|
|
.void_type => ANY_TAG_VOID,
|
|
.boolean => ANY_TAG_BOOL,
|
|
.signed => |w| if (w <= 32) ANY_TAG_S32 else ANY_TAG_S64,
|
|
.unsigned => |w| if (w <= 32) ANY_TAG_S32 else ANY_TAG_S64,
|
|
.f32 => ANY_TAG_F32,
|
|
.f64 => ANY_TAG_F64,
|
|
.string_type => ANY_TAG_STRING,
|
|
.struct_type => |name| try self.getAnyTypeId(name, ty),
|
|
.enum_type => |name| try self.getAnyTypeId(name, ty),
|
|
.union_type => |name| try self.getAnyTypeId(name, ty),
|
|
.vector_type => |info| try self.getAnyTypeId(try std.fmt.allocPrint(self.allocator, "vec[{d}]{s}", .{ info.length, info.element_name }), ty),
|
|
.array_type => |info| try self.getAnyTypeId(try std.fmt.allocPrint(self.allocator, "[{d}]{s}", .{ info.length, info.element_name }), ty),
|
|
.slice_type => |info| try self.getAnyTypeId(try std.fmt.allocPrint(self.allocator, "[]{s}", .{info.element_name}), ty),
|
|
.pointer_type => |info| try self.getAnyTypeId(try std.fmt.allocPrint(self.allocator, "*{s}", .{info.pointee_name}), ty),
|
|
.many_pointer_type => |info| try self.getAnyTypeId(try std.fmt.allocPrint(self.allocator, "[*]{s}", .{info.element_name}), ty),
|
|
.meta_type => ANY_TAG_TYPE,
|
|
else => ANY_TAG_S32,
|
|
};
|
|
const tag_val = c.LLVMConstInt(i64_ty, tag, 0);
|
|
const with_tag = self.insertValue(undef, tag_val, 0, "any_tag");
|
|
|
|
// Convert value to i64
|
|
const val_as_i64 = switch (ty) {
|
|
.void_type => c.LLVMConstInt(i64_ty, 0, 0),
|
|
.boolean => self.zExt(val, i64_ty, "any_bool"),
|
|
.signed => |w| if (w <= 32)
|
|
self.sExt(val, i64_ty, "any_int")
|
|
else
|
|
val,
|
|
.unsigned => |w| if (w <= 32)
|
|
self.zExt(val, i64_ty, "any_uint")
|
|
else
|
|
val,
|
|
.f32 => blk: {
|
|
// f32 -> f64 -> bitcast to i64
|
|
const as_f64 = c.LLVMBuildFPExt(self.builder, val, self.f64Type(), "f32_to_f64");
|
|
break :blk self.bitCast(as_f64, i64_ty, "any_f32");
|
|
},
|
|
.f64 => self.bitCast(val, i64_ty, "any_f64"),
|
|
.string_type => self.allocaStoreAsI64(self.getStringStructType(), val, "any_str"),
|
|
.struct_type => |sname| blk: {
|
|
// Struct — store to alloca, pass pointer as i64
|
|
const info = self.lookupStructInfo(sname) orelse
|
|
return self.getUndef(any_ty);
|
|
break :blk self.allocaStoreAsI64(info.llvm_type, val, "any_struct");
|
|
},
|
|
.enum_type => |ename| blk: {
|
|
// Enum — extend to i64 for Any storage (no-op if already i64)
|
|
const enum_llvm_ty = self.getEnumLLVMType(ename);
|
|
const enum_bits = c.LLVMGetIntTypeWidth(enum_llvm_ty);
|
|
if (enum_bits < 64)
|
|
break :blk self.zExt(val, i64_ty, "any_enum")
|
|
else
|
|
break :blk val;
|
|
},
|
|
.union_type => |uname| blk: {
|
|
// Union — store to alloca, pass pointer as i64
|
|
const info = self.lookupTaggedEnumInfo(uname) orelse
|
|
return self.getUndef(any_ty);
|
|
break :blk self.allocaStoreAsI64(info.llvm_type, val, "any_union");
|
|
},
|
|
.vector_type, .array_type => self.allocaStoreAsI64(self.typeToLLVM(ty), val, "any_vec"),
|
|
.slice_type => self.allocaStoreAsI64(self.getStringStructType(), val, "any_slice"),
|
|
.pointer_type, .many_pointer_type => self.ptrToInt(val, "any_ptr"),
|
|
.meta_type => |mt| self.allocaStoreAsI64(self.getStringStructType(), self.buildStringSlice(val, self.constInt64(mt.name.len)), "any_type"),
|
|
else => self.sExt(val, i64_ty, "any_val"),
|
|
};
|
|
return self.insertValue(with_tag, val_as_i64, 1, "any_value");
|
|
}
|
|
|
|
fn getStringStructType(self: *CodeGen) c.LLVMTypeRef {
|
|
if (self.string_struct_type) |t| return t;
|
|
var field_types = [_]c.LLVMTypeRef{
|
|
self.ptrType(), // ptr
|
|
self.i64Type(), // len
|
|
};
|
|
self.string_struct_type = c.LLVMStructTypeInContext(self.context, &field_types, 2, 0);
|
|
return self.string_struct_type.?;
|
|
}
|
|
|
|
/// Build a fat pointer {ptr, len} struct from a type, pointer, and length value.
|
|
fn buildFatPointer(self: *CodeGen, ty: c.LLVMTypeRef, ptr: c.LLVMValueRef, len: c.LLVMValueRef) c.LLVMValueRef {
|
|
const undef = self.getUndef(ty);
|
|
const with_ptr = self.insertValue(undef, ptr, 0, "ptr");
|
|
return self.insertValue(with_ptr, len, 1, "len");
|
|
}
|
|
|
|
/// Build a string slice {ptr, len} from a raw pointer and a length value.
|
|
fn buildStringSlice(self: *CodeGen, ptr: c.LLVMValueRef, len_val: c.LLVMValueRef) c.LLVMValueRef {
|
|
return self.buildFatPointer(self.getStringStructType(), ptr, len_val);
|
|
}
|
|
|
|
// LLVM type shortcuts (cached — no FFI call)
|
|
fn i1Type(self: *CodeGen) c.LLVMTypeRef { return self.cached_i1.?; }
|
|
fn i8Type(self: *CodeGen) c.LLVMTypeRef { return self.cached_i8.?; }
|
|
fn i32Type(self: *CodeGen) c.LLVMTypeRef { return self.cached_i32.?; }
|
|
fn i64Type(self: *CodeGen) c.LLVMTypeRef { return self.cached_i64.?; }
|
|
fn f32Type(self: *CodeGen) c.LLVMTypeRef { return self.cached_f32.?; }
|
|
fn f64Type(self: *CodeGen) c.LLVMTypeRef { return self.cached_f64.?; }
|
|
fn ptrType(self: *CodeGen) c.LLVMTypeRef { return self.cached_ptr.?; }
|
|
fn voidType(self: *CodeGen) c.LLVMTypeRef { return self.cached_void.?; }
|
|
|
|
fn gepArrayElement(self: *CodeGen, arr_ty: c.LLVMTypeRef, arr_ptr: c.LLVMValueRef, idx: c.LLVMValueRef, name: [*c]const u8) c.LLVMValueRef {
|
|
var indices = [_]c.LLVMValueRef{ self.constInt32(0), idx };
|
|
return c.LLVMBuildGEP2(self.builder, arr_ty, arr_ptr, &indices, 2, name);
|
|
}
|
|
|
|
fn gepPointerElement(self: *CodeGen, elem_ty: c.LLVMTypeRef, ptr: c.LLVMValueRef, idx: c.LLVMValueRef, name: [*c]const u8) c.LLVMValueRef {
|
|
var indices = [_]c.LLVMValueRef{idx};
|
|
return c.LLVMBuildGEP2(self.builder, elem_ty, ptr, &indices, 1, name);
|
|
}
|
|
|
|
fn arrayDecayToPointer(self: *CodeGen, arr_ty: c.LLVMTypeRef, arr_ptr: c.LLVMValueRef, name: [*c]const u8) c.LLVMValueRef {
|
|
const zero = self.constInt64(0);
|
|
var indices = [_]c.LLVMValueRef{ zero, zero };
|
|
return c.LLVMBuildGEP2(self.builder, arr_ty, arr_ptr, &indices, 2, name);
|
|
}
|
|
|
|
fn structGEP(self: *CodeGen, struct_ty: c.LLVMTypeRef, ptr: c.LLVMValueRef, idx: c_uint, name: [*c]const u8) c.LLVMValueRef {
|
|
return c.LLVMBuildStructGEP2(self.builder, struct_ty, ptr, idx, name);
|
|
}
|
|
|
|
fn storeStructField(self: *CodeGen, struct_ty: c.LLVMTypeRef, ptr: c.LLVMValueRef, field_idx: c_uint, val: c.LLVMValueRef) void {
|
|
_ = c.LLVMBuildStore(self.builder, val, self.structGEP(struct_ty, ptr, field_idx, "field"));
|
|
}
|
|
|
|
fn loadStructField(self: *CodeGen, struct_ty: c.LLVMTypeRef, ptr: c.LLVMValueRef, field_idx: c_uint, field_ty: c.LLVMTypeRef) c.LLVMValueRef {
|
|
return c.LLVMBuildLoad2(self.builder, field_ty, self.structGEP(struct_ty, ptr, field_idx, "field"), "fieldval");
|
|
}
|
|
|
|
fn storeUndef(self: *CodeGen, ty: c.LLVMTypeRef, ptr: c.LLVMValueRef) void {
|
|
_ = c.LLVMBuildStore(self.builder, self.getUndef(ty), ptr);
|
|
}
|
|
|
|
fn storeNull(self: *CodeGen, ty: c.LLVMTypeRef, ptr: c.LLVMValueRef) void {
|
|
_ = c.LLVMBuildStore(self.builder, c.LLVMConstNull(ty), ptr);
|
|
}
|
|
|
|
fn loadTyped(self: *CodeGen, ty: Type, ptr: c.LLVMValueRef, name: [*c]const u8) c.LLVMValueRef {
|
|
return c.LLVMBuildLoad2(self.builder, self.typeToLLVM(ty), ptr, name);
|
|
}
|
|
|
|
const SwitchBlock = struct {
|
|
merge_bb: c.LLVMBasicBlockRef,
|
|
default_bb: c.LLVMBasicBlockRef,
|
|
sw: c.LLVMValueRef,
|
|
};
|
|
|
|
fn buildSwitch(self: *CodeGen, cond: c.LLVMValueRef, case_count: c_uint, merge_name: [*c]const u8, default_name: [*c]const u8) SwitchBlock {
|
|
const merge_bb = self.appendBB(merge_name);
|
|
const default_bb = self.appendBB(default_name);
|
|
const sw = c.LLVMBuildSwitch(self.builder, cond, default_bb, case_count);
|
|
return .{ .merge_bb = merge_bb, .default_bb = default_bb, .sw = sw };
|
|
}
|
|
|
|
fn loadIfPointer(self: *CodeGen, val: c.LLVMValueRef, ty: c.LLVMTypeRef, name: [*c]const u8) c.LLVMValueRef {
|
|
if (c.LLVMGetTypeKind(c.LLVMTypeOf(val)) == c.LLVMPointerTypeKind) {
|
|
return c.LLVMBuildLoad2(self.builder, ty, val, name);
|
|
}
|
|
return val;
|
|
}
|
|
|
|
fn loadFromI64Ptr(self: *CodeGen, i64_val: c.LLVMValueRef, ty: c.LLVMTypeRef, name: [*c]const u8) c.LLVMValueRef {
|
|
const ptr = self.intToPtr(i64_val, name);
|
|
return c.LLVMBuildLoad2(self.builder, ty, ptr, name);
|
|
}
|
|
|
|
fn resolveAlias(self: *CodeGen, name: []const u8) []const u8 {
|
|
return self.lookupAlias(name) orelse name;
|
|
}
|
|
|
|
fn buildPhiNode(self: *CodeGen, phi_vals: *std.ArrayList(c.LLVMValueRef), phi_bbs: *std.ArrayList(c.LLVMBasicBlockRef), ty: c.LLVMTypeRef, name: [*c]const u8) !c.LLVMValueRef {
|
|
const vals_slice = try phi_vals.toOwnedSlice(self.allocator);
|
|
const bbs_slice = try phi_bbs.toOwnedSlice(self.allocator);
|
|
const phi = c.LLVMBuildPhi(self.builder, ty, name);
|
|
c.LLVMAddIncoming(phi, vals_slice.ptr, bbs_slice.ptr, @intCast(vals_slice.len));
|
|
return phi;
|
|
}
|
|
|
|
fn addPhiCase(self: *CodeGen, phi_vals: *std.ArrayList(c.LLVMValueRef), phi_bbs: *std.ArrayList(c.LLVMBasicBlockRef), val: c.LLVMValueRef, merge_bb: c.LLVMBasicBlockRef) !void {
|
|
try phi_vals.append(self.allocator, val);
|
|
try phi_bbs.append(self.allocator, self.getCurrentBlock());
|
|
self.br(merge_bb);
|
|
}
|
|
|
|
fn getTypeSize(self: *CodeGen, ty: c.LLVMTypeRef) u64 {
|
|
return c.LLVMStoreSizeOfType(c.LLVMGetModuleDataLayout(self.module), ty);
|
|
}
|
|
|
|
fn appendBlock(self: *CodeGen, function: c.LLVMValueRef, name: [*c]const u8) c.LLVMBasicBlockRef {
|
|
const bb = c.LLVMAppendBasicBlockInContext(self.context, function, name);
|
|
self.positionAt(bb);
|
|
return bb;
|
|
}
|
|
|
|
fn valueToBool(self: *CodeGen, val: c.LLVMValueRef) c.LLVMValueRef {
|
|
if (c.LLVMTypeOf(val) == self.i1Type()) return val;
|
|
return self.icmp(c.LLVMIntNE, val, c.LLVMConstInt(c.LLVMTypeOf(val), 0, 0), "tobool");
|
|
}
|
|
|
|
fn constInt64(self: *CodeGen, val: u64) c.LLVMValueRef {
|
|
return c.LLVMConstInt(self.i64Type(), val, 0);
|
|
}
|
|
|
|
fn constInt32(self: *CodeGen, val: u32) c.LLVMValueRef {
|
|
return c.LLVMConstInt(self.i32Type(), val, 0);
|
|
}
|
|
|
|
fn extractValue(self: *CodeGen, val: c.LLVMValueRef, idx: c_uint, name: [*c]const u8) c.LLVMValueRef {
|
|
return c.LLVMBuildExtractValue(self.builder, val, idx, name);
|
|
}
|
|
|
|
fn buildGlobalString(self: *CodeGen, str: [*c]const u8, name: [*c]const u8) c.LLVMValueRef {
|
|
return c.LLVMBuildGlobalStringPtr(self.builder, str, name);
|
|
}
|
|
|
|
fn insertValue(self: *CodeGen, aggr: c.LLVMValueRef, val: c.LLVMValueRef, idx: c_uint, name: [*c]const u8) c.LLVMValueRef {
|
|
return c.LLVMBuildInsertValue(self.builder, aggr, val, idx, name);
|
|
}
|
|
|
|
fn positionAt(self: *CodeGen, bb: c.LLVMBasicBlockRef) void {
|
|
c.LLVMPositionBuilderAtEnd(self.builder, bb);
|
|
}
|
|
|
|
fn br(self: *CodeGen, dest: c.LLVMBasicBlockRef) void {
|
|
_ = c.LLVMBuildBr(self.builder, dest);
|
|
}
|
|
|
|
fn condBr(self: *CodeGen, cond: c.LLVMValueRef, then_bb: c.LLVMBasicBlockRef, else_bb: c.LLVMBasicBlockRef) void {
|
|
_ = c.LLVMBuildCondBr(self.builder, cond, then_bb, else_bb);
|
|
}
|
|
|
|
fn allocaStoreAsI64(self: *CodeGen, ty: c.LLVMTypeRef, val: c.LLVMValueRef, name: [*c]const u8) c.LLVMValueRef {
|
|
const alloca = self.buildEntryBlockAlloca(ty, name);
|
|
_ = c.LLVMBuildStore(self.builder, val, alloca);
|
|
return self.ptrToInt(alloca, name);
|
|
}
|
|
|
|
fn trunc(self: *CodeGen, val: c.LLVMValueRef, ty: c.LLVMTypeRef, name: [*c]const u8) c.LLVMValueRef {
|
|
return c.LLVMBuildTrunc(self.builder, val, ty, name);
|
|
}
|
|
|
|
fn zExt(self: *CodeGen, val: c.LLVMValueRef, ty: c.LLVMTypeRef, name: [*c]const u8) c.LLVMValueRef {
|
|
return c.LLVMBuildZExt(self.builder, val, ty, name);
|
|
}
|
|
|
|
fn sExt(self: *CodeGen, val: c.LLVMValueRef, ty: c.LLVMTypeRef, name: [*c]const u8) c.LLVMValueRef {
|
|
return c.LLVMBuildSExt(self.builder, val, ty, name);
|
|
}
|
|
|
|
fn bitCast(self: *CodeGen, val: c.LLVMValueRef, ty: c.LLVMTypeRef, name: [*c]const u8) c.LLVMValueRef {
|
|
return c.LLVMBuildBitCast(self.builder, val, ty, name);
|
|
}
|
|
|
|
fn appendBB(self: *CodeGen, name: [*c]const u8) c.LLVMBasicBlockRef {
|
|
return c.LLVMAppendBasicBlockInContext(self.context, self.current_function, name);
|
|
}
|
|
|
|
fn getCurrentBlock(self: *CodeGen) c.LLVMBasicBlockRef {
|
|
return c.LLVMGetInsertBlock(self.builder);
|
|
}
|
|
|
|
fn ret(self: *CodeGen, val: c.LLVMValueRef) void {
|
|
_ = c.LLVMBuildRet(self.builder, val);
|
|
}
|
|
|
|
fn retVoid(self: *CodeGen) void {
|
|
_ = c.LLVMBuildRetVoid(self.builder);
|
|
}
|
|
|
|
fn intToPtr(self: *CodeGen, val: c.LLVMValueRef, name: [*c]const u8) c.LLVMValueRef {
|
|
return c.LLVMBuildIntToPtr(self.builder, val, self.ptrType(), name);
|
|
}
|
|
|
|
fn ptrToInt(self: *CodeGen, ptr: c.LLVMValueRef, name: [*c]const u8) c.LLVMValueRef {
|
|
return c.LLVMBuildPtrToInt(self.builder, ptr, self.i64Type(), name);
|
|
}
|
|
|
|
fn icmp(self: *CodeGen, pred: c.LLVMIntPredicate, lhs: c.LLVMValueRef, rhs: c.LLVMValueRef, name: [*c]const u8) c.LLVMValueRef {
|
|
return c.LLVMBuildICmp(self.builder, pred, lhs, rhs, name);
|
|
}
|
|
|
|
fn getUndef(_: *CodeGen, ty: c.LLVMTypeRef) c.LLVMValueRef {
|
|
return c.LLVMGetUndef(ty);
|
|
}
|
|
|
|
/// Extract .len or .ptr from a fat pointer value ({ptr, len} struct).
|
|
fn extractFatPtrField(self: *CodeGen, val: c.LLVMValueRef, field: []const u8, type_name: []const u8) !c.LLVMValueRef {
|
|
if (std.mem.eql(u8, field, "len")) {
|
|
return self.extractValue(val, 1, "len");
|
|
}
|
|
if (std.mem.eql(u8, field, "ptr")) {
|
|
return self.extractValue(val, 0, "ptr");
|
|
}
|
|
return self.emitErrorFmt("no field '{s}' on {s} (available: .len, .ptr)", .{ field, type_name });
|
|
}
|
|
|
|
fn pushScope(self: *CodeGen) !void {
|
|
var saves = std.ArrayList(ScopeEntry).empty;
|
|
try saves.ensureTotalCapacity(self.allocator, 8);
|
|
var defers = std.ArrayList(*Node).empty;
|
|
try defers.ensureTotalCapacity(self.allocator, 4);
|
|
try self.scope_stack.append(self.allocator, .{ .saves = saves, .defers = defers });
|
|
}
|
|
|
|
fn popScope(self: *CodeGen) !void {
|
|
if (self.scope_stack.items.len == 0) return;
|
|
const scope = self.scope_stack.items[self.scope_stack.items.len - 1];
|
|
|
|
// 1. Execute deferred expressions in LIFO order
|
|
var i: usize = scope.defers.items.len;
|
|
while (i > 0) {
|
|
i -= 1;
|
|
_ = try self.genExpr(scope.defers.items[i]);
|
|
}
|
|
|
|
// 2. Restore shadowed variables in reverse order
|
|
i = scope.saves.items.len;
|
|
while (i > 0) {
|
|
i -= 1;
|
|
const entry = scope.saves.items[i];
|
|
if (entry.prev) |prev| {
|
|
self.named_values.putAssumeCapacity(entry.name, prev);
|
|
} else {
|
|
_ = self.named_values.remove(entry.name);
|
|
}
|
|
}
|
|
|
|
_ = self.scope_stack.pop();
|
|
}
|
|
|
|
/// Emit all pending deferred expressions from all active scopes (LIFO order,
|
|
/// innermost scope first). Does NOT pop the stacks — used before `return`
|
|
/// so that popScope() can still clean up the data structures later.
|
|
fn emitAllDefers(self: *CodeGen) !void {
|
|
var i: usize = self.scope_stack.items.len;
|
|
while (i > 0) {
|
|
i -= 1;
|
|
const defers = self.scope_stack.items[i].defers;
|
|
var j: usize = defers.items.len;
|
|
while (j > 0) {
|
|
j -= 1;
|
|
_ = try self.genExpr(defers.items[j]);
|
|
}
|
|
}
|
|
}
|
|
|
|
fn saveShadowed(self: *CodeGen, name: []const u8) !void {
|
|
if (self.scope_stack.items.len == 0) return;
|
|
const top = &self.scope_stack.items[self.scope_stack.items.len - 1].saves;
|
|
const prev = self.named_values.get(name);
|
|
try top.append(self.allocator, .{ .name = name, .prev = prev });
|
|
}
|
|
|
|
fn registerVariable(self: *CodeGen, name: []const u8, ptr: c.LLVMValueRef, ty: Type) !void {
|
|
try self.saveShadowed(name);
|
|
try self.named_values.put(name, .{ .ptr = ptr, .ty = ty });
|
|
}
|
|
|
|
pub fn generate(self: *CodeGen, root: *Node) !void {
|
|
if (root.data != .root) return self.emitError("expected root node for code generation");
|
|
// Store root decls for VM on-demand function compilation
|
|
self.root_decls = root.data.root.decls;
|
|
// Initialize built-in function declarations (printf, etc.)
|
|
self.builtins = Builtins.init(self.module, self.context);
|
|
|
|
// Pass 1: Register all declarations (signatures only, no bodies)
|
|
for (root.data.root.decls) |decl| {
|
|
switch (decl.data) {
|
|
.fn_decl => |fd| {
|
|
if (fd.body.data == .builtin_expr) {
|
|
try self.builtin_functions.put(fd.name, {});
|
|
} else if (fd.body.data == .foreign_expr) {
|
|
// External C function — register LLVM declaration (no body)
|
|
try self.registerFnDecl(fd, fd.name);
|
|
} else if (fd.type_params.len > 0) {
|
|
try self.generic_templates.put(fd.name, fd);
|
|
} else {
|
|
try self.registerFnDecl(fd, fd.name);
|
|
}
|
|
try self.fn_signatures.put(fd.name, self.buildFnSignature(fd));
|
|
},
|
|
.library_decl => |ld| {
|
|
try self.foreign_libraries.append(self.allocator, ld.lib_name);
|
|
},
|
|
.enum_decl => |ed| {
|
|
if (ed.variant_types.len > 0) {
|
|
// Tagged enum with payloads
|
|
try self.registerTaggedEnum(ed);
|
|
} else {
|
|
// Payload-less enum
|
|
try self.type_registry.put(ed.name, .{ .plain_enum = ed.variant_names });
|
|
_ = try self.getAnyTypeId(ed.name, .{ .enum_type = ed.name });
|
|
|
|
if (ed.is_flags) {
|
|
try self.flags_enum_types.put(ed.name, {});
|
|
}
|
|
|
|
// Register backing type if specified
|
|
if (ed.backing_type) |bt_node| {
|
|
const bt = self.resolveType(bt_node);
|
|
try self.enum_backing_types.put(ed.name, self.typeToLLVM(bt));
|
|
}
|
|
|
|
// Compute and store variant values
|
|
const values = try self.allocator.alloc(i64, ed.variant_names.len);
|
|
for (ed.variant_names, 0..) |_, i| {
|
|
if (ed.variant_values.len > i and ed.variant_values[i] != null) {
|
|
// Explicit value: evaluate comptime int literal
|
|
const val_node = ed.variant_values[i].?;
|
|
values[i] = switch (val_node.data) {
|
|
.int_literal => |il| il.value,
|
|
else => @as(i64, @intCast(i)), // fallback
|
|
};
|
|
} else if (ed.is_flags) {
|
|
// Auto power-of-2: 1, 2, 4, 8, ...
|
|
values[i] = @as(i64, 1) << @intCast(i);
|
|
} else {
|
|
// Regular enum: sequential 0, 1, 2, ...
|
|
values[i] = @intCast(i);
|
|
}
|
|
}
|
|
try self.enum_variant_values.put(ed.name, values);
|
|
}
|
|
},
|
|
.struct_decl => |sd| try self.registerStructType(sd),
|
|
.union_decl => |ud| try self.registerUnionType(ud),
|
|
.const_decl => |cd| {
|
|
if (cd.value.data == .builtin_expr) {
|
|
// #builtin constant — skip codegen
|
|
} else if (cd.value.data == .lambda) {
|
|
try self.registerLambdaAsFunction(cd.name, cd.value.data.lambda);
|
|
} else if (cd.value.data == .type_expr) {
|
|
try self.type_registry.put(cd.name, .{ .alias = cd.value.data.type_expr.name });
|
|
} else if (cd.value.data == .call) {
|
|
// Check if this is a generic struct or type function instantiation
|
|
const callee_name = if (cd.value.data.call.callee.data == .identifier)
|
|
cd.value.data.call.callee.data.identifier.name
|
|
else
|
|
null;
|
|
if (callee_name) |cn| {
|
|
if (self.generic_struct_templates.get(cn)) |tmpl| {
|
|
// Generic struct instantiation: Vec3 :: Vec(3, f32);
|
|
const result_ty = try self.instantiateGenericStruct(cn, tmpl, cd.value.data.call.args);
|
|
if (result_ty.isStruct()) {
|
|
try self.type_registry.put(cd.name, .{ .alias = result_ty.struct_type });
|
|
}
|
|
} else if (self.generic_templates.get(cn)) |tmpl| {
|
|
// Type-returning function: Foo :: Complex(u32);
|
|
const result_ty = try self.instantiateTypeFunction(cd.name, cn, tmpl, cd.value.data.call.args);
|
|
if (result_ty.isStruct()) {
|
|
try self.type_registry.put(cd.name, .{ .alias = result_ty.struct_type });
|
|
} else if (result_ty.isUnion()) {
|
|
try self.type_registry.put(cd.name, .{ .alias = result_ty.union_type });
|
|
}
|
|
} else if (self.builtin_functions.contains(cn)) {
|
|
// Builtin type function (e.g., Vector(4, f32), Array(5, s32))
|
|
if (self.resolveBuiltinType(cn, cd.value.data.call.args)) |result_ty| {
|
|
const display = try result_ty.displayName(self.allocator);
|
|
try self.type_registry.put(cd.name, .{ .alias = display });
|
|
} else {
|
|
try self.registerTopLevelConstant(cd);
|
|
}
|
|
} else {
|
|
try self.registerTopLevelConstant(cd);
|
|
}
|
|
} else {
|
|
try self.registerTopLevelConstant(cd);
|
|
}
|
|
} else if (cd.value.data == .comptime_expr) {
|
|
// Use explicit type annotation if available
|
|
const ct_type_override: ?Type = if (cd.type_annotation) |te| Type.fromTypeExpr(te) else null;
|
|
try self.registerComptimeGlobal(cd.name, cd.value.data.comptime_expr.expr, ct_type_override);
|
|
} else {
|
|
// Top-level value constant (e.g., SPECIAL_VALUE :u8: 42;)
|
|
try self.registerTopLevelConstant(cd);
|
|
}
|
|
},
|
|
.comptime_expr => |ct| {
|
|
try self.comptime_side_effects.append(self.allocator, ct.expr);
|
|
},
|
|
.namespace_decl => |ns| {
|
|
try self.registerNamespace(ns);
|
|
},
|
|
.var_decl => |vd| {
|
|
try self.registerGlobalVar(vd);
|
|
},
|
|
else => {},
|
|
}
|
|
}
|
|
|
|
// Pass 2: Generate all function bodies
|
|
// Functions with Any parameters (like any_to_string) are deferred to Pass 3
|
|
// so that all types are registered before their type-match expressions are compiled.
|
|
for (root.data.root.decls) |decl| {
|
|
switch (decl.data) {
|
|
.fn_decl => |fd| {
|
|
if (fd.body.data == .builtin_expr or fd.body.data == .foreign_expr) {
|
|
// skip — no body to generate
|
|
} else if (fd.type_params.len == 0) {
|
|
if (shouldDeferFnBody(fd)) {
|
|
try self.deferred_fn_bodies.append(self.allocator, .{ .fd = fd, .name = fd.name });
|
|
} else {
|
|
try self.genFnBody(fd, fd.name);
|
|
}
|
|
}
|
|
},
|
|
.const_decl => |cd| {
|
|
if (cd.value.data == .lambda) {
|
|
try self.genLambdaBody(cd.name, cd.value.data.lambda);
|
|
}
|
|
},
|
|
.namespace_decl => |ns| {
|
|
try self.genNamespaceBodies(ns);
|
|
},
|
|
else => {},
|
|
}
|
|
}
|
|
|
|
// Pass 3: Compile deferred function bodies (after all types are registered)
|
|
for (self.deferred_fn_bodies.items) |deferred| {
|
|
const saved_ns = self.current_namespace;
|
|
self.current_namespace = deferred.namespace;
|
|
defer self.current_namespace = saved_ns;
|
|
try self.genFnBody(deferred.fd, deferred.name);
|
|
}
|
|
|
|
// Execute comptime side effects via bytecode VM (e.g., #run main();)
|
|
for (self.comptime_side_effects.items) |expr| {
|
|
_ = try self.comptimeEval(expr, .void_type);
|
|
}
|
|
}
|
|
|
|
/// Evaluate a comptime expression using the bytecode VM.
|
|
/// No LLVM state save/restore needed — the VM operates independently.
|
|
fn comptimeEval(self: *CodeGen, expr: *Node, expected_type: Type) !comptime_mod.Value {
|
|
_ = expected_type; // VM infers types from values; expected_type used by caller for LLVM conversion
|
|
|
|
var compiler = comptime_mod.Compiler.init(self.allocator, if (self.sema_result) |sr| sr else null, self.root_decls, self);
|
|
const chunk = compiler.compile(expr) catch |err| {
|
|
return self.emitErrorFmt("comptime compilation failed: {s}", .{@errorName(err)});
|
|
};
|
|
|
|
var vm = comptime_mod.VM.init(self.allocator, if (self.sema_result) |sr| sr else null, self.root_decls, self);
|
|
return vm.execute(&chunk) catch |err| {
|
|
return self.emitErrorFmt("comptime execution failed: {s}", .{@errorName(err)});
|
|
};
|
|
}
|
|
|
|
/// Substitute comptime param identifiers in an AST expression with their literal nodes.
|
|
/// Used before comptimeEval in #insert to resolve comptime function params.
|
|
fn substituteComptimeNodes(self: *CodeGen, node: *Node) !*Node {
|
|
const cpn = self.comptime_param_nodes orelse return node;
|
|
|
|
// Direct identifier match
|
|
if (node.data == .identifier) {
|
|
if (cpn.get(node.data.identifier.name)) |replacement| {
|
|
return replacement;
|
|
}
|
|
}
|
|
|
|
// Recurse into call arguments
|
|
if (node.data == .call) {
|
|
var new_args = try self.allocator.alloc(*Node, node.data.call.args.len);
|
|
var changed = false;
|
|
for (node.data.call.args, 0..) |arg, i| {
|
|
new_args[i] = try self.substituteComptimeNodes(arg);
|
|
if (new_args[i] != arg) changed = true;
|
|
}
|
|
if (changed) {
|
|
const new_node = try self.allocator.create(Node);
|
|
new_node.* = .{
|
|
.span = node.span,
|
|
.data = .{ .call = .{
|
|
.callee = node.data.call.callee,
|
|
.args = new_args,
|
|
} },
|
|
};
|
|
return new_node;
|
|
}
|
|
}
|
|
|
|
// Recurse into binary ops
|
|
if (node.data == .binary_op) {
|
|
const new_lhs = try self.substituteComptimeNodes(node.data.binary_op.lhs);
|
|
const new_rhs = try self.substituteComptimeNodes(node.data.binary_op.rhs);
|
|
if (new_lhs != node.data.binary_op.lhs or new_rhs != node.data.binary_op.rhs) {
|
|
const new_node = try self.allocator.create(Node);
|
|
new_node.* = .{
|
|
.span = node.span,
|
|
.data = .{ .binary_op = .{
|
|
.op = node.data.binary_op.op,
|
|
.lhs = new_lhs,
|
|
.rhs = new_rhs,
|
|
} },
|
|
};
|
|
return new_node;
|
|
}
|
|
}
|
|
|
|
return node;
|
|
}
|
|
|
|
/// Convert a comptime VM Value to an LLVM constant value.
|
|
fn comptimeValueToLLVM(self: *CodeGen, value: comptime_mod.Value, ty: Type) c.LLVMValueRef {
|
|
return switch (value) {
|
|
.int_val => |v| c.LLVMConstInt(self.typeToLLVM(ty), @bitCast(v), 0),
|
|
.float_val => |v| c.LLVMConstReal(self.f64Type(), v),
|
|
.float32_val => |v| c.LLVMConstReal(self.f32Type(), @as(f64, v)),
|
|
.bool_val => |v| c.LLVMConstInt(self.i1Type(), if (v) 1 else 0, 0),
|
|
.string_val => |v| blk: {
|
|
const z = self.allocator.dupeZ(u8, v) catch unreachable;
|
|
const ptr = self.buildGlobalString(z.ptr, "comptime_str");
|
|
break :blk self.buildStringSlice(ptr, self.constInt64(@intCast(v.len)));
|
|
},
|
|
.void_val => self.constInt32(0),
|
|
.pointer_val => c.LLVMConstNull(self.ptrType()),
|
|
.null_val => c.LLVMConstNull(self.ptrType()),
|
|
.struct_val, .array_val, .type_val, .function_val => unreachable,
|
|
};
|
|
}
|
|
|
|
/// Lazily resolve a comptime global by evaluating its expression via bytecode VM.
|
|
fn resolveComptimeGlobal(self: *CodeGen, ct: *ComptimeGlobal) !void {
|
|
const result = try self.comptimeEval(ct.expr, ct.ty);
|
|
const const_val = self.comptimeValueToLLVM(result, ct.ty);
|
|
c.LLVMSetInitializer(ct.global, const_val);
|
|
c.LLVMSetGlobalConstant(ct.global, 1);
|
|
ct.is_resolved = true;
|
|
}
|
|
|
|
fn resolveType(self: *CodeGen, type_node: ?*Node) Type {
|
|
if (type_node) |tn| {
|
|
if (Type.fromTypeExpr(tn)) |t| return t;
|
|
// Array type: [N]T
|
|
if (tn.data == .array_type_expr) {
|
|
const ate = tn.data.array_type_expr;
|
|
const length: u32 = @intCast(ate.length.data.int_literal.value);
|
|
const elem_type = self.resolveType(ate.element_type);
|
|
const elem_name = elem_type.displayName(self.allocator) catch unreachable;
|
|
return .{ .array_type = .{ .element_name = elem_name, .length = length } };
|
|
}
|
|
// Slice type: []T
|
|
if (tn.data == .slice_type_expr) {
|
|
const ste = tn.data.slice_type_expr;
|
|
const elem_type = self.resolveType(ste.element_type);
|
|
const elem_name = elem_type.displayName(self.allocator) catch unreachable;
|
|
return .{ .slice_type = .{ .element_name = elem_name } };
|
|
}
|
|
// Pointer type: *T
|
|
if (tn.data == .pointer_type_expr) {
|
|
const pte = tn.data.pointer_type_expr;
|
|
const pointee_type = self.resolveType(pte.pointee_type);
|
|
const pointee_name = pointee_type.displayName(self.allocator) catch unreachable;
|
|
return .{ .pointer_type = .{ .pointee_name = pointee_name } };
|
|
}
|
|
// Many-pointer type: [*]T
|
|
if (tn.data == .many_pointer_type_expr) {
|
|
const mpte = tn.data.many_pointer_type_expr;
|
|
const elem_type = self.resolveType(mpte.element_type);
|
|
const elem_name = elem_type.displayName(self.allocator) catch unreachable;
|
|
return .{ .many_pointer_type = .{ .element_name = elem_name } };
|
|
}
|
|
// Function pointer type: (ParamTypes) -> ReturnType
|
|
if (tn.data == .function_type_expr) {
|
|
const fte = tn.data.function_type_expr;
|
|
var param_types = std.ArrayList(Type).empty;
|
|
for (fte.param_types) |pt| {
|
|
param_types.append(self.allocator, self.resolveType(pt)) catch return .void_type;
|
|
}
|
|
const ret_ty = if (fte.return_type) |rt| self.resolveType(rt) else Type.void_type;
|
|
const ret_ptr = self.allocator.create(Type) catch return .void_type;
|
|
ret_ptr.* = ret_ty;
|
|
return .{ .function_type = .{
|
|
.param_types = param_types.toOwnedSlice(self.allocator) catch return .void_type,
|
|
.return_type = ret_ptr,
|
|
} };
|
|
}
|
|
// Parameterized type: Vector(N, T) or generic struct instantiation
|
|
if (tn.data == .parameterized_type_expr) {
|
|
const pte = tn.data.parameterized_type_expr;
|
|
// Direct lookup (unqualified names from flat imports)
|
|
if (self.builtin_functions.contains(pte.name)) {
|
|
if (self.resolveBuiltinType(pte.name, pte.args)) |ty| return ty;
|
|
}
|
|
if (self.generic_struct_templates.get(pte.name)) |tmpl| {
|
|
return self.instantiateGenericStruct(pte.name, tmpl, pte.args) catch .void_type;
|
|
}
|
|
// Progressive namespace resolution for dotted names (e.g. "std.Vector")
|
|
if (std.mem.indexOfScalar(u8, pte.name, '.')) |dot| {
|
|
const ns = pte.name[0..dot];
|
|
if (self.namespaces.contains(ns)) {
|
|
// Namespace verified — look up qualified name in registries
|
|
if (self.builtin_functions.contains(pte.name)) {
|
|
if (self.resolveBuiltinType(pte.name, pte.args)) |ty| return ty;
|
|
}
|
|
if (self.generic_struct_templates.get(pte.name)) |tmpl| {
|
|
return self.instantiateGenericStruct(pte.name, tmpl, pte.args) catch .void_type;
|
|
}
|
|
}
|
|
}
|
|
if (self.diagnostics) |diags| diags.addFmt(.err, tn.span, "unresolved type '{s}'", .{pte.name});
|
|
return .void_type;
|
|
}
|
|
// Call expression as type: Vec(3, f32) → generic struct/type function instantiation
|
|
if (tn.data == .call) {
|
|
const name = self.calleeToQualifiedName(tn.data.call.callee);
|
|
if (name) |n| {
|
|
if (self.builtin_functions.contains(n)) {
|
|
if (self.resolveBuiltinType(n, tn.data.call.args)) |ty| return ty;
|
|
}
|
|
if (self.generic_struct_templates.get(n)) |tmpl| {
|
|
return self.instantiateGenericStruct(n, tmpl, tn.data.call.args) catch .void_type;
|
|
}
|
|
if (self.generic_templates.get(n)) |tmpl| {
|
|
return self.instantiateTypeFunction(n, n, tmpl, tn.data.call.args) catch .void_type;
|
|
}
|
|
}
|
|
return .void_type;
|
|
}
|
|
// Check type parameter bindings (during generic instantiation)
|
|
if (tn.data == .type_expr or tn.data == .identifier) {
|
|
const name = if (tn.data == .type_expr) tn.data.type_expr.name else tn.data.identifier.name;
|
|
// Try primitive type name first
|
|
if (Type.fromName(name)) |t| return t;
|
|
if (self.type_param_bindings) |bindings| {
|
|
if (bindings.get(name)) |t| return t;
|
|
}
|
|
// Unified type registry lookup
|
|
if (self.type_registry.get(name)) |entry| {
|
|
switch (entry) {
|
|
.struct_info => return .{ .struct_type = name },
|
|
.tagged_enum => return .{ .union_type = name },
|
|
.union_info => return .{ .union_type = name },
|
|
.plain_enum => return .{ .enum_type = name },
|
|
.alias => |target| {
|
|
if (Type.fromName(target)) |t| return t;
|
|
if (self.type_registry.get(target)) |inner| {
|
|
switch (inner) {
|
|
.struct_info => return .{ .struct_type = target },
|
|
.tagged_enum => return .{ .union_type = target },
|
|
.union_info => return .{ .union_type = target },
|
|
.plain_enum => return .{ .enum_type = target },
|
|
.alias => {},
|
|
}
|
|
}
|
|
},
|
|
}
|
|
}
|
|
}
|
|
// Safety net: inline declarations that should have been hoisted
|
|
if (tn.data == .struct_decl) {
|
|
const sn = tn.data.struct_decl.name;
|
|
if (self.type_registry.get(sn)) |e| {
|
|
if (e == .struct_info) return .{ .struct_type = sn };
|
|
}
|
|
}
|
|
if (tn.data == .enum_decl) {
|
|
const en = tn.data.enum_decl.name;
|
|
if (self.type_registry.get(en)) |e| switch (e) {
|
|
.tagged_enum => return .{ .union_type = en },
|
|
.plain_enum => return .{ .enum_type = en },
|
|
else => {},
|
|
};
|
|
}
|
|
return .void_type;
|
|
}
|
|
return .void_type;
|
|
}
|
|
|
|
/// Resolve a value argument to an integer — handles int_literal and identifier referencing value_param_bindings.
|
|
fn resolveValueArg(self: *CodeGen, node: *Node) i64 {
|
|
if (node.data == .int_literal) return node.data.int_literal.value;
|
|
if (node.data == .identifier or node.data == .type_expr) {
|
|
const name = if (node.data == .identifier) node.data.identifier.name else node.data.type_expr.name;
|
|
if (self.value_param_bindings) |bindings| {
|
|
if (bindings.get(name)) |val| return val;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/// Instantiate a generic struct template with concrete arguments.
|
|
/// Returns the struct_type for the instantiated struct (possibly cached).
|
|
fn instantiateGenericStruct(self: *CodeGen, template_name: []const u8, tmpl: ast.StructDecl, args: []const *Node) !Type {
|
|
const sd = tmpl;
|
|
|
|
// Build bindings from template params + args
|
|
var type_bindings = std.StringHashMap(Type).init(self.allocator);
|
|
var val_bindings = std.StringHashMap(i64).init(self.allocator);
|
|
|
|
for (sd.type_params, 0..) |tp, i| {
|
|
if (i >= args.len) return self.emitErrorFmt("generic struct '{s}' expects {d} type arguments, got {d}", .{ template_name, sd.type_params.len, args.len });
|
|
const constraint_name = if (tp.constraint.data == .type_expr) tp.constraint.data.type_expr.name else "";
|
|
if (std.mem.eql(u8, constraint_name, "Type")) {
|
|
// Type parameter: resolve arg as type
|
|
const resolved = self.resolveType(args[i]);
|
|
try type_bindings.put(tp.name, resolved);
|
|
} else {
|
|
// Value parameter: resolve arg as integer
|
|
const val = self.resolveValueArg(args[i]);
|
|
try val_bindings.put(tp.name, val);
|
|
}
|
|
}
|
|
|
|
const mangled_name = try self.mangleGenericName(template_name, sd.type_params, type_bindings, val_bindings, null);
|
|
|
|
// Check if already instantiated
|
|
if (self.type_registry.contains(mangled_name)) {
|
|
return .{ .struct_type = mangled_name };
|
|
}
|
|
|
|
// Instantiate: resolve field types with bindings active
|
|
const saved_type_bindings = self.type_param_bindings;
|
|
const saved_value_bindings = self.value_param_bindings;
|
|
self.type_param_bindings = type_bindings;
|
|
self.value_param_bindings = val_bindings;
|
|
defer {
|
|
self.type_param_bindings = saved_type_bindings;
|
|
self.value_param_bindings = saved_value_bindings;
|
|
}
|
|
|
|
const build = try self.buildStructFields(mangled_name, sd.field_types);
|
|
const resolved_defaults = try self.allocator.dupe(?*Node, sd.field_defaults);
|
|
|
|
// Build pretty display name: Vec(3,f32)
|
|
var display_buf = std.ArrayList(u8).empty;
|
|
try display_buf.appendSlice(self.allocator, template_name);
|
|
try display_buf.append(self.allocator, '(');
|
|
for (sd.type_params, 0..) |tp, i| {
|
|
if (i > 0) try display_buf.appendSlice(self.allocator, ",");
|
|
const constraint_name = if (tp.constraint.data == .type_expr) tp.constraint.data.type_expr.name else "";
|
|
if (std.mem.eql(u8, constraint_name, "Type")) {
|
|
if (type_bindings.get(tp.name)) |ty| {
|
|
const dn = ty.displayName(self.allocator) catch "?";
|
|
try display_buf.appendSlice(self.allocator, dn);
|
|
}
|
|
} else {
|
|
if (val_bindings.get(tp.name)) |val| {
|
|
var tmp: [20]u8 = undefined;
|
|
const s = std.fmt.bufPrint(&tmp, "{d}", .{val}) catch "0";
|
|
try display_buf.appendSlice(self.allocator, s);
|
|
}
|
|
}
|
|
}
|
|
try display_buf.append(self.allocator, ')');
|
|
const display_name = try display_buf.toOwnedSlice(self.allocator);
|
|
|
|
// Collect type param names and resolved types for later extraction
|
|
var tp_names = std.ArrayList([]const u8).empty;
|
|
var tp_types = std.ArrayList(Type).empty;
|
|
for (sd.type_params) |tp| {
|
|
const constraint_name = if (tp.constraint.data == .type_expr) tp.constraint.data.type_expr.name else "";
|
|
if (std.mem.eql(u8, constraint_name, "Type")) {
|
|
if (type_bindings.get(tp.name)) |ty| {
|
|
try tp_names.append(self.allocator, tp.name);
|
|
try tp_types.append(self.allocator, ty);
|
|
}
|
|
}
|
|
}
|
|
|
|
const si = StructInfo{
|
|
.field_names = sd.field_names,
|
|
.field_types = build.field_sx_types,
|
|
.field_defaults = resolved_defaults,
|
|
.llvm_type = build.llvm_type,
|
|
.display_name = display_name,
|
|
.type_param_names = try tp_names.toOwnedSlice(self.allocator),
|
|
.type_param_types = try tp_types.toOwnedSlice(self.allocator),
|
|
.template_name = template_name,
|
|
};
|
|
try self.type_registry.put(mangled_name, .{ .struct_info = si });
|
|
_ = try self.getAnyTypeId(mangled_name, .{ .struct_type = mangled_name });
|
|
|
|
return .{ .struct_type = mangled_name };
|
|
}
|
|
|
|
/// Instantiate a type-returning function (e.g. Complex(u32)) by walking the body AST
|
|
/// to find `return struct { ... }` or `return union { ... }` and registering with bindings active.
|
|
fn instantiateTypeFunction(self: *CodeGen, alias_name: []const u8, template_name: []const u8, tmpl: ast.FnDecl, args: []const *Node) !Type {
|
|
const fd = tmpl;
|
|
|
|
// Build type bindings from params + args
|
|
var type_bindings = std.StringHashMap(Type).init(self.allocator);
|
|
for (fd.type_params, 0..) |tp, i| {
|
|
if (i >= args.len) return self.emitErrorFmt("type function '{s}' expects {d} type arguments, got {d}", .{ template_name, fd.type_params.len, args.len });
|
|
const resolved = self.resolveType(args[i]);
|
|
try type_bindings.put(tp.name, resolved);
|
|
}
|
|
|
|
// Activate bindings
|
|
const saved_type_bindings = self.type_param_bindings;
|
|
self.type_param_bindings = type_bindings;
|
|
defer self.type_param_bindings = saved_type_bindings;
|
|
|
|
const mangled_name = try self.mangleGenericName(template_name, fd.type_params, type_bindings, null, null);
|
|
|
|
// Try struct first
|
|
if (self.findStructInBody(fd.body)) |struct_decl| {
|
|
if (self.type_registry.contains(mangled_name)) {
|
|
return .{ .struct_type = mangled_name };
|
|
}
|
|
return self.registerInstantiatedStruct(mangled_name, alias_name, struct_decl);
|
|
}
|
|
|
|
// Try union
|
|
if (self.findUnionInBody(fd.body)) |union_decl| {
|
|
if (self.type_registry.contains(mangled_name)) {
|
|
return .{ .union_type = mangled_name };
|
|
}
|
|
return self.registerInstantiatedTaggedEnum(mangled_name, union_decl);
|
|
}
|
|
|
|
return self.emitErrorFmt("type function '{s}' does not return a struct or enum", .{template_name});
|
|
}
|
|
|
|
fn registerInstantiatedStruct(self: *CodeGen, mangled_name: []const u8, alias_name: []const u8, struct_decl: ast.StructDecl) !Type {
|
|
const build = try self.buildStructFields(mangled_name, struct_decl.field_types);
|
|
|
|
const resolved_defaults = try self.allocator.dupe(?*Node, struct_decl.field_defaults);
|
|
const display_name = try self.allocator.dupe(u8, alias_name);
|
|
|
|
const si2 = StructInfo{
|
|
.field_names = struct_decl.field_names,
|
|
.field_types = build.field_sx_types,
|
|
.field_defaults = resolved_defaults,
|
|
.llvm_type = build.llvm_type,
|
|
.display_name = display_name,
|
|
};
|
|
try self.type_registry.put(mangled_name, .{ .struct_info = si2 });
|
|
_ = try self.getAnyTypeId(mangled_name, .{ .struct_type = mangled_name });
|
|
|
|
return .{ .struct_type = mangled_name };
|
|
}
|
|
|
|
fn registerInstantiatedTaggedEnum(self: *CodeGen, mangled_name: []const u8, union_decl: ast.EnumDecl) !Type {
|
|
const build = try self.buildUnionFields(mangled_name, union_decl.variant_types);
|
|
|
|
const tei = TaggedEnumInfo{
|
|
.variant_names = union_decl.variant_names,
|
|
.variant_types = build.variant_sx_types,
|
|
.llvm_type = build.llvm_type,
|
|
.max_payload_size = build.max_payload_size,
|
|
.payload_field_index = build.payload_field_index,
|
|
};
|
|
try self.type_registry.put(mangled_name, .{ .tagged_enum = tei });
|
|
_ = try self.getAnyTypeId(mangled_name, .{ .union_type = mangled_name });
|
|
|
|
return .{ .union_type = mangled_name };
|
|
}
|
|
|
|
/// Walk an AST body to find a struct declaration (from `return struct { ... }` or bare struct expr).
|
|
fn findDeclInBody(comptime T: type, comptime tag: std.meta.FieldEnum(@TypeOf(@as(Node, undefined).data)), body: *Node) ?T {
|
|
const extract = struct {
|
|
fn get(node: *Node) ?T {
|
|
return if (@field(node.data, @tagName(tag)) != @as(?T, null))
|
|
@field(node.data, @tagName(tag))
|
|
else
|
|
null;
|
|
}
|
|
};
|
|
_ = extract;
|
|
if (body.data == tag) return @field(body.data, @tagName(tag));
|
|
if (body.data == .block) {
|
|
for (body.data.block.stmts) |stmt| {
|
|
if (stmt.data == .return_stmt) {
|
|
if (stmt.data.return_stmt.value) |val| {
|
|
if (val.data == tag) return @field(val.data, @tagName(tag));
|
|
}
|
|
}
|
|
if (stmt.data == tag) return @field(stmt.data, @tagName(tag));
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
fn findStructInBody(_: *CodeGen, body: *Node) ?ast.StructDecl {
|
|
return findDeclInBody(ast.StructDecl, .struct_decl, body);
|
|
}
|
|
|
|
fn findUnionInBody(_: *CodeGen, body: *Node) ?ast.EnumDecl {
|
|
// Tagged enums with payloads are now stored as .enum_decl with variant_types populated
|
|
const isTaggedEnum = struct {
|
|
fn check(node: *Node) ?ast.EnumDecl {
|
|
if (node.data == .enum_decl and node.data.enum_decl.variant_types.len > 0) {
|
|
return node.data.enum_decl;
|
|
}
|
|
return null;
|
|
}
|
|
};
|
|
if (isTaggedEnum.check(body)) |ed| return ed;
|
|
const stmts = if (body.data == .block) body.data.block.stmts else return null;
|
|
for (stmts) |stmt| {
|
|
if (stmt.data == .return_stmt) {
|
|
if (stmt.data.return_stmt.value) |val| {
|
|
if (isTaggedEnum.check(val)) |ed| return ed;
|
|
}
|
|
}
|
|
if (isTaggedEnum.check(stmt)) |ed| return ed;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
fn buildFnType(self: *CodeGen, params: []const ast.Param, return_type: ?*Node, name: []const u8, is_foreign: bool) !c.LLVMTypeRef {
|
|
const ret_sx_type = self.resolveType(return_type);
|
|
const is_main = std.mem.eql(u8, name, "main");
|
|
const ret_llvm_type = if (is_main)
|
|
self.i32Type()
|
|
else if (is_foreign and ret_sx_type.isStruct())
|
|
self.getForeignReturnABIType(ret_sx_type)
|
|
else
|
|
self.typeToLLVM(ret_sx_type);
|
|
|
|
var param_llvm_types = std.ArrayList(c.LLVMTypeRef).empty;
|
|
for (params) |param| {
|
|
if (param.is_comptime) continue;
|
|
if (param.is_variadic) {
|
|
// Variadic param becomes a slice {ptr, i32} in the LLVM signature
|
|
try param_llvm_types.append(self.allocator, self.getStringStructType());
|
|
} else {
|
|
const sx_ty = self.resolveType(param.type_expr);
|
|
if (sx_ty == .void_type) return self.emitErrorFmt("parameter '{s}' in function '{s}' has unresolved type", .{ param.name, name });
|
|
// Foreign functions: apply C ABI lowering
|
|
if (is_foreign and sx_ty == .string_type) {
|
|
try param_llvm_types.append(self.allocator, self.ptrType());
|
|
} else if (is_foreign and sx_ty.isStruct()) {
|
|
try param_llvm_types.append(self.allocator, self.getForeignParamABIType(sx_ty));
|
|
} else if (is_foreign and sx_ty.isArray()) {
|
|
// [N]T → pointer in C ABI (C arrays decay to pointers)
|
|
try param_llvm_types.append(self.allocator, self.ptrType());
|
|
} else {
|
|
try param_llvm_types.append(self.allocator, self.typeToLLVM(sx_ty));
|
|
}
|
|
}
|
|
}
|
|
const params_slice = try param_llvm_types.toOwnedSlice(self.allocator);
|
|
|
|
return c.LLVMFunctionType(
|
|
ret_llvm_type,
|
|
if (params_slice.len > 0) params_slice.ptr else null,
|
|
@intCast(params_slice.len),
|
|
0,
|
|
);
|
|
}
|
|
|
|
/// For foreign (C ABI) functions, struct parameters must be lowered to their
|
|
/// ABI-equivalent types. LLVM does NOT do this automatically on all targets.
|
|
/// Dispatches to architecture-specific lowering based on target config.
|
|
fn getForeignParamABIType(self: *CodeGen, sx_ty: Type) c.LLVMTypeRef {
|
|
if (!sx_ty.isStruct()) return self.typeToLLVM(sx_ty);
|
|
|
|
const sname = self.resolveAlias(sx_ty.struct_type);
|
|
const info = self.lookupStructInfo(sname) orelse return self.typeToLLVM(sx_ty);
|
|
|
|
if (self.target_config.isAarch64()) {
|
|
return self.aarch64ParamABI(info);
|
|
} else if (self.target_config.isX86_64()) {
|
|
if (self.target_config.isWindows()) {
|
|
return self.win64ParamABI(info);
|
|
}
|
|
return self.x86_64SysVParamABI(info);
|
|
}
|
|
// Unknown architecture: pass struct type as-is (let LLVM backend handle it)
|
|
return info.llvm_type;
|
|
}
|
|
|
|
/// AArch64 ABI: struct parameter lowering.
|
|
/// - HFA (1-4 same float/double fields): [N x float/double]
|
|
/// - Non-HFA ≤ 8 bytes: i64
|
|
/// - Non-HFA 9-16 bytes: [2 x i64]
|
|
/// - > 16 bytes: pass as-is (indirect, not yet fully handled)
|
|
fn aarch64ParamABI(self: *CodeGen, info: StructInfo) c.LLVMTypeRef {
|
|
// Check HFA: 1-4 fields all of the same float type
|
|
const field_types = info.field_types;
|
|
if (field_types.len >= 1 and field_types.len <= 4) {
|
|
const first = field_types[0];
|
|
if (first == .f32 or first == .f64) {
|
|
var all_same = true;
|
|
for (field_types[1..]) |ft| {
|
|
if (!ft.eql(first)) {
|
|
all_same = false;
|
|
break;
|
|
}
|
|
}
|
|
if (all_same) {
|
|
const elem_ty = if (first == .f32)
|
|
self.f32Type()
|
|
else
|
|
self.f64Type();
|
|
return c.LLVMArrayType2(elem_ty, @intCast(field_types.len));
|
|
}
|
|
}
|
|
}
|
|
|
|
// Non-HFA: pack into integer registers
|
|
const size = self.getTypeSize(info.llvm_type);
|
|
if (size <= 8) return self.i64Type();
|
|
if (size <= 16) return c.LLVMArrayType2(self.i64Type(), 2);
|
|
return info.llvm_type;
|
|
}
|
|
|
|
/// x86-64 SysV ABI: struct parameter lowering.
|
|
/// Each 8-byte "eightbyte" is classified as INTEGER or SSE:
|
|
/// - If all fields in the eightbyte are float/double: SSE (passed in XMM register)
|
|
/// - If any field is integer/pointer: INTEGER (passed in GPR)
|
|
/// - Structs > 16 bytes: passed in memory (by pointer)
|
|
fn x86_64SysVParamABI(self: *CodeGen, info: StructInfo) c.LLVMTypeRef {
|
|
const data_layout = c.LLVMGetModuleDataLayout(self.module);
|
|
const size = self.getTypeSize(info.llvm_type);
|
|
|
|
// > 16 bytes: MEMORY class (passed by pointer, handled by LLVM backend)
|
|
if (size > 16) return info.llvm_type;
|
|
|
|
// Single eightbyte (≤ 8 bytes)
|
|
if (size <= 8) {
|
|
return self.classifyEightbyte(info.field_types, size);
|
|
}
|
|
|
|
// Two eightbytes (9-16 bytes): classify each half independently
|
|
// Split fields into first eightbyte (offset < 8) and second eightbyte (offset >= 8)
|
|
var first_eb_types = std.ArrayList(Type).empty;
|
|
var second_eb_types = std.ArrayList(Type).empty;
|
|
var second_eb_size: u64 = 0;
|
|
|
|
const struct_ty = info.llvm_type;
|
|
for (info.field_types, 0..) |ft, idx| {
|
|
const offset = c.LLVMOffsetOfElement(data_layout, struct_ty, @intCast(idx));
|
|
if (offset < 8) {
|
|
first_eb_types.append(self.allocator, ft) catch return info.llvm_type;
|
|
} else {
|
|
second_eb_types.append(self.allocator, ft) catch return info.llvm_type;
|
|
const field_llvm = self.typeToLLVM(ft);
|
|
second_eb_size += self.getTypeSize(field_llvm);
|
|
}
|
|
}
|
|
|
|
const eb1 = self.classifyEightbyte(first_eb_types.items, 8);
|
|
const eb2 = self.classifyEightbyte(second_eb_types.items, if (second_eb_size > 0) second_eb_size else size - 8);
|
|
|
|
// Compose the two eightbytes into a struct type
|
|
var members: [2]c.LLVMTypeRef = .{ eb1, eb2 };
|
|
return c.LLVMStructTypeInContext(self.context, &members, 2, 0);
|
|
}
|
|
|
|
/// Classify a single x86-64 eightbyte: if all fields are float, return SSE type;
|
|
/// otherwise return an integer type matching the byte size.
|
|
fn classifyEightbyte(self: *CodeGen, field_types_in_eb: []const Type, byte_size: u64) c.LLVMTypeRef {
|
|
if (field_types_in_eb.len == 0) {
|
|
// No fields in this chunk — use integer padding
|
|
return c.LLVMIntTypeInContext(self.context, @intCast(byte_size * 8));
|
|
}
|
|
|
|
// Check if all fields are SSE (float/double)
|
|
var all_sse = true;
|
|
var float_count: u32 = 0;
|
|
var double_count: u32 = 0;
|
|
for (field_types_in_eb) |ft| {
|
|
if (ft == .f32) {
|
|
float_count += 1;
|
|
} else if (ft == .f64) {
|
|
double_count += 1;
|
|
} else {
|
|
all_sse = false;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (all_sse) {
|
|
// SSE class: return appropriate float type
|
|
if (double_count > 0 and float_count == 0) {
|
|
if (double_count == 1) return self.f64Type();
|
|
// Multiple doubles shouldn't fit in one eightbyte (double = 8 bytes)
|
|
return self.f64Type();
|
|
}
|
|
if (float_count > 0 and double_count == 0) {
|
|
if (float_count == 1) return self.f32Type();
|
|
// 2 floats = 8 bytes, fits in one eightbyte
|
|
return c.LLVMArrayType2(self.f32Type(), @intCast(float_count));
|
|
}
|
|
// Mixed float/double in one eightbyte shouldn't happen (float=4, double=8)
|
|
// but fall through to integer just in case
|
|
}
|
|
|
|
// INTEGER class: coerce to integer matching the byte size
|
|
return c.LLVMIntTypeInContext(self.context, @intCast(byte_size * 8));
|
|
}
|
|
|
|
/// Windows x64 ABI: struct parameter lowering.
|
|
/// Only structs of exactly 1, 2, 4, or 8 bytes are passed in a register.
|
|
/// Everything else is passed by pointer (handled by LLVM backend).
|
|
fn win64ParamABI(self: *CodeGen, info: StructInfo) c.LLVMTypeRef {
|
|
const size = self.getTypeSize(info.llvm_type);
|
|
|
|
// Windows x64: only power-of-2 sizes ≤ 8 passed in register
|
|
if (size == 1 or size == 2 or size == 4 or size == 8) {
|
|
return c.LLVMIntTypeInContext(self.context, @intCast(size * 8));
|
|
}
|
|
// All other sizes: passed by pointer (LLVM handles byval)
|
|
return info.llvm_type;
|
|
}
|
|
|
|
/// For foreign functions returning structs, apply the same ABI lowering as parameters.
|
|
/// The rules for return values match parameter rules on both AArch64 and x86-64 SysV
|
|
/// for small structs (≤ 16 bytes). Larger structs use sret (handled by LLVM).
|
|
fn getForeignReturnABIType(self: *CodeGen, sx_ty: Type) c.LLVMTypeRef {
|
|
// Reuse the same classification as parameters — the rules are identical
|
|
// for small struct returns on both AArch64 and x86-64 SysV.
|
|
return self.getForeignParamABIType(sx_ty);
|
|
}
|
|
|
|
/// Convert a struct value to its C ABI representation for a foreign call.
|
|
/// Stores the struct to memory, then loads as the ABI type.
|
|
fn convertStructToABI(self: *CodeGen, struct_val: c.LLVMValueRef, struct_ty: c.LLVMTypeRef, abi_ty: c.LLVMTypeRef) c.LLVMValueRef {
|
|
const struct_size = self.getTypeSize(struct_ty);
|
|
const abi_size = self.getTypeSize(abi_ty);
|
|
|
|
if (struct_size == abi_size) {
|
|
// Same size (e.g. {float, float} → [2 x float]): store and reload
|
|
const alloca = self.buildEntryBlockAlloca(struct_ty, "abi_tmp");
|
|
_ = c.LLVMBuildStore(self.builder, struct_val, alloca);
|
|
return c.LLVMBuildLoad2(self.builder, abi_ty, alloca, "abi_arg");
|
|
} else {
|
|
// Struct smaller than ABI type (e.g. {i8,i8,i8,i8} → i64): zero-init, then store struct
|
|
const alloca = self.buildEntryBlockAlloca(abi_ty, "abi_tmp");
|
|
self.storeNull(abi_ty, alloca);
|
|
_ = c.LLVMBuildStore(self.builder, struct_val, alloca);
|
|
return c.LLVMBuildLoad2(self.builder, abi_ty, alloca, "abi_arg");
|
|
}
|
|
}
|
|
|
|
fn registerFnDecl(self: *CodeGen, fd: ast.FnDecl, llvm_name: []const u8) !void {
|
|
const is_foreign = fd.body.data == .foreign_expr;
|
|
const fn_type = try self.buildFnType(fd.params, fd.return_type, fd.name, is_foreign);
|
|
const name_z = try self.allocator.dupeZ(u8, llvm_name);
|
|
_ = c.LLVMAddFunction(self.module, name_z.ptr, fn_type);
|
|
// Track foreign functions for ABI lowering at call sites
|
|
if (is_foreign) try self.foreign_fns.put(llvm_name, {});
|
|
// Track resolved parameter types for accurate call-site conversion
|
|
var param_types = std.ArrayList(Type).empty;
|
|
for (fd.params) |param| {
|
|
if (param.is_comptime) continue;
|
|
if (param.is_variadic) {
|
|
const elem_name = if (param.type_expr.data == .type_expr) param.type_expr.data.type_expr.name else "s32";
|
|
try param_types.append(self.allocator, .{ .slice_type = .{ .element_name = elem_name } });
|
|
} else {
|
|
try param_types.append(self.allocator, self.resolveType(param.type_expr));
|
|
}
|
|
}
|
|
try self.fn_param_types.put(llvm_name, try param_types.toOwnedSlice(self.allocator));
|
|
// Track declared return type (preserves signedness lost by LLVM round-trip)
|
|
const ret_ty = if (fd.return_type) |rt| self.resolveType(rt) else Type.void_type;
|
|
try self.function_return_types.put(llvm_name, ret_ty);
|
|
// Track variadic function info for call site packing
|
|
for (fd.params, 0..) |param, i| {
|
|
if (param.is_variadic) {
|
|
const elem_name = if (param.type_expr.data == .type_expr) param.type_expr.data.type_expr.name else "s32";
|
|
try self.variadic_functions.put(llvm_name, .{
|
|
.fixed_param_count = @intCast(i),
|
|
.element_type_name = elem_name,
|
|
});
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
fn registerNamespace(self: *CodeGen, ns: ast.NamespaceDecl) !void {
|
|
try self.namespaces.put(ns.name, {});
|
|
for (ns.decls) |decl| {
|
|
switch (decl.data) {
|
|
.fn_decl => |fd| {
|
|
if (fd.body.data == .builtin_expr) {
|
|
const qualified = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns.name, fd.name });
|
|
try self.builtin_functions.put(qualified, {});
|
|
continue;
|
|
}
|
|
const qualified = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns.name, fd.name });
|
|
if (fd.body.data == .foreign_expr) {
|
|
// External C function in namespace — register LLVM declaration with C name only
|
|
try self.registerFnDecl(fd, fd.name);
|
|
// Also track qualified name as foreign for ABI lowering at call sites
|
|
try self.foreign_fns.put(qualified, {});
|
|
// Store param types under qualified name so call-site type resolution works
|
|
var param_types = std.ArrayList(Type).empty;
|
|
for (fd.params) |param| {
|
|
if (param.is_comptime) continue;
|
|
try param_types.append(self.allocator, self.resolveType(param.type_expr));
|
|
}
|
|
try self.fn_param_types.put(qualified, try param_types.toOwnedSlice(self.allocator));
|
|
} else if (fd.type_params.len > 0) {
|
|
try self.generic_templates.put(qualified, fd);
|
|
} else {
|
|
try self.registerFnDecl(fd, qualified);
|
|
}
|
|
},
|
|
.enum_decl => |ed| {
|
|
if (ed.variant_types.len > 0) {
|
|
// Tagged enum with payloads
|
|
try self.registerTaggedEnum(ed);
|
|
const qualified_u = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns.name, ed.name });
|
|
try self.type_registry.put(qualified_u, .{ .alias = ed.name });
|
|
} else {
|
|
const qualified = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns.name, ed.name });
|
|
try self.type_registry.put(qualified, .{ .plain_enum = ed.variant_names });
|
|
_ = try self.getAnyTypeId(qualified, .{ .enum_type = qualified });
|
|
if (ed.backing_type) |bt_node| {
|
|
const bt = self.resolveType(bt_node);
|
|
try self.enum_backing_types.put(qualified, self.typeToLLVM(bt));
|
|
}
|
|
}
|
|
},
|
|
.struct_decl => |sd| {
|
|
try self.registerStructType(sd);
|
|
// Register qualified alias so rl.Color resolves to Color
|
|
const qualified_s = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns.name, sd.name });
|
|
try self.type_registry.put(qualified_s, .{ .alias = sd.name });
|
|
},
|
|
.union_decl => |ud| {
|
|
try self.registerUnionType(ud);
|
|
const qualified_u = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns.name, ud.name });
|
|
try self.type_registry.put(qualified_u, .{ .alias = ud.name });
|
|
},
|
|
.const_decl => |cd| {
|
|
if (cd.value.data == .builtin_expr) {
|
|
// #builtin constant in namespace — skip codegen
|
|
} else if (cd.value.data == .lambda) {
|
|
const qualified = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns.name, cd.name });
|
|
try self.registerLambdaAsFunction(qualified, cd.value.data.lambda);
|
|
} else if (cd.value.data == .type_expr) {
|
|
const qualified = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns.name, cd.name });
|
|
try self.type_registry.put(qualified, .{ .alias = cd.value.data.type_expr.name });
|
|
}
|
|
},
|
|
.library_decl => |ld| {
|
|
try self.foreign_libraries.append(self.allocator, ld.lib_name);
|
|
},
|
|
.var_decl => |vd| {
|
|
try self.registerGlobalVar(vd);
|
|
},
|
|
else => {},
|
|
}
|
|
}
|
|
}
|
|
|
|
fn genNamespaceBodies(self: *CodeGen, ns: ast.NamespaceDecl) !void {
|
|
const saved_ns = self.current_namespace;
|
|
self.current_namespace = ns.name;
|
|
defer self.current_namespace = saved_ns;
|
|
|
|
for (ns.decls) |decl| {
|
|
switch (decl.data) {
|
|
.fn_decl => |fd| {
|
|
if (fd.body.data == .builtin_expr or fd.body.data == .foreign_expr) {
|
|
// skip — no body to generate
|
|
} else if (fd.type_params.len == 0) {
|
|
const qualified = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns.name, fd.name });
|
|
if (shouldDeferFnBody(fd)) {
|
|
try self.deferred_fn_bodies.append(self.allocator, .{ .fd = fd, .name = qualified, .namespace = ns.name });
|
|
} else {
|
|
try self.genFnBody(fd, qualified);
|
|
}
|
|
}
|
|
},
|
|
.const_decl => |cd| {
|
|
if (cd.value.data == .lambda) {
|
|
const qualified = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns.name, cd.name });
|
|
try self.genLambdaBody(qualified, cd.value.data.lambda);
|
|
}
|
|
},
|
|
else => {},
|
|
}
|
|
}
|
|
}
|
|
|
|
fn inferComptimeReturnType(self: *CodeGen, expr: *Node) Type {
|
|
// xx: see through to inner expression
|
|
if (expr.data == .unary_op and expr.data.unary_op.op == .xx) {
|
|
return self.inferComptimeReturnType(expr.data.unary_op.operand);
|
|
}
|
|
// For function calls, look up the registered function's return type
|
|
if (expr.data == .call) {
|
|
if (self.resolveCalleeName(expr.data.call)) |callee_name| {
|
|
var cnbuf: [256]u8 = undefined;
|
|
const callee_fn = c.LLVMGetNamedFunction(self.module, self.nameToCStr(callee_name, &cnbuf)) orelse return Type.s(64);
|
|
const fn_type = c.LLVMGlobalGetValueType(callee_fn);
|
|
const ret_llvm = c.LLVMGetReturnType(fn_type);
|
|
return self.llvmTypeToSxType(ret_llvm);
|
|
}
|
|
}
|
|
return self.inferType(expr);
|
|
}
|
|
|
|
/// Map an LLVM type back to a sx Type
|
|
fn llvmTypeToSxType(self: *CodeGen, llvm_ty: c.LLVMTypeRef) Type {
|
|
if (llvm_ty == self.i1Type()) return .boolean;
|
|
if (llvm_ty == self.i8Type()) return Type.s(8);
|
|
if (llvm_ty == c.LLVMInt16TypeInContext(self.context)) return Type.s(16);
|
|
if (llvm_ty == self.i32Type()) return Type.s(32);
|
|
if (llvm_ty == self.i64Type()) return Type.s(64);
|
|
if (llvm_ty == self.f32Type()) return .f32;
|
|
if (llvm_ty == self.f64Type()) return .f64;
|
|
if (llvm_ty == self.voidType()) return .void_type;
|
|
if (llvm_ty == self.getStringStructType()) return .string_type;
|
|
if (self.any_struct_type != null and llvm_ty == self.any_struct_type.?) return .any_type;
|
|
if (llvm_ty == self.ptrType()) return .string_type; // raw ptr fallback (meta_type)
|
|
// Handle arbitrary-width integer types (e.g. i3, i7, i12)
|
|
if (c.LLVMGetTypeKind(llvm_ty) == c.LLVMIntegerTypeKind) {
|
|
const width = c.LLVMGetIntTypeWidth(llvm_ty);
|
|
if (width > 0 and width <= 64) return Type.s(@intCast(width));
|
|
}
|
|
// Check for named struct types
|
|
if (c.LLVMGetTypeKind(llvm_ty) == c.LLVMStructTypeKind) {
|
|
const name_ptr = c.LLVMGetStructName(llvm_ty);
|
|
if (name_ptr != null) {
|
|
const name = std.mem.span(name_ptr);
|
|
if (self.type_registry.get(name)) |e| switch (e) {
|
|
.struct_info => return .{ .struct_type = name },
|
|
.tagged_enum => return .{ .union_type = name },
|
|
else => {},
|
|
};
|
|
}
|
|
}
|
|
// Check for array types
|
|
if (c.LLVMGetTypeKind(llvm_ty) == c.LLVMArrayTypeKind) {
|
|
const elem_llvm = c.LLVMGetElementType(llvm_ty);
|
|
const length: u32 = @intCast(c.LLVMGetArrayLength2(llvm_ty));
|
|
const elem_ty = self.llvmTypeToSxType(elem_llvm);
|
|
const elem_name = elem_ty.displayName(self.allocator) catch return Type.s(64);
|
|
return .{ .array_type = .{ .element_name = elem_name, .length = length } };
|
|
}
|
|
// Check for vector types
|
|
if (c.LLVMGetTypeKind(llvm_ty) == c.LLVMVectorTypeKind) {
|
|
const elem_llvm = c.LLVMGetElementType(llvm_ty);
|
|
const length = c.LLVMGetVectorSize(llvm_ty);
|
|
const elem_ty = self.llvmTypeToSxType(elem_llvm);
|
|
const elem_name = elem_ty.displayName(self.allocator) catch return Type.s(64);
|
|
return .{ .vector_type = .{ .element_name = elem_name, .length = length } };
|
|
}
|
|
return Type.s(64);
|
|
}
|
|
|
|
fn registerComptimeGlobal(self: *CodeGen, name: []const u8, expr: *Node, type_override: ?Type) !void {
|
|
const ty = type_override orelse self.inferComptimeReturnType(expr);
|
|
if (ty == .void_type) return self.emitErrorFmt("cannot infer type for comptime global '{s}'", .{name});
|
|
|
|
const llvm_ty = self.typeToLLVM(ty);
|
|
const name_z = try self.allocator.dupeZ(u8, name);
|
|
const global = c.LLVMAddGlobal(self.module, llvm_ty, name_z.ptr);
|
|
c.LLVMSetInitializer(global, c.LLVMConstInt(llvm_ty, 0, 0));
|
|
|
|
try self.comptime_globals.put(name, .{ .global = global, .ty = ty, .expr = expr });
|
|
}
|
|
|
|
/// Evaluate a simple constant expression to an LLVM constant value.
|
|
/// Returns null for expressions that can't be constant-folded at registration time.
|
|
fn evalConstant(self: *CodeGen, node: *Node, target_ty: Type) ?c.LLVMValueRef {
|
|
const llvm_ty = self.typeToLLVM(target_ty);
|
|
switch (node.data) {
|
|
.int_literal => |lit| {
|
|
return c.LLVMConstInt(llvm_ty, @bitCast(@as(i64, lit.value)), 0);
|
|
},
|
|
.float_literal => |lit| {
|
|
return c.LLVMConstReal(llvm_ty, lit.value);
|
|
},
|
|
.bool_literal => |lit| {
|
|
return c.LLVMConstInt(llvm_ty, if (lit.value) 1 else 0, 0);
|
|
},
|
|
else => return null,
|
|
}
|
|
}
|
|
|
|
/// Register a top-level value constant (e.g., `SPECIAL_VALUE :u8: 42;`) as an LLVM global.
|
|
fn registerTopLevelConstant(self: *CodeGen, cd: ast.ConstDecl) !void {
|
|
const ta = cd.type_annotation orelse return; // need explicit type for top-level constants
|
|
const sx_ty = self.resolveType(ta);
|
|
if (sx_ty == .void_type) return;
|
|
|
|
const const_val = self.evalConstant(cd.value, sx_ty) orelse return;
|
|
|
|
const name_z = try self.allocator.dupeZ(u8, cd.name);
|
|
const global = c.LLVMAddGlobal(self.module, self.typeToLLVM(sx_ty), name_z.ptr);
|
|
c.LLVMSetInitializer(global, const_val);
|
|
c.LLVMSetGlobalConstant(global, 1);
|
|
|
|
try self.comptime_globals.put(cd.name, .{
|
|
.global = global,
|
|
.ty = sx_ty,
|
|
.expr = cd.value,
|
|
.is_resolved = true,
|
|
});
|
|
}
|
|
|
|
fn registerGlobalVar(self: *CodeGen, vd: ast.VarDecl) !void {
|
|
const ta = vd.type_annotation orelse return;
|
|
const sx_ty = self.resolveType(ta);
|
|
if (sx_ty == .void_type) return;
|
|
|
|
const llvm_ty = self.typeToLLVM(sx_ty);
|
|
const name_z = try self.allocator.dupeZ(u8, vd.name);
|
|
const global = c.LLVMAddGlobal(self.module, llvm_ty, name_z.ptr);
|
|
// Initialize with undef (will be set at runtime, e.g. by load_gl)
|
|
c.LLVMSetInitializer(global, self.getUndef(llvm_ty));
|
|
// NOT constant — this is a mutable global
|
|
c.LLVMSetGlobalConstant(global, 0);
|
|
|
|
try self.global_mutable_vars.put(vd.name, .{ .ptr = global, .ty = sx_ty });
|
|
}
|
|
|
|
fn bindParam(self: *CodeGen, function: c.LLVMValueRef, name: []const u8, sx_ty: Type, param_idx: u32) !void {
|
|
const llvm_ty = self.typeToLLVM(sx_ty);
|
|
const param_name_z = try self.allocator.dupeZ(u8, name);
|
|
const alloca = c.LLVMBuildAlloca(self.builder, llvm_ty, param_name_z.ptr);
|
|
const param_val = c.LLVMGetParam(function, param_idx);
|
|
_ = c.LLVMBuildStore(self.builder, param_val, alloca);
|
|
try self.named_values.put(name, .{ .ptr = alloca, .ty = sx_ty });
|
|
}
|
|
|
|
/// Prepare a return value: load structs/unions from alloca pointers, convert types.
|
|
fn prepareReturnValue(self: *CodeGen, raw_val: c.LLVMValueRef, ret_type: Type) !c.LLVMValueRef {
|
|
if (ret_type.isStruct()) {
|
|
const sname = self.resolveAlias(ret_type.struct_type);
|
|
const info = try self.getStructInfo(sname);
|
|
return c.LLVMBuildLoad2(self.builder, info.llvm_type, raw_val, "retval");
|
|
} else if (ret_type.isUnion()) {
|
|
const uname = ret_type.union_type;
|
|
const resolved = self.resolveAlias(uname);
|
|
const info = try self.getTaggedEnumInfo(resolved);
|
|
return self.loadIfPointer(raw_val, info.llvm_type, "retval");
|
|
} else {
|
|
const src_ty = self.llvmTypeToSxType(c.LLVMTypeOf(raw_val));
|
|
return self.convertValue(raw_val, src_ty, ret_type);
|
|
}
|
|
}
|
|
|
|
fn genFnBody(self: *CodeGen, fd: ast.FnDecl, llvm_name: []const u8) !void {
|
|
self.named_values.clearRetainingCapacity();
|
|
|
|
const ret_sx_type = self.resolveType(fd.return_type);
|
|
const is_main = std.mem.eql(u8, llvm_name, "main");
|
|
const ret_llvm_type = if (is_main)
|
|
self.i32Type()
|
|
else
|
|
self.typeToLLVM(ret_sx_type);
|
|
|
|
self.current_return_type = if (is_main) Type.s(32) else ret_sx_type;
|
|
|
|
const name_z = try self.allocator.dupeZ(u8, llvm_name);
|
|
const function = c.LLVMGetNamedFunction(self.module, name_z.ptr) orelse return self.emitErrorFmt("function '{s}' not found in LLVM module", .{llvm_name});
|
|
self.current_function = function;
|
|
|
|
_ = self.appendBlock(function, "entry");
|
|
|
|
// Create allocas for parameters and store incoming values
|
|
for (fd.params, 0..) |param, i| {
|
|
const sx_ty = if (param.is_variadic) blk: {
|
|
const elem_name = if (param.type_expr.data == .type_expr) param.type_expr.data.type_expr.name else "s32";
|
|
break :blk Type{ .slice_type = .{ .element_name = elem_name } };
|
|
} else self.resolveType(param.type_expr);
|
|
if (sx_ty == .void_type) return self.emitErrorFmt("parameter '{s}' has unresolved type", .{param.name});
|
|
try self.bindParam(function, param.name, sx_ty, @intCast(i));
|
|
}
|
|
|
|
// Push function-level scope so that function-body defers are tracked
|
|
try self.pushScope();
|
|
|
|
// Generate body
|
|
const body = fd.body;
|
|
if (body.data != .block) return self.emitError("function body must be a block");
|
|
|
|
var last_val: c.LLVMValueRef = null;
|
|
for (body.data.block.stmts) |stmt| {
|
|
last_val = try self.genStmt(stmt);
|
|
}
|
|
|
|
// Return — skip if current block already has a terminator (from explicit return)
|
|
const current_bb = self.getCurrentBlock();
|
|
if (c.LLVMGetBasicBlockTerminator(current_bb) == null) {
|
|
// Implicit return path: pop scope (executes defers) then return
|
|
try self.popScope();
|
|
// Check if last_val is void-typed (e.g. call to void-returning function)
|
|
const effective_last_val: ?c.LLVMValueRef = if (last_val) |val|
|
|
(if (c.LLVMTypeOf(val) == self.voidType()) null else val)
|
|
else
|
|
null;
|
|
|
|
if (ret_sx_type == .void_type and !is_main) {
|
|
self.retVoid();
|
|
} else if (effective_last_val) |val| {
|
|
const ret_val = try self.prepareReturnValue(val, ret_sx_type);
|
|
self.ret(ret_val);
|
|
} else if (is_main) {
|
|
self.ret(c.LLVMConstInt(ret_llvm_type, 0, 0));
|
|
} else if (ret_sx_type != .void_type) {
|
|
_ = c.LLVMBuildUnreachable(self.builder);
|
|
} else {
|
|
self.retVoid();
|
|
}
|
|
} else {
|
|
// Explicit return already emitted defers; just clean up scope stacks
|
|
if (self.scope_stack.items.len > 0) _ = self.scope_stack.pop();
|
|
}
|
|
}
|
|
|
|
fn registerLambdaAsFunction(self: *CodeGen, name: []const u8, lambda: ast.Lambda) !void {
|
|
const ret_sx_type = self.inferType(lambda.body);
|
|
const ret_llvm_type = self.typeToLLVM(ret_sx_type);
|
|
|
|
var param_llvm_types = std.ArrayList(c.LLVMTypeRef).empty;
|
|
for (lambda.params) |param| {
|
|
const sx_ty = self.resolveType(param.type_expr);
|
|
try param_llvm_types.append(self.allocator, self.typeToLLVM(sx_ty));
|
|
}
|
|
const params_slice = try param_llvm_types.toOwnedSlice(self.allocator);
|
|
|
|
const fn_type = c.LLVMFunctionType(
|
|
ret_llvm_type,
|
|
if (params_slice.len > 0) params_slice.ptr else null,
|
|
@intCast(params_slice.len),
|
|
0,
|
|
);
|
|
|
|
const name_z = try self.allocator.dupeZ(u8, name);
|
|
_ = c.LLVMAddFunction(self.module, name_z.ptr, fn_type);
|
|
}
|
|
|
|
fn genLambdaBody(self: *CodeGen, name: []const u8, lambda: ast.Lambda) !void {
|
|
self.named_values.clearRetainingCapacity();
|
|
|
|
const ret_sx_type = self.inferType(lambda.body);
|
|
self.current_return_type = ret_sx_type;
|
|
|
|
const name_z = try self.allocator.dupeZ(u8, name);
|
|
const function = c.LLVMGetNamedFunction(self.module, name_z.ptr) orelse return self.emitErrorFmt("lambda '{s}' not found in LLVM module", .{name});
|
|
self.current_function = function;
|
|
|
|
_ = self.appendBlock(function, "entry");
|
|
|
|
for (lambda.params, 0..) |param, i| {
|
|
const sx_ty = self.resolveType(param.type_expr);
|
|
try self.bindParam(function, param.name, sx_ty, @intCast(i));
|
|
}
|
|
|
|
const ret_val = try self.genExpr(lambda.body);
|
|
if (ret_val) |val| {
|
|
const src_ty = self.llvmTypeToSxType(c.LLVMTypeOf(val));
|
|
const converted = self.convertValue(val, src_ty, ret_sx_type);
|
|
self.ret(converted);
|
|
} else {
|
|
self.retVoid();
|
|
}
|
|
}
|
|
|
|
fn genStmt(self: *CodeGen, node: *Node) !c.LLVMValueRef {
|
|
self.current_span = node.span;
|
|
switch (node.data) {
|
|
.var_decl => |vd| {
|
|
return self.genVarDecl(vd);
|
|
},
|
|
.const_decl => |cd| {
|
|
return self.genConstDecl(cd);
|
|
},
|
|
.fn_decl => |fd| {
|
|
// Local declaration inside a function body
|
|
if (fd.type_params.len > 0) {
|
|
// Generic template / type function: register for lazy instantiation
|
|
try self.generic_templates.put(fd.name, fd);
|
|
} else {
|
|
// Non-generic local function
|
|
// Save outer function state
|
|
const saved_fn = self.current_function;
|
|
const saved_bb = self.getCurrentBlock();
|
|
const saved_ret = self.current_return_type;
|
|
const saved_named = self.named_values;
|
|
self.named_values = std.StringHashMap(NamedValue).init(self.allocator);
|
|
|
|
// Infer return type from body for => lambdas without explicit annotation
|
|
const ret_sx_type = if (fd.return_type != null) self.resolveType(fd.return_type) else if (fd.is_arrow) self.inferType(fd.body) else Type.void_type;
|
|
|
|
// For arrow lambdas with inferred return type, build function manually
|
|
if (fd.is_arrow and fd.return_type == null) {
|
|
const ret_llvm_type = self.typeToLLVM(ret_sx_type);
|
|
var param_llvm_types = std.ArrayList(c.LLVMTypeRef).empty;
|
|
for (fd.params) |param| {
|
|
try param_llvm_types.append(self.allocator, self.typeToLLVM(self.resolveType(param.type_expr)));
|
|
}
|
|
const params_slice = try param_llvm_types.toOwnedSlice(self.allocator);
|
|
const fn_type = c.LLVMFunctionType(
|
|
ret_llvm_type,
|
|
if (params_slice.len > 0) params_slice.ptr else null,
|
|
@intCast(params_slice.len),
|
|
0,
|
|
);
|
|
const name_z2 = try self.allocator.dupeZ(u8, fd.name);
|
|
_ = c.LLVMAddFunction(self.module, name_z2.ptr, fn_type);
|
|
try self.function_return_types.put(fd.name, ret_sx_type);
|
|
} else {
|
|
try self.registerFnDecl(fd, fd.name);
|
|
}
|
|
self.current_return_type = ret_sx_type;
|
|
const name_z = try self.allocator.dupeZ(u8, fd.name);
|
|
const function = c.LLVMGetNamedFunction(self.module, name_z.ptr) orelse
|
|
return self.emitErrorFmt("local function '{s}' not found", .{fd.name});
|
|
self.current_function = function;
|
|
_ = self.appendBlock(function, "entry");
|
|
|
|
for (fd.params, 0..) |param, i| {
|
|
const sx_ty = self.resolveType(param.type_expr);
|
|
try self.bindParam(function, param.name, sx_ty, @intCast(i));
|
|
}
|
|
|
|
var last_val: c.LLVMValueRef = null;
|
|
if (fd.body.data == .block) {
|
|
for (fd.body.data.block.stmts) |stmt| {
|
|
last_val = try self.genStmt(stmt);
|
|
}
|
|
} else {
|
|
last_val = try self.genExpr(fd.body);
|
|
}
|
|
|
|
const current_bb2 = self.getCurrentBlock();
|
|
if (c.LLVMGetBasicBlockTerminator(current_bb2) == null) {
|
|
if (ret_sx_type == .void_type) {
|
|
self.retVoid();
|
|
} else if (last_val) |val| {
|
|
const src_ty = self.llvmTypeToSxType(c.LLVMTypeOf(val));
|
|
const converted = self.convertValue(val, src_ty, ret_sx_type);
|
|
self.ret(converted);
|
|
} else {
|
|
self.retVoid();
|
|
}
|
|
}
|
|
|
|
// Restore outer function state
|
|
self.named_values = saved_named;
|
|
self.current_return_type = saved_ret;
|
|
self.current_function = saved_fn;
|
|
self.positionAt(saved_bb);
|
|
}
|
|
return null;
|
|
},
|
|
.struct_decl => |sd| {
|
|
try self.registerStructType(sd);
|
|
return null;
|
|
},
|
|
.union_decl => {
|
|
// C-style union — registration handled in type pass
|
|
return null;
|
|
},
|
|
.assignment => |asgn| {
|
|
return self.genAssignment(asgn);
|
|
},
|
|
.multi_assign => |ma| {
|
|
return self.genMultiAssign(ma);
|
|
},
|
|
.return_stmt => |rs| {
|
|
// Evaluate return value first, then emit all defers, then return
|
|
if (rs.value) |val_node| {
|
|
const raw_val = try self.genExpr(val_node);
|
|
const ret_val = try self.prepareReturnValue(raw_val, self.current_return_type);
|
|
try self.emitAllDefers();
|
|
self.ret(ret_val);
|
|
} else {
|
|
try self.emitAllDefers();
|
|
self.retVoid();
|
|
}
|
|
// Create a dead basic block for any subsequent instructions
|
|
_ = self.appendBlock(self.current_function, "after_ret");
|
|
return null;
|
|
},
|
|
.defer_stmt => |ds| {
|
|
// Don't generate now — push onto current defer list for later execution
|
|
if (self.scope_stack.items.len > 0) {
|
|
const top = &self.scope_stack.items[self.scope_stack.items.len - 1].defers;
|
|
try top.append(self.allocator, ds.expr);
|
|
}
|
|
return null;
|
|
},
|
|
.insert_expr => |ins| {
|
|
// Substitute comptime param nodes before evaluation (e.g., replace $fmt identifier with literal)
|
|
const expr = if (self.comptime_param_nodes != null)
|
|
try self.substituteComptimeNodes(ins.expr)
|
|
else
|
|
ins.expr;
|
|
// Evaluate the inner expression via bytecode VM to get a string, parse it, generate inline
|
|
const result = try self.comptimeEval(expr, .string_type);
|
|
const code_z = try self.allocator.dupeZ(u8, result.string_val);
|
|
var parser = Parser.init(self.allocator, code_z);
|
|
var last_val: c.LLVMValueRef = null;
|
|
while (parser.current.tag != .eof) {
|
|
const stmt = try parser.parseStmt();
|
|
last_val = try self.genStmt(stmt);
|
|
}
|
|
return last_val;
|
|
},
|
|
else => {
|
|
return self.genExpr(node);
|
|
},
|
|
}
|
|
}
|
|
|
|
fn genVarDecl(self: *CodeGen, vd: ast.VarDecl) !c.LLVMValueRef {
|
|
// Meta type variable: x := f64 or x := Vec4 → runtime string holding the type name
|
|
if (vd.value) |val| {
|
|
const meta_name = self.asTypeName(val);
|
|
if (meta_name) |raw_name| {
|
|
const type_name = try self.allocator.dupeZ(u8, raw_name);
|
|
const name_z = try self.allocator.dupeZ(u8, vd.name);
|
|
const ptr_ty = self.ptrType();
|
|
const alloca = self.buildEntryBlockAlloca(ptr_ty, name_z.ptr);
|
|
const str_val = self.buildGlobalString(type_name.ptr, "type_name");
|
|
_ = c.LLVMBuildStore(self.builder, str_val, alloca);
|
|
try self.registerVariable(vd.name, alloca, .{ .meta_type = .{ .name = raw_name } });
|
|
return null;
|
|
}
|
|
}
|
|
|
|
var sx_ty: Type = Type.s(64);
|
|
|
|
if (vd.type_annotation) |ta| {
|
|
sx_ty = self.resolveType(ta);
|
|
} else if (vd.value) |val| {
|
|
// Infer type from value
|
|
if (val.data == .struct_literal) {
|
|
const sl = val.data.struct_literal;
|
|
if (sl.struct_name) |name| {
|
|
sx_ty = .{ .struct_type = name };
|
|
} else if (sl.type_expr) |te| {
|
|
sx_ty = self.resolveType(te);
|
|
} else {
|
|
return self.emitError("cannot infer struct type from untyped struct literal");
|
|
}
|
|
} else if (val.data == .array_literal and val.data.array_literal.type_expr != null) {
|
|
sx_ty = self.resolveType(val.data.array_literal.type_expr);
|
|
} else {
|
|
sx_ty = self.inferType(val);
|
|
}
|
|
} else {
|
|
return self.emitErrorFmt("variable '{s}' has no type annotation and no initializer", .{vd.name});
|
|
}
|
|
|
|
// Struct-typed variable
|
|
if (sx_ty.isStruct()) {
|
|
// Resolve type aliases (e.g. Vec3 -> Vec__3_f32)
|
|
const sname = self.resolveAlias(sx_ty.struct_type);
|
|
sx_ty = .{ .struct_type = sname };
|
|
const info = try self.getStructInfo(sname);
|
|
const alloca = try self.buildNamedAlloca(info.llvm_type, vd.name);
|
|
|
|
if (vd.value == null) {
|
|
// Default-init: per-field defaults or zero
|
|
try self.genStructDefaultInit(alloca, info);
|
|
} else if (vd.value.?.data == .undef_literal) {
|
|
// Undef-init: entire struct is undefined
|
|
self.storeUndef(info.llvm_type, alloca);
|
|
} else if (vd.value.?.data == .struct_literal) {
|
|
// Struct literal codegen returns an alloca — use it directly instead
|
|
const lit_alloca = try self.genStructLiteral(vd.value.?.data.struct_literal, sname);
|
|
try self.registerVariable(vd.name, lit_alloca, sx_ty);
|
|
return null;
|
|
} else if (vd.value.?.data == .call) {
|
|
// Function call returning a struct — result is a value, store to alloca
|
|
const val = try self.genExpr(vd.value.?);
|
|
_ = c.LLVMBuildStore(self.builder, val, alloca);
|
|
} else {
|
|
// General expression (xx cast, identifier, etc.) — evaluate as target type
|
|
const val = try self.genExprAsType(vd.value.?, sx_ty);
|
|
_ = c.LLVMBuildStore(self.builder, val, alloca);
|
|
}
|
|
|
|
try self.registerVariable(vd.name, alloca, sx_ty);
|
|
return null;
|
|
}
|
|
|
|
// Union-typed variable (tagged enum or C-style union)
|
|
if (sx_ty.isUnion()) {
|
|
const uname = self.resolveAlias(sx_ty.union_type);
|
|
sx_ty = .{ .union_type = uname };
|
|
|
|
// C-style (untagged) union
|
|
if (self.lookupUnionInfo(uname)) |info| {
|
|
const alloca = try self.buildNamedAlloca(info.llvm_type, vd.name);
|
|
|
|
if (vd.value == null) {
|
|
self.storeNull(info.llvm_type, alloca);
|
|
} else if (vd.value.?.data == .undef_literal) {
|
|
self.storeUndef(info.llvm_type, alloca);
|
|
} else {
|
|
return self.emitErrorFmt("union '{s}' must be initialized with '---' or field assignment", .{uname});
|
|
}
|
|
|
|
try self.registerVariable(vd.name, alloca, sx_ty);
|
|
return null;
|
|
}
|
|
|
|
// Tagged enum
|
|
const info = try self.getTaggedEnumInfo(uname);
|
|
const alloca = try self.buildNamedAlloca(info.llvm_type, vd.name);
|
|
|
|
if (vd.value == null) {
|
|
// Zero-init: tag=0, payload zeroed
|
|
self.storeNull(info.llvm_type, alloca);
|
|
} else if (vd.value.?.data == .undef_literal) {
|
|
self.storeUndef(info.llvm_type, alloca);
|
|
} else if (vd.value.?.data == .enum_literal) {
|
|
const el = vd.value.?.data.enum_literal;
|
|
const lit_alloca = try self.genTaggedEnumLiteral(el, uname);
|
|
try self.registerVariable(vd.name, lit_alloca, sx_ty);
|
|
return null;
|
|
} else if (vd.value.?.data == .call) {
|
|
// Call returning a union — could be enum construction (alloca) or function call (value)
|
|
const result = try self.genExpr(vd.value.?);
|
|
_ = c.LLVMBuildStore(self.builder, self.loadIfPointer(result, info.llvm_type, "union_load"), alloca);
|
|
} else {
|
|
// Other expression — try genExprAsType
|
|
const result = try self.genExprAsType(vd.value.?, sx_ty);
|
|
_ = c.LLVMBuildStore(self.builder, self.loadIfPointer(result, info.llvm_type, "union_load"), alloca);
|
|
}
|
|
|
|
try self.registerVariable(vd.name, alloca, sx_ty);
|
|
return null;
|
|
}
|
|
|
|
// Array-typed variable
|
|
if (sx_ty.isArray()) {
|
|
const arr_info = sx_ty.array_type;
|
|
const llvm_arr_ty = self.typeToLLVM(sx_ty);
|
|
const arr_alloca = try self.buildNamedAlloca(llvm_arr_ty, vd.name);
|
|
|
|
if (vd.value == null) {
|
|
self.storeNull(llvm_arr_ty, arr_alloca);
|
|
} else if (vd.value.?.data == .undef_literal) {
|
|
self.storeUndef(llvm_arr_ty, arr_alloca);
|
|
} else if (vd.value.?.data == .array_literal) {
|
|
const al = vd.value.?.data.array_literal;
|
|
const elem_sx_ty = try self.resolveElementType(arr_info.element_name, "array");
|
|
const elem_llvm_ty = self.typeToLLVM(elem_sx_ty);
|
|
const len = @min(al.elements.len, arr_info.length);
|
|
for (0..len) |i| {
|
|
const val = try self.genExprAsType(al.elements[i], elem_sx_ty);
|
|
const gep = self.gepArrayElement(llvm_arr_ty, arr_alloca, self.constInt32(@intCast(i)), "arr_elem");
|
|
_ = c.LLVMBuildStore(self.builder, val, gep);
|
|
}
|
|
// Zero-init remaining elements
|
|
for (len..arr_info.length) |i| {
|
|
const gep = self.gepArrayElement(llvm_arr_ty, arr_alloca, self.constInt32(@intCast(i)), "arr_elem");
|
|
self.storeNull(elem_llvm_ty, gep);
|
|
}
|
|
} else {
|
|
return self.emitErrorFmt("unsupported initializer for array variable '{s}'", .{vd.name});
|
|
}
|
|
|
|
try self.registerVariable(vd.name, arr_alloca, sx_ty);
|
|
return null;
|
|
}
|
|
|
|
// Vector-typed variable
|
|
if (sx_ty.isVector()) {
|
|
const llvm_vec_ty = self.typeToLLVM(sx_ty);
|
|
const vec_alloca = try self.buildNamedAlloca(llvm_vec_ty, vd.name);
|
|
|
|
if (vd.value == null) {
|
|
self.storeNull(llvm_vec_ty, vec_alloca);
|
|
} else if (vd.value.?.data == .undef_literal) {
|
|
self.storeUndef(llvm_vec_ty, vec_alloca);
|
|
} else if (vd.value.?.data == .array_literal) {
|
|
const vec_val = try self.genVectorLiteral(vd.value.?.data.array_literal, sx_ty);
|
|
_ = c.LLVMBuildStore(self.builder, vec_val, vec_alloca);
|
|
} else {
|
|
// Expression (e.g. function call) returning a vector
|
|
const val = try self.genExpr(vd.value.?);
|
|
_ = c.LLVMBuildStore(self.builder, val, vec_alloca);
|
|
}
|
|
|
|
try self.registerVariable(vd.name, vec_alloca, sx_ty);
|
|
return null;
|
|
}
|
|
|
|
// Function pointer typed variable
|
|
if (sx_ty.isFunctionType()) {
|
|
const llvm_ty = self.ptrType();
|
|
const alloca = try self.buildNamedAlloca(llvm_ty, vd.name);
|
|
|
|
if (vd.value == null) {
|
|
self.storeNull(llvm_ty, alloca);
|
|
} else if (vd.value.?.data == .undef_literal) {
|
|
self.storeUndef(llvm_ty, alloca);
|
|
} else if (vd.value.?.data == .unary_op and vd.value.?.data.unary_op.op == .xx) {
|
|
// xx cast: e.g. xx SDL_GL_GetProcAddress("glClear")
|
|
const inner = vd.value.?.data.unary_op.operand;
|
|
const val = try self.genExpr(inner);
|
|
const src_ty = self.inferType(inner);
|
|
const converted = self.convertValue(val, src_ty, sx_ty);
|
|
_ = c.LLVMBuildStore(self.builder, converted, alloca);
|
|
} else {
|
|
// Direct assignment: identifier (function name) or other expression
|
|
const val = try self.genExpr(vd.value.?);
|
|
_ = c.LLVMBuildStore(self.builder, val, alloca);
|
|
}
|
|
|
|
try self.registerVariable(vd.name, alloca, sx_ty);
|
|
return null;
|
|
}
|
|
|
|
// Guard: void type cannot be allocated (would crash LLVM)
|
|
if (sx_ty == .void_type) {
|
|
return self.emitErrorFmt("cannot declare variable '{s}' with void type", .{vd.name});
|
|
}
|
|
|
|
// Non-struct types
|
|
const llvm_ty = self.typeToLLVM(sx_ty);
|
|
const alloca = try self.buildNamedAlloca(llvm_ty, vd.name);
|
|
|
|
if (vd.value == null) {
|
|
// Default-init: zero
|
|
self.storeNull(llvm_ty, alloca);
|
|
} else if (vd.value.?.data == .undef_literal) {
|
|
// Undef-init
|
|
self.storeUndef(llvm_ty, alloca);
|
|
} else {
|
|
const val = vd.value.?;
|
|
const enum_name: ?[]const u8 = if (sx_ty.isEnum()) sx_ty.enum_type else null;
|
|
const init_val = if (val.data == .enum_literal and enum_name != null)
|
|
self.genEnumLiteral(val.data.enum_literal.name, enum_name.?)
|
|
else if (vd.type_annotation != null)
|
|
try self.genExprAsType(val, sx_ty)
|
|
else
|
|
try self.genExpr(val);
|
|
_ = c.LLVMBuildStore(self.builder, init_val, alloca);
|
|
}
|
|
|
|
try self.registerVariable(vd.name, alloca, sx_ty);
|
|
return null;
|
|
}
|
|
|
|
fn genStructDefaultInit(self: *CodeGen, alloca: c.LLVMValueRef, info: StructInfo) !void {
|
|
for (info.field_names, 0..) |_, fi| {
|
|
const ft = info.field_types[fi];
|
|
const ft_llvm = self.typeToLLVM(ft);
|
|
const gep = self.structGEP(info.llvm_type, alloca, @intCast(fi), "dinit");
|
|
|
|
if (info.field_defaults.len > fi and info.field_defaults[fi] != null) {
|
|
const default_node = info.field_defaults[fi].?;
|
|
if (default_node.data == .undef_literal) {
|
|
// Field default is --- → store undef
|
|
self.storeUndef(ft_llvm, gep);
|
|
} else {
|
|
// Field has expression default → evaluate and convert
|
|
const val = try self.genExprAsType(default_node, ft);
|
|
if (ft.isStruct() or ft.isUnion()) {
|
|
// Aggregate types: genExprAsType returns an alloca, load the value first
|
|
const loaded = c.LLVMBuildLoad2(self.builder, ft_llvm, val, "dinit_load");
|
|
_ = c.LLVMBuildStore(self.builder, loaded, gep);
|
|
} else {
|
|
_ = c.LLVMBuildStore(self.builder, val, gep);
|
|
}
|
|
}
|
|
} else {
|
|
// No default → zero
|
|
self.storeNull(ft_llvm, gep);
|
|
}
|
|
}
|
|
}
|
|
|
|
fn genConstDecl(self: *CodeGen, cd: ast.ConstDecl) !c.LLVMValueRef {
|
|
// Compile-time evaluation: register as comptime global for JIT
|
|
if (cd.value.data == .comptime_expr) {
|
|
const ct_type_override: ?Type = if (cd.type_annotation) |te| Type.fromTypeExpr(te) else null;
|
|
try self.registerComptimeGlobal(cd.name, cd.value.data.comptime_expr.expr, ct_type_override);
|
|
return null;
|
|
}
|
|
|
|
// Local lambda: register as function, generate body, done
|
|
if (cd.value.data == .lambda) {
|
|
const saved_fn = self.current_function;
|
|
const saved_bb = self.getCurrentBlock();
|
|
const saved_ret = self.current_return_type;
|
|
const saved_named = self.named_values;
|
|
self.named_values = std.StringHashMap(NamedValue).init(self.allocator);
|
|
|
|
try self.registerLambdaAsFunction(cd.name, cd.value.data.lambda);
|
|
try self.genLambdaBody(cd.name, cd.value.data.lambda);
|
|
|
|
self.named_values.deinit();
|
|
self.named_values = saved_named;
|
|
self.current_return_type = saved_ret;
|
|
self.current_function = saved_fn;
|
|
self.positionAt(saved_bb);
|
|
return null;
|
|
}
|
|
|
|
var sx_ty: Type = Type.s(64);
|
|
|
|
if (cd.type_annotation) |ta| {
|
|
sx_ty = self.resolveType(ta);
|
|
} else {
|
|
sx_ty = self.inferType(cd.value);
|
|
}
|
|
|
|
// Enum-typed constant: delegate to genExprAsType which handles enum_literal
|
|
if (sx_ty.isUnion()) {
|
|
const val = try self.genExprAsType(cd.value, sx_ty);
|
|
try self.registerVariable(cd.name, val, sx_ty);
|
|
return null;
|
|
}
|
|
|
|
// Function pointer typed constant
|
|
if (sx_ty.isFunctionType()) {
|
|
const llvm_ty = self.ptrType();
|
|
const alloca = try self.buildNamedAlloca(llvm_ty, cd.name);
|
|
if (cd.value.data == .unary_op and cd.value.data.unary_op.op == .xx) {
|
|
const inner = cd.value.data.unary_op.operand;
|
|
const val = try self.genExpr(inner);
|
|
const src_inner_ty = self.inferType(inner);
|
|
const converted = self.convertValue(val, src_inner_ty, sx_ty);
|
|
_ = c.LLVMBuildStore(self.builder, converted, alloca);
|
|
} else {
|
|
const val = try self.genExpr(cd.value);
|
|
_ = c.LLVMBuildStore(self.builder, val, alloca);
|
|
}
|
|
try self.registerVariable(cd.name, alloca, sx_ty);
|
|
return null;
|
|
}
|
|
|
|
const enum_name: ?[]const u8 = if (sx_ty.isEnum()) sx_ty.enum_type else null;
|
|
const init_val = if (cd.value.data == .enum_literal and enum_name != null)
|
|
self.genEnumLiteral(cd.value.data.enum_literal.name, enum_name.?)
|
|
else if (cd.type_annotation != null)
|
|
try self.genExprAsType(cd.value, sx_ty)
|
|
else
|
|
try self.genExpr(cd.value);
|
|
|
|
const llvm_ty = self.typeToLLVM(sx_ty);
|
|
const alloca = try self.buildNamedAlloca(llvm_ty, cd.name);
|
|
_ = c.LLVMBuildStore(self.builder, init_val, alloca);
|
|
try self.registerVariable(cd.name, alloca, sx_ty);
|
|
return null;
|
|
}
|
|
|
|
fn genCompoundOp(self: *CodeGen, op: ast.Assignment.Op, cur: c.LLVMValueRef, rhs: c.LLVMValueRef, ty: Type) c.LLVMValueRef {
|
|
return switch (op) {
|
|
.add_assign => if (ty.isFloat()) c.LLVMBuildFAdd(self.builder, cur, rhs, "addtmp") else c.LLVMBuildAdd(self.builder, cur, rhs, "addtmp"),
|
|
.sub_assign => if (ty.isFloat()) c.LLVMBuildFSub(self.builder, cur, rhs, "subtmp") else c.LLVMBuildSub(self.builder, cur, rhs, "subtmp"),
|
|
.mul_assign => if (ty.isFloat()) c.LLVMBuildFMul(self.builder, cur, rhs, "multmp") else c.LLVMBuildMul(self.builder, cur, rhs, "multmp"),
|
|
.div_assign => if (ty.isFloat()) c.LLVMBuildFDiv(self.builder, cur, rhs, "divtmp") else if (ty.isUnsigned()) c.LLVMBuildUDiv(self.builder, cur, rhs, "divtmp") else c.LLVMBuildSDiv(self.builder, cur, rhs, "divtmp"),
|
|
.mod_assign => if (ty.isFloat()) c.LLVMBuildFRem(self.builder, cur, rhs, "modtmp") else if (ty.isUnsigned()) c.LLVMBuildURem(self.builder, cur, rhs, "modtmp") else c.LLVMBuildSRem(self.builder, cur, rhs, "modtmp"),
|
|
.assign => unreachable,
|
|
};
|
|
}
|
|
|
|
fn storeOrCompound(self: *CodeGen, op: ast.Assignment.Op, ptr: c.LLVMValueRef, rhs: c.LLVMValueRef, field_ty: Type, name: [*c]const u8) void {
|
|
if (op == .assign) {
|
|
_ = c.LLVMBuildStore(self.builder, rhs, ptr);
|
|
} else {
|
|
const llvm_ty = self.typeToLLVM(field_ty);
|
|
const cur = c.LLVMBuildLoad2(self.builder, llvm_ty, ptr, name);
|
|
_ = c.LLVMBuildStore(self.builder, self.genCompoundOp(op, cur, rhs, field_ty), ptr);
|
|
}
|
|
}
|
|
|
|
fn genAssignment(self: *CodeGen, asgn: ast.Assignment) !c.LLVMValueRef {
|
|
// Field assignment: expr.field = value;
|
|
if (asgn.target.data == .field_access) {
|
|
return self.genFieldAssignment(asgn);
|
|
}
|
|
|
|
// Index assignment: expr[i] = value;
|
|
if (asgn.target.data == .index_expr) {
|
|
return self.genIndexAssignment(asgn);
|
|
}
|
|
|
|
// Deref assignment: p.* = value;
|
|
if (asgn.target.data == .deref_expr) {
|
|
const de = asgn.target.data.deref_expr;
|
|
const ptr_val = try self.genExpr(de.operand);
|
|
const ptr_ty = self.inferType(de.operand);
|
|
if (!ptr_ty.isPointer()) return self.emitError("dereference assignment requires a pointer");
|
|
const pointee_ty = self.resolveTypeFromName(ptr_ty.pointer_type.pointee_name) orelse return self.emitError("unknown pointee type");
|
|
const new_val = try self.genExprAsType(asgn.value, pointee_ty);
|
|
_ = c.LLVMBuildStore(self.builder, new_val, ptr_val);
|
|
return null;
|
|
}
|
|
|
|
// Target must be an identifier
|
|
if (asgn.target.data != .identifier) return self.emitError("assignment target must be a variable");
|
|
const name = asgn.target.data.identifier.name;
|
|
const lookup = self.lookupValue(name) orelse
|
|
return self.emitErrorFmt("undefined variable '{s}'", .{name});
|
|
const entry = lookup.asNamedValue() orelse
|
|
return self.emitErrorFmt("cannot assign to constant '{s}'", .{name});
|
|
if (entry.is_const)
|
|
return self.emitErrorFmt("cannot assign to '{s}'", .{name});
|
|
|
|
// Meta type reassignment: x = Vec4, x = f64, x = test
|
|
if (entry.ty == .meta_type and asgn.op == .assign) {
|
|
const raw_name = self.asTypeName(asgn.value) orelse blk: {
|
|
// Also accept function names as meta_type values (use signature)
|
|
if (asgn.value.data == .identifier) {
|
|
const fn_name = asgn.value.data.identifier.name;
|
|
if (self.fn_signatures.get(fn_name)) |sig| break :blk sig;
|
|
}
|
|
break :blk @as(?[]const u8, null);
|
|
};
|
|
if (raw_name) |rn| {
|
|
const type_name = try self.allocator.dupeZ(u8, rn);
|
|
const str_val = self.buildGlobalString(type_name.ptr, "type_name");
|
|
_ = c.LLVMBuildStore(self.builder, str_val, entry.ptr);
|
|
if (self.named_values.getPtr(name)) |entry_ptr| {
|
|
entry_ptr.ty = .{ .meta_type = .{ .name = rn } };
|
|
}
|
|
return null;
|
|
}
|
|
return self.emitErrorFmt("cannot assign non-type value to Type variable '{s}'", .{name});
|
|
}
|
|
|
|
// Function pointer reassignment
|
|
if (entry.ty.isFunctionType() and asgn.op == .assign) {
|
|
if (asgn.value.data == .unary_op and asgn.value.data.unary_op.op == .xx) {
|
|
const inner = asgn.value.data.unary_op.operand;
|
|
const val = try self.genExpr(inner);
|
|
const src_ty = self.inferType(inner);
|
|
const converted = self.convertValue(val, src_ty, entry.ty);
|
|
_ = c.LLVMBuildStore(self.builder, converted, entry.ptr);
|
|
} else {
|
|
const val = try self.genExpr(asgn.value);
|
|
_ = c.LLVMBuildStore(self.builder, val, entry.ptr);
|
|
}
|
|
return null;
|
|
}
|
|
|
|
// Tagged enum reassignment: s = .circle(3.14) or s = .none or s = fn_call()
|
|
if (entry.ty.isUnion() and asgn.op == .assign) {
|
|
if (self.lookupTaggedEnumInfo(entry.ty.union_type)) |info| {
|
|
const new_val = try self.genExprAsType(asgn.value, entry.ty);
|
|
// genExprAsType returns alloca for enum literals, loaded value for calls
|
|
_ = c.LLVMBuildStore(self.builder, self.loadIfPointer(new_val, info.llvm_type, "union_load"), entry.ptr);
|
|
return null;
|
|
}
|
|
// C-style union: full assignment not supported, use field assignment
|
|
}
|
|
|
|
const new_val = try self.genExpr(asgn.value);
|
|
const llvm_ty = self.typeToLLVM(entry.ty);
|
|
|
|
const store_val = if (asgn.op == .assign) new_val else blk: {
|
|
const cur = c.LLVMBuildLoad2(self.builder, llvm_ty, entry.ptr, "cur");
|
|
break :blk self.genCompoundOp(asgn.op, cur, new_val, entry.ty);
|
|
};
|
|
|
|
_ = c.LLVMBuildStore(self.builder, store_val, entry.ptr);
|
|
return null;
|
|
}
|
|
|
|
fn genMultiAssign(self: *CodeGen, ma: ast.MultiAssign) !c.LLVMValueRef {
|
|
const n = ma.targets.len;
|
|
|
|
// Phase 1: Evaluate ALL RHS values into temporaries.
|
|
// This ensures correctness for aliased swaps (a, b = b, a).
|
|
const tmp_ptrs = try self.allocator.alloc(c.LLVMValueRef, n);
|
|
const target_types = try self.allocator.alloc(Type, n);
|
|
|
|
for (0..n) |i| {
|
|
target_types[i] = self.inferType(ma.targets[i]);
|
|
const llvm_ty = self.typeToLLVM(target_types[i]);
|
|
const val = try self.genExprAsType(ma.values[i], target_types[i]);
|
|
const tmp = self.buildEntryBlockAlloca(llvm_ty, "swap_tmp");
|
|
_ = c.LLVMBuildStore(self.builder, val, tmp);
|
|
tmp_ptrs[i] = tmp;
|
|
}
|
|
|
|
// Phase 2: Load temporaries and store to each target.
|
|
for (0..n) |i| {
|
|
const llvm_ty = self.typeToLLVM(target_types[i]);
|
|
const val = c.LLVMBuildLoad2(self.builder, llvm_ty, tmp_ptrs[i], "swap_load");
|
|
try self.storeToLvalue(ma.targets[i], val);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
fn storeToLvalue(self: *CodeGen, target: *Node, val: c.LLVMValueRef) !void {
|
|
// Deref assignment: p.* = val
|
|
if (target.data == .deref_expr) {
|
|
const de = target.data.deref_expr;
|
|
const ptr_val = try self.genExpr(de.operand);
|
|
_ = c.LLVMBuildStore(self.builder, val, ptr_val);
|
|
return;
|
|
}
|
|
|
|
// Identifier assignment: x = val (with const check)
|
|
if (target.data == .identifier) {
|
|
const name = target.data.identifier.name;
|
|
const lookup = self.lookupValue(name) orelse
|
|
return self.emitErrorFmt("undefined variable '{s}'", .{name});
|
|
const entry = lookup.asNamedValue() orelse
|
|
return self.emitErrorFmt("cannot assign to constant '{s}'", .{name});
|
|
if (entry.is_const)
|
|
return self.emitErrorFmt("cannot assign to '{s}'", .{name});
|
|
_ = c.LLVMBuildStore(self.builder, val, entry.ptr);
|
|
return;
|
|
}
|
|
|
|
// Field access and index expressions — use genAddressOf to get the target pointer
|
|
if (target.data == .field_access or target.data == .index_expr) {
|
|
const ptr = try self.genAddressOf(target);
|
|
_ = c.LLVMBuildStore(self.builder, val, ptr);
|
|
return;
|
|
}
|
|
|
|
return self.emitError("multi-assign target must be a variable, field, index, or dereference expression");
|
|
}
|
|
|
|
fn genFieldAssignment(self: *CodeGen, asgn: ast.Assignment) !c.LLVMValueRef {
|
|
const fa = asgn.target.data.field_access;
|
|
|
|
// Object must be an identifier for now
|
|
if (fa.object.data != .identifier) return self.emitError("field assignment target must be a variable");
|
|
const obj_name = fa.object.data.identifier.name;
|
|
const entry = self.named_values.get(obj_name) orelse return self.emitErrorFmt("undefined variable '{s}'", .{obj_name});
|
|
|
|
// Pointer auto-deref: p.field = val
|
|
if (entry.ty.isPointer()) {
|
|
const pointee_ty = self.resolveTypeFromName(entry.ty.pointer_type.pointee_name) orelse
|
|
return self.emitError("unknown pointee type for field assignment");
|
|
if (pointee_ty.isStruct()) {
|
|
const sname = pointee_ty.struct_type;
|
|
const info = try self.getStructInfo(sname);
|
|
const fi = try self.findFieldIndex(info.field_names, fa.field, sname);
|
|
const field_ty = info.field_types[fi];
|
|
const loaded_ptr = c.LLVMBuildLoad2(self.builder,
|
|
self.ptrType(), entry.ptr, "ptr_load");
|
|
const gep = self.structGEP(info.llvm_type, loaded_ptr, @intCast(fi), "pfield_ptr");
|
|
const rhs = try self.genExprAsType(asgn.value, field_ty);
|
|
self.storeOrCompound(asgn.op, gep, rhs, field_ty, "pcur");
|
|
return null;
|
|
}
|
|
return self.emitError("field assignment through pointer requires a struct pointee");
|
|
}
|
|
|
|
// C-style union field assignment
|
|
if (entry.ty.isUnion()) {
|
|
const uname = entry.ty.union_type;
|
|
if (self.lookupUnionInfo(uname)) |info| {
|
|
if (self.findNameIndex(info.field_names, fa.field)) |fidx| {
|
|
const field_ty = info.field_types[fidx];
|
|
const rhs = try self.genExprAsType(asgn.value, field_ty);
|
|
self.storeOrCompound(asgn.op, entry.ptr, rhs, field_ty, "ucur");
|
|
return null;
|
|
}
|
|
// Check promoted fields from anonymous structs
|
|
if (info.promoted_fields.get(fa.field)) |pf| {
|
|
const sinfo = try self.getStructInfo(pf.struct_name);
|
|
const gep = self.structGEP(sinfo.llvm_type, entry.ptr, @intCast(pf.field_index), "promoted_ptr");
|
|
const rhs = try self.genExprAsType(asgn.value, pf.field_type);
|
|
self.storeOrCompound(asgn.op, gep, rhs, pf.field_type, "ucur");
|
|
return null;
|
|
}
|
|
return self.emitErrorFmt("no field '{s}' in union '{s}'", .{ fa.field, uname });
|
|
}
|
|
return self.emitErrorFmt("field assignment not supported on tagged enum '{s}'", .{uname});
|
|
}
|
|
|
|
if (!entry.ty.isStruct()) return self.emitErrorFmt("field access on non-struct variable '{s}'", .{obj_name});
|
|
|
|
const sname = entry.ty.struct_type;
|
|
const info = try self.getStructInfo(sname);
|
|
const fi = try self.findFieldIndex(info.field_names, fa.field, sname);
|
|
const field_ty = info.field_types[fi];
|
|
|
|
const gep = self.structGEP(info.llvm_type, entry.ptr, @intCast(fi), "fassign");
|
|
|
|
// Generate RHS and convert to field type
|
|
const rhs = try self.genExprAsType(asgn.value, field_ty);
|
|
self.storeOrCompound(asgn.op, gep, rhs, field_ty, "fcur");
|
|
return null;
|
|
}
|
|
|
|
fn genIndexAssignment(self: *CodeGen, asgn: ast.Assignment) !c.LLVMValueRef {
|
|
const ie = asgn.target.data.index_expr;
|
|
const obj_ty = self.inferType(ie.object);
|
|
if (obj_ty == .string_type) {
|
|
// String index assignment: s[i] = c
|
|
const str_val = try self.genExpr(ie.object);
|
|
const ptr = self.extractValue(str_val, 0, "str_ptr");
|
|
const idx = try self.genExpr(ie.index);
|
|
const val = try self.genExpr(asgn.value);
|
|
const i8_type = self.i8Type();
|
|
const gep_ptr = self.gepPointerElement(i8_type, ptr, idx, "stridx");
|
|
const byte_val = self.trunc(val, i8_type, "trunc_byte");
|
|
_ = c.LLVMBuildStore(self.builder, byte_val, gep_ptr);
|
|
return null;
|
|
}
|
|
if (obj_ty.isArray()) {
|
|
if (ie.object.data == .identifier) {
|
|
if (self.named_values.get(ie.object.data.identifier.name)) |entry| {
|
|
const idx = try self.genExpr(ie.index);
|
|
const val = try self.genExpr(asgn.value);
|
|
const gep_ptr = self.gepArrayElement(self.typeToLLVM(obj_ty), entry.ptr, idx, "arridx");
|
|
_ = c.LLVMBuildStore(self.builder, val, gep_ptr);
|
|
return null;
|
|
}
|
|
}
|
|
// struct.field[i] = val — GEP through struct to array field, then index
|
|
if (ie.object.data == .field_access) {
|
|
const field_ptr = try self.genAddressOf(ie.object);
|
|
const idx = try self.genExpr(ie.index);
|
|
const val = try self.genExpr(asgn.value);
|
|
const gep_ptr = self.gepArrayElement(self.typeToLLVM(obj_ty), field_ptr, idx, "field_arridx");
|
|
_ = c.LLVMBuildStore(self.builder, val, gep_ptr);
|
|
return null;
|
|
}
|
|
}
|
|
if (obj_ty.isSlice()) {
|
|
const slice_info = obj_ty.slice_type;
|
|
const elem_ty = Type.fromName(slice_info.element_name) orelse return self.emitError("unknown slice element type");
|
|
const elem_llvm_ty = self.typeToLLVM(elem_ty);
|
|
// Load slice value to get ptr
|
|
const slice_val = blk: {
|
|
if (ie.object.data == .identifier) {
|
|
if (self.named_values.get(ie.object.data.identifier.name)) |entry| {
|
|
break :blk c.LLVMBuildLoad2(self.builder, self.getStringStructType(), entry.ptr, "slice_load");
|
|
}
|
|
}
|
|
break :blk try self.genExpr(ie.object);
|
|
};
|
|
const ptr = self.extractValue(slice_val, 0, "slice_ptr");
|
|
const idx = try self.genExpr(ie.index);
|
|
const val = try self.genExpr(asgn.value);
|
|
const gep_ptr = self.gepPointerElement(elem_llvm_ty, ptr, idx, "sliceidx");
|
|
_ = c.LLVMBuildStore(self.builder, val, gep_ptr);
|
|
return null;
|
|
}
|
|
// Many-pointer index assignment: mp[i] = val
|
|
if (obj_ty.isManyPointer()) {
|
|
const elem_ty = self.resolveTypeFromName(obj_ty.many_pointer_type.element_name) orelse return self.emitError("unknown many-pointer element type");
|
|
const elem_llvm_ty = self.typeToLLVM(elem_ty);
|
|
const ptr_val = try self.genExpr(ie.object);
|
|
const idx = try self.genExpr(ie.index);
|
|
const val = try self.genExprAsType(asgn.value, elem_ty);
|
|
const gep_ptr = self.gepPointerElement(elem_llvm_ty, ptr_val, idx, "mptridx");
|
|
_ = c.LLVMBuildStore(self.builder, val, gep_ptr);
|
|
return null;
|
|
}
|
|
return self.emitError("index assignment requires a string, array, slice, or [*] pointer target");
|
|
}
|
|
|
|
fn genExpr(self: *CodeGen, node: *Node) anyerror!c.LLVMValueRef {
|
|
self.current_span = node.span;
|
|
switch (node.data) {
|
|
.int_literal => |lit| {
|
|
const i64_type = self.i64Type();
|
|
return c.LLVMConstInt(i64_type, @bitCast(@as(i64, lit.value)), 0);
|
|
},
|
|
.float_literal => |lit| {
|
|
const f32_type = self.f32Type();
|
|
return c.LLVMConstReal(f32_type, lit.value);
|
|
},
|
|
.bool_literal => |lit| {
|
|
const i1_type = self.i1Type();
|
|
return c.LLVMConstInt(i1_type, if (lit.value) 1 else 0, 0);
|
|
},
|
|
.string_literal => |lit| {
|
|
const content = if (lit.is_raw) lit.raw else try unescape.unescapeString(self.allocator, lit.raw);
|
|
const str_z = try self.allocator.dupeZ(u8, content);
|
|
const ptr = self.buildGlobalString(str_z.ptr, "str");
|
|
return self.buildStringSlice(ptr, self.constInt64(@intCast(content.len)));
|
|
},
|
|
.identifier => |ident| {
|
|
if (self.lookupValue(ident.name)) |v| {
|
|
switch (v) {
|
|
.local => |nv| return c.LLVMBuildLoad2(self.builder, self.typeToLLVM(nv.ty), nv.ptr, "loadtmp"),
|
|
.comptime_global => |ct| {
|
|
if (!ct.is_resolved) try self.resolveComptimeGlobal(ct);
|
|
return c.LLVMBuildLoad2(self.builder, self.typeToLLVM(ct.ty), ct.global, "ct_load");
|
|
},
|
|
.global_mutable => |gm| return c.LLVMBuildLoad2(self.builder, self.typeToLLVM(gm.ty), gm.ptr, "global_load"),
|
|
}
|
|
}
|
|
// Fall back to function name → function pointer value
|
|
{
|
|
var nbuf: [256]u8 = undefined;
|
|
var fn_val = c.LLVMGetNamedFunction(self.module, self.nameToCStr(ident.name, &nbuf));
|
|
if (fn_val == null) {
|
|
// Try qualified name with current namespace
|
|
if (self.current_namespace) |ns| {
|
|
const qualified = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns, ident.name });
|
|
var qbuf: [256]u8 = undefined;
|
|
fn_val = c.LLVMGetNamedFunction(self.module, self.nameToCStr(qualified, &qbuf));
|
|
}
|
|
}
|
|
if (fn_val != null) return fn_val.?;
|
|
}
|
|
return self.emitErrorFmt("undefined identifier '{s}'", .{ident.name});
|
|
},
|
|
.binary_op => |binop| {
|
|
if (binop.op == .and_op) return self.genShortCircuitOp(binop, true);
|
|
if (binop.op == .or_op) return self.genShortCircuitOp(binop, false);
|
|
const lhs_ty = self.inferType(binop.lhs);
|
|
const rhs_ty = self.inferType(binop.rhs);
|
|
const result_type = Type.widen(lhs_ty, rhs_ty);
|
|
|
|
// Tagged enum comparison: compare tags only
|
|
if (result_type.isUnion() and (binop.op == .eq or binop.op == .neq)) {
|
|
const uname = result_type.union_type;
|
|
const resolved = self.resolveAlias(uname);
|
|
const info = self.lookupTaggedEnumInfo(resolved) orelse return self.emitError("unknown tagged enum type");
|
|
const tag_ty = self.getEnumLLVMType(resolved);
|
|
|
|
var lhs_val = try self.genExprAsType(binop.lhs, result_type);
|
|
var rhs_val = try self.genExprAsType(binop.rhs, result_type);
|
|
|
|
// If either side is a pointer (alloca from genTaggedEnumLiteral), load it
|
|
lhs_val = self.loadIfPointer(lhs_val, info.llvm_type, "union_load_l");
|
|
rhs_val = self.loadIfPointer(rhs_val, info.llvm_type, "union_load_r");
|
|
|
|
// Extract tags (field 0) and compare
|
|
const lhs_tag = self.extractValue(lhs_val, 0, "lhs_tag");
|
|
const rhs_tag = self.extractValue(rhs_val, 0, "rhs_tag");
|
|
_ = tag_ty;
|
|
const pred: c_uint = if (binop.op == .eq) c.LLVMIntEQ else c.LLVMIntNE;
|
|
return self.icmp(pred, lhs_tag, rhs_tag, "tag_cmp");
|
|
}
|
|
|
|
const lhs = try self.genExprAsType(binop.lhs, result_type);
|
|
const rhs = try self.genExprAsType(binop.rhs, result_type);
|
|
return self.genBinaryOp(binop.op, lhs, rhs, result_type);
|
|
},
|
|
.chained_comparison => |chain| {
|
|
return self.genChainedComparison(chain);
|
|
},
|
|
.unary_op => |unop| {
|
|
if (unop.op == .xx) {
|
|
// xx requires a target type context (assignment, declaration, argument, return)
|
|
return self.emitError("'xx' cast requires a target type context");
|
|
}
|
|
if (unop.op == .address_of) {
|
|
return self.genAddressOf(unop.operand);
|
|
}
|
|
const operand = try self.genExpr(unop.operand);
|
|
return switch (unop.op) {
|
|
.negate => blk: {
|
|
const operand_ty = self.inferType(unop.operand);
|
|
if (operand_ty.isVector()) {
|
|
const elem_ty = operand_ty.vectorElementType();
|
|
break :blk if (elem_ty != null and elem_ty.?.isFloat())
|
|
c.LLVMBuildFNeg(self.builder, operand, "vnegtmp")
|
|
else
|
|
c.LLVMBuildNeg(self.builder, operand, "vnegtmp");
|
|
}
|
|
break :blk if (self.exprIsFloat(unop.operand))
|
|
c.LLVMBuildFNeg(self.builder, operand, "negtmp")
|
|
else
|
|
c.LLVMBuildNeg(self.builder, operand, "negtmp");
|
|
},
|
|
.not => c.LLVMBuildNot(self.builder, operand, "nottmp"),
|
|
.xx, .address_of => unreachable,
|
|
};
|
|
},
|
|
.enum_literal => |el| {
|
|
if (self.current_return_type.isUnion()) {
|
|
return self.genTaggedEnumLiteral(el, self.current_return_type.union_type);
|
|
}
|
|
if (self.current_return_type.isEnum()) {
|
|
return self.genEnumLiteral(el.name, self.current_return_type.enum_type);
|
|
}
|
|
return self.emitError("cannot infer enum type for literal");
|
|
},
|
|
.struct_literal => |sl| {
|
|
const ctx_name: ?[]const u8 = if (self.current_return_type.isStruct()) self.current_return_type.struct_type else null;
|
|
return self.genStructLiteral(sl, ctx_name);
|
|
},
|
|
.array_literal => |al| {
|
|
// Typed array/vector/slice literal: Type.[elems]
|
|
if (al.type_expr) |te| {
|
|
const ty = self.resolveType(te);
|
|
if (ty.isVector()) return self.genVectorLiteral(al, ty);
|
|
if (ty.isArray()) return self.genArrayLiteral(al, ty);
|
|
if (ty.isSlice()) return self.genSliceLiteral(al, ty);
|
|
}
|
|
// If current return type is vector, build as vector SSA value
|
|
if (self.current_return_type.isVector()) {
|
|
return self.genVectorLiteral(al, self.current_return_type);
|
|
}
|
|
return self.genArrayLiteral(al, null);
|
|
},
|
|
.field_access => |fa| {
|
|
return self.genFieldAccess(fa);
|
|
},
|
|
.index_expr => |ie| {
|
|
return self.genIndexExpr(ie);
|
|
},
|
|
.slice_expr => |se| {
|
|
return self.genSliceExpr(se);
|
|
},
|
|
.call => |call_node| {
|
|
return self.genCall(call_node);
|
|
},
|
|
.if_expr => |ie| {
|
|
return self.genIfExpr(ie);
|
|
},
|
|
.match_expr => |me| {
|
|
return self.genMatchExpr(me);
|
|
},
|
|
.while_expr => |we| {
|
|
return self.genWhileExpr(we);
|
|
},
|
|
.for_expr => |fe| {
|
|
return self.genForExpr(fe);
|
|
},
|
|
.break_expr => {
|
|
if (self.loop_break_bb) |break_bb| {
|
|
self.br(break_bb);
|
|
_ = self.appendBlock(self.current_function, "after_break");
|
|
return null;
|
|
}
|
|
return self.emitError("'break' outside of loop");
|
|
},
|
|
.continue_expr => {
|
|
if (self.loop_continue_bb) |continue_bb| {
|
|
self.br(continue_bb);
|
|
_ = self.appendBlock(self.current_function, "after_continue");
|
|
return null;
|
|
}
|
|
return self.emitError("'continue' outside of loop");
|
|
},
|
|
.block => |blk| {
|
|
try self.pushScope();
|
|
var last_val: c.LLVMValueRef = null;
|
|
for (blk.stmts) |stmt| {
|
|
last_val = try self.genStmt(stmt);
|
|
}
|
|
try self.popScope();
|
|
return last_val;
|
|
},
|
|
.var_decl => |vd| {
|
|
return self.genVarDecl(vd);
|
|
},
|
|
.const_decl => |cd| {
|
|
return self.genConstDecl(cd);
|
|
},
|
|
.assignment => |asgn| {
|
|
return self.genAssignment(asgn);
|
|
},
|
|
.multi_assign => |ma| {
|
|
return self.genMultiAssign(ma);
|
|
},
|
|
.return_stmt => |rs| {
|
|
if (rs.value) |val_node| {
|
|
const raw_val = try self.genExpr(val_node);
|
|
const ret_val = try self.prepareReturnValue(raw_val, self.current_return_type);
|
|
try self.emitAllDefers();
|
|
self.ret(ret_val);
|
|
} else {
|
|
try self.emitAllDefers();
|
|
self.retVoid();
|
|
}
|
|
_ = self.appendBlock(self.current_function, "after_ret");
|
|
return null;
|
|
},
|
|
.deref_expr => |de| {
|
|
const ptr_val = try self.genExpr(de.operand);
|
|
const ptr_ty = self.inferType(de.operand);
|
|
if (ptr_ty.isPointer()) {
|
|
const pointee_ty = self.resolveTypeFromName(ptr_ty.pointer_type.pointee_name) orelse return self.emitError("unknown pointee type");
|
|
return self.loadTyped(pointee_ty, ptr_val, "deref");
|
|
}
|
|
return self.emitError("dereference requires a pointer type");
|
|
},
|
|
.null_literal => {
|
|
return c.LLVMConstNull(self.ptrType());
|
|
},
|
|
.comptime_expr => |ct| {
|
|
return self.genExpr(ct.expr);
|
|
},
|
|
else => return self.emitError("unsupported expression"),
|
|
}
|
|
}
|
|
|
|
fn genAddressOf(self: *CodeGen, operand: *Node) !c.LLVMValueRef {
|
|
// &x — return the alloca pointer of the variable
|
|
if (operand.data == .identifier) {
|
|
if (self.named_values.get(operand.data.identifier.name)) |entry| {
|
|
return entry.ptr;
|
|
}
|
|
return self.emitErrorFmt("undefined variable '{s}'", .{operand.data.identifier.name});
|
|
}
|
|
// &expr[i] — return GEP pointer to the indexed element
|
|
if (operand.data == .index_expr) {
|
|
const ie = operand.data.index_expr;
|
|
const obj_ty = self.inferType(ie.object);
|
|
const idx = try self.genExpr(ie.index);
|
|
|
|
if (obj_ty.isArray()) {
|
|
if (ie.object.data == .identifier) {
|
|
if (self.named_values.get(ie.object.data.identifier.name)) |entry| {
|
|
return self.gepArrayElement(self.typeToLLVM(obj_ty), entry.ptr, idx, "addr_elem");
|
|
}
|
|
}
|
|
}
|
|
if (obj_ty.isSlice() or obj_ty == .string_type) {
|
|
const slice_val = try self.genExpr(ie.object);
|
|
const ptr = self.extractValue(slice_val, 0, "slice_ptr");
|
|
const elem_ty = if (obj_ty.isSlice())
|
|
obj_ty.sliceElementType() orelse return self.emitError("unknown slice element type")
|
|
else
|
|
Type.u(8);
|
|
return self.gepPointerElement(self.typeToLLVM(elem_ty), ptr, idx, "addr_elem");
|
|
}
|
|
}
|
|
// &s.field — return GEP pointer to the struct field
|
|
if (operand.data == .field_access) {
|
|
const fa = operand.data.field_access;
|
|
if (fa.object.data == .identifier) {
|
|
if (self.named_values.get(fa.object.data.identifier.name)) |entry| {
|
|
if (entry.ty.isStruct()) {
|
|
const sname = entry.ty.struct_type;
|
|
const info = try self.getStructInfo(sname);
|
|
const idx = try self.findFieldIndex(info.field_names, fa.field, sname);
|
|
return self.structGEP(info.llvm_type, entry.ptr, @intCast(idx), "addr_field");
|
|
}
|
|
// &u.field where u is a C-style union — all fields at offset 0
|
|
if (entry.ty.isUnion()) {
|
|
if (self.lookupUnionInfo(entry.ty.union_type)) |info| {
|
|
if (self.findNameIndex(info.field_names, fa.field) != null) {
|
|
return entry.ptr;
|
|
}
|
|
if (info.promoted_fields.get(fa.field)) |pf| {
|
|
const sinfo = try self.getStructInfo(pf.struct_name);
|
|
return self.structGEP(sinfo.llvm_type, entry.ptr, @intCast(pf.field_index), "addr_promoted");
|
|
}
|
|
}
|
|
}
|
|
// &p.field where p is *Struct — auto-deref through pointer
|
|
if (entry.ty.isPointer()) {
|
|
const pointee_name = entry.ty.pointer_type.pointee_name;
|
|
if (self.lookupStructInfo(pointee_name)) |info| {
|
|
const loaded_ptr = c.LLVMBuildLoad2(self.builder,
|
|
self.ptrType(), entry.ptr, "ptr_load");
|
|
const idx = try self.findFieldIndex(info.field_names, fa.field, pointee_name);
|
|
return self.structGEP(info.llvm_type, loaded_ptr, @intCast(idx), "addr_pfield");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return self.emitError("address-of requires a variable, index, or field expression");
|
|
}
|
|
|
|
const StructBuildResult = struct {
|
|
field_sx_types: []const Type,
|
|
llvm_type: c.LLVMTypeRef,
|
|
};
|
|
|
|
fn buildStructFields(self: *CodeGen, name: []const u8, field_type_nodes: []const *Node) !StructBuildResult {
|
|
var field_sx_types = std.ArrayList(Type).empty;
|
|
var field_llvm_types = std.ArrayList(c.LLVMTypeRef).empty;
|
|
|
|
for (field_type_nodes) |ft| {
|
|
const sx_ty = self.resolveType(ft);
|
|
try field_sx_types.append(self.allocator, sx_ty);
|
|
try field_llvm_types.append(self.allocator, self.typeToLLVM(sx_ty));
|
|
}
|
|
|
|
const llvm_types_slice = try field_llvm_types.toOwnedSlice(self.allocator);
|
|
const name_z = try self.allocator.dupeZ(u8, name);
|
|
const struct_ty = c.LLVMStructCreateNamed(self.context, name_z.ptr);
|
|
c.LLVMStructSetBody(struct_ty, if (llvm_types_slice.len > 0) llvm_types_slice.ptr else null, @intCast(llvm_types_slice.len), 0);
|
|
|
|
return .{
|
|
.field_sx_types = try field_sx_types.toOwnedSlice(self.allocator),
|
|
.llvm_type = struct_ty,
|
|
};
|
|
}
|
|
|
|
const UnionBuildResult = struct {
|
|
variant_sx_types: []const Type,
|
|
llvm_type: c.LLVMTypeRef,
|
|
max_payload_size: u64,
|
|
payload_field_index: c_uint,
|
|
};
|
|
|
|
fn buildUnionFields(self: *CodeGen, name: []const u8, variant_type_nodes: []const ?*Node) !UnionBuildResult {
|
|
var variant_sx_types = std.ArrayList(Type).empty;
|
|
var max_payload_size: u64 = 0;
|
|
|
|
for (variant_type_nodes) |vt| {
|
|
if (vt) |type_node| {
|
|
const sx_ty = self.resolveType(type_node);
|
|
try variant_sx_types.append(self.allocator, sx_ty);
|
|
const llvm_ty = self.typeToLLVM(sx_ty);
|
|
const size = self.getTypeSize(llvm_ty);
|
|
if (size > max_payload_size) max_payload_size = size;
|
|
} else {
|
|
try variant_sx_types.append(self.allocator, .void_type);
|
|
}
|
|
}
|
|
|
|
const name_z = try self.allocator.dupeZ(u8, name);
|
|
const union_ty = c.LLVMStructCreateNamed(self.context, name_z.ptr);
|
|
const tag_ty = self.getEnumLLVMType(name);
|
|
const i8_ty = self.i8Type();
|
|
|
|
const payload_array_ty = c.LLVMArrayType2(i8_ty, max_payload_size);
|
|
var union_fields = [2]c.LLVMTypeRef{ tag_ty, payload_array_ty };
|
|
c.LLVMStructSetBody(union_ty, &union_fields, 2, 0);
|
|
|
|
return .{
|
|
.variant_sx_types = try variant_sx_types.toOwnedSlice(self.allocator),
|
|
.llvm_type = union_ty,
|
|
.max_payload_size = max_payload_size,
|
|
.payload_field_index = 1,
|
|
};
|
|
}
|
|
|
|
fn hoistInlineTypeDecl(self: *CodeGen, parent_name: []const u8, child_name: []const u8, type_node: *Node) anyerror!void {
|
|
const synthetic_name = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ parent_name, child_name });
|
|
switch (type_node.data) {
|
|
.struct_decl => |inline_sd| {
|
|
var hoisted = inline_sd;
|
|
hoisted.name = synthetic_name;
|
|
try self.registerStructType(hoisted);
|
|
type_node.data = .{ .type_expr = .{ .name = synthetic_name } };
|
|
},
|
|
.union_decl => |inline_ud| {
|
|
var hoisted_ud = inline_ud;
|
|
hoisted_ud.name = synthetic_name;
|
|
try self.registerUnionType(hoisted_ud);
|
|
type_node.data = .{ .type_expr = .{ .name = synthetic_name } };
|
|
},
|
|
.enum_decl => |inline_ed| {
|
|
if (inline_ed.variant_types.len > 0) {
|
|
// Tagged enum with payloads
|
|
var hoisted = inline_ed;
|
|
hoisted.name = synthetic_name;
|
|
try self.registerTaggedEnum(hoisted);
|
|
} else {
|
|
try self.type_registry.put(synthetic_name, .{ .plain_enum = inline_ed.variant_names });
|
|
_ = try self.getAnyTypeId(synthetic_name, .{ .enum_type = synthetic_name });
|
|
if (inline_ed.backing_type) |bt_node| {
|
|
const bt = self.resolveType(bt_node);
|
|
try self.enum_backing_types.put(synthetic_name, self.typeToLLVM(bt));
|
|
}
|
|
}
|
|
type_node.data = .{ .type_expr = .{ .name = synthetic_name } };
|
|
},
|
|
else => {},
|
|
}
|
|
}
|
|
|
|
fn registerStructType(self: *CodeGen, sd: ast.StructDecl) anyerror!void {
|
|
// Generic struct: store as template instead of registering now
|
|
if (sd.type_params.len > 0) {
|
|
try self.generic_struct_templates.put(sd.name, sd);
|
|
return;
|
|
}
|
|
|
|
// Pre-pass: hoist inline type declarations from field types
|
|
for (sd.field_types, 0..) |ft, i| {
|
|
try self.hoistInlineTypeDecl(sd.name, sd.field_names[i], ft);
|
|
}
|
|
|
|
const build = try self.buildStructFields(sd.name, sd.field_types);
|
|
|
|
// Process field defaults: replace #run expressions with comptime global references
|
|
var resolved_defaults = try self.allocator.alloc(?*Node, sd.field_defaults.len);
|
|
for (sd.field_defaults, 0..) |fd, i| {
|
|
if (fd != null and fd.?.data == .comptime_expr) {
|
|
const synthetic_name = try std.fmt.allocPrint(self.allocator, "__struct_{s}_field_{d}", .{ sd.name, i });
|
|
const field_type_override: ?Type = if (i < build.field_sx_types.len) build.field_sx_types[i] else null;
|
|
try self.registerComptimeGlobal(synthetic_name, fd.?.data.comptime_expr.expr, field_type_override);
|
|
const id_node = try self.allocator.create(Node);
|
|
id_node.* = .{ .span = .{ .start = 0, .end = 0 }, .data = .{ .identifier = .{ .name = synthetic_name } } };
|
|
resolved_defaults[i] = id_node;
|
|
} else {
|
|
resolved_defaults[i] = fd;
|
|
}
|
|
}
|
|
|
|
const sinfo = StructInfo{
|
|
.field_names = sd.field_names,
|
|
.field_types = build.field_sx_types,
|
|
.field_defaults = resolved_defaults,
|
|
.llvm_type = build.llvm_type,
|
|
};
|
|
try self.type_registry.put(sd.name, .{ .struct_info = sinfo });
|
|
_ = try self.getAnyTypeId(sd.name, .{ .struct_type = sd.name });
|
|
}
|
|
|
|
fn registerTaggedEnum(self: *CodeGen, ud: ast.EnumDecl) !void {
|
|
// Pre-pass: hoist inline type declarations from variant types
|
|
for (ud.variant_types, 0..) |vt_opt, i| {
|
|
if (vt_opt) |vt| {
|
|
try self.hoistInlineTypeDecl(ud.name, ud.variant_names[i], vt);
|
|
}
|
|
}
|
|
|
|
// Check if backing type is a struct layout specification
|
|
const layout_info = try self.resolveEnumLayout(ud);
|
|
|
|
if (layout_info) |layout| {
|
|
// Struct-backed layout: use the struct's LLVM type directly
|
|
try self.enum_backing_types.put(ud.name, layout.tag_llvm_type);
|
|
|
|
// Resolve variant sx types
|
|
var variant_sx_types = std.ArrayList(Type).empty;
|
|
for (ud.variant_types) |vt| {
|
|
if (vt) |type_node| {
|
|
try variant_sx_types.append(self.allocator, self.resolveType(type_node));
|
|
} else {
|
|
try variant_sx_types.append(self.allocator, .void_type);
|
|
}
|
|
}
|
|
|
|
const tei_layout = TaggedEnumInfo{
|
|
.variant_names = ud.variant_names,
|
|
.variant_types = try variant_sx_types.toOwnedSlice(self.allocator),
|
|
.llvm_type = layout.llvm_type,
|
|
.max_payload_size = layout.payload_size,
|
|
.payload_field_index = layout.payload_field_index,
|
|
};
|
|
try self.type_registry.put(ud.name, .{ .tagged_enum = tei_layout });
|
|
} else {
|
|
// Primitive backing type (e.g. enum u32 { ... })
|
|
if (ud.backing_type) |bt_node| {
|
|
const bt = self.resolveType(bt_node);
|
|
try self.enum_backing_types.put(ud.name, self.typeToLLVM(bt));
|
|
}
|
|
|
|
const build = try self.buildUnionFields(ud.name, ud.variant_types);
|
|
|
|
const tei_build = TaggedEnumInfo{
|
|
.variant_names = ud.variant_names,
|
|
.variant_types = build.variant_sx_types,
|
|
.llvm_type = build.llvm_type,
|
|
.max_payload_size = build.max_payload_size,
|
|
.payload_field_index = build.payload_field_index,
|
|
};
|
|
try self.type_registry.put(ud.name, .{ .tagged_enum = tei_build });
|
|
}
|
|
|
|
_ = try self.getAnyTypeId(ud.name, .{ .union_type = ud.name });
|
|
|
|
// Compute and store variant values (explicit or sequential)
|
|
const values = try self.allocator.alloc(i64, ud.variant_names.len);
|
|
for (ud.variant_names, 0..) |_, i| {
|
|
if (ud.variant_values.len > i and ud.variant_values[i] != null) {
|
|
const val_node = ud.variant_values[i].?;
|
|
values[i] = switch (val_node.data) {
|
|
.int_literal => |il| il.value,
|
|
else => @as(i64, @intCast(i)),
|
|
};
|
|
} else {
|
|
values[i] = @intCast(i);
|
|
}
|
|
}
|
|
try self.enum_variant_values.put(ud.name, values);
|
|
}
|
|
|
|
const EnumLayoutInfo = struct {
|
|
llvm_type: c.LLVMTypeRef,
|
|
tag_llvm_type: c.LLVMTypeRef,
|
|
payload_field_index: c_uint,
|
|
payload_size: u64,
|
|
};
|
|
|
|
/// Resolve a struct-backed layout for a tagged enum.
|
|
/// Returns null if the backing type is a primitive (e.g. u32), in which case
|
|
/// the caller should fall back to buildUnionFields.
|
|
///
|
|
/// The layout struct must have:
|
|
/// - A field named `tag` (integer type) — the discriminant
|
|
/// - A field named `payload` (array type) — the overlay area for variant data
|
|
/// - Any other fields are treated as padding/reserved
|
|
fn resolveEnumLayout(self: *CodeGen, ud: ast.EnumDecl) !?EnumLayoutInfo {
|
|
const bt_node = ud.backing_type orelse return null;
|
|
|
|
// Check for inline struct: enum struct { ... } { ... }
|
|
if (bt_node.data == .struct_decl) {
|
|
const layout_name = try std.fmt.allocPrint(self.allocator, "{s}.__layout", .{ud.name});
|
|
var sd = bt_node.data.struct_decl;
|
|
sd.name = layout_name;
|
|
try self.registerStructType(sd);
|
|
return try self.validateEnumLayout(ud.name, layout_name);
|
|
}
|
|
|
|
// Check for named struct reference: enum MyLayout { ... }
|
|
if (bt_node.data == .type_expr) {
|
|
const name = bt_node.data.type_expr.name;
|
|
// If it resolves to a primitive type, it's not a layout struct
|
|
if (Type.fromName(name) != null) return null;
|
|
// Check type aliases
|
|
const resolved = self.resolveAlias(name);
|
|
if (Type.fromName(resolved) != null) return null;
|
|
// Must be a registered struct
|
|
if (self.lookupStructInfo(resolved) != null) {
|
|
return try self.validateEnumLayout(ud.name, resolved);
|
|
}
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
fn validateEnumLayout(self: *CodeGen, enum_name: []const u8, layout_name: []const u8) !EnumLayoutInfo {
|
|
const layout = self.lookupStructInfo(layout_name) orelse {
|
|
return self.emitErrorFmt("enum '{s}': layout type '{s}' is not a registered struct", .{ enum_name, layout_name });
|
|
};
|
|
|
|
// Find 'tag' field
|
|
var tag_index: ?usize = null;
|
|
var payload_index: ?usize = null;
|
|
for (layout.field_names, 0..) |fname, i| {
|
|
if (std.mem.eql(u8, fname, "tag")) {
|
|
tag_index = i;
|
|
} else if (std.mem.eql(u8, fname, "payload")) {
|
|
payload_index = i;
|
|
}
|
|
}
|
|
|
|
if (tag_index == null) {
|
|
return self.emitErrorFmt(
|
|
"enum '{s}': layout struct '{s}' must have a field named 'tag' (the discriminant). Expected layout: struct {{ tag: <int_type>; payload: [N]<type>; }}",
|
|
.{ enum_name, layout_name },
|
|
);
|
|
}
|
|
if (payload_index == null) {
|
|
return self.emitErrorFmt(
|
|
"enum '{s}': layout struct '{s}' must have a field named 'payload' (the variant data area). Expected layout: struct {{ tag: <int_type>; payload: [N]<type>; }}",
|
|
.{ enum_name, layout_name },
|
|
);
|
|
}
|
|
|
|
const tag_ty = layout.field_types[tag_index.?];
|
|
const payload_ty = layout.field_types[payload_index.?];
|
|
|
|
// Validate tag is an integer type
|
|
switch (tag_ty) {
|
|
.signed, .unsigned => {},
|
|
else => return self.emitErrorFmt(
|
|
"enum '{s}': layout field 'tag' must be an integer type (e.g. u32), got '{s}'",
|
|
.{ enum_name, tag_ty.displayName(self.allocator) catch "?" },
|
|
),
|
|
}
|
|
|
|
// Validate payload is an array type
|
|
const payload_size = switch (payload_ty) {
|
|
.array_type => |info| blk: {
|
|
const elem_ty = Type.fromName(info.element_name) orelse {
|
|
return self.emitErrorFmt(
|
|
"enum '{s}': layout field 'payload' has unresolved element type '{s}'",
|
|
.{ enum_name, info.element_name },
|
|
);
|
|
};
|
|
const elem_llvm = self.typeToLLVM(elem_ty);
|
|
break :blk self.getTypeSize(elem_llvm) * info.length;
|
|
},
|
|
else => return self.emitErrorFmt(
|
|
"enum '{s}': layout field 'payload' must be an array type (e.g. [30]u32), got '{s}'",
|
|
.{ enum_name, payload_ty.displayName(self.allocator) catch "?" },
|
|
),
|
|
};
|
|
|
|
return .{
|
|
.llvm_type = layout.llvm_type,
|
|
.tag_llvm_type = self.typeToLLVM(tag_ty),
|
|
.payload_field_index = @intCast(payload_index.?),
|
|
.payload_size = payload_size,
|
|
};
|
|
}
|
|
|
|
fn registerUnionType(self: *CodeGen, ud: ast.UnionDecl) !void {
|
|
// Hoist inline type declarations from field types
|
|
for (ud.field_types, 0..) |ft, i| {
|
|
try self.hoistInlineTypeDecl(ud.name, ud.field_names[i], ft);
|
|
}
|
|
|
|
// Compute max field size and resolve field types
|
|
const data_layout = c.LLVMGetModuleDataLayout(self.module);
|
|
var field_sx_types = std.ArrayList(Type).empty;
|
|
var max_size: u64 = 0;
|
|
for (ud.field_types) |ft| {
|
|
const resolved = self.resolveType(ft);
|
|
try field_sx_types.append(self.allocator, resolved);
|
|
const llvm_ty = self.typeToLLVM(resolved);
|
|
const size = c.LLVMABISizeOfType(data_layout, llvm_ty);
|
|
if (size > max_size) max_size = size;
|
|
}
|
|
|
|
// LLVM type: byte array sized to the largest field
|
|
const byte_ty = self.i8Type();
|
|
const llvm_type = c.LLVMArrayType(byte_ty, @intCast(max_size));
|
|
|
|
const resolved_field_types = try field_sx_types.toOwnedSlice(self.allocator);
|
|
|
|
// Build promoted fields map from anonymous struct members
|
|
var promoted = std.StringHashMap(PromotedField).init(self.allocator);
|
|
for (ud.field_names, 0..) |_, i| {
|
|
const fty = resolved_field_types[i];
|
|
if (fty.isStruct()) {
|
|
// Check if this is an anonymous struct (name contains __anon_)
|
|
const sname = fty.struct_type;
|
|
if (std.mem.indexOf(u8, sname, ".__anon_") != null) {
|
|
if (self.lookupStructInfo(sname)) |sinfo| {
|
|
for (sinfo.field_names, 0..) |sf_name, sf_idx| {
|
|
try promoted.put(sf_name, .{
|
|
.struct_name = sname,
|
|
.field_index = sf_idx,
|
|
.field_type = sinfo.field_types[sf_idx],
|
|
});
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
const uinfo = UnionInfo{
|
|
.field_names = ud.field_names,
|
|
.field_types = resolved_field_types,
|
|
.llvm_type = llvm_type,
|
|
.total_size = max_size,
|
|
.promoted_fields = promoted,
|
|
};
|
|
try self.type_registry.put(ud.name, .{ .union_info = uinfo });
|
|
// Note: C-style unions are not registered with the Any type system.
|
|
// They can't be meaningfully printed as a whole — access individual fields instead.
|
|
}
|
|
|
|
fn genTaggedEnumLiteral(self: *CodeGen, el: ast.EnumLiteral, expected_union_name: ?[]const u8) !c.LLVMValueRef {
|
|
const uname = expected_union_name orelse
|
|
(if (self.current_return_type.isUnion()) self.current_return_type.union_type else null) orelse
|
|
return self.emitError("cannot infer enum type for literal");
|
|
const resolved_name = self.resolveAlias(uname);
|
|
const info = try self.getTaggedEnumInfo(resolved_name);
|
|
|
|
// Find variant index
|
|
var variant_idx: ?u32 = null;
|
|
for (info.variant_names, 0..) |vn, i| {
|
|
if (std.mem.eql(u8, vn, el.name)) {
|
|
variant_idx = @intCast(i);
|
|
break;
|
|
}
|
|
}
|
|
const idx = variant_idx orelse return self.emitErrorFmt("no variant '{s}' in enum '{s}'", .{ el.name, resolved_name });
|
|
|
|
// Alloca union
|
|
const alloca = self.buildEntryBlockAlloca(info.llvm_type, "union_tmp");
|
|
const tag_ty = self.getEnumLLVMType(resolved_name);
|
|
|
|
// Store tag (field 0) — use explicit value if available, otherwise index
|
|
const tag_gep = self.structGEP(info.llvm_type, alloca, 0, "tag");
|
|
const tag_val: u64 = if (self.enum_variant_values.get(resolved_name)) |vals| @bitCast(vals[idx]) else idx;
|
|
_ = c.LLVMBuildStore(self.builder, c.LLVMConstInt(tag_ty, tag_val, 0), tag_gep);
|
|
|
|
// Store payload (field 1) if not void
|
|
if (el.payload) |payload_node| {
|
|
const variant_ty = info.variant_types[idx];
|
|
if (variant_ty != .void_type) {
|
|
const payload_val = try self.genExprAsType(payload_node, variant_ty);
|
|
self.storeStructField(info.llvm_type, alloca, info.payload_field_index, payload_val);
|
|
}
|
|
}
|
|
|
|
return alloca;
|
|
}
|
|
|
|
fn genStructLiteral(self: *CodeGen, sl: ast.StructLiteral, expected_struct_name: ?[]const u8) anyerror!c.LLVMValueRef {
|
|
const raw_name = sl.struct_name orelse blk: {
|
|
if (sl.type_expr) |te| {
|
|
const ty = self.resolveType(te);
|
|
if (ty.isStruct()) break :blk ty.struct_type;
|
|
}
|
|
break :blk expected_struct_name orelse return self.emitError("cannot infer struct type for literal");
|
|
};
|
|
// Resolve type aliases (e.g. Vec3 -> Vec__3_f32)
|
|
const sname = self.resolveAlias(raw_name);
|
|
const info = try self.getStructInfo(sname);
|
|
|
|
// Alloca the struct and default-init all fields (zero or declared defaults)
|
|
const name_z = try self.allocator.dupeZ(u8, sname);
|
|
const alloca = self.buildEntryBlockAlloca(info.llvm_type, name_z.ptr);
|
|
try self.genStructDefaultInit(alloca, info);
|
|
|
|
// Determine if this is named or positional mode
|
|
var has_named = false;
|
|
for (sl.field_inits) |fi| {
|
|
if (fi.name != null) {
|
|
has_named = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (has_named) {
|
|
// Named/shorthand mode: map by field name
|
|
for (sl.field_inits) |fi| {
|
|
const fname = fi.name orelse {
|
|
// Positional field mixed with named — treat as identifier shorthand
|
|
if (fi.value.data == .identifier) {
|
|
const idx = try self.findFieldIndex(info.field_names, fi.value.data.identifier.name, sname);
|
|
const val = try self.genExprAsType(fi.value, info.field_types[idx]);
|
|
self.storeStructField(info.llvm_type, alloca, @intCast(idx), val);
|
|
continue;
|
|
}
|
|
return self.emitError("mixed positional and named fields in struct literal");
|
|
};
|
|
const idx = try self.findFieldIndex(info.field_names, fname, sname);
|
|
const val = try self.genExprAsType(fi.value, info.field_types[idx]);
|
|
self.storeStructField(info.llvm_type, alloca, @intCast(idx), val);
|
|
}
|
|
} else {
|
|
// Positional mode: assign in order
|
|
for (sl.field_inits, 0..) |fi, i| {
|
|
if (i >= info.field_names.len) return self.emitErrorFmt("too many fields in struct literal (expected {d})", .{info.field_names.len});
|
|
const val = try self.genExprAsType(fi.value, info.field_types[i]);
|
|
self.storeStructField(info.llvm_type, alloca, @intCast(i), val);
|
|
}
|
|
}
|
|
|
|
return alloca;
|
|
}
|
|
|
|
/// Generate an array literal as an alloca with elements stored via GEP.
|
|
/// If target_ty is provided, elements are converted to the array's element type.
|
|
/// Otherwise, element type is inferred from the first element.
|
|
fn genArrayLiteral(self: *CodeGen, al: ast.ArrayLiteral, target_ty_opt: ?Type) !c.LLVMValueRef {
|
|
const arr_ty: Type = target_ty_opt orelse blk: {
|
|
// Infer from first element
|
|
if (al.elements.len == 0) return self.emitError("cannot infer type of empty array literal");
|
|
const elem_ty = self.inferType(al.elements[0]);
|
|
const elem_name = try elem_ty.displayName(self.allocator);
|
|
break :blk .{ .array_type = .{ .element_name = elem_name, .length = @intCast(al.elements.len) } };
|
|
};
|
|
const arr_info = arr_ty.array_type;
|
|
const elem_sx_ty = try self.resolveElementType(arr_info.element_name, "array");
|
|
const llvm_arr_ty = self.typeToLLVM(arr_ty);
|
|
const alloca = self.buildEntryBlockAlloca(llvm_arr_ty, "arr");
|
|
|
|
const len = @min(al.elements.len, arr_info.length);
|
|
for (0..len) |i| {
|
|
const val = try self.genExprAsType(al.elements[i], elem_sx_ty);
|
|
const gep = self.gepArrayElement(llvm_arr_ty, alloca, self.constInt32(@intCast(i)), "arr_elem");
|
|
_ = c.LLVMBuildStore(self.builder, val, gep);
|
|
}
|
|
return alloca;
|
|
}
|
|
|
|
fn genSliceLiteral(self: *CodeGen, al: ast.ArrayLiteral, slice_ty: Type) !c.LLVMValueRef {
|
|
const elem_name = slice_ty.slice_type.element_name;
|
|
const elem_sx_ty = try self.resolveElementType(elem_name, "slice");
|
|
const n: u32 = @intCast(al.elements.len);
|
|
|
|
// Create backing array [N]elem on the stack
|
|
const arr_ty: Type = .{ .array_type = .{ .element_name = elem_name, .length = n } };
|
|
const llvm_arr_ty = self.typeToLLVM(arr_ty);
|
|
const arr_alloca = self.buildEntryBlockAlloca(llvm_arr_ty, "slice_backing");
|
|
|
|
// Fill elements
|
|
for (0..n) |i| {
|
|
const val = try self.genExprAsType(al.elements[i], elem_sx_ty);
|
|
const gep = self.gepArrayElement(llvm_arr_ty, arr_alloca, self.constInt32(@intCast(i)), "slice_elem");
|
|
_ = c.LLVMBuildStore(self.builder, val, gep);
|
|
}
|
|
|
|
// Build slice {ptr, len}
|
|
const elem_ptr = self.arrayDecayToPointer(llvm_arr_ty, arr_alloca, "slice_data");
|
|
return self.buildFatPointer(self.getStringStructType(), elem_ptr, self.constInt64(n));
|
|
}
|
|
|
|
fn genVectorLiteral(self: *CodeGen, al: ast.ArrayLiteral, vec_ty: Type) !c.LLVMValueRef {
|
|
const vec_info = vec_ty.vector_type;
|
|
const elem_sx_ty = try self.resolveElementType(vec_info.element_name, "vector");
|
|
const llvm_vec_ty = self.typeToLLVM(vec_ty);
|
|
var vec_val = self.getUndef(llvm_vec_ty);
|
|
|
|
const len = @min(al.elements.len, vec_info.length);
|
|
for (0..len) |i| {
|
|
const elem_val = try self.genExprAsType(al.elements[i], elem_sx_ty);
|
|
const idx = self.constInt32(@intCast(i));
|
|
vec_val = c.LLVMBuildInsertElement(self.builder, vec_val, elem_val, idx, "vec_ins");
|
|
}
|
|
return vec_val;
|
|
}
|
|
|
|
fn broadcastScalar(self: *CodeGen, scalar: c.LLVMValueRef, vec_ty: Type) c.LLVMValueRef {
|
|
const vec_info = vec_ty.vector_type;
|
|
const llvm_vec_ty = self.typeToLLVM(vec_ty);
|
|
// Insert scalar at index 0 of undef vector
|
|
var vec = self.getUndef(llvm_vec_ty);
|
|
const zero = self.constInt32(0);
|
|
vec = c.LLVMBuildInsertElement(self.builder, vec, scalar, zero, "splat_ins");
|
|
// Shuffle with zeroinitializer mask to broadcast element 0 to all lanes
|
|
const mask_ty = c.LLVMVectorType(self.i32Type(), vec_info.length);
|
|
const mask = c.LLVMConstNull(mask_ty);
|
|
return c.LLVMBuildShuffleVector(self.builder, vec, self.getUndef(llvm_vec_ty), mask, "splat");
|
|
}
|
|
|
|
fn genExprAsType(self: *CodeGen, node: *Node, target_ty: Type) !c.LLVMValueRef {
|
|
self.current_span = node.span;
|
|
// xx prefix: unwrap and convert freely (explicit cast)
|
|
if (node.data == .unary_op and node.data.unary_op.op == .xx) {
|
|
const inner = node.data.unary_op.operand;
|
|
const val = try self.genExpr(inner);
|
|
const src_ty = self.inferType(inner);
|
|
return self.convertValue(val, src_ty, target_ty);
|
|
}
|
|
|
|
// Function pointer target: bypass narrowing check, just produce the pointer value
|
|
if (target_ty.isFunctionType()) {
|
|
return try self.genExpr(node);
|
|
}
|
|
|
|
// String literal → pointer context: produce raw pointer directly (no {ptr, len} wrapping)
|
|
if (node.data == .string_literal and target_ty.isPointer()) {
|
|
const lit = node.data.string_literal;
|
|
const content = if (lit.is_raw) lit.raw else try unescape.unescapeString(self.allocator, lit.raw);
|
|
const str_z = try self.allocator.dupeZ(u8, content);
|
|
return self.buildGlobalString(str_z.ptr, "str");
|
|
}
|
|
|
|
// Enum literal assigned to enum type: resolve variant value
|
|
if (node.data == .enum_literal and target_ty.isEnum()) {
|
|
return self.genEnumLiteral(node.data.enum_literal.name, target_ty.enum_type);
|
|
}
|
|
|
|
// Bitwise op on enum type: recursively generate both sides with enum context
|
|
if (node.data == .binary_op and (node.data.binary_op.op == .bit_or or node.data.binary_op.op == .bit_and) and target_ty.isEnum()) {
|
|
const binop = node.data.binary_op;
|
|
const lhs = try self.genExprAsType(binop.lhs, target_ty);
|
|
const rhs = try self.genExprAsType(binop.rhs, target_ty);
|
|
const b = self.builder;
|
|
return if (binop.op == .bit_or)
|
|
c.LLVMBuildOr(b, lhs, rhs, "bortmp")
|
|
else
|
|
c.LLVMBuildAnd(b, lhs, rhs, "bandtmp");
|
|
}
|
|
|
|
// Enum/union literal assigned to union type: construct tagged enum
|
|
if (node.data == .enum_literal and target_ty.isUnion()) {
|
|
const el = node.data.enum_literal;
|
|
return self.genTaggedEnumLiteral(el, target_ty.union_type);
|
|
}
|
|
|
|
// Struct literal targeting union type: .Variant.{fields} pattern
|
|
// Parsed as struct_literal with type_expr = enum_literal("Variant")
|
|
if (node.data == .struct_literal and target_ty.isUnion()) {
|
|
const sl = node.data.struct_literal;
|
|
if (sl.struct_name == null) {
|
|
if (sl.type_expr) |te| {
|
|
if (te.data == .enum_literal) {
|
|
const variant_name = te.data.enum_literal.name;
|
|
const uname = self.resolveAlias(target_ty.union_type);
|
|
const info = try self.getTaggedEnumInfo(uname);
|
|
|
|
// Find variant index
|
|
var variant_idx: ?u32 = null;
|
|
for (info.variant_names, 0..) |vn, vi| {
|
|
if (std.mem.eql(u8, vn, variant_name)) {
|
|
variant_idx = @intCast(vi);
|
|
break;
|
|
}
|
|
}
|
|
const idx = variant_idx orelse
|
|
return self.emitErrorFmt("no variant '{s}' in enum '{s}'", .{ variant_name, uname });
|
|
|
|
const variant_ty = info.variant_types[idx];
|
|
|
|
// Alloca union, store tag
|
|
const alloca = self.buildEntryBlockAlloca(info.llvm_type, "union_lit");
|
|
const tag_llvm_ty = self.getEnumLLVMType(uname);
|
|
self.storeStructField(info.llvm_type, alloca, 0, c.LLVMConstInt(tag_llvm_ty, idx, 0));
|
|
|
|
// Store struct payload
|
|
if (variant_ty != .void_type) {
|
|
const payload_struct_name = if (variant_ty.isStruct()) variant_ty.struct_type else null;
|
|
const payload_alloca = try self.genStructLiteral(.{
|
|
.struct_name = payload_struct_name,
|
|
.type_expr = null,
|
|
.field_inits = sl.field_inits,
|
|
}, payload_struct_name);
|
|
const payload_gep = self.structGEP(info.llvm_type, alloca, info.payload_field_index, "payload");
|
|
const payload_llvm_ty = self.typeToLLVM(variant_ty);
|
|
const struct_val = c.LLVMBuildLoad2(self.builder, payload_llvm_ty, payload_alloca, "struct_load");
|
|
_ = c.LLVMBuildStore(self.builder, struct_val, payload_gep);
|
|
}
|
|
|
|
return alloca;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Struct literal targeting struct type: pass struct name context
|
|
if (node.data == .struct_literal and target_ty.isStruct()) {
|
|
const alloca = try self.genStructLiteral(node.data.struct_literal, target_ty.struct_type);
|
|
// genStructLiteral returns an alloca pointer — load the value for by-value passing
|
|
const sname = self.resolveAlias(target_ty.struct_type);
|
|
if (self.lookupStructInfo(sname)) |si| {
|
|
return c.LLVMBuildLoad2(self.builder, si.llvm_type, alloca, "struct_val");
|
|
}
|
|
return alloca;
|
|
}
|
|
|
|
// Array literal with target array type: generate with element conversion
|
|
if (node.data == .array_literal and target_ty.isArray()) {
|
|
return self.genArrayLiteral(node.data.array_literal, target_ty);
|
|
}
|
|
|
|
// Array literal with target vector type: build via undef + InsertElement
|
|
if (node.data == .array_literal and target_ty.isVector()) {
|
|
return self.genVectorLiteral(node.data.array_literal, target_ty);
|
|
}
|
|
|
|
// Array literal with target slice type: build stack-backed slice
|
|
if (node.data == .array_literal and target_ty.isSlice()) {
|
|
return self.genSliceLiteral(node.data.array_literal, target_ty);
|
|
}
|
|
|
|
// Infer source type once for all coercion checks below
|
|
const src_ty = self.inferType(node);
|
|
|
|
// Array to slice coercion: [N]T → []T
|
|
if (target_ty.isSlice() and src_ty.isArray()) {
|
|
const arr_info = src_ty.array_type;
|
|
// Get the alloca pointer for the array (not the loaded value)
|
|
const arr_alloca = blk: {
|
|
if (node.data == .identifier) {
|
|
if (self.named_values.get(node.data.identifier.name)) |entry| {
|
|
break :blk entry.ptr;
|
|
}
|
|
}
|
|
if (node.data == .field_access) {
|
|
break :blk try self.genAddressOf(node);
|
|
}
|
|
// Fallback: generate the expression and hope it returns a pointer
|
|
break :blk try self.genExpr(node);
|
|
};
|
|
// GEP to get pointer to first element
|
|
const elem_ptr = self.arrayDecayToPointer(self.typeToLLVM(src_ty), arr_alloca, "arr_data");
|
|
// Build slice struct {ptr, len}
|
|
return self.buildFatPointer(self.getStringStructType(), elem_ptr, self.constInt64(arr_info.length));
|
|
}
|
|
|
|
// Array to many-pointer coercion: [N]T → [*]T
|
|
if (target_ty.isManyPointer() and src_ty.isArray()) {
|
|
const arr_info = src_ty.array_type;
|
|
if (std.mem.eql(u8, arr_info.element_name, target_ty.many_pointer_type.element_name)) {
|
|
const arr_alloca = blk: {
|
|
if (node.data == .identifier) {
|
|
if (self.named_values.get(node.data.identifier.name)) |entry| {
|
|
break :blk entry.ptr;
|
|
}
|
|
}
|
|
if (node.data == .field_access) {
|
|
break :blk try self.genAddressOf(node);
|
|
}
|
|
break :blk try self.genExpr(node);
|
|
};
|
|
return self.arrayDecayToPointer(self.typeToLLVM(src_ty), arr_alloca, "arr_decay");
|
|
}
|
|
}
|
|
|
|
// Slice to many-pointer coercion: []T → [*]T (extract .ptr from fat pointer)
|
|
if (target_ty.isManyPointer() and src_ty.isSlice()) {
|
|
if (std.mem.eql(u8, src_ty.slice_type.element_name, target_ty.many_pointer_type.element_name)) {
|
|
const slice_val = try self.genExpr(node);
|
|
return self.extractValue(slice_val, 0, "slice_decay");
|
|
}
|
|
}
|
|
|
|
// Implicit address-of: passing T where *T is expected → auto &
|
|
if (target_ty.isPointer()) {
|
|
const pointee_name = target_ty.pointer_type.pointee_name;
|
|
const src_matches = if (src_ty.isStruct())
|
|
std.mem.eql(u8, src_ty.struct_type, pointee_name) or
|
|
(if (self.lookupAlias(src_ty.struct_type)) |alias| std.mem.eql(u8, alias, pointee_name) else false) or
|
|
(if (self.lookupAlias(pointee_name)) |alias| std.mem.eql(u8, alias, src_ty.struct_type) else false)
|
|
else if (src_ty.isUnion()) blk: {
|
|
const uname = src_ty.union_type;
|
|
break :blk std.mem.eql(u8, uname, pointee_name) or
|
|
(if (self.lookupAlias(uname)) |alias| std.mem.eql(u8, alias, pointee_name) else false) or
|
|
(if (self.lookupAlias(pointee_name)) |alias| std.mem.eql(u8, alias, uname) else false);
|
|
} else if (Type.fromName(pointee_name)) |pointee_ty|
|
|
src_ty.eql(pointee_ty)
|
|
else
|
|
false;
|
|
if (src_matches) {
|
|
if (node.data == .identifier) {
|
|
if (self.named_values.get(node.data.identifier.name)) |entry| {
|
|
return entry.ptr;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
var val = try self.genExpr(node);
|
|
|
|
// Struct literals return alloca pointers — load the value for by-value passing
|
|
if (src_ty.isStruct() and target_ty.isStruct()) {
|
|
if (c.LLVMGetTypeKind(c.LLVMTypeOf(val)) == c.LLVMPointerTypeKind) {
|
|
const info = self.lookupStructInfo(src_ty.struct_type) orelse
|
|
self.lookupStructInfo(self.resolveAlias(src_ty.struct_type));
|
|
if (info) |si| {
|
|
val = c.LLVMBuildLoad2(self.builder, si.llvm_type, val, "struct_load");
|
|
}
|
|
}
|
|
}
|
|
|
|
// Scalar to vector broadcast
|
|
if (target_ty.isVector() and !src_ty.isVector()) {
|
|
const elem_ty = target_ty.vectorElementType() orelse return self.emitError("cannot determine vector element type");
|
|
const converted = self.convertValue(val, src_ty, elem_ty);
|
|
return self.broadcastScalar(converted, target_ty);
|
|
}
|
|
|
|
// Literals are exempt from narrowing checks
|
|
if (node.data == .int_literal or node.data == .float_literal) {
|
|
return self.convertValue(val, src_ty, target_ty);
|
|
}
|
|
|
|
// Check for narrowing conversion
|
|
if (!src_ty.isImplicitlyConvertibleTo(target_ty)) {
|
|
// Narrowing without xx — compile error
|
|
return self.emitErrorFmt("narrowing conversion from '{s}' to '{s}' requires explicit 'xx' cast", .{
|
|
src_ty.displayName(self.allocator) catch "?",
|
|
target_ty.displayName(self.allocator) catch "?",
|
|
});
|
|
}
|
|
|
|
return self.convertValue(val, src_ty, target_ty);
|
|
}
|
|
|
|
/// Convert an LLVM value from src_ty to target_ty, emitting appropriate casts.
|
|
fn convertValue(self: *CodeGen, val: c.LLVMValueRef, src_ty: Type, target_ty: Type) c.LLVMValueRef {
|
|
// Same type → return as-is
|
|
if (src_ty.eql(target_ty)) return val;
|
|
|
|
// string <-> []u8: identical LLVM type {ptr, i64}, no conversion needed
|
|
if ((src_ty == .string_type and target_ty.isSlice() and
|
|
std.mem.eql(u8, target_ty.slice_type.element_name, "u8")) or
|
|
(src_ty.isSlice() and std.mem.eql(u8, src_ty.slice_type.element_name, "u8") and
|
|
target_ty == .string_type))
|
|
return val;
|
|
|
|
const target_llvm = self.typeToLLVM(target_ty);
|
|
|
|
// Any → concrete type: extract the i64 value and convert
|
|
if (src_ty.isAny()) {
|
|
const i64_val = self.extractValue(val, 1, "any_extract");
|
|
if (target_ty.isInt()) {
|
|
if (target_ty.bitWidth() < 64) {
|
|
return self.trunc(i64_val, target_llvm, "any_to_int");
|
|
}
|
|
return i64_val;
|
|
}
|
|
if (target_ty == .boolean) {
|
|
return self.trunc(i64_val, self.i1Type(), "any_to_bool");
|
|
}
|
|
if (target_ty == .f64) {
|
|
return self.bitCast(i64_val, self.f64Type(), "any_to_f64");
|
|
}
|
|
if (target_ty == .f32) {
|
|
const as_f64 = self.bitCast(i64_val, self.f64Type(), "any_f64_tmp");
|
|
return c.LLVMBuildFPTrunc(self.builder, as_f64, self.f32Type(), "any_to_f32");
|
|
}
|
|
if (target_ty == .string_type) {
|
|
// i64 is a pointer to {ptr, i32} on the stack
|
|
return self.loadFromI64Ptr(i64_val, self.getStringStructType(), "any_to_str");
|
|
}
|
|
if (target_ty.isStruct()) {
|
|
const sname = target_ty.struct_type;
|
|
if (self.lookupStructInfo(sname)) |info| {
|
|
return self.loadFromI64Ptr(i64_val, info.llvm_type, "any_to_struct");
|
|
}
|
|
}
|
|
if (target_ty.isEnum()) {
|
|
const enum_llvm_ty = self.getEnumLLVMType(target_ty.enum_type);
|
|
const enum_bits = c.LLVMGetIntTypeWidth(enum_llvm_ty);
|
|
if (enum_bits < 64) return self.trunc(i64_val, enum_llvm_ty, "any_to_enum");
|
|
return i64_val;
|
|
}
|
|
if (target_ty.isUnion()) {
|
|
const uname = target_ty.union_type;
|
|
if (self.lookupTaggedEnumInfo(uname)) |info| {
|
|
return self.loadFromI64Ptr(i64_val, info.llvm_type, "any_to_union");
|
|
}
|
|
}
|
|
if (target_ty.isPointer() or target_ty.isManyPointer()) {
|
|
return self.intToPtr(i64_val, "any_to_ptr");
|
|
}
|
|
return i64_val;
|
|
}
|
|
|
|
// Float → float conversions
|
|
if (src_ty.isFloat() and target_ty.isFloat()) {
|
|
if (target_ty.bitWidth() > src_ty.bitWidth()) {
|
|
return c.LLVMBuildFPExt(self.builder, val, target_llvm, "fext");
|
|
} else {
|
|
return c.LLVMBuildFPTrunc(self.builder, val, target_llvm, "ftrunc");
|
|
}
|
|
}
|
|
|
|
// Int → float conversions
|
|
if (src_ty.isInt() and target_ty.isFloat()) {
|
|
if (src_ty.isSigned()) {
|
|
return c.LLVMBuildSIToFP(self.builder, val, target_llvm, "sitofp");
|
|
} else {
|
|
return c.LLVMBuildUIToFP(self.builder, val, target_llvm, "uitofp");
|
|
}
|
|
}
|
|
|
|
// Float → int conversions
|
|
if (src_ty.isFloat() and target_ty.isInt()) {
|
|
if (target_ty.isSigned()) {
|
|
return c.LLVMBuildFPToSI(self.builder, val, target_llvm, "fptosi");
|
|
} else {
|
|
return c.LLVMBuildFPToUI(self.builder, val, target_llvm, "fptoui");
|
|
}
|
|
}
|
|
|
|
// Pointer → int: PtrToInt
|
|
if ((src_ty.isPointer() or src_ty.isManyPointer()) and target_ty.isInt()) {
|
|
const as_i64 = self.ptrToInt(val, "ptrtoint");
|
|
if (target_ty.bitWidth() < 64) {
|
|
return self.trunc(as_i64, target_llvm, "ptr_trunc");
|
|
}
|
|
return as_i64;
|
|
}
|
|
|
|
// Union → int: extract the tag field (index 0)
|
|
if (src_ty.isUnion() and target_ty.isInt()) {
|
|
const uname = src_ty.union_type;
|
|
if (self.lookupTaggedEnumInfo(uname)) |info| {
|
|
const tag_llvm_ty = self.getEnumLLVMType(uname);
|
|
const tag_bits = c.LLVMGetIntTypeWidth(tag_llvm_ty);
|
|
const tmp = self.buildEntryBlockAlloca(info.llvm_type, "union_cast");
|
|
_ = c.LLVMBuildStore(self.builder, val, tmp);
|
|
const tag_val = self.loadStructField(info.llvm_type, tmp, 0, tag_llvm_ty);
|
|
if (target_ty.bitWidth() == tag_bits) return tag_val;
|
|
if (target_ty.bitWidth() > tag_bits) return self.sExt(tag_val, target_llvm, "tag_ext");
|
|
return self.trunc(tag_val, target_llvm, "tag_trunc");
|
|
}
|
|
}
|
|
|
|
// Int → int conversions
|
|
if (src_ty.isInt() and target_ty.isInt()) {
|
|
const sw = src_ty.bitWidth();
|
|
const tw = target_ty.bitWidth();
|
|
if (tw > sw) {
|
|
// Extend — use SExt if source is signed, ZExt if unsigned
|
|
if (src_ty.isSigned()) {
|
|
return self.sExt(val, target_llvm, "sext");
|
|
} else {
|
|
return self.zExt(val, target_llvm, "zext");
|
|
}
|
|
} else if (tw < sw) {
|
|
// Truncate
|
|
return self.trunc(val, target_llvm, "trunc");
|
|
}
|
|
// Same width, different signedness — no-op (bit pattern is the same)
|
|
return val;
|
|
}
|
|
|
|
// Int → pointer/function_type: IntToPtr (for xx cast from integer to pointer)
|
|
if (src_ty.isInt() and (target_ty.isPointer() or target_ty.isManyPointer() or target_ty.isFunctionType())) {
|
|
return self.intToPtr(val, "inttoptr");
|
|
}
|
|
|
|
// Slice/string → pointer: extract .ptr from fat pointer
|
|
if ((src_ty.isSlice() or src_ty == .string_type) and (target_ty.isPointer() or target_ty.isManyPointer())) {
|
|
return self.extractValue(val, 0, "slice_to_ptr");
|
|
}
|
|
|
|
// Enum → int: extend or truncate from backing type to target int
|
|
if (src_ty.isEnum() and target_ty.isInt()) {
|
|
const enum_bits = c.LLVMGetIntTypeWidth(self.getEnumLLVMType(src_ty.enum_type));
|
|
const target_bits = target_ty.bitWidth();
|
|
if (target_bits > enum_bits) {
|
|
return self.zExt(val, target_llvm, "enum_to_int");
|
|
} else if (target_bits < enum_bits) {
|
|
return self.trunc(val, target_llvm, "enum_to_int");
|
|
}
|
|
return val;
|
|
}
|
|
|
|
// Int → enum: extend or truncate from source int to backing type
|
|
if (src_ty.isInt() and target_ty.isEnum()) {
|
|
const enum_llvm_ty = self.getEnumLLVMType(target_ty.enum_type);
|
|
const enum_bits = c.LLVMGetIntTypeWidth(enum_llvm_ty);
|
|
const src_bits = src_ty.bitWidth();
|
|
if (enum_bits > src_bits) {
|
|
return self.zExt(val, enum_llvm_ty, "int_to_enum");
|
|
} else if (enum_bits < src_bits) {
|
|
return self.trunc(val, enum_llvm_ty, "int_to_enum");
|
|
}
|
|
return val;
|
|
}
|
|
|
|
// *[N]T → [*]T: pointer to array decays to many-pointer (both opaque ptrs, no-op)
|
|
if (src_ty.isPointer() and target_ty.isManyPointer()) {
|
|
return val;
|
|
}
|
|
|
|
// Pointer → function_type or function_type → pointer: both are opaque pointers, no-op
|
|
if ((src_ty.isPointer() or src_ty.isManyPointer()) and target_ty.isFunctionType()) {
|
|
return val;
|
|
}
|
|
if (src_ty.isFunctionType() and (target_ty.isPointer() or target_ty.isManyPointer())) {
|
|
return val;
|
|
}
|
|
|
|
return val;
|
|
}
|
|
|
|
fn findNameIndex(_: *CodeGen, names: []const []const u8, name: []const u8) ?usize {
|
|
for (names, 0..) |n, i| {
|
|
if (std.mem.eql(u8, n, name)) return i;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
fn findFieldIndex(self: *CodeGen, field_names: []const []const u8, field: []const u8, struct_name: []const u8) !usize {
|
|
return self.findNameIndex(field_names, field) orelse
|
|
return self.emitErrorFmt("no field '{s}' in struct '{s}'", .{ field, struct_name });
|
|
}
|
|
|
|
fn componentToIndex(ch: u8) ?u32 {
|
|
return switch (ch) {
|
|
'x', 'r', 'u' => 0,
|
|
'y', 'g', 'v' => 1,
|
|
'z', 'b' => 2,
|
|
'w', 'a' => 3,
|
|
else => null,
|
|
};
|
|
}
|
|
|
|
fn genMathIntrinsic(self: *CodeGen, call_node: ast.Call, comptime name: []const u8) !c.LLVMValueRef {
|
|
if (call_node.args.len != 1) return self.emitError(name ++ " expects exactly 1 argument");
|
|
const arg_val = try self.genExpr(call_node.args[0]);
|
|
const arg_ty = self.inferType(call_node.args[0]);
|
|
|
|
const is_f64 = std.meta.eql(arg_ty, Type.f64);
|
|
const intrinsic_name: [*c]const u8 = if (is_f64) "llvm." ++ name ++ ".f64" else "llvm." ++ name ++ ".f32";
|
|
const llvm_float_ty = if (is_f64) self.f64Type() else self.f32Type();
|
|
|
|
var intrinsic_fn = c.LLVMGetNamedFunction(self.module, intrinsic_name);
|
|
if (intrinsic_fn == null) {
|
|
var param_types = [_]c.LLVMTypeRef{llvm_float_ty};
|
|
const fn_type = c.LLVMFunctionType(llvm_float_ty, ¶m_types, 1, 0);
|
|
intrinsic_fn = c.LLVMAddFunction(self.module, intrinsic_name, fn_type);
|
|
}
|
|
|
|
var args = [_]c.LLVMValueRef{arg_val};
|
|
return c.LLVMBuildCall2(self.builder, c.LLVMGlobalGetValueType(intrinsic_fn.?), intrinsic_fn.?, &args, 1, name.ptr);
|
|
}
|
|
|
|
fn genSizeOf(self: *CodeGen, call_node: ast.Call) !c.LLVMValueRef {
|
|
if (call_node.args.len != 1) return self.emitError("size_of expects exactly 1 argument");
|
|
const arg = call_node.args[0];
|
|
const ty = self.resolveType(arg);
|
|
if (std.meta.eql(ty, Type.void_type)) {
|
|
// Uninstantiated generic type function → size of a function pointer
|
|
if (arg.data == .identifier or arg.data == .type_expr) {
|
|
const name = if (arg.data == .identifier) arg.data.identifier.name else arg.data.type_expr.name;
|
|
if (self.generic_templates.contains(name) or self.generic_struct_templates.contains(name)) {
|
|
return self.constInt64(self.getTypeSize(self.ptrType()));
|
|
}
|
|
}
|
|
return self.constInt64(0);
|
|
}
|
|
const llvm_ty = self.typeToLLVM(ty);
|
|
const size = self.getTypeSize(llvm_ty);
|
|
return self.constInt64(size);
|
|
}
|
|
|
|
fn genTypeOf(self: *CodeGen, call_node: ast.Call) !c.LLVMValueRef {
|
|
if (call_node.args.len != 1) return self.emitError("type_of expects exactly 1 argument");
|
|
const arg = call_node.args[0];
|
|
const arg_ty = self.inferType(arg);
|
|
const i64_ty = self.i64Type();
|
|
|
|
// For Any values: extract the runtime tag (field 0)
|
|
if (arg_ty.isAny()) {
|
|
const val = try self.genExpr(arg);
|
|
return self.extractValue(val, 0, "type_of");
|
|
}
|
|
|
|
// For known types: return the constant tag value
|
|
const tag: u64 = switch (arg_ty) {
|
|
.void_type => ANY_TAG_VOID,
|
|
.boolean => ANY_TAG_BOOL,
|
|
.signed => |w| if (w <= 32) ANY_TAG_S32 else ANY_TAG_S64,
|
|
.unsigned => |w| if (w <= 32) ANY_TAG_S32 else ANY_TAG_S64,
|
|
.f32 => ANY_TAG_F32,
|
|
.f64 => ANY_TAG_F64,
|
|
.string_type => ANY_TAG_STRING,
|
|
.struct_type => |name| try self.getAnyTypeId(name, arg_ty),
|
|
.enum_type => |name| try self.getAnyTypeId(name, arg_ty),
|
|
.union_type => |name| try self.getAnyTypeId(name, arg_ty),
|
|
.meta_type => ANY_TAG_TYPE,
|
|
else => ANY_TAG_S32,
|
|
};
|
|
return c.LLVMConstInt(i64_ty, tag, 0);
|
|
}
|
|
|
|
fn genTypeName(self: *CodeGen, call_node: ast.Call) !c.LLVMValueRef {
|
|
if (call_node.args.len != 1) return self.emitError("type_name expects exactly 1 argument");
|
|
const ty = self.resolveType(call_node.args[0]);
|
|
const name = try ty.displayName(self.allocator);
|
|
return self.buildConstStr(name);
|
|
}
|
|
|
|
fn genFieldCount(self: *CodeGen, call_node: ast.Call) !c.LLVMValueRef {
|
|
if (call_node.args.len != 1) return self.emitError("field_count expects exactly 1 argument");
|
|
const ty = self.resolveType(call_node.args[0]);
|
|
const i64_ty = self.i64Type();
|
|
if (ty.isStruct()) {
|
|
const info = try self.getStructInfo(ty.struct_type);
|
|
return c.LLVMConstInt(i64_ty, info.field_names.len, 0);
|
|
}
|
|
if (ty.isEnum()) {
|
|
const variants = self.lookupEnumVariants(ty.enum_type) orelse
|
|
return self.emitErrorFmt("unknown enum type '{s}'", .{ty.enum_type});
|
|
return c.LLVMConstInt(i64_ty, variants.len, 0);
|
|
}
|
|
if (ty.isVector()) {
|
|
return c.LLVMConstInt(i64_ty, ty.vector_type.length, 0);
|
|
}
|
|
if (ty.isUnion()) {
|
|
const info = try self.getTaggedEnumInfo(ty.union_type);
|
|
return c.LLVMConstInt(i64_ty, info.variant_names.len, 0);
|
|
}
|
|
if (ty.isArray()) {
|
|
return c.LLVMConstInt(i64_ty, ty.array_type.length, 0);
|
|
}
|
|
return self.emitError("field_count requires a struct, enum, vector, or array type");
|
|
}
|
|
|
|
fn genFieldName(self: *CodeGen, call_node: ast.Call) !c.LLVMValueRef {
|
|
if (call_node.args.len != 2) return self.emitError("field_name expects 2 arguments: field_name(T, idx)");
|
|
const ty = self.resolveType(call_node.args[0]);
|
|
|
|
// Get the name list and type key
|
|
const names: []const []const u8, const type_key: []const u8 = if (ty.isStruct()) blk: {
|
|
const info = try self.getStructInfo(ty.struct_type);
|
|
break :blk .{ info.field_names, ty.struct_type };
|
|
} else if (ty.isEnum()) blk: {
|
|
const variants = self.lookupEnumVariants(ty.enum_type) orelse
|
|
return self.emitErrorFmt("unknown enum type '{s}'", .{ty.enum_type});
|
|
break :blk .{ variants, ty.enum_type };
|
|
} else if (ty.isUnion()) blk: {
|
|
const info = try self.getTaggedEnumInfo(ty.union_type);
|
|
break :blk .{ info.variant_names, ty.union_type };
|
|
} else return self.emitError("field_name requires a struct or enum type");
|
|
|
|
// Build a global array of string slices
|
|
const n = names.len;
|
|
const str_ty = self.getStringStructType();
|
|
const arr_ty = c.LLVMArrayType2(str_ty, n);
|
|
|
|
const vals = try self.allocator.alloc(c.LLVMValueRef, n);
|
|
for (names, 0..) |name, i| {
|
|
vals[i] = self.buildConstStrGlobal(name);
|
|
}
|
|
const arr_init = c.LLVMConstArray2(str_ty, vals.ptr, @intCast(n));
|
|
const global_name = try self.allocator.dupeZ(u8, try std.fmt.allocPrint(self.allocator, "field_names.{s}", .{type_key}));
|
|
var global = c.LLVMGetNamedGlobal(self.module, global_name.ptr);
|
|
if (global == null) {
|
|
global = c.LLVMAddGlobal(self.module, arr_ty, global_name.ptr);
|
|
c.LLVMSetInitializer(global, arr_init);
|
|
c.LLVMSetGlobalConstant(global, 1);
|
|
c.LLVMSetLinkage(global, c.LLVMPrivateLinkage);
|
|
}
|
|
|
|
// GEP into the array with runtime index
|
|
const idx = try self.genExpr(call_node.args[1]);
|
|
const elem_ptr = self.gepArrayElement(arr_ty, global, idx, "field_name_ptr");
|
|
return c.LLVMBuildLoad2(self.builder, str_ty, elem_ptr, "field_name");
|
|
}
|
|
|
|
fn genFieldValue(self: *CodeGen, call_node: ast.Call) !c.LLVMValueRef {
|
|
if (call_node.args.len != 2) return self.emitError("field_value expects 2 arguments: field_value(s, idx)");
|
|
|
|
const val = try self.genExpr(call_node.args[0]);
|
|
const val_ty = self.inferType(call_node.args[0]);
|
|
|
|
// Vector: extractelement + box as Any
|
|
if (val_ty.isVector()) {
|
|
const info = val_ty.vector_type;
|
|
const elem_ty = Type.fromName(info.element_name) orelse
|
|
return self.emitErrorFmt("unknown vector element type '{s}'", .{info.element_name});
|
|
const idx = try self.genExpr(call_node.args[1]);
|
|
const elem = c.LLVMBuildExtractElement(self.builder, val, idx, "vec_elem");
|
|
return self.buildAnyValue(elem, elem_ty);
|
|
}
|
|
|
|
// Payload-less enum: return void Any (no payload to extract)
|
|
if (val_ty.isEnum() and !val_ty.isUnion()) {
|
|
return self.buildAnyValue(self.constInt64(0), .void_type);
|
|
}
|
|
|
|
// Tagged enum (with payloads): switch over tag, extract payload with correct type
|
|
if (val_ty.isUnion()) {
|
|
const uinfo = try self.getTaggedEnumInfo(val_ty.union_type);
|
|
|
|
const union_alloca = self.buildEntryBlockAlloca(uinfo.llvm_type, "fv_union");
|
|
_ = c.LLVMBuildStore(self.builder, val, union_alloca);
|
|
|
|
// Read tag (field 0)
|
|
const tag_val = self.loadStructField(uinfo.llvm_type, union_alloca, 0, self.getEnumLLVMType(val_ty.union_type));
|
|
const payload_ptr = self.structGEP(uinfo.llvm_type, union_alloca, uinfo.payload_field_index, "fv_payload_ptr");
|
|
|
|
const n = uinfo.variant_names.len;
|
|
const any_ty = self.getAnyStructType();
|
|
const sb = self.buildSwitch(tag_val, @intCast(n), "fv_merge", "fv_default");
|
|
|
|
var phi_vals = std.ArrayList(c.LLVMValueRef).empty;
|
|
var phi_bbs = std.ArrayList(c.LLVMBasicBlockRef).empty;
|
|
|
|
const tag_llvm_ty = self.getEnumLLVMType(val_ty.union_type);
|
|
for (uinfo.variant_types, 0..) |vty, vi| {
|
|
const case_bb = self.appendBB("fv_ucase");
|
|
c.LLVMAddCase(sb.sw, c.LLVMConstInt(tag_llvm_ty, @intCast(vi), 0), case_bb);
|
|
self.positionAt(case_bb);
|
|
|
|
const any_val = if (vty == .void_type) blk: {
|
|
// Void variant: return Any with void tag
|
|
const undef = self.getUndef(any_ty);
|
|
const void_tag = self.constInt64(ANY_TAG_VOID);
|
|
const with_tag = self.insertValue(undef, void_tag, 0, "void_tag");
|
|
const zero_val = self.constInt64(0);
|
|
break :blk self.insertValue(with_tag, zero_val, 1, "void_any");
|
|
} else blk: {
|
|
const payload = self.loadTyped(vty, payload_ptr, "fv_payload");
|
|
break :blk try self.buildAnyValue(payload, vty);
|
|
};
|
|
try self.addPhiCase(&phi_vals, &phi_bbs, any_val, sb.merge_bb);
|
|
}
|
|
|
|
// Default: undef
|
|
self.positionAt(sb.default_bb);
|
|
try self.addPhiCase(&phi_vals, &phi_bbs, self.getUndef(any_ty), sb.merge_bb);
|
|
|
|
self.positionAt(sb.merge_bb);
|
|
const phi = try self.buildPhiNode(&phi_vals, &phi_bbs, any_ty, "fv_uresult");
|
|
return phi;
|
|
}
|
|
|
|
// Slice: extract ptr, GEP to element, load, box as Any
|
|
if (val_ty.isSlice()) {
|
|
const sinfo = val_ty.slice_type;
|
|
const elem_ty = Type.fromName(sinfo.element_name) orelse
|
|
return self.emitErrorFmt("unknown slice element type '{s}'", .{sinfo.element_name});
|
|
const elem_llvm_ty = self.typeToLLVM(elem_ty);
|
|
// val is {ptr, i32} — extract ptr
|
|
const data_ptr = self.extractValue(val, 0, "fv_sdata");
|
|
const idx = try self.genExpr(call_node.args[1]);
|
|
const elem_ptr = self.gepPointerElement(elem_llvm_ty, data_ptr, idx, "fv_selem");
|
|
const elem = c.LLVMBuildLoad2(self.builder, elem_llvm_ty, elem_ptr, "fv_seval");
|
|
return self.buildAnyValue(elem, elem_ty);
|
|
}
|
|
|
|
// Array: GEP + load + box as Any
|
|
if (val_ty.isArray()) {
|
|
const ainfo = val_ty.array_type;
|
|
const elem_ty = Type.fromName(ainfo.element_name) orelse
|
|
return self.emitErrorFmt("unknown array element type '{s}'", .{ainfo.element_name});
|
|
const arr_llvm_ty = self.typeToLLVM(val_ty);
|
|
const elem_llvm_ty = self.typeToLLVM(elem_ty);
|
|
const arr_alloca = self.buildEntryBlockAlloca(arr_llvm_ty, "fv_arr");
|
|
_ = c.LLVMBuildStore(self.builder, val, arr_alloca);
|
|
const idx = try self.genExpr(call_node.args[1]);
|
|
var gep_indices = [_]c.LLVMValueRef{
|
|
self.constInt32(0),
|
|
idx,
|
|
};
|
|
const elem_ptr = c.LLVMBuildGEP2(self.builder, arr_llvm_ty, arr_alloca, &gep_indices, 2, "fv_aelem");
|
|
const elem = c.LLVMBuildLoad2(self.builder, elem_llvm_ty, elem_ptr, "fv_aeval");
|
|
return self.buildAnyValue(elem, elem_ty);
|
|
}
|
|
|
|
// Struct: switch over field indices
|
|
const struct_val = val;
|
|
const struct_ty = val_ty;
|
|
if (!struct_ty.isStruct()) return self.emitError("field_value requires a struct, vector, enum, or array value");
|
|
|
|
const info = try self.getStructInfo(struct_ty.struct_type);
|
|
|
|
const idx = try self.genExpr(call_node.args[1]);
|
|
const n = info.field_names.len;
|
|
|
|
// Store struct to alloca BEFORE the switch (switch is a terminator)
|
|
const struct_alloca = self.buildEntryBlockAlloca(info.llvm_type, "fv_struct");
|
|
_ = c.LLVMBuildStore(self.builder, struct_val, struct_alloca);
|
|
|
|
// Generate switch on idx with N cases
|
|
const sb = self.buildSwitch(idx, @intCast(n), "fv_merge", "fv_default");
|
|
|
|
const any_ty = self.getAnyStructType();
|
|
var phi_vals = std.ArrayList(c.LLVMValueRef).empty;
|
|
var phi_bbs = std.ArrayList(c.LLVMBasicBlockRef).empty;
|
|
|
|
for (0..n) |i| {
|
|
const case_bb = self.appendBB("fv_case");
|
|
const case_val = self.constInt64(i);
|
|
c.LLVMAddCase(sb.sw, case_val, case_bb);
|
|
|
|
self.positionAt(case_bb);
|
|
// Extract field i via GEP + load
|
|
const field_ptr = self.structGEP(info.llvm_type, struct_alloca, @intCast(i), "fv_field_ptr");
|
|
const field_llvm_ty = c.LLVMStructGetTypeAtIndex(info.llvm_type, @intCast(i));
|
|
const field_val = c.LLVMBuildLoad2(self.builder, field_llvm_ty, field_ptr, "fv_field");
|
|
const any_val = try self.buildAnyValue(field_val, info.field_types[i]);
|
|
try self.addPhiCase(&phi_vals, &phi_bbs, any_val, sb.merge_bb);
|
|
}
|
|
|
|
// Default: return undef Any
|
|
self.positionAt(sb.default_bb);
|
|
try self.addPhiCase(&phi_vals, &phi_bbs, self.getUndef(any_ty), sb.merge_bb);
|
|
|
|
self.positionAt(sb.merge_bb);
|
|
const phi = try self.buildPhiNode(&phi_vals, &phi_bbs, any_ty, "fv_result");
|
|
return phi;
|
|
}
|
|
|
|
fn genIsFlags(self: *CodeGen, call_node: ast.Call) !c.LLVMValueRef {
|
|
if (call_node.args.len != 1) return self.emitError("is_flags expects exactly 1 argument");
|
|
const ty = self.resolveType(call_node.args[0]);
|
|
const i1_type = self.i1Type();
|
|
if (ty.isEnum()) {
|
|
const is_flags = self.flags_enum_types.contains(ty.enum_type);
|
|
return c.LLVMConstInt(i1_type, @intFromBool(is_flags), 0);
|
|
}
|
|
return c.LLVMConstInt(i1_type, 0, 0);
|
|
}
|
|
|
|
fn genFieldValueInt(self: *CodeGen, call_node: ast.Call) !c.LLVMValueRef {
|
|
if (call_node.args.len != 2) return self.emitError("field_value_int expects 2 arguments: field_value_int(T, idx)");
|
|
const ty = self.resolveType(call_node.args[0]);
|
|
const i64_type = self.i64Type();
|
|
// For non-enum types (e.g. tagged enums compiled via dead code), return the index as value
|
|
if (!ty.isEnum()) {
|
|
return try self.genExpr(call_node.args[1]);
|
|
}
|
|
const enum_name = ty.enum_type;
|
|
const values = self.enum_variant_values.get(enum_name);
|
|
const variants = self.lookupEnumVariants(enum_name) orelse return try self.genExpr(call_node.args[1]);
|
|
const n = variants.len;
|
|
|
|
const idx = try self.genExpr(call_node.args[1]);
|
|
const sb = self.buildSwitch(idx, @intCast(n), "fvi_merge", "fvi_default");
|
|
|
|
var phi_vals = std.ArrayList(c.LLVMValueRef).empty;
|
|
var phi_bbs = std.ArrayList(c.LLVMBasicBlockRef).empty;
|
|
|
|
for (0..n) |i| {
|
|
const case_bb = self.appendBB("fvi_case");
|
|
c.LLVMAddCase(sb.sw, c.LLVMConstInt(i64_type, i, 0), case_bb);
|
|
self.positionAt(case_bb);
|
|
const val: u64 = if (values) |vals| @bitCast(vals[i]) else i;
|
|
try self.addPhiCase(&phi_vals, &phi_bbs, c.LLVMConstInt(i64_type, val, 0), sb.merge_bb);
|
|
}
|
|
|
|
self.positionAt(sb.default_bb);
|
|
try self.addPhiCase(&phi_vals, &phi_bbs, c.LLVMConstInt(i64_type, 0, 0), sb.merge_bb);
|
|
|
|
self.positionAt(sb.merge_bb);
|
|
const phi = try self.buildPhiNode(&phi_vals, &phi_bbs, i64_type, "fvi_result");
|
|
return phi;
|
|
}
|
|
|
|
fn genFieldIndex(self: *CodeGen, call_node: ast.Call) !c.LLVMValueRef {
|
|
if (call_node.args.len != 2) return self.emitError("field_index expects 2 arguments: field_index(T, value)");
|
|
const ty = self.resolveType(call_node.args[0]);
|
|
const i64_type = self.i64Type();
|
|
|
|
// Handle tagged enums (union_type) — extract tag from field 0
|
|
if (ty == .union_type) {
|
|
const union_name = ty.union_type;
|
|
const info = self.lookupTaggedEnumInfo(union_name) orelse {
|
|
_ = try self.genExpr(call_node.args[1]);
|
|
return c.LLVMConstInt(i64_type, 0, 0);
|
|
};
|
|
const values = self.enum_variant_values.get(union_name);
|
|
const n = info.variant_names.len;
|
|
|
|
const val = try self.genExpr(call_node.args[1]);
|
|
// Extract tag from field 0 of the { tag, payload } struct
|
|
const tag_val = self.extractValue(val, 0, "fi_tag");
|
|
const enum_llvm_ty = self.getEnumLLVMType(union_name);
|
|
const sw_val = if (c.LLVMTypeOf(tag_val) != enum_llvm_ty)
|
|
c.LLVMBuildIntCast2(self.builder, tag_val, enum_llvm_ty, 0, "fi_cast")
|
|
else
|
|
tag_val;
|
|
|
|
const sb = self.buildSwitch(sw_val, @intCast(n), "fi_merge", "fi_default");
|
|
|
|
var phi_vals = std.ArrayList(c.LLVMValueRef).empty;
|
|
var phi_bbs = std.ArrayList(c.LLVMBasicBlockRef).empty;
|
|
var seen_values = std.ArrayList(u64).empty;
|
|
|
|
for (0..n) |i| {
|
|
const explicit_val: u64 = if (values) |vals| @bitCast(vals[i]) else i;
|
|
var is_dup = false;
|
|
for (seen_values.items) |sv| {
|
|
if (sv == explicit_val) {
|
|
is_dup = true;
|
|
break;
|
|
}
|
|
}
|
|
if (is_dup) continue;
|
|
try seen_values.append(self.allocator, explicit_val);
|
|
const case_bb = self.appendBB("fi_case");
|
|
c.LLVMAddCase(sb.sw, c.LLVMConstInt(enum_llvm_ty, explicit_val, 0), case_bb);
|
|
self.positionAt(case_bb);
|
|
try self.addPhiCase(&phi_vals, &phi_bbs, c.LLVMConstInt(i64_type, i, 0), sb.merge_bb);
|
|
}
|
|
|
|
self.positionAt(sb.default_bb);
|
|
const neg_one = c.LLVMConstInt(i64_type, @bitCast(@as(i64, -1)), 0);
|
|
try self.addPhiCase(&phi_vals, &phi_bbs, neg_one, sb.merge_bb);
|
|
|
|
self.positionAt(sb.merge_bb);
|
|
return try self.buildPhiNode(&phi_vals, &phi_bbs, i64_type, "fi_result");
|
|
}
|
|
|
|
if (!ty.isEnum()) {
|
|
_ = try self.genExpr(call_node.args[1]);
|
|
return c.LLVMConstInt(i64_type, 0, 0);
|
|
}
|
|
const enum_name = ty.enum_type;
|
|
// Flags enums don't use sequential indices
|
|
if (self.flags_enum_types.contains(enum_name)) {
|
|
_ = try self.genExpr(call_node.args[1]);
|
|
return c.LLVMConstInt(i64_type, 0, 0);
|
|
}
|
|
const values = self.enum_variant_values.get(enum_name);
|
|
const variants = self.lookupEnumVariants(enum_name) orelse return try self.genExpr(call_node.args[1]);
|
|
const n = variants.len;
|
|
|
|
const val = try self.genExpr(call_node.args[1]);
|
|
// Ensure the switch value uses the enum's backing type
|
|
const enum_llvm_ty = self.getEnumLLVMType(enum_name);
|
|
const sw_val = if (c.LLVMTypeOf(val) != enum_llvm_ty)
|
|
c.LLVMBuildIntCast2(self.builder, val, enum_llvm_ty, 0, "fi_cast")
|
|
else
|
|
val;
|
|
|
|
const sb = self.buildSwitch(sw_val, @intCast(n), "fi_merge", "fi_default");
|
|
|
|
var phi_vals = std.ArrayList(c.LLVMValueRef).empty;
|
|
var phi_bbs = std.ArrayList(c.LLVMBasicBlockRef).empty;
|
|
var seen_values = std.ArrayList(u64).empty;
|
|
|
|
for (0..n) |i| {
|
|
const explicit_val: u64 = if (values) |vals| @bitCast(vals[i]) else i;
|
|
// Skip duplicate values (first one wins)
|
|
var is_dup = false;
|
|
for (seen_values.items) |sv| {
|
|
if (sv == explicit_val) { is_dup = true; break; }
|
|
}
|
|
if (is_dup) continue;
|
|
try seen_values.append(self.allocator, explicit_val);
|
|
const case_bb = self.appendBB("fi_case");
|
|
c.LLVMAddCase(sb.sw, c.LLVMConstInt(enum_llvm_ty, explicit_val, 0), case_bb);
|
|
self.positionAt(case_bb);
|
|
try self.addPhiCase(&phi_vals, &phi_bbs, c.LLVMConstInt(i64_type, i, 0), sb.merge_bb);
|
|
}
|
|
|
|
self.positionAt(sb.default_bb);
|
|
const neg_one = c.LLVMConstInt(i64_type, @bitCast(@as(i64, -1)), 0);
|
|
try self.addPhiCase(&phi_vals, &phi_bbs, neg_one, sb.merge_bb);
|
|
|
|
self.positionAt(sb.merge_bb);
|
|
const phi = try self.buildPhiNode(&phi_vals, &phi_bbs, i64_type, "fi_result");
|
|
return phi;
|
|
}
|
|
|
|
fn genCast(self: *CodeGen, call_node: ast.Call) !c.LLVMValueRef {
|
|
if (call_node.args.len != 2) return self.emitError("cast expects: cast(Type) expr");
|
|
const target_ty = self.resolveType(call_node.args[0]);
|
|
const src_ty = self.inferType(call_node.args[1]);
|
|
const val = try self.genExpr(call_node.args[1]);
|
|
return self.convertValue(val, src_ty, target_ty);
|
|
}
|
|
|
|
fn genAlloc(self: *CodeGen, args: []const *Node) !c.LLVMValueRef {
|
|
if (args.len != 1) return self.emitError("alloc expects exactly 1 argument: alloc(size)");
|
|
const builtins = try self.requireBuiltins();
|
|
const size_val = try self.genExpr(args[0]);
|
|
const i64_type = self.i64Type();
|
|
// calloc(size + 1, 1) — extra byte for null terminator
|
|
const one_i64 = c.LLVMConstInt(i64_type, 1, 0);
|
|
const size_plus_one = c.LLVMBuildAdd(self.builder, size_val, one_i64, "szp1");
|
|
const calloc_fn = builtins.calloc_fn;
|
|
const calloc_ty = c.LLVMGlobalGetValueType(calloc_fn);
|
|
var calloc_args = [_]c.LLVMValueRef{ size_plus_one, one_i64 };
|
|
const ptr = c.LLVMBuildCall2(self.builder, calloc_ty, calloc_fn, &calloc_args, 2, "alloc_ptr");
|
|
// Build string slice: {ptr, size}
|
|
return self.buildStringSlice(ptr, size_val);
|
|
}
|
|
|
|
fn genMalloc(self: *CodeGen, args: []const *Node) !c.LLVMValueRef {
|
|
if (args.len != 1) return self.emitError("malloc expects exactly 1 argument: malloc(size)");
|
|
const builtins = try self.requireBuiltins();
|
|
const size_val = try self.genExpr(args[0]);
|
|
const fn_ty = c.LLVMGlobalGetValueType(builtins.malloc_fn);
|
|
var call_args = [_]c.LLVMValueRef{size_val};
|
|
return c.LLVMBuildCall2(self.builder, fn_ty, builtins.malloc_fn, &call_args, 1, "malloc_ptr");
|
|
}
|
|
|
|
fn genFree(self: *CodeGen, args: []const *Node) !c.LLVMValueRef {
|
|
if (args.len != 1) return self.emitError("free expects exactly 1 argument: free(ptr)");
|
|
const builtins = try self.requireBuiltins();
|
|
const ptr_val = try self.genExpr(args[0]);
|
|
const fn_ty = c.LLVMGlobalGetValueType(builtins.free_fn);
|
|
var call_args = [_]c.LLVMValueRef{ptr_val};
|
|
_ = c.LLVMBuildCall2(self.builder, fn_ty, builtins.free_fn, &call_args, 1, "");
|
|
return null;
|
|
}
|
|
|
|
fn genMemcpy(self: *CodeGen, args: []const *Node) !c.LLVMValueRef {
|
|
if (args.len != 3) return self.emitError("memcpy expects 3 arguments: memcpy(dst, src, size)");
|
|
const builtins = try self.requireBuiltins();
|
|
const dst = try self.genExpr(args[0]);
|
|
const src = try self.genExpr(args[1]);
|
|
const size_val = try self.genExpr(args[2]);
|
|
const fn_ty = c.LLVMGlobalGetValueType(builtins.memcpy_fn);
|
|
var call_args = [_]c.LLVMValueRef{ dst, src, size_val };
|
|
_ = c.LLVMBuildCall2(self.builder, fn_ty, builtins.memcpy_fn, &call_args, 3, "");
|
|
return null;
|
|
}
|
|
|
|
fn genVectorExtract(self: *CodeGen, vec_val: c.LLVMValueRef, field: []const u8) !c.LLVMValueRef {
|
|
if (field.len == 1) {
|
|
const idx_val = componentToIndex(field[0]) orelse return self.emitErrorFmt("invalid vector component '{c}'", .{field[0]});
|
|
const idx = self.constInt32(idx_val);
|
|
return c.LLVMBuildExtractElement(self.builder, vec_val, idx, "comp");
|
|
}
|
|
return self.emitErrorFmt("unsupported vector swizzle '{s}'", .{field});
|
|
}
|
|
|
|
fn genFieldAccess(self: *CodeGen, fa: ast.FieldAccess) !c.LLVMValueRef {
|
|
// Check if the object is a struct or vector variable
|
|
if (fa.object.data == .identifier) {
|
|
if (self.named_values.get(fa.object.data.identifier.name)) |entry| {
|
|
// Pointer auto-deref: p.field → p.*.field
|
|
if (entry.ty.isPointer()) {
|
|
const pointee_ty = self.resolveTypeFromName(entry.ty.pointer_type.pointee_name) orelse
|
|
return self.emitError("unknown pointee type for auto-deref");
|
|
const loaded_ptr = c.LLVMBuildLoad2(self.builder,
|
|
self.ptrType(), entry.ptr, "ptr_load");
|
|
if (pointee_ty.isStruct()) {
|
|
const sname = pointee_ty.struct_type;
|
|
const info = try self.getStructInfo(sname);
|
|
const idx = self.findNameIndex(info.field_names, fa.field) orelse
|
|
return self.emitErrorFmt("no field '{s}' in struct '{s}'", .{ fa.field, sname });
|
|
const gep = self.structGEP(info.llvm_type, loaded_ptr, @intCast(idx), "pfield");
|
|
return self.loadTyped(info.field_types[idx], gep, "pfieldval");
|
|
}
|
|
if (pointee_ty.isSlice()) {
|
|
const slice_val = c.LLVMBuildLoad2(self.builder, self.getStringStructType(), loaded_ptr, "pslice_load");
|
|
return self.extractFatPtrField(slice_val, fa.field, "*slice");
|
|
}
|
|
return self.emitErrorFmt("no field '{s}' on pointer", .{fa.field});
|
|
}
|
|
if (entry.ty.isStruct()) {
|
|
const sname = entry.ty.struct_type;
|
|
const info = try self.getStructInfo(sname);
|
|
const idx = try self.findFieldIndex(info.field_names, fa.field, sname);
|
|
return self.loadStructField(info.llvm_type, entry.ptr, @intCast(idx), self.typeToLLVM(info.field_types[idx]));
|
|
}
|
|
if (entry.ty.isUnion()) {
|
|
const uname = entry.ty.union_type;
|
|
// C-style (untagged) union: bitcast pointer and load
|
|
if (self.lookupUnionInfo(uname)) |info| {
|
|
if (self.findNameIndex(info.field_names, fa.field)) |fidx| {
|
|
const field_ty = info.field_types[fidx];
|
|
return self.loadTyped(field_ty, entry.ptr, "union_field");
|
|
}
|
|
// Check promoted fields from anonymous structs
|
|
if (info.promoted_fields.get(fa.field)) |pf| {
|
|
const sinfo = try self.getStructInfo(pf.struct_name);
|
|
// GEP through union pointer as struct type, then access field
|
|
return self.loadStructField(sinfo.llvm_type, entry.ptr, @intCast(pf.field_index), self.typeToLLVM(pf.field_type));
|
|
}
|
|
return self.emitErrorFmt("no field '{s}' in union '{s}'", .{ fa.field, uname });
|
|
}
|
|
// Tagged enum: GEP to payload area
|
|
const info = try self.getTaggedEnumInfo(uname);
|
|
// Find variant by name to determine payload type
|
|
var vidx: ?usize = null;
|
|
for (info.variant_names, 0..) |vn, i| {
|
|
if (std.mem.eql(u8, vn, fa.field)) {
|
|
vidx = i;
|
|
break;
|
|
}
|
|
}
|
|
const idx = vidx orelse return self.emitErrorFmt("no variant '{s}' in enum '{s}'", .{ fa.field, uname });
|
|
const variant_ty = info.variant_types[idx];
|
|
if (variant_ty == .void_type) return self.emitErrorFmt("cannot access payload of void variant '{s}'", .{fa.field});
|
|
// GEP to payload area, load as variant type
|
|
return self.loadStructField(info.llvm_type, entry.ptr, info.payload_field_index, self.typeToLLVM(variant_ty));
|
|
}
|
|
if (entry.ty.isVector()) {
|
|
const vec_val = self.loadTyped(entry.ty, entry.ptr, "vec_load");
|
|
return self.genVectorExtract(vec_val, fa.field);
|
|
}
|
|
if (entry.ty == .string_type) {
|
|
const str_val = c.LLVMBuildLoad2(self.builder, self.getStringStructType(), entry.ptr, "str_load");
|
|
return self.extractFatPtrField(str_val, fa.field, "string");
|
|
}
|
|
if (entry.ty.isSlice()) {
|
|
const slice_val = c.LLVMBuildLoad2(self.builder, self.getStringStructType(), entry.ptr, "slice_load");
|
|
return self.extractFatPtrField(slice_val, fa.field, "slice");
|
|
}
|
|
if (entry.ty.isArray()) {
|
|
if (std.mem.eql(u8, fa.field, "len")) {
|
|
return self.constInt64(entry.ty.array_type.length);
|
|
}
|
|
return self.emitErrorFmt("no field '{s}' on array (available: .len)", .{fa.field});
|
|
}
|
|
if (entry.ty.isAny()) {
|
|
const any_val = c.LLVMBuildLoad2(self.builder, self.getAnyStructType(), entry.ptr, "any_load");
|
|
if (std.mem.eql(u8, fa.field, "tag")) {
|
|
return self.extractValue(any_val, 0, "any_tag");
|
|
}
|
|
if (std.mem.eql(u8, fa.field, "value")) {
|
|
return self.extractValue(any_val, 1, "any_value");
|
|
}
|
|
return self.emitErrorFmt("no field '{s}' on Any (available: .tag, .value)", .{fa.field});
|
|
}
|
|
}
|
|
}
|
|
// Non-identifier object: evaluate expression and check type
|
|
const obj_val = try self.genExpr(fa.object);
|
|
const obj_ty = self.inferType(fa.object);
|
|
if (obj_ty.isVector()) {
|
|
return self.genVectorExtract(obj_val, fa.field);
|
|
}
|
|
if (obj_ty == .string_type) {
|
|
return self.extractFatPtrField(obj_val, fa.field, "string");
|
|
}
|
|
return self.emitError("field access on non-struct/non-vector expression");
|
|
}
|
|
|
|
fn genVectorComparison(self: *CodeGen, op: ast.BinaryOp.Op, lhs: c.LLVMValueRef, rhs: c.LLVMValueRef, vec_ty: Type, elem_ty: Type) c.LLVMValueRef {
|
|
const vec_info = vec_ty.vector_type;
|
|
const cmp = if (elem_ty.isFloat())
|
|
(if (op == .eq) c.LLVMBuildFCmp(self.builder, c.LLVMRealOEQ, lhs, rhs, "vcmp") else c.LLVMBuildFCmp(self.builder, c.LLVMRealONE, lhs, rhs, "vcmp"))
|
|
else
|
|
(if (op == .eq) self.icmp(c.LLVMIntEQ, lhs, rhs, "vcmp") else self.icmp(c.LLVMIntNE, lhs, rhs, "vcmp"));
|
|
// Reduce: extract each i1 and AND (eq) or OR (neq)
|
|
var result = c.LLVMBuildExtractElement(self.builder, cmp, self.constInt32(0), "cmp0");
|
|
for (1..vec_info.length) |i| {
|
|
const elem = c.LLVMBuildExtractElement(self.builder, cmp, self.constInt32(@intCast(i)), "cmpi");
|
|
result = if (op == .eq)
|
|
c.LLVMBuildAnd(self.builder, result, elem, "andcmp")
|
|
else
|
|
c.LLVMBuildOr(self.builder, result, elem, "orcmp");
|
|
}
|
|
return result;
|
|
}
|
|
|
|
fn genIndexExpr(self: *CodeGen, ie: ast.IndexExpr) !c.LLVMValueRef {
|
|
const obj_ty = self.inferType(ie.object);
|
|
if (obj_ty.isVector()) {
|
|
const vec_val = try self.genExpr(ie.object);
|
|
const idx = try self.genExpr(ie.index);
|
|
return c.LLVMBuildExtractElement(self.builder, vec_val, idx, "vidx");
|
|
}
|
|
if (obj_ty.isArray()) {
|
|
const arr_info = obj_ty.array_type;
|
|
const elem_ty = try self.resolveElementType(arr_info.element_name, "array");
|
|
// Array index via identifier: load from GEP
|
|
if (ie.object.data == .identifier) {
|
|
if (self.named_values.get(ie.object.data.identifier.name)) |entry| {
|
|
const idx = try self.genExpr(ie.index);
|
|
const gep = self.gepArrayElement(self.typeToLLVM(obj_ty), entry.ptr, idx, "arridx");
|
|
return self.loadTyped(elem_ty, gep, "arrval");
|
|
}
|
|
}
|
|
// Array index via field access: GEP through struct field
|
|
if (ie.object.data == .field_access) {
|
|
const field_ptr = try self.genAddressOf(ie.object);
|
|
const idx = try self.genExpr(ie.index);
|
|
const gep = self.gepArrayElement(self.typeToLLVM(obj_ty), field_ptr, idx, "field_arridx");
|
|
return self.loadTyped(elem_ty, gep, "field_arrval");
|
|
}
|
|
}
|
|
if (obj_ty == .string_type) {
|
|
// String indexing: extract ptr from slice, GEP + load u8
|
|
const str_val = try self.genExpr(ie.object);
|
|
const ptr = self.extractValue(str_val, 0, "str_ptr");
|
|
const idx = try self.genExpr(ie.index);
|
|
const i8_type = self.i8Type();
|
|
const gep = self.gepPointerElement(i8_type, ptr, idx, "stridx");
|
|
return c.LLVMBuildLoad2(self.builder, i8_type, gep, "byte");
|
|
}
|
|
if (obj_ty.isSlice()) {
|
|
// Slice indexing: extract ptr, GEP with element type, load
|
|
const slice_info = obj_ty.slice_type;
|
|
const elem_ty = try self.resolveElementType(slice_info.element_name, "slice");
|
|
const elem_llvm_ty = self.typeToLLVM(elem_ty);
|
|
// For identifier objects, load the slice from alloca
|
|
if (ie.object.data == .identifier) {
|
|
if (self.named_values.get(ie.object.data.identifier.name)) |entry| {
|
|
const slice_val = c.LLVMBuildLoad2(self.builder, self.getStringStructType(), entry.ptr, "slice_load");
|
|
const ptr = self.extractValue(slice_val, 0, "slice_ptr");
|
|
const idx = try self.genExpr(ie.index);
|
|
const gep = self.gepPointerElement(elem_llvm_ty, ptr, idx, "sliceidx");
|
|
return c.LLVMBuildLoad2(self.builder, elem_llvm_ty, gep, "sliceval");
|
|
}
|
|
}
|
|
// Fallback for non-identifier slice expressions
|
|
const slice_val = try self.genExpr(ie.object);
|
|
const ptr = self.extractValue(slice_val, 0, "slice_ptr");
|
|
const idx = try self.genExpr(ie.index);
|
|
const gep = self.gepPointerElement(elem_llvm_ty, ptr, idx, "sliceidx");
|
|
return c.LLVMBuildLoad2(self.builder, elem_llvm_ty, gep, "sliceval");
|
|
}
|
|
// Many-pointer indexing: [*]T — GEP + load
|
|
if (obj_ty.isManyPointer()) {
|
|
const elem_ty = self.resolveTypeFromName(obj_ty.many_pointer_type.element_name) orelse return self.emitError("unknown many-pointer element type");
|
|
const elem_llvm_ty = self.typeToLLVM(elem_ty);
|
|
const ptr_val = try self.genExpr(ie.object);
|
|
const idx = try self.genExpr(ie.index);
|
|
const gep = self.gepPointerElement(elem_llvm_ty, ptr_val, idx, "mptridx");
|
|
return c.LLVMBuildLoad2(self.builder, elem_llvm_ty, gep, "mptrval");
|
|
}
|
|
return self.emitError("index expression requires an array, vector, string, slice, or [*] pointer");
|
|
}
|
|
|
|
fn genSliceExpr(self: *CodeGen, se: ast.SliceExpr) !c.LLVMValueRef {
|
|
const obj_ty = self.inferType(se.object);
|
|
const i64_ty = self.i64Type();
|
|
const zero = c.LLVMConstInt(i64_ty, 0, 0);
|
|
const slice_struct_ty = self.getStringStructType();
|
|
|
|
// Resolve start (default: 0)
|
|
const start_val = if (se.start) |s| try self.genExpr(s) else zero;
|
|
|
|
if (obj_ty.isArray()) {
|
|
const arr_info = obj_ty.array_type;
|
|
// Resolve end (default: array length)
|
|
const end_val = if (se.end) |e| try self.genExpr(e) else c.LLVMConstInt(i64_ty, arr_info.length, 0);
|
|
// Get array alloca
|
|
const arr_ptr = blk: {
|
|
if (se.object.data == .identifier) {
|
|
if (self.named_values.get(se.object.data.identifier.name)) |entry| {
|
|
break :blk entry.ptr;
|
|
}
|
|
}
|
|
break :blk try self.genExpr(se.object);
|
|
};
|
|
// GEP to arr[start]
|
|
var indices = [_]c.LLVMValueRef{ zero, start_val };
|
|
const elem_ptr = c.LLVMBuildGEP2(self.builder, self.typeToLLVM(obj_ty), arr_ptr, &indices, 2, "slice_start");
|
|
// len = end - start
|
|
const len_val = c.LLVMBuildSub(self.builder, end_val, start_val, "slice_len");
|
|
// Build {ptr, len}
|
|
return self.buildFatPointer(slice_struct_ty, elem_ptr, len_val);
|
|
}
|
|
|
|
if (obj_ty == .string_type or obj_ty.isSlice()) {
|
|
// Load {ptr, len} from variable or expression
|
|
const obj_val = blk: {
|
|
if (se.object.data == .identifier) {
|
|
if (self.named_values.get(se.object.data.identifier.name)) |entry| {
|
|
break :blk c.LLVMBuildLoad2(self.builder, slice_struct_ty, entry.ptr, "sslice_load");
|
|
}
|
|
}
|
|
break :blk try self.genExpr(se.object);
|
|
};
|
|
const base_ptr = self.extractValue(obj_val, 0, "sslice_ptr");
|
|
const base_len = self.extractValue(obj_val, 1, "sslice_len");
|
|
// Resolve end (default: original length)
|
|
const end_val = if (se.end) |e| try self.genExpr(e) else base_len;
|
|
// GEP base_ptr + start
|
|
const elem_llvm_ty = if (obj_ty == .string_type)
|
|
self.i8Type()
|
|
else
|
|
self.typeToLLVM(Type.fromName(obj_ty.slice_type.element_name) orelse return self.emitError("unknown slice element type"));
|
|
const new_ptr = self.gepPointerElement(elem_llvm_ty, base_ptr, start_val, "sslice_off");
|
|
// len = end - start
|
|
const len_val = c.LLVMBuildSub(self.builder, end_val, start_val, "sslice_len");
|
|
// Build {ptr, len}
|
|
return self.buildFatPointer(slice_struct_ty, new_ptr, len_val);
|
|
}
|
|
|
|
return self.emitError("slice expression requires an array, string, or slice");
|
|
}
|
|
|
|
fn genBinaryOp(self: *CodeGen, op: ast.BinaryOp.Op, lhs: c.LLVMValueRef, rhs: c.LLVMValueRef, result_type: Type) c.LLVMValueRef {
|
|
// For vectors, dispatch based on element type; LLVM handles element-wise ops automatically
|
|
const effective_ty = if (result_type.isVector())
|
|
result_type.vectorElementType() orelse return lhs
|
|
else
|
|
result_type;
|
|
|
|
// Vector comparison needs special handling (returns vector of i1)
|
|
if (result_type.isVector() and (op == .eq or op == .neq)) {
|
|
return self.genVectorComparison(op, lhs, rhs, result_type, effective_ty);
|
|
}
|
|
|
|
const b = self.builder;
|
|
const is_float = effective_ty.isFloat();
|
|
const is_unsigned = effective_ty.isUnsigned();
|
|
|
|
return switch (op) {
|
|
.add => if (is_float) c.LLVMBuildFAdd(b, lhs, rhs, "addtmp") else c.LLVMBuildAdd(b, lhs, rhs, "addtmp"),
|
|
.sub => if (is_float) c.LLVMBuildFSub(b, lhs, rhs, "subtmp") else c.LLVMBuildSub(b, lhs, rhs, "subtmp"),
|
|
.mul => if (is_float) c.LLVMBuildFMul(b, lhs, rhs, "multmp") else c.LLVMBuildMul(b, lhs, rhs, "multmp"),
|
|
.div => if (is_float) c.LLVMBuildFDiv(b, lhs, rhs, "divtmp") else if (is_unsigned) c.LLVMBuildUDiv(b, lhs, rhs, "divtmp") else c.LLVMBuildSDiv(b, lhs, rhs, "divtmp"),
|
|
.mod => if (is_float) c.LLVMBuildFRem(b, lhs, rhs, "modtmp") else if (is_unsigned) c.LLVMBuildURem(b, lhs, rhs, "modtmp") else c.LLVMBuildSRem(b, lhs, rhs, "modtmp"),
|
|
.eq => if (is_float) c.LLVMBuildFCmp(b, c.LLVMRealOEQ, lhs, rhs, "eqtmp") else c.LLVMBuildICmp(b, c.LLVMIntEQ, lhs, rhs, "eqtmp"),
|
|
.neq => if (is_float) c.LLVMBuildFCmp(b, c.LLVMRealONE, lhs, rhs, "neqtmp") else c.LLVMBuildICmp(b, c.LLVMIntNE, lhs, rhs, "neqtmp"),
|
|
.lt => if (is_float) c.LLVMBuildFCmp(b, c.LLVMRealOLT, lhs, rhs, "lttmp") else if (is_unsigned) c.LLVMBuildICmp(b, c.LLVMIntULT, lhs, rhs, "lttmp") else c.LLVMBuildICmp(b, c.LLVMIntSLT, lhs, rhs, "lttmp"),
|
|
.lte => if (is_float) c.LLVMBuildFCmp(b, c.LLVMRealOLE, lhs, rhs, "letmp") else if (is_unsigned) c.LLVMBuildICmp(b, c.LLVMIntULE, lhs, rhs, "letmp") else c.LLVMBuildICmp(b, c.LLVMIntSLE, lhs, rhs, "letmp"),
|
|
.gt => if (is_float) c.LLVMBuildFCmp(b, c.LLVMRealOGT, lhs, rhs, "gttmp") else if (is_unsigned) c.LLVMBuildICmp(b, c.LLVMIntUGT, lhs, rhs, "gttmp") else c.LLVMBuildICmp(b, c.LLVMIntSGT, lhs, rhs, "gttmp"),
|
|
.gte => if (is_float) c.LLVMBuildFCmp(b, c.LLVMRealOGE, lhs, rhs, "getmp") else if (is_unsigned) c.LLVMBuildICmp(b, c.LLVMIntUGE, lhs, rhs, "getmp") else c.LLVMBuildICmp(b, c.LLVMIntSGE, lhs, rhs, "getmp"),
|
|
.bit_and => c.LLVMBuildAnd(b, lhs, rhs, "bandtmp"),
|
|
.bit_or => c.LLVMBuildOr(b, lhs, rhs, "bortmp"),
|
|
.and_op, .or_op => unreachable,
|
|
};
|
|
}
|
|
|
|
fn genShortCircuitOp(self: *CodeGen, binop: ast.BinaryOp, is_and: bool) !c.LLVMValueRef {
|
|
const lhs_val = self.valueToBool(try self.genExpr(binop.lhs));
|
|
const lhs_bb = self.getCurrentBlock();
|
|
|
|
const rhs_label: [*c]const u8 = if (is_and) "and.rhs" else "or.rhs";
|
|
const merge_label: [*c]const u8 = if (is_and) "and.merge" else "or.merge";
|
|
const rhs_bb = self.appendBB(rhs_label);
|
|
const merge_bb = self.appendBB(merge_label);
|
|
|
|
// AND: true → evaluate rhs, false → short-circuit to merge
|
|
// OR: true → short-circuit to merge, false → evaluate rhs
|
|
if (is_and)
|
|
self.condBr(lhs_val, rhs_bb, merge_bb)
|
|
else
|
|
self.condBr(lhs_val, merge_bb, rhs_bb);
|
|
|
|
self.positionAt(rhs_bb);
|
|
const rhs_val = self.valueToBool(try self.genExpr(binop.rhs));
|
|
const rhs_end_bb = self.getCurrentBlock();
|
|
self.br(merge_bb);
|
|
|
|
self.positionAt(merge_bb);
|
|
const short_circuit_val: u64 = if (is_and) 0 else 1;
|
|
const result_label: [*c]const u8 = if (is_and) "and.result" else "or.result";
|
|
const i1_ty = self.i1Type();
|
|
const phi = c.LLVMBuildPhi(self.builder, i1_ty, result_label);
|
|
var vals = [2]c.LLVMValueRef{ c.LLVMConstInt(i1_ty, short_circuit_val, 0), rhs_val };
|
|
var blocks = [2]c.LLVMBasicBlockRef{ lhs_bb, rhs_end_bb };
|
|
c.LLVMAddIncoming(phi, &vals, &blocks, 2);
|
|
|
|
return phi;
|
|
}
|
|
|
|
fn genChainedComparison(self: *CodeGen, chain: ast.ChainedComparison) !c.LLVMValueRef {
|
|
// Evaluate all operands exactly once
|
|
var operand_vals = std.ArrayList(c.LLVMValueRef).empty;
|
|
for (chain.operands) |operand| {
|
|
const val = try self.genExpr(operand);
|
|
try operand_vals.append(self.allocator, val);
|
|
}
|
|
|
|
// Compare pairwise and AND results together
|
|
var result: c.LLVMValueRef = undefined;
|
|
for (chain.ops, 0..) |op, i| {
|
|
const lhs_ty = self.inferType(chain.operands[i]);
|
|
const rhs_ty = self.inferType(chain.operands[i + 1]);
|
|
const cmp_type = Type.widen(lhs_ty, rhs_ty);
|
|
|
|
const lhs_conv = self.convertValue(operand_vals.items[i], lhs_ty, cmp_type);
|
|
const rhs_conv = self.convertValue(operand_vals.items[i + 1], rhs_ty, cmp_type);
|
|
|
|
const cmp = self.genBinaryOp(op, lhs_conv, rhs_conv, cmp_type);
|
|
|
|
if (i == 0) {
|
|
result = cmp;
|
|
} else {
|
|
result = c.LLVMBuildAnd(self.builder, result, cmp, "chain.and");
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
fn genCall(self: *CodeGen, call_node: ast.Call) !c.LLVMValueRef {
|
|
if (call_node.callee.data == .field_access) {
|
|
const fa = call_node.callee.data.field_access;
|
|
|
|
// Union construction: Shape.variant(payload)
|
|
const resolved_type: ?Type = blk: {
|
|
if (fa.object.data == .identifier) {
|
|
const name = self.resolveAlias(fa.object.data.identifier.name);
|
|
if (self.type_registry.get(name)) |e| switch (e) {
|
|
.tagged_enum => break :blk Type{ .union_type = name },
|
|
.struct_info => break :blk Type{ .struct_type = name },
|
|
else => {},
|
|
};
|
|
} else {
|
|
const ty = self.resolveType(fa.object);
|
|
if (ty.isUnion() or ty.isStruct()) break :blk ty;
|
|
}
|
|
break :blk null;
|
|
};
|
|
if (resolved_type) |rty| {
|
|
if (rty.isUnion()) {
|
|
const type_name = rty.union_type;
|
|
const payload_node: ?*Node = if (call_node.args.len > 0) call_node.args[0] else null;
|
|
return self.genTaggedEnumLiteral(.{
|
|
.name = fa.field,
|
|
.payload = payload_node,
|
|
}, type_name);
|
|
}
|
|
}
|
|
|
|
// Namespaced call: namespace.func(args)
|
|
if (fa.object.data == .identifier) {
|
|
const ns_name = fa.object.data.identifier.name;
|
|
if (self.namespaces.contains(ns_name)) {
|
|
const qualified = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns_name, fa.field });
|
|
return self.genCallByName(qualified, call_node);
|
|
}
|
|
}
|
|
|
|
// UFCS: obj.method(args...) → method(obj, args...)
|
|
const method_name = fa.field;
|
|
const method_z = self.allocator.dupeZ(u8, method_name) catch method_name;
|
|
if (self.generic_templates.contains(method_name) or
|
|
c.LLVMGetNamedFunction(self.module, method_z.ptr) != null)
|
|
{
|
|
var ufcs_args = try self.allocator.alloc(*Node, call_node.args.len + 1);
|
|
ufcs_args[0] = fa.object;
|
|
for (call_node.args, 0..) |arg, i| {
|
|
ufcs_args[i + 1] = arg;
|
|
}
|
|
return self.genCallByName(method_name, .{
|
|
.callee = call_node.callee,
|
|
.args = ufcs_args,
|
|
});
|
|
}
|
|
}
|
|
|
|
// Resolve callee — must be an identifier
|
|
if (call_node.callee.data != .identifier) return self.emitError("callee must be an identifier");
|
|
const callee_name = call_node.callee.data.identifier.name;
|
|
return self.genCallByName(callee_name, call_node);
|
|
}
|
|
|
|
fn genCallByName(self: *CodeGen, callee_name: []const u8, call_node: ast.Call) !c.LLVMValueRef {
|
|
// Check if this is a generic function call
|
|
if (self.generic_templates.get(callee_name)) |template| {
|
|
return self.genGenericCall(callee_name, template, call_node);
|
|
}
|
|
// Intra-namespace fallback for generic templates
|
|
if (self.current_namespace) |ns| {
|
|
const qualified = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns, callee_name });
|
|
if (self.generic_templates.get(qualified)) |template| {
|
|
return self.genGenericCall(qualified, template, call_node);
|
|
}
|
|
}
|
|
|
|
// Check for #builtin function (only available when imported)
|
|
if (self.builtin_functions.contains(callee_name)) {
|
|
return self.dispatchBuiltin(callee_name, call_node);
|
|
}
|
|
// Intra-namespace fallback for builtins
|
|
if (self.current_namespace) |ns| {
|
|
const qualified_builtin = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns, callee_name });
|
|
if (self.builtin_functions.contains(qualified_builtin)) {
|
|
return self.dispatchBuiltin(qualified_builtin, call_node);
|
|
}
|
|
}
|
|
|
|
// Compiler intrinsics (always available, no #builtin declaration needed)
|
|
if (std.mem.eql(u8, callee_name, "sqrt")) {
|
|
return self.genMathIntrinsic(call_node, "sqrt");
|
|
}
|
|
if (std.mem.eql(u8, callee_name, "sin")) {
|
|
return self.genMathIntrinsic(call_node, "sin");
|
|
}
|
|
if (std.mem.eql(u8, callee_name, "cos")) {
|
|
return self.genMathIntrinsic(call_node, "cos");
|
|
}
|
|
if (std.mem.eql(u8, callee_name, "cast")) {
|
|
return self.genCast(call_node);
|
|
}
|
|
if (std.mem.eql(u8, callee_name, "malloc")) {
|
|
return self.genMalloc(call_node.args);
|
|
}
|
|
if (std.mem.eql(u8, callee_name, "free")) {
|
|
return self.genFree(call_node.args);
|
|
}
|
|
if (std.mem.eql(u8, callee_name, "memcpy")) {
|
|
return self.genMemcpy(call_node.args);
|
|
}
|
|
|
|
var nbuf: [256]u8 = undefined;
|
|
var callee_fn = c.LLVMGetNamedFunction(self.module, self.nameToCStr(callee_name, &nbuf));
|
|
// Foreign function fallback: qualified name "ns.Func" → try unqualified "Func" (the C symbol)
|
|
if (callee_fn == null) {
|
|
if (std.mem.lastIndexOfScalar(u8, callee_name, '.')) |dot_idx| {
|
|
const base_name = callee_name[dot_idx + 1 ..];
|
|
var bbuf: [256]u8 = undefined;
|
|
callee_fn = c.LLVMGetNamedFunction(self.module, self.nameToCStr(base_name, &bbuf));
|
|
}
|
|
}
|
|
// Intra-namespace fallback: try qualified name
|
|
if (callee_fn == null) {
|
|
if (self.current_namespace) |ns| {
|
|
const qualified2 = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns, callee_name });
|
|
var qbuf: [256]u8 = undefined;
|
|
callee_fn = c.LLVMGetNamedFunction(self.module, self.nameToCStr(qualified2, &qbuf));
|
|
}
|
|
}
|
|
// Function pointer indirect call: callee is a variable with function_type
|
|
if (callee_fn == null) {
|
|
if (self.lookupValue(callee_name)) |v| {
|
|
const entry = v.asNamedValue();
|
|
if (entry) |e| {
|
|
if (e.ty.isFunctionType()) {
|
|
return self.genIndirectCall(e, call_node);
|
|
}
|
|
}
|
|
}
|
|
return self.emitErrorFmt("undefined function '{s}'", .{callee_name});
|
|
}
|
|
|
|
// Get function type (opaque pointers: use LLVMGlobalGetValueType)
|
|
const fn_type = c.LLVMGlobalGetValueType(callee_fn.?);
|
|
|
|
// Check if this is a variadic function call
|
|
const var_info = self.variadic_functions.get(callee_name) orelse blk: {
|
|
// Try qualified name lookup
|
|
if (self.current_namespace) |ns| {
|
|
const qualified = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns, callee_name });
|
|
break :blk self.variadic_functions.get(qualified);
|
|
}
|
|
break :blk null;
|
|
};
|
|
|
|
// Generate arguments with type conversion to match parameter types
|
|
const num_params = c.LLVMCountParamTypes(fn_type);
|
|
var param_llvm_types: [64]c.LLVMTypeRef = undefined;
|
|
if (num_params > 0) {
|
|
c.LLVMGetParamTypes(fn_type, ¶m_llvm_types);
|
|
}
|
|
|
|
var arg_vals = std.ArrayList(c.LLVMValueRef).empty;
|
|
|
|
if (var_info) |vi| {
|
|
// Variadic call: generate fixed args, then pack remaining into slice
|
|
const fixed_count = vi.fixed_param_count;
|
|
// Generate fixed args
|
|
for (0..fixed_count) |i| {
|
|
if (i < call_node.args.len) {
|
|
const param_ty = self.llvmTypeToSxType(param_llvm_types[i]);
|
|
try arg_vals.append(self.allocator, try self.genExprAsType(call_node.args[i], param_ty));
|
|
}
|
|
}
|
|
// Pack variadic args into a slice {ptr, len}
|
|
const elem_ty = Type.fromName(vi.element_type_name) orelse Type.s(64);
|
|
const elem_llvm_ty = self.typeToLLVM(elem_ty);
|
|
const var_arg_count = if (call_node.args.len > fixed_count) call_node.args.len - fixed_count else 0;
|
|
|
|
// Check for spread operator: fn(..array) — single spread arg
|
|
if (var_arg_count == 1 and call_node.args[fixed_count].data == .spread_expr) {
|
|
const spread_operand = call_node.args[fixed_count].data.spread_expr.operand;
|
|
const spread_ty = self.inferType(spread_operand);
|
|
if (spread_ty.isArray()) {
|
|
// Spread an array: construct slice from array pointer + known length
|
|
const arr_info = spread_ty.array_type;
|
|
if (spread_operand.data == .identifier) {
|
|
if (self.named_values.get(spread_operand.data.identifier.name)) |entry| {
|
|
const arr_llvm_ty = self.typeToLLVM(spread_ty);
|
|
const arr_ptr = self.arrayDecayToPointer(arr_llvm_ty, entry.ptr, "spread_ptr");
|
|
const len_val = self.constInt64(arr_info.length);
|
|
const slice_val = self.buildStringSlice(arr_ptr, len_val);
|
|
try arg_vals.append(self.allocator, slice_val);
|
|
} else {
|
|
return self.emitError("spread operand not found");
|
|
}
|
|
} else {
|
|
return self.emitError("spread operator requires a named variable");
|
|
}
|
|
} else if (spread_ty.isSlice()) {
|
|
// Spread a slice: pass through as-is
|
|
const slice_val = try self.genExpr(spread_operand);
|
|
try arg_vals.append(self.allocator, slice_val);
|
|
} else {
|
|
return self.emitError("spread operator requires an array or slice");
|
|
}
|
|
} else if (var_arg_count > 0) {
|
|
// Allocate array on stack: [N x elem_type]
|
|
const arr_ty = c.LLVMArrayType2(elem_llvm_ty, @intCast(var_arg_count));
|
|
const arr_alloca = self.buildEntryBlockAlloca(arr_ty, "varargs_arr");
|
|
// Store each variadic arg
|
|
for (0..var_arg_count) |vi_idx| {
|
|
const arg_val = if (elem_ty.isAny()) blk: {
|
|
// ..Any: wrap each arg in Any{tag, value}
|
|
const raw_val = try self.genExpr(call_node.args[fixed_count + vi_idx]);
|
|
const arg_ty = self.inferType(call_node.args[fixed_count + vi_idx]);
|
|
break :blk try self.buildAnyValue(raw_val, arg_ty);
|
|
} else try self.genExprAsType(call_node.args[fixed_count + vi_idx], elem_ty);
|
|
const gep = self.gepArrayElement(arr_ty, arr_alloca, self.constInt32(@intCast(vi_idx)), "vararg_elem");
|
|
_ = c.LLVMBuildStore(self.builder, arg_val, gep);
|
|
}
|
|
// Build slice: {ptr, len}
|
|
const arr_ptr = self.arrayDecayToPointer(arr_ty, arr_alloca, "varargs_ptr");
|
|
const len_val = self.constInt64(@intCast(var_arg_count));
|
|
const slice_val = self.buildStringSlice(arr_ptr, len_val);
|
|
try arg_vals.append(self.allocator, slice_val);
|
|
} else {
|
|
// Zero variadic args: pass empty slice {null, 0}
|
|
const null_ptr = c.LLVMConstNull(self.ptrType());
|
|
const zero_len = self.constInt64(0);
|
|
const slice_val = self.buildStringSlice(null_ptr, zero_len);
|
|
try arg_vals.append(self.allocator, slice_val);
|
|
}
|
|
} else {
|
|
// Normal (non-variadic) call — use stored sx param types when available
|
|
const stored_param_types = self.fn_param_types.get(callee_name) orelse blk: {
|
|
if (self.current_namespace) |ns| {
|
|
const qualified = try std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns, callee_name });
|
|
break :blk self.fn_param_types.get(qualified);
|
|
}
|
|
break :blk null;
|
|
};
|
|
for (call_node.args, 0..) |arg, i| {
|
|
if (i < num_params) {
|
|
const param_ty = if (stored_param_types != null and i < stored_param_types.?.len)
|
|
stored_param_types.?[i]
|
|
else
|
|
self.llvmTypeToSxType(param_llvm_types[i]);
|
|
// For #foreign functions, [:0]u8 params have LLVM type ptr
|
|
const arg_ty = self.inferType(arg);
|
|
const llvm_param_is_ptr = (i < num_params and
|
|
c.LLVMGetTypeKind(param_llvm_types[i]) == c.LLVMPointerTypeKind);
|
|
const ptr_ty = Type{ .pointer_type = .{ .pointee_name = "u8" } };
|
|
|
|
var val = if (llvm_param_is_ptr and arg.data == .string_literal) blk: {
|
|
// String literal → pointer: produce raw ptr directly (context-dependent)
|
|
break :blk try self.genExprAsType(arg, ptr_ty);
|
|
} else if (llvm_param_is_ptr and arg_ty == .string_type) blk: {
|
|
// String variable → pointer: extract .ptr from {ptr, len}
|
|
const str_val = try self.genExpr(arg);
|
|
break :blk self.extractValue(str_val, 0, "str_ptr");
|
|
} else if ((param_ty.isPointer() or llvm_param_is_ptr) and arg_ty.isSlice() and
|
|
std.mem.eql(u8, arg_ty.slice_type.element_name, "u8"))
|
|
{
|
|
return self.emitError(
|
|
"cannot pass []u8 to *u8: slice may not be null-terminated; use a string literal or xx cast",
|
|
);
|
|
} else try self.genExprAsType(arg, param_ty);
|
|
// Foreign calls: convert struct values to C ABI representation
|
|
if (param_ty.isStruct() and self.foreign_fns.contains(callee_name)) {
|
|
const struct_llvm_ty = self.typeToLLVM(param_ty);
|
|
if (struct_llvm_ty != param_llvm_types[i]) {
|
|
val = self.convertStructToABI(val, struct_llvm_ty, param_llvm_types[i]);
|
|
}
|
|
}
|
|
try arg_vals.append(self.allocator, val);
|
|
} else {
|
|
try arg_vals.append(self.allocator, try self.genExpr(arg));
|
|
}
|
|
}
|
|
}
|
|
const args_slice = try arg_vals.toOwnedSlice(self.allocator);
|
|
|
|
const ret_ty = c.LLVMGetReturnType(fn_type);
|
|
const call_name: [*c]const u8 = if (ret_ty == self.voidType()) "" else "calltmp";
|
|
return c.LLVMBuildCall2(
|
|
self.builder,
|
|
fn_type,
|
|
callee_fn.?,
|
|
if (args_slice.len > 0) args_slice.ptr else null,
|
|
@intCast(args_slice.len),
|
|
call_name,
|
|
);
|
|
}
|
|
|
|
fn genIndirectCall(self: *CodeGen, entry: NamedValue, call_node: ast.Call) !c.LLVMValueRef {
|
|
const fti = entry.ty.function_type;
|
|
|
|
// Load the function pointer from the alloca
|
|
const ptr_ty = self.ptrType();
|
|
const fn_ptr = c.LLVMBuildLoad2(self.builder, ptr_ty, entry.ptr, "fn_ptr");
|
|
|
|
// Build LLVM function type from FunctionTypeInfo
|
|
const ptr_ty_llvm = self.ptrType();
|
|
var param_llvm_types: [64]c.LLVMTypeRef = undefined;
|
|
for (fti.param_types, 0..) |pt, i| {
|
|
// [N]T and [:0]T params are pointers at the ABI level
|
|
param_llvm_types[i] = if (pt.isArray() or pt == .string_type) ptr_ty_llvm else self.typeToLLVM(pt);
|
|
}
|
|
const ret_llvm = self.typeToLLVM(fti.return_type.*);
|
|
const fn_type = c.LLVMFunctionType(
|
|
ret_llvm,
|
|
if (fti.param_types.len > 0) ¶m_llvm_types else null,
|
|
@intCast(fti.param_types.len),
|
|
0,
|
|
);
|
|
|
|
// Generate arguments with type conversion
|
|
var arg_vals = std.ArrayList(c.LLVMValueRef).empty;
|
|
for (call_node.args, 0..) |arg, i| {
|
|
if (i < fti.param_types.len) {
|
|
const pt = fti.param_types[i];
|
|
// [N]T params: pass pointer via array decay
|
|
if (pt.isArray()) {
|
|
const decay_target: Type = .{ .many_pointer_type = .{ .element_name = pt.array_type.element_name } };
|
|
try arg_vals.append(self.allocator, try self.genExprAsType(arg, decay_target));
|
|
} else if (pt == .string_type) {
|
|
// [:0]u8 params: extract .ptr from fat pointer
|
|
const val = try self.genExprAsType(arg, pt);
|
|
try arg_vals.append(self.allocator, self.extractValue(val, 0, "str_ptr"));
|
|
} else {
|
|
try arg_vals.append(self.allocator, try self.genExprAsType(arg, pt));
|
|
}
|
|
} else {
|
|
try arg_vals.append(self.allocator, try self.genExpr(arg));
|
|
}
|
|
}
|
|
const args_slice = try arg_vals.toOwnedSlice(self.allocator);
|
|
|
|
const call_name: [*c]const u8 = if (ret_llvm == self.voidType()) "" else "calltmp";
|
|
return c.LLVMBuildCall2(
|
|
self.builder,
|
|
fn_type,
|
|
fn_ptr,
|
|
if (args_slice.len > 0) args_slice.ptr else null,
|
|
@intCast(args_slice.len),
|
|
call_name,
|
|
);
|
|
}
|
|
|
|
fn genGenericCall(self: *CodeGen, qualified_name: []const u8, template: ast.FnDecl, call_node: ast.Call) !c.LLVMValueRef {
|
|
const fd = template;
|
|
|
|
// Check for runtime type dispatch: cast(runtime_type_var, any_val) as argument
|
|
if (self.current_match_tags) |match_tags| {
|
|
if (match_tags.len > 0) {
|
|
for (call_node.args) |arg| {
|
|
if (arg.data == .call) {
|
|
if (arg.data.call.callee.data == .identifier) {
|
|
const cast_name = arg.data.call.callee.data.identifier.name;
|
|
if (std.mem.eql(u8, cast_name, "cast") or std.mem.eql(u8, cast_name, "std.cast")) {
|
|
if (arg.data.call.args.len == 2) {
|
|
const type_arg = arg.data.call.args[0];
|
|
// Check if first arg of cast is a runtime variable (not a type expression)
|
|
if (type_arg.data == .identifier) {
|
|
const name = type_arg.data.identifier.name;
|
|
// It's a runtime type if it's a named_value, not a type name
|
|
if (self.named_values.contains(name) and
|
|
Type.fromName(name) == null and
|
|
!self.type_registry.contains(name))
|
|
{
|
|
return self.genGenericCallWithRuntimeDispatch(template, call_node, match_tags);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Check for comptime value params
|
|
var has_comptime_values = false;
|
|
var comptime_nodes = std.StringHashMap(*Node).init(self.allocator);
|
|
for (fd.type_params) |tp| {
|
|
const constraint_name = if (tp.constraint.data == .type_expr) tp.constraint.data.type_expr.name else "";
|
|
if (!std.mem.eql(u8, constraint_name, "Type")) {
|
|
// Value param — extract comptime value from call arg
|
|
has_comptime_values = true;
|
|
for (fd.params, 0..) |param, pi| {
|
|
if (std.mem.eql(u8, param.name, tp.name)) {
|
|
if (pi < call_node.args.len) {
|
|
try comptime_nodes.put(tp.name, @constCast(call_node.args[pi]));
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Normal generic call: Infer type bindings from arguments, widening across all args for the same type param
|
|
var bindings = std.StringHashMap(Type).init(self.allocator);
|
|
// Track bindings derived from parameterized struct types — these are authoritative and should not be widened
|
|
var firm_bindings = std.StringHashMap(void).init(self.allocator);
|
|
for (fd.params, 0..) |param, i| {
|
|
if (param.is_comptime) continue;
|
|
// Direct type param: (a: $T) introduces/widens, (a: T) only binds if not yet bound
|
|
if (param.type_expr.data == .type_expr) {
|
|
const type_name = param.type_expr.data.type_expr.name;
|
|
// Check if this type name is a type parameter
|
|
for (fd.type_params) |tp| {
|
|
if (std.mem.eql(u8, tp.name, type_name)) {
|
|
if (i < call_node.args.len) {
|
|
// Skip widening if binding was derived from a parameterized struct
|
|
if (!firm_bindings.contains(type_name)) {
|
|
const arg_ty = self.inferType(call_node.args[i]);
|
|
if (bindings.get(type_name)) |existing| {
|
|
try bindings.put(type_name, Type.widen(existing, arg_ty));
|
|
} else {
|
|
try bindings.put(type_name, arg_ty);
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
// Pointer to parameterized type: (p: *Foo($T)) — extract T from concrete struct
|
|
if (param.type_expr.data == .pointer_type_expr) {
|
|
const pointee = param.type_expr.data.pointer_type_expr.pointee_type;
|
|
if (pointee.data == .parameterized_type_expr) {
|
|
const pte = pointee.data.parameterized_type_expr;
|
|
if (i < call_node.args.len) {
|
|
const arg_ty = self.inferType(call_node.args[i]);
|
|
// arg should be *StructName — get the struct's stored type param bindings
|
|
const struct_name = if (arg_ty.isPointer())
|
|
arg_ty.pointer_type.pointee_name
|
|
else if (arg_ty.isStruct())
|
|
arg_ty.struct_type
|
|
else
|
|
"";
|
|
if (self.lookupStructInfo(struct_name)) |info| {
|
|
if (info.template_name) |tmpl_name| {
|
|
if (std.mem.eql(u8, tmpl_name, pte.name)) {
|
|
// Match generic args against stored type param bindings
|
|
for (pte.args, 0..) |arg, ai| {
|
|
if (arg.data == .type_expr and arg.data.type_expr.is_generic) {
|
|
const gen_name = arg.data.type_expr.name;
|
|
if (ai < info.type_param_types.len) {
|
|
try bindings.put(gen_name, info.type_param_types[ai]);
|
|
try firm_bindings.put(gen_name, {});
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
// Slice type param: (items: []$T) — infer T from array or slice element type
|
|
if (param.type_expr.data == .slice_type_expr) {
|
|
const elem_node = param.type_expr.data.slice_type_expr.element_type;
|
|
if (elem_node.data == .type_expr) {
|
|
const type_name = elem_node.data.type_expr.name;
|
|
for (fd.type_params) |tp| {
|
|
if (std.mem.eql(u8, tp.name, type_name)) {
|
|
if (i < call_node.args.len) {
|
|
const arg_ty = self.inferType(call_node.args[i]);
|
|
const elem_ty = if (arg_ty.isArray())
|
|
Type.fromName(arg_ty.array_type.element_name) orelse arg_ty
|
|
else if (arg_ty.isSlice())
|
|
Type.fromName(arg_ty.slice_type.element_name) orelse arg_ty
|
|
else
|
|
arg_ty;
|
|
if (bindings.get(type_name)) |existing| {
|
|
try bindings.put(type_name, Type.widen(existing, elem_ty));
|
|
} else {
|
|
try bindings.put(type_name, elem_ty);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (has_comptime_values) {
|
|
return self.genComptimeCall(qualified_name, fd, call_node, bindings, comptime_nodes);
|
|
}
|
|
|
|
// Generate mangled name
|
|
const mangled = try self.mangleGenericName(fd.name, fd.type_params, bindings, null, null);
|
|
|
|
// Check cache
|
|
const callee_fn = if (self.generic_instances.get(mangled)) |cached|
|
|
cached
|
|
else
|
|
try self.instantiateGeneric(fd, bindings, mangled);
|
|
|
|
// Generate arguments with type conversion to match parameter types
|
|
const saved_call_bindings = self.type_param_bindings;
|
|
self.type_param_bindings = bindings;
|
|
var arg_vals = std.ArrayList(c.LLVMValueRef).empty;
|
|
for (call_node.args, 0..) |arg, i| {
|
|
if (i < fd.params.len) {
|
|
const param_ty = self.resolveType(fd.params[i].type_expr);
|
|
try arg_vals.append(self.allocator, try self.genExprAsType(arg, param_ty));
|
|
} else {
|
|
try arg_vals.append(self.allocator, try self.genExpr(arg));
|
|
}
|
|
}
|
|
self.type_param_bindings = saved_call_bindings;
|
|
const args_slice = try arg_vals.toOwnedSlice(self.allocator);
|
|
|
|
const fn_type = c.LLVMGlobalGetValueType(callee_fn);
|
|
const ret_ty = c.LLVMGetReturnType(fn_type);
|
|
const call_name: [*c]const u8 = if (ret_ty == self.voidType()) "" else "calltmp";
|
|
return c.LLVMBuildCall2(
|
|
self.builder,
|
|
fn_type,
|
|
callee_fn,
|
|
if (args_slice.len > 0) args_slice.ptr else null,
|
|
@intCast(args_slice.len),
|
|
call_name,
|
|
);
|
|
}
|
|
|
|
/// Generate a call to a generic function with comptime value parameters.
|
|
/// Instantiates the function with the specific comptime values, then delegates to genCallByName
|
|
/// with the mangled name and adjusted args (comptime args removed).
|
|
fn genComptimeCall(
|
|
self: *CodeGen,
|
|
qualified_name: []const u8,
|
|
fd: ast.FnDecl,
|
|
call_node: ast.Call,
|
|
type_bindings: std.StringHashMap(Type),
|
|
comptime_nodes: std.StringHashMap(*Node),
|
|
) !c.LLVMValueRef {
|
|
const mangled = try self.mangleGenericName(qualified_name, fd.type_params, type_bindings, null, comptime_nodes);
|
|
|
|
// Instantiate if not cached
|
|
if (!self.generic_instances.contains(mangled)) {
|
|
// Set comptime param nodes for #insert substitution
|
|
const saved_comptime_nodes = self.comptime_param_nodes;
|
|
self.comptime_param_nodes = comptime_nodes;
|
|
defer self.comptime_param_nodes = saved_comptime_nodes;
|
|
|
|
// Set namespace context if the qualified name is namespaced (e.g. "std.print")
|
|
const saved_namespace = self.current_namespace;
|
|
if (std.mem.indexOfScalar(u8, qualified_name, '.')) |dot_pos| {
|
|
self.current_namespace = qualified_name[0..dot_pos];
|
|
}
|
|
defer self.current_namespace = saved_namespace;
|
|
|
|
// Pre-register Any type IDs for variadic args before function instantiation,
|
|
// so type category matching (case slice:, case array:, etc.) in any_to_string
|
|
// can find registered types during compilation of the function body.
|
|
for (fd.params, 0..) |param, pi| {
|
|
if (param.is_variadic) {
|
|
const elem_name_raw = if (param.type_expr.data == .type_expr) param.type_expr.data.type_expr.name else "";
|
|
if (std.mem.eql(u8, elem_name_raw, "Any") or std.mem.eql(u8, elem_name_raw, "std.Any")) {
|
|
for (pi..call_node.args.len) |ai| {
|
|
const arg_ty = self.inferType(call_node.args[ai]);
|
|
try self.preRegisterAnyType(arg_ty);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
_ = try self.instantiateGeneric(fd, type_bindings, mangled);
|
|
|
|
// Register variadic info for the mangled function (adjusted for removed comptime params)
|
|
var comptime_before_variadic: u32 = 0;
|
|
for (fd.params) |param| {
|
|
if (param.is_variadic) break;
|
|
if (param.is_comptime) comptime_before_variadic += 1;
|
|
}
|
|
for (fd.params, 0..) |param, i| {
|
|
if (param.is_variadic) {
|
|
const elem_name = if (param.type_expr.data == .type_expr) param.type_expr.data.type_expr.name else "s32";
|
|
try self.variadic_functions.put(mangled, .{
|
|
.fixed_param_count = @intCast(i - comptime_before_variadic),
|
|
.element_type_name = elem_name,
|
|
});
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Build adjusted call args (skip comptime args)
|
|
var adjusted_args = std.ArrayList(*Node).empty;
|
|
for (call_node.args, 0..) |arg, i| {
|
|
if (i < fd.params.len and fd.params[i].is_comptime) continue;
|
|
try adjusted_args.append(self.allocator, @constCast(arg));
|
|
}
|
|
const adjusted_args_slice = try adjusted_args.toOwnedSlice(self.allocator);
|
|
|
|
const adjusted_call = ast.Call{
|
|
.callee = call_node.callee,
|
|
.args = adjusted_args_slice,
|
|
};
|
|
|
|
// Call the instantiated function through normal path (handles variadic packing etc.)
|
|
return self.genCallByName(mangled, adjusted_call);
|
|
}
|
|
|
|
/// Generate a generic function call with runtime type dispatch.
|
|
/// For each type tag in match_tags, monomorphize the generic function and dispatch via switch.
|
|
fn genGenericCallWithRuntimeDispatch(
|
|
self: *CodeGen,
|
|
template: ast.FnDecl,
|
|
call_node: ast.Call,
|
|
match_tags: []const u64,
|
|
) !c.LLVMValueRef {
|
|
const fd = template;
|
|
|
|
// Find the cast argument and extract the runtime type tag + any value source
|
|
var cast_arg_idx: usize = 0;
|
|
var type_tag_node: *Node = undefined;
|
|
var any_val_node: *Node = undefined;
|
|
for (call_node.args, 0..) |arg, i| {
|
|
if (arg.data == .call and arg.data.call.callee.data == .identifier) {
|
|
const name = arg.data.call.callee.data.identifier.name;
|
|
if ((std.mem.eql(u8, name, "cast") or std.mem.eql(u8, name, "std.cast")) and arg.data.call.args.len == 2) {
|
|
cast_arg_idx = i;
|
|
type_tag_node = arg.data.call.args[0];
|
|
any_val_node = arg.data.call.args[1];
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Generate the runtime type tag value and the Any value
|
|
const type_tag_val = try self.genExpr(type_tag_node);
|
|
const any_val = try self.genExpr(any_val_node);
|
|
// Generate non-cast arguments (evaluated once, before the switch)
|
|
var other_arg_vals = std.ArrayList(?c.LLVMValueRef).empty;
|
|
for (call_node.args, 0..) |arg, i| {
|
|
if (i == cast_arg_idx) {
|
|
try other_arg_vals.append(self.allocator, null); // placeholder
|
|
} else {
|
|
try other_arg_vals.append(self.allocator, try self.genExpr(arg));
|
|
}
|
|
}
|
|
|
|
// Extract Any value i64 BEFORE the switch (switch is a terminator, nothing can follow it in the same BB)
|
|
const any_i64 = self.extractValue(any_val, 1, "any_payload");
|
|
|
|
// Build dispatch switch
|
|
const sb = self.buildSwitch(type_tag_val, @intCast(match_tags.len), "dispatch_merge", "dispatch_default");
|
|
|
|
// Determine return type from function signature
|
|
const ret_ty = if (fd.return_type) |rt| self.resolveType(rt) else Type.void_type;
|
|
// We'll use the first monomorphized function's return to determine LLVM type
|
|
var result_llvm_ty: c.LLVMTypeRef = null;
|
|
|
|
var phi_vals = std.ArrayList(c.LLVMValueRef).empty;
|
|
var phi_bbs = std.ArrayList(c.LLVMBasicBlockRef).empty;
|
|
|
|
for (match_tags) |tag| {
|
|
// Find the AnyTypeEntry for this tag
|
|
var entry_type: ?Type = null;
|
|
var it = self.any_type_entries.iterator();
|
|
while (it.next()) |entry| {
|
|
if (entry.value_ptr.tag_id == tag) {
|
|
entry_type = entry.value_ptr.sx_type;
|
|
break;
|
|
}
|
|
}
|
|
const sx_type = entry_type orelse continue;
|
|
|
|
// Create case BB
|
|
const case_bb = self.appendBB("dispatch_case");
|
|
c.LLVMAddCase(sb.sw, self.constInt64(tag), case_bb);
|
|
self.positionAt(case_bb);
|
|
|
|
// Convert Any payload to the concrete type
|
|
const concrete_val = try self.extractAnyToConcreteType(any_i64, sx_type);
|
|
|
|
// Monomorphize the generic function with this type
|
|
var bindings = std.StringHashMap(Type).init(self.allocator);
|
|
// Bind the type parameter from the cast argument's position
|
|
if (cast_arg_idx < fd.params.len) {
|
|
if (fd.params[cast_arg_idx].type_expr.data == .type_expr) {
|
|
const tp_name = fd.params[cast_arg_idx].type_expr.data.type_expr.name;
|
|
for (fd.type_params) |tp| {
|
|
if (std.mem.eql(u8, tp.name, tp_name)) {
|
|
try bindings.put(tp.name, sx_type);
|
|
break;
|
|
}
|
|
}
|
|
} else if (fd.params[cast_arg_idx].type_expr.data == .slice_type_expr) {
|
|
// Slice type param: (items: []$T) — extract element type from concrete slice
|
|
const elem_node = fd.params[cast_arg_idx].type_expr.data.slice_type_expr.element_type;
|
|
if (elem_node.data == .type_expr) {
|
|
const tp_name = elem_node.data.type_expr.name;
|
|
for (fd.type_params) |tp| {
|
|
if (std.mem.eql(u8, tp.name, tp_name)) {
|
|
// Extract element type from concrete slice type
|
|
const elem_ty = if (sx_type.isSlice())
|
|
Type.fromName(sx_type.slice_type.element_name) orelse sx_type
|
|
else if (sx_type.isArray())
|
|
Type.fromName(sx_type.array_type.element_name) orelse sx_type
|
|
else
|
|
sx_type;
|
|
try bindings.put(tp.name, elem_ty);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
const mangled = try self.mangleGenericName(fd.name, fd.type_params, bindings, null, null);
|
|
const callee_fn = if (self.generic_instances.get(mangled)) |cached|
|
|
cached
|
|
else
|
|
try self.instantiateGeneric(fd, bindings, mangled);
|
|
|
|
// Build argument list
|
|
self.type_param_bindings = bindings;
|
|
var arg_vals_list = std.ArrayList(c.LLVMValueRef).empty;
|
|
for (other_arg_vals.items, 0..) |maybe_val, ai| {
|
|
if (ai == cast_arg_idx) {
|
|
// Use the converted concrete value
|
|
try arg_vals_list.append(self.allocator, concrete_val);
|
|
} else if (maybe_val) |v| {
|
|
try arg_vals_list.append(self.allocator, v);
|
|
}
|
|
}
|
|
self.type_param_bindings = null;
|
|
|
|
const args_slice = try arg_vals_list.toOwnedSlice(self.allocator);
|
|
const fn_type = c.LLVMGlobalGetValueType(callee_fn);
|
|
const call_result = c.LLVMBuildCall2(
|
|
self.builder,
|
|
fn_type,
|
|
callee_fn,
|
|
if (args_slice.len > 0) args_slice.ptr else null,
|
|
@intCast(args_slice.len),
|
|
if (ret_ty != .void_type) "dispatch_result" else "",
|
|
);
|
|
|
|
if (result_llvm_ty == null and ret_ty != .void_type) {
|
|
result_llvm_ty = c.LLVMTypeOf(call_result);
|
|
}
|
|
|
|
if (ret_ty != .void_type) {
|
|
try phi_vals.append(self.allocator, call_result);
|
|
try phi_bbs.append(self.allocator, self.getCurrentBlock());
|
|
}
|
|
self.br(sb.merge_bb);
|
|
}
|
|
|
|
// Default case: return undef (should not be reached)
|
|
self.positionAt(sb.default_bb);
|
|
if (ret_ty != .void_type and result_llvm_ty != null) {
|
|
try phi_vals.append(self.allocator, self.getUndef(result_llvm_ty.?));
|
|
try phi_bbs.append(self.allocator, sb.default_bb);
|
|
}
|
|
self.br(sb.merge_bb);
|
|
|
|
// Merge
|
|
self.positionAt(sb.merge_bb);
|
|
if (ret_ty != .void_type and result_llvm_ty != null) {
|
|
const phi = try self.buildPhiNode(&phi_vals, &phi_bbs, result_llvm_ty.?, "dispatch_phi");
|
|
return phi;
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
/// Extract a concrete typed value from an Any i64 payload.
|
|
fn extractAnyToConcreteType(self: *CodeGen, any_i64: c.LLVMValueRef, sx_type: Type) !c.LLVMValueRef {
|
|
return switch (sx_type) {
|
|
.boolean => self.trunc(any_i64, self.i1Type(), "any_to_bool"),
|
|
.signed => |w| if (w <= 32)
|
|
self.trunc(any_i64, c.LLVMIntTypeInContext(self.context, w), "any_to_int")
|
|
else
|
|
any_i64,
|
|
.unsigned => |w| if (w <= 32)
|
|
self.trunc(any_i64, c.LLVMIntTypeInContext(self.context, w), "any_to_uint")
|
|
else
|
|
any_i64,
|
|
.f32 => blk: {
|
|
const as_f64 = self.bitCast(any_i64, self.f64Type(), "i64_to_f64");
|
|
break :blk c.LLVMBuildFPTrunc(self.builder, as_f64, self.f32Type(), "any_to_f32");
|
|
},
|
|
.f64 => self.bitCast(any_i64, self.f64Type(), "any_to_f64"),
|
|
.string_type => self.loadFromI64Ptr(any_i64, self.getStringStructType(), "any_to_str"),
|
|
.struct_type => |sname| blk: {
|
|
const info = try self.getStructInfo(sname);
|
|
break :blk self.loadFromI64Ptr(any_i64, info.llvm_type, "any_to_struct");
|
|
},
|
|
.enum_type => |ename| blk: {
|
|
const enum_llvm_ty = self.getEnumLLVMType(ename);
|
|
const enum_bits = c.LLVMGetIntTypeWidth(enum_llvm_ty);
|
|
if (enum_bits < 64)
|
|
break :blk self.trunc(any_i64, enum_llvm_ty, "any_to_enum")
|
|
else
|
|
break :blk any_i64;
|
|
},
|
|
.union_type => |uname| blk: {
|
|
const info = try self.getTaggedEnumInfo(uname);
|
|
break :blk self.loadFromI64Ptr(any_i64, info.llvm_type, "any_to_union");
|
|
},
|
|
.vector_type, .array_type => blk: {
|
|
const llvm_ty = self.typeToLLVM(sx_type);
|
|
break :blk self.loadFromI64Ptr(any_i64, llvm_ty, "any_to_vec");
|
|
},
|
|
.slice_type => self.loadFromI64Ptr(any_i64, self.getStringStructType(), "any_to_slice"),
|
|
.pointer_type, .many_pointer_type => self.intToPtr(any_i64, "any_to_ptr"),
|
|
else => any_i64,
|
|
};
|
|
}
|
|
|
|
fn mangleGenericName(
|
|
self: *CodeGen,
|
|
base: []const u8,
|
|
type_params: []const ast.StructTypeParam,
|
|
type_bindings: std.StringHashMap(Type),
|
|
val_bindings: ?std.StringHashMap(i64),
|
|
comptime_nodes: ?std.StringHashMap(*Node),
|
|
) ![]const u8 {
|
|
var buf = std.ArrayList(u8).empty;
|
|
try buf.appendSlice(self.allocator, base);
|
|
try buf.appendSlice(self.allocator, "__");
|
|
for (type_params, 0..) |tp, i| {
|
|
if (i > 0) try buf.append(self.allocator, '_');
|
|
const constraint_name = if (tp.constraint.data == .type_expr) tp.constraint.data.type_expr.name else "";
|
|
if (std.mem.eql(u8, constraint_name, "Type")) {
|
|
if (type_bindings.get(tp.name)) |ty| {
|
|
const name = try ty.displayName(self.allocator);
|
|
try buf.appendSlice(self.allocator, name);
|
|
}
|
|
} else if (comptime_nodes != null) {
|
|
if (comptime_nodes.?.get(tp.name)) |node| {
|
|
if (node.data == .string_literal) {
|
|
const hash = std.hash.Wyhash.hash(0, node.data.string_literal.raw);
|
|
var hash_buf: [16]u8 = undefined;
|
|
const hash_str = std.fmt.bufPrint(&hash_buf, "{x}", .{hash}) catch "0";
|
|
try buf.appendSlice(self.allocator, hash_str);
|
|
} else if (node.data == .int_literal) {
|
|
var int_buf: [20]u8 = undefined;
|
|
const int_str = std.fmt.bufPrint(&int_buf, "{d}", .{node.data.int_literal.value}) catch "0";
|
|
try buf.appendSlice(self.allocator, int_str);
|
|
}
|
|
}
|
|
} else if (val_bindings != null) {
|
|
if (val_bindings.?.get(tp.name)) |val| {
|
|
var tmp: [20]u8 = undefined;
|
|
const s = std.fmt.bufPrint(&tmp, "{d}", .{val}) catch "0";
|
|
try buf.appendSlice(self.allocator, s);
|
|
}
|
|
}
|
|
}
|
|
return try buf.toOwnedSlice(self.allocator);
|
|
}
|
|
|
|
fn instantiateGeneric(self: *CodeGen, fd: ast.FnDecl, bindings: std.StringHashMap(Type), mangled: []const u8) !c.LLVMValueRef {
|
|
// Save current codegen state
|
|
const saved_function = self.current_function;
|
|
const saved_return_type = self.current_return_type;
|
|
const saved_insert_bb = self.getCurrentBlock();
|
|
|
|
// Save named_values
|
|
var saved_named_values = std.StringHashMap(NamedValue).init(self.allocator);
|
|
var nv_iter = self.named_values.iterator();
|
|
while (nv_iter.next()) |entry| {
|
|
try saved_named_values.put(entry.key_ptr.*, entry.value_ptr.*);
|
|
}
|
|
|
|
// Save scope stack — generic body must not pollute caller's scope tracking
|
|
const saved_scope_stack = self.scope_stack;
|
|
self.scope_stack = std.ArrayList(Scope).empty;
|
|
|
|
// Set type param bindings (save/restore to support nested generic instantiation)
|
|
const saved_bindings = self.type_param_bindings;
|
|
self.type_param_bindings = bindings;
|
|
defer self.type_param_bindings = saved_bindings;
|
|
|
|
// Build the specialized function type
|
|
const fn_type = try self.buildFnType(fd.params, fd.return_type, mangled, false);
|
|
const mangled_z = try self.allocator.dupeZ(u8, mangled);
|
|
const function = c.LLVMAddFunction(self.module, mangled_z.ptr, fn_type);
|
|
|
|
// Cache before generating body (in case of recursion)
|
|
try self.generic_instances.put(mangled, function);
|
|
|
|
// Generate body
|
|
self.named_values.clearRetainingCapacity();
|
|
self.current_function = function;
|
|
|
|
_ = self.appendBlock(function, "entry");
|
|
|
|
// Create allocas for parameters
|
|
var llvm_param_idx: u32 = 0;
|
|
for (fd.params) |param| {
|
|
if (param.is_comptime) {
|
|
// Comptime param: create a constant in named_values from the call-site value
|
|
if (self.comptime_param_nodes) |cpn| {
|
|
if (cpn.get(param.name)) |node| {
|
|
if (node.data == .string_literal) {
|
|
const slit = node.data.string_literal;
|
|
const raw = slit.raw;
|
|
const inner = if (!slit.is_raw and raw.len >= 2 and raw[0] == '"' and raw[raw.len - 1] == '"')
|
|
raw[1 .. raw.len - 1]
|
|
else
|
|
raw;
|
|
const content = if (slit.is_raw) inner else try unescape.unescapeString(self.allocator, inner);
|
|
const str_val = self.buildConstStr(content);
|
|
const param_name_z = try self.allocator.dupeZ(u8, param.name);
|
|
const alloca = c.LLVMBuildAlloca(self.builder, self.getStringStructType(), param_name_z.ptr);
|
|
_ = c.LLVMBuildStore(self.builder, str_val, alloca);
|
|
try self.named_values.put(param.name, .{ .ptr = alloca, .ty = .string_type });
|
|
} else if (node.data == .int_literal) {
|
|
const ct_sx_ty = self.resolveType(param.type_expr);
|
|
const ct_llvm_ty = self.typeToLLVM(ct_sx_ty);
|
|
const const_val = c.LLVMConstInt(ct_llvm_ty, @bitCast(node.data.int_literal.value), 0);
|
|
const param_name_z = try self.allocator.dupeZ(u8, param.name);
|
|
const alloca = c.LLVMBuildAlloca(self.builder, ct_llvm_ty, param_name_z.ptr);
|
|
_ = c.LLVMBuildStore(self.builder, const_val, alloca);
|
|
try self.named_values.put(param.name, .{ .ptr = alloca, .ty = ct_sx_ty });
|
|
}
|
|
}
|
|
}
|
|
continue;
|
|
}
|
|
// Variadic params: use slice_type (same as genFnBody)
|
|
const sx_ty = if (param.is_variadic) blk: {
|
|
const elem_name = if (param.type_expr.data == .type_expr) param.type_expr.data.type_expr.name else "s32";
|
|
break :blk Type{ .slice_type = .{ .element_name = elem_name } };
|
|
} else self.resolveType(param.type_expr);
|
|
try self.bindParam(function, param.name, sx_ty, llvm_param_idx);
|
|
llvm_param_idx += 1;
|
|
}
|
|
|
|
// Generate body statements
|
|
const body = fd.body;
|
|
if (body.data != .block) return self.emitError("generic function body must be a block");
|
|
|
|
const ret_sx_type = self.resolveType(fd.return_type);
|
|
self.current_return_type = ret_sx_type;
|
|
|
|
var last_val: c.LLVMValueRef = null;
|
|
for (body.data.block.stmts) |stmt| {
|
|
last_val = try self.genStmt(stmt);
|
|
}
|
|
|
|
// Emit return if current block has no terminator
|
|
const current_bb = self.getCurrentBlock();
|
|
if (c.LLVMGetBasicBlockTerminator(current_bb) == null) {
|
|
if (ret_sx_type == .void_type) {
|
|
self.retVoid();
|
|
} else if (last_val) |val| {
|
|
if (ret_sx_type.isStruct()) {
|
|
const sname = ret_sx_type.struct_type;
|
|
const info = try self.getStructInfo(sname);
|
|
const loaded = c.LLVMBuildLoad2(self.builder, info.llvm_type, val, "retval");
|
|
self.ret(loaded);
|
|
} else {
|
|
const src_ty = self.llvmTypeToSxType(c.LLVMTypeOf(val));
|
|
const converted = self.convertValue(val, src_ty, ret_sx_type);
|
|
self.ret(converted);
|
|
}
|
|
} else {
|
|
_ = c.LLVMBuildUnreachable(self.builder);
|
|
}
|
|
}
|
|
|
|
// Restore codegen state
|
|
self.current_function = saved_function;
|
|
self.current_return_type = saved_return_type;
|
|
if (saved_insert_bb) |bb| {
|
|
self.positionAt(bb);
|
|
}
|
|
self.named_values.clearRetainingCapacity();
|
|
var restore_iter = saved_named_values.iterator();
|
|
while (restore_iter.next()) |entry| {
|
|
try self.named_values.put(entry.key_ptr.*, entry.value_ptr.*);
|
|
}
|
|
saved_named_values.deinit();
|
|
|
|
// Restore scope stack
|
|
self.scope_stack = saved_scope_stack;
|
|
|
|
return function;
|
|
}
|
|
|
|
fn genIfExpr(self: *CodeGen, if_expr: ast.IfExpr) !c.LLVMValueRef {
|
|
// Generate condition
|
|
const cond_val = self.valueToBool(try self.genExpr(if_expr.condition));
|
|
|
|
const has_else = if_expr.else_branch != null;
|
|
|
|
var then_bb = self.appendBB("then");
|
|
var else_bb: c.LLVMBasicBlockRef = if (has_else)
|
|
self.appendBB("else")
|
|
else
|
|
null;
|
|
const merge_bb = self.appendBB("merge");
|
|
|
|
const false_dest = if (has_else) else_bb else merge_bb;
|
|
self.condBr(cond_val, then_bb, false_dest);
|
|
|
|
// Then branch
|
|
self.positionAt(then_bb);
|
|
const then_val = try self.genExpr(if_expr.then_branch);
|
|
then_bb = self.getCurrentBlock(); // may have changed due to nested control flow
|
|
self.br(merge_bb);
|
|
|
|
// Else branch
|
|
var else_val: c.LLVMValueRef = null;
|
|
if (if_expr.else_branch) |else_branch| {
|
|
self.positionAt(else_bb);
|
|
else_val = try self.genExpr(else_branch);
|
|
else_bb = self.getCurrentBlock();
|
|
self.br(merge_bb);
|
|
}
|
|
|
|
// Merge block
|
|
self.positionAt(merge_bb);
|
|
|
|
// PHI node if both branches produced values (skip for void type)
|
|
if (then_val != null and else_val != null) {
|
|
const ty = c.LLVMTypeOf(then_val);
|
|
if (c.LLVMGetTypeKind(ty) != c.LLVMVoidTypeKind) {
|
|
const phi = c.LLVMBuildPhi(self.builder, ty, "iftmp");
|
|
var vals = [2]c.LLVMValueRef{ then_val, else_val };
|
|
var blocks = [2]c.LLVMBasicBlockRef{ then_bb, else_bb };
|
|
c.LLVMAddIncoming(phi, &vals, &blocks, 2);
|
|
return phi;
|
|
}
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
fn genWhileExpr(self: *CodeGen, while_expr: ast.WhileExpr) !c.LLVMValueRef {
|
|
// Create basic blocks: condition, body, after
|
|
const cond_bb = self.appendBB("while.cond");
|
|
const body_bb = self.appendBB("while.body");
|
|
const after_bb = self.appendBB("while.after");
|
|
|
|
// Branch from current block to condition check
|
|
self.br(cond_bb);
|
|
|
|
// Condition block
|
|
self.positionAt(cond_bb);
|
|
const cond_val = self.valueToBool(try self.genExpr(while_expr.condition));
|
|
|
|
self.condBr(cond_val, body_bb, after_bb);
|
|
|
|
// Body block — save and set loop context for break/continue
|
|
self.positionAt(body_bb);
|
|
const saved_break_bb = self.loop_break_bb;
|
|
const saved_continue_bb = self.loop_continue_bb;
|
|
self.loop_break_bb = after_bb;
|
|
self.loop_continue_bb = cond_bb;
|
|
|
|
_ = try self.genExpr(while_expr.body);
|
|
|
|
// Restore loop context
|
|
self.loop_break_bb = saved_break_bb;
|
|
self.loop_continue_bb = saved_continue_bb;
|
|
|
|
// Branch back to condition (if not already terminated by break/return)
|
|
const current_bb = self.getCurrentBlock();
|
|
if (c.LLVMGetBasicBlockTerminator(current_bb) == null) {
|
|
self.br(cond_bb);
|
|
}
|
|
|
|
// Position at after block
|
|
self.positionAt(after_bb);
|
|
|
|
return null;
|
|
}
|
|
|
|
fn genForExpr(self: *CodeGen, for_expr: ast.ForExpr) !c.LLVMValueRef {
|
|
const i64_type = self.i64Type();
|
|
|
|
// Determine iterable type and get length + element access info
|
|
const iter_ty = self.inferType(for_expr.iterable);
|
|
var len_val: c.LLVMValueRef = undefined;
|
|
var elem_ty: Type = Type.s(64);
|
|
var iter_ptr: c.LLVMValueRef = undefined; // pointer to data
|
|
var is_slice = false;
|
|
|
|
if (iter_ty.isSlice()) {
|
|
is_slice = true;
|
|
const info = iter_ty.slice_type;
|
|
elem_ty = Type.fromName(info.element_name) orelse Type.s(64);
|
|
// Load slice value from alloca
|
|
if (for_expr.iterable.data == .identifier) {
|
|
if (self.named_values.get(for_expr.iterable.data.identifier.name)) |entry| {
|
|
const slice_val = c.LLVMBuildLoad2(self.builder, self.getStringStructType(), entry.ptr, "for_slice");
|
|
iter_ptr = self.extractValue(slice_val, 0, "for_ptr");
|
|
len_val = self.extractValue(slice_val, 1, "for_len");
|
|
} else return self.emitError("for: iterable not found");
|
|
} else return self.emitError("for: slice iterable must be a variable");
|
|
} else if (iter_ty.isArray()) {
|
|
const info = iter_ty.array_type;
|
|
elem_ty = Type.fromName(info.element_name) orelse Type.s(64);
|
|
len_val = c.LLVMConstInt(i64_type, info.length, 0);
|
|
// Get pointer to array
|
|
if (for_expr.iterable.data == .identifier) {
|
|
if (self.named_values.get(for_expr.iterable.data.identifier.name)) |entry| {
|
|
iter_ptr = entry.ptr;
|
|
} else return self.emitError("for: iterable not found");
|
|
} else return self.emitError("for: array iterable must be a variable");
|
|
} else {
|
|
return self.emitError("for loop requires a slice or array iterable");
|
|
}
|
|
|
|
const elem_llvm_ty = self.typeToLLVM(elem_ty);
|
|
|
|
// Allocate index variable
|
|
const idx_alloca = self.buildEntryBlockAlloca(i64_type, "for_idx");
|
|
_ = c.LLVMBuildStore(self.builder, c.LLVMConstInt(i64_type, 0, 0), idx_alloca);
|
|
|
|
// Push scope and bind index if requested
|
|
try self.pushScope();
|
|
if (for_expr.index_name) |idx_name| {
|
|
if (!std.mem.eql(u8, idx_name, "_")) {
|
|
try self.named_values.put(idx_name, .{ .ptr = idx_alloca, .ty = Type.s(64), .is_const = true });
|
|
}
|
|
}
|
|
|
|
// Create basic blocks
|
|
const cond_bb = self.appendBB("for.cond");
|
|
const body_bb = self.appendBB("for.body");
|
|
const incr_bb = self.appendBB("for.incr");
|
|
const after_bb = self.appendBB("for.after");
|
|
|
|
self.br(cond_bb);
|
|
|
|
// Condition: index < len
|
|
self.positionAt(cond_bb);
|
|
const cur_idx = c.LLVMBuildLoad2(self.builder, i64_type, idx_alloca, "cur_idx");
|
|
const cond_val = self.icmp(c.LLVMIntSLT, cur_idx, len_val, "for_cond");
|
|
self.condBr(cond_val, body_bb, after_bb);
|
|
|
|
// Body: compute element GEP, bind capture, execute body
|
|
self.positionAt(body_bb);
|
|
const body_idx = c.LLVMBuildLoad2(self.builder, i64_type, idx_alloca, "body_idx");
|
|
|
|
const elem_gep = if (is_slice)
|
|
self.gepPointerElement(elem_llvm_ty, iter_ptr, body_idx, "for_elem")
|
|
else blk: {
|
|
const arr_llvm_ty = self.typeToLLVM(iter_ty);
|
|
const zero = c.LLVMConstInt(i64_type, 0, 0);
|
|
var indices = [_]c.LLVMValueRef{ zero, body_idx };
|
|
break :blk c.LLVMBuildGEP2(self.builder, arr_llvm_ty, iter_ptr, &indices, 2, "for_elem");
|
|
};
|
|
|
|
if (!std.mem.eql(u8, for_expr.capture_name, "_")) {
|
|
// Alias mode: capture points directly to element in array/slice
|
|
try self.named_values.put(for_expr.capture_name, .{
|
|
.ptr = elem_gep,
|
|
.ty = elem_ty,
|
|
.is_const = true,
|
|
});
|
|
}
|
|
|
|
// Save and set loop context for break/continue
|
|
const saved_break_bb = self.loop_break_bb;
|
|
const saved_continue_bb = self.loop_continue_bb;
|
|
self.loop_break_bb = after_bb;
|
|
self.loop_continue_bb = incr_bb;
|
|
|
|
_ = try self.genExpr(for_expr.body);
|
|
|
|
self.loop_break_bb = saved_break_bb;
|
|
self.loop_continue_bb = saved_continue_bb;
|
|
|
|
// Fall through to increment block
|
|
const current_bb = self.getCurrentBlock();
|
|
if (c.LLVMGetBasicBlockTerminator(current_bb) == null) {
|
|
self.br(incr_bb);
|
|
}
|
|
|
|
// Increment index, then branch back to condition
|
|
self.positionAt(incr_bb);
|
|
const inc_idx = c.LLVMBuildLoad2(self.builder, i64_type, idx_alloca, "inc_idx");
|
|
const next_idx = c.LLVMBuildAdd(self.builder, inc_idx, c.LLVMConstInt(i64_type, 1, 0), "next_idx");
|
|
_ = c.LLVMBuildStore(self.builder, next_idx, idx_alloca);
|
|
self.br(cond_bb);
|
|
|
|
self.positionAt(after_bb);
|
|
|
|
try self.popScope();
|
|
|
|
return null;
|
|
}
|
|
|
|
fn genEnumLiteral(self: *CodeGen, variant_name: []const u8, enum_type_name: []const u8) c.LLVMValueRef {
|
|
const enum_ty = self.getEnumLLVMType(enum_type_name);
|
|
const variants = self.lookupEnumVariants(enum_type_name) orelse return c.LLVMConstInt(enum_ty, 0, 0);
|
|
const values = self.enum_variant_values.get(enum_type_name);
|
|
for (variants, 0..) |v, i| {
|
|
if (std.mem.eql(u8, v, variant_name)) {
|
|
const val: u64 = if (values) |vals| @bitCast(vals[i]) else @intCast(i);
|
|
return c.LLVMConstInt(enum_ty, val, 0);
|
|
}
|
|
}
|
|
return c.LLVMConstInt(enum_ty, 0, 0);
|
|
}
|
|
|
|
fn lookupVariantValue(self: *CodeGen, enum_name: ?[]const u8, variants: ?[]const []const u8, name: []const u8) u64 {
|
|
if (variants) |vs| {
|
|
for (vs, 0..) |v, i| {
|
|
if (std.mem.eql(u8, v, name)) {
|
|
// Use resolved values if available (flags enums, explicit values)
|
|
if (enum_name) |en| {
|
|
if (self.enum_variant_values.get(en)) |vals| {
|
|
return @bitCast(vals[i]);
|
|
}
|
|
}
|
|
return i;
|
|
}
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
fn genMatchExpr(self: *CodeGen, match: ast.MatchExpr) !c.LLVMValueRef {
|
|
// Determine subject type for enum vs union dispatch
|
|
var enum_name: ?[]const u8 = null;
|
|
var union_name: ?[]const u8 = null;
|
|
const subject_ty = self.inferType(match.subject);
|
|
if (subject_ty.isEnum()) enum_name = subject_ty.enum_type;
|
|
if (subject_ty.isUnion()) union_name = subject_ty.union_type;
|
|
|
|
// Get the switch value: for unions, load the tag from field 0; for enums, use the value directly
|
|
const subject_val: c.LLVMValueRef = if (union_name != null) blk: {
|
|
// Union: load tag from field 0 of the alloca
|
|
const entry = self.named_values.get(match.subject.data.identifier.name).?;
|
|
const info = self.lookupTaggedEnumInfo(union_name.?).?;
|
|
break :blk self.loadStructField(info.llvm_type, entry.ptr, 0, self.getEnumLLVMType(union_name.?));
|
|
} else try self.genExpr(match.subject);
|
|
|
|
const variants: ?[]const []const u8 = if (union_name) |un|
|
|
(if (self.lookupTaggedEnumInfo(un)) |info| info.variant_names else null)
|
|
else if (enum_name) |en|
|
|
self.lookupEnumVariants(en)
|
|
else
|
|
null;
|
|
|
|
const i64_type = self.i64Type();
|
|
// Enum/union case constants use the backing type; Any dispatch uses i64
|
|
const case_int_type = if (enum_name) |en| self.getEnumLLVMType(en) else if (union_name) |un| self.getEnumLLVMType(un) else i64_type;
|
|
const merge_bb = self.appendBB("match_end");
|
|
|
|
// Create case basic blocks
|
|
var case_bbs = std.ArrayList(c.LLVMBasicBlockRef).empty;
|
|
for (match.arms) |_| {
|
|
try case_bbs.append(self.allocator, self.appendBB("case"));
|
|
}
|
|
|
|
// Find else arm (null pattern) — use its BB as the switch default
|
|
var else_arm_idx: ?usize = null;
|
|
for (match.arms, 0..) |arm, i| {
|
|
if (arm.pattern == null) {
|
|
else_arm_idx = i;
|
|
break;
|
|
}
|
|
}
|
|
const default_bb = if (else_arm_idx) |idx|
|
|
case_bbs.items[idx]
|
|
else
|
|
self.appendBB("match_default");
|
|
|
|
// Build switch instruction
|
|
const sw = c.LLVMBuildSwitch(self.builder, subject_val, default_bb, @intCast(match.arms.len));
|
|
for (match.arms, 0..) |arm, i| {
|
|
const pat = arm.pattern orelse continue; // skip else arm
|
|
if (pat.data == .enum_literal) {
|
|
const idx = self.lookupVariantValue(enum_name orelse union_name, variants, pat.data.enum_literal.name);
|
|
const case_val = c.LLVMConstInt(case_int_type, idx, 0);
|
|
c.LLVMAddCase(sw, case_val, case_bbs.items[i]);
|
|
} else if (pat.data == .type_expr) {
|
|
// Type-match: resolve type name to Any tag value(s)
|
|
const tag_values = try self.resolveTypeMatchTags(pat.data.type_expr.name);
|
|
for (tag_values) |tag| {
|
|
c.LLVMAddCase(sw, c.LLVMConstInt(i64_type, tag, 0), case_bbs.items[i]);
|
|
}
|
|
} else if (pat.data == .identifier) {
|
|
// Named type (struct/enum/union name) or category (int/float)
|
|
const tag_values = try self.resolveTypeMatchTags(pat.data.identifier.name);
|
|
for (tag_values) |tag| {
|
|
c.LLVMAddCase(sw, c.LLVMConstInt(i64_type, tag, 0), case_bbs.items[i]);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Generate arm bodies and collect PHI info
|
|
var phi_vals = std.ArrayList(c.LLVMValueRef).empty;
|
|
var phi_bbs = std.ArrayList(c.LLVMBasicBlockRef).empty;
|
|
var has_value = false;
|
|
var value_type: c.LLVMTypeRef = null;
|
|
|
|
// Pre-collect tag values for each arm (for runtime dispatch context)
|
|
var arm_tag_values = std.ArrayList([]const u64).empty;
|
|
for (match.arms) |arm| {
|
|
const tag_values: []const u64 = if (arm.pattern) |pat| blk: {
|
|
break :blk if (pat.data == .type_expr)
|
|
try self.resolveTypeMatchTags(pat.data.type_expr.name)
|
|
else if (pat.data == .identifier)
|
|
try self.resolveTypeMatchTags(pat.data.identifier.name)
|
|
else
|
|
&.{};
|
|
} else &.{};
|
|
try arm_tag_values.append(self.allocator, tag_values);
|
|
}
|
|
|
|
for (match.arms, 0..) |arm, i| {
|
|
self.positionAt(case_bbs.items[i]);
|
|
if (arm.is_break) {
|
|
self.br(merge_bb);
|
|
} else if (arm.pattern != null and arm_tag_values.items[i].len == 0 and
|
|
(arm.pattern.?.data == .identifier or arm.pattern.?.data == .type_expr))
|
|
{
|
|
// Category/type arm with no matching types — BB is unreachable, skip body
|
|
self.br(merge_bb);
|
|
} else {
|
|
// Payload capture: bind variant payload as a local variable
|
|
if (arm.capture) |cap_name| {
|
|
if (union_name) |un| {
|
|
const uinfo = self.lookupTaggedEnumInfo(un).?;
|
|
const pat = arm.pattern.?;
|
|
if (pat.data == .enum_literal) {
|
|
const vname = pat.data.enum_literal.name;
|
|
var vidx: ?usize = null;
|
|
for (uinfo.variant_names, 0..) |vn, vi| {
|
|
if (std.mem.eql(u8, vn, vname)) { vidx = vi; break; }
|
|
}
|
|
if (vidx) |vi| {
|
|
const variant_ty = uinfo.variant_types[vi];
|
|
if (variant_ty != .void_type) {
|
|
const subject_entry = self.named_values.get(match.subject.data.identifier.name).?;
|
|
const payload_gep = self.structGEP(uinfo.llvm_type, subject_entry.ptr, uinfo.payload_field_index, "cap_payload");
|
|
const payload_llvm_ty = self.typeToLLVM(variant_ty);
|
|
const payload_val = c.LLVMBuildLoad2(self.builder, payload_llvm_ty, payload_gep, "cap_load");
|
|
const cap_alloca = c.LLVMBuildAlloca(self.builder, payload_llvm_ty, @ptrCast(cap_name.ptr));
|
|
_ = c.LLVMBuildStore(self.builder, payload_val, cap_alloca);
|
|
try self.named_values.put(cap_name, .{ .ptr = cap_alloca, .ty = variant_ty });
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
// Set match arm context for runtime type dispatch
|
|
const saved_match_tags = self.current_match_tags;
|
|
self.current_match_tags = arm_tag_values.items[i];
|
|
const val = try self.genExpr(arm.body);
|
|
self.current_match_tags = saved_match_tags;
|
|
const bb = self.getCurrentBlock();
|
|
self.br(merge_bb);
|
|
if (val != null and c.LLVMGetTypeKind(c.LLVMTypeOf(val)) != c.LLVMVoidTypeKind) {
|
|
has_value = true;
|
|
if (value_type == null) value_type = c.LLVMTypeOf(val);
|
|
try phi_vals.append(self.allocator, val);
|
|
try phi_bbs.append(self.allocator, bb);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Default block branches to merge (only if no else arm — else arm's body already generated above)
|
|
if (else_arm_idx == null) {
|
|
self.positionAt(default_bb);
|
|
self.br(merge_bb);
|
|
}
|
|
|
|
// Merge block
|
|
self.positionAt(merge_bb);
|
|
|
|
if (has_value and value_type != null) {
|
|
const undef_val = self.getUndef(value_type);
|
|
// Add undef entries for break arms and default block
|
|
for (match.arms, 0..) |arm, i| {
|
|
if (arm.is_break) {
|
|
try phi_vals.append(self.allocator, undef_val);
|
|
try phi_bbs.append(self.allocator, case_bbs.items[i]);
|
|
}
|
|
}
|
|
if (else_arm_idx == null) {
|
|
try phi_vals.append(self.allocator, undef_val);
|
|
try phi_bbs.append(self.allocator, default_bb);
|
|
}
|
|
|
|
const phi = try self.buildPhiNode(&phi_vals, &phi_bbs, value_type, "matchtmp");
|
|
return phi;
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
/// Resolve a type name to one or more Any tag values for type-switch matching.
|
|
/// Categories: "int" matches s32+s64, "float" matches f32+f64.
|
|
/// Specific types: "s32", "f64", "string", "bool", "Type".
|
|
/// Named types: struct/enum/union names get dynamic IDs.
|
|
fn resolveTypeMatchTags(self: *CodeGen, name: []const u8) ![]const u64 {
|
|
// Category aliases
|
|
if (std.mem.eql(u8, name, "int")) {
|
|
const tags = try self.allocator.alloc(u64, 2);
|
|
tags[0] = ANY_TAG_S32;
|
|
tags[1] = ANY_TAG_S64;
|
|
return tags;
|
|
}
|
|
if (std.mem.eql(u8, name, "float")) {
|
|
const tags = try self.allocator.alloc(u64, 2);
|
|
tags[0] = ANY_TAG_F32;
|
|
tags[1] = ANY_TAG_F64;
|
|
return tags;
|
|
}
|
|
// Type category aliases: "struct", "enum", "union", "vector", "array", "slice"
|
|
const category: ?TypeCategory = if (std.mem.eql(u8, name, "struct"))
|
|
.struct_cat
|
|
else if (std.mem.eql(u8, name, "enum"))
|
|
.enum_cat
|
|
else if (std.mem.eql(u8, name, "union"))
|
|
.enum_cat
|
|
else if (std.mem.eql(u8, name, "vector"))
|
|
.vector_cat
|
|
else if (std.mem.eql(u8, name, "array"))
|
|
.array_cat
|
|
else if (std.mem.eql(u8, name, "slice"))
|
|
.slice_cat
|
|
else if (std.mem.eql(u8, name, "pointer"))
|
|
.pointer_cat
|
|
else
|
|
null;
|
|
if (category) |cat| {
|
|
var tag_list = std.ArrayList(u64).empty;
|
|
var it = self.any_type_entries.iterator();
|
|
while (it.next()) |entry| {
|
|
if (entry.value_ptr.category == cat) {
|
|
try tag_list.append(self.allocator, entry.value_ptr.tag_id);
|
|
}
|
|
}
|
|
if (tag_list.items.len > 0) {
|
|
return try tag_list.toOwnedSlice(self.allocator);
|
|
}
|
|
// No types registered for this category — return empty slice
|
|
return &.{};
|
|
}
|
|
// Specific builtin types
|
|
const single_tag: ?u64 = if (std.mem.eql(u8, name, "bool"))
|
|
ANY_TAG_BOOL
|
|
else if (std.mem.eql(u8, name, "s32"))
|
|
ANY_TAG_S32
|
|
else if (std.mem.eql(u8, name, "s64"))
|
|
ANY_TAG_S64
|
|
else if (std.mem.eql(u8, name, "f32"))
|
|
ANY_TAG_F32
|
|
else if (std.mem.eql(u8, name, "f64"))
|
|
ANY_TAG_F64
|
|
else if (std.mem.eql(u8, name, "string"))
|
|
ANY_TAG_STRING
|
|
else if (std.mem.eql(u8, name, "Type") or std.mem.eql(u8, name, "type"))
|
|
ANY_TAG_TYPE
|
|
else if (std.mem.eql(u8, name, "void"))
|
|
ANY_TAG_VOID
|
|
else
|
|
null;
|
|
if (single_tag) |t| {
|
|
const tags = try self.allocator.alloc(u64, 1);
|
|
tags[0] = t;
|
|
return tags;
|
|
}
|
|
// Named type (struct/enum/union) — get dynamic ID
|
|
const sx_type: Type = if (self.type_registry.get(name)) |e| switch (e) {
|
|
.struct_info => Type{ .struct_type = name },
|
|
.plain_enum => Type{ .enum_type = name },
|
|
.tagged_enum => Type{ .union_type = name },
|
|
.union_info => Type{ .union_type = name },
|
|
.alias => Type{ .struct_type = name },
|
|
} else .{ .struct_type = name }; // fallback
|
|
const id = try self.getAnyTypeId(name, sx_type);
|
|
const tags = try self.allocator.alloc(u64, 1);
|
|
tags[0] = id;
|
|
return tags;
|
|
}
|
|
|
|
/// Resolve a callee node to a function name string for type inference.
|
|
/// Handles identifiers, namespaced calls, and intra-namespace fallback.
|
|
fn resolveCalleeName(self: *CodeGen, call_node: ast.Call) ?[]const u8 {
|
|
if (call_node.callee.data == .identifier) {
|
|
return call_node.callee.data.identifier.name;
|
|
}
|
|
if (call_node.callee.data == .field_access) {
|
|
const fa = call_node.callee.data.field_access;
|
|
if (fa.object.data == .identifier) {
|
|
if (self.namespaces.contains(fa.object.data.identifier.name)) {
|
|
return std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ fa.object.data.identifier.name, fa.field }) catch return null;
|
|
}
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
/// Resolve a builtin parameterized type (e.g. Vector(3, f32)).
|
|
/// Strips namespace prefix to get base name, then dispatches.
|
|
fn resolveBuiltinType(self: *CodeGen, name: []const u8, args: []const *Node) ?Type {
|
|
const base = baseName(name);
|
|
if (std.mem.eql(u8, base, "Vector")) {
|
|
if (args.len >= 2) {
|
|
const n: u32 = @intCast(self.resolveValueArg(args[0]));
|
|
const elem = self.resolveType(args[1]);
|
|
const elem_name = elem.displayName(self.allocator) catch return null;
|
|
const ty: Type = .{ .vector_type = .{ .element_name = elem_name, .length = n } };
|
|
// Pre-register in any_type_entries so runtime dispatch knows about this type
|
|
const any_name = std.fmt.allocPrint(self.allocator, "vec[{d}]{s}", .{ n, elem_name }) catch return null;
|
|
_ = self.getAnyTypeId(any_name, ty) catch return null;
|
|
return ty;
|
|
}
|
|
}
|
|
if (std.mem.eql(u8, base, "Array")) {
|
|
if (args.len >= 2) {
|
|
const n: u32 = @intCast(self.resolveValueArg(args[0]));
|
|
const elem = self.resolveType(args[1]);
|
|
const elem_name = elem.displayName(self.allocator) catch return null;
|
|
const ty: Type = .{ .array_type = .{ .element_name = elem_name, .length = n } };
|
|
const any_name = std.fmt.allocPrint(self.allocator, "[{d}]{s}", .{ n, elem_name }) catch return null;
|
|
_ = self.getAnyTypeId(any_name, ty) catch return null;
|
|
return ty;
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
fn dispatchBuiltin(self: *CodeGen, name: []const u8, call_node: ast.Call) !c.LLVMValueRef {
|
|
// Extract base name (strip namespace prefix)
|
|
const base = baseName(name);
|
|
if (std.mem.eql(u8, base, "write")) return self.genWriteCall(call_node.args);
|
|
if (std.mem.eql(u8, base, "sqrt")) return self.genMathIntrinsic(call_node, "sqrt");
|
|
if (std.mem.eql(u8, base, "sin")) return self.genMathIntrinsic(call_node, "sin");
|
|
if (std.mem.eql(u8, base, "cos")) return self.genMathIntrinsic(call_node, "cos");
|
|
if (std.mem.eql(u8, base, "size_of")) return self.genSizeOf(call_node);
|
|
if (std.mem.eql(u8, base, "cast")) return self.genCast(call_node);
|
|
if (std.mem.eql(u8, base, "alloc")) return self.genAlloc(call_node.args);
|
|
if (std.mem.eql(u8, base, "malloc")) return self.genMalloc(call_node.args);
|
|
if (std.mem.eql(u8, base, "free")) return self.genFree(call_node.args);
|
|
if (std.mem.eql(u8, base, "memcpy")) return self.genMemcpy(call_node.args);
|
|
if (std.mem.eql(u8, base, "type_of")) return self.genTypeOf(call_node);
|
|
if (std.mem.eql(u8, base, "type_name")) return self.genTypeName(call_node);
|
|
if (std.mem.eql(u8, base, "field_count")) return self.genFieldCount(call_node);
|
|
if (std.mem.eql(u8, base, "field_name")) return self.genFieldName(call_node);
|
|
if (std.mem.eql(u8, base, "field_value")) return self.genFieldValue(call_node);
|
|
if (std.mem.eql(u8, base, "is_flags")) return self.genIsFlags(call_node);
|
|
if (std.mem.eql(u8, base, "field_value_int")) return self.genFieldValueInt(call_node);
|
|
if (std.mem.eql(u8, base, "field_index")) return self.genFieldIndex(call_node);
|
|
return self.emitErrorFmt("unknown builtin function '{s}'", .{name});
|
|
}
|
|
|
|
fn genWriteCall(self: *CodeGen, args: []const *Node) !c.LLVMValueRef {
|
|
if (args.len != 1) return self.emitError("write expects exactly 1 argument");
|
|
const builtins = try self.requireBuiltins();
|
|
const val = try self.genExpr(args[0]);
|
|
// Extract ptr and len from string slice
|
|
const ptr = self.extractValue(val, 0, "str_ptr");
|
|
const len_i64 = self.extractValue(val, 1, "str_len");
|
|
// printf %.*s precision is C int (i32) — truncate from i64
|
|
const len = self.trunc(len_i64, self.i32Type(), "len_trunc");
|
|
const fmt = self.buildGlobalString("%.*s", "write_fmt");
|
|
const printf_fn = builtins.printf_fn;
|
|
const fn_type = c.LLVMGlobalGetValueType(printf_fn);
|
|
var call_args = [_]c.LLVMValueRef{ fmt, len, ptr };
|
|
_ = c.LLVMBuildCall2(self.builder, fn_type, printf_fn, &call_args, 3, "");
|
|
return null;
|
|
}
|
|
|
|
/// Helper: build a constant string slice in the current function
|
|
fn buildConstStr(self: *CodeGen, s: []const u8) c.LLVMValueRef {
|
|
const sz = self.allocator.dupeZ(u8, s) catch unreachable;
|
|
const ptr = self.buildGlobalString(sz.ptr, "cstr");
|
|
return self.buildStringSlice(ptr, self.constInt64(@intCast(s.len)));
|
|
}
|
|
|
|
/// Helper: build a constant string slice as a global constant (no builder needed).
|
|
fn buildConstStrGlobal(self: *CodeGen, s: []const u8) c.LLVMValueRef {
|
|
const sz = self.allocator.dupeZ(u8, s) catch unreachable;
|
|
const i64_ty = self.i64Type();
|
|
const i8_ty = self.i8Type();
|
|
// Create a global string constant
|
|
const str_const = c.LLVMConstStringInContext(self.context, sz.ptr, @intCast(s.len), 0);
|
|
const global_name = (self.allocator.dupeZ(u8, std.fmt.allocPrint(self.allocator, ".str.{s}", .{s}) catch unreachable)) catch unreachable;
|
|
var global = c.LLVMGetNamedGlobal(self.module, global_name.ptr);
|
|
if (global == null) {
|
|
const arr_ty = c.LLVMArrayType2(i8_ty, s.len + 1);
|
|
global = c.LLVMAddGlobal(self.module, arr_ty, global_name.ptr);
|
|
c.LLVMSetInitializer(global, str_const);
|
|
c.LLVMSetGlobalConstant(global, 1);
|
|
c.LLVMSetLinkage(global, c.LLVMPrivateLinkage);
|
|
}
|
|
// Build constant struct {ptr, i64}
|
|
var fields = [_]c.LLVMValueRef{
|
|
c.LLVMConstBitCast(global.?, self.ptrType()),
|
|
c.LLVMConstInt(i64_ty, s.len, 0),
|
|
};
|
|
return c.LLVMConstStructInContext(self.context, &fields, 2, 0);
|
|
}
|
|
|
|
/// Check if a node refers to a type name. Returns the raw name or null.
|
|
fn asTypeName(self: *CodeGen, node: *const Node) ?[]const u8 {
|
|
if (node.data == .type_expr) return node.data.type_expr.name;
|
|
if (node.data == .identifier) {
|
|
const id = node.data.identifier.name;
|
|
if (self.resolveTypeName(id) != null) return id;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
/// Resolve a type name to its display string (null-terminated) for runtime use.
|
|
fn resolveDisplayName(self: *CodeGen, name: []const u8) [:0]const u8 {
|
|
const display = self.resolveTypeName(name) orelse name;
|
|
return self.allocator.dupeZ(u8, display) catch unreachable;
|
|
}
|
|
|
|
/// Convert a type expression AST node to its source string representation.
|
|
fn typeNodeToString(self: *CodeGen, node: *const Node) []const u8 {
|
|
return switch (node.data) {
|
|
.type_expr => |te| te.name,
|
|
.identifier => |id| id.name,
|
|
.pointer_type_expr => |pte| std.fmt.allocPrint(self.allocator, "*{s}", .{self.typeNodeToString(pte.pointee_type)}) catch "?",
|
|
.many_pointer_type_expr => |mpte| std.fmt.allocPrint(self.allocator, "[*]{s}", .{self.typeNodeToString(mpte.element_type)}) catch "?",
|
|
.slice_type_expr => |ste| std.fmt.allocPrint(self.allocator, "[]{s}", .{self.typeNodeToString(ste.element_type)}) catch "?",
|
|
.array_type_expr => |ate| blk: {
|
|
const elem = self.typeNodeToString(ate.element_type);
|
|
if (ate.length.data == .int_literal) {
|
|
break :blk std.fmt.allocPrint(self.allocator, "[{d}]{s}", .{ ate.length.data.int_literal.value, elem }) catch "?";
|
|
}
|
|
break :blk std.fmt.allocPrint(self.allocator, "[?]{s}", .{elem}) catch "?";
|
|
},
|
|
else => "?",
|
|
};
|
|
}
|
|
|
|
/// Build a function type string like "() -> s32" from an fn_decl.
|
|
fn buildFnSignature(self: *CodeGen, fd: ast.FnDecl) []const u8 {
|
|
var buf = std.ArrayList(u8).empty;
|
|
buf.appendSlice(self.allocator, "(") catch return "?";
|
|
for (fd.params, 0..) |param, i| {
|
|
if (i > 0) buf.appendSlice(self.allocator, ", ") catch {};
|
|
if (param.is_variadic) buf.appendSlice(self.allocator, "..") catch {};
|
|
const ty_str = self.typeNodeToString(param.type_expr);
|
|
buf.appendSlice(self.allocator, ty_str) catch {};
|
|
}
|
|
buf.appendSlice(self.allocator, ")") catch {};
|
|
if (fd.return_type) |rt| {
|
|
buf.appendSlice(self.allocator, " -> ") catch {};
|
|
buf.appendSlice(self.allocator, self.typeNodeToString(rt)) catch {};
|
|
}
|
|
return buf.toOwnedSlice(self.allocator) catch "?";
|
|
}
|
|
|
|
/// Extract a qualified name from a callee expression (identifier or field_access chain).
|
|
fn calleeToQualifiedName(self: *CodeGen, callee: *Node) ?[]const u8 {
|
|
if (callee.data == .identifier) return callee.data.identifier.name;
|
|
if (callee.data == .field_access) {
|
|
const obj_name = self.calleeToQualifiedName(callee.data.field_access.object) orelse return null;
|
|
return std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ obj_name, callee.data.field_access.field }) catch null;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
/// Resolve a name to a type display string, or null if not a type.
|
|
fn resolveTypeName(self: *CodeGen, name: []const u8) ?[]const u8 {
|
|
const resolved = self.resolveAlias(name);
|
|
if (self.lookupStructInfo(resolved)) |info| return info.display_name orelse resolved;
|
|
if (self.resolveTypeFromName(name) != null) return resolved;
|
|
return null;
|
|
}
|
|
|
|
/// Resolve a type name to a Type, checking primitives + registered structs/unions/enums.
|
|
/// Unlike Type.fromName which only handles primitives.
|
|
fn resolveTypeFromName(self: *CodeGen, name: []const u8) ?Type {
|
|
// Primitives
|
|
if (Type.fromName(name)) |t| return t;
|
|
// Unified type registry lookup
|
|
if (self.type_registry.get(name)) |entry| switch (entry) {
|
|
.struct_info => return .{ .struct_type = name },
|
|
.tagged_enum => return .{ .union_type = name },
|
|
.union_info => return .{ .union_type = name },
|
|
.plain_enum => return .{ .enum_type = name },
|
|
.alias => |target| {
|
|
if (Type.fromName(target)) |t| return t;
|
|
if (self.type_registry.get(target)) |inner| switch (inner) {
|
|
.struct_info => return .{ .struct_type = target },
|
|
.tagged_enum => return .{ .union_type = target },
|
|
.union_info => return .{ .union_type = target },
|
|
.plain_enum => return .{ .enum_type = target },
|
|
.alias => {},
|
|
};
|
|
},
|
|
};
|
|
// Vector display name: "Vector(N,T)"
|
|
if (name.len > 8 and std.mem.startsWith(u8, name, "Vector(") and name[name.len - 1] == ')') {
|
|
const inner = name[7 .. name.len - 1]; // "N,T"
|
|
if (std.mem.indexOfScalar(u8, inner, ',')) |comma| {
|
|
const n_str = inner[0..comma];
|
|
const elem = inner[comma + 1 ..];
|
|
const length = std.fmt.parseInt(u32, n_str, 10) catch return null;
|
|
return .{ .vector_type = .{ .element_name = elem, .length = length } };
|
|
}
|
|
}
|
|
// Type aliases
|
|
if (self.lookupAlias(name)) |target| return self.resolveTypeFromName(target);
|
|
return null;
|
|
}
|
|
|
|
fn inferType(self: *CodeGen, node: *Node) Type {
|
|
return switch (node.data) {
|
|
.int_literal => Type.s(64),
|
|
.float_literal => .f32,
|
|
.bool_literal => .boolean,
|
|
.string_literal => .string_type,
|
|
.insert_expr => .void_type,
|
|
.comptime_expr => |ct| self.inferType(ct.expr),
|
|
.binary_op => |binop| {
|
|
switch (binop.op) {
|
|
.eq, .neq, .lt, .lte, .gt, .gte, .and_op, .or_op => return .boolean,
|
|
else => {
|
|
const lhs_ty = self.inferType(binop.lhs);
|
|
const rhs_ty = self.inferType(binop.rhs);
|
|
return Type.widen(lhs_ty, rhs_ty);
|
|
},
|
|
}
|
|
},
|
|
.chained_comparison => return .boolean,
|
|
.identifier => |ident| {
|
|
if (self.lookupValue(ident.name)) |v| return v.ty();
|
|
return Type.s(64);
|
|
},
|
|
.if_expr => |ie| {
|
|
return self.inferType(ie.then_branch);
|
|
},
|
|
.block => |blk| {
|
|
if (blk.stmts.len > 0) {
|
|
return self.inferType(blk.stmts[blk.stmts.len - 1]);
|
|
}
|
|
return .void_type;
|
|
},
|
|
.enum_literal => {
|
|
if (self.current_return_type.isEnum()) return self.current_return_type;
|
|
if (self.current_return_type.isUnion()) return self.current_return_type;
|
|
return .{ .enum_type = "" };
|
|
},
|
|
.match_expr => |me| {
|
|
for (me.arms) |arm| {
|
|
if (!arm.is_break) return self.inferType(arm.body);
|
|
}
|
|
return .void_type;
|
|
},
|
|
.call => |call_node| {
|
|
// Check for union literal pattern: Type.variant(payload)
|
|
if (call_node.callee.data == .field_access) {
|
|
const fa = call_node.callee.data.field_access;
|
|
const obj_ty = blk: {
|
|
if (fa.object.data == .identifier) {
|
|
const name = self.resolveAlias(fa.object.data.identifier.name);
|
|
if (self.lookupTaggedEnumInfo(name) != null) break :blk Type{ .union_type = name };
|
|
}
|
|
const ty = self.resolveType(fa.object);
|
|
if (ty.isUnion()) break :blk ty;
|
|
break :blk @as(?Type, null);
|
|
};
|
|
if (obj_ty) |uty| return uty;
|
|
}
|
|
const callee_name = self.resolveCalleeName(call_node) orelse return Type.s(64);
|
|
const base_name = baseName(callee_name);
|
|
// Built-in: sqrt/sin/cos returns same type as argument
|
|
if (std.mem.eql(u8, base_name, "sqrt") or
|
|
std.mem.eql(u8, base_name, "sin") or
|
|
std.mem.eql(u8, base_name, "cos"))
|
|
{
|
|
if (call_node.args.len > 0) return self.inferType(call_node.args[0]);
|
|
return .f32;
|
|
}
|
|
// Built-in: size_of returns s64
|
|
if (std.mem.eql(u8, base_name, "size_of")) return Type.s(64);
|
|
// Built-in: type_of returns s64 (type tag)
|
|
if (std.mem.eql(u8, base_name, "type_of")) return Type.s(64);
|
|
// Built-in: type_name returns string
|
|
if (std.mem.eql(u8, base_name, "type_name")) return .string_type;
|
|
// Built-in: field_count returns s64
|
|
if (std.mem.eql(u8, base_name, "field_count")) return Type.s(64);
|
|
// Built-in: field_name returns string
|
|
if (std.mem.eql(u8, base_name, "field_name")) return .string_type;
|
|
// Built-in: field_value returns Any
|
|
if (std.mem.eql(u8, base_name, "field_value")) return .{ .any_type = {} };
|
|
// Built-in: is_flags returns bool
|
|
if (std.mem.eql(u8, base_name, "is_flags")) return .boolean;
|
|
// Built-in: field_value_int returns s64
|
|
if (std.mem.eql(u8, base_name, "field_value_int")) return Type.s(64);
|
|
// Built-in: field_index returns s64
|
|
if (std.mem.eql(u8, base_name, "field_index")) return Type.s(64);
|
|
// Built-in: cast returns the target type (first arg)
|
|
if (std.mem.eql(u8, base_name, "cast")) {
|
|
if (call_node.args.len > 0) return self.resolveType(call_node.args[0]);
|
|
return Type.s(64);
|
|
}
|
|
// Built-in: alloc returns string
|
|
if (std.mem.eql(u8, base_name, "alloc")) return .string_type;
|
|
if (std.mem.eql(u8, base_name, "malloc")) return .{ .pointer_type = .{ .pointee_name = "void" } };
|
|
if (std.mem.eql(u8, base_name, "free")) return .void_type;
|
|
if (std.mem.eql(u8, base_name, "memcpy")) return .void_type;
|
|
// Check generic templates — infer return type from widened bindings
|
|
const template = self.generic_templates.get(callee_name) orelse blk: {
|
|
// Intra-namespace fallback
|
|
if (self.current_namespace) |ns| {
|
|
const qualified = std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns, callee_name }) catch return Type.s(64);
|
|
break :blk self.generic_templates.get(qualified);
|
|
}
|
|
break :blk null;
|
|
};
|
|
if (template) |tmpl| {
|
|
const gfd = tmpl;
|
|
// Build widened type bindings from all call args
|
|
var inferred_bindings = std.StringHashMap(Type).init(self.allocator);
|
|
for (gfd.params, 0..) |param, pi| {
|
|
if (param.type_expr.data == .type_expr) {
|
|
for (gfd.type_params) |tp| {
|
|
if (std.mem.eql(u8, tp.name, param.type_expr.data.type_expr.name)) {
|
|
if (pi < call_node.args.len) {
|
|
const arg_ty = self.inferType(call_node.args[pi]);
|
|
if (inferred_bindings.get(tp.name)) |existing| {
|
|
inferred_bindings.put(tp.name, Type.widen(existing, arg_ty)) catch {};
|
|
} else {
|
|
inferred_bindings.put(tp.name, arg_ty) catch {};
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
// Resolve return type from bindings
|
|
if (gfd.return_type) |rt| {
|
|
if (rt.data == .type_expr) {
|
|
if (inferred_bindings.get(rt.data.type_expr.name)) |bound_ty| {
|
|
return bound_ty;
|
|
}
|
|
}
|
|
// Try resolving as a concrete type (e.g. -> string, -> s32)
|
|
const resolved = self.resolveType(rt);
|
|
if (!std.meta.eql(resolved, Type.void_type)) return resolved;
|
|
}
|
|
return Type.s(64);
|
|
}
|
|
// Check declared return types (preserves signedness)
|
|
if (self.function_return_types.get(callee_name)) |ret_ty| return ret_ty;
|
|
if (self.current_namespace) |ns| {
|
|
const qualified = std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns, callee_name }) catch return Type.s(64);
|
|
if (self.function_return_types.get(qualified)) |ret_ty| return ret_ty;
|
|
}
|
|
// Fallback: check non-generic LLVM functions
|
|
var cnbuf2: [256]u8 = undefined;
|
|
var callee_fn_opt = c.LLVMGetNamedFunction(self.module, self.nameToCStr(callee_name, &cnbuf2));
|
|
// Intra-namespace fallback
|
|
if (callee_fn_opt == null) {
|
|
if (self.current_namespace) |ns2| {
|
|
const q = std.fmt.allocPrint(self.allocator, "{s}.{s}", .{ ns2, callee_name }) catch return Type.s(64);
|
|
var qbuf2: [256]u8 = undefined;
|
|
callee_fn_opt = c.LLVMGetNamedFunction(self.module, self.nameToCStr(q, &qbuf2));
|
|
}
|
|
}
|
|
if (callee_fn_opt) |callee_fn| {
|
|
const fn_type = c.LLVMGlobalGetValueType(callee_fn);
|
|
const ret_llvm = c.LLVMGetReturnType(fn_type);
|
|
return self.llvmTypeToSxType(ret_llvm);
|
|
}
|
|
// Check if callee is a variable with function pointer type
|
|
{
|
|
if (self.lookupValue(callee_name)) |v| {
|
|
if (v.ty().isFunctionType()) {
|
|
return v.ty().function_type.return_type.*;
|
|
}
|
|
}
|
|
}
|
|
return Type.s(64);
|
|
},
|
|
.unary_op => |unop| {
|
|
if (unop.op == .address_of) {
|
|
const operand_ty = self.inferType(unop.operand);
|
|
const name = operand_ty.displayName(self.allocator) catch return Type.s(64);
|
|
return .{ .pointer_type = .{ .pointee_name = name } };
|
|
}
|
|
return self.inferType(unop.operand);
|
|
},
|
|
.deref_expr => |de| {
|
|
const ptr_ty = self.inferType(de.operand);
|
|
if (ptr_ty.isPointer()) return self.resolveTypeFromName(ptr_ty.pointer_type.pointee_name) orelse Type.s(64);
|
|
return Type.s(64);
|
|
},
|
|
.null_literal => return .{ .pointer_type = .{ .pointee_name = "void" } },
|
|
.field_access => |fa| {
|
|
var obj_ty = self.inferType(fa.object);
|
|
// Auto-deref: if pointer, unwrap to pointee
|
|
if (obj_ty.isPointer()) {
|
|
obj_ty = self.resolveTypeFromName(obj_ty.pointer_type.pointee_name) orelse Type.s(64);
|
|
}
|
|
if (obj_ty == .string_type) {
|
|
if (std.mem.eql(u8, fa.field, "len")) return Type.s(64);
|
|
if (std.mem.eql(u8, fa.field, "ptr")) return .{ .pointer_type = .{ .pointee_name = "u8" } };
|
|
}
|
|
if (obj_ty.isSlice()) {
|
|
if (std.mem.eql(u8, fa.field, "len")) return Type.s(64);
|
|
}
|
|
if (obj_ty.isArray()) {
|
|
if (std.mem.eql(u8, fa.field, "len")) return Type.s(64);
|
|
}
|
|
if (obj_ty.isAny()) {
|
|
if (std.mem.eql(u8, fa.field, "tag")) return Type.s(64);
|
|
if (std.mem.eql(u8, fa.field, "value")) return Type.s(64);
|
|
}
|
|
if (obj_ty.isVector()) {
|
|
return obj_ty.vectorElementType() orelse Type.s(64);
|
|
}
|
|
if (obj_ty.isStruct()) {
|
|
if (self.lookupStructInfo(obj_ty.struct_type)) |info| {
|
|
if (self.findNameIndex(info.field_names, fa.field)) |idx| {
|
|
return info.field_types[idx];
|
|
}
|
|
}
|
|
}
|
|
if (obj_ty.isUnion()) {
|
|
if (self.lookupUnionInfo(obj_ty.union_type)) |info| {
|
|
if (self.findNameIndex(info.field_names, fa.field)) |idx| {
|
|
return info.field_types[idx];
|
|
}
|
|
if (info.promoted_fields.get(fa.field)) |pf| {
|
|
return pf.field_type;
|
|
}
|
|
}
|
|
if (self.lookupTaggedEnumInfo(obj_ty.union_type)) |info| {
|
|
for (info.variant_names, 0..) |vn, i| {
|
|
if (std.mem.eql(u8, vn, fa.field)) {
|
|
return info.variant_types[i];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return Type.s(64);
|
|
},
|
|
.index_expr => |ie| {
|
|
const obj_ty = self.inferType(ie.object);
|
|
if (obj_ty == .string_type) return Type.u(8);
|
|
if (obj_ty.isVector()) {
|
|
return obj_ty.vectorElementType() orelse Type.s(64);
|
|
}
|
|
if (obj_ty.isArray()) {
|
|
return Type.fromName(obj_ty.array_type.element_name) orelse Type.s(64);
|
|
}
|
|
if (obj_ty.isSlice()) {
|
|
return obj_ty.sliceElementType() orelse Type.s(64);
|
|
}
|
|
if (obj_ty.isManyPointer()) {
|
|
return self.resolveTypeFromName(obj_ty.many_pointer_type.element_name) orelse Type.s(64);
|
|
}
|
|
return Type.s(64);
|
|
},
|
|
.slice_expr => |se| {
|
|
const obj_ty = self.inferType(se.object);
|
|
if (obj_ty == .string_type) return .{ .slice_type = .{ .element_name = "u8" } };
|
|
if (obj_ty.isArray()) return .{ .slice_type = .{ .element_name = obj_ty.array_type.element_name } };
|
|
if (obj_ty.isSlice()) return obj_ty;
|
|
return .void_type;
|
|
},
|
|
.array_literal => |al| {
|
|
if (al.elements.len == 0) return .void_type;
|
|
const elem_ty = self.inferType(al.elements[0]);
|
|
const elem_name = elem_ty.displayName(self.allocator) catch return Type.s(64);
|
|
return .{ .array_type = .{ .element_name = elem_name, .length = @intCast(al.elements.len) } };
|
|
},
|
|
.while_expr, .for_expr, .break_expr, .continue_expr => .void_type,
|
|
else => Type.s(64),
|
|
};
|
|
}
|
|
|
|
fn exprIsFloat(self: *CodeGen, node: *Node) bool {
|
|
return self.inferType(node).isFloat();
|
|
}
|
|
|
|
pub fn verify(self: *CodeGen) !void {
|
|
var err_msg: [*c]u8 = null;
|
|
if (c.LLVMVerifyModule(self.module, c.LLVMReturnStatusAction, &err_msg) != 0) {
|
|
defer c.LLVMDisposeMessage(err_msg);
|
|
const msg = std.mem.span(err_msg);
|
|
return self.emitErrorFmt("LLVM verification failed: {s}", .{msg});
|
|
}
|
|
}
|
|
|
|
pub fn printIR(self: *CodeGen) void {
|
|
const ir = c.LLVMPrintModuleToString(self.module);
|
|
defer c.LLVMDisposeMessage(ir);
|
|
const len = std.mem.len(ir);
|
|
std.debug.print("{s}\n", .{ir[0..len]});
|
|
}
|
|
|
|
fn emitToFile(self: *CodeGen, output_path: [*:0]const u8, file_type: c.LLVMCodeGenFileType) !void {
|
|
const tm = self.target_machine orelse return self.emitError("no target machine available");
|
|
|
|
var err_msg: [*c]u8 = null;
|
|
if (c.LLVMTargetMachineEmitToFile(tm, self.module, output_path, file_type, &err_msg) != 0) {
|
|
defer c.LLVMDisposeMessage(err_msg);
|
|
const msg = std.mem.span(err_msg);
|
|
return self.emitErrorFmt("failed to emit file: {s}", .{msg});
|
|
}
|
|
}
|
|
|
|
pub fn emitObject(self: *CodeGen, output_path: [*:0]const u8) !void {
|
|
return self.emitToFile(output_path, c.LLVMObjectFile);
|
|
}
|
|
|
|
pub fn emitAssembly(self: *CodeGen, output_path: [*:0]const u8) !void {
|
|
return self.emitToFile(output_path, c.LLVMAssemblyFile);
|
|
}
|
|
|
|
/// Emit the module as an object file to a memory buffer.
|
|
/// Caller owns the returned buffer and must dispose or pass to JIT.
|
|
pub fn emitObjectToMemory(self: *CodeGen) !c.LLVMMemoryBufferRef {
|
|
const tm = self.target_machine orelse return self.emitError("no target machine available");
|
|
var err_msg: [*c]u8 = null;
|
|
var buf: c.LLVMMemoryBufferRef = null;
|
|
if (c.LLVMTargetMachineEmitToMemoryBuffer(tm, self.module, c.LLVMObjectFile, &err_msg, &buf) != 0) {
|
|
if (err_msg != null) {
|
|
defer c.LLVMDisposeMessage(err_msg);
|
|
const msg = std.mem.span(err_msg);
|
|
return self.emitErrorFmt("failed to emit object to memory: {s}", .{msg});
|
|
}
|
|
return error.CompileError;
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
/// Execute a precompiled object file in-process using LLVM's ORC JIT.
|
|
/// Takes ownership of obj_buf. Returns the exit code from main().
|
|
pub fn runJITFromObject(obj_buf: c.LLVMMemoryBufferRef) !u8 {
|
|
// Create LLJIT with default builder (no custom TM needed — .o is precompiled)
|
|
var jit: c.LLVMOrcLLJITRef = null;
|
|
var err = c.LLVMOrcCreateLLJIT(&jit, null);
|
|
if (err != null) {
|
|
const msg = c.LLVMGetErrorMessage(err);
|
|
defer c.LLVMDisposeErrorMessage(msg);
|
|
std.debug.print("JIT error: {s}\n", .{std.mem.span(msg)});
|
|
return error.CompileError;
|
|
}
|
|
defer _ = c.LLVMOrcDisposeLLJIT(jit);
|
|
|
|
// Add process symbols so JIT can find libc (printf, write, etc.)
|
|
const jd = c.LLVMOrcLLJITGetMainJITDylib(jit);
|
|
const prefix = c.LLVMOrcLLJITGetGlobalPrefix(jit);
|
|
var gen: c.LLVMOrcDefinitionGeneratorRef = null;
|
|
err = c.LLVMOrcCreateDynamicLibrarySearchGeneratorForProcess(&gen, prefix, null, null);
|
|
if (err != null) {
|
|
const msg = c.LLVMGetErrorMessage(err);
|
|
defer c.LLVMDisposeErrorMessage(msg);
|
|
std.debug.print("JIT symbol gen error: {s}\n", .{std.mem.span(msg)});
|
|
return error.CompileError;
|
|
}
|
|
c.LLVMOrcJITDylibAddGenerator(jd, gen);
|
|
|
|
// Add precompiled object file (transfers ownership of obj_buf)
|
|
err = c.LLVMOrcLLJITAddObjectFile(jit, jd, obj_buf);
|
|
if (err != null) {
|
|
const msg = c.LLVMGetErrorMessage(err);
|
|
defer c.LLVMDisposeErrorMessage(msg);
|
|
std.debug.print("JIT add object error: {s}\n", .{std.mem.span(msg)});
|
|
return error.CompileError;
|
|
}
|
|
|
|
// Look up the "main" function
|
|
var main_addr: c.LLVMOrcExecutorAddress = 0;
|
|
err = c.LLVMOrcLLJITLookup(jit, &main_addr, "main");
|
|
if (err != null) {
|
|
const msg = c.LLVMGetErrorMessage(err);
|
|
defer c.LLVMDisposeErrorMessage(msg);
|
|
std.debug.print("JIT lookup error: {s}\n", .{std.mem.span(msg)});
|
|
return error.CompileError;
|
|
}
|
|
|
|
// Cast to function pointer and call
|
|
const main_fn: *const fn () callconv(.c) i32 = @ptrFromInt(main_addr);
|
|
const result = main_fn();
|
|
return if (result >= 0 and result <= 255) @intCast(result) else 1;
|
|
}
|
|
|
|
pub fn link(allocator: std.mem.Allocator, io: std.Io, output_obj: []const u8, output_bin: []const u8, libraries: []const []const u8, target_config: TargetConfig) !void {
|
|
var argv = std.ArrayList([]const u8).empty;
|
|
|
|
if (target_config.isWindows()) {
|
|
// Windows: MSVC-style linker flags
|
|
const linker = target_config.linker orelse "link.exe";
|
|
try argv.appendSlice(allocator, &.{ linker, output_obj });
|
|
try argv.append(allocator, try std.fmt.allocPrint(allocator, "/OUT:{s}", .{output_bin}));
|
|
|
|
for (target_config.lib_paths) |lp| {
|
|
try argv.append(allocator, try std.fmt.allocPrint(allocator, "/LIBPATH:{s}", .{lp}));
|
|
}
|
|
for (libraries) |lib| {
|
|
try argv.append(allocator, try std.fmt.allocPrint(allocator, "{s}.lib", .{lib}));
|
|
}
|
|
} else {
|
|
// Unix: cc-style linker flags
|
|
try argv.appendSlice(allocator, &.{ target_config.getLinker(), output_obj, "-o", output_bin });
|
|
|
|
if (target_config.sysroot) |sr| {
|
|
try argv.append(allocator, try std.fmt.allocPrint(allocator, "--sysroot={s}", .{sr}));
|
|
}
|
|
|
|
// User-supplied library paths first
|
|
for (target_config.lib_paths) |lp| {
|
|
try argv.append(allocator, try std.fmt.allocPrint(allocator, "-L{s}", .{lp}));
|
|
}
|
|
|
|
// Auto-detect host OS library paths when linking foreign libraries
|
|
if (libraries.len > 0 and target_config.triple == null) {
|
|
for (host_lib_paths) |path| {
|
|
try argv.append(allocator, try std.fmt.allocPrint(allocator, "-L{s}", .{path}));
|
|
}
|
|
}
|
|
|
|
for (libraries) |lib| {
|
|
try argv.append(allocator, try std.fmt.allocPrint(allocator, "-l{s}", .{lib}));
|
|
}
|
|
}
|
|
|
|
const argv_slice = try argv.toOwnedSlice(allocator);
|
|
var child = std.process.spawn(io, .{
|
|
.argv = argv_slice,
|
|
}) catch return error.LinkError;
|
|
const result = child.wait(io) catch return error.LinkError;
|
|
if (result != .exited) return error.LinkError;
|
|
if (result.exited != 0) return error.LinkError;
|
|
}
|
|
|
|
/// Common library paths for the host OS, computed at comptime.
|
|
const host_lib_paths = blk: {
|
|
const builtin = @import("builtin");
|
|
var paths: []const []const u8 = &.{};
|
|
if (builtin.os.tag == .macos) {
|
|
if (builtin.cpu.arch == .aarch64) {
|
|
// Apple Silicon Homebrew
|
|
paths = &.{ "/opt/homebrew/lib", "/usr/local/lib" };
|
|
} else {
|
|
// Intel Mac Homebrew
|
|
paths = &.{"/usr/local/lib"};
|
|
}
|
|
} else if (builtin.os.tag == .linux) {
|
|
paths = &.{ "/usr/local/lib", "/usr/lib" };
|
|
}
|
|
break :blk paths;
|
|
};
|
|
};
|