@antv/layout 2.0.0 → 2.0.1

package/esm/algorithm/dagre/index.js

@@ -0,0 +1,143 @@
+import { __awaiter } from 'tslib';
+import { pick, isNil, isBoolean } from '@antv/util';
+import dagre, { graphlib } from 'dagre';
+import '@antv/expr';
+import { formatSizeFn, formatNumberFn, formatFn } from '../../util/format.js';
+import { parsePoint } from '../../util/point.js';
+import { parseSize } from '../../util/size.js';
+import { BaseLayout } from '../base-layout.js';
+
+/**
+ * <zh/> Dagre 布局
+ *
+ * <en/> Dagre layout
+ */
+class DagreLayout extends BaseLayout {
+    constructor() {
+        super(...arguments);
+        this.id = 'dagre';
+        this.isCompoundGraph = null;
+        this.config = {
+            graphAttributes: [
+                'rankdir',
+                'align',
+                'nodesep',
+                'edgesep',
+                'ranksep',
+                'marginx',
+                'marginy',
+                'acyclicer',
+                'ranker',
+            ],
+            nodeAttributes: ['width', 'height'],
+            edgeAttributes: [
+                'minlen',
+                'weight',
+                'width',
+                'height',
+                'labelpos',
+                'labeloffset',
+            ],
+        };
+    }
+    getDefaultOptions() {
+        return {
+            directed: true,
+            multigraph: true,
+            rankdir: 'TB',
+            align: undefined,
+            nodesep: 50,
+            edgesep: 10,
+            ranksep: 50,
+            marginx: 0,
+            marginy: 0,
+            acyclicer: undefined,
+            ranker: 'network-simplex',
+            nodeSize: [0, 0],
+            edgeMinLen: 1,
+            edgeWeight: 1,
+            edgeLabelSize: [0, 0],
+            edgeLabelPos: 'r',
+            edgeLabelOffset: 10,
+        };
+    }
+    layout() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const g = new graphlib.Graph({
+                directed: !!this.options.directed,
+                multigraph: !!this.options.multigraph,
+                compound: this.isCompound(),
+            });
+            g.setGraph(pick(this.options, this.config.graphAttributes));
+            g.setDefaultEdgeLabel(() => ({}));
+            const nodeSizeFn = formatSizeFn(this.options.nodeSize, 0);
+            this.model.forEachNode((node) => {
+                const raw = node._original;
+                const [width, height] = parseSize(nodeSizeFn(raw));
+                const label = { width, height };
+                g.setNode(String(node.id), label);
+                if (this.isCompound()) {
+                    if (isNil(node.parentId))
+                        return;
+                    g.setParent(String(node.id), String(node.parentId));
+                }
+            });
+            const { edgeLabelSize, edgeLabelOffset, edgeLabelPos, edgeMinLen, edgeWeight, } = this.options;
+            const edgeLabelSizeFn = formatSizeFn(edgeLabelSize, 0, 'edge');
+            const edgeLabelOffsetFn = formatNumberFn(edgeLabelOffset, 10, 'edge');
+            const edgeLabelPosFn = typeof edgeLabelPos === 'string'
+                ? () => edgeLabelPos
+                : formatFn(edgeLabelPos, ['edge']);
+            const edgeMinLenFn = formatNumberFn(edgeMinLen, 1, 'edge');
+            const edgeWeightFn = formatNumberFn(edgeWeight, 1, 'edge');
+            this.model.forEachEdge((edge) => {
+                const raw = edge._original;
+                const [lw, lh] = parseSize(edgeLabelSizeFn(raw));
+                const label = {
+                    width: lw,
+                    height: lh,
+                    labelpos: edgeLabelPosFn(raw),
+                    labeloffset: edgeLabelOffsetFn(raw),
+                    minlen: edgeMinLenFn(raw),
+                    weight: edgeWeightFn(raw),
+                };
+                g.setEdge(String(edge.source), String(edge.target), label, String(edge.id));
+            });
+            dagre.layout(g);
+            this.model.forEachNode((node) => {
+                const data = g.node(String(node.id));
+                if (!data)
+                    return;
+                node.x = data.x;
+                node.y = data.y;
+                node.size = [data.width, data.height];
+            });
+            this.model.forEachEdge((edge) => {
+                const data = g.edge(String(edge.source), String(edge.target), String(edge.id));
+                if (!data)
+                    return;
+                const { width, height, weight, minlen, labelpos, labeloffset, points } = data;
+                edge.labelSize = [width, height];
+                edge.weight = weight;
+                edge.minLen = minlen;
+                edge.labelPos = labelpos;
+                edge.labelOffset = labeloffset;
+                edge.points = points.map(parsePoint);
+            });
+        });
+    }
+    isCompound() {
+        if (this.isCompoundGraph !== null)
+            return this.isCompoundGraph;
+        if (isBoolean(this.options.compound)) {
+            return (this.isCompoundGraph = this.options.compound);
+        }
+        this.isCompoundGraph = this.model
+            .nodes()
+            .some((node) => !isNil(node.parentId));
+        return this.isCompoundGraph;
+    }
+}
+
+export { DagreLayout };
+//# sourceMappingURL=index.js.map

package/esm/algorithm/dagre/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sources":["../../../src/algorithm/dagre/src/algorithm/dagre/index.ts"],"sourcesContent":[null],"names":[],"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;IAoIH;IAlIY,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,IAAI,QAAQ,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,EAAE,MAAM,CAAC;YAC9D,MAAM,iBAAiB,GAAG,cAAc,CAAC,eAAe,EAAE,EAAE,EAAE,MAAM,CAAC;AACrE,YAAA,MAAM,cAAc,GAClB,OAAO,YAAY,KAAK;AACtB,kBAAE,MAAM;kBACN,QAAQ,CAAC,YAAY,EAAE,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,YAAY,GAAG,cAAc,CAAC,UAAU,EAAE,CAAC,EAAE,MAAM,CAAC;YAC1D,MAAM,YAAY,GAAG,cAAc,CAAC,UAAU,EAAE,CAAC,EAAE,MAAM,CAAC;YAE1D,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/esm/algorithm/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/esm/algorithm/force/attractive.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"attractive.js","sources":["../../../src/algorithm/force/src/algorithm/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,KAAe,EAAE,MAAc,EAAA;AAC5C,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/esm/algorithm/force/centripetal.js

@@ -0,0 +1,87 @@
+import { isNumber } from '@antv/util';
+
+/**
+ * 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/esm/algorithm/force/centripetal.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"centripetal.js","sources":["../../../src/algorithm/force/src/algorithm/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,KAAe,EAAE,MAAc,EAAA;AAC5C,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/esm/algorithm/force/collide.js

@@ -0,0 +1,172 @@
+import { isNumber } from '@antv/util';
+import { quadtree } from 'd3-quadtree';
+
+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/esm/algorithm/force/collide.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"collide.js","sources":["../../../src/algorithm/force/src/algorithm/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,KAAe,EAAE,MAAc,EAAA;AAC5C,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/esm/algorithm/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/esm/algorithm/force/gravity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gravity.js","sources":["../../../src/algorithm/force/src/algorithm/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,KAAe,EAAE,MAAc,EAAA;AAC5C,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;;;;"}