@antv/layout 1.2.14-beta.8 → 2.0.0-alpha.1

package/lib/dagre/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sources":["../../src/dagre/src/dagre/index.ts"],"sourcesContent":[null],"names":["graphlib"],"mappings":";;;;;;;;;;AAUA;;;;AAIG;AACG,MAAO,WAAY,SAAQ,UAA8B,CAAA;AAA/D,IAAA,WAAA,GAAA;;QACE,IAAA,CAAA,EAAE,GAAG,OAAO;QAEJ,IAAA,CAAA,eAAe,GAAmB,IAAI;AAEpC,QAAA,IAAA,CAAA,MAAM,GAAG;AACjB,YAAA,eAAe,EAAE;gBACf,SAAS;gBACT,OAAO;gBACP,SAAS;gBACT,SAAS;gBACT,SAAS;gBACT,SAAS;gBACT,SAAS;gBACT,WAAW;gBACX,QAAQ;AACT,aAAA;AACD,YAAA,cAAc,EAAE,CAAC,OAAO,EAAE,QAAQ,CAAC;AACnC,YAAA,cAAc,EAAE;gBACd,QAAQ;gBACR,QAAQ;gBACR,OAAO;gBACP,QAAQ;gBACR,UAAU;gBACV,aAAa;AACd,aAAA;SACF;IAkIH;IAhIY,iBAAiB,GAAA;QACzB,OAAO;AACL,YAAA,QAAQ,EAAE,IAAI;AACd,YAAA,UAAU,EAAE,IAAI;AAChB,YAAA,OAAO,EAAE,IAAI;AACb,YAAA,KAAK,EAAE,SAAS;AAChB,YAAA,OAAO,EAAE,EAAE;AACX,YAAA,OAAO,EAAE,EAAE;AACX,YAAA,OAAO,EAAE,EAAE;AACX,YAAA,OAAO,EAAE,CAAC;AACV,YAAA,OAAO,EAAE,CAAC;AACV,YAAA,SAAS,EAAE,SAAS;AACpB,YAAA,MAAM,EAAE,iBAAiB;AACzB,YAAA,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;AAChB,YAAA,UAAU,EAAE,CAAC;AACb,YAAA,UAAU,EAAE,CAAC;AACb,YAAA,aAAa,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;AACrB,YAAA,YAAY,EAAE,GAAG;AACjB,YAAA,eAAe,EAAE,EAAE;SACpB;IACH;IAEgB,MAAM,GAAA;;AACpB,YAAA,MAAM,CAAC,GAAG,IAAIA,qBAAQ,CAAC,KAAK,CAAC;AAC3B,gBAAA,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ;AACjC,gBAAA,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU;AACrC,gBAAA,QAAQ,EAAE,IAAI,CAAC,UAAU,EAAE;AAC5B,aAAA,CAAC;AAEF,YAAA,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;YAE3D,CAAC,CAAC,mBAAmB,CAAC,OAAO,EAAE,CAAC,CAAC;AAEjC,YAAA,MAAM,UAAU,GAAG,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;YAEzD,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAgB,KAAI;AAC1C,gBAAA,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS;AAC1B,gBAAA,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;AAElD,gBAAA,MAAM,KAAK,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE;AAC/B,gBAAA,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC;AAEjC,gBAAA,IAAI,IAAI,CAAC,UAAU,EAAE,EAAE;AACrB,oBAAA,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC;wBAAE;AAE1B,oBAAA,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AACpD,gBAAA;AACH,YAAA,CAAC,CAAC;AAEF,YAAA,MAAM,EACJ,aAAa,EACb,eAAe,EACf,YAAY,EACZ,UAAU,EACV,UAAU,GACX,GAAG,IAAI,CAAC,OAAO;YAEhB,MAAM,eAAe,GAAG,YAAY,CAAC,aAAa,EAAE,CAAC,CAAC;YACtD,MAAM,iBAAiB,GAAG,cAAc,CAAC,eAAe,EAAE,EAAE,CAAC;AAC7D,YAAA,MAAM,cAAc,GAClB,OAAO,YAAY,KAAK,UAAU,GAAG,YAAY,GAAG,MAAM,YAAY;YACxE,MAAM,YAAY,GAAG,cAAc,CAAC,UAAU,EAAE,CAAC,CAAC;YAClD,MAAM,YAAY,GAAG,cAAc,CAAC,UAAU,EAAE,CAAC,CAAC;YAElD,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,KAAI;AAC9B,gBAAA,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS;AAC1B,gBAAA,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,SAAS,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;AAEhD,gBAAA,MAAM,KAAK,GAAG;AACZ,oBAAA,KAAK,EAAE,EAAE;AACT,oBAAA,MAAM,EAAE,EAAE;AACV,oBAAA,QAAQ,EAAE,cAAc,CAAC,GAAG,CAAC;AAC7B,oBAAA,WAAW,EAAE,iBAAiB,CAAC,GAAG,CAAC;AACnC,oBAAA,MAAM,EAAE,YAAY,CAAC,GAAG,CAAC;AACzB,oBAAA,MAAM,EAAE,YAAY,CAAC,GAAG,CAAC;iBAC1B;gBAED,CAAC,CAAC,OAAO,CACP,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EACnB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EACnB,KAAK,EACL,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAChB;AACH,YAAA,CAAC,CAAC;AAEF,YAAA,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;YAEf,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,KAAI;AAC9B,gBAAA,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AACpC,gBAAA,IAAI,CAAC,IAAI;oBAAE;AAEX,gBAAA,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;AACf,gBAAA,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;AACf,gBAAA,IAAI,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC;AACvC,YAAA,CAAC,CAAC;YAEF,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,KAAI;gBAC9B,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,CACjB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EACnB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EACnB,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAChB;AAED,gBAAA,IAAI,CAAC,IAAI;oBAAE;AAEX,gBAAA,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE,GACpE,IAAI;gBACN,IAAI,CAAC,SAAS,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC;AAChC,gBAAA,IAAI,CAAC,MAAM,GAAG,MAAM;AACpB,gBAAA,IAAI,CAAC,MAAM,GAAG,MAAM;AACpB,gBAAA,IAAI,CAAC,QAAQ,GAAG,QAAQ;AACxB,gBAAA,IAAI,CAAC,WAAW,GAAG,WAAW;gBAC9B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC;AACtC,YAAA,CAAC,CAAC;QACJ,CAAC,CAAA;AAAA,IAAA;IAEO,UAAU,GAAA;AAChB,QAAA,IAAI,IAAI,CAAC,eAAe,KAAK,IAAI;YAAE,OAAO,IAAI,CAAC,eAAe;QAE9D,IAAI,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;YACpC,QAAQ,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ;AACrD,QAAA;AAED,QAAA,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;AACzB,aAAA,KAAK;AACL,aAAA,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxC,OAAO,IAAI,CAAC,eAAe;IAC7B;AACD;;;;"}

package/lib/dagre/types.d.ts

@@ -0,0 +1,79 @@
+import { GraphLabel } from 'dagre';
+import { BaseLayoutOptions } from '../core/types.js';
+import { NodeData } from '../types/data.js';
+import { Size } from '../types/size.js';
+
+/**
+ * <zh/> 边标签位置：'l' 左侧，'c' 中心，'r' 右侧
+ *
+ * <en/> Edge label position: 'l' for left, 'c' for center, 'r' for right
+ */
+type EdgeLabelPos = 'l' | 'c' | 'r';
+/**
+ * <zh/> Dagre 层次布局配置项，用于有向图的分层排列
+ *
+ * <en/> Dagre hierarchical layout options for layered arrangement of directed graphs
+ */
+interface DagreLayoutOptions extends BaseLayoutOptions, GraphLabel {
+    /**
+     * <zh/> 控制边的方向性，true 表示边有方向（箭头），影响布局的层次关系
+     *
+     * <en/> Controls edge directionality; true means edges have direction (arrows), affecting hierarchical relationships in layout
+     * @defaultValue true
+     */
+    directed?: boolean;
+    /**
+     * <zh/> 支持嵌套结构，允许节点包含子图，用于复杂的层级关系
+     *
+     * <en/> Supports nested structures where nodes can contain subgraphs for complex hierarchical relationships
+     * @defaultValue true
+     */
+    compound?: boolean;
+    /**
+     * <zh/> 允许两节点间存在多条边，用于表示多重关系
+     *
+     * <en/> Allows multiple edges between two nodes to represent multiple relationships
+     * @defaultValue true
+     */
+    multigraph?: boolean;
+    /**
+     * <zh/> 定义节点占用的空间大小，影响节点间距和整体布局疏密
+     *
+     * <en/> Defines space occupied by nodes, affecting inter-node spacing and overall layout density
+     * @defaultValue [0, 0]
+     */
+    nodeSize?: Size | ((d?: NodeData) => Size);
+    /**
+     * <zh/> 设置边跨越的最小层数，值越大节点间距越远，用于控制布局紧凑度
+     *
+     * <en/> Sets minimum number of layers an edge spans; larger values create more distance between nodes, controlling layout compactness
+     * @defaultValue 1
+     */
+    edgeMinLen?: number | ((d?: NodeData) => number);
+    /**
+     * <zh/> 边的权重，影响边的长度优化优先级，权重大的边倾向于更短
+     *
+     * <en/> Edge weight affecting length optimization priority; higher weight edges tend to be shorter
+     */
+    edgeWeight?: number | ((d?: NodeData) => number);
+    /**
+     * <zh/> 边标签的尺寸，用于为标签预留空间，避免与节点重叠
+     *
+     * <en/> Size of edge labels for reserving space to prevent overlap with nodes
+     */
+    edgeLabelSize?: Size | ((d?: NodeData) => Size);
+    /**
+     * <zh/> 标签在边上的位置，控制标签相对于边的对齐方式
+     *
+     * <en/> Label position on edge, controlling label alignment relative to the edge
+     */
+    edgeLabelPos?: EdgeLabelPos | ((d?: NodeData) => EdgeLabelPos);
+    /**
+     * <zh/> 标签与边的偏移距离，用于微调标签位置避免视觉重叠
+     *
+     * <en/> Offset distance between label and edge for fine-tuning label position to avoid visual overlap
+     */
+    edgeLabelOffset?: number | ((d?: NodeData) => number);
+}
+
+export type { DagreLayoutOptions, EdgeLabelPos };

package/lib/force/attractive.js

@@ -0,0 +1,60 @@
+/**
+ * Attractive force based on Hooke's law
+ * Applies spring-like forces between connected nodes
+ */
+function forceAttractive(dimensions = 2) {
+    let preventOverlap = false;
+    function force(model, accMap) {
+        model.forEachEdge((edge) => {
+            var _a, _b;
+            const { source, target } = edge;
+            const sourceNode = model.node(source);
+            const targetNode = model.node(target);
+            if (!sourceNode || !targetNode)
+                return;
+            let vecX = targetNode.x - sourceNode.x;
+            let vecY = targetNode.y - sourceNode.y;
+            let vecZ = dimensions === 3 ? ((_a = targetNode.z) !== null && _a !== void 0 ? _a : 0) - ((_b = sourceNode.z) !== null && _b !== void 0 ? _b : 0) : 0;
+            if (!vecX && !vecY) {
+                vecX = 0.01;
+                vecY = 0.01;
+                if (dimensions === 3 && !vecZ) {
+                    vecZ = 0.01;
+                }
+            }
+            const vecLength = Math.sqrt(vecX * vecX + vecY * vecY + vecZ * vecZ);
+            if (vecLength < Number(sourceNode.size) + Number(targetNode.size))
+                return;
+            const direX = vecX / vecLength;
+            const direY = vecY / vecLength;
+            const direZ = vecZ / vecLength;
+            const { linkDistance = 200, edgeStrength = 200 } = edge;
+            const diff = linkDistance - vecLength;
+            const param = diff * edgeStrength;
+            const massSource = sourceNode.mass || 1;
+            const massTarget = targetNode.mass || 1;
+            // 质量占比越大，对另一端影响程度越大
+            const sourceMassRatio = 1 / massSource;
+            const targetMassRatio = 1 / massTarget;
+            const disX = direX * param;
+            const disY = direY * param;
+            const disZ = direZ * param;
+            accMap[source].x -= disX * sourceMassRatio;
+            accMap[source].y -= disY * sourceMassRatio;
+            accMap[source].z -= disZ * sourceMassRatio;
+            accMap[target].x += disX * targetMassRatio;
+            accMap[target].y += disY * targetMassRatio;
+            accMap[target].z += disZ * targetMassRatio;
+        });
+    }
+    force.dimensions = function (_) {
+        return arguments.length ? ((dimensions = _), force) : dimensions;
+    };
+    force.preventOverlap = function (_) {
+        return arguments.length ? ((preventOverlap = _), force) : preventOverlap;
+    };
+    return force;
+}
+
+export { forceAttractive };
+//# sourceMappingURL=attractive.js.map

package/lib/force/attractive.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"attractive.js","sources":["../../src/force/src/force/attractive.ts"],"sourcesContent":[null],"names":[],"mappings":"AAGA;;;AAGG;AACG,SAAU,eAAe,CAAC,UAAA,GAAqB,CAAC,EAAA;IACpD,IAAI,cAAc,GAAY,KAAK;AAEnC,IAAA,SAAS,KAAK,CAAC,KAAkB,EAAE,MAAc,EAAA;AAC/C,QAAA,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,KAAI;;AACzB,YAAA,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI;YAC/B,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAE;YACtC,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAE;AACtC,YAAA,IAAI,CAAC,UAAU,IAAI,CAAC,UAAU;gBAAE;YAEhC,IAAI,IAAI,GAAG,UAAU,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC;YACtC,IAAI,IAAI,GAAG,UAAU,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC;AACtC,YAAA,IAAI,IAAI,GACN,UAAU,KAAK,CAAC,GAAG,CAAC,CAAA,EAAA,GAAA,UAAU,CAAC,CAAC,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,EAAA,GAAI,CAAC,KAAK,CAAA,EAAA,GAAA,UAAU,CAAC,CAAC,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,EAAA,GAAI,CAAC,CAAC,GAAG,CAAC;AAElE,YAAA,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,EAAE;gBAClB,IAAI,GAAG,IAAI;gBACX,IAAI,GAAG,IAAI;AACX,gBAAA,IAAI,UAAU,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE;oBAC7B,IAAI,GAAG,IAAI;AACZ,gBAAA;AACF,YAAA;AAED,YAAA,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;AACpE,YAAA,IAAI,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC;gBAAE;AAEnE,YAAA,MAAM,KAAK,GAAG,IAAI,GAAG,SAAS;AAC9B,YAAA,MAAM,KAAK,GAAG,IAAI,GAAG,SAAS;AAC9B,YAAA,MAAM,KAAK,GAAG,IAAI,GAAG,SAAS;YAE9B,MAAM,EAAE,YAAY,GAAG,GAAG,EAAE,YAAY,GAAG,GAAG,EAAE,GAAG,IAAW;AAC9D,YAAA,MAAM,IAAI,GAAG,YAAY,GAAG,SAAS;AACrC,YAAA,MAAM,KAAK,GAAG,IAAI,GAAG,YAAY;AAEjC,YAAA,MAAM,UAAU,GAAG,UAAU,CAAC,IAAI,IAAI,CAAC;AACvC,YAAA,MAAM,UAAU,GAAG,UAAU,CAAC,IAAI,IAAI,CAAC;;AAGvC,YAAA,MAAM,eAAe,GAAG,CAAC,GAAG,UAAU;AACtC,YAAA,MAAM,eAAe,GAAG,CAAC,GAAG,UAAU;AAEtC,YAAA,MAAM,IAAI,GAAG,KAAK,GAAG,KAAK;AAC1B,YAAA,MAAM,IAAI,GAAG,KAAK,GAAG,KAAK;AAC1B,YAAA,MAAM,IAAI,GAAG,KAAK,GAAG,KAAK;YAE1B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,GAAG,eAAe;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,GAAG,eAAe;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,GAAG,eAAe;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,GAAG,eAAe;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,GAAG,eAAe;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,GAAG,eAAe;AAC5C,QAAA,CAAC,CAAC;IACJ;AAEA,IAAA,KAAK,CAAC,UAAU,GAAG,UAAU,CAAU,EAAA;AACrC,QAAA,OAAO,SAAS,CAAC,MAAM,IAAI,CAAC,UAAU,GAAG,CAAE,GAAG,KAAK,IAAI,UAAU;AACnE,IAAA,CAAC;AAED,IAAA,KAAK,CAAC,cAAc,GAAG,UAAU,CAAW,EAAA;AAC1C,QAAA,OAAO,SAAS,CAAC,MAAM,IAAI,CAAC,cAAc,GAAG,CAAE,GAAG,KAAK,IAAI,cAAc;AAC3E,IAAA,CAAC;AAED,IAAA,OAAO,KAAK;AACd;;;;"}

package/lib/force/centripetal.js

@@ -0,0 +1,87 @@
+import isNumber from '../node_modules/.pnpm/@antv_util@3.3.11/node_modules/@antv/util/esm/lodash/is-number.js';
+
+/**
+ * Centripetal force with different strengths for leaf, single, and other nodes
+ * Unique feature of Force layout
+ */
+function forceCentripetal(options) {
+    let centripetalOptions;
+    let width = 800;
+    let height = 600;
+    function force(model, accMap) {
+        if (!centripetalOptions)
+            return;
+        const { leaf, single, others, center: centriCenter } = centripetalOptions;
+        const nodes = model.nodes();
+        const edges = model.edges();
+        // Convert LayoutNode[] to Node[] for callback compatibility
+        const nodesForCallback = nodes.map((n) => (Object.assign(Object.assign({}, n), { data: n.data || {} })));
+        const edgesForCallback = edges.map((e) => (Object.assign(Object.assign({}, e), { data: e.data || {} })));
+        model.forEachNode((node) => {
+            const { id, mass, x, y, z = 0, data } = node;
+            const inDegree = model.degree(id, 'in');
+            const outDegree = model.degree(id, 'out');
+            const degree = model.degree(id, 'both');
+            // Create Node type from LayoutNode for callback compatibility
+            const nodeForCallback = Object.assign(Object.assign({}, node), { data: data || {} });
+            const { x: centriX, y: centriY, z: centriZ, centerStrength, } = (centriCenter === null || centriCenter === void 0 ? void 0 : centriCenter(nodeForCallback, nodesForCallback, edgesForCallback, width, height)) || {
+                x: 0,
+                y: 0,
+                z: 0,
+                centerStrength: 0,
+            };
+            if (!isNumber(centriX) || !isNumber(centriY))
+                return;
+            const vx = (x - centriX) / mass;
+            const vy = (y - centriY) / mass;
+            const vz = (z - (centriZ || 0)) / mass;
+            if (centerStrength) {
+                accMap[id].x -= centerStrength * vx;
+                accMap[id].y -= centerStrength * vy;
+                accMap[id].z -= centerStrength * vz;
+            }
+            // 孤点
+            if (degree === 0) {
+                const singleStrength = single(nodeForCallback);
+                if (!singleStrength)
+                    return;
+                accMap[id].x -= singleStrength * vx;
+                accMap[id].y -= singleStrength * vy;
+                accMap[id].z -= singleStrength * vz;
+                return;
+            }
+            // 没有出度或没有入度，都认为是叶子节点
+            if (inDegree === 0 || outDegree === 0) {
+                const leafStrength = leaf(nodeForCallback, nodesForCallback, edgesForCallback);
+                if (!leafStrength)
+                    return;
+                accMap[id].x -= leafStrength * vx;
+                accMap[id].y -= leafStrength * vy;
+                accMap[id].z -= leafStrength * vz;
+                return;
+            }
+            // others
+            const othersStrength = others(nodeForCallback);
+            if (!othersStrength)
+                return;
+            accMap[id].x -= othersStrength * vx;
+            accMap[id].y -= othersStrength * vy;
+            accMap[id].z -= othersStrength * vz;
+        });
+    }
+    force.options = function (_) {
+        return arguments.length
+            ? ((centripetalOptions = _), force)
+            : centripetalOptions;
+    };
+    force.width = function (_) {
+        return arguments.length ? ((width = _), force) : width;
+    };
+    force.height = function (_) {
+        return arguments.length ? ((height = _), force) : height;
+    };
+    return force;
+}
+
+export { forceCentripetal };
+//# sourceMappingURL=centripetal.js.map

package/lib/force/centripetal.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"centripetal.js","sources":["../../src/force/src/force/centripetal.ts"],"sourcesContent":[null],"names":[],"mappings":";;AAuBA;;;AAGG;AACG,SAAU,gBAAgB,CAAC,OAAyC,EAAA;AACxE,IAAA,IAAI,kBAA2C;IAC/C,IAAI,KAAK,GAAW,GAAG;IACvB,IAAI,MAAM,GAAW,GAAG;AAExB,IAAA,SAAS,KAAK,CAAC,KAAkB,EAAE,MAAc,EAAA;AAC/C,QAAA,IAAI,CAAC,kBAAkB;YAAE;AAEzB,QAAA,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,GAAG,kBAAkB;AAEzE,QAAA,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,EAAE;AAC3B,QAAA,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,EAAE;;QAG3B,MAAM,gBAAgB,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,sCAAW,CAAC,CAAA,EAAA,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,EAAE,EAAA,CAAA,CAAG,CAAC;QACzE,MAAM,gBAAgB,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,sCAAW,CAAC,CAAA,EAAA,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,EAAE,EAAA,CAAA,CAAG,CAAC;AAEzE,QAAA,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,KAAI;AACzB,YAAA,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,GAAG,IAAI;YAE5C,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC;YACvC,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC;YACzC,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC;;YAGvC,MAAM,eAAe,GAAA,MAAA,CAAA,MAAA,CAAA,MAAA,CAAA,MAAA,CAAA,EAAA,EAAQ,IAAI,CAAA,EAAA,EAAE,IAAI,EAAE,IAAI,IAAI,EAAE,EAAA,CAAE;AAErD,YAAA,MAAM,EACJ,CAAC,EAAE,OAAO,EACV,CAAC,EAAE,OAAO,EACV,CAAC,EAAE,OAAO,EACV,cAAc,GACf,GAAG,CAAA,YAAY,KAAA,IAAA,IAAZ,YAAY,KAAA,MAAA,GAAA,MAAA,GAAZ,YAAY,CACd,eAAe,EACf,gBAAgB,EAChB,gBAAgB,EAChB,KAAK,EACL,MAAM,CACP,KAAI;AACH,gBAAA,CAAC,EAAE,CAAC;AACJ,gBAAA,CAAC,EAAE,CAAC;AACJ,gBAAA,CAAC,EAAE,CAAC;AACJ,gBAAA,cAAc,EAAE,CAAC;aAClB;YAED,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;gBAAE;YAE9C,MAAM,EAAE,GAAG,CAAC,CAAC,GAAG,OAAO,IAAI,IAAI;YAC/B,MAAM,EAAE,GAAG,CAAC,CAAC,GAAG,OAAO,IAAI,IAAI;AAC/B,YAAA,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,OAAO,IAAI,CAAC,CAAC,IAAI,IAAI;AAEtC,YAAA,IAAI,cAAc,EAAE;gBAClB,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,cAAc,GAAG,EAAE;gBACnC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,cAAc,GAAG,EAAE;gBACnC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,cAAc,GAAG,EAAE;AACpC,YAAA;;YAGD,IAAI,MAAM,KAAK,CAAC,EAAE;AAChB,gBAAA,MAAM,cAAc,GAAG,MAAM,CAAC,eAAe,CAAC;AAC9C,gBAAA,IAAI,CAAC,cAAc;oBAAE;gBACrB,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,cAAc,GAAG,EAAE;gBACnC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,cAAc,GAAG,EAAE;gBACnC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,cAAc,GAAG,EAAE;gBACnC;AACD,YAAA;;AAGD,YAAA,IAAI,QAAQ,KAAK,CAAC,IAAI,SAAS,KAAK,CAAC,EAAE;gBACrC,MAAM,YAAY,GAAG,IAAI,CACvB,eAAe,EACf,gBAAgB,EAChB,gBAAgB,CACjB;AACD,gBAAA,IAAI,CAAC,YAAY;oBAAE;gBACnB,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,YAAY,GAAG,EAAE;gBACjC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,YAAY,GAAG,EAAE;gBACjC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,YAAY,GAAG,EAAE;gBACjC;AACD,YAAA;;AAGD,YAAA,MAAM,cAAc,GAAG,MAAM,CAAC,eAAe,CAAC;AAC9C,YAAA,IAAI,CAAC,cAAc;gBAAE;YACrB,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,cAAc,GAAG,EAAE;YACnC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,cAAc,GAAG,EAAE;YACnC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,cAAc,GAAG,EAAE;AACrC,QAAA,CAAC,CAAC;IACJ;AAEA,IAAA,KAAK,CAAC,OAAO,GAAG,UAAU,CAAoC,EAAA;QAC5D,OAAO,SAAS,CAAC;eACZ,CAAC,kBAAkB,GAAG,CAA4B,GAAG,KAAK;cAC3D,kBAAkB;AACxB,IAAA,CAAC;AAED,IAAA,KAAK,CAAC,KAAK,GAAG,UAAU,CAAU,EAAA;AAChC,QAAA,OAAO,SAAS,CAAC,MAAM,IAAI,CAAC,KAAK,GAAG,CAAE,GAAG,KAAK,IAAI,KAAK;AACzD,IAAA,CAAC;AAED,IAAA,KAAK,CAAC,MAAM,GAAG,UAAU,CAAU,EAAA;AACjC,QAAA,OAAO,SAAS,CAAC,MAAM,IAAI,CAAC,MAAM,GAAG,CAAE,GAAG,KAAK,IAAI,MAAM;AAC3D,IAAA,CAAC;AAED,IAAA,OAAO,KAAK;AACd;;;;"}

package/lib/force/collide.js

@@ -0,0 +1,172 @@
+import quadtree from '../node_modules/.pnpm/d3-quadtree@3.0.1/node_modules/d3-quadtree/src/quadtree.js';
+import isNumber from '../node_modules/.pnpm/@antv_util@3.3.11/node_modules/@antv/util/esm/lodash/is-number.js';
+
+const EPSILON = 1e-6;
+const COLLIDE_SCALE = 500;
+/**
+ * Collision force to prevent node overlap.
+ */
+function forceCollide(dimensions = 2) {
+    let strength = 1;
+    function force(model, accMap) {
+        const nodes = model.nodes();
+        if (nodes.length < 2 || strength <= 0)
+            return;
+        if (dimensions === 2) {
+            const data = nodes.map((node, index) => {
+                return {
+                    id: node.id,
+                    index,
+                    x: node.x,
+                    y: node.y,
+                    r: Number(node.size) / 2,
+                    mass: node.mass || 1,
+                    fx: node.fx,
+                    fy: node.fy,
+                };
+            });
+            const tree = quadtree(data, (d) => d.x, (d) => d.y).visitAfter(accumulate2D);
+            for (const node of data) {
+                tree.visit((quad, x0, y0, x1, y1) => {
+                    const quadR = quad.r || 0;
+                    const searchR = node.r + quadR;
+                    if (x0 > node.x + searchR ||
+                        x1 < node.x - searchR ||
+                        y0 > node.y + searchR ||
+                        y1 < node.y - searchR) {
+                        return true;
+                    }
+                    if (!quad.data)
+                        return false;
+                    let q = quad;
+                    do {
+                        const other = q.data;
+                        if (other && other.index > node.index) {
+                            collide2D(node, other, accMap);
+                        }
+                        q = q.next;
+                    } while (q);
+                    return false;
+                });
+            }
+            return;
+        }
+        const data3d = nodes.map((node, index) => {
+            var _a;
+            return {
+                id: node.id,
+                index,
+                x: node.x,
+                y: node.y,
+                z: (_a = node.z) !== null && _a !== void 0 ? _a : 0,
+                r: Number(node.size) / 2,
+                mass: node.mass || 1,
+                fx: node.fx,
+                fy: node.fy,
+                fz: node.fz,
+            };
+        });
+        for (let i = 0; i < data3d.length; i++) {
+            for (let j = i + 1; j < data3d.length; j++) {
+                collide3D(data3d[i], data3d[j], accMap);
+            }
+        }
+    }
+    function invMass2D(d) {
+        const fixed = isNumber(d.fx) && isNumber(d.fy);
+        return fixed ? 0 : 1 / (d.mass || 1);
+    }
+    function invMass3D(d) {
+        const fixed = isNumber(d.fx) && isNumber(d.fy) && isNumber(d.fz);
+        return fixed ? 0 : 1 / (d.mass || 1);
+    }
+    function collide2D(a, b, accMap) {
+        let dx = a.x - b.x;
+        let dy = a.y - b.y;
+        let dist = Math.hypot(dx, dy);
+        const minDist = a.r + b.r;
+        if (dist >= minDist)
+            return;
+        if (dist < EPSILON) {
+            dx = a.index < b.index ? EPSILON : -EPSILON;
+            dy = 0;
+            dist = Math.abs(dx);
+        }
+        const overlap = minDist - dist;
+        const nx = dx / dist;
+        const ny = dy / dist;
+        const f = overlap * strength * COLLIDE_SCALE;
+        const ia = invMass2D(a);
+        const ib = invMass2D(b);
+        if (ia) {
+            accMap[a.id].x += nx * f * ia;
+            accMap[a.id].y += ny * f * ia;
+        }
+        if (ib) {
+            accMap[b.id].x -= nx * f * ib;
+            accMap[b.id].y -= ny * f * ib;
+        }
+    }
+    function collide3D(a, b, accMap) {
+        let dx = a.x - b.x;
+        let dy = a.y - b.y;
+        let dz = a.z - b.z;
+        let dist = Math.sqrt(dx * dx + dy * dy + dz * dz);
+        const minDist = a.r + b.r;
+        if (dist >= minDist)
+            return;
+        if (dist < EPSILON) {
+            dx = a.index < b.index ? EPSILON : -EPSILON;
+            dy = 0;
+            dz = 0;
+            dist = Math.abs(dx);
+        }
+        const overlap = minDist - dist;
+        const nx = dx / dist;
+        const ny = dy / dist;
+        const nz = dz / dist;
+        const f = overlap * strength * COLLIDE_SCALE;
+        const ia = invMass3D(a);
+        const ib = invMass3D(b);
+        if (ia) {
+            accMap[a.id].x += nx * f * ia;
+            accMap[a.id].y += ny * f * ia;
+            accMap[a.id].z += nz * f * ia;
+        }
+        if (ib) {
+            accMap[b.id].x -= nx * f * ib;
+            accMap[b.id].y -= ny * f * ib;
+            accMap[b.id].z -= nz * f * ib;
+        }
+    }
+    force.dimensions = function (_) {
+        return arguments.length ? ((dimensions = _), force) : dimensions;
+    };
+    force.strength = function (_) {
+        return arguments.length ? ((strength = _), force) : strength;
+    };
+    return force;
+}
+function accumulate2D(quad) {
+    var _a;
+    let r = 0;
+    if (quad.length) {
+        for (let i = 0; i < quad.length; i++) {
+            const c = quad[i];
+            if (c && c.r > r)
+                r = c.r;
+        }
+    }
+    else if (quad.data) {
+        r = quad.data.r || 0;
+        let next = quad.next;
+        while (next) {
+            r = Math.max(r, ((_a = next.data) === null || _a === void 0 ? void 0 : _a.r) || 0);
+            next = next.next;
+        }
+    }
+    quad.r = r;
+}
+
+export { forceCollide };
+//# sourceMappingURL=collide.js.map

package/lib/force/collide.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"collide.js","sources":["../../src/force/src/force/collide.ts"],"sourcesContent":[null],"names":[],"mappings":";;;AAsBA,MAAM,OAAO,GAAG,IAAI;AACpB,MAAM,aAAa,GAAG,GAAG;AAEzB;;AAEG;AACG,SAAU,YAAY,CAAC,UAAA,GAAqB,CAAC,EAAA;IACjD,IAAI,QAAQ,GAAW,CAAC;AAExB,IAAA,SAAS,KAAK,CAAC,KAAkB,EAAE,MAAc,EAAA;AAC/C,QAAA,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,EAAE;QAC3B,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,QAAQ,IAAI,CAAC;YAAE;QAEvC,IAAI,UAAU,KAAK,CAAC,EAAE;YACpB,MAAM,IAAI,GAAqB,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,KAAI;gBACvD,OAAO;oBACL,EAAE,EAAE,IAAI,CAAC,EAAE;oBACX,KAAK;oBACL,CAAC,EAAE,IAAI,CAAC,CAAC;oBACT,CAAC,EAAE,IAAI,CAAC,CAAC;oBACT,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;AACxB,oBAAA,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC;oBACpB,EAAE,EAAE,IAAI,CAAC,EAAE;oBACX,EAAE,EAAE,IAAI,CAAC,EAAE;iBACZ;AACH,YAAA,CAAC,CAAC;AAEF,YAAA,MAAM,IAAI,GAAG,QAAQ,CACnB,IAAI,EACJ,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EACV,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CACX,CAAC,UAAU,CAAC,YAAY,CAAC;AAE1B,YAAA,KAAK,MAAM,IAAI,IAAI,IAAI,EAAE;AACvB,gBAAA,IAAI,CAAC,KAAK,CACR,CAAC,IAAS,EAAE,EAAU,EAAE,EAAU,EAAE,EAAU,EAAE,EAAU,KAAI;AAC5D,oBAAA,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC;AACzB,oBAAA,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,GAAG,KAAK;AAC9B,oBAAA,IACE,EAAE,GAAG,IAAI,CAAC,CAAC,GAAG,OAAO;AACrB,wBAAA,EAAE,GAAG,IAAI,CAAC,CAAC,GAAG,OAAO;AACrB,wBAAA,EAAE,GAAG,IAAI,CAAC,CAAC,GAAG,OAAO;AACrB,wBAAA,EAAE,GAAG,IAAI,CAAC,CAAC,GAAG,OAAO,EACrB;AACA,wBAAA,OAAO,IAAI;AACZ,oBAAA;oBAED,IAAI,CAAC,IAAI,CAAC,IAAI;AAAE,wBAAA,OAAO,KAAK;oBAE5B,IAAI,CAAC,GAAQ,IAAI;oBACjB,GAAG;AACD,wBAAA,MAAM,KAAK,GAAmB,CAAC,CAAC,IAAI;wBACpC,IAAI,KAAK,IAAI,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,EAAE;AACrC,4BAAA,SAAS,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC;AAC/B,wBAAA;AACD,wBAAA,CAAC,GAAG,CAAC,CAAC,IAAI;AACX,oBAAA,CAAA,QAAQ,CAAC;AAEV,oBAAA,OAAO,KAAK;AACd,gBAAA,CAAC,CACF;AACF,YAAA;YAED;AACD,QAAA;QAED,MAAM,MAAM,GAAqB,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,KAAI;;YACzD,OAAO;gBACL,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,KAAK;gBACL,CAAC,EAAE,IAAI,CAAC,CAAC;gBACT,CAAC,EAAE,IAAI,CAAC,CAAC;AACT,gBAAA,CAAC,EAAE,CAAA,EAAA,GAAA,IAAI,CAAC,CAAC,mCAAI,CAAC;gBACd,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;AACxB,gBAAA,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC;gBACpB,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,EAAE,EAAE,IAAI,CAAC,EAAE;aACZ;AACH,QAAA,CAAC,CAAC;AAEF,QAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;AACtC,YAAA,KAAK,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;AAC1C,gBAAA,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC;AACxC,YAAA;AACF,QAAA;IACH;IAEA,SAAS,SAAS,CAAC,CAAiB,EAAA;AAClC,QAAA,MAAM,KAAK,GAAG,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;AAC9C,QAAA,OAAO,KAAK,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC;IACtC;IAEA,SAAS,SAAS,CAAC,CAAiB,EAAA;QAClC,MAAM,KAAK,GAAG,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;AAChE,QAAA,OAAO,KAAK,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC;IACtC;AAEA,IAAA,SAAS,SAAS,CAAC,CAAiB,EAAE,CAAiB,EAAE,MAAc,EAAA;QACrE,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAClB,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAClB,IAAI,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC;QAE7B,MAAM,OAAO,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACzB,IAAI,IAAI,IAAI,OAAO;YAAE;QAErB,IAAI,IAAI,GAAG,OAAO,EAAE;AAClB,YAAA,EAAE,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,GAAG,OAAO,GAAG,CAAC,OAAO;YAC3C,EAAE,GAAG,CAAC;AACN,YAAA,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;AACpB,QAAA;AAED,QAAA,MAAM,OAAO,GAAG,OAAO,GAAG,IAAI;AAC9B,QAAA,MAAM,EAAE,GAAG,EAAE,GAAG,IAAI;AACpB,QAAA,MAAM,EAAE,GAAG,EAAE,GAAG,IAAI;AACpB,QAAA,MAAM,CAAC,GAAG,OAAO,GAAG,QAAQ,GAAG,aAAa;AAE5C,QAAA,MAAM,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC;AACvB,QAAA,MAAM,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC;AAEvB,QAAA,IAAI,EAAE,EAAE;AACN,YAAA,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,GAAG,EAAE;AAC7B,YAAA,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,GAAG,EAAE;AAC9B,QAAA;AACD,QAAA,IAAI,EAAE,EAAE;AACN,YAAA,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,GAAG,EAAE;AAC7B,YAAA,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,GAAG,EAAE;AAC9B,QAAA;IACH;AAEA,IAAA,SAAS,SAAS,CAAC,CAAiB,EAAE,CAAiB,EAAE,MAAc,EAAA;QACrE,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAClB,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAClB,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;AAClB,QAAA,IAAI,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC;QAEjD,MAAM,OAAO,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACzB,IAAI,IAAI,IAAI,OAAO;YAAE;QAErB,IAAI,IAAI,GAAG,OAAO,EAAE;AAClB,YAAA,EAAE,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,GAAG,OAAO,GAAG,CAAC,OAAO;YAC3C,EAAE,GAAG,CAAC;YACN,EAAE,GAAG,CAAC;AACN,YAAA,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;AACpB,QAAA;AAED,QAAA,MAAM,OAAO,GAAG,OAAO,GAAG,IAAI;AAC9B,QAAA,MAAM,EAAE,GAAG,EAAE,GAAG,IAAI;AACpB,QAAA,MAAM,EAAE,GAAG,EAAE,GAAG,IAAI;AACpB,QAAA,MAAM,EAAE,GAAG,EAAE,GAAG,IAAI;AACpB,QAAA,MAAM,CAAC,GAAG,OAAO,GAAG,QAAQ,GAAG,aAAa;AAE5C,QAAA,MAAM,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC;AACvB,QAAA,MAAM,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC;AAEvB,QAAA,IAAI,EAAE,EAAE;AACN,YAAA,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,GAAG,EAAE;AAC7B,YAAA,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,GAAG,EAAE;AAC7B,YAAA,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,GAAG,EAAE;AAC9B,QAAA;AACD,QAAA,IAAI,EAAE,EAAE;AACN,YAAA,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,GAAG,EAAE;AAC7B,YAAA,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,GAAG,EAAE;AAC7B,YAAA,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,GAAG,EAAE;AAC9B,QAAA;IACH;AAEA,IAAA,KAAK,CAAC,UAAU,GAAG,UAAU,CAAU,EAAA;AACrC,QAAA,OAAO,SAAS,CAAC,MAAM,IAAI,CAAC,UAAU,GAAG,CAAE,GAAG,KAAK,IAAI,UAAU;AACnE,IAAA,CAAC;AAED,IAAA,KAAK,CAAC,QAAQ,GAAG,UAAU,CAAU,EAAA;AACnC,QAAA,OAAO,SAAS,CAAC,MAAM,IAAI,CAAC,QAAQ,GAAG,CAAE,GAAG,KAAK,IAAI,QAAQ;AAC/D,IAAA,CAAC;AAED,IAAA,OAAO,KAAK;AACd;AAEA,SAAS,YAAY,CAAC,IAAS,EAAA;;IAC7B,IAAI,CAAC,GAAG,CAAC;IAET,IAAI,IAAI,CAAC,MAAM,EAAE;AACf,QAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;AACpC,YAAA,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;AACjB,YAAA,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC;AAAE,gBAAA,CAAC,GAAG,CAAC,CAAC,CAAC;AAC1B,QAAA;AACF,IAAA;SAAM,IAAI,IAAI,CAAC,IAAI,EAAE;QACpB,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC;AACpB,QAAA,IAAI,IAAI,GAAG,IAAI,CAAC,IAAI;AACpB,QAAA,OAAO,IAAI,EAAE;AACX,YAAA,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAA,CAAA,EAAA,GAAA,IAAI,CAAC,IAAI,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAE,CAAC,KAAI,CAAC,CAAC;AAClC,YAAA,IAAI,GAAG,IAAI,CAAC,IAAI;AACjB,QAAA;AACF,IAAA;AAED,IAAA,IAAI,CAAC,CAAC,GAAG,CAAC;AACZ;;;;"}

package/lib/force/gravity.js

@@ -0,0 +1,50 @@
+/**
+ * Gravity force toward center
+ * Pulls all nodes toward a specified center point
+ */
+function forceGravity(center = [0, 0, 0], gravity = 10) {
+    let getCenter;
+    function force(model, accMap) {
+        const nodes = model.nodes();
+        if (!nodes)
+            return;
+        model.forEachNode((node) => {
+            const { id, mass, x, y, z = 0 } = node;
+            let vecX = 0;
+            let vecY = 0;
+            let vecZ = 0;
+            let currentGravity = gravity;
+            const degree = model.degree(id);
+            const forceCenter = getCenter === null || getCenter === void 0 ? void 0 : getCenter(node, degree);
+            if (forceCenter) {
+                const [centerX, centerY, strength] = forceCenter;
+                vecX = x - centerX;
+                vecY = y - centerY;
+                currentGravity = strength;
+            }
+            else {
+                vecX = x - center[0];
+                vecY = y - center[1];
+                vecZ = z - (center[2] || 0);
+            }
+            if (currentGravity) {
+                accMap[id].x -= (currentGravity * vecX) / mass;
+                accMap[id].y -= (currentGravity * vecY) / mass;
+                accMap[id].z -= (currentGravity * vecZ) / mass;
+            }
+        });
+    }
+    force.center = function (_) {
+        return arguments.length ? ((center = _), force) : center;
+    };
+    force.gravity = function (_) {
+        return arguments.length ? ((gravity = _), force) : gravity;
+    };
+    force.getCenter = function (_) {
+        return arguments.length ? ((getCenter = _), force) : getCenter;
+    };
+    return force;
+}
+
+export { forceGravity };
+//# sourceMappingURL=gravity.js.map

package/lib/force/gravity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gravity.js","sources":["../../src/force/src/force/gravity.ts"],"sourcesContent":[null],"names":[],"mappings":"AAIA;;;AAGG;AACG,SAAU,YAAY,CAAC,MAAA,GAAgB,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,UAAkB,EAAE,EAAA;AAC1E,IAAA,IAAI,SAAuE;AAE3E,IAAA,SAAS,KAAK,CAAC,KAAkB,EAAE,MAAc,EAAA;AAC/C,QAAA,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,EAAE;AAC3B,QAAA,IAAI,CAAC,KAAK;YAAE;AAEZ,QAAA,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,KAAI;AACzB,YAAA,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,GAAG,IAAI;YAEtC,IAAI,IAAI,GAAG,CAAC;YACZ,IAAI,IAAI,GAAG,CAAC;YACZ,IAAI,IAAI,GAAG,CAAC;YACZ,IAAI,cAAc,GAAG,OAAO;YAE5B,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;AAC/B,YAAA,MAAM,WAAW,GAAG,SAAS,KAAA,IAAA,IAAT,SAAS,KAAA,MAAA,GAAA,MAAA,GAAT,SAAS,CAAG,IAAI,EAAE,MAAM,CAAC;AAE7C,YAAA,IAAI,WAAW,EAAE;gBACf,MAAM,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,CAAC,GAAG,WAAW;AAChD,gBAAA,IAAI,GAAG,CAAC,GAAG,OAAO;AAClB,gBAAA,IAAI,GAAG,CAAC,GAAG,OAAO;gBAClB,cAAc,GAAG,QAAQ;AAC1B,YAAA;AAAM,iBAAA;AACL,gBAAA,IAAI,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC;AACpB,gBAAA,IAAI,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC;gBACpB,IAAI,GAAG,CAAC,IAAI,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;AAC5B,YAAA;AAED,YAAA,IAAI,cAAc,EAAE;AAClB,gBAAA,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,IAAI,IAAI;AAC9C,gBAAA,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,IAAI,IAAI;AAC9C,gBAAA,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,IAAI,IAAI;AAC/C,YAAA;AACH,QAAA,CAAC,CAAC;IACJ;AAEA,IAAA,KAAK,CAAC,MAAM,GAAG,UAAU,CAAS,EAAA;AAChC,QAAA,OAAO,SAAS,CAAC,MAAM,IAAI,CAAC,MAAM,GAAG,CAAE,GAAG,KAAK,IAAI,MAAM;AAC3D,IAAA,CAAC;AAED,IAAA,KAAK,CAAC,OAAO,GAAG,UAAU,CAAU,EAAA;AAClC,QAAA,OAAO,SAAS,CAAC,MAAM,IAAI,CAAC,OAAO,GAAG,CAAE,GAAG,KAAK,IAAI,OAAO;AAC7D,IAAA,CAAC;AAED,IAAA,KAAK,CAAC,SAAS,GAAG,UAChB,CAAkD,EAAA;AAElD,QAAA,OAAO,SAAS,CAAC,MAAM,IAAI,CAAC,SAAS,GAAG,CAAC,GAAG,KAAK,IAAI,SAAS;AAChE,IAAA,CAAC;AAED,IAAA,OAAO,KAAK;AACd;;;;"}