gbuoy/lib/ir/block.ts

import type { AbsInsn2 } from "./insn"
import type { Loc } from "./loc"

export class BasicBlock {
    insns: Array<AbsInsn2>

    constructor(insns: Array<AbsInsn2>) {
        this.insns = insns
    }

    locs(): Set<Loc> {
        return new Set(
            this.insns.flatMap(insn => {
                const insnLocs = []

                if (insn.type === "copy" || insn.type === "unary") {
                    insnLocs.push(insn.dest)

                    if (typeof insn.source !== "number") {
                        insnLocs.push(insn.source)
                    }
                }

                return insnLocs
            })
        )
    }
}