Initial commit

Author: marijnz

.gitignore

@@ -0,0 +1,73 @@
+
+# Created by https://www.gitignore.io/api/unity
+# Edit at https://www.gitignore.io/?templates=unity
+
+### Unity ###
+# This .gitignore file should be placed at the root of your Unity project directory
+#
+# Get latest from https://github.com/github/gitignore/blob/master/Unity.gitignore
+/[Ll]ibrary/
+/[Tt]emp/
+/[Oo]bj/
+/[Bb]uild/
+/[Bb]uilds/
+/[Ll]ogs/
+/[Mm]emoryCaptures/
+
+# Never ignore Asset meta data
+!/[Aa]ssets/**/*.meta
+
+# Uncomment this line if you wish to ignore the asset store tools plugin
+# /[Aa]ssets/AssetStoreTools*
+
+# TextMesh Pro files
+[Aa]ssets/TextMesh*Pro/
+
+# Autogenerated Jetbrains Rider plugin
+[Aa]ssets/Plugins/Editor/JetBrains*
+
+# Visual Studio cache directory
+.vs/
+
+# Gradle cache directory
+.gradle/
+
+# Rider
+.idea/
+
+# Autogenerated VS/MD/Consulo solution and project files
+ExportedObj/
+.consulo/
+*.csproj
+*.unityproj
+*.sln
+*.suo
+*.tmp
+*.user
+*.userprefs
+*.pidb
+*.booproj
+*.svd
+*.pdb
+*.mdb
+*.opendb
+*.VC.db
+
+# Unity3D generated meta files
+*.pidb.meta
+*.pdb.meta
+*.mdb.meta
+
+# Unity3D generated file on crash reports
+sysinfo.txt
+
+# Builds
+*.apk
+*.unitypackage
+*.symbols.zip
+
+# Crashlytics generated file
+crashlytics-build.properties
+
+
+# End of https://www.gitignore.io/api/unity

Assets/AABB2D.cs

@@ -0,0 +1,46 @@
+using System;
+using Unity.Mathematics;
+
+namespace QuadTree
+{
+	[Serializable]
+	public struct AABB2D {
+		public float2 Center;
+		public float2 Extents;
+
+		public float2 Size => Extents * 2;
+		public float2 Min => Center - Extents;
+		public float2 Max => Center + Extents;
+
+		public AABB2D(float2 center, float2 extents)
+		{
+			Center = center;
+			Extents = extents;
+		}
+
+		public bool Contains(float2 point) {
+			if (point[0] < Center[0] - Extents[0]) {
+				return false;
+			}
+			if (point[0] > Center[0] + Extents[0]) {
+				return false;
+			}
+
+			if (point[1] < Center[1] - Extents[1]) {
+				return false;
+			}
+			if (point[1] > Center[1] + Extents[1]) {
+				return false;
+			}
+
+			return true;
+		}
+
+		public bool Contains(AABB2D b) {
+			return Contains(b.Center + new float2(-b.Extents.x, -b.Extents.y)) &&
+			       Contains(b.Center + new float2(-b.Extents.x, b.Extents.y)) &&
+			       Contains(b.Center + new float2(b.Extents.x, -b.Extents.y)) &&
+			       Contains(b.Center + new float2(b.Extents.x, b.Extents.y));
+		}
+	}
+}

Assets/AABB2D.cs.meta

@@ -0,0 +1,49 @@
+using QuadTree;
+using UnityEditor;
+using UnityEngine;
+
+public class QuadTreeDrawer : EditorWindow
+{
+	[MenuItem("Window/QuadTreeDrawer")]
+	static void Init()
+	{
+		GetWindow(typeof(QuadTreeDrawer)).Show();
+	}
+
+	public static void Draw<T>(NativeQuadTree<T> quadTree) where T : unmanaged
+	{
+		QuadTreeDrawer window = (QuadTreeDrawer)GetWindow(typeof(QuadTreeDrawer));
+		window.DoDraw(quadTree);
+	}
+
+	[SerializeField]
+	Color[][] pixels;
+
+	void DoDraw<T>(NativeQuadTree<T> quadTree) where T : unmanaged
+	{
+		pixels = new Color[512][];
+		for (var i = 0; i < pixels.Length; i++)
+		{
+			pixels[i] = new Color[512];
+		}
+		NativeQuadTree<T>.Draw(quadTree, pixels);
+	}
+
+	void OnGUI()
+	{
+		if(pixels != null)
+		{
+			var texture = new Texture2D(512, 512);
+			for (var x = 0; x < pixels.Length; x++)
+			{
+				for (int y = 0; y < pixels[x].Length; y++)
+				{
+					texture.SetPixel(x, y, pixels[x][y]);
+				}
+			}
+			texture.Apply();
+
+			GUI.DrawTexture(new Rect(0, 0, position.width, position.height), texture);
+		}
+	}
+}

Assets/Editor.meta

@@ -0,0 +1,56 @@
+using System.Diagnostics;
+using NUnit.Framework;
+using QuadTree;
+using Unity.Burst;
+using Unity.Jobs;
+using Debug = UnityEngine.Debug;
+
+public class QuadTreeTests
+{
+    [Test]
+    public void Insert()
+    {
+        var quadTree = new NativeQuadTree<int>(new AABB2D(0, 50));
+        quadTree.InsertElement(new QuadElement<int>
+        {
+            pos = 5,
+            element = 53
+        });
+    }
+
+    [Test]
+    public void InsertTriggerDivide()
+    {
+        var s = Stopwatch.StartNew();
+        var job = new AddJob()
+        {
+            QuadTree = new NativeQuadTree<int>(new AABB2D(0, 50))
+        };
+        job.Run();
+        s.Stop();
+        Debug.Log(s.Elapsed.TotalMilliseconds);
+
+        QuadTreeDrawer.Draw(job.QuadTree);
+        job.QuadTree.Dispose();
+    }
+
+    [BurstCompile]
+    struct AddJob : IJob
+    {
+        public NativeQuadTree<int> QuadTree;
+
+        public void Execute()
+        {
+            var rnd = new Unity.Mathematics.Random(1);
+
+            for (int i = 0; i < 2000; i++)
+            {
+                QuadTree.InsertElement(new QuadElement<int>
+                {
+                    pos = rnd.NextFloat2(-40, 40),
+                    element = 53
+                });
+            }
+        }
+    }
+}