(use '[rtree.core])

; Default values, can also use {:node-split :linear}
(def empty-tree (rtree {:node-split :quadratic,
                        :min-children 2,
                        :max-children 4}))

(def tree
  (reduce insert empty-tree [(->Point 0.5 10.5)
                             (->Point 33.3 45.0)
                             (->Rectangle 0.0 0.0 10.0 10.0)
                             (->Rectangle 5.0 15.0 30.0 55.0)
                             (->Point 3.0 10.0)]))

(search tree (->Rectangle 0.0 0.0 50.0 50.0))
;=>
(#archery.shape.Point{:x 33.3, :y 45.0}
 #archery.shape.Rectangle{:x1 0.0, :y1 0.0, :x2 10.0, :y2 10.0}
 #archery.shape.Point{:x 0.5, :y 10.5}
 #archery.shape.Point{:x 3.0, :y 10.0})

Archery Benchmark:
Evaluation count : 120 in 60 samples of 2 calls.
             Execution time mean : 754.215969 ms
    Execution time std-deviation : 34.708262 ms
   Execution time lower quantile : 702.300994 ms ( 2.5%)
   Execution time upper quantile : 797.469882 ms (97.5%)
                   Overhead used : 6.236561 ns

Found 1 outliers in 60 samples (1.6667 %)
        low-severe       1 (1.6667 %)
 Variance from outliers : 31.9670 % Variance is moderately inflated by outliers
 
Java RTree Benchmark:
Evaluation count : 1020 in 60 samples of 17 calls.
             Execution time mean : 55.397792 ms
    Execution time std-deviation : 3.061980 ms
   Execution time lower quantile : 51.176994 ms ( 2.5%)
   Execution time upper quantile : 60.853864 ms (97.5%)
                   Overhead used : 6.236561 ns

Found 2 outliers in 60 samples (3.3333 %)
        low-severe       1 (1.6667 %)
        low-mild         1 (1.6667 %)
 Variance from outliers : 40.1868 % Variance is moderately inflated by outliers

(datafy tree)
;=>
{:type :RTree,
 :max-children 4,
 :min-children 2,
 :root {:type :RectangleNode,
        :leaf? false,
        :shape [0.0 0.0 33.3 55.0],
        :children [{:type :RectangleNode,
                    :leaf? true,
                    :shape [5.0 15.0 33.3 55.0],
                    :children [{:type :Point, :shape [33.3 45.0]}
                               {:type :Rectangle, :shape [5.0 15.0 30.0 55.0]}]}
                   {:type :RectangleNode,
                    :leaf? true,
                    :shape [0.0 0.0 10.0 10.5],
                    :children [{:type :Rectangle, :shape [0.0 0.0 10.0 10.0]}
                               {:type :Point, :shape [0.5 10.5]}
                               {:type :Point, :shape [3.0 10.0]}]}]}}
                               

 
(use '[archery.visualization])
(tree->vega-lite tree)
;=>
{:background "white",
 :width 1000,
 :height 1000,
 :layer [{:data {:values [{:x1 0.0, :y1 0.0, :x2 33.3, :y2 55.0}
                          {:x1 5.0, :y1 15.0, :x2 33.3, :y2 55.0}
                          {:x1 0.0, :y1 0.0, :x2 10.0, :y2 10.5}]},
          :mark "rect",
          :encoding {:x {:field "x1", :type "quantitative"},
                     :x2 {:field "x2"},
                     :y {:field "y1", :type "quantitative"},
                     :y2 {:field "y2"},
                     :opacity {:value 1},
                     :stroke {:value "green"}}}
         {:data {:values [{:x1 33.3, :y1 45.0, :x2 33.3, :y2 45.0}
                          {:x1 0.5, :y1 10.5, :x2 0.5, :y2 10.5}
                          {:x1 3.0, :y1 10.0, :x2 3.0, :y2 10.0}]},
          :mark "circle",
          :encoding {:color {:value "red"},
                     :x {:field "x1", :type "quantitative"},
                     :x2 {:field "x2"},
                     :y {:field "y1", :type "quantitative"},
                     :y2 {:field "y2"}}}
         {:data {:values [{:x1 5.0, :y1 15.0, :x2 30.0, :y2 55.0}
                          {:x1 0.0, :y1 0.0, :x2 10.0, :y2 10.0}]},
          :mark "rect",
          :encoding {:x {:field "x1", :type "quantitative"},
                     :x2 {:field "x2"},
                     :y {:field "y1", :type "quantitative"},
                     :y2 {:field "y2"},
                     :stroke {:value "blue"},
                     :opacity {:value 0.8}}}]}