tiny.choir.ir.dialects.shape

1pub const shape = struct {2    pub fn of(comptime spec: anytype) interfaces.OperationShape {3        comptime validateSpec(@TypeOf(spec));4        return .{5            .operands = field(spec, "operands"),6            .results = field(spec, "results"),7            .regions = field(spec, "regions"),8            .successors = field(spec, "successors"),9        };10    }1112    pub fn fixed(13        comptime operands: anytype,14        comptime results: anytype,15        comptime regions: anytype,16        comptime successors: anytype,17    ) interfaces.OperationShape {18        return of(.{19            .operands = operands,20            .results = results,21            .regions = regions,22            .successors = successors,23        });24    }2526    pub fn leaf(comptime operands: anytype, comptime results: anytype) interfaces.OperationShape {27        return of(.{28            .operands = operands,29            .results = results,30            .regions = 0,31            .successors = 0,32        });33    }3435    pub fn noNested(comptime spec: anytype) interfaces.OperationShape {36        comptime validateNoNestedSpec(@TypeOf(spec));37        const base = of(spec);38        return .{39            .operands = base.operands,40            .results = base.results,41            .regions = interfaces.CountRange.exactly(0),42            .successors = interfaces.CountRange.exactly(0),43        };44    }4546    pub fn exactly(comptime count: usize) interfaces.CountRange {47        return interfaces.CountRange.exactly(count);48    }4950    pub fn atLeast(comptime count: usize) interfaces.CountRange {51        return interfaces.CountRange.atLeast(count);52    }5354    pub fn atMost(comptime count: usize) interfaces.CountRange {55        return interfaces.CountRange.atMost(count);56    }5758    pub fn between(comptime min: usize, comptime max: usize) interfaces.CountRange {59        return interfaces.CountRange.between(min, max);60    }6162    pub fn any() interfaces.CountRange {63        return .{};64    }6566    pub fn range(comptime value: anytype) interfaces.CountRange {67        return rangeFrom(value);68    }6970    fn field(comptime spec: anytype, comptime name: []const u8) interfaces.CountRange {71        if (comptime @hasField(@TypeOf(spec), name)) {72            return rangeFrom(@field(spec, name));73        }74        return .{};75    }7677    fn rangeFrom(comptime value: anytype) interfaces.CountRange {78        const Value = @TypeOf(value);79        if (Value == interfaces.CountRange) return value;80        return switch (@typeInfo(Value)) {81            .comptime_int => blk: {82                if (value < 0) @compileError("operation shape count cannot be negative");83                break :blk interfaces.CountRange.exactly(@as(usize, value));84            },85            .int => |int_info| blk: {86                if (int_info.signedness == .signed and value < 0) {87                    @compileError("operation shape count cannot be negative");88                }89                break :blk interfaces.CountRange.exactly(@intCast(value));90            },91            .pointer => |pointer_info| switch (pointer_info.size) {92                .one => switch (@typeInfo(pointer_info.child)) {93                    .array => |array_info| blk: {94                        if (array_info.child == u8) {95                            @compileError("operation shape fields must use a list of component names, not one component name");96                        }97                        tryComponentNames(value.*, "shape");98                        break :blk interfaces.CountRange.exactly(array_info.len);99                    },100                    .@"struct" => |struct_info| blk: {101                        tryComponentNames(value.*, "shape");102                        break :blk interfaces.CountRange.exactly(struct_info.field_names.len);103                    },104                    else => @compileError("operation shape fields must be integer counts, CountRange values, or component name lists"),105                },106                .slice => blk: {107                    tryComponentNames(value, "shape");108                    break :blk interfaces.CountRange.exactly(value.len);109                },110                else => @compileError("operation shape fields must be integer counts, CountRange values, or component name lists"),111            },112            .array => |array_info| blk: {113                if (array_info.child == u8) {114                    @compileError("operation shape fields must use a list of component names, not one component name");115                }116                tryComponentNames(value, "shape");117                break :blk interfaces.CountRange.exactly(array_info.len);118            },119            .@"struct" => |struct_info| blk: {120                tryComponentNames(value, "shape");121                break :blk interfaces.CountRange.exactly(struct_info.field_names.len);122            },123            else => @compileError("operation shape fields must be integer counts or CountRange values"),124        };125    }126127    fn validateSpec(comptime Spec: type) void {128        switch (@typeInfo(Spec)) {129            .@"struct" => |struct_info| {130                inline for (struct_info.field_names) |field_name| {131                    if (!std.mem.eql(u8, field_name, "operands") and132                        !std.mem.eql(u8, field_name, "results") and133                        !std.mem.eql(u8, field_name, "regions") and134                        !std.mem.eql(u8, field_name, "successors"))135                    {136                        @compileError("unknown operation shape field: " ++ field_name);137                    }138                }139            },140            else => @compileError("operation shape spec must be a struct literal"),141        }142    }143144    fn validateNoNestedSpec(comptime Spec: type) void {145        validateSpec(Spec);146        if (@hasField(Spec, "regions")) @compileError("noNested operation shape cannot specify regions");147        if (@hasField(Spec, "successors")) @compileError("noNested operation shape cannot specify successors");148    }149};

1pub const shape = spec.shape;