@antv/layout 1.2.10 → 1.2.11-beta.0

package/lib/force/forceNBody.js

@@ -0,0 +1,123 @@
+import { quadtree } from 'd3-quadtree';
+// @ts-ignore
+import { octree } from 'd3-octree';
+var theta2 = 0.81; // Barnes-Hut approximation threshold
+var epsilon = 0.1; // 为了防止出现除0的情况，加一个epsilon
+export function forceNBody(calcGraph, factor, coulombDisScale2, accMap, dimensions) {
+    if (dimensions === void 0) { dimensions = 2; }
+    var weightParam = factor / coulombDisScale2;
+    var calcNodes = calcGraph.getAllNodes();
+    var data = calcNodes.map(function (calcNode, i) {
+        var _a = calcNode.data, nodeStrength = _a.nodeStrength, x = _a.x, y = _a.y, z = _a.z, size = _a.size;
+        return {
+            x: x,
+            y: y,
+            z: z,
+            size: size,
+            index: i,
+            id: calcNode.id,
+            vx: 0,
+            vy: 0,
+            vz: 0,
+            weight: weightParam * nodeStrength,
+        };
+    });
+    var tree = (dimensions === 2
+        ? quadtree(data, function (d) { return d.x; }, function (d) { return d.y; })
+        : octree(data, function (d) { return d.x; }, function (d) { return d.y; }, function (d) { return d.z; })).visitAfter(accumulate); // init internal node
+    var nodeMap = new Map();
+    data.forEach(function (n) {
+        nodeMap.set(n.id, n);
+        // @ts-ignore
+        computeForce(n, tree, dimensions);
+    });
+    data.map(function (n, i) {
+        var _a = calcNodes[i], id = _a.id, data = _a.data;
+        var _b = data.mass, mass = _b === void 0 ? 1 : _b;
+        // 从 0 开始，= 初始化 + 加斥力
+        accMap[id] = {
+            x: n.vx / mass,
+            y: n.vy / mass,
+            z: n.vz / mass,
+        };
+    });
+    return accMap;
+}
+function accumulate(treeNode) {
+    var accWeight = 0;
+    var accX = 0;
+    var accY = 0;
+    var accZ = 0;
+    var accSize = 0;
+    var numChildren = treeNode.length;
+    if (numChildren) {
+        // internal node, accumulate 4 child quads
+        for (var i = 0; i < numChildren; i++) {
+            var q = treeNode[i];
+            if (q && q.weight) {
+                accWeight += q.weight;
+                accX += q.x * q.weight;
+                accY += q.y * q.weight;
+                accZ += q.z * q.weight;
+                accSize += q.size * q.weight;
+            }
+        }
+        treeNode.x = accX / accWeight;
+        treeNode.y = accY / accWeight;
+        treeNode.z = accZ / accWeight;
+        treeNode.size = accSize / accWeight;
+        treeNode.weight = accWeight;
+    }
+    else {
+        // leaf node
+        var q = treeNode;
+        treeNode.x = q.data.x;
+        treeNode.y = q.data.y;
+        treeNode.z = q.data.z;
+        treeNode.size = q.data.size;
+        treeNode.weight = q.data.weight;
+    }
+}
+var apply = function (treeNode, x1, arg1, arg2, arg3, node, dimensions) {
+    var _a;
+    if (((_a = treeNode.data) === null || _a === void 0 ? void 0 : _a.id) === node.id)
+        return;
+    var x2 = [arg1, arg2, arg3][dimensions - 1];
+    var dx = node.x - treeNode.x || epsilon;
+    var dy = node.y - treeNode.y || epsilon;
+    var dz = node.z - treeNode.z || epsilon;
+    var pos = [dx, dy, dz];
+    var width = x2 - x1;
+    var len2 = 0;
+    for (var i = 0; i < dimensions; i++) {
+        len2 += pos[i] * pos[i];
+    }
+    var len1 = Math.sqrt(len2);
+    var len3 = len1 * len2;
+    // far node, apply Barnes-Hut approximation
+    if (width * width * theta2 < len2) {
+        var param = treeNode.weight / len3;
+        node.vx += dx * param;
+        node.vy += dy * param;
+        node.vz += dz * param;
+        return true;
+    }
+    // near quad, compute force directly
+    if (treeNode.length)
+        return false; // internal node, visit children
+    // leaf node
+    if (treeNode.data !== node) {
+        var param = treeNode.data.weight / len3;
+        node.vx += dx * param;
+        node.vy += dy * param;
+        node.vz += dz * param;
+    }
+};
+// @ts-ignore
+function computeForce(node, tree, dimensions) {
+    // @ts-ignore
+    tree.visit(function (treeNode, x1, y1, x2, y2) {
+        return apply(treeNode, x1, y1, x2, y2, node, dimensions);
+    });
+}
+//# sourceMappingURL=forceNBody.js.map

package/lib/force/forceNBody.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"forceNBody.js","sourceRoot":"","sources":["../../src/force/forceNBody.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,aAAa;AACb,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AAInC,IAAM,MAAM,GAAG,IAAI,CAAC,CAAC,qCAAqC;AAC1D,IAAM,OAAO,GAAG,GAAG,CAAC,CAAC,yBAAyB;AAc9C,MAAM,UAAU,UAAU,CACxB,SAAoB,EACpB,MAAc,EACd,gBAAwB,EACxB,MAA+B,EAC/B,UAAsB;IAAtB,2BAAA,EAAA,cAAsB;IAEtB,IAAM,WAAW,GAAG,MAAM,GAAG,gBAAgB,CAAC;IAC9C,IAAM,SAAS,GAAG,SAAS,CAAC,WAAW,EAAE,CAAC;IAC1C,IAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,UAAC,QAAQ,EAAE,CAAC;QAC/B,IAAA,KAAkC,QAAQ,CAAC,IAAI,EAA7C,YAAY,kBAAA,EAAE,CAAC,OAAA,EAAE,CAAC,OAAA,EAAE,CAAC,OAAA,EAAE,IAAI,UAAkB,CAAC;QACtD,OAAO;YACL,CAAC,GAAA;YACD,CAAC,GAAA;YACD,CAAC,GAAA;YACD,IAAI,MAAA;YACJ,KAAK,EAAE,CAAC;YACR,EAAE,EAAE,QAAQ,CAAC,EAAE;YACf,EAAE,EAAE,CAAC;YACL,EAAE,EAAE,CAAC;YACL,EAAE,EAAE,CAAC;YACL,MAAM,EAAE,WAAW,GAAG,YAAY;SACnC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,IAAM,IAAI,GAAG,CACX,UAAU,KAAK,CAAC;QACd,CAAC,CAAC,QAAQ,CACN,IAAI,EACJ,UAAC,CAAM,IAAK,OAAA,CAAC,CAAC,CAAC,EAAH,CAAG,EACf,UAAC,CAAM,IAAK,OAAA,CAAC,CAAC,CAAC,EAAH,CAAG,CAChB;QACH,CAAC,CAAC,MAAM,CACJ,IAAI,EACJ,UAAC,CAAM,IAAK,OAAA,CAAC,CAAC,CAAC,EAAH,CAAG,EACf,UAAC,CAAM,IAAK,OAAA,CAAC,CAAC,CAAC,EAAH,CAAG,EACf,UAAC,CAAM,IAAK,OAAA,CAAC,CAAC,CAAC,EAAH,CAAG,CAChB,CACN,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,qBAAqB;IAE/C,IAAM,OAAO,GAAG,IAAI,GAAG,EAAE,CAAC;IAC1B,IAAI,CAAC,OAAO,CAAC,UAAC,CAAC;QACb,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QACrB,aAAa;QACb,YAAY,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,GAAG,CAAC,UAAC,CAAC,EAAE,CAAC;QACN,IAAA,KAAe,SAAS,CAAC,CAAC,CAAC,EAAzB,EAAE,QAAA,EAAE,IAAI,UAAiB,CAAC;QAC1B,IAAA,KAAa,IAAI,KAAT,EAAR,IAAI,mBAAG,CAAC,KAAA,CAAU;QAC1B,qBAAqB;QACrB,MAAM,CAAC,EAAE,CAAC,GAAG;YACX,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI;YACd,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI;YACd,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI;SACf,CAAC;IACJ,CAAC,CAAC,CAAC;IACH,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,UAAU,CAAC,QAAa;IAC/B,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,IAAM,WAAW,GAAG,QAAQ,CAAC,MAAM,CAAC;IAEpC,IAAI,WAAW,EAAE;QACf,0CAA0C;QAC1C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,EAAE,CAAC,EAAE,EAAE;YACpC,IAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;YACtB,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE;gBACjB,SAAS,IAAI,CAAC,CAAC,MAAM,CAAC;gBACtB,IAAI,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC;gBACvB,IAAI,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC;gBACvB,IAAI,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC;gBACvB,OAAO,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,MAAM,CAAC;aAC9B;SACF;QACD,QAAQ,CAAC,CAAC,GAAG,IAAI,GAAG,SAAS,CAAC;QAC9B,QAAQ,CAAC,CAAC,GAAG,IAAI,GAAG,SAAS,CAAC;QAC9B,QAAQ,CAAC,CAAC,GAAG,IAAI,GAAG,SAAS,CAAC;QAC9B,QAAQ,CAAC,IAAI,GAAG,OAAO,GAAG,SAAS,CAAC;QACpC,QAAQ,CAAC,MAAM,GAAG,SAAS,CAAC;KAC7B;SAAM;QACL,YAAY;QACZ,IAAM,CAAC,GAAG,QAAQ,CAAC;QACnB,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACtB,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACtB,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACtB,QAAQ,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QAC5B,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;KACjC;AACH,CAAC;AAED,IAAM,KAAK,GAAG,UACZ,QAAa,EACb,EAAU,EACV,IAAY,EACZ,IAAY,EACZ,IAAY,EACZ,IAAkB,EAClB,UAAkB;;IAElB,IAAI,CAAA,MAAA,QAAQ,CAAC,IAAI,0CAAE,EAAE,MAAK,IAAI,CAAC,EAAE;QAAE,OAAO;IAC1C,IAAM,EAAE,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;IAE9C,IAAI,EAAE,GAAG,IAAI,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,IAAI,OAAO,CAAC;IACxC,IAAI,EAAE,GAAG,IAAI,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,IAAI,OAAO,CAAC;IACxC,IAAI,EAAE,GAAG,IAAI,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,IAAI,OAAO,CAAC;IACxC,IAAM,GAAG,GAAG,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;IACzB,IAAM,KAAK,GAAG,EAAE,GAAG,EAAE,CAAC;IACtB,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE;QACnC,IAAI,IAAI,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;KACzB;IACD,IAAI,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3B,IAAI,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;IAEvB,2CAA2C;IAC3C,IAAI,KAAK,GAAG,KAAK,GAAG,MAAM,GAAG,IAAI,EAAE;QACjC,IAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,GAAG,IAAI,CAAC;QACrC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC;QACtB,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC;QACtB,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC;QACtB,OAAO,IAAI,CAAC;KACb;IACD,oCAAoC;IACpC,IAAI,QAAQ,CAAC,MAAM;QAAE,OAAO,KAAK,CAAC,CAAC,gCAAgC;IAEnE,YAAY;IAEZ,IAAI,QAAQ,CAAC,IAAI,KAAK,IAAI,EAAE;QAC1B,IAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QAC1C,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC;QACtB,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC;QACtB,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC;KACvB;AACH,CAAC,CAAC;AAEF,aAAa;AACb,SAAS,YAAY,CAAC,IAAkB,EAAE,IAAS,EAAE,UAAkB;IACrE,aAAa;IACb,IAAI,CAAC,KAAK,CAAC,UAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE;QAClC,OAAA,KAAK,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,UAAU,CAAC;IAAjD,CAAiD,CAClD,CAAC;AACJ,CAAC"}