tiny.choir.Location

1pub const Location = union(enum) {2    unknown: void,34    file: FileLocation,56    file_range: FileRangeLocation,78    name: NameLocation,910    fused: FusedLocation,1112    call_site: CallSiteLocation,1314    pub const FileLocation = struct {15        filename: []const u8,16        line: u32,17        column: u32,18    };1920    pub const FilePosition = struct {21        byte: u64,22        line: u32,23        column: u32,24    };2526    pub const FileRangeLocation = struct {27        filename: []const u8,28        start: FilePosition,29        end: FilePosition,30    };3132    pub const NameLocation = struct {33        name: []const u8,34        child: ?*const Location,35    };3637    pub const FusedLocation = struct {38        locations: []const Location,39        metadata: ?*const anyopaque,40    };4142    pub const CallSiteLocation = struct {43        callee: *const Location,44        caller: *const Location,45    };4647    pub fn getUnknown() Location {48        return .{ .unknown = {} };49    }5051    pub fn getFile(filename: []const u8, line: u32, column: u32) Location {52        return .{ .file = .{53            .filename = filename,54            .line = line,55            .column = column,56        } };57    }5859    pub fn getFileRange(60        filename: []const u8,61        start: FilePosition,62        end: FilePosition,63    ) Location {64        std.debug.assert(start.byte <= end.byte);65        return .{ .file_range = .{66            .filename = filename,67            .start = start,68            .end = end,69        } };70    }7172    pub fn getName(n: []const u8, child: ?*const Location) Location {73        return .{ .name = .{74            .name = n,75            .child = child,76        } };77    }7879    pub fn eql(self: Location, other: Location) bool {80        const Tag = std.meta.Tag(Location);81        if (@as(Tag, self) != @as(Tag, other)) return false;8283        return switch (self) {84            .unknown => true,85            .file => |f| {86                const o = other.file;87                return f.line == o.line and f.column == o.column and std.mem.eql(u8, f.filename, o.filename);88            },89            .file_range => |range| {90                const o = other.file_range;91                return range.start.byte == o.start.byte and92                    range.start.line == o.start.line and93                    range.start.column == o.start.column and94                    range.end.byte == o.end.byte and95                    range.end.line == o.end.line and96                    range.end.column == o.end.column and97                    std.mem.eql(u8, range.filename, o.filename);98            },99            .name => |n| {100                const o = other.name;101                if (!std.mem.eql(u8, n.name, o.name)) return false;102                if (n.child == null and o.child == null) return true;103                if (n.child == null or o.child == null) return false;104                return n.child.?.eql(o.child.?.*);105            },106            .fused => |f| {107                const o = other.fused;108                if (f.locations.len != o.locations.len) return false;109                for (f.locations, o.locations) |a, b| {110                    if (!a.eql(b)) return false;111                }112                return f.metadata == o.metadata;113            },114            .call_site => |c| {115                const o = other.call_site;116                return c.callee.eql(o.callee.*) and c.caller.eql(o.caller.*);117            },118        };119    }120121    pub fn format(self: Location, writer: *std.Io.Writer) std.Io.Writer.Error!void {122        switch (self) {123            .unknown => try writer.writeAll("loc(unknown)"),124            .file => |f| try writer.print("loc(\"{s}\":{d}:{d})", .{ f.filename, f.line, f.column }),125            .file_range => |range| try writer.print(126                "loc(\"{s}\":{d}:{d}-{d}:{d}@{d}..{d})",127                .{128                    range.filename,129                    range.start.line,130                    range.start.column,131                    range.end.line,132                    range.end.column,133                    range.start.byte,134                    range.end.byte,135                },136            ),137            .name => |n| try writer.print("loc(\"{s}\")", .{n.name}),138            .fused => try writer.writeAll("loc(fused)"),139            .call_site => try writer.writeAll("loc(callsite)"),140        }141    }142};

1pub const Location = ir.Location;