[Merged by Bors] - expose stages and system containers

crates/bevy_ecs/src/schedule/mod.rs

@@ -201,6 +201,13 @@ impl Schedule {
             stage.run(world);
         }
     }
+
+    /// Iterates over all of schedule's stages and their labels, in execution order.
+    pub fn iter_stages(&self) -> impl Iterator<Item = (&dyn StageLabel, &dyn Stage)> {
+        self.stage_order
+            .iter()
+            .map(move |label| (&**label, &*self.stages[label]))
+    }
 }
 
 impl Stage for Schedule {

crates/bevy_ecs/src/schedule/stage.rs

@@ -167,6 +167,26 @@ impl SystemStage {
         self.add_system_to_set(system, 0)
     }
 
+    /// Topologically sorted parallel systems.
+    ///
+    /// Note that this method won't output fully-formed data until the stage has been ran at least once.
+    pub fn parallel_systems(&self) -> &[impl SystemContainer] {
+        &self.parallel
+    }
+    /// Topologically sorted exclusive systems that want to be ran at the start of the stage.
+    pub fn exclusive_at_start_systems(&self) -> &[impl SystemContainer] {
+        &self.exclusive_at_start
+    }
+    /// Topologically sorted exclusive systems that want to be ran at the end of the stage.
+    pub fn exclusive_at_end_systems(&self) -> &[impl SystemContainer] {
+        &self.exclusive_at_end
+    }
+    /// Topologically sorted exclusive systems that want to be ran after parallel systems but
+    /// before the application of their command buffers.
+    pub fn exclusive_before_commands_systems(&self) -> &[impl SystemContainer] {
+        &self.exclusive_before_commands
+    }
+
     // TODO: consider exposing
     fn add_system_to_set(&mut self, system: impl Into<SystemDescriptor>, set: usize) -> &mut Self {
         self.systems_modified = true;

crates/bevy_ecs/src/schedule/system_container.rs

@@ -9,9 +9,12 @@ use crate::{
 };
 use std::{borrow::Cow, ptr::NonNull};
 
-pub(super) trait SystemContainer {
+/// System metadata like its name, labels, order requirements and component access.
+pub trait SystemContainer {
     fn name(&self) -> Cow<'static, str>;
+    #[doc(hidden)]
     fn dependencies(&self) -> &[usize];
+    #[doc(hidden)]
     fn set_dependencies(&mut self, dependencies: impl IntoIterator<Item = usize>);
     fn system_set(&self) -> usize;
     fn labels(&self) -> &[BoxedSystemLabel];