def number.stage
import gralha-text-scan@0.1.0.1/scan.bend as Scanimport Base
158def number.stage(+k: U32, +stage: U32) -> U32:
159 choose(U32, U32.is_eq(k, 0), choose(U32, U32.is_eq(stage, 0), 1, choose(U32, U32.is_eq(stage, 3), 4, stage)),
160 choose(U32, U32.is_eq(k, 1), 2, choose(U32, U32.is_eq(k, 2), 3, stage)))