Util

insert_qubit_idx_from_address

insert_qubit_idx_from_address(
    address: Address, stmt_to_insert_before: Statement
) -> tuple[ir.SSAValue, ...] | None

Extract qubit indices from an Address and insert them into the SSA form.

Source code in .venv/lib/python3.12/site-packages/bloqade/stim/rewrite/util.py

def insert_qubit_idx_from_address(
    address: Address, stmt_to_insert_before: ir.Statement
) -> tuple[ir.SSAValue, ...] | None:
    """
    Extract qubit indices from an Address and insert them into the SSA form.
    """
    qubit_idx_ssas = []
    if isinstance(address, AddressReg):
        for qubit_idx in address.qubits:
            create_and_insert_qubit_idx_stmt(
                qubit_idx.data, stmt_to_insert_before, qubit_idx_ssas
            )
    elif isinstance(address, AddressQubit):
        create_and_insert_qubit_idx_stmt(
            address.data, stmt_to_insert_before, qubit_idx_ssas
        )
    else:
        return

    return tuple(qubit_idx_ssas)