update parents list on remove (#511)

c9e45e50db2fdae4c838d091f1a5dc9d3be60216 · by Max Stoumen · 2023-07-17T14:32:06Z (3 years ago) · parent 0914058db · browse the tree at this commit

Semantic delta · 1

src/tree/taffy_tree/tree.rs

body changed fn remove pub

1 files · +6 −0

src/tree/taffy_tree/tree.rs +6 −0

diff --git a/src/tree/taffy_tree/tree.rs b/src/tree/taffy_tree/tree.rs
index a55e6f334..2949bd2a7 100644
--- a/src/tree/taffy_tree/tree.rs
+++ b/src/tree/taffy_tree/tree.rs
@@ -229,6 +229,12 @@ impl Taffy {
             }
         }
 
+        if let Some(children) = self.children.get(key) {
+            for child in children.iter().copied() {
+                self.parents[child.into()] = None;
+            }
+        }
+
         let _ = self.children.remove(key);
         let _ = self.parents.remove(key);
         let _ = self.nodes.remove(key);