D3 Examples - Brush & Zoom with choropleth

xxxxxxxxxx
 
<!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>
​
<script>
 
  
var svg = d3.select(".graphsvg"),
    margin = {top: 20, right: 100, bottom: 110, left: 40},
    margin2 = {top: 430, right: 20, bottom: 30, left: 40},
    width = +svg.attr("width") - margin.left - margin.right,
    height = +svg.attr("height") - margin.top - margin.bottom,
    height2 = +svg.attr("height") - margin2.top - margin2.bottom;
​
// 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 area = d3.area()
    .curve(d3.curveMonotoneX)
    .x(function(d) { return x(d.date); })
    .y0(height)
    .y1(function(d) { return y(d.price); });
  
// var line = d3.line()
// //       .curve(d3.curveMonotoneX)
//     .x(function(d) {
//       return x(d.date);
//     })
//     .y(function(d) {
//       return y(d.price);
//     });
  
  
var line = d3.line()
.x(function(d) {
  return x(parseTime(d.x));
})
.y(function(d) {
  return y(d.y);
});  
​
// 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 + ")");
​
var context = svg.append("g")
    .attr("class", "context")
    .attr("transform", "translate(" + margin2.left + "," + margin2.top + ")");
  
  
  var mapsvg = d3.select(".mapsvg"),
    mapwidth = +mapsvg.attr("width"),
    mapheight = +mapsvg.attr("height");
    
  var points = [
    {
      date: 20111001,
      coords: [-122.490402, 37.786453],
      county_number: 06087,
      county_name: 'Santa Cruz',
      sentiment: 0
    },
    {
      date: 20111112,
      coords: [-94.5786, 39.0997],
      county_number: 20091,
      county_name: 'Johnson',
      sentiment: 0
    },
    {
      date: 20111200,
      coords: [-97.3308, 32.7555],
      county_number: 48439,
      county_name: 'Tarrant',
      sentiment: 0
    }
  ];
//    var dots = mapsvg.selectAll("circle");
​
  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>
​
    