function ForceGraph(
{
nodes,
links
},
{
nodeId,
nodeGroup,
nodeGroups,
nodeTitle,
nodeLabel,
nodeFill = "currentColor",
nodeStroke,
nodeStrokeWidth = 12,
nodeStrokeOpacity = 1,
nodeRadius,
collisionOffset,
labelOffset,
nodeStrength,
linkSource = ({ source }) => source,
linkTarget = ({ target }) => target,
linkStroke = "#222",
linkStrokeOpacity = 0.7,
linkStrokeWidth = 1.5,
linkStrokeLinecap = "round",
linkStrength,
colors = colourArray,
width = 3000,
height = 4000,
invalidation
} = {}
) {
// Compute values.
const N = d3.map(nodes, nodeId).map(intern);
const LS = d3.map(links, linkSource).map(intern);
const LT = d3.map(links, linkTarget).map(intern);
if (nodeTitle === undefined) nodeTitle = (_, i) => N[i];
const T = nodeTitle == null ? null : d3.map(nodes, nodeTitle);
const B = nodeLabel == null ? null : d3.map(nodes, nodeLabel);
const G = nodeGroup == null ? null : d3.map(nodes, nodeGroup).map(intern);
const R = nodeRadius == null ? null : d3.map(nodes, nodeRadius).map(intern);
const W =
typeof linkStrokeWidth !== "function"
? null
: d3.map(links, linkStrokeWidth);
const L = typeof linkStroke !== "function" ? null : d3.map(links, linkStroke);
const S = nodeStroke == null ? null : d3.map(nodes, nodeStroke);
// Replace the input nodes and links with mutable objects for the simulation.
nodes = d3.map(nodes, (_, i) => ({ id: N[i] }));
links = d3.map(links, (_, i) => ({ source: LS[i], target: LT[i] }));
// Compute default domains.
if (G && nodeGroups === undefined) nodeGroups = d3.sort(G);
// Construct the scales.
const color = nodeGroup == null ? null : d3.scaleOrdinal(nodeGroups, colors);
// Construct the forces.
const forceNode = d3.forceManyBody();
const forceLink = d3.forceLink(links).id(({ index: i }) => N[i]);
// Amendment by Kathy Reid to prevent the circles from overlapping
const forceCollide = d3
.forceCollide()
.radius(({ index: i }) => R[i] + collisionOffset);
if (nodeStrength !== undefined) forceNode.strength(nodeStrength);
if (linkStrength !== undefined) forceLink.strength(linkStrength);
const simulation = d3
.forceSimulation(nodes)
.force("link", forceLink)
.force("charge", forceNode)
.force("center", d3.forceCenter())
.force("collision", forceCollide)
.on("tick", ticked);
const svg = d3
.create("svg")
.attr("width", width)
.attr("height", height)
.attr("viewBox", [-width / 2, -height / 2, width, height])
.attr("style", "max-width: 100%; height: auto; height: intrinsic;");
const link = svg
.append("g")
.attr("stroke", typeof linkStroke !== "function" ? linkStroke : null)
.attr("stroke-opacity", linkStrokeOpacity)
.attr(
"stroke-width",
typeof linkStrokeWidth !== "function" ? linkStrokeWidth : null
)
.attr("stroke-linecap", linkStrokeLinecap)
.selectAll("line")
.data(links)
.join("line");
const node = svg
.append("g")
.attr("fill", nodeFill)
//.attr("stroke", nodeStroke)
.attr("stroke-opacity", nodeStrokeOpacity)
.attr("stroke-width", nodeStrokeWidth)
.selectAll("circle")
.data(nodes)
.join("circle")
.call(drag(simulation))
// isolate the links of a node for easier exploration
.on("mouseenter", (evt, d) => {
link
.attr("display", "none")
.filter((l) => l.source.id === d.id || l.target.id === d.id)
.attr("display", "block");
})
.on("mouseleave", (evt) => {
link.attr("display", "block");
});
// labels for the nodes
const label = svg
.append("g")
.selectAll("text")
.data(nodes)
.enter()
.append("text")
.text(({ index: i }) => T[i])
.attr("class", "labels")
.attr("text-anchor", "middle")
.attr("dx", 0)
.attr("y", ({ index: i }) => R[i] + labelOffset);
if (W) link.attr("stroke-width", ({ index: i }) => W[i]);
if (L) link.attr("stroke", ({ index: i }) => L[i]);
if (G) node.attr("fill", ({ index: i }) => color(G[i]));
if (T) node.append("title").text(({ index: i }) => T[i]);
// use the predetermined attribute of a node to determine stroke colour
if (S)
node.attr("stroke", ({ index: i }) => {
if (S[i]) {
return "#000000";
} else {
return "#fefefe";
}
});
if (B) label.text(({ index: i }) => B[i]);
if (R) node.attr("r", ({ index: i }) => R[i]);
if (invalidation != null) invalidation.then(() => simulation.stop());
function intern(value) {
return value !== null && typeof value === "object"
? value.valueOf()
: value;
}
function ticked() {
link
.attr("x1", (d) => d.source.x)
.attr("y1", (d) => d.source.y)
.attr("x2", (d) => d.target.x)
.attr("y2", (d) => d.target.y);
node.attr("cx", (d) => d.x).attr("cy", (d) => d.y);
label.attr("dx", (d) => d.x).attr("dy", (d) => d.y);
}
function drag(simulation) {
function dragstarted(event) {
if (!event.active) simulation.alphaTarget(0.3).restart();
event.subject.fx = event.subject.x;
event.subject.fy = event.subject.y;
}
function dragged(event) {
event.subject.fx = event.x;
event.subject.fy = event.y;
}
function dragended(event) {
if (!event.active) simulation.alphaTarget(0);
event.subject.fx = null;
event.subject.fy = null;
}
return d3
.drag()
.on("start", dragstarted)
.on("drag", dragged)
.on("end", dragended);
}
return Object.assign(svg.node(), { scales: { color } });
}