flow: perf

packages/math/quadtree.js

@@ -5,7 +5,10 @@ import * as Rectangle from './rectangle';
 export class QuadTree {
 
   constructor() {
+    this._visitNodes = [];
+    this._queryRectangle = [0, 0, 0, 0];
     this.quadTree = quadtree();
+    this.onVisit = this.onVisit.bind(this);
   }
 
   add(datum) {
@@ -20,28 +23,31 @@ export class QuadTree {
     }
   }
 
-  remove(datum) {
-    this.quadTree.remove(datum);
-  }
-
-  search(queryRectangle) {
-    const nodes = [];
-    this.quadTree.visit((node, x, y, x0, y0) => {
+  onVisit(node, x, y, x0, y0) {
     const w = x0 - x;
     const h = y0 - y;
-      if (!Rectangle.intersects(queryRectangle, [x, y, w, h])) {
+    if (!Rectangle.intersects(this._queryRectangle, [x, y, w, h])) {
       return true;
     }
     if (4 === node.length) {
       return;
     }
     do {
-        if (Rectangle.intersects(queryRectangle, node.data)) {
-          nodes.push(node);
+      if (Rectangle.isTouching(this._queryRectangle, node.data)) {
+        this.nodes.push(node);
       }
     } while (node = node.next);
-    });
-    return nodes;
+  }
+
+  remove(datum) {
+    this.quadTree.remove(datum);
+  }
+
+  search(queryRectangle) {
+    this._queryRectangle = queryRectangle;
+    this.nodes = [];
+    this.quadTree.visit(this.onVisit);
+    return this.nodes;
   }
 
 }