Add dynamic reference methods

bevyengine · Sep 12, 2023 · 25623dc · 25623dc
1 parent f6382c1
commit 25623dc
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/crates/bevy_ecs/src/term_query/builder.rs b/crates/bevy_ecs/src/term_query/builder.rs
@@ -54,6 +54,17 @@ impl<'w, Q: QueryTermGroup> QueryBuilder<'w, Q> {
         self
     }
 
+    pub fn ref_id(&mut self, id: ComponentId) -> &mut Self {
+        self.terms.push(Term::Component(ComponentTerm::read_id(id)));
+        self
+    }
+
+    pub fn mut_id(&mut self, id: ComponentId) -> &mut Self {
+        self.terms
+            .push(Term::Component(ComponentTerm::write_id(id)));
+        self
+    }
+
     pub fn set<T: Component>(&mut self) -> &mut Self {
         let id = self.world.init_component::<T>();
         self.set_id(id);