fix issue 2

deepfabric · Oct 17, 2017 · 5ce0888 · 5ce0888
1 parent 18c7585
commit 5ce0888
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/bkdtree_intersect.go b/bkdtree_intersect.go
@@ -65,10 +65,18 @@ func (bkd *BkdTree) intersectNode(visitor IntersectVisitor, data []byte,
 	if err != nil {
 		return
 	}
-	for _, child := range node.Children {
+	lowVal := lowP.Vals[node.SplitDim]
+	highVal := highP.Vals[node.SplitDim]
+	for i, child := range node.Children {
 		if child.NumPoints <= 0 {
 			continue
 		}
+		if i < int(node.NumStrips)-1 && node.SplitValues[i] < lowVal {
+			continue
+		}
+		if i != 0 && node.SplitValues[i-1] > highVal {
+			continue
+		}
 		if child.Offset < meta.PointsOffEnd {
 			//leaf node
 			pae := PointArrayExt{