Skip to content

Instantly share code, notes, and snippets.

@cgroll
Last active August 29, 2015 14:16
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save cgroll/3234c697c9d611f9bfc4 to your computer and use it in GitHub Desktop.
Save cgroll/3234c697c9d611f9bfc4 to your computer and use it in GitHub Desktop.
Tree layout

First attempt to visualize possible conditional probability integral transforms through trees.

// define margins
var margin = {top: 20, right: 80, bottom: 30, left: 150};
var nodeRadius = 18;
// graphics size without axis
var width = 960 - margin.left - margin.right;
var height = 500 - margin.top - margin.bottom;
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 + ")");
// var treeData = {
// name: "3",
// copulaLink: [
// {
// name: "2",
// copulaLink: [
// { name: "1" },
// { name: "5" },
// ]
// },
// {
// name: "4",
// copulaLink: [
// ]
// },
// {
// name: "6",
// copulaLink: [
// { name: "7" },
// ]
// }
// ]
// };
var data = [
{ "name": "3", "parent": "null" },
{ "name": "2", "parent": "3" },
{ "name": "1", "parent": "2" },
{ "name": "5", "parent": "2" },
{ "name": "6", "parent": "3" },
{ "name": "4", "parent": "3" },
{ "name": "7", "parent": "6" }
]
var dataMap = data.reduce(function(map, node) {
map[node.name] = node;
return map;
}, {});
var treeData = [];
data.forEach(function(node) {
// add to parent
var parent = dataMap[node.parent];
if (parent) {
// create child array if it doesn't exist
(parent.copulaLink || (parent.copulaLink = []))
// add node to child array
.push(node);
} else {
// parent is null or missing
treeData.push(node);
}
});
var tree = d3.layout.tree()
.sort(function comparator(a, b) {
return +a.name - +b.name;
})
// .size([height, width - maxLabelLength*options.fontSize])
.size([height, width])
.children(function(d)
{
return (!d.copulaLink || d.copulaLink.length === 0) ? null : d.copulaLink;
});
var nodes = tree.nodes(treeData[0]);
var links = tree.links(nodes);
nodes.forEach(function(d) { d.y = d.depth * 200; });
var link = d3.svg.diagonal()
.projection(function(d)
{
return [d.x, d.y];
});
svg.selectAll("path.link")
.data(links)
.enter()
.append("svg:path")
.attr("class", "link")
.attr("d", link);
var nodeGroup = svg.selectAll("g.node")
.data(nodes)
.enter()
.append("svg:g")
.attr("class", "node")
.attr("transform", function(d)
{
return "translate(" + d.x + "," + d.y + ")";
});
nodeGroup.append("svg:circle")
.attr("class", "node-dot")
.attr("r", nodeRadius)
nodeGroup.append("svg:text")
.attr("text-anchor", "middle")
.attr("alignment-baseline", "central")
.text(function(d) { return d.name; });
// nodeGroup.append("svg:text")
// // .attr("text-anchor", function(d)
// // {
// // return d.children ? "end" : "start";
// // })
// // .attr("dx", function(d)
// // {
// // var gap = 2 * nodeRadius;
// // return d.children ? -gap : gap;
// // })
// .attr("dy", 3)
// .text(function(d)
// {
// return d.name;
// });
<!DOCTYPE html>
<meta charset="utf-8">
<style>
body {
font: 16px sans-serif;
}
.node circle {
fill: #fff;
stroke: steelblue;
stroke-width: 3px;
}
.node text { font: 16px sans-serif; }
.link {
fill: none;
stroke: #ccc;
stroke-width: 2px;
}
</style>
<body>
<script src="http://d3js.org/d3.v3.min.js" charset="utf-8">
</script>
<script src="assembly_code.js">
</script>
</body>
</html>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment