tiny.choir.composition.source.boundary

1const std = @import("std");2const source = @import("root.zig");34const Allocator = std.mem.Allocator;56pub const ElementType = enum(u8) {7    i1 = 1,8    i8 = 2,9    i16 = 3,10    i32 = 4,11    i64 = 5,12    u8 = 6,13    u16 = 7,14    u32 = 8,15    u64 = 9,16    f16 = 10,17    bf16 = 11,18    f32 = 12,19    f64 = 13,20    key = 14,2122    pub fn byteSize(self: ElementType) u64 {23        return switch (self) {24            .i1, .i8, .u8 => 1,25            .i16, .u16, .f16, .bf16 => 2,26            .i32, .u32, .f32 => 4,27            .i64, .u64, .f64, .key => 8,28        };29    }30};3132/// A logical value crossing partition boundaries before physical placement is chosen.33/// The byte size must match the static shape, and ownership and alias facts are mandatory.34pub const BoundarySpec = struct {35    id: source.BoundaryId,36    name: []const u8,37    element_type: ElementType,38    dimensions: []const u64,39    byte_size: u64,40    access: source.Access,41    ownership: source.Ownership,42    alias: Alias,43    provenance: source.ProvenanceSpec,44};4546pub const Alias = union(enum(u8)) {47    disjoint,48    boundary: source.BoundaryId,4950    pub fn eql(self: Alias, other: Alias) bool {51        const Tag = std.meta.Tag(Alias);52        if (@as(Tag, self) != @as(Tag, other)) return false;53        return switch (self) {54            .disjoint => true,55            .boundary => |id| id.value == other.boundary.value,56        };57    }58};5960/// An owned, context-independent boundary descriptor.61pub const Boundary = struct {62    id: source.BoundaryId,63    name: []const u8,64    element_type: ElementType,65    dimensions: []const u64,66    byte_size: u64,67    access: source.Access,68    ownership: source.Ownership,69    alias: Alias,70    provenance: source.Provenance,7172    pub fn init(allocator: Allocator, spec: BoundarySpec) Allocator.Error!Boundary {73        const name = try allocator.dupe(u8, spec.name);74        errdefer allocator.free(name);75        const dimensions = try source.slice.dupeOrEmpty(u64, allocator, spec.dimensions);76        errdefer source.slice.freeOrEmpty(u64, allocator, dimensions);77        return .{78            .id = spec.id,79            .name = name,80            .element_type = spec.element_type,81            .dimensions = dimensions,82            .byte_size = spec.byte_size,83            .access = spec.access,84            .ownership = spec.ownership,85            .alias = spec.alias,86            .provenance = try source.Provenance.init(allocator, spec.provenance),87        };88    }8990    pub fn clone(self: Boundary, allocator: Allocator) Allocator.Error!Boundary {91        return try init(allocator, .{92            .id = self.id,93            .name = self.name,94            .element_type = self.element_type,95            .dimensions = self.dimensions,96            .byte_size = self.byte_size,97            .access = self.access,98            .ownership = self.ownership,99            .alias = self.alias,100            .provenance = .{101                .path = self.provenance.path,102                .symbol = self.provenance.symbol,103                .line = self.provenance.line,104                .column = self.provenance.column,105            },106        });107    }108109    pub fn deinit(self: *Boundary, allocator: Allocator) void {110        self.provenance.deinit(allocator);111        source.slice.freeOrEmpty(u64, allocator, self.dimensions);112        allocator.free(self.name);113        self.* = undefined;114    }115};116117pub fn staticByteSize(element_type: ElementType, dimensions: []const u64) ?u64 {118    var byte_size = element_type.byteSize();119    for (dimensions) |dimension| {120        if (dimension == 0) return null;121        byte_size = std.math.mul(u64, byte_size, dimension) catch return null;122    }123    return byte_size;124}

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