1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
#!/usr/bin/env python
import sys
from collections import defaultdict
from pngchart import SimpleLineChart
#from pygooglechart import SimpleLineChart
from colour import hash_colour
WIDTH = 800
HEIGHT = 300
RESULTS_FILE = 'results.csv'
def main():
data = {}
order = []
snakes = []
for line in open(RESULTS_FILE):
game_id, name, length, life = line[:-1].split(',')
game_id = int(game_id)
length = int(length)
life = float(life)
if name not in data:
snakes.append(name)
data[name] = {}
if game_id not in order:
order.append(game_id)
data[name][game_id] = (length, life)
length_data = []
time_data = []
colours = []
for name in snakes:
time_series = []
length_series = []
for game_id in order:
length, time = data[name].get(game_id, (None, None))
time_series.append(time)
length_series.append(length)
colours.append('%02X%02X%02X' % hash_colour(name))
time_data.append(time_series)
length_data.append(length_series)
for filename, data in (('length_chart.png', length_data),
('time_chart.png', time_data)):
chart = SimpleLineChart(WIDTH, HEIGHT, colours=colours, legend=snakes)
for series in data:
chart.add_data(series)
print 'Updating', filename, '... ',
sys.stdout.flush()
chart.download(filename)
print 'done!'
if __name__ == '__main__':
main()
|
