From 98143d6240b11c89ca45adb986a3a4d443be58ee Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?David=20Proch=C3=A1zka?= <david@prochazka.dev>
Date: Fri, 2 Apr 2021 09:32:09 +0200
Subject: [PATCH] FIX: changed methods to public
---
src/mhtree/LeafNode.java | 5 +++--
src/mhtree/MHTree.java | 16 ++++++++--------
2 files changed, 11 insertions(+), 10 deletions(-)
diff --git a/src/mhtree/LeafNode.java b/src/mhtree/LeafNode.java
index 66d41e1..1415df4 100644
--- a/src/mhtree/LeafNode.java
+++ b/src/mhtree/LeafNode.java
@@ -12,7 +12,8 @@ import java.util.List;
/**
* Represents a leaf node in MH-Tree.
*/
-class LeafNode extends Node implements Serializable {
+
+public class LeafNode extends Node implements Serializable {
/**
* Serialization ID
@@ -52,7 +53,7 @@ class LeafNode extends Node implements Serializable {
*
* @return a list of objects in node's bucket
*/
- protected List<LocalAbstractObject> getObjects() {
+ public List<LocalAbstractObject> getObjects() {
List<LocalAbstractObject> objects = new ArrayList<>(bucket.getObjectCount());
bucket
diff --git a/src/mhtree/MHTree.java b/src/mhtree/MHTree.java
index 9ec88c0..1ab345e 100644
--- a/src/mhtree/MHTree.java
+++ b/src/mhtree/MHTree.java
@@ -105,13 +105,9 @@ public class MHTree extends Algorithm implements Serializable {
operation.endOperation();
}
- private boolean isPrunable(Node child, LocalAbstractObject queryObject, ApproxKNNQueryOperation operation) {
- return operation.getAnswerDistance() < child.getDistanceToNearest(queryObject);
- }
-
- public void kNN(KNNQueryOperation knnQueryOperation) {
- root.getObjects().forEach(knnQueryOperation::addToAnswer);
- knnQueryOperation.endOperation();
+ public void kNN(KNNQueryOperation operation) {
+ root.getObjects().forEach(operation::addToAnswer);
+ operation.endOperation();
}
public int getObjectCount() {
@@ -151,7 +147,7 @@ public class MHTree extends Algorithm implements Serializable {
*
* @return a list of leaf nodes
*/
- private List<LeafNode> getLeafNodes() {
+ public List<LeafNode> getLeafNodes() {
List<LeafNode> leafNodes = new ArrayList<>();
root.gatherLeafNodes(leafNodes);
return leafNodes;
@@ -206,6 +202,10 @@ public class MHTree extends Algorithm implements Serializable {
'}';
}
+ private boolean isPrunable(Node child, LocalAbstractObject queryObject, ApproxKNNQueryOperation operation) {
+ return operation.getAnswerDistance() < child.getDistanceToNearest(queryObject);
+ }
+
public static class Builder {
/**
--
GitLab