chart = {
const width = 928;
const height = 300;
const format = d3.format(",.0f");
const svg = d3
.create("svg")
.attr("width", width)
.attr("height", height)
.attr("viewBox", [0, 0, width, height])
.attr("style", "max-width: 100%; height: auto; font: 10px sans-serif;");
const sankey = d3
.sankey()
.nodeId((d) => d.name)
.nodeAlign(d3[nodeAlign])
.nodeWidth(15)
.nodePadding(10)
.extent([
[1, 5],
[width - 1, height - 5]
]);
const { nodes, links } = sankey({
nodes: data.nodes.map((d) => Object.assign({}, d)),
links: data.links.map((d) => Object.assign({}, d))
});
console.log({ nodes, links });
const color = d3.scaleOrdinal(d3.schemeCategory10);
const rect = svg
.append("g")
.attr("stroke", "#000")
.selectAll()
.data(nodes)
.join("rect")
.attr("x", (d) => d.x0)
.attr("y", (d) => d.y0)
.attr("height", (d) => d.y1 - d.y0)
.attr("width", (d) => d.x1 - d.x0)
.attr("fill", (d) => color(d.category));
rect.append("title").text((d) => `${d.name}\n${format(d.value)} TWh`);
const link = svg
.append("g")
.attr("fill", "none")
.attr("stroke-opacity", 0.5)
.selectAll()
.data(links)
.join("g")
.style("mix-blend-mode", "multiply");
if (linkColor === "source-target") {
const gradient = link
.append("linearGradient")
.attr("id", (d) => (d.uid = DOM.uid("link")).id)
.attr("gradientUnits", "userSpaceOnUse")
.attr("x1", (d) => d.source.x1)
.attr("x2", (d) => d.target.x0);
gradient
.append("stop")
.attr("offset", "0%")
.attr("stop-color", (d) => color(d.source.category));
gradient
.append("stop")
.attr("offset", "100%")
.attr("stop-color", (d) => color(d.target.category));
}
link
.append("path")
.attr("d", d3.sankeyLinkHorizontal())
.attr(
"stroke",
linkColor === "source-target"
? (d) => d.uid
: linkColor === "source"
? (d) => color(d.source.category)
: linkColor === "target"
? (d) => color(d.target.category)
: linkColor
)
.attr("stroke-width", (d) => Math.max(1, d.width));
link
.append("title")
.text((d) => `${d.source.name} → ${d.target.name}\n${format(d.value)} TWh`);
svg
.append("g")
.selectAll()
.data(nodes)
.join("text")
.attr("x", (d) => (d.x0 < width / 2 ? d.x1 + 6 : d.x0 - 6))
.attr("y", (d) => (d.y1 + d.y0) / 2)
.attr("dy", "0.35em")
.attr("text-anchor", (d) => (d.x0 < width / 2 ? "start" : "end"))
.text((d) => d.name);
return svg.node();
}