1 files changed, 12 insertions, 0 deletions
diff --git a/ecs/src/component/storage/graph.rs b/ecs/src/component/storage/graph.rs
index a221877..11160e7 100644
--- a/ecs/src/component/storage/graph.rs
+++ b/ecs/src/component/storage/graph.rs
@@ -71,6 +71,12 @@ impl Graph
         }))
     }
 
+    #[cfg(feature = "vizoxide")]
+    pub fn iter_nodes(&self) -> impl Iterator<Item = &ArchetypeNode>
+    {
+        self.nodes.iter()
+    }
+
     pub fn dfs_archetype_add_edges(
         &self,
         archetype_id: ArchetypeId,
@@ -233,6 +239,12 @@ impl ArchetypeNode
         self.edges.entry(component_id).or_insert_with(insert_fn)
     }
 
+    #[cfg(feature = "vizoxide")]
+    pub fn iter_edges(&self) -> impl Iterator<Item = (&Uid, &ArchetypeEdges)>
+    {
+        self.edges.iter()
+    }
+
     pub fn get_edges_mut(&mut self, component_id: Uid) -> Option<&mut ArchetypeEdges>
     {
         debug_assert_eq!(component_id.kind(), UidKind::Component);