Add sorts/types/names to classes

src/lib.rs

@@ -60,6 +60,9 @@ pub struct EGraph {
     pub nodes: IndexMap<NodeId, Node>,
     #[cfg_attr(feature = "serde", serde(default))]
     pub root_eclasses: Vec<ClassId>,
+    // Optional mapping of e-class ids to some additional data about the e-class
+    #[cfg_attr(feature = "serde", serde(default))]
+    pub class_data: IndexMap<ClassId, ClassData>,
     #[cfg_attr(feature = "serde", serde(skip))]
     once_cell_classes: OnceCell<IndexMap<ClassId, Class>>,
 }
@@ -180,3 +183,10 @@ pub struct Class {
     pub id: ClassId,
     pub nodes: Vec<NodeId>,
 }
+
+#[cfg_attr(feature = "serde", derive(serde::Deserialize, serde::Serialize))]
+#[derive(Debug, Clone, PartialEq)]
+pub struct ClassData {
+    #[cfg_attr(feature = "serde", serde(rename = "type"))]
+    pub typ: Option<String>,
+}

tests/round_trip.rs

@@ -3,7 +3,7 @@ use egraph_serialize::*;
 #[test]
 fn test_round_trip() {
     let mut n_tested = 0;
-    let pattern = "../extraction-gym/data/**/*.json";
+    let pattern = "tests/*.json";
     for entry in glob::glob(pattern).expect("Failed to read glob pattern") {
         let entry = entry.unwrap();
         println!("Testing {:?}", entry);

tests/tiny.json

@@ -18,5 +18,10 @@
             "eclass": "12",
             "cost": 6.0
         }
+    },
+    "class_data": {
+        "18": {
+            "type": "foobar"
+        }
     }
 }