Rough. Re-do and delete.
forked from fogonwater's block: Delete and rework
license: mit |
Rough. Re-do and delete.
forked from fogonwater's block: Delete and rework
<!DOCTYPE html> | |
<meta charset="utf-8"> | |
<title>SANKEY Experiment</title> | |
<style> | |
body { | |
font-family: Helvetica; | |
font-size:0.6em; | |
} | |
.node rect { | |
cursor: move; | |
fill-opacity: .9; | |
shape-rendering: crispEdges; | |
} | |
.node text { | |
pointer-events: none; | |
text-shadow: 0 1px 0 #fff; | |
} | |
.link { | |
fill: none; | |
stroke: #000; | |
} | |
.link:hover { | |
stroke-opacity: .5; | |
} | |
</style> | |
<body> | |
<script src="https://d3js.org/d3.v4.min.js"></script> | |
<script src="sankey.js"></script> | |
<script> | |
console.clear() | |
var units = "percent", | |
iterations = 0; | |
// set the dimensions and margins of the graph | |
var margin = {top: 10, right: 10, bottom: 10, left: 10}, | |
width = 420 - margin.left - margin.right, | |
height = 690 - margin.top - margin.bottom; | |
// format variables | |
var formatNumber = d3.format(",.0f"), // zero decimal places | |
format = function(d) { return formatNumber(d) + " " + units; } | |
// append the svg object to the body of the page | |
var svg = d3.select("body").append("svg") | |
.attr("width", width + margin.left + margin.right) | |
.attr("height", height + margin.top + margin.bottom) | |
.append("g") | |
.attr("transform", | |
"translate(" + margin.left + "," + margin.top + ")"); | |
// Set the sankey diagram properties | |
var sankey = d3.sankey() | |
.nodeWidth(2) | |
.nodePadding(2) | |
.size([width, height]); | |
var path = sankey.link(); | |
// load the data | |
d3.json("sankey.json", function(error, graph) { | |
sankey | |
.nodes(graph.nodes) | |
.links(graph.links) | |
.layout(iterations); | |
// add in the links | |
var link = svg.append("g").selectAll(".link") | |
.data(graph.links) | |
.enter().append("path") | |
.attr("class", "link") | |
.attr("d", path) | |
.style("stroke-width", function(d) { return Math.max(1, d.dy); }) | |
.style("stroke", (d) => color(d.source.name)) | |
.style("opacity", (d) => 1) | |
.sort(function(a, b) { return b.dy - a.dy; }); | |
// add the link titles | |
/* | |
link.append("title") | |
.text(function(d) { | |
return d.source.label + " → " + | |
d.target.label + "\n" + format(d.value); });*/ | |
// add in the nodes | |
var node = svg.append("g").selectAll(".node") | |
.data(graph.nodes) | |
.enter().append("g") | |
.attr("class", "node") | |
.attr("transform", function(d) { | |
return "translate(" + d.x + "," + d.y + ")"; }) | |
.call(d3.drag() | |
.subject(function(d) { | |
return d; | |
}) | |
.on("start", function() { | |
this.parentNode.appendChild(this); | |
}) | |
.on("drag", dragmove)); | |
// add the rectangles for the nodes | |
node.append("rect") | |
.attr("height", function(d) { return d.dy; }) | |
.attr("width", sankey.nodeWidth()) | |
.style("fill", (d) => color(d.name)) | |
.text(function(d) { | |
return d.name + "\n" + format(d.value); }); | |
// add in the title for the nodes | |
node.append("text") | |
.attr("x", -6) | |
.attr("y", function(d) { return d.dy / 2; }) | |
.attr("dy", ".35em") | |
.attr("text-anchor", "end") | |
.attr("transform", null) | |
.text((d)=> d.name + ': ' + Math.round(d.value * 100) + '%') | |
.filter(function(d) { return d.x < width / 2; }) | |
.attr("x", 6 + sankey.nodeWidth()) | |
.attr("text-anchor", "start"); | |
// the function for moving the nodes | |
function dragmove(d) { | |
d3.select(this) | |
.attr("transform", | |
"translate(" | |
+ d.x + "," | |
+ (d.y = Math.max( | |
0, Math.min(height - d.dy, d3.event.y)) | |
) + ")"); | |
sankey.relayout(); | |
link.attr("d", path); | |
} | |
}); | |
function color(s) { | |
switch(s) { | |
case "Non-communicable diseases": | |
return '#009a93' | |
case "_Misc": | |
return '#009cde' | |
case "_Injuries": | |
return '#006ab4' | |
} | |
return '#ccc' | |
} | |
function opacity(s) { | |
switch(s) { | |
case "Cancers": | |
return 0.2 | |
case "Cardiovascular disease": | |
return 0.3 | |
case "Other non-communicable diseases": | |
return 0.4 | |
case "Musculoskeletal conditions": | |
return 0.5 | |
case "Mental disorders": | |
return 0.6 | |
case "Neurological disorders": | |
return 0.7 | |
case "Chronic respiratory diseases": | |
return 0.8 | |
case "Diabetes, chronic kidney diseases": | |
return 0.9 | |
case "Substance use disorders": | |
return 1.0 | |
default: | |
return 1.0 | |
} | |
} | |
</script> | |
</body> |
d3.sankey = function() { | |
var sankey = {}, | |
nodeWidth = 24, | |
nodePadding = 8, | |
size = [1, 1], | |
nodes = [], | |
links = []; | |
sankey.nodeWidth = function(_) { | |
if (!arguments.length) return nodeWidth; | |
nodeWidth = +_; | |
return sankey; | |
}; | |
sankey.nodePadding = function(_) { | |
if (!arguments.length) return nodePadding; | |
nodePadding = +_; | |
return sankey; | |
}; | |
sankey.nodes = function(_) { | |
if (!arguments.length) return nodes; | |
nodes = _; | |
return sankey; | |
}; | |
sankey.links = function(_) { | |
if (!arguments.length) return links; | |
links = _; | |
return sankey; | |
}; | |
sankey.size = function(_) { | |
if (!arguments.length) return size; | |
size = _; | |
return sankey; | |
}; | |
sankey.layout = function(iterations) { | |
computeNodeLinks(); | |
computeNodeValues(); | |
computeNodeBreadths(); | |
computeNodeDepths(iterations); | |
computeLinkDepths(); | |
return sankey; | |
}; | |
sankey.relayout = function() { | |
computeLinkDepths(); | |
return sankey; | |
}; | |
sankey.link = function() { | |
var curvature = .5; | |
function link(d) { | |
var x0 = d.source.x + d.source.dx, | |
x1 = d.target.x, | |
xi = d3.interpolateNumber(x0, x1), | |
x2 = xi(curvature), | |
x3 = xi(1 - curvature), | |
y0 = d.source.y + d.sy + d.dy / 2, | |
y1 = d.target.y + d.ty + d.dy / 2; | |
return "M" + x0 + "," + y0 | |
+ "C" + x2 + "," + y0 | |
+ " " + x3 + "," + y1 | |
+ " " + x1 + "," + y1; | |
} | |
link.curvature = function(_) { | |
if (!arguments.length) return curvature; | |
curvature = +_; | |
return link; | |
}; | |
return link; | |
}; | |
// Populate the sourceLinks and targetLinks for each node. | |
// Also, if the source and target are not objects, assume they are indices. | |
function computeNodeLinks() { | |
nodes.forEach(function(node) { | |
node.sourceLinks = []; | |
node.targetLinks = []; | |
}); | |
links.forEach(function(link) { | |
var source = link.source, | |
target = link.target; | |
if (typeof source === "number") source = link.source = nodes[link.source]; | |
if (typeof target === "number") target = link.target = nodes[link.target]; | |
source.sourceLinks.push(link); | |
target.targetLinks.push(link); | |
}); | |
} | |
// Compute the value (size) of each node by summing the associated links. | |
function computeNodeValues() { | |
nodes.forEach(function(node) { | |
node.value = Math.max( | |
d3.sum(node.sourceLinks, value), | |
d3.sum(node.targetLinks, value) | |
); | |
}); | |
} | |
// Iteratively assign the breadth (x-position) for each node. | |
// Nodes are assigned the maximum breadth of incoming neighbors plus one; | |
// nodes with no incoming links are assigned breadth zero, while | |
// nodes with no outgoing links are assigned the maximum breadth. | |
function computeNodeBreadths() { | |
var remainingNodes = nodes, | |
nextNodes, | |
x = 0; | |
while (remainingNodes.length) { | |
nextNodes = []; | |
remainingNodes.forEach(function(node) { | |
node.x = x; | |
node.dx = nodeWidth; | |
node.sourceLinks.forEach(function(link) { | |
if (nextNodes.indexOf(link.target) < 0) { | |
nextNodes.push(link.target); | |
} | |
}); | |
}); | |
remainingNodes = nextNodes; | |
++x; | |
} | |
// | |
moveSinksRight(x); | |
scaleNodeBreadths((size[0] - nodeWidth) / (x - 1)); | |
} | |
function moveSourcesRight() { | |
nodes.forEach(function(node) { | |
if (!node.targetLinks.length) { | |
node.x = d3.min(node.sourceLinks, function(d) { return d.target.x; }) - 1; | |
} | |
}); | |
} | |
function moveSinksRight(x) { | |
nodes.forEach(function(node) { | |
if (!node.sourceLinks.length) { | |
node.x = x - 1; | |
} | |
}); | |
} | |
function scaleNodeBreadths(kx) { | |
nodes.forEach(function(node) { | |
node.x *= kx; | |
}); | |
} | |
function computeNodeDepths(iterations) { | |
var nodesByBreadth = d3.nest() | |
.key(function(d) { return d.x; }) | |
.sortKeys(d3.ascending) | |
.entries(nodes) | |
.map(function(d) { return d.values; }); | |
// | |
initializeNodeDepth(); | |
resolveCollisions(); | |
for (var alpha = 1; iterations > 0; --iterations) { | |
relaxRightToLeft(alpha *= .99); | |
resolveCollisions(); | |
relaxLeftToRight(alpha); | |
resolveCollisions(); | |
} | |
function initializeNodeDepth() { | |
var ky = d3.min(nodesByBreadth, function(nodes) { | |
return (size[1] - (nodes.length - 1) * nodePadding) / d3.sum(nodes, value); | |
}); | |
nodesByBreadth.forEach(function(nodes) { | |
nodes.forEach(function(node, i) { | |
node.y = i; | |
node.dy = node.value * ky; | |
}); | |
}); | |
links.forEach(function(link) { | |
link.dy = link.value * ky; | |
}); | |
} | |
function relaxLeftToRight(alpha) { | |
nodesByBreadth.forEach(function(nodes, breadth) { | |
nodes.forEach(function(node) { | |
if (node.targetLinks.length) { | |
var y = d3.sum(node.targetLinks, weightedSource) / d3.sum(node.targetLinks, value); | |
node.y += (y - center(node)) * alpha; | |
} | |
}); | |
}); | |
function weightedSource(link) { | |
return center(link.source) * link.value; | |
} | |
} | |
function relaxRightToLeft(alpha) { | |
nodesByBreadth.slice().reverse().forEach(function(nodes) { | |
nodes.forEach(function(node) { | |
if (node.sourceLinks.length) { | |
var y = d3.sum(node.sourceLinks, weightedTarget) / d3.sum(node.sourceLinks, value); | |
node.y += (y - center(node)) * alpha; | |
} | |
}); | |
}); | |
function weightedTarget(link) { | |
return center(link.target) * link.value; | |
} | |
} | |
function resolveCollisions() { | |
nodesByBreadth.forEach(function(nodes) { | |
var node, | |
dy, | |
y0 = 0, | |
n = nodes.length, | |
i; | |
// Push any overlapping nodes down. | |
nodes.sort(ascendingDepth); | |
for (i = 0; i < n; ++i) { | |
node = nodes[i]; | |
dy = y0 - node.y; | |
if (dy > 0) node.y += dy; | |
y0 = node.y + node.dy + nodePadding; | |
} | |
// If the bottommost node goes outside the bounds, push it back up. | |
dy = y0 - nodePadding - size[1]; | |
if (dy > 0) { | |
y0 = node.y -= dy; | |
// Push any overlapping nodes back up. | |
for (i = n - 2; i >= 0; --i) { | |
node = nodes[i]; | |
dy = node.y + node.dy + nodePadding - y0; | |
if (dy > 0) node.y -= dy; | |
y0 = node.y; | |
} | |
} | |
}); | |
} | |
function ascendingDepth(a, b) { | |
return a.y - b.y; | |
} | |
} | |
function computeLinkDepths() { | |
nodes.forEach(function(node) { | |
node.sourceLinks.sort(ascendingTargetDepth); | |
node.targetLinks.sort(ascendingSourceDepth); | |
}); | |
nodes.forEach(function(node) { | |
var sy = 0, ty = 0; | |
node.sourceLinks.forEach(function(link) { | |
link.sy = sy; | |
sy += link.dy; | |
}); | |
node.targetLinks.forEach(function(link) { | |
link.ty = ty; | |
ty += link.dy; | |
}); | |
}); | |
function ascendingSourceDepth(a, b) { | |
return a.source.y - b.source.y; | |
} | |
function ascendingTargetDepth(a, b) { | |
return a.target.y - b.target.y; | |
} | |
} | |
function center(node) { | |
return node.y + node.dy / 2; | |
} | |
function value(link) { | |
return link.value; | |
} | |
return sankey; | |
}; |
{ | |
"links": [ | |
{ | |
"source": 0, | |
"target": 13, | |
"value": 0.82 | |
}, | |
{ | |
"source": 13, | |
"target": 14, | |
"value": 0.174 | |
}, | |
{ | |
"source": 0, | |
"target": 6, | |
"value": 0.14 | |
}, | |
{ | |
"source": 6, | |
"target": 7, | |
"value": 0.14 | |
}, | |
{ | |
"source": 13, | |
"target": 15, | |
"value": 0.139 | |
}, | |
{ | |
"source": 13, | |
"target": 16, | |
"value": 0.125 | |
}, | |
{ | |
"source": 13, | |
"target": 17, | |
"value": 0.116 | |
}, | |
{ | |
"source": 14, | |
"target": 27, | |
"value": 0.092 | |
}, | |
{ | |
"source": 16, | |
"target": 33, | |
"value": 0.091 | |
}, | |
{ | |
"source": 13, | |
"target": 18, | |
"value": 0.085 | |
}, | |
{ | |
"source": 17, | |
"target": 34, | |
"value": 0.079 | |
}, | |
{ | |
"source": 13, | |
"target": 19, | |
"value": 0.071 | |
}, | |
{ | |
"source": 15, | |
"target": 28, | |
"value": 0.07 | |
}, | |
{ | |
"source": 13, | |
"target": 20, | |
"value": 0.052 | |
}, | |
{ | |
"source": 7, | |
"target": 8, | |
"value": 0.046 | |
}, | |
{ | |
"source": 0, | |
"target": 1, | |
"value": 0.04 | |
}, | |
{ | |
"source": 1, | |
"target": 2, | |
"value": 0.04 | |
}, | |
{ | |
"source": 13, | |
"target": 21, | |
"value": 0.038 | |
}, | |
{ | |
"source": 15, | |
"target": 30, | |
"value": 0.037 | |
}, | |
{ | |
"source": 20, | |
"target": 43, | |
"value": 0.035 | |
}, | |
{ | |
"source": 18, | |
"target": 39, | |
"value": 0.034 | |
}, | |
{ | |
"source": 15, | |
"target": 29, | |
"value": 0.032 | |
}, | |
{ | |
"source": 14, | |
"target": 23, | |
"value": 0.031 | |
}, | |
{ | |
"source": 7, | |
"target": 9, | |
"value": 0.028 | |
}, | |
{ | |
"source": 18, | |
"target": 37, | |
"value": 0.028 | |
}, | |
{ | |
"source": 19, | |
"target": 40, | |
"value": 0.028 | |
}, | |
{ | |
"source": 7, | |
"target": 12, | |
"value": 0.027 | |
}, | |
{ | |
"source": 19, | |
"target": 41, | |
"value": 0.027 | |
}, | |
{ | |
"source": 21, | |
"target": 46, | |
"value": 0.026 | |
}, | |
{ | |
"source": 14, | |
"target": 24, | |
"value": 0.024 | |
}, | |
{ | |
"source": 17, | |
"target": 36, | |
"value": 0.023 | |
}, | |
{ | |
"source": 18, | |
"target": 38, | |
"value": 0.023 | |
}, | |
{ | |
"source": 2, | |
"target": 3, | |
"value": 0.022 | |
}, | |
{ | |
"source": 7, | |
"target": 10, | |
"value": 0.022 | |
}, | |
{ | |
"source": 13, | |
"target": 22, | |
"value": 0.02 | |
}, | |
{ | |
"source": 16, | |
"target": 31, | |
"value": 0.02 | |
}, | |
{ | |
"source": 7, | |
"target": 11, | |
"value": 0.017 | |
}, | |
{ | |
"source": 14, | |
"target": 25, | |
"value": 0.017 | |
}, | |
{ | |
"source": 19, | |
"target": 42, | |
"value": 0.016 | |
}, | |
{ | |
"source": 16, | |
"target": 32, | |
"value": 0.014 | |
}, | |
{ | |
"source": 17, | |
"target": 35, | |
"value": 0.014 | |
}, | |
{ | |
"source": 21, | |
"target": 47, | |
"value": 0.013 | |
}, | |
{ | |
"source": 20, | |
"target": 44, | |
"value": 0.012 | |
}, | |
{ | |
"source": 2, | |
"target": 4, | |
"value": 0.011 | |
}, | |
{ | |
"source": 22, | |
"target": 48, | |
"value": 0.011 | |
}, | |
{ | |
"source": 14, | |
"target": 26, | |
"value": 0.01 | |
}, | |
{ | |
"source": 22, | |
"target": 49, | |
"value": 0.009 | |
}, | |
{ | |
"source": 2, | |
"target": 5, | |
"value": 0.007 | |
}, | |
{ | |
"source": 20, | |
"target": 45, | |
"value": 0.005 | |
} | |
], | |
"nodes": [ | |
{ | |
"name": "total" | |
}, | |
{ | |
"name": "_Communicable" | |
}, | |
{ | |
"name": "Communicable" | |
}, | |
{ | |
"name": "Neonatal" | |
}, | |
{ | |
"name": "Congenital birth defects" | |
}, | |
{ | |
"name": "Other communicable" | |
}, | |
{ | |
"name": "_Injuries" | |
}, | |
{ | |
"name": "Injuries" | |
}, | |
{ | |
"name": "Falls" | |
}, | |
{ | |
"name": "Road injuries" | |
}, | |
{ | |
"name": "Self-harm" | |
}, | |
{ | |
"name": "Mechanical forces" | |
}, | |
{ | |
"name": "Other injuries" | |
}, | |
{ | |
"name": "Non-communicable diseases" | |
}, | |
{ | |
"name": "Cancers" | |
}, | |
{ | |
"name": "Cardiovascular disease" | |
}, | |
{ | |
"name": "Other non-communicable diseases" | |
}, | |
{ | |
"name": "Musculoskeletal conditions" | |
}, | |
{ | |
"name": "Mental disorders" | |
}, | |
{ | |
"name": "Neurological disorders" | |
}, | |
{ | |
"name": "Chronic respiratory diseases" | |
}, | |
{ | |
"name": "Diabetes, chronic kidney diseases" | |
}, | |
{ | |
"name": "Substance use disorders" | |
}, | |
{ | |
"name": "Tracheal, bronchus, lung cancer" | |
}, | |
{ | |
"name": "Colon and rectum cancer" | |
}, | |
{ | |
"name": "Breast cancer" | |
}, | |
{ | |
"name": "Prostate cancer" | |
}, | |
{ | |
"name": "Other cancers" | |
}, | |
{ | |
"name": "Ischemic heart disease" | |
}, | |
{ | |
"name": "Stroke" | |
}, | |
{ | |
"name": "Other cardiovascular" | |
}, | |
{ | |
"name": "Age-related and other hearing loss" | |
}, | |
{ | |
"name": "Oral disorders" | |
}, | |
{ | |
"name": "Other other non-communicable" | |
}, | |
{ | |
"name": "Low back pain" | |
}, | |
{ | |
"name": "Neck pain" | |
}, | |
{ | |
"name": "Other musculoskeletal" | |
}, | |
{ | |
"name": "Anxiety disorders" | |
}, | |
{ | |
"name": "Depressive disorders" | |
}, | |
{ | |
"name": "Other mental" | |
}, | |
{ | |
"name": "Headache disorders" | |
}, | |
{ | |
"name": "Alzheimer's disease and other dementias" | |
}, | |
{ | |
"name": "Other neurological" | |
}, | |
{ | |
"name": "Chronic obstructive pulmonary disease" | |
}, | |
{ | |
"name": "Asthma" | |
}, | |
{ | |
"name": "Other respiratory" | |
}, | |
{ | |
"name": "Diabetes mellitus" | |
}, | |
{ | |
"name": "Chronic kidney disease" | |
}, | |
{ | |
"name": "Drug use disorders" | |
}, | |
{ | |
"name": "Alcohol use disorders" | |
} | |
] | |
} |