Ifs handling

IsIfElseSimplifiable dataclass

IsIfElseSimplifiable()

contains_ifelse

contains_ifelse(stmt: IfElse) -> bool

Check if the IfElse statement contains another IfElse statement.

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

def contains_ifelse(self, stmt: scf.IfElse) -> bool:
    """Check if the IfElse statement contains another IfElse statement."""
    for child in stmt.walk(include_self=False):
        if isinstance(child, scf.IfElse):
            return True
    return False

has_else_body

has_else_body(stmt: IfElse) -> bool

Check if the IfElse statement has an else body.

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

def has_else_body(self, stmt: scf.IfElse) -> bool:
    """Check if the IfElse statement has an else body."""
    if stmt.else_body.blocks and not (
        len(stmt.else_body.blocks[0].stmts) == 1
        and isinstance(stmt.else_body.blocks[0].last_stmt, scf.Yield)
    ):
        return True

    return False

is_nested_ifelse

is_nested_ifelse(stmt: IfElse) -> bool

Check if the IfElse statement is nested.

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

def is_nested_ifelse(self, stmt: scf.IfElse) -> bool:
    """Check if the IfElse statement is nested."""
    if stmt.parent_stmt is not None:
        if isinstance(stmt.parent_stmt, scf.IfElse) or isinstance(
            stmt.parent_stmt.parent_stmt, scf.IfElse
        ):
            return True
        else:
            return False
    else:
        return False

StimSplitIfStmts dataclass

StimSplitIfStmts()

Bases: IsIfElseSimplifiable, SplitIfStmts

Splits the then body of an if-else statement into multiple if statements

Given an IfElse with multiple valid statements in the then-body:

if measure_result

squin.x(q0) squin.y(q1)

this should be rewritten to:

if measure_result

squin.x(q0)

if measure_result

squin.y(q1)