fix: Ignore hierarchy elements that are not in the graph when displaying

CQCL · Jul 11, 2023 · ac1891c · ac1891c
1 parent bc221b8
commit ac1891c
Showing 1 changed file with 13 additions and 10 deletions.
diff --git a/src/dot.rs b/src/dot.rs
@@ -336,17 +336,20 @@ where
  dot.push_str(&node_str);
 
  // Connect the parent to any existing children
- forest.children(n).for_each(|child| {
- dot.push_str(&{
- let from_node = n;
- let to_node = child;
- format!(
- "{} -> {} [style = \"dashed\"] \n",
- hier_node_id(from_node),
- hier_node_id(to_node),
- )
+ forest
+ .children(n)
+ .filter(|c| self.graph.contains_node(c))
+ .for_each(|child| {
+ dot.push_str(&{
+ let from_node = n;
+ let to_node = child;
+ format!(
+ "{} -> {} [style = \"dashed\"] \n",
+ hier_node_id(from_node),
+ hier_node_id(to_node),
+ )
+ });
  });
- });
  }
  }
  }