ecs: add ability to set the current entity in world builders

lassade · Jun 25, 2020 · dfde160 · dfde160
1 parent 4a0f8b8
commit dfde160
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/crates/bevy_core/src/transform/world_builder.rs b/crates/bevy_core/src/transform/world_builder.rs
@@ -34,6 +34,11 @@ impl<'a> WorldBuilder<'a> {
         self
     }
 
+    pub fn set_entity(&mut self, entity: Entity) -> &mut Self {
+        self.current_entity = Some(entity);
+        self
+    }
+
     /// note: this is slow and does a full entity copy
     pub fn with<T>(&mut self, component: T) -> &mut Self
     where
@@ -114,6 +119,11 @@ impl<'a> CommandBufferBuilder<'a> {
         self
     }
 
+    pub fn set_entity(&mut self, entity: Entity) -> &mut Self {
+        self.current_entity = Some(entity);
+        self
+    }
+
     // note: this is slow and does a full entity copy
     pub fn with<T>(&mut self, component: T) -> &mut Self
     where