bevy/crates/bevy_ecs/src/schedule/system_set.rs

use crate::{
    component::Component,
    schedule::{RunCriteria, ShouldRun, State, SystemDescriptor},
    system::System,
};

/// Describes a group of systems sharing one run criterion.
pub struct SystemSet {
    pub(crate) run_criteria: RunCriteria,
    pub(crate) descriptors: Vec<SystemDescriptor>,
}

impl Default for SystemSet {
    fn default() -> SystemSet {
        SystemSet {
            run_criteria: Default::default(),
            descriptors: vec![],
        }
    }
}

impl SystemSet {
    pub fn new() -> Self {
        Default::default()
    }

    pub fn with_run_criteria<S: System<In = (), Out = ShouldRun>>(mut self, system: S) -> Self {
        self.add_run_criteria(system);
        self
    }

    pub fn add_run_criteria<S: System<In = (), Out = ShouldRun>>(
        &mut self,
        system: S,
    ) -> &mut Self {
        self.run_criteria.set(Box::new(system));
        self
    }

    pub fn with_system(mut self, system: impl Into<SystemDescriptor>) -> Self {
        self.add_system(system);
        self
    }

    pub fn add_system(&mut self, system: impl Into<SystemDescriptor>) -> &mut Self {
        self.descriptors.push(system.into());
        self
    }

    pub fn on_update<T: Component + Clone + Eq>(s: T) -> SystemSet {
        Self::new().with_run_criteria(State::<T>::on_update(s))
    }

    pub fn on_inactive_update<T: Component + Clone + Eq>(s: T) -> SystemSet {
        Self::new().with_run_criteria(State::<T>::on_inactive_update(s))
    }

    pub fn on_enter<T: Component + Clone + Eq>(s: T) -> SystemSet {
        Self::new().with_run_criteria(State::<T>::on_enter(s))
    }

    pub fn on_exit<T: Component + Clone + Eq>(s: T) -> SystemSet {
        Self::new().with_run_criteria(State::<T>::on_exit(s))
    }

    pub fn on_pause<T: Component + Clone + Eq>(s: T) -> SystemSet {
        Self::new().with_run_criteria(State::<T>::on_pause(s))
    }

    pub fn on_resume<T: Component + Clone + Eq>(s: T) -> SystemSet {
        Self::new().with_run_criteria(State::<T>::on_resume(s))
    }
}