chore: Add convenience method for extracting variables from an SAtom

plaans · Apr 18, 2024 · 77b0aae · 77b0aae
1 parent ccbdbeb
commit 77b0aae
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/solver/src/model/lang/sym.rs b/solver/src/model/lang/sym.rs
@@ -64,6 +64,13 @@ impl SAtom {
             SAtom::Cst(s) => IAtom::new(IVar::ZERO, usize::from(s.sym) as IntCst),
         }
     }
+
+    pub fn variable(&self) -> VarRef {
+        match self {
+            SAtom::Var(v) => v.var,
+            SAtom::Cst(_) => VarRef::ZERO,
+        }
+    }
 }
 
 impl From<SVar> for VarRef {