+ plugins project

1 files changed, 76 insertions, 0 deletions

diff --git a/Plugins/MonoGame.Extended/source/MonoGame.Extended.Collisions/QuadTree/QuadTreeSpace.cs b/Plugins/MonoGame.Extended/source/MonoGame.Extended.Collisions/QuadTree/QuadTreeSpace.cs
new file mode 100644
index 0000000..3e9625f
--- /dev/null
+++ b/Plugins/MonoGame.Extended/source/MonoGame.Extended.Collisions/QuadTree/QuadTreeSpace.cs
@@ -0,0 +1,76 @@
+using System.Collections.Generic;
+using System.Linq;
+
+namespace MonoGame.Extended.Collisions.QuadTree;
+
+public class QuadTreeSpace: ISpaceAlgorithm
+{
+    private readonly QuadTree _collisionTree;
+    private readonly List<ICollisionActor> _actors = new();
+    private readonly Dictionary<ICollisionActor, QuadtreeData> _targetDataDictionary = new();
+
+    public QuadTreeSpace(RectangleF boundary)
+    {
+        _collisionTree = new QuadTree(boundary);
+    }
+
+    /// <summary>
+    /// Inserts the target into the collision tree.
+    /// The target will have its OnCollision called when collisions occur.
+    /// </summary>
+    /// <param name="target">Target to insert.</param>
+    public void Insert(ICollisionActor target)
+    {
+        if (!_targetDataDictionary.ContainsKey(target))
+        {
+            var data = new QuadtreeData(target);
+            _targetDataDictionary.Add(target, data);
+            _collisionTree.Insert(data);
+            _actors.Add(target);
+        }
+    }
+
+    /// <summary>
+    /// Removes the target from the collision tree.
+    /// </summary>
+    /// <param name="target">Target to remove.</param>
+    public bool Remove(ICollisionActor target)
+    {
+        if (_targetDataDictionary.ContainsKey(target))
+        {
+            var data = _targetDataDictionary[target];
+            data.RemoveFromAllParents();
+            _targetDataDictionary.Remove(target);
+            _collisionTree.Shake();
+            _actors.Remove(target);
+            return true;
+        }
+
+        return false;
+    }
+
+    /// <summary>
+    /// Restructure a inner collection, if layer is dynamic, because actors can change own position
+    /// </summary>
+    public void Reset()
+    {
+        _collisionTree.ClearAll();
+        foreach (var value in _targetDataDictionary.Values)
+        {
+            _collisionTree.Insert(value);
+        }
+        _collisionTree.Shake();
+    }
+
+    /// <summary>
+    /// foreach support
+    /// </summary>
+    /// <returns></returns>
+    public List<ICollisionActor>.Enumerator GetEnumerator() => _actors.GetEnumerator();
+
+    /// <inheritdoc cref="QuadTree.Query"/>
+    public IEnumerable<ICollisionActor> Query(RectangleF boundsBoundingRectangle)
+    {
+        return _collisionTree.Query(ref boundsBoundingRectangle).Select(x => x.Target);
+    }
+}