assets/weights_graphs.js

   1 function WeightsDrawer(selector) {
   2   this.selector = selector;
   3   this.current_source = "consensus_weight_fraction";
   4   this.current_period = "1_month";
   5   this.weights_data = {};
   6   this.svg = null;
   7 };
   8 WeightsDrawer.prototype = new WeightsDrawer();
   9
  10 WeightsDrawer.margin = {top: 50, right: 10, bottom: 90, left: 130};
  11 WeightsDrawer.width = 600 - WeightsDrawer.margin.left - WeightsDrawer.margin.right;
  12 WeightsDrawer.height = 400 - WeightsDrawer.margin.top - WeightsDrawer.margin.bottom;
  13
  14 WeightsDrawer.parseTime = d3.time.format("%Y-%m-%d %H:%M:%S").parse;
  15 WeightsDrawer.percentFormatter = d3.format(".2%");
  16
  17 WeightsDrawer.x = d3.time.scale()
  18     .range([0, WeightsDrawer.width]);
  19
  20 WeightsDrawer.y = d3.scale.linear()
  21     .range([WeightsDrawer.height, 0]);
  22
  23 WeightsDrawer.xAxis = d3.svg.axis()
  24     .scale(WeightsDrawer.x)
  25     .orient("bottom");
  26
  27 WeightsDrawer.yAxis = d3.svg.axis()
  28     .scale(WeightsDrawer.y)
  29     .orient("left")
  30     .tickFormat(function(d) { return (d == 0) ? "" : WeightsDrawer.percentFormatter(d); });
  31
  32 WeightsDrawer.area = d3.svg.area()
  33     .x(function(d) { return WeightsDrawer.x(d.date); })
  34     .y0(function(d) { return WeightsDrawer.y(d.y0); })
  35     .y1(function(d) { return WeightsDrawer.y(d.y0 + d.y); });
  36
  37 WeightsDrawer.stack = d3.layout.stack()
  38     .values(function(d) { return d.values; });
  39
  40 WeightsDrawer.onionoo_url = "https://onionoo.torproject.org/weights?type=relay&contact=adminsys@nos-oignons.net";
  41
  42 WeightsDrawer.periods = [
  43     { id: "1_month", label: L10n.t_1_month },
  44     { id: "3_months", label: L10n.t_3_months },
  45     { id: "1_year", label: L10n.t_1_year },
  46     { id: "5_years", label: L10n.t_5_years },
  47   ];
  48
  49 WeightsDrawer.current_period = WeightsDrawer.periods[0].id;
  50
  51 WeightsDrawer.sources = [
  52     { id: "consensus_weight_fraction", label: L10n.consensus_weight },
  53     { id: "exit_probability", label: L10n.exit_probability },
  54   ];
  55
  56 WeightsDrawer.current_source = WeightsDrawer.sources[0].id;
  57
  58 WeightsDrawer.extract_values = function(history, interval, minTime, maxTime) {
  59   var values = [];
  60   var first = history ? WeightsDrawer.parseTime(history.first) : maxTime;
  61   var last = history ? WeightsDrawer.parseTime(history.last) : minTime;
  62   var i = 0;
  63   for (var current = minTime; current <= maxTime; current = d3.time.second.offset(current, interval)) {
  64     values.push({ date: current,
  65                   y: (first <= current && current <= last) ? history.factor * history.values[i++] : 0
  66                 });
  67   }
  68   return values;
  69 }
  70
  71 WeightsDrawer.color = d3.scale.ordinal();
  72 WeightsDrawer.color.domain(nos_oignons_relays.map(function(r) {return r.fingerprint}));
  73 WeightsDrawer.color.range(nos_oignons_relays.map(function(r) {return r.color}));
  74
  75 WeightsDrawer.prototype.draw_weights_graph = function(raw_data) {
  76   var drawer = this;
  77
  78   // Purge non running relays
  79   raw_data.relays.forEach(function(r, i) {
  80     if (typeof r.consensus_weight_fraction === 'undefined') {
  81       raw_data.relays.splice(i, 1);
  82     }
  83   });
  84
  85   var form_source = d3.select(drawer.selector).append("form")
  86       .attr("id", "weight-sources")
  87       .attr("action", "#");
  88   WeightsDrawer.sources.forEach(function(s) {
  89     var div = form_source.append("div");
  90     var radio = div.append("input")
  91       .attr("type", "radio")
  92       .attr("name", "source")
  93       .attr("id", "source_" + s.id)
  94       .on("click", function() { drawer.update_source(s.id); });
  95     div.append("label")
  96       .attr("for", "source_" + s.id)
  97       .text(s.label);
  98     if (s.id == WeightsDrawer.sources[0].id) {
  99       radio.attr("checked", true);
 100     }
 101   });
 102
 103   drawer.svg = d3.select(drawer.selector).append("svg")
 104       .attr("width", WeightsDrawer.width + WeightsDrawer.margin.left + WeightsDrawer.margin.right)
 105       .attr("height", WeightsDrawer.height + WeightsDrawer.margin.top + WeightsDrawer.margin.bottom)
 106     .append("g")
 107       .attr("transform", "translate(" + WeightsDrawer.margin.left + "," + WeightsDrawer.margin.top + ")");
 108
 109   var form_period = d3.select(drawer.selector).append("form")
 110       .attr("class", "graph-period")
 111       .attr("action", "#");
 112   WeightsDrawer.periods.forEach(function(p) {
 113     var div = form_period.append("div");
 114     var radio = div.append("input")
 115       .attr("type", "radio")
 116       .attr("name", "period")
 117       .attr("id", "weights_period_" + p.id)
 118       .on("click", function() { drawer.update_period(p.id); });
 119     div.append("label")
 120       .attr("for", "weights_period_" + p.id)
 121       .text(p.label);
 122     if (p.id == WeightsDrawer.periods[0].id) {
 123       radio.attr("checked", true);
 124     }
 125   });
 126
 127   WeightsDrawer.sources.map(function(s) { return s.id; }).forEach(function(source) {
 128     drawer.weights_data[source] = {};
 129     WeightsDrawer.periods.map(function(p) { return p.id; }).forEach(function(period) {
 130       var interval = d3.max(raw_data.relays, function(d) {
 131         return d[source][period] && d[source][period].interval;
 132       });
 133       raw_data.relays.forEach(function(d) {
 134         if ((d[source][period] && d[source][period].interval != interval)) {
 135           throw "PANIC: Different interval for different relays in the same time period.";
 136         }
 137       });
 138       var minTime = d3.min(raw_data.relays, function(d) {
 139         return d[source][period] && WeightsDrawer.parseTime(d[source][period].first);
 140       });
 141       var maxTime = d3.min(raw_data.relays, function(d) {
 142         return d[source][period] && WeightsDrawer.parseTime(d[source][period].last);
 143       });
 144
 145       var maxValue = 0;
 146
 147       var relays = WeightsDrawer.color.domain().map(function(fingerprint) {
 148         var relay_data = raw_data["relays"].filter(function(d) { return d.fingerprint == fingerprint; })[0];
 149
 150         var history = relay_data[source][period];
 151         var values = WeightsDrawer.extract_values(history, interval, minTime, maxTime);
 152         maxValue = maxValue + d3.max(values, function(d) { return d.y; });
 153
 154         return {
 155           fingerprint: fingerprint,
 156           values: values,
 157         };
 158       });
 159       drawer.weights_data[source][period] = {
 160         minTime: minTime,
 161         maxTime: maxTime,
 162         maxValue: maxValue,
 163         relays: relays
 164       };
 165     });
 166   });
 167
 168   WeightsDrawer.y.domain([0, drawer.weights_data[WeightsDrawer.current_source][WeightsDrawer.current_period].maxValue]);
 169
 170   WeightsDrawer.x.domain([drawer.weights_data[WeightsDrawer.current_source][WeightsDrawer.current_period].minTime, drawer.weights_data[WeightsDrawer.current_source][WeightsDrawer.current_period].maxTime]);
 171
 172   var weight_graph = drawer.svg.selectAll(".weight_graph")
 173       .data(WeightsDrawer.stack(drawer.weights_data[WeightsDrawer.current_source][WeightsDrawer.current_period].relays))
 174     .enter().append("path")
 175       .attr("class", "weight_graph area")
 176       .attr("d", function(d) { return WeightsDrawer.area(d.values); })
 177       .style("fill", function(d) { return WeightsDrawer.color(d.fingerprint); });
 178
 179   drawer.svg.append("g")
 180       .attr("class", "x axis")
 181       .attr("transform", "translate(0," + WeightsDrawer.height + ")")
 182       .call(WeightsDrawer.xAxis)
 183         .selectAll("text")
 184         .style("text-anchor", "end")
 185         .attr("transform", "rotate(-90) translate(-10, 0)");
 186
 187   drawer.svg.append("g")
 188       .attr("class", "y axis")
 189       .call(WeightsDrawer.yAxis);
 190
 191   var legend = drawer.svg.selectAll(".legend")
 192       .data(WeightsDrawer.color.domain().slice().reverse())
 193     .enter().append("g")
 194       .attr("class", "legend")
 195       .attr("transform", function(d, i) { return "translate(0," + ((i * 20) - WeightsDrawer.margin.top) + ")"; });
 196
 197   legend.append("rect")
 198       .attr("x", WeightsDrawer.width - 18)
 199       .attr("width", 18)
 200       .attr("height", 18)
 201       .style("fill", WeightsDrawer.color);
 202
 203   legend.append("text")
 204       .attr("x", WeightsDrawer.width - 24)
 205       .attr("y", 9)
 206       .attr("dy", ".35em")
 207       .style("text-anchor", "end")
 208       .text(function(d) {
 209         return nos_oignons_relays.filter(function(r) { return r.fingerprint == d; })[0].name;
 210       });
 211 };
 212
 213 WeightsDrawer.prototype.refresh_graph = function() {
 214   var drawer = this;
 215
 216   WeightsDrawer.x.domain([drawer.weights_data[drawer.current_source][drawer.current_period].minTime, drawer.weights_data[drawer.current_source][drawer.current_period].maxTime]);
 217   WeightsDrawer.y.domain([0, drawer.weights_data[drawer.current_source][drawer.current_period].maxValue]);
 218   var t = drawer.svg.transition().duration(300);
 219   t.select(".x.axis").call(WeightsDrawer.xAxis);
 220   t.select(".y.axis").call(WeightsDrawer.yAxis);
 221   t.selectAll(".weight_graph").style("fill-opacity", 0);
 222   t.each("end", function() {
 223     d3.selectAll(".weight_graph").data(WeightsDrawer.stack(drawer.weights_data[drawer.current_source][drawer.current_period].relays));
 224     d3.selectAll(".weight_graph").attr("d", function(d) { return WeightsDrawer.area(d.values); })
 225     var t2 = drawer.svg.transition().duration(100);
 226     t2.selectAll(".weight_graph").style("fill-opacity", 1);
 227   });
 228   d3.selectAll(".x.axis text")
 229     .style("text-anchor", "end")
 230     .attr("transform", "rotate(-90) translate(-10, 0)");
 231 }
 232
 233 WeightsDrawer.prototype.update_source = function(source) {
 234   this.current_source = source;
 235   this.refresh_graph();
 236 }
 237
 238 WeightsDrawer.prototype.update_period = function(period) {
 239   this.current_period = period;
 240   this.refresh_graph();
 241 }
 242
 243 WeightsDrawer.prototype.draw = function() {
 244   var drawer = this;
 245   d3.json(WeightsDrawer.onionoo_url, function(error, raw_data) {
 246     d3.select(drawer.selector).text("");
 247     drawer.draw_weights_graph(raw_data);
 248   });
 249 };