dev-chiefworks
327 lines
10 KiB
JavaScript
327 lines
10 KiB
JavaScript
/* ------------------------------------------------------------------------------
|
|
*
|
|
* # D3.js - bracket tree layout
|
|
*
|
|
* Demo of double sided bracket layout setup with pan and zoom
|
|
*
|
|
* Version: 1.0
|
|
* Latest update: August 1, 2015
|
|
*
|
|
* ---------------------------------------------------------------------------- */
|
|
|
|
$(function () {
|
|
|
|
// Initialize chart
|
|
brackets('#d3-tree-bracket', 600);
|
|
|
|
// Chart setup
|
|
function brackets(element, height) {
|
|
|
|
|
|
// Basic setup
|
|
// ------------------------------
|
|
|
|
// Define main variables
|
|
var d3Container = d3.select(element),
|
|
margin = {top: 0, right: 0, bottom: 0, left: 0},
|
|
width = d3Container.node().getBoundingClientRect().width - margin.left - margin.right,
|
|
halfWidth = width / 2,
|
|
height = height - margin.top - margin.bottom - 5,
|
|
i = 0,
|
|
duration = 500,
|
|
root;
|
|
|
|
|
|
|
|
// Create chart
|
|
// ------------------------------
|
|
|
|
// Add SVG element
|
|
var container = d3Container.append("svg");
|
|
|
|
// Add SVG group
|
|
var svg = container
|
|
.attr("width", width + margin.left + margin.right)
|
|
.attr("height", height + margin.top + margin.bottom)
|
|
.append("g")
|
|
.attr("transform", "translate(" + margin.left + "," + margin.top + ")");
|
|
|
|
|
|
// Get children
|
|
var getChildren = function(d) {
|
|
var a = [];
|
|
if(d.winners) for(var i = 0; i < d.winners.length; i++){
|
|
d.winners[i].isRight = false;
|
|
d.winners[i].parent = d;
|
|
a.push(d.winners[i]);
|
|
}
|
|
if(d.challengers) for(var i = 0; i < d.challengers.length; i++){
|
|
d.challengers[i].isRight = true;
|
|
d.challengers[i].parent = d;
|
|
a.push(d.challengers[i]);
|
|
}
|
|
return a.length?a:null;
|
|
};
|
|
|
|
|
|
|
|
// Add zoom behavior
|
|
// ------------------------------
|
|
|
|
// Add zoom with scale
|
|
var zoom = d3.behavior.zoom()
|
|
.scaleExtent([1,2])
|
|
.on('zoom', function(){
|
|
svg.attr("transform", "translate(" + d3.event.translate + ") scale(" + d3.event.scale + ")");
|
|
});
|
|
|
|
// Initialize zoom
|
|
container.call(zoom);
|
|
|
|
|
|
|
|
// Construct chart layout
|
|
// ------------------------------
|
|
|
|
// Tree
|
|
var tree = d3.layout.tree()
|
|
.size([height, width]);
|
|
|
|
// Diagonal projection
|
|
var diagonal = d3.svg.diagonal()
|
|
.projection(function(d) { return [d.y, d.x]; });
|
|
|
|
|
|
|
|
// Helper functions
|
|
// ------------------------------
|
|
|
|
// Connector
|
|
var elbow = function (d, i){
|
|
var source = calcLeft(d.source),
|
|
target = calcLeft(d.target),
|
|
hy = (target.y-source.y) / 2;
|
|
|
|
if(d.isRight) hy = -hy;
|
|
return "M" + source.y + "," + source.x + "H" + (source.y + hy) + "V" + target.x + "H" + target.y;
|
|
};
|
|
var connector = elbow;
|
|
|
|
// Calculate horizontal position
|
|
var calcLeft = function(d) {
|
|
var l = d.y;
|
|
if(!d.isRight) {
|
|
l = d.y-halfWidth;
|
|
l = halfWidth - l;
|
|
}
|
|
return {x : d.x, y : l};
|
|
};
|
|
|
|
var toArray = function(item, arr){
|
|
arr = arr || [];
|
|
var i = 0,
|
|
l = item.children?item.children.length : 0;
|
|
|
|
arr.push(item);
|
|
for(; i < l; i++) {
|
|
toArray(item.children[i], arr);
|
|
}
|
|
return arr;
|
|
};
|
|
|
|
|
|
|
|
// Load data
|
|
// ------------------------------
|
|
|
|
d3.json("assets/demo_data/d3/tree/tree_bracket.json", function(json) {
|
|
root = json;
|
|
root.x0 = height / 2;
|
|
root.y0 = width / 2;
|
|
|
|
// Add tree layout
|
|
var t1 = d3.layout.tree().size([height, halfWidth]).children(function(d){return d.winners;}),
|
|
t2 = d3.layout.tree().size([height, halfWidth]).children(function(d){return d.challengers;});
|
|
t1.nodes(root);
|
|
t2.nodes(root);
|
|
|
|
// Rebuild children nodes
|
|
var rebuildChildren = function(node){
|
|
node.children = getChildren(node);
|
|
if(node.children) node.children.forEach(rebuildChildren);
|
|
}
|
|
rebuildChildren(root);
|
|
root.isRight = false;
|
|
update(root);
|
|
});
|
|
|
|
|
|
|
|
// Layout setup
|
|
// ------------------------------
|
|
|
|
// Update nodes
|
|
function update(source) {
|
|
|
|
// Compute the new tree layout.
|
|
var nodes = toArray(source);
|
|
|
|
// Normalize for fixed-depth.
|
|
nodes.forEach(function(d) { d.y = d.depth * 180 + halfWidth; });
|
|
|
|
// Update the nodes…
|
|
var node = svg.selectAll("g.node")
|
|
.data(nodes, function(d) { return d.id || (d.id = ++i); });
|
|
|
|
// Stash the old positions for transition.
|
|
nodes.forEach(function(d) {
|
|
var p = calcLeft(d);
|
|
d.x0 = p.x;
|
|
d.y0 = p.y;
|
|
});
|
|
|
|
|
|
// Enter nodes
|
|
// ------------------------------
|
|
|
|
// Enter any new nodes at the parent's previous position.
|
|
var nodeEnter = node.enter().append("g")
|
|
.attr("class", "node")
|
|
.attr("transform", function(d) { return "translate(" + source.y0 + "," + source.x0 + ")"; })
|
|
.on("click", click);
|
|
|
|
// Add node circles
|
|
nodeEnter.append("circle")
|
|
.attr("r", 1e-6)
|
|
.style("stroke", "#546E7A")
|
|
.style("stroke-width", 1.5)
|
|
.style("cursor", "pointer")
|
|
.style("fill", function(d) { return d._children ? "#546E7A" : "#fff"; });
|
|
|
|
// Add node text
|
|
nodeEnter.append("text")
|
|
.attr("dy", function(d) { return d.isRight?18:-12;})
|
|
.attr("text-anchor", "middle")
|
|
.text(function(d) { return d.name; })
|
|
.style("font-size", 12)
|
|
.style("fill-opacity", 1e-6);
|
|
|
|
|
|
// Update nodes
|
|
// ------------------------------
|
|
|
|
// Transition nodes to their new position.
|
|
var nodeUpdate = node.transition()
|
|
.duration(duration)
|
|
.attr("transform", function(d) { p = calcLeft(d); return "translate(" + p.y + "," + p.x + ")"; });
|
|
|
|
// Update circle
|
|
nodeUpdate.select("circle")
|
|
.attr("r", 4.5)
|
|
.style("fill", function(d) { return d._children ? "#546E7A" : "#fff"; });
|
|
|
|
// Update text
|
|
nodeUpdate.select("text")
|
|
.style("fill-opacity", 1);
|
|
|
|
|
|
// Exit nodes
|
|
// ------------------------------
|
|
|
|
// Transition exiting nodes to the parent's new position.
|
|
var nodeExit = node.exit().transition()
|
|
.duration(duration)
|
|
.attr("transform", function(d) { p = calcLeft(d.parent||source); return "translate(" + p.y + "," + p.x + ")"; })
|
|
.remove();
|
|
|
|
// Update circles
|
|
nodeExit.select("circle")
|
|
.attr("r", 1e-6);
|
|
|
|
// Update text
|
|
nodeExit.select("text")
|
|
.style("fill-opacity", 1e-6);
|
|
|
|
|
|
|
|
// Links
|
|
// ------------------------------
|
|
|
|
// Update the links
|
|
var link = svg.selectAll("path.link")
|
|
.data(tree.links(nodes), function(d) { return d.target.id; });
|
|
|
|
// Enter any new links at the parent's previous position
|
|
link.enter().insert("path", "g")
|
|
.attr("class", "link")
|
|
.style("stroke", "#546E7A")
|
|
.style("fill", "none")
|
|
.style("stroke-width", 1.5)
|
|
.attr("d", function(d) {
|
|
var o = {x: source.x0, y: source.y0};
|
|
return connector({source: o, target: o});
|
|
});
|
|
|
|
// Transition links to their new position
|
|
link.transition()
|
|
.duration(duration)
|
|
.attr("d", connector);
|
|
|
|
// Transition exiting nodes to the parent's new position
|
|
link.exit().transition()
|
|
.duration(duration)
|
|
.attr("d", function(d) {
|
|
var o = calcLeft(d.source||source);
|
|
if(d.source.isRight) o.y -= halfWidth - (d.target.y - d.source.y);
|
|
else o.y += halfWidth - (d.target.y - d.source.y);
|
|
return connector({source: o, target: o});
|
|
})
|
|
.remove();
|
|
|
|
|
|
|
|
// Toggle children on click.
|
|
function click(d) {
|
|
if (d.children) {
|
|
d._children = d.children;
|
|
d.children = null;
|
|
} else {
|
|
d.children = d._children;
|
|
d._children = null;
|
|
}
|
|
update(source);
|
|
}
|
|
|
|
|
|
// Resize chart
|
|
// ------------------------------
|
|
|
|
// Call function on window resize
|
|
$(window).on('resize', resize);
|
|
|
|
// Call function on sidebar width change
|
|
$('.sidebar-control').on('click', resize);
|
|
|
|
|
|
// Resize function
|
|
//
|
|
// Since D3 doesn't support SVG resize by default,
|
|
// we need to manually specify parts of the graph that need to
|
|
// be updated on window resize
|
|
function resize() {
|
|
|
|
// Layout variables
|
|
width = d3Container.node().getBoundingClientRect().width - margin.left - margin.right,
|
|
|
|
// Layout
|
|
// -------------------------
|
|
|
|
// Main svg width
|
|
container.attr("width", width + margin.left + margin.right);
|
|
|
|
// Width of appended group
|
|
svg.attr("width", width + margin.left + margin.right);
|
|
}
|
|
}
|
|
}
|
|
}); |