serialize ur egraphs

mostly for use in extraction gym rn

snippet for egg

One day egg will natively export to this format, but for now you can use this:

pub fn egg_to_serialized_egraph<L, A>(egraph: &EGraph<L, A>) -> egraph_serialize::EGraph
where
    L: Language + Display,
    A: Analysis<L>,
{
    use egraph_serialize::*;
    let mut out = EGraph::default();
    for class in egraph.classes() {
        for (i, node) in class.nodes.iter().enumerate() {
            out.add_node(
                format!("{}.{}", class.id, i),
                Node {
                    op: node.to_string(),
                    children: node
                        .children()
                        .iter()
                        .map(|id| NodeId::from(format!("{}.0", id)))
                        .collect(),
                    eclass: ClassId::from(format!("{}", class.id)),
                    cost: Cost::new(1.0).unwrap(),
                },
            )
        }
    }
    out
}

Don't forget to add something to root_eclasses on the resulting serialized egraph!

Visualization

Check out the ./tests-viz directory to view visualizations of all the test cases with Graphviz.

To remake them, run make tests from the root of this repo. Any tests that don't have SVG visualizations created yet will be generated. You'll need to have Graphviz installed.