Shared CFG: allow init stmts for IfStmt

This commit is contained in:
Owen Mansel-Chan
2026-05-30 07:27:56 +01:00
parent e68a77db63
commit 6befac2412

View File

@@ -95,6 +95,9 @@ signature module AstSig<LocationSig Location> {
Stmt getElse();
}
/** Gets the initializer of this `if` statement, if any. */
default AstNode getIfInit(IfStmt ifstmt) { none() }
/**
* A loop statement. Loop statements are further subclassed into specific
* types of loops.
@@ -1529,6 +1532,13 @@ module Make0<LocationSig Location, AstSig<Location> Ast> {
or
exists(IfStmt ifstmt |
n1.isBefore(ifstmt) and
(
n2.isBefore(getIfInit(ifstmt))
or
not exists(getIfInit(ifstmt)) and n2.isBefore(ifstmt.getCondition())
)
or
n1.isAfter(getIfInit(ifstmt)) and
n2.isBefore(ifstmt.getCondition())
or
n1.isAfterTrue(ifstmt.getCondition()) and