tiny.choir.backends.wasm.emission.function.instruction

1pub fn Writer(comptime Out: type) type {2    return struct {3        out: Out,4        plan: *const emission.Plan,5        function_ordinal: usize,67        const Self = @This();89        pub fn writeOperation(self: *Self, operation: *ir.Operation) emission.Error!void {10            const name = operation.name.name;11            if (std.mem.eql(u8, name, ArithDialect.ConstantOp.operation_name)) {12                return instruction.arithmetic.lowering.writeConstant(Out, self, operation);13            }14            if (std.mem.eql(u8, name, ArithDialect.AddOp.operation_name)) {15                return instruction.arithmetic.lowering.writeBinary(Out, self, operation, .add);16            }17            if (std.mem.eql(u8, name, ArithDialect.SubOp.operation_name)) {18                return instruction.arithmetic.lowering.writeBinary(Out, self, operation, .sub);19            }20            if (std.mem.eql(u8, name, ArithDialect.MulOp.operation_name)) {21                return instruction.arithmetic.lowering.writeBinary(Out, self, operation, .mul);22            }23            if (std.mem.eql(u8, name, ArithDialect.DivOp.operation_name)) {24                return instruction.arithmetic.lowering.writeBinary(Out, self, operation, .div);25            }26            if (std.mem.eql(u8, name, ArithDialect.RemOp.operation_name)) {27                return instruction.arithmetic.lowering.writeBinary(Out, self, operation, .rem);28            }29            if (std.mem.eql(u8, name, ArithDialect.AndOp.operation_name)) {30                return instruction.arithmetic.lowering.writeBinary(Out, self, operation, .band);31            }32            if (std.mem.eql(u8, name, ArithDialect.OrOp.operation_name)) {33                return instruction.arithmetic.lowering.writeBinary(Out, self, operation, .bor);34            }35            if (std.mem.eql(u8, name, ArithDialect.XorOp.operation_name)) {36                return instruction.arithmetic.lowering.writeBinary(Out, self, operation, .bxor);37            }38            if (std.mem.eql(u8, name, ArithDialect.ShlOp.operation_name)) {39                return instruction.arithmetic.lowering.writeBinary(Out, self, operation, .shl);40            }41            if (std.mem.eql(u8, name, ArithDialect.ShrOp.operation_name)) {42                return instruction.arithmetic.lowering.writeBinary(Out, self, operation, .shr);43            }44            if (std.mem.eql(u8, name, ArithDialect.UshrOp.operation_name)) {45                return instruction.arithmetic.lowering.writeBinary(Out, self, operation, .ushr);46            }47            if (std.mem.eql(u8, name, ArithDialect.CmpOp.operation_name)) {48                return instruction.arithmetic.lowering.writeCmp(Out, self, operation);49            }50            if (std.mem.eql(u8, name, ArithDialect.SelectOp.operation_name)) {51                return instruction.arithmetic.lowering.writeSelect(Out, self, operation);52            }53            if (std.mem.eql(u8, name, ArithDialect.NegOp.operation_name)) {54                return instruction.arithmetic.lowering.writeNeg(Out, self, operation);55            }56            if (std.mem.eql(u8, name, ArithDialect.AbsOp.operation_name)) {57                return instruction.arithmetic.lowering.writeAbs(Out, self, operation);58            }59            if (std.mem.eql(u8, name, ArithDialect.SqrtOp.operation_name)) {60                return instruction.arithmetic.lowering.writeSqrt(Out, self, operation);61            }62            if (std.mem.eql(u8, name, ArithDialect.BitcastOp.operation_name)) {63                return instruction.cast.write(Out, self, operation, true);64            }65            if (std.mem.eql(u8, name, ArithDialect.CastOp.operation_name)) {66                return instruction.cast.write(Out, self, operation, false);67            }68            if (std.mem.eql(u8, name, FuncDialect.CallOp.operation_name)) {69                return instruction.call.write(Out, self, operation);70            }71            if (std.mem.eql(u8, name, MemrefDialect.LoadOp.operation_name)) {72                return instruction.memory.writeLoad(Out, self, operation);73            }74            if (std.mem.eql(u8, name, MemrefDialect.StoreOp.operation_name)) {75                return instruction.memory.writeStore(Out, self, operation);76            }77            inline for (.{ ArithDialect.AddoOp, ArithDialect.SuboOp, ArithDialect.MuloOp }) |Op| {78                if (std.mem.eql(u8, name, Op.operation_name)) return error.UnsupportedOperation;79            }80            return error.CodeGenFailed;81        }8283        pub fn writeValue(self: *Self, value: *ir.Value) emission.Error!void {84            try self.writeLocalGet(try self.localFor(value));85        }8687        pub fn writeLocalGet(self: *Self, local: u32) emission.Error!void {88            try self.out.writeByte(0x20);89            try wasm.binary.writeUleb(self.out, local);90        }9192        pub fn writeLocalSet(self: *Self, local: u32) emission.Error!void {93            try self.out.writeByte(0x21);94            try wasm.binary.writeUleb(self.out, local);95        }9697        pub fn localFor(self: *const Self, value: *ir.Value) emission.Error!u32 {98            return self.plan.localFor(self.function_ordinal, value) orelse99                error.CodeGenFailed;100        }101    };102}

1pub const arithmetic = @import("arithmetic/root.zig");2pub const cast = @import("cast.zig");3pub const call = @import("call.zig");4pub const memory = @import("memory.zig");56pub const Writer = @import("writer.zig").Writer;

1pub const instruction = @import("instruction/root.zig");