Skip to content
Snippets Groups Projects
hierarchical_edge_bundling.html.in 4.97 KiB
Newer Older
<!DOCTYPE html>
<meta charset="utf-8">
<style>

.node {
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	font: 300 11px "Helvetica Neue", Helvetica, Arial, sans-serif;
	fill: #bbb;
}

.node:hover {
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	fill: #000;
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	stroke: steelblue;
	stroke-opacity: 0.4;
	fill: none;
	pointer-events: none;
}

.node:hover,
.node--source,
.node--target {
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	font-weight: 700;
}

.node--source {
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	fill: #2ca02c;
}

.node--target {
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	fill: #d62728;
}

.link--source,
.link--target {
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	stroke-opacity: 1;
	stroke-width: 2px;
}

.link--source {
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	stroke: #d62728;
}

.link--target {
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	stroke: #2ca02c;
}

</style>
<body style="text-align: center;">
<script>@D3_SCRIPT@</script>
<script>

///////////////////////////// Parse input
var tension = @TENSION@;

var diameter = @WIDTH@,
    radius = diameter / 2,
    innerRadius = radius - @PADDING@;

var root = packageHierarchy(JSON.parse('@JSON_STRING@'))
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	.sum(function(d) { return d.size; });
var cluster = d3.cluster()
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	.size([360, innerRadius]);
cluster(root);

var imports = packageImports(root.leaves());



///////////////////////////// Create canvas
var link, node; // Required by callbacks.
var svg = d3.select("body").append("svg")
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	.call(d3.zoom().scaleExtent([0.1, 10]).on("zoom", function () {
		svg.attr("transform",
		"translate(" +
			(d3.event.transform.applyX(radius)) + "," +
			(d3.event.transform.applyY(radius)) + ") " +
		"scale(" + d3.event.transform.k + ")")
	}))
	.attr("width", diameter)
	.attr("height", diameter)
	.append("g")
	.attr("transform", "translate(" + radius + "," + radius + ")");
update();



///////////////////////////// Add user interaction
document.onkeydown = function(e) {
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	switch (e.keyCode) {
		case 48: // 0
			tension = 0.0
			update();
			break;
		case 49: // 1
			tension = 0.1
			update();
			break;
		case 50: // 2
			tension = 0.2
			update();
			break;
		case 51: // 3
			tension = 0.3
			update();
			break;
		case 52: // 4
			tension = 0.4
			update();
			break;
		case 53: // 5
			tension = 0.5
			update();
			break;
		case 54: // 6
			tension = 0.6
			update();
			break;
		case 55: // 7
			tension = 0.7
			update();
			break;
		case 56: // 8
			tension = 0.8
			update();
			break;
		case 57: // 9
			tension = 0.9
			update();
			break;
		case 173: // -
			tension = 1.0
			update();
			break;
		case 87: // w
			tension = Math.min(tension + 0.01, 1);
			update();
			break;
		case 83: // s
			tension = Math.max(tension - 0.01, 0);
			update();
			break;
	}
};



///////////////////////////// Callbacks
function update() {
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	d3.selectAll("g > *").remove();
	svg.append("g")

	var line = d3.radialLine()
		.curve(d3.curveBundle.beta(tension))
		.radius(function(d) { return d.y; })
		.angle(function(d) { return d.x / 180 * Math.PI; });

	link = svg.append("g").selectAll(".link"),
	node = svg.append("g").selectAll(".node");

	link = link
		.data(imports)
		.enter().append("path")
		.each(function(d) { d.source = d[0], d.target = d[d.length - 1]; })
		.attr("class", "link")
		.attr("d", line);

	node = node
		.data(root.leaves())
		.enter().append("text")
		.attr("class", "node")
		.attr("dy", "0.31em")
		.attr("transform", function(d) { return "rotate(" + (d.x - 90) + ")translate(" + (d.y + 8) + ",0)" + (d.x < 180 ? "" : "rotate(180)"); })
		.attr("text-anchor", function(d) { return d.x < 180 ? "start" : "end"; })
		.text(function(d) { return d.data.key; })
		.on("mouseover", mouseovered)
		.on("mouseout", mouseouted);
}

function mouseovered(d) {
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	node
		.each(function(n) { n.target = n.source = false; });

	link
		.classed("link--target", function(l) { if (l.target === d) return l.source.source = true; })
		.classed("link--source", function(l) { if (l.source === d) return l.target.target = true; })
		.filter(function(l) { return l.target === d || l.source === d; })
		.raise();

	node
		.classed("node--target", function(n) { return n.target; })
		.classed("node--source", function(n) { return n.source; });
}

function mouseouted(d) {
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	link
		.classed("link--target", false)
		.classed("link--source", false);
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	node
		.classed("node--target", false)
		.classed("node--source", false);
}

// Lazily construct the package hierarchy from class names.
function packageHierarchy(classes) {
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	var map = {};

	function find(name, data) {
		var node = map[name], i;
		if (!node) {
			node = map[name] = data || {name: name, children: []};
			if (name.length) {
				node.parent = find(name.substring(0, i = name.lastIndexOf(".")));
				node.parent.children.push(node);
				node.key = name.substring(i + 1);
			}
		}
		return node;
	}

	classes.forEach(function(d) {
		find(d.name, d);
	});

	return d3.hierarchy(map[""]);
}

// Return a list of imports for the given array of nodes.
function packageImports(nodes) {
Marcel Steinbeck's avatar
Marcel Steinbeck committed
	var map = {},
	    imports = [];

	// Compute a map from name to node.
	nodes.forEach(function(d) {
		map[d.data.name] = d;
	});

	// For each import, construct a link from the source to target node.
	nodes.forEach(function(d) {
		if (d.data.imports) d.data.imports.forEach(function(i) {
			if (map[i]) {
				imports.push(map[d.data.name].path(map[i]));
			}
		});
	});

	return imports;