Fix color wraparound when number of types exceed the max colors

egraphs-good · Oct 10, 2023 · 686c89a · 686c89a
1 parent ea6dd17
commit 686c89a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/graphviz.rs b/src/graphviz.rs
@@ -81,7 +81,7 @@ impl EGraph {
         let mut typ_colors = std::collections::HashMap::new();
 
         for (typ, class_to_node) in class_nodes {
-            let next_color = typ_colors.len() % N_COLORS + INITIAL_COLOR;
+            let next_color = (typ_colors.len() + INITIAL_COLOR) % N_COLORS;
             let color = typ_colors.entry(typ).or_insert(next_color);
             stmts.push(stmt!(attr!("fillcolor", color)));
             for (class_id, nodes) in class_to_node {