77 lines
1.6 KiB
Zig
77 lines
1.6 KiB
Zig
|
const std = @import("std");
|
||
|
|
|
||
|
|
const heap = @import("./heap.zig");
|
||
|
|
|
||
|
|
const map = @import("./map.zig");
|
||
|
|
|
||
|
|
pub const Table = struct {
|
||
|
|
arena: std.heap.ArenaAllocator,
|
||
|
|
table: map.Hashed(TypeID, Entry, map.enum_traits(TypeID)),
|
||
|
|
|
||
|
|
const Entry = struct {
|
||
|
|
ptr: *anyopaque,
|
||
|
|
};
|
||
|
|
|
||
|
|
pub fn deinit(self: *Table) void {
|
||
|
|
self.table.deinit();
|
||
|
|
self.arena.deinit();
|
||
|
|
|
||
|
|
self.* = undefined;
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn get(self: Table, comptime Resource: type) ?*Resource {
|
||
|
|
if (self.table.get_ptr(type_id(Resource))) |entry| {
|
||
|
|
return @ptrCast(@alignCast(entry.ptr));
|
||
|
|
}
|
||
|
|
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn init() Table {
|
||
|
|
return .{
|
||
|
|
.arena = std.heap.ArenaAllocator.init(heap.allocator),
|
||
|
|
.table = .{.allocator = heap.allocator},
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn set_get(self: *Table, value: anytype) std.mem.Allocator.Error!*@TypeOf(value) {
|
||
|
|
try self.set(value);
|
||
|
|
|
||
|
|
return self.get(@TypeOf(value)).?;
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn set(self: *Table, value: anytype) std.mem.Allocator.Error!void {
|
||
|
|
const Value = @TypeOf(value);
|
||
|
|
const value_id = type_id(Value);
|
||
|
|
|
||
|
|
if (self.table.get_ptr(value_id)) |entry| {
|
||
|
|
@as(*Value, @ptrCast(@alignCast(entry.ptr))).* = value;
|
||
|
|
} else {
|
||
|
|
const resource_allocator = self.arena.allocator();
|
||
|
|
const allocated_resource = try resource_allocator.create(Value);
|
||
|
|
|
||
|
|
errdefer resource_allocator.destroy(allocated_resource);
|
||
|
|
|
||
|
|
std.debug.assert(try self.table.emplace(value_id, .{
|
||
|
|
.ptr = allocated_resource,
|
||
|
|
}));
|
||
|
|
|
||
|
|
allocated_resource.* = value;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
pub const TypeID = enum (usize) { _ };
|
||
|
|
|
||
|
|
pub fn type_id(comptime T: type) TypeID {
|
||
|
|
const TypeHandle = struct {
|
||
|
|
comptime {
|
||
|
|
_ = T;
|
||
|
|
}
|
||
|
|
|
||
|
|
var byte: u8 = 0;
|
||
|
|
};
|
||
|
|
|
||
|
|
return @enumFromInt(@intFromPtr(&TypeHandle.byte));
|
||
|
|
}
|