BallTree WIP

tzaeschke · Aug 11, 2024 · 9a4bd15 · 9a4bd15
1 parent 4da0dff
commit 9a4bd15
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -6,6 +6,8 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 
 ## [Unreleased]
 
+TODO implement extent/iterator(), see mokeTest line 150 (Map) and 165 (MultiMap) 
+
 - Nothing yet
 
 ## [2.1.4] - 2024-08-01

diff --git a/src/test/java/org/tinspin/index/test/PointMultimapTest.java b/src/test/java/org/tinspin/index/test/PointMultimapTest.java
@@ -51,6 +51,7 @@ public static Iterable<Object[]> candidates() {
         ArrayList<Object[]> l = new ArrayList<>();
         // l.add(new Object[]{IDX.ARRAY});
         // l.add(new Object[]{IDX.COVER});
+        l.add(new Object[]{IDX.BALL});
         l.add(new Object[]{IDX.KDTREE});
         l.add(new Object[]{IDX.PHTREE_MM});
         l.add(new Object[]{IDX.QUAD_HC});
@@ -160,7 +161,7 @@ private void smokeTest(List<Entry> data) {
         }
 
         if (candidate != IDX.COVER && candidate != IDX.KDTREE && candidate != IDX.QUAD_PLAIN
-                && candidate != IDX.QUAD_HC && candidate != IDX.QUAD_HC2) {
+                && candidate != IDX.QUAD_HC && candidate != IDX.QUAD_HC2 && candidate != IDX.BALL) {
             int nExtent = 0;
             PointIterator<Entry> extent = tree.iterator();
             while (extent.hasNext()) {
@@ -382,6 +383,8 @@ private <T> PointMultimap<T> createTree(int size, int dims) {
         switch (candidate) {
             case ARRAY:
                 return PointMultimap.Factory.createArray(dims, size);
+            case BALL:
+                return PointMultimap.Factory.createKdTree(dims);
 //            //case CRITBIT: return new PointArray<>(dims, size);
             case KDTREE:
                 return PointMultimap.Factory.createKdTree(dims);