From 91ad130f8beb765cf8b503dbee433a68acd54464 Mon Sep 17 00:00:00 2001
From: Scott Bell <scott@traclabs.com>
Date: Thu, 20 Apr 2023 15:18:42 +0200
Subject: [PATCH] switch plots to use flatbush

---
 .../imagery/components/AnnotationsCanvas.vue  |  1 -
 src/plugins/plot/MctPlot.vue                  | 44 +++++++++++++++----
 2 files changed, 35 insertions(+), 10 deletions(-)

diff --git a/src/plugins/imagery/components/AnnotationsCanvas.vue b/src/plugins/imagery/components/AnnotationsCanvas.vue
index dc68863840..0335115839 100644
--- a/src/plugins/imagery/components/AnnotationsCanvas.vue
+++ b/src/plugins/imagery/components/AnnotationsCanvas.vue
@@ -33,7 +33,6 @@
 
 <script>
 
-import e from 'cors';
 import Flatbush from 'flatbush';
 const EXISTING_ANNOTATION_STROKE_STYLE = "#D79078";
 const EXISTING_ANNOTATION_FILL_STYLE = "rgba(202, 202, 142, 0.2)";
diff --git a/src/plugins/plot/MctPlot.vue b/src/plugins/plot/MctPlot.vue
index 8227ba044e..9cd355b351 100644
--- a/src/plugins/plot/MctPlot.vue
+++ b/src/plugins/plot/MctPlot.vue
@@ -215,6 +215,7 @@ import MctChart from "./chart/MctChart.vue";
 import XAxis from "./axis/XAxis.vue";
 import YAxis from "./axis/YAxis.vue";
 import KDBush from 'kdbush';
+import Flatbush from 'flatbush';
 import _ from "lodash";
 
 const OFFSET_THRESHOLD = 10;
@@ -1348,6 +1349,32 @@ export default {
 
             return annotationsByPoints.flat();
         },
+        searchWithKDTree(seriesData, seriesModel, boundingBox) {
+            const kdTree = new KDBush(seriesData,
+                (point) => {
+                    return seriesModel.getXVal(point);
+                },
+                (point) => {
+                    return seriesModel.getYVal(point);
+                }
+            );
+            const rangeResults = kdTree.range(boundingBox.minX, boundingBox.minY, boundingBox.maxX, boundingBox.maxY);
+
+            return rangeResults;
+        },
+        searchWithFlatbush(seriesData, seriesModel, boundingBox) {
+            const flatbush = new Flatbush(seriesData.length);
+            seriesData.forEach(point => {
+                const x = seriesModel.getXVal(point);
+                const y = seriesModel.getYVal(point);
+                flatbush.add(x, y, x, y);
+            });
+            flatbush.finish();
+
+            const rangeResults = flatbush.search(boundingBox.minX, boundingBox.minY, boundingBox.maxX, boundingBox.maxY);
+
+            return rangeResults;
+        },
         getPointsInBox(boundingBoxPerYAxis, rawAnnotation) {
             // load series models in KD-Trees
             const seriesKDTrees = [];
@@ -1361,16 +1388,15 @@ export default {
 
                 const seriesData = seriesModel.getSeriesData();
                 if (seriesData && seriesData.length) {
-                    const kdTree = new KDBush(seriesData,
-                        (point) => {
-                            return seriesModel.getXVal(point);
-                        },
-                        (point) => {
-                            return seriesModel.getYVal(point);
-                        }
-                    );
                     const searchResults = [];
-                    const rangeResults = kdTree.range(boundingBox.minX, boundingBox.minY, boundingBox.maxX, boundingBox.maxY);
+                    let startTime = Date.now();
+                    let rangeResults = this.searchWithKDTree(seriesData, seriesModel, boundingBox);
+                    let endTime = Date.now();
+                    console.debug(`KD Tree Annotation search took ${endTime - startTime} ms for ${seriesData.length} points`);
+                    startTime = Date.now();
+                    rangeResults = this.searchWithFlatbush(seriesData, seriesModel, boundingBox);
+                    endTime = Date.now();
+                    console.debug(`Flatbush Tree Annotation search took ${endTime - startTime} ms for ${seriesData.length} points`);
                     rangeResults.forEach(id => {
                         const seriesDatum = seriesData[id];
                         if (seriesDatum) {