archive-gh-me/flatbush
1
0
Fork 0
mirror of https://github.com/mourner/flatbush.git synced 2025-12-08 17:36:26 +00:00
Code Issues Projects Releases Wiki Activity

optimize knn search by using non-negative queue indices

Browse Source
This commit is contained in:
Vladimir Agafonkin 2021-02-17 23:25:55 +02:00
parent a71e9a5da7
commit 6903d4b2c4
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
index.js
View File

@ -228,22 +228,22 @@ export default class Flatbush {
if (nodeIndex < this.numItems * 4) { // leaf node
if (filterFn === undefined || filterFn(index)) {
// put a negative index if it's an item rather than a node, to recognize later
q.push(-index - 1, dist);
// put an odd index if it's an item rather than a node, to recognize later
q.push((index << 1) + 1, dist);
}
} else {
q.push(index, dist);
q.push(index << 1, dist);
}
}
// pop items from the queue
while (q.length && q.peek() < 0) {
while (q.length && (q.peek() & 1)) {
const dist = q.peekValue();
if (dist > maxDistSquared) {
q.clear();
return results;
}
results.push(-q.pop() - 1);
results.push(q.pop() >> 1);
if (results.length === maxResults) {
q.clear();
@ -251,7 +251,7 @@ export default class Flatbush {
}
}
nodeIndex = q.pop();
nodeIndex = q.pop() >> 1;
}
q.clear();