only update components for entities in map

bevyengine · Dec 7, 2020 · e8855fa · e8855fa
1 parent 1398d78
commit e8855fa
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/crates/bevy_reflect/src/impls/bevy_ecs.rs b/crates/bevy_reflect/src/impls/bevy_ecs.rs
@@ -202,8 +202,10 @@ impl<C: Component + MapEntities> FromType<C> for ReflectMapEntities {
     fn from_type() -> Self {
         ReflectMapEntities {
             map_entities: |world, entity_map| {
-                for mut component in &mut world.query_mut::<&mut C>() {
-                    component.map_entities(entity_map)?;
+                for entity in entity_map.values() {
+                    if let Ok(mut component) = world.get_mut::<C>(entity) {
+                        component.map_entities(entity_map)?;
+                    }
                 }
 
                 Ok(())