From 1df8e8bee04c191f6802c88ea4c1044bd86fb3ab Mon Sep 17 00:00:00 2001
From: Adrian Lehmann <info@adrianlehmann.net>
Date: Fri, 7 Mar 2025 15:24:26 -0600
Subject: [PATCH] Add multi-condition rewrites

---
 src/rewrite.rs | 99 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 99 insertions(+)

diff --git a/src/rewrite.rs b/src/rewrite.rs
index eae5755e..0ceb39e6 100644
--- a/src/rewrite.rs
+++ b/src/rewrite.rs
@@ -538,6 +538,105 @@ where
     }
 }
 
+#[derive(Clone)]
+pub struct AndCondition<L, N>
+where
+    L: Language,
+    N: Analysis<L>,
+{
+    pub c1: Arc<dyn Condition<L, N> + Send + Sync>,
+    pub c2: Arc<dyn Condition<L, N> + Send + Sync>,
+}
+
+impl<L: Language, N: Analysis<L>> Condition<L, N> for AndCondition<L, N> {
+    #[inline]
+    fn check(&self, egraph: &mut EGraph<L, N>, id: Id, subst: &Subst) -> bool {
+        self.c1.check(egraph, id, subst) && self.c2.check(egraph, id, subst)
+    }
+    fn vars(&self) -> Vec<Var> {
+        let mut vars = self.c1.vars();
+        vars.extend(self.c2.vars());
+        vars
+    }
+}
+
+#[derive(Clone)]
+pub struct OrCondition<L, N>
+where
+    L: Language,
+    N: Analysis<L>,
+{
+    pub c1: Arc<dyn Condition<L, N> + Send + Sync>,
+    pub c2: Arc<dyn Condition<L, N> + Send + Sync>,
+}
+
+impl<L: Language, N: Analysis<L>> Condition<L, N> for OrCondition<L, N> {
+    #[inline]
+    fn check(&self, egraph: &mut EGraph<L, N>, id: Id, subst: &Subst) -> bool {
+        self.c1.check(egraph, id, subst) || self.c2.check(egraph, id, subst)
+    }
+    fn vars(&self) -> Vec<Var> {
+        let mut vars = self.c1.vars();
+        vars.extend(self.c2.vars());
+        vars
+    }
+}
+
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub struct TrueCondition {}
+
+impl<L: Language, N: Analysis<L>> Condition<L, N> for TrueCondition {
+    #[inline(always)]
+    fn check(&self, _egraph: &mut EGraph<L, N>, _id: Id, _subst: &Subst) -> bool {
+        true
+    }
+    #[inline(always)]
+    fn vars(&self) -> Vec<Var> {
+        vec![]
+    }
+}
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub struct FalseCondition {}
+
+impl<L: Language, N: Analysis<L>> Condition<L, N> for FalseCondition {
+    #[inline(always)]
+    fn check(&self, _egraph: &mut EGraph<L, N>, _id: Id, _subst: &Subst) -> bool {
+        false
+    }
+    #[inline(always)]
+    fn vars(&self) -> Vec<Var> {
+        vec![]
+    }
+}
+
+pub fn any<L, N>(
+    conds: Vec<Arc<dyn Condition<L, N> + Send + Sync>>,
+) -> Arc<dyn Condition<L, N> + Send + Sync>
+where
+    L: Language + 'static,
+    N: Analysis<L> + 'static,
+{
+    conds
+        .into_iter()
+        .fold(Arc::new(FalseCondition {}), |acc, c| {
+            Arc::new(OrCondition { c1: acc, c2: c })
+        })
+}
+
+pub fn all<L, N>(
+    conds: Vec<Arc<dyn Condition<L, N> + Send + Sync>>,
+) -> Arc<dyn Condition<L, N> + Send + Sync>
+where
+    L: Language + 'static,
+    N: Analysis<L> + 'static,
+{
+    conds
+        .into_iter()
+        .fold(Arc::new(TrueCondition {}), |acc, c| {
+            Arc::new(AndCondition { c1: acc, c2: c })
+        })
+}
+
 #[cfg(test)]
 mod tests {