add support for storing if a node is subsumed

egraphs-good · Sep 11, 2024 · 325f7c6 · 325f7c6
1 parent 9ce2812
commit 325f7c6
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -162,6 +162,14 @@ impl std::ops::Index<&ClassId> for EGraph {
     }
 }
 
+#[cfg_attr(feature = "serde", derive(serde::Deserialize, serde::Serialize))]
+#[derive(Debug, Default, Clone, Hash, PartialEq, Eq, PartialOrd, Ord)]
+pub struct NodeData {
+    //
+    #[cfg_attr(feature = "serde", serde(default))]
+    pub subsumed: bool,
+}
+
 #[cfg_attr(feature = "serde", derive(serde::Deserialize, serde::Serialize))]
 #[derive(Clone, Debug, PartialEq, Eq, Hash, PartialOrd, Ord)]
 pub struct Node {
@@ -171,6 +179,9 @@ pub struct Node {
     pub eclass: ClassId,
     #[cfg_attr(feature = "serde", serde(default = "one"))]
     pub cost: Cost,
+    // Optional data about the node
+    #[cfg_attr(feature = "serde", serde(default))]
+    pub data: NodeData,
 }
 
 impl Node {