852 lines
23 KiB
JavaScript
Executable file
852 lines
23 KiB
JavaScript
Executable file
// https://github.com/vasturiano/d3-force-3d v3.0.5
|
|
(function (global, factory) {
|
|
typeof exports === 'object' && typeof module !== 'undefined' ? factory(exports, require('d3-binarytree'), require('d3-quadtree'), require('d3-octree'), require('d3-dispatch'), require('d3-timer')) :
|
|
typeof define === 'function' && define.amd ? define(['exports', 'd3-binarytree', 'd3-quadtree', 'd3-octree', 'd3-dispatch', 'd3-timer'], factory) :
|
|
(global = typeof globalThis !== 'undefined' ? globalThis : global || self, factory(global.d3 = global.d3 || {}, global.d3, global.d3, global.d3, global.d3, global.d3));
|
|
})(this, (function (exports, d3Binarytree, d3Quadtree, d3Octree, d3Dispatch, d3Timer) { 'use strict';
|
|
|
|
function center(x, y, z) {
|
|
var nodes, strength = 1;
|
|
|
|
if (x == null) x = 0;
|
|
if (y == null) y = 0;
|
|
if (z == null) z = 0;
|
|
|
|
function force() {
|
|
var i,
|
|
n = nodes.length,
|
|
node,
|
|
sx = 0,
|
|
sy = 0,
|
|
sz = 0;
|
|
|
|
for (i = 0; i < n; ++i) {
|
|
node = nodes[i], sx += node.x || 0, sy += node.y || 0, sz += node.z || 0;
|
|
}
|
|
|
|
for (sx = (sx / n - x) * strength, sy = (sy / n - y) * strength, sz = (sz / n - z) * strength, i = 0; i < n; ++i) {
|
|
node = nodes[i];
|
|
if (sx) { node.x -= sx; }
|
|
if (sy) { node.y -= sy; }
|
|
if (sz) { node.z -= sz; }
|
|
}
|
|
}
|
|
|
|
force.initialize = function(_) {
|
|
nodes = _;
|
|
};
|
|
|
|
force.x = function(_) {
|
|
return arguments.length ? (x = +_, force) : x;
|
|
};
|
|
|
|
force.y = function(_) {
|
|
return arguments.length ? (y = +_, force) : y;
|
|
};
|
|
|
|
force.z = function(_) {
|
|
return arguments.length ? (z = +_, force) : z;
|
|
};
|
|
|
|
force.strength = function(_) {
|
|
return arguments.length ? (strength = +_, force) : strength;
|
|
};
|
|
|
|
return force;
|
|
}
|
|
|
|
function constant(x) {
|
|
return function() {
|
|
return x;
|
|
};
|
|
}
|
|
|
|
function jiggle(random) {
|
|
return (random() - 0.5) * 1e-6;
|
|
}
|
|
|
|
function x$2(d) {
|
|
return d.x + d.vx;
|
|
}
|
|
|
|
function y$2(d) {
|
|
return d.y + d.vy;
|
|
}
|
|
|
|
function z$2(d) {
|
|
return d.z + d.vz;
|
|
}
|
|
|
|
function collide(radius) {
|
|
var nodes,
|
|
nDim,
|
|
radii,
|
|
random,
|
|
strength = 1,
|
|
iterations = 1;
|
|
|
|
if (typeof radius !== "function") radius = constant(radius == null ? 1 : +radius);
|
|
|
|
function force() {
|
|
var i, n = nodes.length,
|
|
tree,
|
|
node,
|
|
xi,
|
|
yi,
|
|
zi,
|
|
ri,
|
|
ri2;
|
|
|
|
for (var k = 0; k < iterations; ++k) {
|
|
tree =
|
|
(nDim === 1 ? d3Binarytree.binarytree(nodes, x$2)
|
|
:(nDim === 2 ? d3Quadtree.quadtree(nodes, x$2, y$2)
|
|
:(nDim === 3 ? d3Octree.octree(nodes, x$2, y$2, z$2)
|
|
:null
|
|
))).visitAfter(prepare);
|
|
|
|
for (i = 0; i < n; ++i) {
|
|
node = nodes[i];
|
|
ri = radii[node.index], ri2 = ri * ri;
|
|
xi = node.x + node.vx;
|
|
if (nDim > 1) { yi = node.y + node.vy; }
|
|
if (nDim > 2) { zi = node.z + node.vz; }
|
|
tree.visit(apply);
|
|
}
|
|
}
|
|
|
|
function apply(treeNode, arg1, arg2, arg3, arg4, arg5, arg6) {
|
|
var args = [arg1, arg2, arg3, arg4, arg5, arg6];
|
|
var x0 = args[0],
|
|
y0 = args[1],
|
|
z0 = args[2],
|
|
x1 = args[nDim],
|
|
y1 = args[nDim+1],
|
|
z1 = args[nDim+2];
|
|
|
|
var data = treeNode.data, rj = treeNode.r, r = ri + rj;
|
|
if (data) {
|
|
if (data.index > node.index) {
|
|
var x = xi - data.x - data.vx,
|
|
y = (nDim > 1 ? yi - data.y - data.vy : 0),
|
|
z = (nDim > 2 ? zi - data.z - data.vz : 0),
|
|
l = x * x + y * y + z * z;
|
|
if (l < r * r) {
|
|
if (x === 0) x = jiggle(random), l += x * x;
|
|
if (nDim > 1 && y === 0) y = jiggle(random), l += y * y;
|
|
if (nDim > 2 && z === 0) z = jiggle(random), l += z * z;
|
|
l = (r - (l = Math.sqrt(l))) / l * strength;
|
|
|
|
node.vx += (x *= l) * (r = (rj *= rj) / (ri2 + rj));
|
|
if (nDim > 1) { node.vy += (y *= l) * r; }
|
|
if (nDim > 2) { node.vz += (z *= l) * r; }
|
|
|
|
data.vx -= x * (r = 1 - r);
|
|
if (nDim > 1) { data.vy -= y * r; }
|
|
if (nDim > 2) { data.vz -= z * r; }
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
return x0 > xi + r || x1 < xi - r
|
|
|| (nDim > 1 && (y0 > yi + r || y1 < yi - r))
|
|
|| (nDim > 2 && (z0 > zi + r || z1 < zi - r));
|
|
}
|
|
}
|
|
|
|
function prepare(treeNode) {
|
|
if (treeNode.data) return treeNode.r = radii[treeNode.data.index];
|
|
for (var i = treeNode.r = 0; i < Math.pow(2, nDim); ++i) {
|
|
if (treeNode[i] && treeNode[i].r > treeNode.r) {
|
|
treeNode.r = treeNode[i].r;
|
|
}
|
|
}
|
|
}
|
|
|
|
function initialize() {
|
|
if (!nodes) return;
|
|
var i, n = nodes.length, node;
|
|
radii = new Array(n);
|
|
for (i = 0; i < n; ++i) node = nodes[i], radii[node.index] = +radius(node, i, nodes);
|
|
}
|
|
|
|
force.initialize = function(_nodes, ...args) {
|
|
nodes = _nodes;
|
|
random = args.find(arg => typeof arg === 'function') || Math.random;
|
|
nDim = args.find(arg => [1, 2, 3].includes(arg)) || 2;
|
|
initialize();
|
|
};
|
|
|
|
force.iterations = function(_) {
|
|
return arguments.length ? (iterations = +_, force) : iterations;
|
|
};
|
|
|
|
force.strength = function(_) {
|
|
return arguments.length ? (strength = +_, force) : strength;
|
|
};
|
|
|
|
force.radius = function(_) {
|
|
return arguments.length ? (radius = typeof _ === "function" ? _ : constant(+_), initialize(), force) : radius;
|
|
};
|
|
|
|
return force;
|
|
}
|
|
|
|
function index(d) {
|
|
return d.index;
|
|
}
|
|
|
|
function find(nodeById, nodeId) {
|
|
var node = nodeById.get(nodeId);
|
|
if (!node) throw new Error("node not found: " + nodeId);
|
|
return node;
|
|
}
|
|
|
|
function link(links) {
|
|
var id = index,
|
|
strength = defaultStrength,
|
|
strengths,
|
|
distance = constant(30),
|
|
distances,
|
|
nodes,
|
|
nDim,
|
|
count,
|
|
bias,
|
|
random,
|
|
iterations = 1;
|
|
|
|
if (links == null) links = [];
|
|
|
|
function defaultStrength(link) {
|
|
return 1 / Math.min(count[link.source.index], count[link.target.index]);
|
|
}
|
|
|
|
function force(alpha) {
|
|
for (var k = 0, n = links.length; k < iterations; ++k) {
|
|
for (var i = 0, link, source, target, x = 0, y = 0, z = 0, l, b; i < n; ++i) {
|
|
link = links[i], source = link.source, target = link.target;
|
|
x = target.x + target.vx - source.x - source.vx || jiggle(random);
|
|
if (nDim > 1) { y = target.y + target.vy - source.y - source.vy || jiggle(random); }
|
|
if (nDim > 2) { z = target.z + target.vz - source.z - source.vz || jiggle(random); }
|
|
l = Math.sqrt(x * x + y * y + z * z);
|
|
l = (l - distances[i]) / l * alpha * strengths[i];
|
|
x *= l, y *= l, z *= l;
|
|
|
|
target.vx -= x * (b = bias[i]);
|
|
if (nDim > 1) { target.vy -= y * b; }
|
|
if (nDim > 2) { target.vz -= z * b; }
|
|
|
|
source.vx += x * (b = 1 - b);
|
|
if (nDim > 1) { source.vy += y * b; }
|
|
if (nDim > 2) { source.vz += z * b; }
|
|
}
|
|
}
|
|
}
|
|
|
|
function initialize() {
|
|
if (!nodes) return;
|
|
|
|
var i,
|
|
n = nodes.length,
|
|
m = links.length,
|
|
nodeById = new Map(nodes.map((d, i) => [id(d, i, nodes), d])),
|
|
link;
|
|
|
|
for (i = 0, count = new Array(n); i < m; ++i) {
|
|
link = links[i], link.index = i;
|
|
if (typeof link.source !== "object") link.source = find(nodeById, link.source);
|
|
if (typeof link.target !== "object") link.target = find(nodeById, link.target);
|
|
count[link.source.index] = (count[link.source.index] || 0) + 1;
|
|
count[link.target.index] = (count[link.target.index] || 0) + 1;
|
|
}
|
|
|
|
for (i = 0, bias = new Array(m); i < m; ++i) {
|
|
link = links[i], bias[i] = count[link.source.index] / (count[link.source.index] + count[link.target.index]);
|
|
}
|
|
|
|
strengths = new Array(m), initializeStrength();
|
|
distances = new Array(m), initializeDistance();
|
|
}
|
|
|
|
function initializeStrength() {
|
|
if (!nodes) return;
|
|
|
|
for (var i = 0, n = links.length; i < n; ++i) {
|
|
strengths[i] = +strength(links[i], i, links);
|
|
}
|
|
}
|
|
|
|
function initializeDistance() {
|
|
if (!nodes) return;
|
|
|
|
for (var i = 0, n = links.length; i < n; ++i) {
|
|
distances[i] = +distance(links[i], i, links);
|
|
}
|
|
}
|
|
|
|
force.initialize = function(_nodes, ...args) {
|
|
nodes = _nodes;
|
|
random = args.find(arg => typeof arg === 'function') || Math.random;
|
|
nDim = args.find(arg => [1, 2, 3].includes(arg)) || 2;
|
|
initialize();
|
|
};
|
|
|
|
force.links = function(_) {
|
|
return arguments.length ? (links = _, initialize(), force) : links;
|
|
};
|
|
|
|
force.id = function(_) {
|
|
return arguments.length ? (id = _, force) : id;
|
|
};
|
|
|
|
force.iterations = function(_) {
|
|
return arguments.length ? (iterations = +_, force) : iterations;
|
|
};
|
|
|
|
force.strength = function(_) {
|
|
return arguments.length ? (strength = typeof _ === "function" ? _ : constant(+_), initializeStrength(), force) : strength;
|
|
};
|
|
|
|
force.distance = function(_) {
|
|
return arguments.length ? (distance = typeof _ === "function" ? _ : constant(+_), initializeDistance(), force) : distance;
|
|
};
|
|
|
|
return force;
|
|
}
|
|
|
|
// https://en.wikipedia.org/wiki/Linear_congruential_generator#Parameters_in_common_use
|
|
const a = 1664525;
|
|
const c = 1013904223;
|
|
const m = 4294967296; // 2^32
|
|
|
|
function lcg() {
|
|
let s = 1;
|
|
return () => (s = (a * s + c) % m) / m;
|
|
}
|
|
|
|
var MAX_DIMENSIONS = 3;
|
|
|
|
function x$1(d) {
|
|
return d.x;
|
|
}
|
|
|
|
function y$1(d) {
|
|
return d.y;
|
|
}
|
|
|
|
function z$1(d) {
|
|
return d.z;
|
|
}
|
|
|
|
var initialRadius = 10,
|
|
initialAngleRoll = Math.PI * (3 - Math.sqrt(5)), // Golden ratio angle
|
|
initialAngleYaw = Math.PI * 20 / (9 + Math.sqrt(221)); // Markov irrational number
|
|
|
|
function simulation(nodes, numDimensions) {
|
|
numDimensions = numDimensions || 2;
|
|
|
|
var nDim = Math.min(MAX_DIMENSIONS, Math.max(1, Math.round(numDimensions))),
|
|
simulation,
|
|
alpha = 1,
|
|
alphaMin = 0.001,
|
|
alphaDecay = 1 - Math.pow(alphaMin, 1 / 300),
|
|
alphaTarget = 0,
|
|
velocityDecay = 0.6,
|
|
forces = new Map(),
|
|
stepper = d3Timer.timer(step),
|
|
event = d3Dispatch.dispatch("tick", "end"),
|
|
random = lcg();
|
|
|
|
if (nodes == null) nodes = [];
|
|
|
|
function step() {
|
|
tick();
|
|
event.call("tick", simulation);
|
|
if (alpha < alphaMin) {
|
|
stepper.stop();
|
|
event.call("end", simulation);
|
|
}
|
|
}
|
|
|
|
function tick(iterations) {
|
|
var i, n = nodes.length, node;
|
|
|
|
if (iterations === undefined) iterations = 1;
|
|
|
|
for (var k = 0; k < iterations; ++k) {
|
|
alpha += (alphaTarget - alpha) * alphaDecay;
|
|
|
|
forces.forEach(function (force) {
|
|
force(alpha);
|
|
});
|
|
|
|
for (i = 0; i < n; ++i) {
|
|
node = nodes[i];
|
|
if (node.fx == null) node.x += node.vx *= velocityDecay;
|
|
else node.x = node.fx, node.vx = 0;
|
|
if (nDim > 1) {
|
|
if (node.fy == null) node.y += node.vy *= velocityDecay;
|
|
else node.y = node.fy, node.vy = 0;
|
|
}
|
|
if (nDim > 2) {
|
|
if (node.fz == null) node.z += node.vz *= velocityDecay;
|
|
else node.z = node.fz, node.vz = 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
return simulation;
|
|
}
|
|
|
|
function initializeNodes() {
|
|
for (var i = 0, n = nodes.length, node; i < n; ++i) {
|
|
node = nodes[i], node.index = i;
|
|
if (node.fx != null) node.x = node.fx;
|
|
if (node.fy != null) node.y = node.fy;
|
|
if (node.fz != null) node.z = node.fz;
|
|
if (isNaN(node.x) || (nDim > 1 && isNaN(node.y)) || (nDim > 2 && isNaN(node.z))) {
|
|
var radius = initialRadius * (nDim > 2 ? Math.cbrt(0.5 + i) : (nDim > 1 ? Math.sqrt(0.5 + i) : i)),
|
|
rollAngle = i * initialAngleRoll,
|
|
yawAngle = i * initialAngleYaw;
|
|
|
|
if (nDim === 1) {
|
|
node.x = radius;
|
|
} else if (nDim === 2) {
|
|
node.x = radius * Math.cos(rollAngle);
|
|
node.y = radius * Math.sin(rollAngle);
|
|
} else { // 3 dimensions: use spherical distribution along 2 irrational number angles
|
|
node.x = radius * Math.sin(rollAngle) * Math.cos(yawAngle);
|
|
node.y = radius * Math.cos(rollAngle);
|
|
node.z = radius * Math.sin(rollAngle) * Math.sin(yawAngle);
|
|
}
|
|
}
|
|
if (isNaN(node.vx) || (nDim > 1 && isNaN(node.vy)) || (nDim > 2 && isNaN(node.vz))) {
|
|
node.vx = 0;
|
|
if (nDim > 1) { node.vy = 0; }
|
|
if (nDim > 2) { node.vz = 0; }
|
|
}
|
|
}
|
|
}
|
|
|
|
function initializeForce(force) {
|
|
if (force.initialize) force.initialize(nodes, random, nDim);
|
|
return force;
|
|
}
|
|
|
|
initializeNodes();
|
|
|
|
return simulation = {
|
|
tick: tick,
|
|
|
|
restart: function() {
|
|
return stepper.restart(step), simulation;
|
|
},
|
|
|
|
stop: function() {
|
|
return stepper.stop(), simulation;
|
|
},
|
|
|
|
numDimensions: function(_) {
|
|
return arguments.length
|
|
? (nDim = Math.min(MAX_DIMENSIONS, Math.max(1, Math.round(_))), forces.forEach(initializeForce), simulation)
|
|
: nDim;
|
|
},
|
|
|
|
nodes: function(_) {
|
|
return arguments.length ? (nodes = _, initializeNodes(), forces.forEach(initializeForce), simulation) : nodes;
|
|
},
|
|
|
|
alpha: function(_) {
|
|
return arguments.length ? (alpha = +_, simulation) : alpha;
|
|
},
|
|
|
|
alphaMin: function(_) {
|
|
return arguments.length ? (alphaMin = +_, simulation) : alphaMin;
|
|
},
|
|
|
|
alphaDecay: function(_) {
|
|
return arguments.length ? (alphaDecay = +_, simulation) : +alphaDecay;
|
|
},
|
|
|
|
alphaTarget: function(_) {
|
|
return arguments.length ? (alphaTarget = +_, simulation) : alphaTarget;
|
|
},
|
|
|
|
velocityDecay: function(_) {
|
|
return arguments.length ? (velocityDecay = 1 - _, simulation) : 1 - velocityDecay;
|
|
},
|
|
|
|
randomSource: function(_) {
|
|
return arguments.length ? (random = _, forces.forEach(initializeForce), simulation) : random;
|
|
},
|
|
|
|
force: function(name, _) {
|
|
return arguments.length > 1 ? ((_ == null ? forces.delete(name) : forces.set(name, initializeForce(_))), simulation) : forces.get(name);
|
|
},
|
|
|
|
find: function() {
|
|
var args = Array.prototype.slice.call(arguments);
|
|
var x = args.shift() || 0,
|
|
y = (nDim > 1 ? args.shift() : null) || 0,
|
|
z = (nDim > 2 ? args.shift() : null) || 0,
|
|
radius = args.shift() || Infinity;
|
|
|
|
var i = 0,
|
|
n = nodes.length,
|
|
dx,
|
|
dy,
|
|
dz,
|
|
d2,
|
|
node,
|
|
closest;
|
|
|
|
radius *= radius;
|
|
|
|
for (i = 0; i < n; ++i) {
|
|
node = nodes[i];
|
|
dx = x - node.x;
|
|
dy = y - (node.y || 0);
|
|
dz = z - (node.z ||0);
|
|
d2 = dx * dx + dy * dy + dz * dz;
|
|
if (d2 < radius) closest = node, radius = d2;
|
|
}
|
|
|
|
return closest;
|
|
},
|
|
|
|
on: function(name, _) {
|
|
return arguments.length > 1 ? (event.on(name, _), simulation) : event.on(name);
|
|
}
|
|
};
|
|
}
|
|
|
|
function manyBody() {
|
|
var nodes,
|
|
nDim,
|
|
node,
|
|
random,
|
|
alpha,
|
|
strength = constant(-30),
|
|
strengths,
|
|
distanceMin2 = 1,
|
|
distanceMax2 = Infinity,
|
|
theta2 = 0.81;
|
|
|
|
function force(_) {
|
|
var i,
|
|
n = nodes.length,
|
|
tree =
|
|
(nDim === 1 ? d3Binarytree.binarytree(nodes, x$1)
|
|
:(nDim === 2 ? d3Quadtree.quadtree(nodes, x$1, y$1)
|
|
:(nDim === 3 ? d3Octree.octree(nodes, x$1, y$1, z$1)
|
|
:null
|
|
))).visitAfter(accumulate);
|
|
|
|
for (alpha = _, i = 0; i < n; ++i) node = nodes[i], tree.visit(apply);
|
|
}
|
|
|
|
function initialize() {
|
|
if (!nodes) return;
|
|
var i, n = nodes.length, node;
|
|
strengths = new Array(n);
|
|
for (i = 0; i < n; ++i) node = nodes[i], strengths[node.index] = +strength(node, i, nodes);
|
|
}
|
|
|
|
function accumulate(treeNode) {
|
|
var strength = 0, q, c, weight = 0, x, y, z, i;
|
|
var numChildren = treeNode.length;
|
|
|
|
// For internal nodes, accumulate forces from children.
|
|
if (numChildren) {
|
|
for (x = y = z = i = 0; i < numChildren; ++i) {
|
|
if ((q = treeNode[i]) && (c = Math.abs(q.value))) {
|
|
strength += q.value, weight += c, x += c * (q.x || 0), y += c * (q.y || 0), z += c * (q.z || 0);
|
|
}
|
|
}
|
|
strength *= Math.sqrt(4 / numChildren); // scale accumulated strength according to number of dimensions
|
|
|
|
treeNode.x = x / weight;
|
|
if (nDim > 1) { treeNode.y = y / weight; }
|
|
if (nDim > 2) { treeNode.z = z / weight; }
|
|
}
|
|
|
|
// For leaf nodes, accumulate forces from coincident nodes.
|
|
else {
|
|
q = treeNode;
|
|
q.x = q.data.x;
|
|
if (nDim > 1) { q.y = q.data.y; }
|
|
if (nDim > 2) { q.z = q.data.z; }
|
|
do strength += strengths[q.data.index];
|
|
while (q = q.next);
|
|
}
|
|
|
|
treeNode.value = strength;
|
|
}
|
|
|
|
function apply(treeNode, x1, arg1, arg2, arg3) {
|
|
if (!treeNode.value) return true;
|
|
var x2 = [arg1, arg2, arg3][nDim-1];
|
|
|
|
var x = treeNode.x - node.x,
|
|
y = (nDim > 1 ? treeNode.y - node.y : 0),
|
|
z = (nDim > 2 ? treeNode.z - node.z : 0),
|
|
w = x2 - x1,
|
|
l = x * x + y * y + z * z;
|
|
|
|
// Apply the Barnes-Hut approximation if possible.
|
|
// Limit forces for very close nodes; randomize direction if coincident.
|
|
if (w * w / theta2 < l) {
|
|
if (l < distanceMax2) {
|
|
if (x === 0) x = jiggle(random), l += x * x;
|
|
if (nDim > 1 && y === 0) y = jiggle(random), l += y * y;
|
|
if (nDim > 2 && z === 0) z = jiggle(random), l += z * z;
|
|
if (l < distanceMin2) l = Math.sqrt(distanceMin2 * l);
|
|
node.vx += x * treeNode.value * alpha / l;
|
|
if (nDim > 1) { node.vy += y * treeNode.value * alpha / l; }
|
|
if (nDim > 2) { node.vz += z * treeNode.value * alpha / l; }
|
|
}
|
|
return true;
|
|
}
|
|
|
|
// Otherwise, process points directly.
|
|
else if (treeNode.length || l >= distanceMax2) return;
|
|
|
|
// Limit forces for very close nodes; randomize direction if coincident.
|
|
if (treeNode.data !== node || treeNode.next) {
|
|
if (x === 0) x = jiggle(random), l += x * x;
|
|
if (nDim > 1 && y === 0) y = jiggle(random), l += y * y;
|
|
if (nDim > 2 && z === 0) z = jiggle(random), l += z * z;
|
|
if (l < distanceMin2) l = Math.sqrt(distanceMin2 * l);
|
|
}
|
|
|
|
do if (treeNode.data !== node) {
|
|
w = strengths[treeNode.data.index] * alpha / l;
|
|
node.vx += x * w;
|
|
if (nDim > 1) { node.vy += y * w; }
|
|
if (nDim > 2) { node.vz += z * w; }
|
|
} while (treeNode = treeNode.next);
|
|
}
|
|
|
|
force.initialize = function(_nodes, ...args) {
|
|
nodes = _nodes;
|
|
random = args.find(arg => typeof arg === 'function') || Math.random;
|
|
nDim = args.find(arg => [1, 2, 3].includes(arg)) || 2;
|
|
initialize();
|
|
};
|
|
|
|
force.strength = function(_) {
|
|
return arguments.length ? (strength = typeof _ === "function" ? _ : constant(+_), initialize(), force) : strength;
|
|
};
|
|
|
|
force.distanceMin = function(_) {
|
|
return arguments.length ? (distanceMin2 = _ * _, force) : Math.sqrt(distanceMin2);
|
|
};
|
|
|
|
force.distanceMax = function(_) {
|
|
return arguments.length ? (distanceMax2 = _ * _, force) : Math.sqrt(distanceMax2);
|
|
};
|
|
|
|
force.theta = function(_) {
|
|
return arguments.length ? (theta2 = _ * _, force) : Math.sqrt(theta2);
|
|
};
|
|
|
|
return force;
|
|
}
|
|
|
|
function radial(radius, x, y, z) {
|
|
var nodes,
|
|
nDim,
|
|
strength = constant(0.1),
|
|
strengths,
|
|
radiuses;
|
|
|
|
if (typeof radius !== "function") radius = constant(+radius);
|
|
if (x == null) x = 0;
|
|
if (y == null) y = 0;
|
|
if (z == null) z = 0;
|
|
|
|
function force(alpha) {
|
|
for (var i = 0, n = nodes.length; i < n; ++i) {
|
|
var node = nodes[i],
|
|
dx = node.x - x || 1e-6,
|
|
dy = (node.y || 0) - y || 1e-6,
|
|
dz = (node.z || 0) - z || 1e-6,
|
|
r = Math.sqrt(dx * dx + dy * dy + dz * dz),
|
|
k = (radiuses[i] - r) * strengths[i] * alpha / r;
|
|
node.vx += dx * k;
|
|
if (nDim>1) { node.vy += dy * k; }
|
|
if (nDim>2) { node.vz += dz * k; }
|
|
}
|
|
}
|
|
|
|
function initialize() {
|
|
if (!nodes) return;
|
|
var i, n = nodes.length;
|
|
strengths = new Array(n);
|
|
radiuses = new Array(n);
|
|
for (i = 0; i < n; ++i) {
|
|
radiuses[i] = +radius(nodes[i], i, nodes);
|
|
strengths[i] = isNaN(radiuses[i]) ? 0 : +strength(nodes[i], i, nodes);
|
|
}
|
|
}
|
|
|
|
force.initialize = function(initNodes, ...args) {
|
|
nodes = initNodes;
|
|
nDim = args.find(arg => [1, 2, 3].includes(arg)) || 2;
|
|
initialize();
|
|
};
|
|
|
|
force.strength = function(_) {
|
|
return arguments.length ? (strength = typeof _ === "function" ? _ : constant(+_), initialize(), force) : strength;
|
|
};
|
|
|
|
force.radius = function(_) {
|
|
return arguments.length ? (radius = typeof _ === "function" ? _ : constant(+_), initialize(), force) : radius;
|
|
};
|
|
|
|
force.x = function(_) {
|
|
return arguments.length ? (x = +_, force) : x;
|
|
};
|
|
|
|
force.y = function(_) {
|
|
return arguments.length ? (y = +_, force) : y;
|
|
};
|
|
|
|
force.z = function(_) {
|
|
return arguments.length ? (z = +_, force) : z;
|
|
};
|
|
|
|
return force;
|
|
}
|
|
|
|
function x(x) {
|
|
var strength = constant(0.1),
|
|
nodes,
|
|
strengths,
|
|
xz;
|
|
|
|
if (typeof x !== "function") x = constant(x == null ? 0 : +x);
|
|
|
|
function force(alpha) {
|
|
for (var i = 0, n = nodes.length, node; i < n; ++i) {
|
|
node = nodes[i], node.vx += (xz[i] - node.x) * strengths[i] * alpha;
|
|
}
|
|
}
|
|
|
|
function initialize() {
|
|
if (!nodes) return;
|
|
var i, n = nodes.length;
|
|
strengths = new Array(n);
|
|
xz = new Array(n);
|
|
for (i = 0; i < n; ++i) {
|
|
strengths[i] = isNaN(xz[i] = +x(nodes[i], i, nodes)) ? 0 : +strength(nodes[i], i, nodes);
|
|
}
|
|
}
|
|
|
|
force.initialize = function(_) {
|
|
nodes = _;
|
|
initialize();
|
|
};
|
|
|
|
force.strength = function(_) {
|
|
return arguments.length ? (strength = typeof _ === "function" ? _ : constant(+_), initialize(), force) : strength;
|
|
};
|
|
|
|
force.x = function(_) {
|
|
return arguments.length ? (x = typeof _ === "function" ? _ : constant(+_), initialize(), force) : x;
|
|
};
|
|
|
|
return force;
|
|
}
|
|
|
|
function y(y) {
|
|
var strength = constant(0.1),
|
|
nodes,
|
|
strengths,
|
|
yz;
|
|
|
|
if (typeof y !== "function") y = constant(y == null ? 0 : +y);
|
|
|
|
function force(alpha) {
|
|
for (var i = 0, n = nodes.length, node; i < n; ++i) {
|
|
node = nodes[i], node.vy += (yz[i] - node.y) * strengths[i] * alpha;
|
|
}
|
|
}
|
|
|
|
function initialize() {
|
|
if (!nodes) return;
|
|
var i, n = nodes.length;
|
|
strengths = new Array(n);
|
|
yz = new Array(n);
|
|
for (i = 0; i < n; ++i) {
|
|
strengths[i] = isNaN(yz[i] = +y(nodes[i], i, nodes)) ? 0 : +strength(nodes[i], i, nodes);
|
|
}
|
|
}
|
|
|
|
force.initialize = function(_) {
|
|
nodes = _;
|
|
initialize();
|
|
};
|
|
|
|
force.strength = function(_) {
|
|
return arguments.length ? (strength = typeof _ === "function" ? _ : constant(+_), initialize(), force) : strength;
|
|
};
|
|
|
|
force.y = function(_) {
|
|
return arguments.length ? (y = typeof _ === "function" ? _ : constant(+_), initialize(), force) : y;
|
|
};
|
|
|
|
return force;
|
|
}
|
|
|
|
function z(z) {
|
|
var strength = constant(0.1),
|
|
nodes,
|
|
strengths,
|
|
zz;
|
|
|
|
if (typeof z !== "function") z = constant(z == null ? 0 : +z);
|
|
|
|
function force(alpha) {
|
|
for (var i = 0, n = nodes.length, node; i < n; ++i) {
|
|
node = nodes[i], node.vz += (zz[i] - node.z) * strengths[i] * alpha;
|
|
}
|
|
}
|
|
|
|
function initialize() {
|
|
if (!nodes) return;
|
|
var i, n = nodes.length;
|
|
strengths = new Array(n);
|
|
zz = new Array(n);
|
|
for (i = 0; i < n; ++i) {
|
|
strengths[i] = isNaN(zz[i] = +z(nodes[i], i, nodes)) ? 0 : +strength(nodes[i], i, nodes);
|
|
}
|
|
}
|
|
|
|
force.initialize = function(_) {
|
|
nodes = _;
|
|
initialize();
|
|
};
|
|
|
|
force.strength = function(_) {
|
|
return arguments.length ? (strength = typeof _ === "function" ? _ : constant(+_), initialize(), force) : strength;
|
|
};
|
|
|
|
force.z = function(_) {
|
|
return arguments.length ? (z = typeof _ === "function" ? _ : constant(+_), initialize(), force) : z;
|
|
};
|
|
|
|
return force;
|
|
}
|
|
|
|
exports.forceCenter = center;
|
|
exports.forceCollide = collide;
|
|
exports.forceLink = link;
|
|
exports.forceManyBody = manyBody;
|
|
exports.forceRadial = radial;
|
|
exports.forceSimulation = simulation;
|
|
exports.forceX = x;
|
|
exports.forceY = y;
|
|
exports.forceZ = z;
|
|
|
|
}));
|