add a predicates function to theory

potassco · Nov 22, 2024 · 70f2c4a · 70f2c4a
1 parent ef9777e
commit 70f2c4a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/syntax_tree/fol.rs b/src/syntax_tree/fol.rs
@@ -906,6 +906,14 @@ pub struct Theory {
 impl_node!(Theory, Format, TheoryParser);
 
 impl Theory {
+    pub fn predicates(&self) -> IndexSet<Predicate> {
+        let mut preds = IndexSet::new();
+        for formula in self {
+            preds.extend(formula.predicates())
+        }
+        preds
+    }
+
     pub fn replace_placeholders(self, mapping: &IndexMap<String, FunctionConstant>) -> Self {
         self.into_iter()
             .map(|f| f.replace_placeholders(mapping))