// CSS require('./sankey.css'); // JS var px = window.px || require('../javascripts/modules/caravel.js'); var d3 = window.d3 || require('d3'); d3.sankey = require('d3-sankey').sankey; function sankeyVis(slice) { var div = d3.select(slice.selector); var render = function () { var margin = { top: 5, right: 5, bottom: 5, left: 5 }; var width = slice.width() - margin.left - margin.right; var height = slice.height() - margin.top - margin.bottom; var formatNumber = d3.format(",.2f"); div.selectAll("*").remove(); var svg = div.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 + ")"); var tooltip = div.append("div") .attr("class", "sankey-tooltip") .style("opacity", 0); var sankey = d3.sankey() .nodeWidth(15) .nodePadding(10) .size([width, height]); var path = sankey.link(); d3.json(slice.jsonEndpoint(), function (error, json) { if (error !== null) { slice.error(error.responseText, error); return ''; } var links = json.data; var nodes = {}; // Compute the distinct nodes from the links. links.forEach(function (link) { link.source = nodes[link.source] || (nodes[link.source] = { name: link.source }); link.target = nodes[link.target] || (nodes[link.target] = { name: link.target }); link.value = Number(link.value); }); nodes = d3.values(nodes); sankey .nodes(nodes) .links(links) .layout(32); var link = svg.append("g").selectAll(".link") .data(links) .enter().append("path") .attr("class", "link") .attr("d", path) .style("stroke-width", function (d) { return Math.max(1, d.dy); }) .sort(function (a, b) { return b.dy - a.dy; }) .on("mouseover", onmouseover) .on("mouseout", onmouseout); var node = svg.append("g").selectAll(".node") .data(nodes) .enter().append("g") .attr("class", "node") .attr("transform", function (d) { return "translate(" + d.x + "," + d.y + ")"; }) .call(d3.behavior.drag() .origin(function (d) { return d; }) .on("dragstart", function () { this.parentNode.appendChild(this); }) .on("drag", dragmove)); node.append("rect") .attr("height", function (d) { return d.dy; }) .attr("width", sankey.nodeWidth()) .style("fill", function (d) { d.color = px.color.category21(d.name.replace(/ .*/, "")); return d.color; }) .style("stroke", function (d) { return d3.rgb(d.color).darker(2); }) .on("mouseover", onmouseover) .on("mouseout", onmouseout); 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(function (d) { return d.name; }) .filter(function (d) { return d.x < width / 2; }) .attr("x", 6 + sankey.nodeWidth()) .attr("text-anchor", "start"); 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 getTooltipHtml(d) { var html; if (d.sourceLinks) { // is node html = d.name + " Value: " + formatNumber(d.value) + ""; } else { var val = formatNumber(d.value); var sourcePercent = d3.round((d.value / d.source.value) * 100, 1); var targetPercent = d3.round((d.value / d.target.value) * 100, 1); html = [ "
Path Value: ", val, "
", "
", "", (isFinite(sourcePercent) ? sourcePercent : "100"), "% of ", d.source.name, "
", "" + (isFinite(targetPercent) ? targetPercent : "--") + "% of ", d.target.name, "target", "
" ].join(""); } return html; } function onmouseover(d) { tooltip .html(function () { return getTooltipHtml(d); }) .transition() .duration(200) .style("left", (d3.event.layerX + 10) + "px") .style("top", (d3.event.layerY + 10) + "px") .style("opacity", 0.95); } function onmouseout(d) { tooltip.transition() .duration(100) .style("opacity", 0); } slice.done(json); }); }; return { render: render, resize: render }; } module.exports = sankeyVis;