Sankey-diagram-Sample 2 (using a csv file with v4)

This is a sankey diagram used as an example in the book D3 Tips and Tricks v4. It is based on a simpler version here.

It demonstrates the construction of a sankey diagram using d3.js from external csv data that is formatted using just link data and named values.

forked from d3noob's block: Sankey diagram (using a csv file with v4)

forked from AJ-Mengistu's block: Sankey-diagram-Sample 2 (using a csv file with v4)

<!DOCTYPE html>
<meta charset="utf-8">
<title>Sankey Experiment</title>
<style>

.node rect {
  cursor: move;
  fill-opacity: 9;
  shape-rendering: crispEdges;
}
  
.node text {
  pointer-events: none;
  text-shadow: 0 1px 0 #fff;
  color: green
}

.link {
  fill: none;
  stroke: #000;
  stroke-opacity: .3;
}

.link:hover {
  stroke-opacity: .5;
}

</style>
<body>
  <p align="right"><b>Sample 2</b></p> 
  <button type="button" id="start" onclick="window.location.reload()">Start</button>
<script src="https://d3js.org/d3.v4.min.js"></script>
<script src="sankey.js"></script>
<script>
	
// var units = "Widgets";

// set the dimensions and margins of the graph
var margin = {top: 10, right: 10, bottom:10, left: 20},
    width = 750 - margin.left - margin.right,
    height = 450 - margin.top - margin.bottom;

// format variables
var formatNumber = d3.format(",.0f"),    // zero decimal places
//     format = function(d) { return formatNumber(d) + " " + units; },
    format = function(d) { return formatNumber(d) + " ";},
    color = d3.scaleOrdinal(d3.schemeCategory10);
//

// 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(19)
    .nodePadding(32)
    .size([width, height]);

var path = sankey.link();

// load the data
d3.csv("sample2.csv", function(error, data) {
 
  //set up graph in same style as original example but empty
  graph = {"nodes" : [], "links" : []};

data.forEach(function (d) {
    graph.nodes.push({ "name": d.source });
    graph.nodes.push({ "name": d.target });
    graph.links.push({ "source": d.source,
                       "target": d.target,
                       "value": +d.value });
   });

// return only the distinct / unique nodes
  graph.nodes = d3.keys(d3.nest()
    .key(function (d) { return d.name; })
    .object(graph.nodes));

// loop through each link replacing the text with its index from node
  graph.links.forEach(function (d, i) {
    graph.links[i].source = graph.nodes.indexOf(graph.links[i].source);
    graph.links[i].target = graph.nodes.indexOf(graph.links[i].target);
  });

// now loop through each nodes to make nodes an array of objects
  // rather than an array of strings
  graph.nodes.forEach(function (d, i) {
    graph.nodes[i] = { "name": d };
  });

sankey
      .nodes(graph.nodes)
      .links(graph.links)
      .layout(32);

//   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); })
      .sort(function(a, b) { return b.dy - a.dy; });

// add the link titles
  link.append("title")
        .text(function(d) {
    		return d.source.name + " → " + 
                d.target.name; });
//              d.target.name + "\n" + format(d.value); });

// graph.links[12].source.x = 140;
// graph.links[13].source.x = 155;
// graph.links[14].target.x = 160;
// graph.links[15].target.x = 175;
// graph.links[9].target.x = 185;

//   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("click", click)
        .on("drag", dragmove))
  			.on("click", fade(0.2))
  			.on("mouseover", fade(.02)) 
			.on("mouseout", fade(1));

//   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 + ")"; })
//     .on("click",function(d){
//         if (d3.event.defaultPrevented) return;
//         alert("clicked!"+d.value);
//     })
   
  
  
  
  
  
  
  
  // add the rectangles for the nodes
  node.append("rect")
      .attr("height", function(d) { return d.dy; })
      .attr("width", sankey.nodeWidth())
      .style("fill", function(d) { 
		  return d.color = color(d.name.replace(/ .*/, "")); })
      .style("stroke", function(d) { 
		  return d3.rgb(d.color).darker(2); })
    .append("title")
      .text(function(d) { 
		  return d.name; });
//   		  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(function(d) { return d.name; })
    .filter(function(d) { return d.x < width / 2; })
      .attr("x", 6 + sankey.nodeWidth())
      .attr("text-anchor", "start");

// Returns an event handler for fading a given chord group.
// https://bl.ocks.org/mbostock/4062006
function fade(opacity) {
  return function(g, i) {
    var elements = svg.selectAll(".node");
    elements = elements.filter(function(d) { return d.name != graph.nodes[i].name });
    elements.transition()
        .style("opacity", opacity);

svg.selectAll(".link")
//         .filter(function(d) { return d.source.name != graph.nodes[i].name && d.target.name != graph.nodes[i].name })
    .filter(function(d) { return d.source.name != graph.nodes[i].name && d.target.name != graph.nodes[i].name })
    
    
    
      .transition()
        .style("opacity", opacity);
  };
}	
  
  
  
  
  
  
  
  
  // 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);
  }
});

</script>

</body>

https://d3js.org/d3.v4.min.js