forestia/bevy
2
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add with_enter_stage (and other variants) (#1091)

Browse Source
This commit is contained in:
Carter Anderson 2020-12-19 10:31:47 -08:00 committed by GitHub
parent 9db38b7b97
commit caf3d8b3ef
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
crates/bevy_ecs/src/schedule/state.rs
View File

@ -33,6 +33,21 @@ impl<T> Default for StateStage<T> {
#[allow(clippy::mem_discriminant_non_enum)] #[allow(clippy::mem_discriminant_non_enum)]
impl<T> StateStage<T> { impl<T> StateStage<T> {
pub fn with_enter_stage<S: Stage>(mut self, state: T, stage: S) -> Self {
self.set_enter_stage(state, stage);
self
}
pub fn with_exit_stage<S: Stage>(mut self, state: T, stage: S) -> Self {
self.set_exit_stage(state, stage);
self
}
pub fn with_update_stage<S: Stage>(mut self, state: T, stage: S) -> Self {
self.set_update_stage(state, stage);
self
}
pub fn set_enter_stage<S: Stage>(&mut self, state: T, stage: S) -> &mut Self { pub fn set_enter_stage<S: Stage>(&mut self, state: T, stage: S) -> &mut Self {
let stages = self.state_stages(state); let stages = self.state_stages(state);
stages.enter = Box::new(stage); stages.enter = Box::new(stage);