Skip to content

Instantly share code, notes, and snippets.

@tejaser

tejaser/.block

Last active January 3, 2018 08:37
Show Gist options
  • Save tejaser/c74b971b4e69e7e05c3e2e72935ba45e to your computer and use it in GitHub Desktop.
Save tejaser/c74b971b4e69e7e05c3e2e72935ba45e to your computer and use it in GitHub Desktop.
Download ZIP
Zoomable Treemaps w/ Color1
Raw
.block
license: mit
Raw
index.html
<!DOCTYPE html>
<meta charset="utf-8">
<title>Zoomable Treemaps with Color</title>
<style>
@import url(../style.css?aea6f0a);
#chart {
width: 960px;
height: 500px;
}
text {
pointer-events: none;
}
.grandparent text {
font-weight: bold;
}
rect {
/* fill: none; */
stroke: #fff;
stroke-width: 1px;
}
rect.parent,
.grandparent rect {
stroke-width: 2px;
}
.grandparent:hover rect {
fill: darkgrey;
}
.children rect.parent,
.grandparent rect {
cursor: pointer;
}
.children rect.child {
opacity: 0;
}
.children rect.parent {
}
.children:hover rect.child {
opacity: 1;
stroke-width: 1px;
}
.children:hover rect.parent {
opacity: 0;
}
.legend rect {
stroke-width: 0px;
}
.legend text {
text-anchor: middle;
pointer-events: auto;
font-size: 15px;
font-family: sans-serif;
fill: black;
}
</style>
<header>
<aside>Adapted from Mike Bostock's <a href="http://bost.ocks.org/mike/treemap/">Zoomable Treemaps</a></aside>
</header>
<h1>Zoomable Treemaps, with Color</h1>
<p>Hover on any cell to see next level of detail, or click on a cell to zoom in. Click on the top label to zoom out.</p>
<p id="chart">
<h3 style="font-weight: bold">Legend</h3>
<p style="width: 960px">Color: Percent change in number of cases of TB from 2006 to 2012 (negative is good).</p>
<div id="legend"></div>
<p>Size of box: number of TB cases in 2012.</p>
<footer>
<h3>References</h3>
<aside>Data from US Center for Disease Control <a href="http://wonder.cdc.gov/TB-v2012.html">Wonder database</a>.</aside>
<aside>Original <a href="http://bost.ocks.org/mike/treemap/">treemap</a> created June, 2012 by Mike Bostock. Adapted by Zan Armstrong in Sept 2014.</aside>
</footer>
<script src="http://d3js.org/d3.v3.min.js"></script>
<script>
var margin = {top: 30, right: 0, bottom: 20, left: 0},
width = 960,
height = 500 - margin.top - margin.bottom,
formatNumber = d3.format(",%"),
colorDomain = [-.6, 0, .6],
colorRange = ["#373a93", 'white', "#936638"],
transitioning;
// sets x and y scale to determine size of visible boxes
var x = d3.scale.linear()
.domain([0, width])
.range([0, width]);
var y = d3.scale.linear()
.domain([0, height])
.range([0, height]);
// adding a color scale
var color = d3.scale.linear()
.domain(colorDomain)
.range(colorRange);
// introduce color scale here
var treemap = d3.layout.treemap()
.children(function(d, depth) { return depth ? null : d._children; })
.sort(function(a, b) { return a.value - b.value; })
.ratio(height / width * 0.5 * (1 + Math.sqrt(5)))
.round(false);
var svg = d3.select("#chart").append("svg")
.attr("width", width + margin.left + margin.right)
.attr("height", height + margin.bottom + margin.top)
.style("margin-left", -margin.left + "px")
.style("margin.right", -margin.right + "px")
.append("g")
.attr("transform", "translate(" + margin.left + "," + margin.top + ")")
.style("shape-rendering", "crispEdges");
var grandparent = svg.append("g")
.attr("class", "grandparent");
grandparent.append("rect")
.attr("y", -margin.top)
.attr("width", width)
.attr("height", margin.top);
grandparent.append("text")
.attr("x", 6)
.attr("y", 6 - margin.top)
.attr("dy", ".75em");
var legend = d3.select("#legend").append("svg")
.attr("width", width + margin.left + margin.right)
.attr("height", 30)
.attr('class', 'legend')
.selectAll("g")
.data([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18])
.enter()
.append('g')
// functions
function initialize(root) {
root.x = root.y = 0;
root.dx = width;
root.dy = height;
root.depth = 0;
}
// Aggregate the values for internal nodes. This is normally done by the
// treemap layout, but not here because of our custom implementation.
// We also take a snapshot of the original children (_children) to avoid
// the children being overwritten when when layout is computed.
function accumulate(d) {
return (d._children = d.children)
// recursion step, note that p and v are defined by reduce
? d.value = d.children.reduce(function(p, v) {return p + accumulate(v); }, 0)
: d.value;
}
// Compute the treemap layout recursively such that each group of siblings
// uses the same size (1×1) rather than the dimensions of the parent cell.
// This optimizes the layout for the current zoom state. Note that a wrapper
// object is created for the parent node for each group of siblings so that
// the parent’s dimensions are not discarded as we recurse. Since each group
// of sibling was laid out in 1×1, we must rescale to fit using absolute
// coordinates. This lets us use a viewport to zoom.
function layout(d) {
if (d._children) {
// treemap nodes comes from the treemap set of functions as part of d3
treemap.nodes({_children: d._children});
d._children.forEach(function(c) {
c.x = d.x + c.x * d.dx;
c.y = d.y + c.y * d.dy;
c.dx *= d.dx;
c.dy *= d.dy;
c.parent = d;
// recursion
layout(c);
});
}
}
function colorIncrements(d){
return (colorDomain[colorDomain.length - 1] - colorDomain[0])/18*d + colorDomain[0];
}
legend.append("rect")
.attr("x", function(d){return margin.left + d * 40})
.attr("y", 0)
.attr("fill", function(d) {return color(colorIncrements(d))})
.attr('width', '40px')
.attr('height', '40px')
legend.append("text")
.text(function(d){return formatNumber(colorIncrements(d))})
.attr('y', 20)
.attr('x', function(d){return margin.left + d * 40 + 20});
// determines if white or black will be better contrasting color
function getContrast50(hexcolor){
return (parseInt(hexcolor.replace('#', ''), 16) > 0xffffff/3) ? 'black':'white';
}
d3.json("TBchangeInCases2006to2012.json", function(root) {
console.log(root)
initialize(root);
accumulate(root);
layout(root);
display(root);
function display(d) {
grandparent
.datum(d.parent)
.on("click", transition)
.select("text")
.text(name(d))
// color header based on grandparent's rate
grandparent
.datum(d.parent)
.select("rect")
.attr("fill", function(){console.log(color(d.rate)); return color(d['rate'])})
var g1 = svg.insert("g", ".grandparent")
.datum(d)
.attr("class", "depth");
var g = g1.selectAll("g")
.data(d._children)
.enter().append("g");
g.filter(function(d) { return d._children; })
.classed("children", true)
.on("click", transition);
g.selectAll(".child")
.data(function(d) { return d._children || [d]; })
.enter().append("rect")
.attr("class", "child")
.call(rect);
g.append("rect")
.attr("class", "parent")
.call(rect)
.append("title")
.text(function(d) {console.log(typeof(d.value), d.value); return d.name + ', Cases of TB: ' + d.value + ', percent change: ' + formatNumber(d.rate); });
g.append("text")
.attr("dy", ".75em")
.text(function(d) { return d.name; })
.call(text);
function transition(d) {
if (transitioning || !d) return;
transitioning = true;
var g2 = display(d),
t1 = g1.transition().duration(750),
t2 = g2.transition().duration(750);
// Update the domain only after entering new elements.
x.domain([d.x, d.x + d.dx]);
y.domain([d.y, d.y + d.dy]);
// Enable anti-aliasing during the transition.
svg.style("shape-rendering", null);
// Draw child nodes on top of parent nodes.
svg.selectAll(".depth").sort(function(a, b) { return a.depth - b.depth; });
// Fade-in entering text.
g2.selectAll("text").style("fill-opacity", 0);
// Transition to the new view.
t1.selectAll("text").call(text).style("fill-opacity", 0);
t2.selectAll("text").call(text).style("fill-opacity", 1);
t1.selectAll("rect").call(rect);
t2.selectAll("rect").call(rect);
// Remove the old node when the transition is finished.
t1.remove().each("end", function() {
svg.style("shape-rendering", "crispEdges");
transitioning = false;
});
}
return g;
}
function text(text) {
text.attr("x", function(d) { return x(d.x) + 6; })
.attr("y", function(d) { return y(d.y) + 6; })
.attr("fill", function (d) {return getContrast50(color(parseFloat(d.rate)))});
}
function rect(rect) {
rect.attr("x", function(d) { return x(d.x); })
.attr("y", function(d) { return y(d.y); })
.attr("width", function(d) { return x(d.x + d.dx) - x(d.x); })
.attr("height", function(d) { return y(d.y + d.dy) - y(d.y); })
.attr("fill", function(d){return color(parseFloat(d.rate));});
}
function name(d) {
return d.parent
? name(d.parent) + "." + d.name
: d.name;
}
});
</script>
Raw
TBchangeInCases2006to2012.json
[{"name": "Ames Walker", "children": [{"name": "November", "children": [{"name": "2742"}]}, {"name": "October", "children": [{"name": "831"}]}]}, {"name": "Burgers Smokehouse", "children": [{"name": "November", "children": [{"name": "1057"}]}, {"name": "October", "children": [{"name": "347"}]}]}, {"name": "Circa Lighting", "children": [{"name": "November", "children": [{"name": "9954"}]}, {"name": "October", "children": [{"name": "5656"}]}]}, {"name": "Hanky Panky", "children": [{"name": "November", "children": [{"name": "26077"}]}, {"name": "October", "children": [{"name": "2910"}]}]}, {"name": "Kansas City Steak", "children": [{"name": "November", "children": [{"name": "7903"}]}, {"name": "October", "children": [{"name": "1418"}]}]}, {"name": "Kittenish", "children": [{"name": "November", "children": [{"name": "129"}]}, {"name": "October", "children": [{"name": "0"}]}]}, {"name": "Model Train Stuff", "children": [{"name": "November", "children": [{"name": "8675"}]}, {"name": "October", "children": [{"name": "9419"}]}]}]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment