1669 lines
62 KiB
Zig
1669 lines
62 KiB
Zig
//! This file has to do with parsing and rendering the ZIR text format.
|
|
|
|
const std = @import("std");
|
|
const mem = std.mem;
|
|
const Allocator = std.mem.Allocator;
|
|
const assert = std.debug.assert;
|
|
const BigIntConst = std.math.big.int.Const;
|
|
const BigIntMutable = std.math.big.int.Mutable;
|
|
const Type = @import("type.zig").Type;
|
|
const Value = @import("value.zig").Value;
|
|
const TypedValue = @import("TypedValue.zig");
|
|
const ir = @import("ir.zig");
|
|
const IrModule = @import("Module.zig");
|
|
|
|
/// These are instructions that correspond to the ZIR text format. See `ir.Inst` for
|
|
/// in-memory, analyzed instructions with types and values.
|
|
pub const Inst = struct {
|
|
tag: Tag,
|
|
/// Byte offset into the source.
|
|
src: usize,
|
|
name: []const u8,
|
|
|
|
/// Slice into the source of the part after the = and before the next instruction.
|
|
contents: []const u8 = &[0]u8{},
|
|
|
|
/// These names are used directly as the instruction names in the text format.
|
|
pub const Tag = enum {
|
|
breakpoint,
|
|
call,
|
|
compileerror,
|
|
/// Represents a pointer to a global decl by name.
|
|
declref,
|
|
/// The syntax `@foo` is equivalent to `declval("foo")`.
|
|
/// declval is equivalent to declref followed by deref.
|
|
declval,
|
|
str,
|
|
int,
|
|
ptrtoint,
|
|
fieldptr,
|
|
deref,
|
|
as,
|
|
@"asm",
|
|
@"unreachable",
|
|
@"return",
|
|
@"fn",
|
|
@"export",
|
|
primitive,
|
|
ref,
|
|
fntype,
|
|
intcast,
|
|
bitcast,
|
|
elemptr,
|
|
add,
|
|
cmp,
|
|
condbr,
|
|
isnull,
|
|
isnonnull,
|
|
};
|
|
|
|
pub fn TagToType(tag: Tag) type {
|
|
return switch (tag) {
|
|
.breakpoint => Breakpoint,
|
|
.call => Call,
|
|
.declref => DeclRef,
|
|
.declval => DeclVal,
|
|
.compileerror => CompileError,
|
|
.str => Str,
|
|
.int => Int,
|
|
.ptrtoint => PtrToInt,
|
|
.fieldptr => FieldPtr,
|
|
.deref => Deref,
|
|
.as => As,
|
|
.@"asm" => Asm,
|
|
.@"unreachable" => Unreachable,
|
|
.@"return" => Return,
|
|
.@"fn" => Fn,
|
|
.@"export" => Export,
|
|
.primitive => Primitive,
|
|
.ref => Ref,
|
|
.fntype => FnType,
|
|
.intcast => IntCast,
|
|
.bitcast => BitCast,
|
|
.elemptr => ElemPtr,
|
|
.add => Add,
|
|
.cmp => Cmp,
|
|
.condbr => CondBr,
|
|
.isnull => IsNull,
|
|
.isnonnull => IsNonNull,
|
|
};
|
|
}
|
|
|
|
pub fn cast(base: *Inst, comptime T: type) ?*T {
|
|
if (base.tag != T.base_tag)
|
|
return null;
|
|
|
|
return @fieldParentPtr(T, "base", base);
|
|
}
|
|
|
|
pub const Breakpoint = struct {
|
|
pub const base_tag = Tag.breakpoint;
|
|
base: Inst,
|
|
|
|
positionals: struct {},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const Call = struct {
|
|
pub const base_tag = Tag.call;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
func: *Inst,
|
|
args: []*Inst,
|
|
},
|
|
kw_args: struct {
|
|
modifier: std.builtin.CallOptions.Modifier = .auto,
|
|
},
|
|
};
|
|
|
|
pub const DeclRef = struct {
|
|
pub const base_tag = Tag.declref;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
name: *Inst,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const DeclVal = struct {
|
|
pub const base_tag = Tag.declval;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
name: []const u8,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const CompileError = struct {
|
|
pub const base_tag = Tag.compileerror;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
msg: []const u8,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const Str = struct {
|
|
pub const base_tag = Tag.str;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
bytes: []const u8,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const Int = struct {
|
|
pub const base_tag = Tag.int;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
int: BigIntConst,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const PtrToInt = struct {
|
|
pub const base_tag = Tag.ptrtoint;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
ptr: *Inst,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const FieldPtr = struct {
|
|
pub const base_tag = Tag.fieldptr;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
object_ptr: *Inst,
|
|
field_name: *Inst,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const Deref = struct {
|
|
pub const base_tag = Tag.deref;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
ptr: *Inst,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const As = struct {
|
|
pub const base_tag = Tag.as;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
dest_type: *Inst,
|
|
value: *Inst,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const Asm = struct {
|
|
pub const base_tag = Tag.@"asm";
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
asm_source: *Inst,
|
|
return_type: *Inst,
|
|
},
|
|
kw_args: struct {
|
|
@"volatile": bool = false,
|
|
output: ?*Inst = null,
|
|
inputs: []*Inst = &[0]*Inst{},
|
|
clobbers: []*Inst = &[0]*Inst{},
|
|
args: []*Inst = &[0]*Inst{},
|
|
},
|
|
};
|
|
|
|
pub const Unreachable = struct {
|
|
pub const base_tag = Tag.@"unreachable";
|
|
base: Inst,
|
|
|
|
positionals: struct {},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const Return = struct {
|
|
pub const base_tag = Tag.@"return";
|
|
base: Inst,
|
|
|
|
positionals: struct {},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const Fn = struct {
|
|
pub const base_tag = Tag.@"fn";
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
fn_type: *Inst,
|
|
body: Module.Body,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const Export = struct {
|
|
pub const base_tag = Tag.@"export";
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
symbol_name: *Inst,
|
|
value: *Inst,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const Ref = struct {
|
|
pub const base_tag = Tag.ref;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
operand: *Inst,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const Primitive = struct {
|
|
pub const base_tag = Tag.primitive;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
tag: Builtin,
|
|
},
|
|
kw_args: struct {},
|
|
|
|
pub const Builtin = enum {
|
|
isize,
|
|
usize,
|
|
c_short,
|
|
c_ushort,
|
|
c_int,
|
|
c_uint,
|
|
c_long,
|
|
c_ulong,
|
|
c_longlong,
|
|
c_ulonglong,
|
|
c_longdouble,
|
|
c_void,
|
|
f16,
|
|
f32,
|
|
f64,
|
|
f128,
|
|
bool,
|
|
void,
|
|
noreturn,
|
|
type,
|
|
anyerror,
|
|
comptime_int,
|
|
comptime_float,
|
|
@"true",
|
|
@"false",
|
|
@"null",
|
|
@"undefined",
|
|
void_value,
|
|
|
|
pub fn toTypedValue(self: Builtin) TypedValue {
|
|
return switch (self) {
|
|
.isize => .{ .ty = Type.initTag(.type), .val = Value.initTag(.isize_type) },
|
|
.usize => .{ .ty = Type.initTag(.type), .val = Value.initTag(.usize_type) },
|
|
.c_short => .{ .ty = Type.initTag(.type), .val = Value.initTag(.c_short_type) },
|
|
.c_ushort => .{ .ty = Type.initTag(.type), .val = Value.initTag(.c_ushort_type) },
|
|
.c_int => .{ .ty = Type.initTag(.type), .val = Value.initTag(.c_int_type) },
|
|
.c_uint => .{ .ty = Type.initTag(.type), .val = Value.initTag(.c_uint_type) },
|
|
.c_long => .{ .ty = Type.initTag(.type), .val = Value.initTag(.c_long_type) },
|
|
.c_ulong => .{ .ty = Type.initTag(.type), .val = Value.initTag(.c_ulong_type) },
|
|
.c_longlong => .{ .ty = Type.initTag(.type), .val = Value.initTag(.c_longlong_type) },
|
|
.c_ulonglong => .{ .ty = Type.initTag(.type), .val = Value.initTag(.c_ulonglong_type) },
|
|
.c_longdouble => .{ .ty = Type.initTag(.type), .val = Value.initTag(.c_longdouble_type) },
|
|
.c_void => .{ .ty = Type.initTag(.type), .val = Value.initTag(.c_void_type) },
|
|
.f16 => .{ .ty = Type.initTag(.type), .val = Value.initTag(.f16_type) },
|
|
.f32 => .{ .ty = Type.initTag(.type), .val = Value.initTag(.f32_type) },
|
|
.f64 => .{ .ty = Type.initTag(.type), .val = Value.initTag(.f64_type) },
|
|
.f128 => .{ .ty = Type.initTag(.type), .val = Value.initTag(.f128_type) },
|
|
.bool => .{ .ty = Type.initTag(.type), .val = Value.initTag(.bool_type) },
|
|
.void => .{ .ty = Type.initTag(.type), .val = Value.initTag(.void_type) },
|
|
.noreturn => .{ .ty = Type.initTag(.type), .val = Value.initTag(.noreturn_type) },
|
|
.type => .{ .ty = Type.initTag(.type), .val = Value.initTag(.type_type) },
|
|
.anyerror => .{ .ty = Type.initTag(.type), .val = Value.initTag(.anyerror_type) },
|
|
.comptime_int => .{ .ty = Type.initTag(.type), .val = Value.initTag(.comptime_int_type) },
|
|
.comptime_float => .{ .ty = Type.initTag(.type), .val = Value.initTag(.comptime_float_type) },
|
|
.@"true" => .{ .ty = Type.initTag(.bool), .val = Value.initTag(.bool_true) },
|
|
.@"false" => .{ .ty = Type.initTag(.bool), .val = Value.initTag(.bool_false) },
|
|
.@"null" => .{ .ty = Type.initTag(.@"null"), .val = Value.initTag(.null_value) },
|
|
.@"undefined" => .{ .ty = Type.initTag(.@"undefined"), .val = Value.initTag(.undef) },
|
|
.void_value => .{ .ty = Type.initTag(.void), .val = Value.initTag(.the_one_possible_value) },
|
|
};
|
|
}
|
|
};
|
|
};
|
|
|
|
pub const FnType = struct {
|
|
pub const base_tag = Tag.fntype;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
param_types: []*Inst,
|
|
return_type: *Inst,
|
|
},
|
|
kw_args: struct {
|
|
cc: std.builtin.CallingConvention = .Unspecified,
|
|
},
|
|
};
|
|
|
|
pub const IntCast = struct {
|
|
pub const base_tag = Tag.intcast;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
dest_type: *Inst,
|
|
value: *Inst,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const BitCast = struct {
|
|
pub const base_tag = Tag.bitcast;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
dest_type: *Inst,
|
|
operand: *Inst,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const ElemPtr = struct {
|
|
pub const base_tag = Tag.elemptr;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
array_ptr: *Inst,
|
|
index: *Inst,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const Add = struct {
|
|
pub const base_tag = Tag.add;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
lhs: *Inst,
|
|
rhs: *Inst,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const Cmp = struct {
|
|
pub const base_tag = Tag.cmp;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
lhs: *Inst,
|
|
op: std.math.CompareOperator,
|
|
rhs: *Inst,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const CondBr = struct {
|
|
pub const base_tag = Tag.condbr;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
condition: *Inst,
|
|
true_body: Module.Body,
|
|
false_body: Module.Body,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const IsNull = struct {
|
|
pub const base_tag = Tag.isnull;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
operand: *Inst,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
|
|
pub const IsNonNull = struct {
|
|
pub const base_tag = Tag.isnonnull;
|
|
base: Inst,
|
|
|
|
positionals: struct {
|
|
operand: *Inst,
|
|
},
|
|
kw_args: struct {},
|
|
};
|
|
};
|
|
|
|
pub const ErrorMsg = struct {
|
|
byte_offset: usize,
|
|
msg: []const u8,
|
|
};
|
|
|
|
pub const Module = struct {
|
|
decls: []*Inst,
|
|
arena: std.heap.ArenaAllocator,
|
|
error_msg: ?ErrorMsg = null,
|
|
|
|
pub const Body = struct {
|
|
instructions: []*Inst,
|
|
};
|
|
|
|
pub fn deinit(self: *Module, allocator: *Allocator) void {
|
|
allocator.free(self.decls);
|
|
self.arena.deinit();
|
|
self.* = undefined;
|
|
}
|
|
|
|
/// This is a debugging utility for rendering the tree to stderr.
|
|
pub fn dump(self: Module) void {
|
|
self.writeToStream(std.heap.page_allocator, std.io.getStdErr().outStream()) catch {};
|
|
}
|
|
|
|
const InstPtrTable = std.AutoHashMap(*Inst, struct { inst: *Inst, index: ?usize });
|
|
|
|
/// TODO Look into making a table to speed this up.
|
|
pub fn findDecl(self: Module, name: []const u8) ?*Inst {
|
|
for (self.decls) |decl| {
|
|
if (mem.eql(u8, decl.name, name)) {
|
|
return decl;
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
/// The allocator is used for temporary storage, but this function always returns
|
|
/// with no resources allocated.
|
|
pub fn writeToStream(self: Module, allocator: *Allocator, stream: var) !void {
|
|
// First, build a map of *Inst to @ or % indexes
|
|
var inst_table = InstPtrTable.init(allocator);
|
|
defer inst_table.deinit();
|
|
|
|
try inst_table.ensureCapacity(self.decls.len);
|
|
|
|
for (self.decls) |decl, decl_i| {
|
|
try inst_table.putNoClobber(decl, .{ .inst = decl, .index = null });
|
|
|
|
if (decl.cast(Inst.Fn)) |fn_inst| {
|
|
for (fn_inst.positionals.body.instructions) |inst, inst_i| {
|
|
try inst_table.putNoClobber(inst, .{ .inst = inst, .index = inst_i });
|
|
}
|
|
}
|
|
}
|
|
|
|
for (self.decls) |decl, i| {
|
|
try stream.print("@{} ", .{decl.name});
|
|
try self.writeInstToStream(stream, decl, &inst_table);
|
|
try stream.writeByte('\n');
|
|
}
|
|
}
|
|
|
|
fn writeInstToStream(
|
|
self: Module,
|
|
stream: var,
|
|
decl: *Inst,
|
|
inst_table: *const InstPtrTable,
|
|
) @TypeOf(stream).Error!void {
|
|
// TODO I tried implementing this with an inline for loop and hit a compiler bug
|
|
switch (decl.tag) {
|
|
.breakpoint => return self.writeInstToStreamGeneric(stream, .breakpoint, decl, inst_table),
|
|
.call => return self.writeInstToStreamGeneric(stream, .call, decl, inst_table),
|
|
.declref => return self.writeInstToStreamGeneric(stream, .declref, decl, inst_table),
|
|
.declval => return self.writeInstToStreamGeneric(stream, .declval, decl, inst_table),
|
|
.compileerror => return self.writeInstToStreamGeneric(stream, .compileerror, decl, inst_table),
|
|
.str => return self.writeInstToStreamGeneric(stream, .str, decl, inst_table),
|
|
.int => return self.writeInstToStreamGeneric(stream, .int, decl, inst_table),
|
|
.ptrtoint => return self.writeInstToStreamGeneric(stream, .ptrtoint, decl, inst_table),
|
|
.fieldptr => return self.writeInstToStreamGeneric(stream, .fieldptr, decl, inst_table),
|
|
.deref => return self.writeInstToStreamGeneric(stream, .deref, decl, inst_table),
|
|
.as => return self.writeInstToStreamGeneric(stream, .as, decl, inst_table),
|
|
.@"asm" => return self.writeInstToStreamGeneric(stream, .@"asm", decl, inst_table),
|
|
.@"unreachable" => return self.writeInstToStreamGeneric(stream, .@"unreachable", decl, inst_table),
|
|
.@"return" => return self.writeInstToStreamGeneric(stream, .@"return", decl, inst_table),
|
|
.@"fn" => return self.writeInstToStreamGeneric(stream, .@"fn", decl, inst_table),
|
|
.@"export" => return self.writeInstToStreamGeneric(stream, .@"export", decl, inst_table),
|
|
.ref => return self.writeInstToStreamGeneric(stream, .ref, decl, inst_table),
|
|
.primitive => return self.writeInstToStreamGeneric(stream, .primitive, decl, inst_table),
|
|
.fntype => return self.writeInstToStreamGeneric(stream, .fntype, decl, inst_table),
|
|
.intcast => return self.writeInstToStreamGeneric(stream, .intcast, decl, inst_table),
|
|
.bitcast => return self.writeInstToStreamGeneric(stream, .bitcast, decl, inst_table),
|
|
.elemptr => return self.writeInstToStreamGeneric(stream, .elemptr, decl, inst_table),
|
|
.add => return self.writeInstToStreamGeneric(stream, .add, decl, inst_table),
|
|
.cmp => return self.writeInstToStreamGeneric(stream, .cmp, decl, inst_table),
|
|
.condbr => return self.writeInstToStreamGeneric(stream, .condbr, decl, inst_table),
|
|
.isnull => return self.writeInstToStreamGeneric(stream, .isnull, decl, inst_table),
|
|
.isnonnull => return self.writeInstToStreamGeneric(stream, .isnonnull, decl, inst_table),
|
|
}
|
|
}
|
|
|
|
fn writeInstToStreamGeneric(
|
|
self: Module,
|
|
stream: var,
|
|
comptime inst_tag: Inst.Tag,
|
|
base: *Inst,
|
|
inst_table: *const InstPtrTable,
|
|
) !void {
|
|
const SpecificInst = Inst.TagToType(inst_tag);
|
|
const inst = @fieldParentPtr(SpecificInst, "base", base);
|
|
const Positionals = @TypeOf(inst.positionals);
|
|
try stream.writeAll("= " ++ @tagName(inst_tag) ++ "(");
|
|
const pos_fields = @typeInfo(Positionals).Struct.fields;
|
|
inline for (pos_fields) |arg_field, i| {
|
|
if (i != 0) {
|
|
try stream.writeAll(", ");
|
|
}
|
|
try self.writeParamToStream(stream, @field(inst.positionals, arg_field.name), inst_table);
|
|
}
|
|
|
|
comptime var need_comma = pos_fields.len != 0;
|
|
const KW_Args = @TypeOf(inst.kw_args);
|
|
inline for (@typeInfo(KW_Args).Struct.fields) |arg_field, i| {
|
|
if (@typeInfo(arg_field.field_type) == .Optional) {
|
|
if (@field(inst.kw_args, arg_field.name)) |non_optional| {
|
|
if (need_comma) try stream.writeAll(", ");
|
|
try stream.print("{}=", .{arg_field.name});
|
|
try self.writeParamToStream(stream, non_optional, inst_table);
|
|
need_comma = true;
|
|
}
|
|
} else {
|
|
if (need_comma) try stream.writeAll(", ");
|
|
try stream.print("{}=", .{arg_field.name});
|
|
try self.writeParamToStream(stream, @field(inst.kw_args, arg_field.name), inst_table);
|
|
need_comma = true;
|
|
}
|
|
}
|
|
|
|
try stream.writeByte(')');
|
|
}
|
|
|
|
fn writeParamToStream(self: Module, stream: var, param: var, inst_table: *const InstPtrTable) !void {
|
|
if (@typeInfo(@TypeOf(param)) == .Enum) {
|
|
return stream.writeAll(@tagName(param));
|
|
}
|
|
switch (@TypeOf(param)) {
|
|
*Inst => return self.writeInstParamToStream(stream, param, inst_table),
|
|
[]*Inst => {
|
|
try stream.writeByte('[');
|
|
for (param) |inst, i| {
|
|
if (i != 0) {
|
|
try stream.writeAll(", ");
|
|
}
|
|
try self.writeInstParamToStream(stream, inst, inst_table);
|
|
}
|
|
try stream.writeByte(']');
|
|
},
|
|
Module.Body => {
|
|
try stream.writeAll("{\n");
|
|
for (param.instructions) |inst, i| {
|
|
try stream.print(" %{} ", .{i});
|
|
try self.writeInstToStream(stream, inst, inst_table);
|
|
try stream.writeByte('\n');
|
|
}
|
|
try stream.writeByte('}');
|
|
},
|
|
bool => return stream.writeByte("01"[@boolToInt(param)]),
|
|
[]u8, []const u8 => return std.zig.renderStringLiteral(param, stream),
|
|
BigIntConst => return stream.print("{}", .{param}),
|
|
else => |T| @compileError("unimplemented: rendering parameter of type " ++ @typeName(T)),
|
|
}
|
|
}
|
|
|
|
fn writeInstParamToStream(self: Module, stream: var, inst: *Inst, inst_table: *const InstPtrTable) !void {
|
|
if (inst_table.getValue(inst)) |info| {
|
|
if (info.index) |i| {
|
|
try stream.print("%{}", .{info.index});
|
|
} else {
|
|
try stream.print("@{}", .{info.inst.name});
|
|
}
|
|
} else if (inst.cast(Inst.DeclVal)) |decl_val| {
|
|
try stream.print("@{}", .{decl_val.positionals.name});
|
|
} else {
|
|
//try stream.print("?", .{});
|
|
unreachable;
|
|
}
|
|
}
|
|
};
|
|
|
|
pub fn parse(allocator: *Allocator, source: [:0]const u8) Allocator.Error!Module {
|
|
var global_name_map = std.StringHashMap(usize).init(allocator);
|
|
defer global_name_map.deinit();
|
|
|
|
var parser: Parser = .{
|
|
.allocator = allocator,
|
|
.arena = std.heap.ArenaAllocator.init(allocator),
|
|
.i = 0,
|
|
.source = source,
|
|
.global_name_map = &global_name_map,
|
|
.decls = .{},
|
|
.unnamed_index = 0,
|
|
};
|
|
errdefer parser.arena.deinit();
|
|
|
|
parser.parseRoot() catch |err| switch (err) {
|
|
error.ParseFailure => {
|
|
assert(parser.error_msg != null);
|
|
},
|
|
else => |e| return e,
|
|
};
|
|
|
|
return Module{
|
|
.decls = parser.decls.toOwnedSlice(allocator),
|
|
.arena = parser.arena,
|
|
.error_msg = parser.error_msg,
|
|
};
|
|
}
|
|
|
|
const Parser = struct {
|
|
allocator: *Allocator,
|
|
arena: std.heap.ArenaAllocator,
|
|
i: usize,
|
|
source: [:0]const u8,
|
|
decls: std.ArrayListUnmanaged(*Inst),
|
|
global_name_map: *std.StringHashMap(usize),
|
|
error_msg: ?ErrorMsg = null,
|
|
unnamed_index: usize,
|
|
|
|
const Body = struct {
|
|
instructions: std.ArrayList(*Inst),
|
|
name_map: std.StringHashMap(usize),
|
|
};
|
|
|
|
fn parseBody(self: *Parser) !Module.Body {
|
|
var body_context = Body{
|
|
.instructions = std.ArrayList(*Inst).init(self.allocator),
|
|
.name_map = std.StringHashMap(usize).init(self.allocator),
|
|
};
|
|
defer body_context.instructions.deinit();
|
|
defer body_context.name_map.deinit();
|
|
|
|
try requireEatBytes(self, "{");
|
|
skipSpace(self);
|
|
|
|
while (true) : (self.i += 1) switch (self.source[self.i]) {
|
|
';' => _ = try skipToAndOver(self, '\n'),
|
|
'%' => {
|
|
self.i += 1;
|
|
const ident = try skipToAndOver(self, ' ');
|
|
skipSpace(self);
|
|
try requireEatBytes(self, "=");
|
|
skipSpace(self);
|
|
const inst = try parseInstruction(self, &body_context, ident);
|
|
const ident_index = body_context.instructions.items.len;
|
|
if (try body_context.name_map.put(ident, ident_index)) |_| {
|
|
return self.fail("redefinition of identifier '{}'", .{ident});
|
|
}
|
|
try body_context.instructions.append(inst);
|
|
continue;
|
|
},
|
|
' ', '\n' => continue,
|
|
'}' => {
|
|
self.i += 1;
|
|
break;
|
|
},
|
|
else => |byte| return self.failByte(byte),
|
|
};
|
|
|
|
// Move the instructions to the arena
|
|
const instrs = try self.arena.allocator.alloc(*Inst, body_context.instructions.items.len);
|
|
mem.copy(*Inst, instrs, body_context.instructions.items);
|
|
return Module.Body{ .instructions = instrs };
|
|
}
|
|
|
|
fn parseStringLiteral(self: *Parser) ![]u8 {
|
|
const start = self.i;
|
|
try self.requireEatBytes("\"");
|
|
|
|
while (true) : (self.i += 1) switch (self.source[self.i]) {
|
|
'"' => {
|
|
self.i += 1;
|
|
const span = self.source[start..self.i];
|
|
var bad_index: usize = undefined;
|
|
const parsed = std.zig.parseStringLiteral(&self.arena.allocator, span, &bad_index) catch |err| switch (err) {
|
|
error.InvalidCharacter => {
|
|
self.i = start + bad_index;
|
|
const bad_byte = self.source[self.i];
|
|
return self.fail("invalid string literal character: '{c}'\n", .{bad_byte});
|
|
},
|
|
else => |e| return e,
|
|
};
|
|
return parsed;
|
|
},
|
|
'\\' => {
|
|
self.i += 1;
|
|
continue;
|
|
},
|
|
0 => return self.failByte(0),
|
|
else => continue,
|
|
};
|
|
}
|
|
|
|
fn parseIntegerLiteral(self: *Parser) !BigIntConst {
|
|
const start = self.i;
|
|
if (self.source[self.i] == '-') self.i += 1;
|
|
while (true) : (self.i += 1) switch (self.source[self.i]) {
|
|
'0'...'9' => continue,
|
|
else => break,
|
|
};
|
|
const number_text = self.source[start..self.i];
|
|
const base = 10;
|
|
// TODO reuse the same array list for this
|
|
const limbs_buffer_len = std.math.big.int.calcSetStringLimbsBufferLen(base, number_text.len);
|
|
const limbs_buffer = try self.allocator.alloc(std.math.big.Limb, limbs_buffer_len);
|
|
defer self.allocator.free(limbs_buffer);
|
|
const limb_len = std.math.big.int.calcSetStringLimbCount(base, number_text.len);
|
|
const limbs = try self.arena.allocator.alloc(std.math.big.Limb, limb_len);
|
|
var result = BigIntMutable{ .limbs = limbs, .positive = undefined, .len = undefined };
|
|
result.setString(base, number_text, limbs_buffer, self.allocator) catch |err| switch (err) {
|
|
error.InvalidCharacter => {
|
|
self.i = start;
|
|
return self.fail("invalid digit in integer literal", .{});
|
|
},
|
|
};
|
|
return result.toConst();
|
|
}
|
|
|
|
fn parseRoot(self: *Parser) !void {
|
|
// The IR format is designed so that it can be tokenized and parsed at the same time.
|
|
while (true) {
|
|
switch (self.source[self.i]) {
|
|
';' => _ = try skipToAndOver(self, '\n'),
|
|
'@' => {
|
|
self.i += 1;
|
|
const ident = try skipToAndOver(self, ' ');
|
|
skipSpace(self);
|
|
try requireEatBytes(self, "=");
|
|
skipSpace(self);
|
|
const inst = try parseInstruction(self, null, ident);
|
|
const ident_index = self.decls.items.len;
|
|
if (try self.global_name_map.put(ident, ident_index)) |_| {
|
|
return self.fail("redefinition of identifier '{}'", .{ident});
|
|
}
|
|
try self.decls.append(self.allocator, inst);
|
|
},
|
|
' ', '\n' => self.i += 1,
|
|
0 => break,
|
|
else => |byte| return self.fail("unexpected byte: '{c}'", .{byte}),
|
|
}
|
|
}
|
|
}
|
|
|
|
fn eatByte(self: *Parser, byte: u8) bool {
|
|
if (self.source[self.i] != byte) return false;
|
|
self.i += 1;
|
|
return true;
|
|
}
|
|
|
|
fn skipSpace(self: *Parser) void {
|
|
while (self.source[self.i] == ' ' or self.source[self.i] == '\n') {
|
|
self.i += 1;
|
|
}
|
|
}
|
|
|
|
fn requireEatBytes(self: *Parser, bytes: []const u8) !void {
|
|
const start = self.i;
|
|
for (bytes) |byte| {
|
|
if (self.source[self.i] != byte) {
|
|
self.i = start;
|
|
return self.fail("expected '{}'", .{bytes});
|
|
}
|
|
self.i += 1;
|
|
}
|
|
}
|
|
|
|
fn skipToAndOver(self: *Parser, byte: u8) ![]const u8 {
|
|
const start_i = self.i;
|
|
while (self.source[self.i] != 0) : (self.i += 1) {
|
|
if (self.source[self.i] == byte) {
|
|
const result = self.source[start_i..self.i];
|
|
self.i += 1;
|
|
return result;
|
|
}
|
|
}
|
|
return self.fail("unexpected EOF", .{});
|
|
}
|
|
|
|
/// ParseFailure is an internal error code; handled in `parse`.
|
|
const InnerError = error{ ParseFailure, OutOfMemory };
|
|
|
|
fn failByte(self: *Parser, byte: u8) InnerError {
|
|
if (byte == 0) {
|
|
return self.fail("unexpected EOF", .{});
|
|
} else {
|
|
return self.fail("unexpected byte: '{c}'", .{byte});
|
|
}
|
|
}
|
|
|
|
fn fail(self: *Parser, comptime format: []const u8, args: var) InnerError {
|
|
@setCold(true);
|
|
self.error_msg = ErrorMsg{
|
|
.byte_offset = self.i,
|
|
.msg = try std.fmt.allocPrint(&self.arena.allocator, format, args),
|
|
};
|
|
return error.ParseFailure;
|
|
}
|
|
|
|
fn parseInstruction(self: *Parser, body_ctx: ?*Body, name: []const u8) InnerError!*Inst {
|
|
const contents_start = self.i;
|
|
const fn_name = try skipToAndOver(self, '(');
|
|
inline for (@typeInfo(Inst.Tag).Enum.fields) |field| {
|
|
if (mem.eql(u8, field.name, fn_name)) {
|
|
const tag = @field(Inst.Tag, field.name);
|
|
return parseInstructionGeneric(self, field.name, Inst.TagToType(tag), body_ctx, name, contents_start);
|
|
}
|
|
}
|
|
return self.fail("unknown instruction '{}'", .{fn_name});
|
|
}
|
|
|
|
fn parseInstructionGeneric(
|
|
self: *Parser,
|
|
comptime fn_name: []const u8,
|
|
comptime InstType: type,
|
|
body_ctx: ?*Body,
|
|
inst_name: []const u8,
|
|
contents_start: usize,
|
|
) InnerError!*Inst {
|
|
const inst_specific = try self.arena.allocator.create(InstType);
|
|
inst_specific.base = .{
|
|
.name = inst_name,
|
|
.src = self.i,
|
|
.tag = InstType.base_tag,
|
|
};
|
|
|
|
if (@hasField(InstType, "ty")) {
|
|
inst_specific.ty = opt_type orelse {
|
|
return self.fail("instruction '" ++ fn_name ++ "' requires type", .{});
|
|
};
|
|
}
|
|
|
|
const Positionals = @TypeOf(inst_specific.positionals);
|
|
inline for (@typeInfo(Positionals).Struct.fields) |arg_field| {
|
|
if (self.source[self.i] == ',') {
|
|
self.i += 1;
|
|
skipSpace(self);
|
|
} else if (self.source[self.i] == ')') {
|
|
return self.fail("expected positional parameter '{}'", .{arg_field.name});
|
|
}
|
|
@field(inst_specific.positionals, arg_field.name) = try parseParameterGeneric(
|
|
self,
|
|
arg_field.field_type,
|
|
body_ctx,
|
|
);
|
|
skipSpace(self);
|
|
}
|
|
|
|
const KW_Args = @TypeOf(inst_specific.kw_args);
|
|
inst_specific.kw_args = .{}; // assign defaults
|
|
skipSpace(self);
|
|
while (eatByte(self, ',')) {
|
|
skipSpace(self);
|
|
const name = try skipToAndOver(self, '=');
|
|
inline for (@typeInfo(KW_Args).Struct.fields) |arg_field| {
|
|
const field_name = arg_field.name;
|
|
if (mem.eql(u8, name, field_name)) {
|
|
const NonOptional = switch (@typeInfo(arg_field.field_type)) {
|
|
.Optional => |info| info.child,
|
|
else => arg_field.field_type,
|
|
};
|
|
@field(inst_specific.kw_args, field_name) = try parseParameterGeneric(self, NonOptional, body_ctx);
|
|
break;
|
|
}
|
|
} else {
|
|
return self.fail("unrecognized keyword parameter: '{}'", .{name});
|
|
}
|
|
skipSpace(self);
|
|
}
|
|
try requireEatBytes(self, ")");
|
|
|
|
inst_specific.base.contents = self.source[contents_start..self.i];
|
|
//std.debug.warn("parsed {} = '{}'\n", .{ inst_specific.base.name, inst_specific.base.contents });
|
|
|
|
return &inst_specific.base;
|
|
}
|
|
|
|
fn parseParameterGeneric(self: *Parser, comptime T: type, body_ctx: ?*Body) !T {
|
|
if (@typeInfo(T) == .Enum) {
|
|
const start = self.i;
|
|
while (true) : (self.i += 1) switch (self.source[self.i]) {
|
|
' ', '\n', ',', ')' => {
|
|
const enum_name = self.source[start..self.i];
|
|
return std.meta.stringToEnum(T, enum_name) orelse {
|
|
return self.fail("tag '{}' not a member of enum '{}'", .{ enum_name, @typeName(T) });
|
|
};
|
|
},
|
|
0 => return self.failByte(0),
|
|
else => continue,
|
|
};
|
|
}
|
|
switch (T) {
|
|
Module.Body => return parseBody(self),
|
|
bool => {
|
|
const bool_value = switch (self.source[self.i]) {
|
|
'0' => false,
|
|
'1' => true,
|
|
else => |byte| return self.fail("expected '0' or '1' for boolean value, found {c}", .{byte}),
|
|
};
|
|
self.i += 1;
|
|
return bool_value;
|
|
},
|
|
[]*Inst => {
|
|
try requireEatBytes(self, "[");
|
|
skipSpace(self);
|
|
if (eatByte(self, ']')) return &[0]*Inst{};
|
|
|
|
var instructions = std.ArrayList(*Inst).init(&self.arena.allocator);
|
|
while (true) {
|
|
skipSpace(self);
|
|
try instructions.append(try parseParameterInst(self, body_ctx));
|
|
skipSpace(self);
|
|
if (!eatByte(self, ',')) break;
|
|
}
|
|
try requireEatBytes(self, "]");
|
|
return instructions.toOwnedSlice();
|
|
},
|
|
*Inst => return parseParameterInst(self, body_ctx),
|
|
[]u8, []const u8 => return self.parseStringLiteral(),
|
|
BigIntConst => return self.parseIntegerLiteral(),
|
|
else => @compileError("Unimplemented: ir parseParameterGeneric for type " ++ @typeName(T)),
|
|
}
|
|
return self.fail("TODO parse parameter {}", .{@typeName(T)});
|
|
}
|
|
|
|
fn parseParameterInst(self: *Parser, body_ctx: ?*Body) !*Inst {
|
|
const local_ref = switch (self.source[self.i]) {
|
|
'@' => false,
|
|
'%' => true,
|
|
else => |byte| return self.fail("unexpected byte: '{c}'", .{byte}),
|
|
};
|
|
const map = if (local_ref)
|
|
if (body_ctx) |bc|
|
|
&bc.name_map
|
|
else
|
|
return self.fail("referencing a % instruction in global scope", .{})
|
|
else
|
|
self.global_name_map;
|
|
|
|
self.i += 1;
|
|
const name_start = self.i;
|
|
while (true) : (self.i += 1) switch (self.source[self.i]) {
|
|
0, ' ', '\n', ',', ')', ']' => break,
|
|
else => continue,
|
|
};
|
|
const ident = self.source[name_start..self.i];
|
|
const kv = map.get(ident) orelse {
|
|
const bad_name = self.source[name_start - 1 .. self.i];
|
|
const src = name_start - 1;
|
|
if (local_ref) {
|
|
self.i = src;
|
|
return self.fail("unrecognized identifier: {}", .{bad_name});
|
|
} else {
|
|
const declval = try self.arena.allocator.create(Inst.DeclVal);
|
|
declval.* = .{
|
|
.base = .{
|
|
.name = try self.generateName(),
|
|
.src = src,
|
|
.tag = Inst.DeclVal.base_tag,
|
|
},
|
|
.positionals = .{ .name = ident },
|
|
.kw_args = .{},
|
|
};
|
|
return &declval.base;
|
|
}
|
|
};
|
|
if (local_ref) {
|
|
return body_ctx.?.instructions.items[kv.value];
|
|
} else {
|
|
return self.decls.items[kv.value];
|
|
}
|
|
}
|
|
|
|
fn generateName(self: *Parser) ![]u8 {
|
|
const result = try std.fmt.allocPrint(&self.arena.allocator, "unnamed${}", .{self.unnamed_index});
|
|
self.unnamed_index += 1;
|
|
return result;
|
|
}
|
|
};
|
|
|
|
pub fn emit(allocator: *Allocator, old_module: IrModule) !Module {
|
|
var ctx: EmitZIR = .{
|
|
.allocator = allocator,
|
|
.decls = .{},
|
|
.arena = std.heap.ArenaAllocator.init(allocator),
|
|
.old_module = &old_module,
|
|
.next_auto_name = 0,
|
|
.names = std.StringHashMap(void).init(allocator),
|
|
.primitive_table = std.AutoHashMap(Inst.Primitive.Builtin, *Inst).init(allocator),
|
|
};
|
|
defer ctx.decls.deinit(allocator);
|
|
defer ctx.names.deinit();
|
|
defer ctx.primitive_table.deinit();
|
|
errdefer ctx.arena.deinit();
|
|
|
|
try ctx.emit();
|
|
|
|
return Module{
|
|
.decls = ctx.decls.toOwnedSlice(allocator),
|
|
.arena = ctx.arena,
|
|
};
|
|
}
|
|
|
|
const EmitZIR = struct {
|
|
allocator: *Allocator,
|
|
arena: std.heap.ArenaAllocator,
|
|
old_module: *const IrModule,
|
|
decls: std.ArrayListUnmanaged(*Inst),
|
|
names: std.StringHashMap(void),
|
|
next_auto_name: usize,
|
|
primitive_table: std.AutoHashMap(Inst.Primitive.Builtin, *Inst),
|
|
|
|
fn emit(self: *EmitZIR) !void {
|
|
// Put all the Decls in a list and sort them by name to avoid nondeterminism introduced
|
|
// by the hash table.
|
|
var src_decls = std.ArrayList(*IrModule.Decl).init(self.allocator);
|
|
defer src_decls.deinit();
|
|
try src_decls.ensureCapacity(self.old_module.decl_table.size);
|
|
try self.decls.ensureCapacity(self.allocator, self.old_module.decl_table.size);
|
|
try self.names.ensureCapacity(self.old_module.decl_table.size);
|
|
|
|
var decl_it = self.old_module.decl_table.iterator();
|
|
while (decl_it.next()) |kv| {
|
|
const decl = kv.value;
|
|
src_decls.appendAssumeCapacity(decl);
|
|
self.names.putAssumeCapacityNoClobber(mem.spanZ(decl.name), {});
|
|
}
|
|
std.sort.sort(*IrModule.Decl, src_decls.items, {}, (struct {
|
|
fn lessThan(context: void, a: *IrModule.Decl, b: *IrModule.Decl) bool {
|
|
return a.src < b.src;
|
|
}
|
|
}).lessThan);
|
|
|
|
// Emit all the decls.
|
|
for (src_decls.items) |ir_decl| {
|
|
if (self.old_module.export_owners.getValue(ir_decl)) |exports| {
|
|
for (exports) |module_export| {
|
|
const declval = try self.emitDeclVal(ir_decl.src, mem.spanZ(module_export.exported_decl.name));
|
|
const symbol_name = try self.emitStringLiteral(module_export.src, module_export.options.name);
|
|
const export_inst = try self.arena.allocator.create(Inst.Export);
|
|
export_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = module_export.src,
|
|
.tag = Inst.Export.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.symbol_name = symbol_name,
|
|
.value = declval,
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
try self.decls.append(self.allocator, &export_inst.base);
|
|
}
|
|
} else {
|
|
const new_decl = try self.emitTypedValue(ir_decl.src, ir_decl.typed_value.most_recent.typed_value);
|
|
new_decl.name = try self.arena.allocator.dupe(u8, mem.spanZ(ir_decl.name));
|
|
}
|
|
}
|
|
}
|
|
|
|
fn resolveInst(self: *EmitZIR, inst_table: *std.AutoHashMap(*ir.Inst, *Inst), inst: *ir.Inst) !*Inst {
|
|
if (inst.cast(ir.Inst.Constant)) |const_inst| {
|
|
const new_decl = if (const_inst.val.cast(Value.Payload.Function)) |func_pl| blk: {
|
|
const owner_decl = func_pl.func.owner_decl;
|
|
break :blk try self.emitDeclVal(inst.src, mem.spanZ(owner_decl.name));
|
|
} else if (const_inst.val.cast(Value.Payload.DeclRef)) |declref| blk: {
|
|
break :blk try self.emitDeclRef(inst.src, declref.decl);
|
|
} else blk: {
|
|
break :blk try self.emitTypedValue(inst.src, .{ .ty = inst.ty, .val = const_inst.val });
|
|
};
|
|
try inst_table.putNoClobber(inst, new_decl);
|
|
return new_decl;
|
|
} else {
|
|
return inst_table.getValue(inst).?;
|
|
}
|
|
}
|
|
|
|
fn emitDeclVal(self: *EmitZIR, src: usize, decl_name: []const u8) !*Inst {
|
|
const declval = try self.arena.allocator.create(Inst.DeclVal);
|
|
declval.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = src,
|
|
.tag = Inst.DeclVal.base_tag,
|
|
},
|
|
.positionals = .{ .name = try self.arena.allocator.dupe(u8, decl_name) },
|
|
.kw_args = .{},
|
|
};
|
|
return &declval.base;
|
|
}
|
|
|
|
fn emitComptimeIntVal(self: *EmitZIR, src: usize, val: Value) !*Inst {
|
|
const big_int_space = try self.arena.allocator.create(Value.BigIntSpace);
|
|
const int_inst = try self.arena.allocator.create(Inst.Int);
|
|
int_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = src,
|
|
.tag = Inst.Int.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.int = val.toBigInt(big_int_space),
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
try self.decls.append(self.allocator, &int_inst.base);
|
|
return &int_inst.base;
|
|
}
|
|
|
|
fn emitDeclRef(self: *EmitZIR, src: usize, decl: *IrModule.Decl) !*Inst {
|
|
const declval = try self.emitDeclVal(src, mem.spanZ(decl.name));
|
|
const ref_inst = try self.arena.allocator.create(Inst.Ref);
|
|
ref_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = src,
|
|
.tag = Inst.Ref.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.operand = declval,
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
try self.decls.append(self.allocator, &ref_inst.base);
|
|
|
|
return &ref_inst.base;
|
|
}
|
|
|
|
fn emitTypedValue(self: *EmitZIR, src: usize, typed_value: TypedValue) Allocator.Error!*Inst {
|
|
const allocator = &self.arena.allocator;
|
|
if (typed_value.val.cast(Value.Payload.DeclRef)) |decl_ref| {
|
|
const decl = decl_ref.decl;
|
|
return self.emitDeclRef(src, decl);
|
|
}
|
|
switch (typed_value.ty.zigTypeTag()) {
|
|
.Pointer => {
|
|
const ptr_elem_type = typed_value.ty.elemType();
|
|
switch (ptr_elem_type.zigTypeTag()) {
|
|
.Array => {
|
|
// TODO more checks to make sure this can be emitted as a string literal
|
|
//const array_elem_type = ptr_elem_type.elemType();
|
|
//if (array_elem_type.eql(Type.initTag(.u8)) and
|
|
// ptr_elem_type.hasSentinel(Value.initTag(.zero)))
|
|
//{
|
|
//}
|
|
const bytes = typed_value.val.toAllocatedBytes(allocator) catch |err| switch (err) {
|
|
error.AnalysisFail => unreachable,
|
|
else => |e| return e,
|
|
};
|
|
return self.emitStringLiteral(src, bytes);
|
|
},
|
|
else => |t| std.debug.panic("TODO implement emitTypedValue for pointer to {}", .{@tagName(t)}),
|
|
}
|
|
},
|
|
.ComptimeInt => return self.emitComptimeIntVal(src, typed_value.val),
|
|
.Int => {
|
|
const as_inst = try self.arena.allocator.create(Inst.As);
|
|
as_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = src,
|
|
.tag = Inst.As.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.dest_type = try self.emitType(src, typed_value.ty),
|
|
.value = try self.emitComptimeIntVal(src, typed_value.val),
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
|
|
return &as_inst.base;
|
|
},
|
|
.Type => {
|
|
const ty = typed_value.val.toType();
|
|
return self.emitType(src, ty);
|
|
},
|
|
.Fn => {
|
|
const module_fn = typed_value.val.cast(Value.Payload.Function).?.func;
|
|
|
|
var inst_table = std.AutoHashMap(*ir.Inst, *Inst).init(self.allocator);
|
|
defer inst_table.deinit();
|
|
|
|
var instructions = std.ArrayList(*Inst).init(self.allocator);
|
|
defer instructions.deinit();
|
|
|
|
switch (module_fn.analysis) {
|
|
.queued => unreachable,
|
|
.in_progress => unreachable,
|
|
.success => |body| {
|
|
try self.emitBody(body, &inst_table, &instructions);
|
|
},
|
|
.sema_failure => {
|
|
const err_msg = self.old_module.failed_decls.getValue(module_fn.owner_decl).?;
|
|
const fail_inst = try self.arena.allocator.create(Inst.CompileError);
|
|
fail_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = src,
|
|
.tag = Inst.CompileError.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.msg = try self.arena.allocator.dupe(u8, err_msg.msg),
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
try instructions.append(&fail_inst.base);
|
|
},
|
|
.dependency_failure => {
|
|
const fail_inst = try self.arena.allocator.create(Inst.CompileError);
|
|
fail_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = src,
|
|
.tag = Inst.CompileError.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.msg = try self.arena.allocator.dupe(u8, "depends on another failed Decl"),
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
try instructions.append(&fail_inst.base);
|
|
},
|
|
}
|
|
|
|
const fn_type = try self.emitType(src, module_fn.fn_type);
|
|
|
|
const arena_instrs = try self.arena.allocator.alloc(*Inst, instructions.items.len);
|
|
mem.copy(*Inst, arena_instrs, instructions.items);
|
|
|
|
const fn_inst = try self.arena.allocator.create(Inst.Fn);
|
|
fn_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = src,
|
|
.tag = Inst.Fn.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.fn_type = fn_type,
|
|
.body = .{ .instructions = arena_instrs },
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
try self.decls.append(self.allocator, &fn_inst.base);
|
|
return &fn_inst.base;
|
|
},
|
|
.Array => {
|
|
// TODO more checks to make sure this can be emitted as a string literal
|
|
//const array_elem_type = ptr_elem_type.elemType();
|
|
//if (array_elem_type.eql(Type.initTag(.u8)) and
|
|
// ptr_elem_type.hasSentinel(Value.initTag(.zero)))
|
|
//{
|
|
//}
|
|
const bytes = typed_value.val.toAllocatedBytes(allocator) catch |err| switch (err) {
|
|
error.AnalysisFail => unreachable,
|
|
else => |e| return e,
|
|
};
|
|
const str_inst = try self.arena.allocator.create(Inst.Str);
|
|
str_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = src,
|
|
.tag = Inst.Str.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.bytes = bytes,
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
try self.decls.append(self.allocator, &str_inst.base);
|
|
return &str_inst.base;
|
|
},
|
|
.Void => return self.emitPrimitive(src, .void_value),
|
|
else => |t| std.debug.panic("TODO implement emitTypedValue for {}", .{@tagName(t)}),
|
|
}
|
|
}
|
|
|
|
fn emitTrivial(self: *EmitZIR, src: usize, comptime T: type) Allocator.Error!*Inst {
|
|
const new_inst = try self.arena.allocator.create(T);
|
|
new_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = src,
|
|
.tag = T.base_tag,
|
|
},
|
|
.positionals = .{},
|
|
.kw_args = .{},
|
|
};
|
|
return &new_inst.base;
|
|
}
|
|
|
|
fn emitBody(
|
|
self: *EmitZIR,
|
|
body: IrModule.Body,
|
|
inst_table: *std.AutoHashMap(*ir.Inst, *Inst),
|
|
instructions: *std.ArrayList(*Inst),
|
|
) Allocator.Error!void {
|
|
for (body.instructions) |inst| {
|
|
const new_inst = switch (inst.tag) {
|
|
.breakpoint => try self.emitTrivial(inst.src, Inst.Breakpoint),
|
|
.call => blk: {
|
|
const old_inst = inst.cast(ir.Inst.Call).?;
|
|
const new_inst = try self.arena.allocator.create(Inst.Call);
|
|
|
|
const args = try self.arena.allocator.alloc(*Inst, old_inst.args.args.len);
|
|
for (args) |*elem, i| {
|
|
elem.* = try self.resolveInst(inst_table, old_inst.args.args[i]);
|
|
}
|
|
new_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = inst.src,
|
|
.tag = Inst.Call.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.func = try self.resolveInst(inst_table, old_inst.args.func),
|
|
.args = args,
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
break :blk &new_inst.base;
|
|
},
|
|
.unreach => try self.emitTrivial(inst.src, Inst.Unreachable),
|
|
.ret => try self.emitTrivial(inst.src, Inst.Return),
|
|
.constant => unreachable, // excluded from function bodies
|
|
.assembly => blk: {
|
|
const old_inst = inst.cast(ir.Inst.Assembly).?;
|
|
const new_inst = try self.arena.allocator.create(Inst.Asm);
|
|
|
|
const inputs = try self.arena.allocator.alloc(*Inst, old_inst.args.inputs.len);
|
|
for (inputs) |*elem, i| {
|
|
elem.* = try self.emitStringLiteral(inst.src, old_inst.args.inputs[i]);
|
|
}
|
|
|
|
const clobbers = try self.arena.allocator.alloc(*Inst, old_inst.args.clobbers.len);
|
|
for (clobbers) |*elem, i| {
|
|
elem.* = try self.emitStringLiteral(inst.src, old_inst.args.clobbers[i]);
|
|
}
|
|
|
|
const args = try self.arena.allocator.alloc(*Inst, old_inst.args.args.len);
|
|
for (args) |*elem, i| {
|
|
elem.* = try self.resolveInst(inst_table, old_inst.args.args[i]);
|
|
}
|
|
|
|
new_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = inst.src,
|
|
.tag = Inst.Asm.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.asm_source = try self.emitStringLiteral(inst.src, old_inst.args.asm_source),
|
|
.return_type = try self.emitType(inst.src, inst.ty),
|
|
},
|
|
.kw_args = .{
|
|
.@"volatile" = old_inst.args.is_volatile,
|
|
.output = if (old_inst.args.output) |o|
|
|
try self.emitStringLiteral(inst.src, o)
|
|
else
|
|
null,
|
|
.inputs = inputs,
|
|
.clobbers = clobbers,
|
|
.args = args,
|
|
},
|
|
};
|
|
break :blk &new_inst.base;
|
|
},
|
|
.ptrtoint => blk: {
|
|
const old_inst = inst.cast(ir.Inst.PtrToInt).?;
|
|
const new_inst = try self.arena.allocator.create(Inst.PtrToInt);
|
|
new_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = inst.src,
|
|
.tag = Inst.PtrToInt.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.ptr = try self.resolveInst(inst_table, old_inst.args.ptr),
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
break :blk &new_inst.base;
|
|
},
|
|
.bitcast => blk: {
|
|
const old_inst = inst.cast(ir.Inst.BitCast).?;
|
|
const new_inst = try self.arena.allocator.create(Inst.BitCast);
|
|
new_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = inst.src,
|
|
.tag = Inst.BitCast.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.dest_type = try self.emitType(inst.src, inst.ty),
|
|
.operand = try self.resolveInst(inst_table, old_inst.args.operand),
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
break :blk &new_inst.base;
|
|
},
|
|
.cmp => blk: {
|
|
const old_inst = inst.cast(ir.Inst.Cmp).?;
|
|
const new_inst = try self.arena.allocator.create(Inst.Cmp);
|
|
new_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = inst.src,
|
|
.tag = Inst.Cmp.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.lhs = try self.resolveInst(inst_table, old_inst.args.lhs),
|
|
.rhs = try self.resolveInst(inst_table, old_inst.args.rhs),
|
|
.op = old_inst.args.op,
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
break :blk &new_inst.base;
|
|
},
|
|
.condbr => blk: {
|
|
const old_inst = inst.cast(ir.Inst.CondBr).?;
|
|
|
|
var true_body = std.ArrayList(*Inst).init(self.allocator);
|
|
var false_body = std.ArrayList(*Inst).init(self.allocator);
|
|
|
|
defer true_body.deinit();
|
|
defer false_body.deinit();
|
|
|
|
try self.emitBody(old_inst.args.true_body, inst_table, &true_body);
|
|
try self.emitBody(old_inst.args.false_body, inst_table, &false_body);
|
|
|
|
const new_inst = try self.arena.allocator.create(Inst.CondBr);
|
|
new_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = inst.src,
|
|
.tag = Inst.CondBr.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.condition = try self.resolveInst(inst_table, old_inst.args.condition),
|
|
.true_body = .{ .instructions = true_body.toOwnedSlice() },
|
|
.false_body = .{ .instructions = false_body.toOwnedSlice() },
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
break :blk &new_inst.base;
|
|
},
|
|
.isnull => blk: {
|
|
const old_inst = inst.cast(ir.Inst.IsNull).?;
|
|
const new_inst = try self.arena.allocator.create(Inst.IsNull);
|
|
new_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = inst.src,
|
|
.tag = Inst.IsNull.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.operand = try self.resolveInst(inst_table, old_inst.args.operand),
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
break :blk &new_inst.base;
|
|
},
|
|
.isnonnull => blk: {
|
|
const old_inst = inst.cast(ir.Inst.IsNonNull).?;
|
|
const new_inst = try self.arena.allocator.create(Inst.IsNonNull);
|
|
new_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = inst.src,
|
|
.tag = Inst.IsNonNull.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.operand = try self.resolveInst(inst_table, old_inst.args.operand),
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
break :blk &new_inst.base;
|
|
},
|
|
};
|
|
try instructions.append(new_inst);
|
|
try inst_table.putNoClobber(inst, new_inst);
|
|
}
|
|
}
|
|
|
|
fn emitType(self: *EmitZIR, src: usize, ty: Type) Allocator.Error!*Inst {
|
|
switch (ty.tag()) {
|
|
.isize => return self.emitPrimitive(src, .isize),
|
|
.usize => return self.emitPrimitive(src, .usize),
|
|
.c_short => return self.emitPrimitive(src, .c_short),
|
|
.c_ushort => return self.emitPrimitive(src, .c_ushort),
|
|
.c_int => return self.emitPrimitive(src, .c_int),
|
|
.c_uint => return self.emitPrimitive(src, .c_uint),
|
|
.c_long => return self.emitPrimitive(src, .c_long),
|
|
.c_ulong => return self.emitPrimitive(src, .c_ulong),
|
|
.c_longlong => return self.emitPrimitive(src, .c_longlong),
|
|
.c_ulonglong => return self.emitPrimitive(src, .c_ulonglong),
|
|
.c_longdouble => return self.emitPrimitive(src, .c_longdouble),
|
|
.c_void => return self.emitPrimitive(src, .c_void),
|
|
.f16 => return self.emitPrimitive(src, .f16),
|
|
.f32 => return self.emitPrimitive(src, .f32),
|
|
.f64 => return self.emitPrimitive(src, .f64),
|
|
.f128 => return self.emitPrimitive(src, .f128),
|
|
.anyerror => return self.emitPrimitive(src, .anyerror),
|
|
else => switch (ty.zigTypeTag()) {
|
|
.Bool => return self.emitPrimitive(src, .bool),
|
|
.Void => return self.emitPrimitive(src, .void),
|
|
.NoReturn => return self.emitPrimitive(src, .noreturn),
|
|
.Type => return self.emitPrimitive(src, .type),
|
|
.ComptimeInt => return self.emitPrimitive(src, .comptime_int),
|
|
.ComptimeFloat => return self.emitPrimitive(src, .comptime_float),
|
|
.Fn => {
|
|
const param_types = try self.allocator.alloc(Type, ty.fnParamLen());
|
|
defer self.allocator.free(param_types);
|
|
|
|
ty.fnParamTypes(param_types);
|
|
const emitted_params = try self.arena.allocator.alloc(*Inst, param_types.len);
|
|
for (param_types) |param_type, i| {
|
|
emitted_params[i] = try self.emitType(src, param_type);
|
|
}
|
|
|
|
const fntype_inst = try self.arena.allocator.create(Inst.FnType);
|
|
fntype_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = src,
|
|
.tag = Inst.FnType.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.param_types = emitted_params,
|
|
.return_type = try self.emitType(src, ty.fnReturnType()),
|
|
},
|
|
.kw_args = .{
|
|
.cc = ty.fnCallingConvention(),
|
|
},
|
|
};
|
|
try self.decls.append(self.allocator, &fntype_inst.base);
|
|
return &fntype_inst.base;
|
|
},
|
|
else => std.debug.panic("TODO implement emitType for {}", .{ty}),
|
|
},
|
|
}
|
|
}
|
|
|
|
fn autoName(self: *EmitZIR) ![]u8 {
|
|
while (true) {
|
|
const proposed_name = try std.fmt.allocPrint(&self.arena.allocator, "unnamed${}", .{self.next_auto_name});
|
|
self.next_auto_name += 1;
|
|
const gop = try self.names.getOrPut(proposed_name);
|
|
if (!gop.found_existing) {
|
|
gop.kv.value = {};
|
|
return proposed_name;
|
|
}
|
|
}
|
|
}
|
|
|
|
fn emitPrimitive(self: *EmitZIR, src: usize, tag: Inst.Primitive.Builtin) !*Inst {
|
|
const gop = try self.primitive_table.getOrPut(tag);
|
|
if (!gop.found_existing) {
|
|
const primitive_inst = try self.arena.allocator.create(Inst.Primitive);
|
|
primitive_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = src,
|
|
.tag = Inst.Primitive.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.tag = tag,
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
try self.decls.append(self.allocator, &primitive_inst.base);
|
|
gop.kv.value = &primitive_inst.base;
|
|
}
|
|
return gop.kv.value;
|
|
}
|
|
|
|
fn emitStringLiteral(self: *EmitZIR, src: usize, str: []const u8) !*Inst {
|
|
const str_inst = try self.arena.allocator.create(Inst.Str);
|
|
str_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = src,
|
|
.tag = Inst.Str.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.bytes = str,
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
try self.decls.append(self.allocator, &str_inst.base);
|
|
|
|
const ref_inst = try self.arena.allocator.create(Inst.Ref);
|
|
ref_inst.* = .{
|
|
.base = .{
|
|
.name = try self.autoName(),
|
|
.src = src,
|
|
.tag = Inst.Ref.base_tag,
|
|
},
|
|
.positionals = .{
|
|
.operand = &str_inst.base,
|
|
},
|
|
.kw_args = .{},
|
|
};
|
|
try self.decls.append(self.allocator, &ref_inst.base);
|
|
|
|
return &ref_inst.base;
|
|
}
|
|
};
|