From 46863ad070cb36c0d743255da2b675b9d4171e5a Mon Sep 17 00:00:00 2001
From: Luiz Fernando Silva <luizinho_mack@yahoo.com.br>
Date: Wed, 24 Jul 2024 19:34:27 -0300
Subject: [PATCH] Fixing storage reference bug in KDTree

---
 .../SpatialPartitioning/KDTree/KDTree.swift               | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/Sources/GeometriaAlgorithms/SpatialPartitioning/KDTree/KDTree.swift b/Sources/GeometriaAlgorithms/SpatialPartitioning/KDTree/KDTree.swift
index 50f7d47b..bc7e6f17 100644
--- a/Sources/GeometriaAlgorithms/SpatialPartitioning/KDTree/KDTree.swift
+++ b/Sources/GeometriaAlgorithms/SpatialPartitioning/KDTree/KDTree.swift
@@ -110,6 +110,8 @@ public struct KDTree<Element: KDTreeLocatable> where Element.Vector: VectorCompa
 
     /// Inserts a given element on this k-d tree.
     public mutating func insert(_ element: Element) {
+        ensureUnique()
+
         guard let root else {
             root = Subdivision(
                 state: .empty(
@@ -128,12 +130,13 @@ public struct KDTree<Element: KDTreeLocatable> where Element.Vector: VectorCompa
         let path = subdivisionPath(forInserting: element)
         let reversed = path.reversed
 
-        ensureUnique()
         root.inserting(element, path: reversed.asPath)
     }
 
     /// If `element` is contained within this k-d tree, it is removed in-place.
     public mutating func remove(_ element: Element) where Element: Equatable {
+        ensureUnique()
+
         guard let root else {
             return
         }
@@ -156,11 +159,12 @@ public struct KDTree<Element: KDTreeLocatable> where Element.Vector: VectorCompa
 
     /// Removes an element at a given index in this k-d tree.
     public mutating func remove(at index: Index) {
+        ensureUnique()
+
         guard let root, root.pathExists(index.path) else {
             fatalError("Index \(index) is not part of this k-d tree.")
         }
 
-        ensureUnique()
         self.root = root.removing(at: index.path)
     }