Add DemoNearestNeighbour01.kt for orx-kdtree

2020-05-01 15:41:15 +02:00
parent 542ca90379
commit 2540873ba4
3 changed files with 53 additions and 0 deletions
--- a/orx-kdtree/src/demo/kotlin/DemoNearestNeighbour01.kt
+++ b/orx-kdtree/src/demo/kotlin/DemoNearestNeighbour01.kt
@@ -0,0 +1,34 @@
+import org.openrndr.application
+import org.openrndr.extensions.SingleScreenshot
+import org.openrndr.extra.kdtree.buildKDTree
+import org.openrndr.extra.kdtree.findNearest
+import org.openrndr.extra.kdtree.vector2Mapper
+import org.openrndr.math.Vector2
+
+fun main() {
+    application {
+        configure {
+            width = 1280
+            height = 720
+        }
+        program {
+            if (System.getProperty("takeScreenshot") == "true") {
+                extend(SingleScreenshot()) {
+                    this.outputFile = System.getProperty("screenshotPath")
+                }
+            }
+
+            val points = MutableList(1000) {
+                Vector2(Math.random() * width, Math.random() * height)
+            }
+            val tree = buildKDTree(points, 2, ::vector2Mapper)
+            extend {
+                drawer.circles(points, 5.0)
+                val nearest = findNearest(tree, mouse.position, 2, ::vector2Mapper)
+                nearest?.let {
+                    drawer.circle(it.x, it.y, 20.0)
+                }
+            }
+        }
+    }
+}