Brush & Zoom with choropleth

Combining d3-brush and d3-zoom to implement Focus + Context. Another approach is to zoom to the brushed region.

forked from mbostock's block: Brush & Zoom

forked from eric-bunch's block: Brush & Zoom

<!DOCTYPE html>
<meta charset="utf-8">
<style>

.area {
  fill: steelblue;
/*   fill: none; */
  clip-path: url(#clip);
}
  
 .overlay {
  fill: none;
  pointer-events: all;
}
  
  .focus .area{
    fill: none;
  }
  
/*   .circ circle {
  fill: none;
  stroke: steelblue;
} */
  
.line {
  fill: none;
  clip-path: url(#clip);
}

.zoom {
  cursor: move;
  fill: none;
  pointer-events: all;
}
  
.counties {
  fill: none;
}

.states {
  fill: none;
  stroke: #fff;
  stroke-linejoin: round;
}

.q0-9 { fill:rgb(247,251,255); }
.q1-9 { fill:rgb(222,235,247); }
.q2-9 { fill:rgb(198,219,239); }
.q3-9 { fill:rgb(158,202,225); }
.q4-9 { fill:rgb(107,174,214); }
.q5-9 { fill:rgb(66,146,198); }
.q6-9 { fill:rgb(33,113,181); }
.q7-9 { fill:rgb(8,81,156); }
.q8-9 { fill:rgb(8,48,107); }

</style>
<svg class='mapsvg' width="960" height="500"></svg>
<svg class='graphsvg' width="960" height="500"></svg>
<script src="https://d3js.org/d3.v4.min.js"></script>
<script src="//d3js.org/topojson.v1.min.js"></script>

// var parseDate = d3.timeParse("%b %Y"),
  var parseTime = d3.timeParse("%Y%m%d"),
    bisectDate = d3.bisector(function(d) { return d.date; }).left,
    bisect = d3.bisector(function(d) { return parseTime(d.x); }).left;

var x = d3.scaleTime().range([0, width]),
    x2 = d3.scaleTime().range([0, width]),
    x3 = d3.scaleLinear().range([0,width]),
    y = d3.scaleLinear().range([height, 0]),
    y2 = d3.scaleLinear().range([height2, 0])
		z = d3.scaleOrdinal(d3.schemeCategory10);

var xAxis = d3.axisBottom(x),
    xAxis2 = d3.axisBottom(x2),
    yAxis = d3.axisLeft(y);

var zoom = d3.zoom()
    .scaleExtent([1, Infinity])
    .translateExtent([[0, 0], [width, height]])
    .extent([[0, 0], [width, height]])
    .on("zoom", zoomed);

// var area2 = d3.area()
//     .curve(d3.curveMonotoneX)
//     .x(function(d) { return x2(d.date); })
//     .y0(height2)
//     .y1(function(d) { return y2(d.price); });

var area2 = d3.area()
    .curve(d3.curveMonotoneX)
    .x(function(d) { return x2(parseTime(d.x)); })
    .y0(height2)
    .y1(function(d) { return y2(d.y); });

svg.append("defs").append("clipPath")
    .attr("id", "clip")
  .append("rect")
    .attr("width", width)
    .attr("height", height);

var focus = svg.append("g")
    .attr("class", "focus")
    .attr("transform", "translate(" + margin.left + "," + margin.top + ")");

d3.json("temperature3.json", function(error, data) {
//     console.log(data);
// d3.csv("sp500.csv", type, function(error, data) {
  if (error) throw error;
    
  
    
    
pts = [];
    
for(var i=0; i<data.length; i++){
  tmp = data[i].values.filter(function(x) {return x.hasOwnProperty('coords'); });
  
  for(var j=0; j<tmp.length; j++){
    for(var k=0; k<tmp[j].coords.length; k++){
      pts.push({
        	x: tmp[j].x,
        	coords: tmp[j].coords[k]
        }        
      );
    }
  }
  
//   pts = pts.concat(data[i].values.filter(function(x) {return x.hasOwnProperty('coords'); }));
}
    
// console.log(pts);
    
    
    
var rateById = d3.map();
  
var quantize = d3.scaleQuantize()
    .domain([0, 0.15])
    .range(d3.range(9).map(function(i) { return "q" + i + "-9"; }));

var projection = d3.geoAlbersUsa()
    .scale(1100)
    .translate([mapwidth / 2, mapheight / 2]);

var path = d3.geoPath()
    .projection(projection);

//   x.domain(d3.extent(data, function(d) { return d.date; }));
//   y.domain([0, d3.max(data, function(d) { return d.price; })]);
    
x.domain(d3.extent(data['0'].values, function(c){return parseTime(c.x)}));
  y.domain([
    d3.min(data, function(c) { return d3.min(c.values, function(d) { return d.y; }); }),
    d3.max(data, function(c) { return d3.max(c.values, function(d) { return d.y; }); })
  ]);    
  x2.domain(x.domain());
  y2.domain(y.domain());
    
    
  x3.domain([0, data[0].values.length - 1]);
    
  z.domain(data, function(c) { return c.id; });

//////// choropleth stuff vvvv

d3.queue()
    .defer(d3.json, "https://gist.githubusercontent.com/mbostock/4090846/raw/d534aba169207548a8a3d670c9c2cc719ff05c47/us.json")
    .defer(d3.tsv, "unemployment.tsv", function(d) { rateById.set(d.id, +d.rate); })
    .await(ready);
 
function ready(error, us) {
  if (error) throw error;

mapsvg.append("g")
      .attr("class", "counties")
    .selectAll("path")
      .data(topojson.feature(us, us.objects.counties).features)
    .enter().append("path")
//       .attr("class", function(d) { return quantize(rateById.get(d.id)); })
  		.attr("class", function(d) {return 'q3-9'} )
      .attr("d", path);

mapsvg.append("path")
      .datum(topojson.mesh(us, us.objects.states, function(a, b) { return a !== b; }))
      .attr("class", "states")
      .attr("d", path);
  
  
  mapsvg.selectAll('circle')
   .data(pts)
   .enter()
   .append("circle")
   .attr("cx", function (d) { return projection(d.coords)[0]; })
		.attr("cy", function (d) { return projection(d.coords)[1]; })
		.attr("r", 4)
    .style("opacity", .7)      // set the element opacity
    .style("stroke", "#9f5045")    // set the line colour
    .style("fill", "#F0C059");
  
  
    context.append("g")
      .attr("class", "brush")
      .call(brush)
      .call(brush.move, x.range());
  
  
}  
  
  
////// choropleth stuff ^^^

var brush = d3.brushX()
    .extent([[0, 0], [width, height2]])
    .on("brush end", brushed);

focus.append("g")
      .attr("class", "axis axis--x")
      .attr("transform", "translate(0," + height + ")")
      .call(xAxis);

focus.append("g")
      .attr("class", "axis axis--y")
      .call(yAxis);
    
    
var city = focus.selectAll(".city")
    .data(data)
    .enter().append("g")
    .attr("class", "city");

city.append("path")
      .attr("class", "line")
      .attr("d", function(d) { return line(d.values); })
      .style("stroke", function(d) { return z(d.id); });

city.append("text")
      .datum(function(d) { return {id: d.id, value: d.values[d.values.length - 1]}; })
      .attr("transform", function(d) { return "translate(" + x(parseTime(d.value.x)) + "," + y(d.value.y) + ")"; })
      .attr("x", 3)
      .attr("dy", "0.35em")
      .style("font", "10px sans-serif")
      .text(function(d) { return d.id; });

context.append("path")
      .data(data)
      .attr("class", "area")
//       .attr("d", area2)
  		.attr('d', function(d) { return area2(d.values); });

context.append("g")
      .attr("class", "axis axis--x")
      .attr("transform", "translate(0," + height2 + ")")
      .call(xAxis2);

var circ = svg.append("g")
      .attr("class", "circ")
      .style("display", "none");

for(var i=0;i<data.length;i++){
              circ.append("g")
                .attr("class", "circ"+i)
                .append("circle")
              	.style("stroke",  z(data[i].id))
              	.style("fill", z(data[i].id))
              	.attr("transform", "translate(" + margin.left  + "," + margin.top + ")")
                .attr("r", 2);
              svg.select(".circ"+i)
                .append("text")
              	.attr("transform", "translate(" + margin.left  + "," + margin.top + ")")
                .attr("x", 9)
                .attr("dy", ".35em");
          }

svg.append("rect")
          .attr("class", "overlay")
          .attr("width", width)
          .attr("height", height)
      		.attr("transform", "translate(" + margin.left + "," + margin.top + ")")
          .on("mouseover", function() { circ.style("display", null); })
          .on("mouseout", function() { circ.style("display", "none"); })
          .on("mousemove",  mousemove);
  
 
   
    
    
    
  
//   function mousemove() {
// //   console.log("in mosemove");
//     var x0 = x.invert(d3.mouse(this)[0]),
//         i = bisectDate(data, x0, 1),
//         d0 = data[i - 1],
//         d1 = data[i],
//         d = x0 - d0.date > d1.date - x0 ? d1 : d0;
//     circ.attr("transform", "translate(" + x(d.date) + "," + y(d.price) + ")");
//     circ.select("text").text(d.price);
//   } 
    
    
    function mousemove() {
              var x0 = x.invert(d3.mouse(this)[0]);
              var series = data.map(function(e) {
                    var i = bisect(e.values, x0, 1),
                        d0 = e.values[i - 1],
                        d1 = e.values[i];
                    return x0 - d0.x > d1.x - x0 ? d1 : d0;
                  });
//     console.log(series);
              for(var i=0; i<series.length;i++){
                var selectedFocus = svg.selectAll(".circ"+i);
                selectedFocus.attr("transform", "translate(" + x(parseTime(series[i].x)) + "," + y(series[i].y) + ")");
                selectedFocus.select("text").text(series[i].y);
              }
            }

//     console.log(points);
    
 function brushed() {
  if (d3.event.sourceEvent && d3.event.sourceEvent.type === "zoom") return; // ignore brush-by-zoom
  var s = d3.event.selection || x2.range();
//    console.log(s[0],s[1], Math.floor(x3.invert(s[1])));
  x.domain(s.map(x2.invert, x2));
//   focus.select(".area").attr("d", area);
//   focus.select(".line").attr("d", line);
  		focus.selectAll(".line").attr("d", function(d) { return line(d.values); });
//   city.select(".line").attr("d", function(d) { return line(d.values); });
   city.select("text")
   		.data(data)
      .datum(function(d) { return {id: d.id, value: d.values[Math.floor(x3.invert(s[1]))]}; })
      .attr("transform", function(d) { return "translate(" + x(parseTime(d.value.x)) + "," + y(d.value.y) + ")"; });
  focus.select(".axis--x").call(xAxis);
  svg.select(".zoom").call(zoom.transform, d3.zoomIdentity
      .scale(width / (s[1] - s[0]))
      .translate(-s[0], 0));
   
//    console.log(x2.invert(s[0]),x2.invert(s[1]));
   
   // handle map points
   
   var points_filter = points.filter(function(x) {return ((parseTime(x.date) >= x2.invert(s[0])) && (parseTime(x.date) <= x2.invert(s[1]))  )});
   
   var pts_filter = pts.filter(function(x) {return ((parseTime(x.x) >= x2.invert(s[0])) && (parseTime(x.x) <= x2.invert(s[1]))  )});
   
   
//    console.log(pts_filter);
//    console.log(parseTime(points_filter[0].date),x2.invert(s[0]));
//    console.log(points_filter, points_filter.length);
   
//    mapsvg.selectAll("circle")
// 		.data(points_filter).enter()
// 		.append("circle")
// 		.attr("cx", function (d) { return projection(d.coords)[0]; })
// 		.attr("cy", function (d) { return projection(d.coords)[1]; })
// 		.attr("r", 4)
// // 		.attr("fill", "red")
//     .style("opacity", .7)      // set the element opacity
//     .style("stroke", "#9f5045")    // set the line colour
//     .style("fill", "#F0C059");
   
   mapsvg.selectAll('circle').remove();
   
   mapsvg.selectAll('circle')
   .data(pts_filter)
   .enter()
   .append("circle")
   .attr("cx", function (d) { return projection(d.coords)[0]; })
		.attr("cy", function (d) { return projection(d.coords)[1]; })
		.attr("r", 4)
// 		.attr("fill", "red")
    .style("opacity", .7)      // set the element opacity
    .style("stroke", "#9f5045")    // set the line colour
    .style("fill", "#F0C059");
   
   
   
   
}

});

function zoomed() {
  if (d3.event.sourceEvent && d3.event.sourceEvent.type === "brush") return; // ignore zoom-by-brush
  var t = d3.event.transform;
  x.domain(t.rescaleX(x2).domain());
  focus.select(".area").attr("d", area);
  focus.select(".line").attr("d", line);
  focus.select(".axis--x").call(xAxis);
  context.select(".brush").call(brush.move, x.range().map(t.invertX, t));
}

function type(d) {
  d.date = parseDate(d.date);
  d.price = +d.price;
  return d;
}

</script>

https://d3js.org/d3.v4.min.js

https://d3js.org/topojson.v1.min.js