summaryrefslogtreecommitdiff
path: root/ecs/src/relationship.rs
diff options
context:
space:
mode:
authorHampusM <hampus@hampusmat.com>2024-08-21 20:18:37 +0200
committerHampusM <hampus@hampusmat.com>2024-08-21 20:18:37 +0200
commit58f29444ac234c285ce0e3a729ade2b52a303f58 (patch)
tree03bb210d62bf98acc1ee30ce9df1344d8239f7c6 /ecs/src/relationship.rs
parentc3c96f2d7b63d1d7e55f5a9a85da85444089c2a2 (diff)
refactor(ecs): fix clippy lints
Diffstat (limited to 'ecs/src/relationship.rs')
-rw-r--r--ecs/src/relationship.rs30
1 files changed, 28 insertions, 2 deletions
diff --git a/ecs/src/relationship.rs b/ecs/src/relationship.rs
index 1a8c842..0fe776e 100644
--- a/ecs/src/relationship.rs
+++ b/ecs/src/relationship.rs
@@ -26,6 +26,7 @@ where
ComponentT: Component,
{
/// Creates a new `Relationship` with a single target.
+ #[must_use]
pub fn new(entity_uid: EntityUid) -> Self
{
Self {
@@ -35,6 +36,7 @@ where
}
/// Creates a new `Relationship` with multiple targets.
+ #[must_use]
pub fn new_multiple(entity_uids: impl IntoIterator<Item = EntityUid>) -> Self
{
Self {
@@ -132,6 +134,11 @@ where
ComponentT: Component,
{
/// Returns the component of the target at the specified index.
+ ///
+ /// # Panics
+ /// Will panic if the entity does not exist in the archetype it belongs to. This
+ /// should hopefully never happend.
+ #[must_use]
pub fn get(&self, index: usize) -> Option<ComponentRefMut<'_, ComponentT>>
{
let target = self.get_target(index)?;
@@ -162,22 +169,24 @@ where
}
/// Returns a reference to the target at the specified index.
+ #[must_use]
pub fn get_target(&self, index: usize) -> Option<&EntityUid>
{
match &self.relationship_comp.entity_uid {
SingleOrMultiple::Single(entity_uid) if index == 0 => Some(entity_uid),
SingleOrMultiple::Multiple(entity_uids) => entity_uids.get(index),
- _ => None,
+ SingleOrMultiple::Single(_) => None,
}
}
/// Returns a mutable reference to the target at the specified index.
+ #[must_use]
pub fn get_target_mut(&mut self, index: usize) -> Option<&mut EntityUid>
{
match &mut self.relationship_comp.entity_uid {
SingleOrMultiple::Single(entity_uid) if index == 0 => Some(entity_uid),
SingleOrMultiple::Multiple(entity_uids) => entity_uids.get_mut(index),
- _ => None,
+ SingleOrMultiple::Single(_) => None,
}
}
@@ -215,6 +224,7 @@ where
}
}
+ #[must_use]
pub fn target_count(&self) -> usize
{
match &self.relationship_comp.entity_uid {
@@ -224,12 +234,28 @@ where
}
/// Returns a iterator of the components of the targets of this relationship.
+ #[must_use]
pub fn iter(&self) -> TargetComponentIter<'_, 'rel_comp, Kind, ComponentT>
{
TargetComponentIter { relation: self, index: 0 }
}
}
+impl<'relationship, 'rel_comp, Kind, ComponentT> IntoIterator
+ for &'relationship Relation<'rel_comp, Kind, ComponentT>
+where
+ 'relationship: 'rel_comp,
+ ComponentT: Component,
+{
+ type IntoIter = TargetComponentIter<'relationship, 'rel_comp, Kind, ComponentT>;
+ type Item = ComponentRefMut<'rel_comp, ComponentT>;
+
+ fn into_iter(self) -> Self::IntoIter
+ {
+ self.iter()
+ }
+}
+
/// Iterator of the components of the targets of a relationship.
pub struct TargetComponentIter<'relationship, 'rel_comp, Kind, ComponentT>
where