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

add get_mut and contains methods

Browse Source
This commit is contained in:
Danila 2025-06-08 18:03:54 +03:00
parent c17c742d67
commit d2914bfd72
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
crates/bevy_ecs/src/system/related_system.rs
View File

@ -63,13 +63,25 @@ impl<'w, 's, D: QueryData, F1: QueryFilter, R: RelationshipTarget, F2: QueryFilt
&'w self,
entity: Entity,
) -> Result<<<D as QueryData>::ReadOnly as QueryData>::Item<'w>, BevyError> {
if self
if self.contains(entity) {
Ok(self.data_query.get(entity)?)
} else {
Err(RelatedQueryError.into())
}
}
pub fn contains(&self, entity: Entity) -> bool {
return self
.filter_query
.iter()
.map(|r| r.get())
.any(|e| e == entity)
{
return Ok(self.data_query.get(entity)?);
&& self.data_query.contains(entity);
}
pub fn get_mut(&'w mut self, entity: Entity) -> Result<<D as QueryData>::Item<'w>, BevyError> {
if self.contains(entity) {
Ok(self.data_query.get_mut(entity)?)
} else {
Err(RelatedQueryError.into())
}