Getter instead of deref

bevyengine · Mar 23, 2023 · a68126a · a68126a
1 parent c81f1fa
commit a68126a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/crates/bevy_ecs/src/schedule/condition.rs b/crates/bevy_ecs/src/schedule/condition.rs
@@ -304,7 +304,7 @@ pub mod common_conditions {
     ///
     /// The condition will panic if the resource does not exist.
     pub fn in_state<S: States>(state: S) -> impl FnMut(Res<State<S>>) -> bool {
-        move |current_state: Res<State<S>>| current_state.0 == state
+        move |current_state: Res<State<S>>| *current_state.get() == state
     }
 
     /// Generates a [`Condition`](super::Condition)-satisfying closure that returns `true`
@@ -315,7 +315,7 @@ pub mod common_conditions {
         state: S,
     ) -> impl FnMut(Option<Res<State<S>>>) -> bool {
         move |current_state: Option<Res<State<S>>>| match current_state {
-            Some(current_state) => current_state.0 == state,
+            Some(current_state) => *current_state.get() == state,
             None => false,
         }
     }

diff --git a/crates/bevy_ecs/src/schedule/state.rs b/crates/bevy_ecs/src/schedule/state.rs
@@ -73,9 +73,9 @@ pub struct OnUpdate<S: States>(pub S);
 #[derive(Resource, Default, Debug)]
 pub struct State<S: States>(S);
 
-impl<S: States> std::ops::Deref for State<S> {
-    type Target = S;
-    fn deref(&self) -> &Self::Target {
+impl<S: States> State<S> {
+    /// Get the current state.
+    pub fn get(&self) -> &S {
         &self.0
     }
 }