fix: fix edges which connect to combo were filtered in layout stage

antvis · Jan 7, 2025 · 294072d · 294072d
1 parent 48c5c19
commit 294072d
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/packages/g6/src/runtime/layout.ts b/packages/g6/src/runtime/layout.ts
@@ -320,12 +320,11 @@ export class LayoutController {
 
     const nodesToLayout = nodes.filter(filterFn);
 
-    const nodesIdMap = new Map<ID, NodeData>(nodesToLayout.map((node) => [idOf(node), node]));
+    const nodeLikeIdsMap = new Map<ID, NodeData>(nodesToLayout.map((node) => [idOf(node), node]));
+    combos.forEach((combo) => nodeLikeIdsMap.set(idOf(combo), combo));
 
-    const edgesToLayout = edges.filter((edge) => {
-      const { source, target } = edge;
-      if (!nodesIdMap.has(source) || !nodesIdMap.has(target)) return false;
-      return true;
+    const edgesToLayout = edges.filter(({ source, target }) => {
+      return nodeLikeIdsMap.has(source) && nodeLikeIdsMap.has(target);
     });
 
     return {