alicestrt
4 years ago
3 changed files with 4796 additions and 2 deletions
@ -0,0 +1,149 @@ |
|||||
|
var CodeFlower = function(selector, w, h) { |
||||
|
this.w = w; |
||||
|
this.h = h; |
||||
|
|
||||
|
d3.select(selector).selectAll("svg").remove(); |
||||
|
|
||||
|
this.svg = d3.select(selector).append("svg:svg") |
||||
|
.attr('width', w) |
||||
|
.attr('height', h); |
||||
|
|
||||
|
this.svg.append("svg:rect") |
||||
|
.style("stroke", "#999") |
||||
|
.style("fill", "#fff") |
||||
|
.attr('width', w) |
||||
|
.attr('height', h); |
||||
|
|
||||
|
this.force = d3.layout.force() |
||||
|
.on("tick", this.tick.bind(this)) |
||||
|
.charge(function(d) { return d._children ? -d.size / 100 : -40; }) |
||||
|
.linkDistance(function(d) { return d.target._children ? 80 : 25; }) |
||||
|
.size([h, w]); |
||||
|
}; |
||||
|
|
||||
|
CodeFlower.prototype.update = function(json) { |
||||
|
if (json) this.json = json; |
||||
|
|
||||
|
this.json.fixed = true; |
||||
|
this.json.x = this.w / 2; |
||||
|
this.json.y = this.h / 2; |
||||
|
|
||||
|
var nodes = this.flatten(this.json); |
||||
|
var links = d3.layout.tree().links(nodes); |
||||
|
var total = nodes.length || 1; |
||||
|
|
||||
|
// remove existing text (will readd it afterwards to be sure it's on top)
|
||||
|
this.svg.selectAll("text").remove(); |
||||
|
|
||||
|
// Restart the force layout
|
||||
|
this.force |
||||
|
.gravity(Math.atan(total / 50) / Math.PI * 0.4) |
||||
|
.nodes(nodes) |
||||
|
.links(links) |
||||
|
.start(); |
||||
|
|
||||
|
// Update the links
|
||||
|
this.link = this.svg.selectAll("line.link") |
||||
|
.data(links, function(d) { return d.target.name; }); |
||||
|
|
||||
|
// Enter any new links
|
||||
|
this.link.enter().insert("svg:line", ".node") |
||||
|
.attr("class", "link") |
||||
|
.attr("x1", function(d) { return d.source.x; }) |
||||
|
.attr("y1", function(d) { return d.source.y; }) |
||||
|
.attr("x2", function(d) { return d.target.x; }) |
||||
|
.attr("y2", function(d) { return d.target.y; }); |
||||
|
|
||||
|
// Exit any old links.
|
||||
|
this.link.exit().remove(); |
||||
|
|
||||
|
// Update the nodes
|
||||
|
this.node = this.svg.selectAll("circle.node") |
||||
|
.data(nodes, function(d) { return d.name; }) |
||||
|
.classed("collapsed", function(d) { return d._children ? 1 : 0; }); |
||||
|
|
||||
|
this.node.transition() |
||||
|
.attr("r", function(d) { return d.children ? 3.5 : Math.pow(d.size, 2/5) || 1; }); |
||||
|
|
||||
|
// Enter any new nodes
|
||||
|
this.node.enter().append('svg:circle') |
||||
|
.attr("class", "node") |
||||
|
.classed('directory', function(d) { return (d._children || d.children) ? 1 : 0; }) |
||||
|
.attr("r", function(d) { return d.children ? 3.5 : Math.pow(d.size, 2/5) || 1; }) |
||||
|
.style("fill", function color(d) { |
||||
|
return "hsl(" + parseInt(360 / total * d.id, 10) + ",90%,70%)"; |
||||
|
}) |
||||
|
.call(this.force.drag) |
||||
|
.on("click", this.click.bind(this)) |
||||
|
.on("mouseover", this.mouseover.bind(this)) |
||||
|
.on("mouseout", this.mouseout.bind(this)); |
||||
|
|
||||
|
// Exit any old nodes
|
||||
|
this.node.exit().remove(); |
||||
|
|
||||
|
this.text = this.svg.append('svg:text') |
||||
|
.attr('class', 'nodetext') |
||||
|
.attr('dy', 0) |
||||
|
.attr('dx', 0) |
||||
|
.attr('text-anchor', 'middle'); |
||||
|
|
||||
|
return this; |
||||
|
}; |
||||
|
|
||||
|
CodeFlower.prototype.flatten = function(root) { |
||||
|
var nodes = [], i = 0; |
||||
|
|
||||
|
function recurse(node) { |
||||
|
if (node.children) { |
||||
|
node.size = node.children.reduce(function(p, v) { |
||||
|
return p + recurse(v); |
||||
|
}, 0); |
||||
|
} |
||||
|
if (!node.id) node.id = ++i; |
||||
|
nodes.push(node); |
||||
|
return node.size; |
||||
|
} |
||||
|
|
||||
|
root.size = recurse(root); |
||||
|
return nodes; |
||||
|
}; |
||||
|
|
||||
|
CodeFlower.prototype.click = function(d) { |
||||
|
// Toggle children on click.
|
||||
|
if (d.children) { |
||||
|
d._children = d.children; |
||||
|
d.children = null; |
||||
|
} else { |
||||
|
d.children = d._children; |
||||
|
d._children = null; |
||||
|
} |
||||
|
this.update(); |
||||
|
}; |
||||
|
|
||||
|
CodeFlower.prototype.mouseover = function(d) { |
||||
|
this.text.attr('transform', 'translate(' + d.x + ',' + (d.y - 5 - (d.children ? 3.5 : Math.sqrt(d.size) / 2)) + ')') |
||||
|
.text(d.name + ": " + d.size + " loc") |
||||
|
.style('display', null); |
||||
|
}; |
||||
|
|
||||
|
CodeFlower.prototype.mouseout = function(d) { |
||||
|
this.text.style('display', 'none'); |
||||
|
}; |
||||
|
|
||||
|
CodeFlower.prototype.tick = function() { |
||||
|
var h = this.h; |
||||
|
var w = this.w; |
||||
|
this.link.attr("x1", function(d) { return d.source.x; }) |
||||
|
.attr("y1", function(d) { return d.source.y; }) |
||||
|
.attr("x2", function(d) { return d.target.x; }) |
||||
|
.attr("y2", function(d) { return d.target.y; }); |
||||
|
|
||||
|
this.node.attr("transform", function(d) { |
||||
|
return "translate(" + Math.max(5, Math.min(w - 5, d.x)) + "," + Math.max(5, Math.min(h - 5, d.y)) + ")"; |
||||
|
}); |
||||
|
}; |
||||
|
|
||||
|
CodeFlower.prototype.cleanup = function() { |
||||
|
this.update([]); |
||||
|
this.force.stop(); |
||||
|
}; |
File diff suppressed because it is too large
Loading…
Reference in new issue