tiny.choir.ir.types

1pub const types = @import("types.zig");

1const std = @import("std");2const interfaces = @import("interfaces/root.zig");3const TypeRegistry = interfaces.TypeRegistry;4const Type = @import("type.zig").Type;56pub const TypeStorage = struct {7    dialect_types: std.HashMapUnmanaged(DialectTypeKey, Type, DialectTypeKeyContext, std.hash_map.default_max_load_percentage),8    next_type_id: u64,910    pub const DialectTypeKey = struct {11        name: []const u8,12        param_key: []const u8,1314        pub fn eql(self: DialectTypeKey, other: DialectTypeKey) bool {15            return std.mem.eql(u8, self.name, other.name) and std.mem.eql(u8, self.param_key, other.param_key);16        }1718        pub fn hash(self: DialectTypeKey) u64 {19            var hasher = std.hash.Wyhash.init(0);20            hasher.update(self.name);21            hasher.update(&[_]u8{0});22            hasher.update(self.param_key);23            return hasher.final();24        }25    };2627    pub const DialectTypeKeyContext = struct {28        pub fn hash(_: DialectTypeKeyContext, key: DialectTypeKey) u64 {29            return key.hash();30        }3132        pub fn eql(_: DialectTypeKeyContext, a: DialectTypeKey, b: DialectTypeKey) bool {33            return a.eql(b);34        }35    };3637    pub fn init(allocator: std.mem.Allocator) TypeStorage {38        _ = allocator;39        return .{40            .dialect_types = .{},41            .next_type_id = 1,42        };43    }4445    pub fn allocateTypeId(self: *TypeStorage) u64 {46        const id = self.next_type_id;47        self.next_type_id += 1;48        return id;49    }5051    pub fn deinit(52        self: *TypeStorage,53        table_allocator: std.mem.Allocator,54        key_allocator: std.mem.Allocator,55        payload_allocator: std.mem.Allocator,56    ) void {57        var dialect_iter = self.dialect_types.valueIterator();58        while (dialect_iter.next()) |typ| {59            const impl: *Type.DialectTypeStorage = @ptrCast(@alignCast(@constCast(typ.impl)));60            if (impl.param_key.len > 0) {61                key_allocator.free(impl.param_key);62            }63            payload_allocator.destroy(impl);64        }65        self.dialect_types.deinit(table_allocator);66    }67};6869const default_hash_load = std.hash_map.default_max_load_percentage;7071pub const TypeParamCache = std.HashMapUnmanaged(72    u64,73    interfaces.TypeParamPayload,74    std.hash_map.AutoContext(u64),75    default_hash_load,76);7778pub const TypeInterner = struct {79    storage: TypeStorage,80    registry: TypeRegistry,8182    pub fn init(83        table_allocator: std.mem.Allocator,84        registry_allocator: std.mem.Allocator,85    ) TypeInterner {86        return .{87            .storage = TypeStorage.init(table_allocator),88            .registry = TypeRegistry.init(registry_allocator),89        };90    }9192    pub fn deinit(93        self: *TypeInterner,94        table_allocator: std.mem.Allocator,95        key_allocator: std.mem.Allocator,96        payload_allocator: std.mem.Allocator,97    ) void {98        self.storage.deinit(table_allocator, key_allocator, payload_allocator);99        self.registry.deinit();100    }101};