@antv/layout 1.1.0 → 1.2.1

package/lib/d3Force/forceInBox.js

@@ -0,0 +1,331 @@
+import { __read } from "tslib";
+import * as d3Force from "d3-force";
+// https://github.com/john-guerra/forceInABox/blob/master/src/forceInABox.js
+export default function forceInBox() {
+    function constant(_) {
+        return function () { return _; };
+    }
+    var groupBy = function (d) {
+        return d.cluster;
+    };
+    var forceNodeSize = constant(1);
+    var forceCharge = constant(-1);
+    var forceLinkDistance = constant(100);
+    var forceLinkStrength = constant(0.1);
+    var offset = [0, 0];
+    var nodes = [];
+    var nodesMap = {};
+    var links = [];
+    var centerX = 100;
+    var centerY = 100;
+    var foci = {
+        none: {
+            x: 0,
+            y: 0,
+        },
+    };
+    var templateNodes = [];
+    var templateForce;
+    var template = "force";
+    var enableGrouping = true;
+    var strength = 0.1;
+    function force(alpha) {
+        if (!enableGrouping) {
+            return force;
+        }
+        templateForce.tick();
+        getFocisFromTemplate();
+        for (var i = 0, n = nodes.length, node = void 0, k = alpha * strength; i < n; ++i) {
+            node = nodes[i];
+            node.vx += (foci[groupBy(node)].x - node.x) * k;
+            node.vy += (foci[groupBy(node)].y - node.y) * k;
+        }
+    }
+    function initialize() {
+        if (!nodes)
+            return;
+        initializeWithForce();
+    }
+    function initializeWithForce() {
+        if (!nodes || !nodes.length) {
+            return;
+        }
+        if (groupBy(nodes[0]) === undefined) {
+            throw Error("Couldnt find the grouping attribute for the nodes. Make sure to set it up with forceInBox.groupBy('clusterAttr') before calling .links()");
+        }
+        // checkLinksAsObjects();
+        var net = getGroupsGraph();
+        templateForce = d3Force
+            .forceSimulation(net.nodes)
+            .force("x", d3Force.forceX(centerX).strength(0.1))
+            .force("y", d3Force.forceY(centerY).strength(0.1))
+            .force("collide", d3Force.forceCollide(function (d) { return d.r; }).iterations(4))
+            .force("charge", d3Force.forceManyBody().strength(forceCharge))
+            .force("links", d3Force
+            .forceLink(net.nodes.length ? net.links : [])
+            .distance(forceLinkDistance)
+            .strength(forceLinkStrength));
+        templateNodes = templateForce.nodes();
+        getFocisFromTemplate();
+    }
+    function getGroupsGraph() {
+        var gnodes = [];
+        var glinks = [];
+        var dNodes = {};
+        var clustersList = [];
+        var clustersCounts = {};
+        var clustersLinks = [];
+        clustersCounts = computeClustersNodeCounts(nodes);
+        clustersLinks = computeClustersLinkCounts(links);
+        clustersList = Object.keys(clustersCounts);
+        clustersList.forEach(function (key, index) {
+            var val = clustersCounts[key];
+            // Uses approx meta-node size
+            gnodes.push({
+                id: key,
+                size: val.count,
+                r: Math.sqrt(val.sumforceNodeSize / Math.PI),
+            });
+            dNodes[key] = index;
+        });
+        clustersLinks.forEach(function (link) {
+            var sourceTerminal = link.source;
+            var targetTerminal = link.target;
+            var source = dNodes[sourceTerminal];
+            var target = dNodes[targetTerminal];
+            if (source !== undefined && target !== undefined) {
+                glinks.push({
+                    source: source,
+                    target: target,
+                    count: link.count,
+                });
+            }
+        });
+        return {
+            nodes: gnodes,
+            links: glinks,
+        };
+    }
+    function computeClustersNodeCounts(nodes) {
+        var clustersCounts = {};
+        nodes.forEach(function (d) {
+            var key = groupBy(d);
+            if (!clustersCounts[key]) {
+                clustersCounts[key] = {
+                    count: 0,
+                    sumforceNodeSize: 0,
+                };
+            }
+        });
+        nodes.forEach(function (d) {
+            var key = groupBy(d);
+            var nodeSize = forceNodeSize(d);
+            var tmpCount = clustersCounts[key];
+            tmpCount.count = tmpCount.count + 1;
+            tmpCount.sumforceNodeSize =
+                tmpCount.sumforceNodeSize + Math.PI * (nodeSize * nodeSize) * 1.3;
+            clustersCounts[key] = tmpCount;
+        });
+        return clustersCounts;
+    }
+    function computeClustersLinkCounts(links) {
+        var dClusterLinks = {};
+        var clusterLinks = [];
+        links.forEach(function (l) {
+            var key = getLinkKey(l);
+            var count = 0;
+            if (dClusterLinks[key] !== undefined) {
+                count = dClusterLinks[key];
+            }
+            count += 1;
+            dClusterLinks[key] = count;
+        });
+        // @ts-ignore
+        var entries = Object.entries(dClusterLinks);
+        entries.forEach(function (_a) {
+            var _b = __read(_a, 2), key = _b[0], count = _b[1];
+            var source = key.split("~")[0];
+            var target = key.split("~")[1];
+            if (source !== undefined && target !== undefined) {
+                clusterLinks.push({
+                    source: source,
+                    target: target,
+                    count: count,
+                });
+            }
+        });
+        return clusterLinks;
+    }
+    function getFocisFromTemplate() {
+        foci = {
+            none: {
+                x: 0,
+                y: 0,
+            },
+        };
+        templateNodes.forEach(function (d) {
+            foci[d.id] = {
+                x: d.x - offset[0],
+                y: d.y - offset[1],
+            };
+        });
+        return foci;
+    }
+    function getLinkKey(link) {
+        var source = link.source;
+        var target = link.target;
+        var sourceID = groupBy(nodesMap[source]);
+        var targetID = groupBy(nodesMap[target]);
+        return sourceID <= targetID
+            ? sourceID + "~" + targetID
+            : targetID + "~" + sourceID;
+    }
+    function genNodesMap(nodes) {
+        nodesMap = {};
+        nodes.forEach(function (node) {
+            nodesMap[node.id] = node;
+        });
+    }
+    function setTemplate(x) {
+        if (!arguments.length)
+            return template;
+        template = x;
+        initialize();
+        return force;
+    }
+    function setGroupBy(x) {
+        if (!arguments.length)
+            return groupBy;
+        if (typeof x === "string") {
+            groupBy = function (d) {
+                return d[x];
+            };
+            return force;
+        }
+        groupBy = x;
+        return force;
+    }
+    function setEnableGrouping(x) {
+        if (!arguments.length)
+            return enableGrouping;
+        enableGrouping = x;
+        return force;
+    }
+    function setStrength(x) {
+        if (!arguments.length)
+            return strength;
+        strength = x;
+        return force;
+    }
+    function setCenterX(_) {
+        if (arguments.length) {
+            centerX = _;
+            return force;
+        }
+        return centerX;
+    }
+    function setCenterY(_) {
+        if (arguments.length) {
+            centerY = _;
+            return force;
+        }
+        return centerY;
+    }
+    function setNodes(_) {
+        if (arguments.length) {
+            genNodesMap(_ || []);
+            nodes = _ || [];
+            return force;
+        }
+        return nodes;
+    }
+    function setLinks(_) {
+        if (arguments.length) {
+            links = _ || [];
+            initialize();
+            return force;
+        }
+        return links;
+    }
+    function setForceNodeSize(_) {
+        if (arguments.length) {
+            if (typeof _ === "function") {
+                forceNodeSize = _;
+            }
+            else {
+                forceNodeSize = constant(+_);
+            }
+            initialize();
+            return force;
+        }
+        return forceNodeSize;
+    }
+    function setForceCharge(_) {
+        if (arguments.length) {
+            if (typeof _ === "function") {
+                forceCharge = _;
+            }
+            else {
+                forceCharge = constant(+_);
+            }
+            initialize();
+            return force;
+        }
+        return forceCharge;
+    }
+    function setForceLinkDistance(_) {
+        if (arguments.length) {
+            if (typeof _ === "function") {
+                forceLinkDistance = _;
+            }
+            else {
+                forceLinkDistance = constant(+_);
+            }
+            initialize();
+            return force;
+        }
+        return forceLinkDistance;
+    }
+    function setForceLinkStrength(_) {
+        if (arguments.length) {
+            if (typeof _ === "function") {
+                forceLinkStrength = _;
+            }
+            else {
+                forceLinkStrength = constant(+_);
+            }
+            initialize();
+            return force;
+        }
+        return forceLinkStrength;
+    }
+    function setOffset(_) {
+        if (arguments.length) {
+            offset = _;
+            return force;
+        }
+        return offset;
+    }
+    force.initialize = function (_) {
+        nodes = _;
+        initialize();
+    };
+    force.template = setTemplate;
+    force.groupBy = setGroupBy;
+    force.enableGrouping = setEnableGrouping;
+    force.strength = setStrength;
+    force.centerX = setCenterX;
+    force.centerY = setCenterY;
+    force.nodes = setNodes;
+    force.links = setLinks;
+    force.forceNodeSize = setForceNodeSize;
+    // Legacy support
+    force.nodeSize = force.forceNodeSize;
+    force.forceCharge = setForceCharge;
+    force.forceLinkDistance = setForceLinkDistance;
+    force.forceLinkStrength = setForceLinkStrength;
+    force.offset = setOffset;
+    force.getFocis = getFocisFromTemplate;
+    return force;
+}
+//# sourceMappingURL=forceInBox.js.map

package/lib/d3Force/forceInBox.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"forceInBox.js","sourceRoot":"","sources":["../../src/d3Force/forceInBox.ts"],"names":[],"mappings":";AAAA,OAAO,KAAK,OAAO,MAAM,UAAU,CAAC;AAWpC,4EAA4E;AAC5E,MAAM,CAAC,OAAO,UAAU,UAAU;IAChC,SAAS,QAAQ,CAAC,CAAM;QACtB,OAAO,cAAM,OAAA,CAAC,EAAD,CAAC,CAAC;IACjB,CAAC;IAED,IAAI,OAAO,GAAG,UAAC,CAAQ;QACrB,OAAO,CAAC,CAAC,OAAO,CAAC;IACnB,CAAC,CAAC;IACF,IAAI,aAAa,GAA0C,QAAQ,CAAC,CAAC,CAAC,CAAC;IACvE,IAAI,WAAW,GAA0C,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IACtE,IAAI,iBAAiB,GAA0C,QAAQ,CAAC,GAAG,CAAC,CAAC;IAC7E,IAAI,iBAAiB,GAA0C,QAAQ,CAAC,GAAG,CAAC,CAAC;IAC7E,IAAI,MAAM,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAEpB,IAAI,KAAK,GAAY,EAAE,CAAC;IACxB,IAAI,QAAQ,GAAQ,EAAE,CAAC;IACvB,IAAI,KAAK,GAAU,EAAE,CAAC;IACtB,IAAI,OAAO,GAAG,GAAG,CAAC;IAClB,IAAI,OAAO,GAAG,GAAG,CAAC;IAClB,IAAI,IAAI,GAAQ;QACd,IAAI,EAAE;YACJ,CAAC,EAAE,CAAC;YACJ,CAAC,EAAE,CAAC;SACL;KACF,CAAC;IACF,IAAI,aAAa,GAAY,EAAE,CAAC;IAChC,IAAI,aAAkB,CAAC;IACvB,IAAI,QAAQ,GAAG,OAAO,CAAC;IACvB,IAAI,cAAc,GAAG,IAAI,CAAC;IAC1B,IAAI,QAAQ,GAAG,GAAG,CAAC;IAEnB,SAAS,KAAK,CAAC,KAAa;QAC1B,IAAI,CAAC,cAAc,EAAE;YACnB,OAAO,KAAK,CAAC;SACd;QACD,aAAa,CAAC,IAAI,EAAE,CAAC;QACrB,oBAAoB,EAAE,CAAC;QAEvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,IAAI,SAAA,EAAE,CAAC,GAAG,KAAK,GAAG,QAAQ,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,EAAE;YACxE,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YAChB,IAAI,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YAChD,IAAI,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;SACjD;IACH,CAAC;IAED,SAAS,UAAU;QACjB,IAAI,CAAC,KAAK;YAAE,OAAO;QACnB,mBAAmB,EAAE,CAAC;IACxB,CAAC;IAED,SAAS,mBAAmB;QAC1B,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE;YAC3B,OAAO;SACR;QAED,IAAI,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,SAAS,EAAE;YACnC,MAAM,KAAK,CACT,0IAA0I,CAC3I,CAAC;SACH;QAED,yBAAyB;QAEzB,IAAM,GAAG,GAAG,cAAc,EAAE,CAAC;QAC7B,aAAa,GAAG,OAAO;aACpB,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC;aAC1B,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;aACjD,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;aACjD,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,YAAY,CAAC,UAAC,CAAM,IAAK,OAAA,CAAC,CAAC,CAAC,EAAH,CAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;aACrE,KAAK,CAAC,QAAQ,EAAE,OAAO,CAAC,aAAa,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;aAC9D,KAAK,CACJ,OAAO,EACP,OAAO;aACJ,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;aAC5C,QAAQ,CAAC,iBAAiB,CAAC;aAC3B,QAAQ,CAAC,iBAAiB,CAAC,CAC/B,CAAC;QAEJ,aAAa,GAAG,aAAa,CAAC,KAAK,EAAE,CAAC;QAEtC,oBAAoB,EAAE,CAAC;IACzB,CAAC;IAED,SAAS,cAAc;QACrB,IAAM,MAAM,GAAQ,EAAE,CAAC;QACvB,IAAM,MAAM,GAAQ,EAAE,CAAC;QACvB,IAAM,MAAM,GAAQ,EAAE,CAAC;QACvB,IAAI,YAAY,GAAa,EAAE,CAAC;QAChC,IAAI,cAAc,GAAQ,EAAE,CAAC;QAC7B,IAAI,aAAa,GAAQ,EAAE,CAAC;QAE5B,cAAc,GAAG,yBAAyB,CAAC,KAAK,CAAC,CAAC;QAClD,aAAa,GAAG,yBAAyB,CAAC,KAAK,CAAC,CAAC;QAEjD,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAE3C,YAAY,CAAC,OAAO,CAAC,UAAC,GAAG,EAAE,KAAK;YAC9B,IAAM,GAAG,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;YAChC,6BAA6B;YAC7B,MAAM,CAAC,IAAI,CAAC;gBACV,EAAE,EAAE,GAAG;gBACP,IAAI,EAAE,GAAG,CAAC,KAAK;gBACf,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,gBAAgB,GAAG,IAAI,CAAC,EAAE,CAAC;aAC7C,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QACtB,CAAC,CAAC,CAAC;QAEH,aAAa,CAAC,OAAO,CAAC,UAAC,IAAS;YAC9B,IAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC;YACnC,IAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC;YACnC,IAAM,MAAM,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC;YACtC,IAAM,MAAM,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC;YACtC,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,SAAS,EAAE;gBAChD,MAAM,CAAC,IAAI,CAAC;oBACV,MAAM,QAAA;oBACN,MAAM,QAAA;oBACN,KAAK,EAAE,IAAI,CAAC,KAAK;iBAClB,CAAC,CAAC;aACJ;QACH,CAAC,CAAC,CAAC;QAEH,OAAO;YACL,KAAK,EAAE,MAAM;YACb,KAAK,EAAE,MAAM;SACd,CAAC;IACJ,CAAC;IAED,SAAS,yBAAyB,CAAC,KAAU;QAC3C,IAAM,cAAc,GAAQ,EAAE,CAAC;QAE/B,KAAK,CAAC,OAAO,CAAC,UAAC,CAAM;YACnB,IAAM,GAAG,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACvB,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,EAAE;gBACxB,cAAc,CAAC,GAAG,CAAC,GAAG;oBACpB,KAAK,EAAE,CAAC;oBACR,gBAAgB,EAAE,CAAC;iBACpB,CAAC;aACH;QACH,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,OAAO,CAAC,UAAC,CAAM;YACnB,IAAM,GAAG,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACvB,IAAM,QAAQ,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;YAClC,IAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;YACrC,QAAQ,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,GAAG,CAAC,CAAC;YACpC,QAAQ,CAAC,gBAAgB;gBACvB,QAAQ,CAAC,gBAAgB,GAAG,IAAI,CAAC,EAAE,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC,GAAG,GAAG,CAAC;YACpE,cAAc,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,SAAS,yBAAyB,CAAC,KAAU;QAC3C,IAAM,aAAa,GAAQ,EAAE,CAAC;QAC9B,IAAM,YAAY,GAAQ,EAAE,CAAC;QAC7B,KAAK,CAAC,OAAO,CAAC,UAAC,CAAM;YACnB,IAAM,GAAG,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YAC1B,IAAI,KAAK,GAAG,CAAC,CAAC;YACd,IAAI,aAAa,CAAC,GAAG,CAAC,KAAK,SAAS,EAAE;gBACpC,KAAK,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;aAC5B;YACD,KAAK,IAAI,CAAC,CAAC;YACX,aAAa,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,aAAa;QACb,IAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAE9C,OAAO,CAAC,OAAO,CAAC,UAAC,EAAiB;gBAAjB,KAAA,aAAiB,EAAhB,GAAG,QAAA,EAAE,KAAK,QAAA;YAC1B,IAAM,MAAM,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACjC,IAAM,MAAM,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACjC,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,SAAS,EAAE;gBAChD,YAAY,CAAC,IAAI,CAAC;oBAChB,MAAM,QAAA;oBACN,MAAM,QAAA;oBACN,KAAK,OAAA;iBACN,CAAC,CAAC;aACJ;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,SAAS,oBAAoB;QAC3B,IAAI,GAAG;YACL,IAAI,EAAE;gBACJ,CAAC,EAAE,CAAC;gBACJ,CAAC,EAAE,CAAC;aACL;SACF,CAAC;QACF,aAAa,CAAC,OAAO,CAAC,UAAC,CAAC;YACtB,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG;gBACX,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC;gBAClB,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC;aACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAED,SAAS,UAAU,CAAC,IAAS;QAC3B,IAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;QAC3C,IAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;QAE3C,OAAO,QAAQ,IAAI,QAAQ;YACzB,CAAC,CAAI,QAAQ,SAAI,QAAU;YAC3B,CAAC,CAAI,QAAQ,SAAI,QAAU,CAAC;IAChC,CAAC;IAED,SAAS,WAAW,CAAC,KAAU;QAC7B,QAAQ,GAAG,EAAE,CAAC;QACd,KAAK,CAAC,OAAO,CAAC,UAAC,IAAS;YACtB,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC;IAED,SAAS,WAAW,CAAC,CAAM;QACzB,IAAI,CAAC,SAAS,CAAC,MAAM;YAAE,OAAO,QAAQ,CAAC;QACvC,QAAQ,GAAG,CAAC,CAAC;QACb,UAAU,EAAE,CAAC;QACb,OAAO,KAAK,CAAC;IACf,CAAC;IAED,SAAS,UAAU,CAAC,CAAM;QACxB,IAAI,CAAC,SAAS,CAAC,MAAM;YAAE,OAAO,OAAO,CAAC;QACtC,IAAI,OAAO,CAAC,KAAK,QAAQ,EAAE;YACzB,OAAO,GAAG,UAAC,CAAM;gBACf,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;YACd,CAAC,CAAC;YACF,OAAO,KAAK,CAAC;SACd;QACD,OAAO,GAAG,CAAC,CAAC;QACZ,OAAO,KAAK,CAAC;IACf,CAAC;IAED,SAAS,iBAAiB,CAAC,CAAM;QAC/B,IAAI,CAAC,SAAS,CAAC,MAAM;YAAE,OAAO,cAAc,CAAC;QAC7C,cAAc,GAAG,CAAC,CAAC;QACnB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,SAAS,WAAW,CAAC,CAAM;QACzB,IAAI,CAAC,SAAS,CAAC,MAAM;YAAE,OAAO,QAAQ,CAAC;QACvC,QAAQ,GAAG,CAAC,CAAC;QACb,OAAO,KAAK,CAAC;IACf,CAAC;IAED,SAAS,UAAU,CAAC,CAAM;QACxB,IAAI,SAAS,CAAC,MAAM,EAAE;YACpB,OAAO,GAAG,CAAC,CAAC;YACZ,OAAO,KAAK,CAAC;SACd;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,SAAS,UAAU,CAAC,CAAM;QACxB,IAAI,SAAS,CAAC,MAAM,EAAE;YACpB,OAAO,GAAG,CAAC,CAAC;YACZ,OAAO,KAAK,CAAC;SACd;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,SAAS,QAAQ,CAAC,CAAM;QACtB,IAAI,SAAS,CAAC,MAAM,EAAE;YACpB,WAAW,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YACrB,KAAK,GAAG,CAAC,IAAI,EAAE,CAAC;YAChB,OAAO,KAAK,CAAC;SACd;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,SAAS,QAAQ,CAAC,CAAM;QACtB,IAAI,SAAS,CAAC,MAAM,EAAE;YACpB,KAAK,GAAG,CAAC,IAAI,EAAE,CAAC;YAChB,UAAU,EAAE,CAAC;YACb,OAAO,KAAK,CAAC;SACd;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,SAAS,gBAAgB,CAAC,CAAM;QAC9B,IAAI,SAAS,CAAC,MAAM,EAAE;YACpB,IAAI,OAAO,CAAC,KAAK,UAAU,EAAE;gBAC3B,aAAa,GAAG,CAAC,CAAC;aACnB;iBAAM;gBACL,aAAa,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;aAC9B;YACD,UAAU,EAAE,CAAC;YACb,OAAO,KAAK,CAAC;SACd;QAED,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,SAAS,cAAc,CAAC,CAAM;QAC5B,IAAI,SAAS,CAAC,MAAM,EAAE;YACpB,IAAI,OAAO,CAAC,KAAK,UAAU,EAAE;gBAC3B,WAAW,GAAG,CAAC,CAAC;aACjB;iBAAM;gBACL,WAAW,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;aAC5B;YACD,UAAU,EAAE,CAAC;YACb,OAAO,KAAK,CAAC;SACd;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,SAAS,oBAAoB,CAAC,CAAM;QAClC,IAAI,SAAS,CAAC,MAAM,EAAE;YACpB,IAAI,OAAO,CAAC,KAAK,UAAU,EAAE;gBAC3B,iBAAiB,GAAG,CAAC,CAAC;aACvB;iBAAM;gBACL,iBAAiB,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;aAClC;YACD,UAAU,EAAE,CAAC;YACb,OAAO,KAAK,CAAC;SACd;QAED,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAED,SAAS,oBAAoB,CAAC,CAAM;QAClC,IAAI,SAAS,CAAC,MAAM,EAAE;YACpB,IAAI,OAAO,CAAC,KAAK,UAAU,EAAE;gBAC3B,iBAAiB,GAAG,CAAC,CAAC;aACvB;iBAAM;gBACL,iBAAiB,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;aAClC;YACD,UAAU,EAAE,CAAC;YACb,OAAO,KAAK,CAAC;SACd;QAED,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAED,SAAS,SAAS,CAAC,CAAM;QACvB,IAAI,SAAS,CAAC,MAAM,EAAE;YACpB,MAAM,GAAG,CAAC,CAAC;YACX,OAAO,KAAK,CAAC;SACd;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,UAAU,GAAG,UAAC,CAAM;QACxB,KAAK,GAAG,CAAC,CAAC;QACV,UAAU,EAAE,CAAC;IACf,CAAC,CAAC;IAEF,KAAK,CAAC,QAAQ,GAAG,WAAW,CAAC;IAE7B,KAAK,CAAC,OAAO,GAAG,UAAU,CAAC;IAE3B,KAAK,CAAC,cAAc,GAAG,iBAAiB,CAAC;IAEzC,KAAK,CAAC,QAAQ,GAAG,WAAW,CAAC;IAE7B,KAAK,CAAC,OAAO,GAAG,UAAU,CAAC;IAE3B,KAAK,CAAC,OAAO,GAAG,UAAU,CAAC;IAE3B,KAAK,CAAC,KAAK,GAAG,QAAQ,CAAC;IAEvB,KAAK,CAAC,KAAK,GAAG,QAAQ,CAAC;IAEvB,KAAK,CAAC,aAAa,GAAG,gBAAgB,CAAC;IAEvC,iBAAiB;IACjB,KAAK,CAAC,QAAQ,GAAG,KAAK,CAAC,aAAa,CAAC;IAErC,KAAK,CAAC,WAAW,GAAG,cAAc,CAAC;IAEnC,KAAK,CAAC,iBAAiB,GAAG,oBAAoB,CAAC;IAE/C,KAAK,CAAC,iBAAiB,GAAG,oBAAoB,CAAC;IAE/C,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC;IAEzB,KAAK,CAAC,QAAQ,GAAG,oBAAoB,CAAC;IAEtC,OAAO,KAAK,CAAC;AACf,CAAC"}

package/lib/d3Force/index.js

@@ -0,0 +1,339 @@
+import { __assign, __awaiter, __generator, __rest } from "tslib";
+import * as d3Force from "d3-force";
+import { isFunction, isNumber, isObject } from "@antv/util";
+import { cloneFormatData, isArray } from "../util";
+import forceInBox from "./forceInBox";
+var DEFAULTS_LAYOUT_OPTIONS = {
+    center: [0, 0],
+    preventOverlap: false,
+    nodeSize: undefined,
+    nodeSpacing: undefined,
+    linkDistance: 50,
+    forceSimulation: null,
+    alphaDecay: 0.028,
+    alphaMin: 0.001,
+    alpha: 0.3,
+    collideStrength: 1,
+    clustering: false,
+    clusterNodeStrength: -1,
+    clusterEdgeStrength: 0.1,
+    clusterEdgeDistance: 100,
+    clusterFociStrength: 0.8,
+    clusterNodeSize: 10,
+};
+/**
+ * Layout the nodes' positions with d3's basic classic force
+ *
+ * @example
+ * // Assign layout options when initialization.
+ * const layout = new D3ForceLayout({ center: [100, 100] });
+ * const positions = await layout.execute(graph); // { nodes: [], edges: [] }
+ *
+ * // Or use different options later.
+ * const layout = new D3ForceLayout({ center: [100, 100] });
+ * const positions = await layout.execute(graph, { center: [100, 100] }); // { nodes: [], edges: [] }
+ *
+ * // If you want to assign the positions directly to the nodes, use assign method.
+ * await layout.assign(graph, { center: [100, 100] });
+ */
+var D3ForceLayout = /** @class */ (function () {
+    function D3ForceLayout(options) {
+        if (options === void 0) { options = {}; }
+        this.options = options;
+        this.id = "d3force";
+        this.options = __assign(__assign({}, DEFAULTS_LAYOUT_OPTIONS), options);
+    }
+    /**
+     * Return the positions of nodes and edges(if needed).
+     */
+    D3ForceLayout.prototype.execute = function (graph, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, this.genericForceLayout(false, graph, options)];
+            });
+        });
+    };
+    /**
+     * To directly assign the positions to the nodes.
+     */
+    D3ForceLayout.prototype.assign = function (graph, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                this.genericForceLayout(true, graph, options);
+                return [2 /*return*/];
+            });
+        });
+    };
+    /**
+     * Stop simulation immediately.
+     */
+    D3ForceLayout.prototype.stop = function () {
+        var _a;
+        (_a = this.forceSimulation) === null || _a === void 0 ? void 0 : _a.stop();
+    };
+    /**
+     * Manually steps the simulation by the specified number of iterations.
+     * @see https://github.com/d3/d3-force#simulation_tick
+     */
+    D3ForceLayout.prototype.tick = function (iterations) {
+        var _this = this;
+        if (iterations === void 0) { iterations = 1; }
+        this.forceSimulation.tick(iterations);
+        var result = {
+            nodes: formatOutNodes(this.lastLayoutNodes),
+            edges: formatOutEdges(this.lastLayoutEdges),
+        };
+        if (this.lastAssign) {
+            result.nodes.forEach(function (node) {
+                return _this.lastGraph.mergeNodeData(node.id, {
+                    x: node.data.x,
+                    y: node.data.y,
+                });
+            });
+        }
+        return result;
+    };
+    D3ForceLayout.prototype.genericForceLayout = function (assign, graph, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            var mergedOptions, nodes, edges, layoutNodes, layoutEdges, alphaMin, alphaDecay, alpha, nodeStrength, edgeStrength, linkDistance, clustering, clusterFociStrength, clusterEdgeDistance, clusterEdgeStrength, clusterNodeStrength, clusterNodeSize, _a, collideStrength, _b, center, preventOverlap, nodeSize, nodeSpacing, onTick, forceSimulation;
+            var _this = this;
+            return __generator(this, function (_c) {
+                mergedOptions = __assign(__assign({}, this.options), options);
+                nodes = graph.getAllNodes();
+                edges = graph.getAllEdges();
+                layoutNodes = nodes.map(function (node) {
+                    var _a, _b;
+                    return (__assign(__assign({}, cloneFormatData(node)), { x: (_a = node.data) === null || _a === void 0 ? void 0 : _a.x, y: (_b = node.data) === null || _b === void 0 ? void 0 : _b.y }));
+                });
+                layoutEdges = edges.map(function (edge) { return cloneFormatData(edge); });
+                // Use them later in `tick`.
+                this.lastLayoutNodes = layoutNodes;
+                this.lastLayoutEdges = layoutEdges;
+                this.lastAssign = assign;
+                this.lastGraph = graph;
+                alphaMin = mergedOptions.alphaMin, alphaDecay = mergedOptions.alphaDecay, alpha = mergedOptions.alpha, nodeStrength = mergedOptions.nodeStrength, edgeStrength = mergedOptions.edgeStrength, linkDistance = mergedOptions.linkDistance, clustering = mergedOptions.clustering, clusterFociStrength = mergedOptions.clusterFociStrength, clusterEdgeDistance = mergedOptions.clusterEdgeDistance, clusterEdgeStrength = mergedOptions.clusterEdgeStrength, clusterNodeStrength = mergedOptions.clusterNodeStrength, clusterNodeSize = mergedOptions.clusterNodeSize, _a = mergedOptions.collideStrength, collideStrength = _a === void 0 ? 1 : _a, _b = mergedOptions.center, center = _b === void 0 ? [0, 0] : _b, preventOverlap = mergedOptions.preventOverlap, nodeSize = mergedOptions.nodeSize, nodeSpacing = mergedOptions.nodeSpacing, onTick = mergedOptions.onTick;
+                forceSimulation = mergedOptions.forceSimulation;
+                return [2 /*return*/, new Promise(function (resolve) {
+                        if (!forceSimulation) {
+                            try {
+                                // 定义节点的力
+                                var nodeForce = d3Force.forceManyBody();
+                                if (nodeStrength) {
+                                    nodeForce.strength(nodeStrength);
+                                }
+                                forceSimulation = d3Force.forceSimulation().nodes(layoutNodes);
+                                if (clustering) {
+                                    var clusterForce = forceInBox();
+                                    clusterForce
+                                        .centerX(center[0])
+                                        .centerY(center[1])
+                                        .template("force")
+                                        .strength(clusterFociStrength);
+                                    if (layoutEdges) {
+                                        clusterForce.links(layoutEdges);
+                                    }
+                                    if (layoutNodes) {
+                                        clusterForce.nodes(layoutNodes);
+                                    }
+                                    clusterForce
+                                        .forceLinkDistance(clusterEdgeDistance)
+                                        .forceLinkStrength(clusterEdgeStrength)
+                                        .forceCharge(clusterNodeStrength)
+                                        .forceNodeSize(clusterNodeSize);
+                                    forceSimulation.force("group", clusterForce);
+                                }
+                                forceSimulation
+                                    .force("center", d3Force.forceCenter(center[0], center[1]))
+                                    .force("charge", nodeForce)
+                                    .alpha(alpha)
+                                    .alphaDecay(alphaDecay)
+                                    .alphaMin(alphaMin);
+                                if (preventOverlap) {
+                                    _this.overlapProcess(forceSimulation, {
+                                        nodeSize: nodeSize,
+                                        nodeSpacing: nodeSpacing,
+                                        collideStrength: collideStrength,
+                                    });
+                                }
+                                // 如果有边，定义边的力
+                                if (layoutEdges) {
+                                    // d3 的 forceLayout 会重新生成边的数据模型，为了避免污染源数据
+                                    var edgeForce = d3Force
+                                        .forceLink()
+                                        .id(function (d) { return d.id; })
+                                        .links(layoutEdges);
+                                    if (edgeStrength) {
+                                        edgeForce.strength(edgeStrength);
+                                    }
+                                    if (linkDistance) {
+                                        edgeForce.distance(linkDistance);
+                                    }
+                                    forceSimulation.force("link", edgeForce);
+                                }
+                                forceSimulation
+                                    .on("tick", function () {
+                                    var outNodes = formatOutNodes(layoutNodes);
+                                    onTick === null || onTick === void 0 ? void 0 : onTick({
+                                        nodes: outNodes,
+                                        edges: formatOutEdges(layoutEdges),
+                                    });
+                                    if (assign) {
+                                        outNodes.forEach(function (node) {
+                                            return graph.mergeNodeData(node.id, {
+                                                x: node.data.x,
+                                                y: node.data.y,
+                                            });
+                                        });
+                                    }
+                                })
+                                    .on("end", function () {
+                                    var outNodes = formatOutNodes(layoutNodes);
+                                    if (assign) {
+                                        outNodes.forEach(function (node) {
+                                            return graph.mergeNodeData(node.id, {
+                                                x: node.data.x,
+                                                y: node.data.y,
+                                            });
+                                        });
+                                    }
+                                    resolve({
+                                        nodes: outNodes,
+                                        edges: formatOutEdges(layoutEdges),
+                                    });
+                                });
+                            }
+                            catch (e) {
+                                console.warn(e);
+                            }
+                        }
+                        else {
+                            // forceSimulation is defined
+                            if (clustering) {
+                                var clusterForce = forceInBox();
+                                clusterForce.nodes(layoutNodes);
+                                clusterForce.links(layoutEdges);
+                            }
+                            forceSimulation.nodes(layoutNodes);
+                            if (layoutEdges) {
+                                // d3 的 forceLayout 会重新生成边的数据模型，为了避免污染源数据
+                                var edgeForce = d3Force
+                                    .forceLink()
+                                    .id(function (d) { return d.id; })
+                                    .links(layoutEdges);
+                                if (edgeStrength) {
+                                    edgeForce.strength(edgeStrength);
+                                }
+                                if (linkDistance) {
+                                    edgeForce.distance(linkDistance);
+                                }
+                                forceSimulation.force("link", edgeForce);
+                            }
+                            if (preventOverlap) {
+                                _this.overlapProcess(forceSimulation, {
+                                    nodeSize: nodeSize,
+                                    nodeSpacing: nodeSpacing,
+                                    collideStrength: collideStrength,
+                                });
+                            }
+                            forceSimulation.alpha(alpha).restart();
+                            // since d3 writes x and y as node's first level properties, format them into data
+                            var outNodes = formatOutNodes(layoutNodes);
+                            var outEdges = formatOutEdges(layoutEdges);
+                            if (assign) {
+                                outNodes.forEach(function (node) {
+                                    return graph.mergeNodeData(node.id, {
+                                        x: node.data.x,
+                                        y: node.data.y,
+                                    });
+                                });
+                            }
+                            resolve({
+                                nodes: outNodes,
+                                edges: outEdges,
+                            });
+                        }
+                        _this.forceSimulation = forceSimulation;
+                    })];
+            });
+        });
+    };
+    /**
+     * Prevent overlappings.
+     * @param {object} simulation force simulation of d3
+     */
+    D3ForceLayout.prototype.overlapProcess = function (simulation, options) {
+        var nodeSize = options.nodeSize, nodeSpacing = options.nodeSpacing, collideStrength = options.collideStrength;
+        var nodeSizeFunc;
+        var nodeSpacingFunc;
+        if (isNumber(nodeSpacing)) {
+            nodeSpacingFunc = function () { return nodeSpacing; };
+        }
+        else if (isFunction(nodeSpacing)) {
+            nodeSpacingFunc = nodeSpacing;
+        }
+        else {
+            nodeSpacingFunc = function () { return 0; };
+        }
+        if (!nodeSize) {
+            nodeSizeFunc = function (d) {
+                if (d.size) {
+                    if (isArray(d.size)) {
+                        var res = d.size[0] > d.size[1] ? d.size[0] : d.size[1];
+                        return res / 2 + nodeSpacingFunc(d);
+                    }
+                    if (isObject(d.size)) {
+                        var res = d.size.width > d.size.height ? d.size.width : d.size.height;
+                        return res / 2 + nodeSpacingFunc(d);
+                    }
+                    return d.size / 2 + nodeSpacingFunc(d);
+                }
+                return 10 + nodeSpacingFunc(d);
+            };
+        }
+        else if (isFunction(nodeSize)) {
+            nodeSizeFunc = function (d) {
+                var size = nodeSize(d);
+                return size + nodeSpacingFunc(d);
+            };
+        }
+        else if (isArray(nodeSize)) {
+            var larger = nodeSize[0] > nodeSize[1] ? nodeSize[0] : nodeSize[1];
+            var radius_1 = larger / 2;
+            nodeSizeFunc = function (d) { return radius_1 + nodeSpacingFunc(d); };
+        }
+        else if (isNumber(nodeSize)) {
+            var radius_2 = nodeSize / 2;
+            nodeSizeFunc = function (d) { return radius_2 + nodeSpacingFunc(d); };
+        }
+        else {
+            nodeSizeFunc = function () { return 10; };
+        }
+        // forceCollide's parameter is a radius
+        simulation.force("collisionForce", d3Force.forceCollide(nodeSizeFunc).strength(collideStrength));
+    };
+    return D3ForceLayout;
+}());
+export { D3ForceLayout };
+/**
+ * Format the calculation nodes into output nodes.
+ * Since d3 reads properties in plain node data object which is not compact to the OutNode
+ * @param layoutNodes
+ * @returns
+ */
+var formatOutNodes = function (layoutNodes) {
+    return layoutNodes.map(function (node) {
+        var x = node.x, y = node.y, others = __rest(node, ["x", "y"]);
+        return __assign(__assign({}, others), { data: __assign(__assign({}, others.data), { x: x,
+                y: y }) });
+    });
+};
+/**
+ * d3 will modify `source` and `target` on edge object.
+ */
+var formatOutEdges = function (edges) {
+    return edges.map(function (edge) {
+        var source = edge.source, target = edge.target, rest = __rest(edge, ["source", "target"]);
+        return __assign(__assign({}, rest), { source: source.id, target: target.id });
+    });
+};
+//# sourceMappingURL=index.js.map