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

Force all components of an entity that had Disabled removed to be marked as changed

Browse Source
This commit is contained in:
Lucas Farias 2025-04-29 21:24:25 -03:00
parent 9fca353782
commit a65277dc9c
1 changed files with 27 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
crates/bevy_ecs/src/entity_disabling.rs
View File

@ -67,10 +67,15 @@
//! [`World`]: crate::prelude::World //! [`World`]: crate::prelude::World
//! [`Query` performance]: crate::prelude::Query#performance //! [`Query` performance]: crate::prelude::Query#performance
use alloc::vec::Vec;
use log::warn;
use crate::{ use crate::{
component::{ComponentId, Components, StorageType}, component::{ComponentId, Components, HookContext, StorageType},
entity::Entity,
prelude::DetectChangesMut,
query::FilteredAccess, query::FilteredAccess,
world::{FromWorld, World}, world::{DeferredWorld, FromWorld, World},
}; };
use bevy_ecs_macros::{Component, Resource}; use bevy_ecs_macros::{Component, Resource};
use smallvec::SmallVec; use smallvec::SmallVec;
@ -102,9 +107,29 @@ use {
reflect(Component), reflect(Component),
reflect(Debug, Clone, Default) reflect(Debug, Clone, Default)
)] )]
#[component(on_remove = Disabled::on_remove)]
// This component is registered as a disabling component during World::bootstrap // This component is registered as a disabling component during World::bootstrap
pub struct Disabled; pub struct Disabled;
impl Disabled {
fn on_remove<'w>(mut world: DeferredWorld<'w>, hook_context: HookContext) {
Self::set_all_components_as_changed(hook_context.entity, &mut world);
}
fn set_all_components_as_changed(entity: Entity, world: &mut DeferredWorld) {
let mut entity = world.entity_mut(entity);
let archetype = entity.archetype();
let archetype_id = archetype.id();
for comp_id in archetype.components().collect::<Vec<_>>() {
if let Ok(mut comp) = entity.get_mut_by_id(comp_id) {
comp.set_changed();
} else {
warn!("Entity {} was part of archetype {:?}. It reported that component {:?} is part of the archetype but entity did not have it. Maybe this is a bug.", entity.id(), archetype_id, comp_id);
}
}
}
}
/// Default query filters work by excluding entities with certain components from most queries. /// Default query filters work by excluding entities with certain components from most queries.
/// ///
/// If a query does not explicitly mention a given disabling component, it will not include entities with that component. /// If a query does not explicitly mention a given disabling component, it will not include entities with that component.