import type { BinaryOp, UnaryOp } from "../ast"
import type { Loc } from "./loc"

export type Operand = Loc | number

export type AbsInsnCopy = {
    type: "copy",
    dest: Loc,
    source: Operand,
}

type UnaryInsn<Op> = {
    type: "unary",
    dest: Loc,
    op: Op,
    source: Operand,
}

// NOTE: We convert binary -> unary when going SSA3 -> SSA2
export type AbsInsn2Unary = UnaryInsn<UnaryOp | BinaryOp>

// Abstract instruction in two-address form
export type AbsInsn2 = AbsInsnCopy | AbsInsn2Unary

export type AbsInsn3Unary = UnaryInsn<UnaryOp>

export type AbsInsnBinary = {
    type: "binary",
    dest: Loc,
    source: Operand,
    op: BinaryOp,
    source1: Operand
}

// Abstract instruction in three-address form
export type AbsInsn3 = AbsInsnCopy | AbsInsn3Unary | AbsInsnBinary

export const CopyInsn = (dest: Loc, source: Operand): AbsInsnCopy => ({
    type: 'copy',
    dest,
    source,
})

export const UnaryInsn = <Op>(dest: Loc, op: Op, source: Operand): UnaryInsn<Op> => ({
    type: 'unary',
    dest,
    op,
    source,
})

export const BinaryInsn = (dest: Loc, source: Operand, op: BinaryOp, source1: Operand): AbsInsnBinary => ({
    type: 'binary',
    dest,
    source,
    op,
    source1,
})