]> nos-oignons.net Git - website.git/blob - assets/weights_graphs.js
test
[website.git] / 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: "6_months", label: L10n.t_6_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 };