Research Application Manager with the Alan Turing Institute / PhD from @GC_CUNY / formerly Research Software Engineer @LivingwMachines and @dhinstitutes project mgr
Public
Edited
Nov 24, 2022
4 forks
Importers
24 stars
groupedGraph(miserables, {
margin: margin,
curve: curveType
}).node()
groupedGraph(miserables, {
height: 350,
label: (node) => node.id,
labelStyle: {
fontSize: 12,
fontFamily: "serif",
color: true
}
}).node()
groupedGraph(miserables, {
height: 350,
color: d3.scaleOrdinal().range(["#ff8c00", "#d0743c", "#a05d56"]),
nodeStyle: {
stroke: "black",
strokeWidth: "1px"
},
strokeStyle: {
color: "green",
opacity: 0.1
},
groupStyle: {
fillOpacity: 0.4,
strokeOpacity: 0.6
}
}).node()
groupedGraph(miserables, {
onHover: (evt, d) => {
console.log(d.id, "is in group", d.group);
}
}).node()
groupedGraph(miserables, {
height: 350,
excludeGroups: excludeGroups,
fade: false
}).node()
groupedGraph()
groupedGraph({
nodes: [],
links: []
})
groupedGraph({
nodes: [],
links: [
{
source: "source-id",
target: "target-id"
}
]
})
groupedGraph(miserables, {
groupX: [100, 200, 300]
})
groupedGraph(miserables)
groupedGraph = (
graph = {},
{
height = 600,
color = d3
.scaleOrdinal()
.range([
"red",
"green",
"blue",
"#6b486b",
"#a05d56",
"#d0743c",
"#ff8c00"
]),
groupColorNames = [],
margin = 1.2,
curve = "curveCatmullRomClosed",
r = 5,
label = undefined,
labelStyle = {
fontFamily: "sans-serif",
fontSize: 10,
color: true
},
strokeWidth = (d) => Math.sqrt(d.value),
strokeStyle = {
color: "gray",
opacity: 0.6
},
nodeStyle = {
stroke: "white",
strokeWidth: "1.5px"
},
groupStyle = {
fillOpacity: 0.1,
strokeOpacity: 1
},
onHover = undefined,
onClick = undefined,
onLeave = undefined,
groupX = [], // in development
groupY = [], // in development
displayGroupOnHover = false,
chargeStrength = undefined,
linkStrength = undefined,
excludeGroups = [],
fade = true
} = {}
) => {
if (
!Object.keys(graph).includes("links") ||
!Object.keys(graph).includes("nodes")
)
throw new Error(
"The visualization must be provided a graph object containing links and nodes."
);
if (graph.links.length === 0)
throw new Error(
`No links were provided. See documentation for detailed instructions on the required data structure for the graph.`
);
if (graph.nodes.length === 0)
throw new Error(
`No nodes were provided. See documentation for detailed instructions on the required data structure for the graph.`
);
if (groupX.length || groupY.length) {
const _ = [...new Set(graph.nodes.map((d) => d.group))]
.sort((a, b) => a - b)
.forEach((group) => {
if (groupX.length && groupX[group] === undefined)
throw new Error(
`groupX provided but not of correct length: group ${group} has no value.`
);
if (groupY.length && groupY[group] === undefined)
throw new Error(
`groupY provided but not of correct length: group ${group} has no value.`
);
});
}
const svg = d3.create("svg").attr("viewBox", [0, 0, width, height]);
// create groups, links and nodes
const groups = svg.append("g").attr("class", "groups");
const link = svg
.append("g")
.attr("class", "links")
.selectAll("line")
.data(graph.links)
.join("line")
.attr("stroke-width", strokeWidth)
.attr("stroke", strokeStyle.color)
.attr("stroke-opacity", strokeStyle.opacity);
const node = svg
.append("g")
.attr("class", "nodes")
.selectAll("circle")
.data(graph.nodes)
.join("circle")
.attr("r", r)
.attr("stroke", nodeStyle.stroke)
.attr("stroke-width", nodeStyle.strokeWidth)
.attr("fill", (d) => {
if (!groupColorNames.length) return color(d.group);
return color(groupColorNames[d.group]);
})
.call(
d3
.drag()
.on("start", dragstarted)
.on("drag", dragged)
.on("end", dragended)
);
let _label = svg
.append("g")
.attr("class", "labels")
.selectAll("text")
.data(graph.nodes)
.join("text")
.text(label)
.attr("text-anchor", "middle")
.attr("font-family", labelStyle.fontFamily)
.attr("font-size", labelStyle.fontSize)
.attr("fill", (d) => {
if (!labelStyle.color) return "black";
if (!groupColorNames.length) return color(d.group);
return color(groupColorNames[d.group]);
})
.call(
d3
.drag()
.on("start", dragstarted)
.on("drag", dragged)
.on("end", dragended)
);
if (typeof onHover === "function") {
node.on("mouseover", onHover);
}
if (typeof onLeave === "function") {
node.on("mouseout", onLeave);
}
if (typeof onClick === "function") {
node.on("click", onClick);
}
const simulation = d3.forceSimulation().nodes(graph.nodes);
if (linkStrength && typeof linkStrength === "number") {
simulation.force(
"link",
d3
.forceLink()
.links(graph.links)
.id((d) => d.id)
.strength(linkStrength)
);
} else {
simulation.force(
"link",
d3
.forceLink()
.links(graph.links)
.id((d) => d.id)
);
}
if (chargeStrength && typeof chargeStrength === "number") {
simulation.force("charge", d3.forceManyBody().strength(chargeStrength));
} else {
simulation.force("charge", d3.forceManyBody());
}
if (groupX.length === 0 && groupY.length === 0) {
simulation.force("center", d3.forceCenter(width / 2, height / 2));
}
if (groupX.length) {
simulation
.force(
"x",
d3.forceX().x((d) => groupX[+d.group])
)
.force(
"y",
d3.forceY().y((d) => height / 2)
);
}
if (groupY.length) {
simulation
.force(
"y",
d3.forceY().y((d) => groupY[+d.group])
)
.force(
"x",
d3.forceX().x((d) => width / 2)
);
}
simulation.on("tick", tick);
function tick() {
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("x", (d) => d.x).attr("y", (d) => d.y);
updateGroups(paths, groupIds, node, curve, margin);
}
// count members of each group. Groups with less
// than 3 member will not be considered (creating
// a convex hull need 3 points at least)
const groupIds = [...new Set(graph.nodes.map((n) => +n.group))]
.map((groupId) => ({
groupId: groupId,
count: graph.nodes.filter((n) => +n.group == groupId).length
}))
.filter(
(group) => group.count > 2 && !excludeGroups.includes(group.groupId)
)
.map((group) => group.groupId);
const paths = groups
.selectAll(".path_placeholder")
.data(groupIds, (d) => +d)
.join("g")
.attr("class", "path_placeholder")
.attr("fill-opacity", groupStyle.fillOpacity)
.attr("stroke-opacity", groupStyle.strokeOpacity)
.append("path")
.attr("stroke", (d) => {
if (!groupColorNames.length) return color(d);
return color(groupColorNames[d]);
})
.attr("fill", (d) => {
if (!groupColorNames.length) return color(d);
return color(groupColorNames[d]);
})
.attr("opacity", 0);
// TODO: THIS LOOK UNFINISHED
if (!displayGroupOnHover) {
fade
? paths.transition().duration(2000).attr("opacity", 1)
: paths.attr("opacity", 1);
} else {
}
// add interaction to the groups
groups
.selectAll(".path_placeholder")
.call(
d3
.drag()
.on("start", group_dragstarted)
.on("drag", group_dragged)
.on("end", group_dragended)
);
node.append("title").text(function (d) {
return d.id;
});
// drag nodes
function dragstarted(evt, d) {
if (!evt.active) simulation.alphaTarget(0.3).restart();
d.fx = d.x;
d.fy = d.y;
}
function dragged(evt, d) {
d.fx = evt.x;
d.fy = evt.y;
}
function dragended(evt, d) {
if (!evt.active) simulation.alphaTarget(0);
d.fx = null;
d.fy = null;
}
// drag groups
function group_dragstarted(evt, groupId) {
if (!evt.active) simulation.alphaTarget(0.3).restart();
d3.select(this).select("path").style("stroke-width", 3);
}
function group_dragged(evt, groupId) {
node
.filter(function (d) {
return d.group == groupId;
})
.each(function (d) {
d.x += evt.dx;
d.y += evt.dy;
});
}
function group_dragended(evt, groupId) {
if (!evt.active) simulation.alphaTarget(0.3).restart();
d3.select(this).select("path").style("stroke-width", 1);
}
invalidation.then(() => simulation.stop());
return svg;
}
polygonGenerator = function(groupId, node) {
// select nodes of the group, retrieve its positions and return the convex hull of the specified points
// (3 points as minimum, otherwise returns null)
var node_coords = node
.filter(d => d.group == groupId)
.data()
.map(d => [d.x, d.y]);
return d3.polygonHull(node_coords);
};
function updateGroups(paths, groupIds, node, curve, margin) {
const valueline = d3.line()
.x(d => d[0])
.y(d => d[1])
.curve(d3[curveType])
groupIds.forEach(groupId => {
let centroid = [];
let path = paths.filter(d => d == groupId)
.attr('transform', 'scale(1) translate(0,0)')
.attr('d', d => {
const polygon = polygonGenerator(d, node);
centroid = d3.polygonCentroid(polygon);
// to scale the shape properly around its points: move the 'g' element to the centroid point, translate
// all the path around the center of the 'g' and then we can scale the 'g' element properly
return valueline(polygon.map(point => [point[0] - centroid[0], point[1] - centroid[1]]));
});
d3.select(path.node().parentNode)
.attr('transform', 'translate(' + centroid[0] + ',' + (centroid[1]) + ') scale(' + margin + ')');
});
}
basicStyling = html`<style>
.links line {
// stroke: #999;
// stroke-opacity: 0.6;
}
.nodes circle {
// stroke: #fff;
// stroke-width: 1.5px;
}
.path_placeholder {
// fill-opacity: .1;
// stroke-opacity: 1;
}
.labels text {
-webkit-user-select: none; /* Safari */
-moz-user-select: none; /* Firefox */
-ms-user-select: none; /* IE10+/Edge */
user-select: none; /* Standard */
}
</style>`
ClickedElement = ({message: undefined})
One platform to build and deploy the best data apps
Experiment and prototype by building visualizations in live JavaScript notebooks. Collaborate with your team and decide which concepts to build out.
Use Observable Framework to build data apps locally. Use data loaders to build in any language or library, including Python, SQL, and R.
Seamlessly deploy to Observable. Test before you ship, use automatic deploy-on-commit, and ensure your projects are always up-to-date.