tiny.choir.ir.dialects.segments

1pub const segments = struct {2    pub const operand_attribute_name = "operand_segment_sizes";3    pub const result_attribute_name = "result_segment_sizes";45    pub fn operands(comptime values: anytype) interfaces.OperationSegmentSpec {6        return sized(operand_attribute_name, values);7    }89    pub fn results(comptime values: anytype) interfaces.OperationSegmentSpec {10        return sized(result_attribute_name, values);11    }1213    pub fn sized(comptime attribute_name: []const u8, comptime values: anytype) interfaces.OperationSegmentSpec {14        if (attribute_name.len == 0) @compileError("operation segment attribute name cannot be empty");15        const segment_ranges = ranges(values);16        if (segment_ranges.len == 0) @compileError("operation segment declaration cannot be empty");17        return .{18            .attribute_name = attribute_name,19            .segments = segment_ranges,20        };21    }2223    fn ranges(comptime values: anytype) []const interfaces.CountRange {24        const Values = @TypeOf(values);25        if (Values == []const interfaces.CountRange) return values;26        return switch (@typeInfo(Values)) {27            .pointer => |pointer_info| switch (pointer_info.size) {28                .one => switch (@typeInfo(pointer_info.child)) {29                    .array => rangesFromArray(values.*),30                    .@"struct" => |struct_info| blk: {31                        if (!struct_info.is_tuple) @compileError("operation segment values must be a comptime slice, array, or tuple");32                        break :blk rangesFromTuple(values.*);33                    },34                    else => @compileError("operation segment values must be a comptime slice, array, or tuple"),35                },36                .slice => rangesFromSlice(values),37                else => @compileError("operation segment values must be a comptime slice, array, or tuple"),38            },39            .array => rangesFromArray(values),40            .@"struct" => |struct_info| blk: {41                if (!struct_info.is_tuple) @compileError("operation segment values must be a comptime slice, array, or tuple");42                break :blk rangesFromTuple(values);43            },44            else => @compileError("operation segment values must be a comptime slice, array, or tuple"),45        };46    }4748    fn rangesFromSlice(comptime values: anytype) []const interfaces.CountRange {49        const value = comptime blk: {50            var out: [values.len]interfaces.CountRange = undefined;51            for (values, 0..) |segment, index| {52                out[index] = shape.range(segment);53            }54            break :blk out;55        };56        return &value;57    }5859    fn rangesFromArray(comptime values: anytype) []const interfaces.CountRange {60        const value = comptime blk: {61            var out: [values.len]interfaces.CountRange = undefined;62            for (values, 0..) |segment, index| {63                out[index] = shape.range(segment);64            }65            break :blk out;66        };67        return &value;68    }6970    fn rangesFromTuple(comptime values: anytype) []const interfaces.CountRange {71        const struct_info = @typeInfo(@TypeOf(values)).@"struct";72        const value = comptime blk: {73            var out: [struct_info.field_names.len]interfaces.CountRange = undefined;74            for (struct_info.field_names, 0..) |name, index| {75                out[index] = shape.range(@field(values, name));76            }77            break :blk out;78        };79        return &value;80    }81};

1pub const segments = spec.segments;