perf: node sort by area

gkd-kit · Oct 25, 2024 · c05a68f · c05a68f
1 parent f151765
commit c05a68f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/utils/node.ts b/src/utils/node.ts
@@ -114,9 +114,18 @@ export const findNodesByXy = (
         !equalRectNode(node, other),
     );
   });
+  results.sort((a, b) => {
+    return getNodeArea(a) - getNodeArea(b);
+  });
   return results;
 };
 
+const getNodeArea = (node: RawNode) => {
+  const w = node.attr.width ?? node.attr.right - node.attr.left;
+  const h = node.attr.height ?? node.attr.bottom - node.attr.top;
+  return w * h;
+};
+
 export function* traverseNode(node: RawNode, skipKeys: number[] = []) {
   const stack: RawNode[] = [];
   stack.push(node);

diff --git a/src/utils/types.ts b/src/utils/types.ts
@@ -54,6 +54,8 @@ export interface RawAttr {
   top: number;
   right: number;
   bottom: number;
+  width: number;
+  height: number;
   _id?: number;
   _pid?: number;
 }