expose subgraph iterator (#25)

superdump · Jul 14, 2021 · b340e2d · b340e2d
1 parent fb804e0
commit b340e2d
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/pipelined/bevy_render2/src/render_graph/graph.rs b/pipelined/bevy_render2/src/render_graph/graph.rs
@@ -261,6 +261,18 @@ impl RenderGraph {
         self.nodes.values_mut()
     }
 
+    pub fn iter_sub_graphs(&self) -> impl Iterator<Item = (&str, &RenderGraph)> {
+        self.sub_graphs
+            .iter()
+            .map(|(name, graph)| (name.as_ref(), graph))
+    }
+
+    pub fn iter_sub_graphs_mut(&mut self) -> impl Iterator<Item = (&str, &mut RenderGraph)> {
+        self.sub_graphs
+            .iter_mut()
+            .map(|(name, graph)| (name.as_ref(), graph))
+    }
+
     pub fn iter_node_inputs(
         &self,
         label: impl Into<NodeLabel>,