From 551c0231a0014958ed1ae21ed877ee47590595da Mon Sep 17 00:00:00 2001
From: Izumi Kawashima <schedul.xor@gmail.com>
Date: Sat, 18 May 2019 03:21:51 +0900
Subject: [PATCH] R-tree k-nearest neighbors search improvements (#727)

---
 vtm/src/org/oscim/utils/RTree.java | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/vtm/src/org/oscim/utils/RTree.java b/vtm/src/org/oscim/utils/RTree.java
index fd8cc306..41848781 100644
--- a/vtm/src/org/oscim/utils/RTree.java
+++ b/vtm/src/org/oscim/utils/RTree.java
@@ -419,9 +419,8 @@ public class RTree<T> implements SpatialIndex<T>, Iterable<T> {
 
             while (!queue.isEmpty() && queue.peek().isLeaf) {
                 KnnItem knnItem = queue.poll();
-                T obj = (T) (knnItem.branch);
-                results.add(obj);
-                if (results.size() >= k)
+                results.add((T) knnItem.branch.node);
+                if (results.size() == k)
                     return results;
             }