From f67ccdd0e10cfde6851610d5d96008c8a0a50ab6 Mon Sep 17 00:00:00 2001
From: HampusM <hampus@hampusmat.com>
Date: Sat, 5 Apr 2025 16:32:57 +0200
Subject: refactor(ecs): make only component storage graph fns set node edges

---
 ecs/src/component/storage/graph.rs | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

(limited to 'ecs/src/component/storage/graph.rs')

diff --git a/ecs/src/component/storage/graph.rs b/ecs/src/component/storage/graph.rs
index 11160e7..d38223a 100644
--- a/ecs/src/component/storage/graph.rs
+++ b/ecs/src/component/storage/graph.rs
@@ -140,7 +140,7 @@ impl Graph
     }
 
     fn create_missing_subset_node_edges(
-        target_node: &ArchetypeNode,
+        target_node: &mut ArchetypeNode,
         subset_node: &mut ArchetypeNode,
     )
     {
@@ -153,6 +153,14 @@ impl Graph
         subset_node
             .get_or_insert_edges(uniq_comp_id, ArchetypeEdges::default)
             .add = Some(subset_node.make_add_edge(uniq_comp_id).0);
+
+        if target_node.archetype().component_cnt()
+            == subset_node.archetype().component_cnt() + 1
+        {
+            target_node
+                .get_or_insert_edges(uniq_comp_id, ArchetypeEdges::default)
+                .remove = Some(subset_node.archetype().id());
+        }
     }
 
     fn create_missing_superset_node_edges(
@@ -245,13 +253,6 @@ impl ArchetypeNode
         self.edges.iter()
     }
 
-    pub fn get_edges_mut(&mut self, component_id: Uid) -> Option<&mut ArchetypeEdges>
-    {
-        debug_assert_eq!(component_id.kind(), UidKind::Component);
-
-        self.edges.get_mut(&component_id)
-    }
-
     pub fn make_add_edge(&self, component_id: Uid) -> (ArchetypeId, Vec<Uid>)
     {
         let mut edge_comp_ids = self
-- 
cgit v1.2.3-18-g5258