jvmMain.ReadwriteQuadtree.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of orx-quadtree-jvm Show documentation

orx-quadtree

The newest version!

package org.openrndr.extra.quadtree

import org.openrndr.math.Vector2
import java.util.concurrent.locks.ReentrantReadWriteLock
import kotlin.concurrent.read
import kotlin.concurrent.write

/**
 * Wraps a quadtree with a ReentrantReadWriteLock, which allows multiple concurrent
 * readers or one writer at a time.
 */
class ReadwriteQuadtree(val qt: Quadtree) : IQuadtree {
    val lock = ReentrantReadWriteLock()

    override fun clear() {
        lock.write {
            qt.clear()
        }
    }

    override fun nearestToPoint(point: Vector2, radius: Double): QuadtreeQuery? {
        lock.read {
            return qt.nearestToPoint(point, radius)
        }
    }

    override fun nearest(element: T, radius: Double): QuadtreeQuery? {
        lock.read {
            return qt.nearest(element, radius)
        }
    }

    override fun insert(element: T): Boolean {
        lock.write {
            return qt.insert(element)
        }
    }

    override fun remove(element: T): Boolean {
        lock.write {
            return qt.remove(element)
        }
    }

    override fun findNode(element: T): Quadtree? {
        lock.read {
            return qt.findNode(element)
        }
    }
}