c8f7dd52c6
- Adjust some demo window sizes. - Replace Random.double by Double.uniform - Tweak some demos so screenshots look more interesting
74 lines
2.0 KiB
Kotlin
74 lines
2.0 KiB
Kotlin
import org.openrndr.application
|
|
import org.openrndr.color.ColorRGBa
|
|
import org.openrndr.draw.rectangleBatch
|
|
import org.openrndr.extra.noise.Random
|
|
import org.openrndr.extra.noise.gaussian
|
|
import org.openrndr.extra.quadtree.Quadtree
|
|
import org.openrndr.math.Vector2
|
|
import org.openrndr.shape.Rectangle
|
|
|
|
fun main() = application {
|
|
configure {
|
|
width = 800
|
|
height = 800
|
|
title = "QuadTree"
|
|
}
|
|
program {
|
|
val box = Rectangle.fromCenter(Vector2(400.0), 750.0)
|
|
|
|
val points = (0 until 100).map {
|
|
Vector2.gaussian(box.center, Vector2(95.0), Random.rnd)
|
|
}
|
|
|
|
val quadTree = Quadtree<Vector2>(box) { it }
|
|
|
|
for (point in points) {
|
|
quadTree.insert(point)
|
|
}
|
|
|
|
val selected = points[3]
|
|
val radius = 40.0
|
|
|
|
val nearestQuery = quadTree.nearest(selected, radius)
|
|
|
|
val batch = drawer.rectangleBatch {
|
|
this.fill = null
|
|
this.stroke = ColorRGBa.GRAY
|
|
this.strokeWeight = 0.5
|
|
quadTree.batch(this)
|
|
}
|
|
|
|
extend {
|
|
drawer.clear(ColorRGBa.BLACK)
|
|
|
|
drawer.rectangles(batch)
|
|
|
|
drawer.fill = ColorRGBa.PINK.opacify(0.7)
|
|
drawer.stroke = null
|
|
drawer.circles(points, 5.0)
|
|
|
|
nearestQuery?.let { (nearest, neighbours, nodes) ->
|
|
drawer.stroke = null
|
|
drawer.fill = ColorRGBa.YELLOW.opacify(0.2)
|
|
|
|
for (node in nodes) {
|
|
node.draw(drawer)
|
|
}
|
|
|
|
drawer.fill = ColorRGBa.GREEN.opacify(0.7)
|
|
drawer.circles(neighbours, 5.0)
|
|
|
|
drawer.fill = ColorRGBa.RED.opacify(0.9)
|
|
drawer.circle(nearest, 5.0)
|
|
|
|
drawer.fill = ColorRGBa.PINK
|
|
drawer.circle(selected, 5.0)
|
|
|
|
drawer.stroke = ColorRGBa.PINK
|
|
drawer.fill = null
|
|
drawer.circle(selected, radius)
|
|
}
|
|
}
|
|
}
|
|
}
|