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