sonolus.py 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

sonolus/backend/ops.py

@@ -1,197 +1,197 @@
-from enum import StrEnum
-
-
-class Op(StrEnum):
-    def __new__(cls, name: str, side_effects: bool, pure: bool, control_flow: bool):
-        obj = str.__new__(cls, name)
-        obj._value_ = name
-        obj.side_effects = side_effects
-        obj.pure = pure
-        obj.control_flow = control_flow
-        return obj
-
-    Abs = ("Abs", False, True, False)
-    Add = ("Add", False, True, False)
-    And = ("And", False, True, True)
-    Arccos = ("Arccos", False, True, False)
-    Arcsin = ("Arcsin", False, True, False)
-    Arctan = ("Arctan", False, True, False)
-    Arctan2 = ("Arctan2", False, True, False)
-    BeatToBPM = ("BeatToBPM", False, True, False)
-    BeatToStartingBeat = ("BeatToStartingBeat", False, True, False)
-    BeatToStartingTime = ("BeatToStartingTime", False, True, False)
-    BeatToTime = ("BeatToTime", False, True, False)
-    Block = ("Block", False, True, True)
-    Break = ("Break", True, False, True)
-    Ceil = ("Ceil", False, True, False)
-    Clamp = ("Clamp", False, True, False)
-    Copy = ("Copy", True, False, False)
-    Cos = ("Cos", False, True, False)
-    Cosh = ("Cosh", False, True, False)
-    DebugLog = ("DebugLog", True, False, False)
-    DebugPause = ("DebugPause", True, False, False)
-    DecrementPost = ("DecrementPost", True, False, False)
-    DecrementPostPointed = ("DecrementPostPointed", True, False, False)
-    DecrementPostShifted = ("DecrementPostShifted", True, False, False)
-    DecrementPre = ("DecrementPre", True, False, False)
-    DecrementPrePointed = ("DecrementPrePointed", True, False, False)
-    DecrementPreShifted = ("DecrementPreShifted", True, False, False)
-    Degree = ("Degree", False, True, False)
-    DestroyParticleEffect = ("DestroyParticleEffect", True, False, False)
-    Divide = ("Divide", False, True, False)
-    DoWhile = ("DoWhile", False, True, True)
-    Draw = ("Draw", True, False, False)
-    DrawCurvedB = ("DrawCurvedB", True, False, False)
-    DrawCurvedBT = ("DrawCurvedBT", True, False, False)
-    DrawCurvedL = ("DrawCurvedL", True, False, False)
-    DrawCurvedLR = ("DrawCurvedLR", True, False, False)
-    DrawCurvedR = ("DrawCurvedR", True, False, False)
-    DrawCurvedT = ("DrawCurvedT", True, False, False)
-    EaseInBack = ("EaseInBack", False, True, False)
-    EaseInCirc = ("EaseInCirc", False, True, False)
-    EaseInCubic = ("EaseInCubic", False, True, False)
-    EaseInElastic = ("EaseInElastic", False, True, False)
-    EaseInExpo = ("EaseInExpo", False, True, False)
-    EaseInOutBack = ("EaseInOutBack", False, True, False)
-    EaseInOutCirc = ("EaseInOutCirc", False, True, False)
-    EaseInOutCubic = ("EaseInOutCubic", False, True, False)
-    EaseInOutElastic = ("EaseInOutElastic", False, True, False)
-    EaseInOutExpo = ("EaseInOutExpo", False, True, False)
-    EaseInOutQuad = ("EaseInOutQuad", False, True, False)
-    EaseInOutQuart = ("EaseInOutQuart", False, True, False)
-    EaseInOutQuint = ("EaseInOutQuint", False, True, False)
-    EaseInOutSine = ("EaseInOutSine", False, True, False)
-    EaseInQuad = ("EaseInQuad", False, True, False)
-    EaseInQuart = ("EaseInQuart", False, True, False)
-    EaseInQuint = ("EaseInQuint", False, True, False)
-    EaseInSine = ("EaseInSine", False, True, False)
-    EaseOutBack = ("EaseOutBack", False, True, False)
-    EaseOutCirc = ("EaseOutCirc", False, True, False)
-    EaseOutCubic = ("EaseOutCubic", False, True, False)
-    EaseOutElastic = ("EaseOutElastic", False, True, False)
-    EaseOutExpo = ("EaseOutExpo", False, True, False)
-    EaseOutInBack = ("EaseOutInBack", False, True, False)
-    EaseOutInCirc = ("EaseOutInCirc", False, True, False)
-    EaseOutInCubic = ("EaseOutInCubic", False, True, False)
-    EaseOutInElastic = ("EaseOutInElastic", False, True, False)
-    EaseOutInExpo = ("EaseOutInExpo", False, True, False)
-    EaseOutInQuad = ("EaseOutInQuad", False, True, False)
-    EaseOutInQuart = ("EaseOutInQuart", False, True, False)
-    EaseOutInQuint = ("EaseOutInQuint", False, True, False)
-    EaseOutInSine = ("EaseOutInSine", False, True, False)
-    EaseOutQuad = ("EaseOutQuad", False, True, False)
-    EaseOutQuart = ("EaseOutQuart", False, True, False)
-    EaseOutQuint = ("EaseOutQuint", False, True, False)
-    EaseOutSine = ("EaseOutSine", False, True, False)
-    Equal = ("Equal", False, True, False)
-    Execute = ("Execute", False, True, False)
-    Execute0 = ("Execute0", False, True, False)
-    ExportValue = ("ExportValue", True, False, False)
-    Floor = ("Floor", False, True, False)
-    Frac = ("Frac", False, True, False)
-    Get = ("Get", False, False, False)
-    GetPointed = ("GetPointed", False, False, False)
-    GetShifted = ("GetShifted", False, False, False)
-    Greater = ("Greater", False, True, False)
-    GreaterOr = ("GreaterOr", False, True, False)
-    HasEffectClip = ("HasEffectClip", False, True, False)
-    HasParticleEffect = ("HasParticleEffect", False, True, False)
-    HasSkinSprite = ("HasSkinSprite", False, True, False)
-    If = ("If", False, True, True)
-    IncrementPost = ("IncrementPost", True, False, False)
-    IncrementPostPointed = ("IncrementPostPointed", True, False, False)
-    IncrementPostShifted = ("IncrementPostShifted", True, False, False)
-    IncrementPre = ("IncrementPre", True, False, False)
-    IncrementPrePointed = ("IncrementPrePointed", True, False, False)
-    IncrementPreShifted = ("IncrementPreShifted", True, False, False)
-    Judge = ("Judge", False, True, False)
-    JudgeSimple = ("JudgeSimple", False, True, False)
-    JumpLoop = ("JumpLoop", False, True, True)
-    Lerp = ("Lerp", False, True, False)
-    LerpClamped = ("LerpClamped", False, True, False)
-    Less = ("Less", False, True, False)
-    LessOr = ("LessOr", False, True, False)
-    Log = ("Log", False, True, False)
-    Max = ("Max", False, True, False)
-    Min = ("Min", False, True, False)
-    Mod = ("Mod", False, True, False)
-    MoveParticleEffect = ("MoveParticleEffect", True, False, False)
-    Multiply = ("Multiply", False, True, False)
-    Negate = ("Negate", False, True, False)
-    Not = ("Not", False, True, False)
-    NotEqual = ("NotEqual", False, True, False)
-    Or = ("Or", False, True, True)
-    Paint = ("Paint", True, False, False)
-    Play = ("Play", True, False, False)
-    PlayLooped = ("PlayLooped", True, False, False)
-    PlayLoopedScheduled = ("PlayLoopedScheduled", True, False, False)
-    PlayScheduled = ("PlayScheduled", True, False, False)
-    Power = ("Power", False, True, False)
-    Print = ("Print", True, False, False)
-    Radian = ("Radian", False, True, False)
-    Random = ("Random", False, False, False)
-    RandomInteger = ("RandomInteger", False, False, False)
-    Rem = ("Rem", False, True, False)
-    Remap = ("Remap", False, True, False)
-    RemapClamped = ("RemapClamped", False, True, False)
-    Round = ("Round", False, True, False)
-    Set = ("Set", True, False, False)
-    SetAdd = ("SetAdd", True, False, False)
-    SetAddPointed = ("SetAddPointed", True, False, False)
-    SetAddShifted = ("SetAddShifted", True, False, False)
-    SetDivide = ("SetDivide", True, False, False)
-    SetDividePointed = ("SetDividePointed", True, False, False)
-    SetDivideShifted = ("SetDivideShifted", True, False, False)
-    SetMod = ("SetMod", True, False, False)
-    SetModPointed = ("SetModPointed", True, False, False)
-    SetModShifted = ("SetModShifted", True, False, False)
-    SetMultiply = ("SetMultiply", True, False, False)
-    SetMultiplyPointed = ("SetMultiplyPointed", True, False, False)
-    SetMultiplyShifted = ("SetMultiplyShifted", True, False, False)
-    SetPointed = ("SetPointed", True, False, False)
-    SetPower = ("SetPower", True, False, False)
-    SetPowerPointed = ("SetPowerPointed", True, False, False)
-    SetPowerShifted = ("SetPowerShifted", True, False, False)
-    SetRem = ("SetRem", True, False, False)
-    SetRemPointed = ("SetRemPointed", True, False, False)
-    SetRemShifted = ("SetRemShifted", True, False, False)
-    SetShifted = ("SetShifted", True, False, False)
-    SetSubtract = ("SetSubtract", True, False, False)
-    SetSubtractPointed = ("SetSubtractPointed", True, False, False)
-    SetSubtractShifted = ("SetSubtractShifted", True, False, False)
-    Sign = ("Sign", False, True, False)
-    Sin = ("Sin", False, True, False)
-    Sinh = ("Sinh", False, True, False)
-    Spawn = ("Spawn", True, False, False)
-    SpawnParticleEffect = ("SpawnParticleEffect", True, False, False)
-    StackEnter = ("StackEnter", True, False, False)
-    StackGet = ("StackGet", False, False, False)
-    StackGetFrame = ("StackGetFrame", False, False, False)
-    StackGetFramePointer = ("StackGetFramePointer", False, False, False)
-    StackGetPointer = ("StackGetPointer", False, False, False)
-    StackGrow = ("StackGrow", True, False, False)
-    StackInit = ("StackInit", True, False, False)
-    StackLeave = ("StackLeave", True, False, False)
-    StackPop = ("StackPop", True, False, False)
-    StackPush = ("StackPush", True, False, False)
-    StackSet = ("StackSet", True, False, False)
-    StackSetFrame = ("StackSetFrame", True, False, False)
-    StackSetFramePointer = ("StackSetFramePointer", True, False, False)
-    StackSetPointer = ("StackSetPointer", True, False, False)
-    StopLooped = ("StopLooped", True, False, False)
-    StopLoopedScheduled = ("StopLoopedScheduled", True, False, False)
-    Subtract = ("Subtract", False, True, False)
-    Switch = ("Switch", False, True, True)
-    SwitchInteger = ("SwitchInteger", False, True, True)
-    SwitchIntegerWithDefault = ("SwitchIntegerWithDefault", False, True, True)
-    SwitchWithDefault = ("SwitchWithDefault", False, True, True)
-    Tan = ("Tan", False, True, False)
-    Tanh = ("Tanh", False, True, False)
-    TimeToScaledTime = ("TimeToScaledTime", False, True, False)
-    TimeToStartingScaledTime = ("TimeToStartingScaledTime", False, True, False)
-    TimeToStartingTime = ("TimeToStartingTime", False, True, False)
-    TimeToTimeScale = ("TimeToTimeScale", False, True, False)
-    Trunc = ("Trunc", False, True, False)
-    Unlerp = ("Unlerp", False, True, False)
-    UnlerpClamped = ("UnlerpClamped", False, True, False)
-    While = ("While", False, True, True)
+from enum import StrEnum
+
+
+class Op(StrEnum):
+    def __new__(cls, name: str, side_effects: bool, pure: bool, control_flow: bool):
+        obj = str.__new__(cls, name)
+        obj._value_ = name
+        obj.side_effects = side_effects
+        obj.pure = pure
+        obj.control_flow = control_flow
+        return obj
+
+    Abs = ("Abs", False, True, False)
+    Add = ("Add", False, True, False)
+    And = ("And", False, True, True)
+    Arccos = ("Arccos", False, True, False)
+    Arcsin = ("Arcsin", False, True, False)
+    Arctan = ("Arctan", False, True, False)
+    Arctan2 = ("Arctan2", False, True, False)
+    BeatToBPM = ("BeatToBPM", False, True, False)
+    BeatToStartingBeat = ("BeatToStartingBeat", False, True, False)
+    BeatToStartingTime = ("BeatToStartingTime", False, True, False)
+    BeatToTime = ("BeatToTime", False, True, False)
+    Block = ("Block", False, True, True)
+    Break = ("Break", True, False, True)
+    Ceil = ("Ceil", False, True, False)
+    Clamp = ("Clamp", False, True, False)
+    Copy = ("Copy", True, False, False)
+    Cos = ("Cos", False, True, False)
+    Cosh = ("Cosh", False, True, False)
+    DebugLog = ("DebugLog", True, False, False)
+    DebugPause = ("DebugPause", True, False, False)
+    DecrementPost = ("DecrementPost", True, False, False)
+    DecrementPostPointed = ("DecrementPostPointed", True, False, False)
+    DecrementPostShifted = ("DecrementPostShifted", True, False, False)
+    DecrementPre = ("DecrementPre", True, False, False)
+    DecrementPrePointed = ("DecrementPrePointed", True, False, False)
+    DecrementPreShifted = ("DecrementPreShifted", True, False, False)
+    Degree = ("Degree", False, True, False)
+    DestroyParticleEffect = ("DestroyParticleEffect", True, False, False)
+    Divide = ("Divide", False, True, False)
+    DoWhile = ("DoWhile", False, True, True)
+    Draw = ("Draw", True, False, False)
+    DrawCurvedB = ("DrawCurvedB", True, False, False)
+    DrawCurvedBT = ("DrawCurvedBT", True, False, False)
+    DrawCurvedL = ("DrawCurvedL", True, False, False)
+    DrawCurvedLR = ("DrawCurvedLR", True, False, False)
+    DrawCurvedR = ("DrawCurvedR", True, False, False)
+    DrawCurvedT = ("DrawCurvedT", True, False, False)
+    EaseInBack = ("EaseInBack", False, True, False)
+    EaseInCirc = ("EaseInCirc", False, True, False)
+    EaseInCubic = ("EaseInCubic", False, True, False)
+    EaseInElastic = ("EaseInElastic", False, True, False)
+    EaseInExpo = ("EaseInExpo", False, True, False)
+    EaseInOutBack = ("EaseInOutBack", False, True, False)
+    EaseInOutCirc = ("EaseInOutCirc", False, True, False)
+    EaseInOutCubic = ("EaseInOutCubic", False, True, False)
+    EaseInOutElastic = ("EaseInOutElastic", False, True, False)
+    EaseInOutExpo = ("EaseInOutExpo", False, True, False)
+    EaseInOutQuad = ("EaseInOutQuad", False, True, False)
+    EaseInOutQuart = ("EaseInOutQuart", False, True, False)
+    EaseInOutQuint = ("EaseInOutQuint", False, True, False)
+    EaseInOutSine = ("EaseInOutSine", False, True, False)
+    EaseInQuad = ("EaseInQuad", False, True, False)
+    EaseInQuart = ("EaseInQuart", False, True, False)
+    EaseInQuint = ("EaseInQuint", False, True, False)
+    EaseInSine = ("EaseInSine", False, True, False)
+    EaseOutBack = ("EaseOutBack", False, True, False)
+    EaseOutCirc = ("EaseOutCirc", False, True, False)
+    EaseOutCubic = ("EaseOutCubic", False, True, False)
+    EaseOutElastic = ("EaseOutElastic", False, True, False)
+    EaseOutExpo = ("EaseOutExpo", False, True, False)
+    EaseOutInBack = ("EaseOutInBack", False, True, False)
+    EaseOutInCirc = ("EaseOutInCirc", False, True, False)
+    EaseOutInCubic = ("EaseOutInCubic", False, True, False)
+    EaseOutInElastic = ("EaseOutInElastic", False, True, False)
+    EaseOutInExpo = ("EaseOutInExpo", False, True, False)
+    EaseOutInQuad = ("EaseOutInQuad", False, True, False)
+    EaseOutInQuart = ("EaseOutInQuart", False, True, False)
+    EaseOutInQuint = ("EaseOutInQuint", False, True, False)
+    EaseOutInSine = ("EaseOutInSine", False, True, False)
+    EaseOutQuad = ("EaseOutQuad", False, True, False)
+    EaseOutQuart = ("EaseOutQuart", False, True, False)
+    EaseOutQuint = ("EaseOutQuint", False, True, False)
+    EaseOutSine = ("EaseOutSine", False, True, False)
+    Equal = ("Equal", False, True, False)
+    Execute = ("Execute", False, True, False)
+    Execute0 = ("Execute0", False, True, False)
+    ExportValue = ("ExportValue", True, False, False)
+    Floor = ("Floor", False, True, False)
+    Frac = ("Frac", False, True, False)
+    Get = ("Get", False, False, False)
+    GetPointed = ("GetPointed", False, False, False)
+    GetShifted = ("GetShifted", False, False, False)
+    Greater = ("Greater", False, True, False)
+    GreaterOr = ("GreaterOr", False, True, False)
+    HasEffectClip = ("HasEffectClip", False, True, False)
+    HasParticleEffect = ("HasParticleEffect", False, True, False)
+    HasSkinSprite = ("HasSkinSprite", False, True, False)
+    If = ("If", False, True, True)
+    IncrementPost = ("IncrementPost", True, False, False)
+    IncrementPostPointed = ("IncrementPostPointed", True, False, False)
+    IncrementPostShifted = ("IncrementPostShifted", True, False, False)
+    IncrementPre = ("IncrementPre", True, False, False)
+    IncrementPrePointed = ("IncrementPrePointed", True, False, False)
+    IncrementPreShifted = ("IncrementPreShifted", True, False, False)
+    Judge = ("Judge", False, True, False)
+    JudgeSimple = ("JudgeSimple", False, True, False)
+    JumpLoop = ("JumpLoop", False, True, True)
+    Lerp = ("Lerp", False, True, False)
+    LerpClamped = ("LerpClamped", False, True, False)
+    Less = ("Less", False, True, False)
+    LessOr = ("LessOr", False, True, False)
+    Log = ("Log", False, True, False)
+    Max = ("Max", False, True, False)
+    Min = ("Min", False, True, False)
+    Mod = ("Mod", False, True, False)
+    MoveParticleEffect = ("MoveParticleEffect", True, False, False)
+    Multiply = ("Multiply", False, True, False)
+    Negate = ("Negate", False, True, False)
+    Not = ("Not", False, True, False)
+    NotEqual = ("NotEqual", False, True, False)
+    Or = ("Or", False, True, True)
+    Paint = ("Paint", True, False, False)
+    Play = ("Play", True, False, False)
+    PlayLooped = ("PlayLooped", True, False, False)
+    PlayLoopedScheduled = ("PlayLoopedScheduled", True, False, False)
+    PlayScheduled = ("PlayScheduled", True, False, False)
+    Power = ("Power", False, True, False)
+    Print = ("Print", True, False, False)
+    Radian = ("Radian", False, True, False)
+    Random = ("Random", False, False, False)
+    RandomInteger = ("RandomInteger", False, False, False)
+    Rem = ("Rem", False, True, False)
+    Remap = ("Remap", False, True, False)
+    RemapClamped = ("RemapClamped", False, True, False)
+    Round = ("Round", False, True, False)
+    Set = ("Set", True, False, False)
+    SetAdd = ("SetAdd", True, False, False)
+    SetAddPointed = ("SetAddPointed", True, False, False)
+    SetAddShifted = ("SetAddShifted", True, False, False)
+    SetDivide = ("SetDivide", True, False, False)
+    SetDividePointed = ("SetDividePointed", True, False, False)
+    SetDivideShifted = ("SetDivideShifted", True, False, False)
+    SetMod = ("SetMod", True, False, False)
+    SetModPointed = ("SetModPointed", True, False, False)
+    SetModShifted = ("SetModShifted", True, False, False)
+    SetMultiply = ("SetMultiply", True, False, False)
+    SetMultiplyPointed = ("SetMultiplyPointed", True, False, False)
+    SetMultiplyShifted = ("SetMultiplyShifted", True, False, False)
+    SetPointed = ("SetPointed", True, False, False)
+    SetPower = ("SetPower", True, False, False)
+    SetPowerPointed = ("SetPowerPointed", True, False, False)
+    SetPowerShifted = ("SetPowerShifted", True, False, False)
+    SetRem = ("SetRem", True, False, False)
+    SetRemPointed = ("SetRemPointed", True, False, False)
+    SetRemShifted = ("SetRemShifted", True, False, False)
+    SetShifted = ("SetShifted", True, False, False)
+    SetSubtract = ("SetSubtract", True, False, False)
+    SetSubtractPointed = ("SetSubtractPointed", True, False, False)
+    SetSubtractShifted = ("SetSubtractShifted", True, False, False)
+    Sign = ("Sign", False, True, False)
+    Sin = ("Sin", False, True, False)
+    Sinh = ("Sinh", False, True, False)
+    Spawn = ("Spawn", True, False, False)
+    SpawnParticleEffect = ("SpawnParticleEffect", True, False, False)
+    StackEnter = ("StackEnter", True, False, False)
+    StackGet = ("StackGet", False, False, False)
+    StackGetFrame = ("StackGetFrame", False, False, False)
+    StackGetFramePointer = ("StackGetFramePointer", False, False, False)
+    StackGetPointer = ("StackGetPointer", False, False, False)
+    StackGrow = ("StackGrow", True, False, False)
+    StackInit = ("StackInit", True, False, False)
+    StackLeave = ("StackLeave", True, False, False)
+    StackPop = ("StackPop", True, False, False)
+    StackPush = ("StackPush", True, False, False)
+    StackSet = ("StackSet", True, False, False)
+    StackSetFrame = ("StackSetFrame", True, False, False)
+    StackSetFramePointer = ("StackSetFramePointer", True, False, False)
+    StackSetPointer = ("StackSetPointer", True, False, False)
+    StopLooped = ("StopLooped", True, False, False)
+    StopLoopedScheduled = ("StopLoopedScheduled", True, False, False)
+    Subtract = ("Subtract", False, True, False)
+    Switch = ("Switch", False, True, True)
+    SwitchInteger = ("SwitchInteger", False, True, True)
+    SwitchIntegerWithDefault = ("SwitchIntegerWithDefault", False, True, True)
+    SwitchWithDefault = ("SwitchWithDefault", False, True, True)
+    Tan = ("Tan", False, True, False)
+    Tanh = ("Tanh", False, True, False)
+    TimeToScaledTime = ("TimeToScaledTime", False, True, False)
+    TimeToStartingScaledTime = ("TimeToStartingScaledTime", False, True, False)
+    TimeToStartingTime = ("TimeToStartingTime", False, True, False)
+    TimeToTimeScale = ("TimeToTimeScale", False, True, False)
+    Trunc = ("Trunc", False, True, False)
+    Unlerp = ("Unlerp", False, True, False)
+    UnlerpClamped = ("UnlerpClamped", False, True, False)
+    While = ("While", False, True, True)

sonolus/backend/optimize.py

@@ -1,9 +1,37 @@
-from sonolus.backend.allocate import AllocateBasic
-from sonolus.backend.flow import BasicBlock
-from sonolus.backend.simplify import CoalesceFlow
-
-
-def optimize_and_allocate(cfg: BasicBlock):
-    cfg = CoalesceFlow().run(cfg)
-    cfg = AllocateBasic().run(cfg)
-    return cfg
+from sonolus.backend.allocate import Allocate, AllocateBasic
+from sonolus.backend.coalesce import CopyCoalesce
+from sonolus.backend.constant_evaluation import SparseConditionalConstantPropagation
+from sonolus.backend.dead_code import DeadCodeElimination, UnreachableCodeElimination
+from sonolus.backend.flow import BasicBlock
+from sonolus.backend.inlining import InlineVars
+from sonolus.backend.passes import run_passes
+from sonolus.backend.simplify import CoalesceFlow
+from sonolus.backend.ssa import FromSSA, ToSSA
+
+MINIMAL_PASSES = [
+    CoalesceFlow(),
+    UnreachableCodeElimination(),
+    AllocateBasic(),
+]
+
+STANDARD_PASSES = [
+    CoalesceFlow(),
+    UnreachableCodeElimination(),
+    DeadCodeElimination(),
+    ToSSA(),
+    SparseConditionalConstantPropagation(),
+    UnreachableCodeElimination(),
+    DeadCodeElimination(),
+    CoalesceFlow(),
+    InlineVars(),
+    FromSSA(),
+    CoalesceFlow(),
+    CopyCoalesce(),
+    DeadCodeElimination(),
+    CoalesceFlow(),
+    Allocate(),
+]
+
+
+def optimize_and_allocate(cfg: BasicBlock):
+    return run_passes(cfg, STANDARD_PASSES)

sonolus/backend/passes.py

@@ -1,6 +1,52 @@
-from sonolus.backend.flow import BasicBlock
-
-
-class CompilerPass:
-    def run(self, entry: BasicBlock) -> BasicBlock:
-        pass
+from __future__ import annotations
+
+from abc import ABC, abstractmethod
+from collections import deque
+
+from sonolus.backend.flow import BasicBlock
+
+
+class CompilerPass(ABC):
+    def requires(self) -> set[CompilerPass]:
+        return set()
+
+    def preserves(self) -> set[CompilerPass] | None:
+        return None
+
+    def destroys(self) -> set[CompilerPass] | None:
+        return None
+
+    def applies(self) -> set[CompilerPass]:
+        return {self}
+
+    def exists_after(self, passes: set[CompilerPass]) -> set[CompilerPass]:
+        preserved = self.preserves()
+        destroyed = self.destroys()
+        if destroyed is None and preserved is None:
+            return self.applies()
+        if preserved is not None:
+            passes = {p for p in passes if p in preserved}
+        if destroyed is not None:
+            passes = {p for p in passes if p not in destroyed}
+        return passes | self.applies()
+
+    @abstractmethod
+    def run(self, entry: BasicBlock) -> BasicBlock:
+        pass
+
+
+def run_passes(entry: BasicBlock, passes: list[CompilerPass]) -> BasicBlock:
+    active_passes = set()
+    queue = deque(passes)
+    while queue:
+        if len(queue) > 99:
+            raise RuntimeError("Likely unsatisfiable pass requirements")
+        current_pass = queue.popleft()
+        missing_requirements = current_pass.requires() - active_passes
+        if missing_requirements:
+            queue.appendleft(current_pass)
+            queue.extendleft(missing_requirements)
+            continue
+        entry = current_pass.run(entry)
+        active_passes = current_pass.exists_after(active_passes)
+    return entry

sonolus/backend/simplify.py

@@ -1,30 +1,47 @@
-from sonolus.backend.flow import BasicBlock
-from sonolus.backend.passes import CompilerPass
-
-
-class CoalesceFlow(CompilerPass):
-    def run(self, entry: BasicBlock) -> BasicBlock:
-        queue = [entry]
-        processed = set()
-        while queue:
-            block = queue.pop()
-            if block in processed:
-                continue
-            processed.add(block)
-            if len(block.outgoing) != 1:
-                queue.extend(edge.dst for edge in block.outgoing)
-                continue
-            next_block = next(iter(block.outgoing)).dst
-            if len(next_block.incoming) != 1:
-                queue.append(next_block)
-                continue
-            block.statements.extend(next_block.statements)
-            block.test = next_block.test
-            for edge in next_block.outgoing:
-                edge.src = block
-            block.outgoing = next_block.outgoing
-            processed.add(next_block)
-            queue.extend(edge.dst for edge in block.outgoing)
-            processed.remove(block)
-            queue.append(block)
-        return entry
+from sonolus.backend.flow import BasicBlock
+from sonolus.backend.ir import IRGet, IRSet
+from sonolus.backend.passes import CompilerPass
+
+
+class CoalesceFlow(CompilerPass):
+    def run(self, entry: BasicBlock) -> BasicBlock:
+        queue = [entry]
+        processed = set()
+        while queue:
+            block = queue.pop()
+            if block in processed:
+                continue
+            processed.add(block)
+            if len(block.outgoing) != 1:
+                queue.extend(edge.dst for edge in block.outgoing)
+                continue
+            next_block = next(iter(block.outgoing)).dst
+            if len(next_block.incoming) != 1:
+                queue.append(next_block)
+                if not block.statements and not block.phis and not next_block.phis:
+                    for edge in block.incoming:
+                        edge.dst = next_block
+                        next_block.incoming.add(edge)
+                    for edge in block.outgoing:  # There should be exactly one
+                        next_block.incoming.remove(edge)
+                    if block is entry:
+                        entry = next_block
+                continue
+            for p, args in next_block.phis.items():
+                if block not in args:
+                    continue
+                block.statements.append(IRSet(p, IRGet(args[block])))
+            block.statements.extend(next_block.statements)
+            block.test = next_block.test
+            block.outgoing = next_block.outgoing
+            for edge in block.outgoing:
+                edge.src = block
+                dst = edge.dst
+                for args in dst.phis.values():
+                    if next_block in args:
+                        args[block] = args.pop(next_block)
+            processed.add(next_block)
+            queue.extend(edge.dst for edge in block.outgoing)
+            processed.remove(block)
+            queue.append(block)
+        return entry