mirror of
https://github.com/penpot/penpot.git
synced 2025-01-10 17:00:36 -05:00
119 lines
2.5 KiB
Clojure
119 lines
2.5 KiB
Clojure
(ns bench.core
|
|
(:require [kdtree.core :as k]
|
|
[intervaltree.core :as it]
|
|
[cljs.pprint :refer (pprint)]
|
|
[cljs.nodejs :as node]))
|
|
|
|
(enable-console-print!)
|
|
|
|
;; --- Helpers
|
|
|
|
(defn generate-points
|
|
([n] (generate-points n 1))
|
|
([n step]
|
|
(into-array
|
|
(for [x (range 0 n step)
|
|
y (range 0 n step)]
|
|
#js [x y]))))
|
|
|
|
;; --- Index Initialization Bechmark
|
|
|
|
(defn- bench-init-10000
|
|
[]
|
|
(println "1000x1000,10 -> 10000 points")
|
|
(let [tree (k/create)]
|
|
(time
|
|
(k/setup tree 1000 1000 10 10))))
|
|
|
|
(defn- bench-init-250000
|
|
[]
|
|
(println "5000x5000,10 -> 250000 points")
|
|
(let [tree (k/create)]
|
|
(time
|
|
(k/setup tree 5000 5000 10 10))))
|
|
|
|
(defn bench-init
|
|
[]
|
|
(bench-init-10000)
|
|
(bench-init-250000))
|
|
|
|
;; --- Nearest Search Benchmark
|
|
|
|
(defn- bench-knn-160000
|
|
[]
|
|
(let [tree (k/create)]
|
|
(k/setup tree 4000 4000 10 10)
|
|
(println "KNN Search (160000 points) 1000 times")
|
|
(time
|
|
(dotimes [i 1000]
|
|
(let [pt #js [(rand-int 400)
|
|
(rand-int 400)]]
|
|
(k/nearest tree pt 2))))))
|
|
|
|
|
|
(defn- bench-knn-360000
|
|
[]
|
|
(let [tree (k/create)]
|
|
(k/initialize tree 6000 6000 10 10)
|
|
(println "KNN Search (360000 points) 1000 times")
|
|
(time
|
|
(dotimes [i 1000]
|
|
(let [pt #js [(rand-int 600)
|
|
(rand-int 600)]]
|
|
(k/nearest tree pt 2))))))
|
|
|
|
(defn bench-knn
|
|
[]
|
|
(bench-knn-160000)
|
|
(bench-knn-360000))
|
|
|
|
;; --- Accuracity tests
|
|
|
|
(defn test-accuracity
|
|
[]
|
|
(let [tree (k/create)]
|
|
(k/setup tree 4000 4000 20 20)
|
|
(print "[1742 1419]")
|
|
(pprint (js->clj (k/nearest tree #js [1742 1419] 6)))
|
|
(print "[1742 1420]")
|
|
(pprint (js->clj (k/nearest tree #js [1742 1420] 6)))
|
|
))
|
|
|
|
(defn test-interval
|
|
[]
|
|
(let [tree (doto (it/create)
|
|
(.add #js [1 5])
|
|
(.add #js [5 7])
|
|
(.add #js [-4 -1])
|
|
(.add #js [-10 -3])
|
|
(.add #js [-20 -10])
|
|
(.add #js [20 30])
|
|
(.add #js [3 9])
|
|
(.add #js [100 200])
|
|
(.add #js [1000 2000])
|
|
(.add #js [6 9])
|
|
)]
|
|
(js/console.dir tree #js {"depth" nil})
|
|
(js/console.log "contains", 4, (.contains tree 4))
|
|
(js/console.log "contains", 0, (.contains tree 0))
|
|
))
|
|
|
|
(defn main
|
|
[& [type]]
|
|
(cond
|
|
(= type "init")
|
|
(bench-init)
|
|
|
|
(= type "knn")
|
|
(bench-knn)
|
|
|
|
(= type "test")
|
|
(test-accuracity)
|
|
|
|
(= type "interval")
|
|
(test-interval)
|
|
|
|
:else
|
|
(println "not implemented")))
|
|
|
|
(set! *main-cli-fn* main)
|