110 lines
2.7 KiB
Python
110 lines
2.7 KiB
Python
from __future__ import division
|
|
|
|
import axi
|
|
import numpy as np
|
|
import sys
|
|
|
|
NUMBER = 1
|
|
TITLE = 'Five Seconds of Donkey Kong'
|
|
LABEL = '#%d' % NUMBER
|
|
|
|
COLUMNS = 6
|
|
SECONDS = 5
|
|
|
|
def stack_drawings(ds, spacing=0):
|
|
result = axi.Drawing()
|
|
y = 0
|
|
for d in ds:
|
|
d = d.origin().translate(-d.width / 2, y)
|
|
result.add(d)
|
|
y += d.height + spacing
|
|
return result
|
|
|
|
def title():
|
|
d = axi.Drawing(axi.text(TITLE, axi.FUTURAM))
|
|
d = d.scale_to_fit_height(0.25)
|
|
d = d.move(6, 8.5, 0.5, 1)
|
|
d = d.join_paths(0.01)
|
|
return d
|
|
|
|
def label():
|
|
d = axi.Drawing(axi.text(LABEL, axi.FUTURAL))
|
|
d = d.scale_to_fit_height(0.125)
|
|
d = d.rotate(-90)
|
|
d = d.move(12, 8.5, 1, 1)
|
|
d = d.join_paths(0.01)
|
|
return d
|
|
|
|
def main():
|
|
# read file
|
|
with open(sys.argv[1], 'r') as fp:
|
|
data = fp.read()
|
|
|
|
# strip and split lines
|
|
lines = data.split('\n')
|
|
lines = [x.strip() for x in lines]
|
|
lines = [x.strip(',') for x in lines]
|
|
lines = filter(None, lines)
|
|
|
|
# read values and transpose
|
|
data = [map(int, line.split(',')) for line in lines]
|
|
data = np.transpose(data)
|
|
|
|
print '%d series in file' % len(data)
|
|
|
|
# trim to SECONDS worth of data
|
|
n = len(data[0])
|
|
m = SECONDS * 60 / 2
|
|
a = max(0, int(n // 2 - m))
|
|
b = min(n, int(n // 2 + m))
|
|
data = [x[a:b] for x in data]
|
|
|
|
# remove addresses with too few values
|
|
data = [x for x in data if len(set(x)) > 1]
|
|
|
|
print '%d series that changed' % len(data)
|
|
|
|
# trim so all rows are full
|
|
data = data[:int((len(data) // COLUMNS) * COLUMNS)]
|
|
|
|
print '%d series after trimming' % len(data)
|
|
|
|
print '%d data points each' % len(data[0])
|
|
|
|
# create sparklines in a grid pattern
|
|
paths = []
|
|
for i, row in enumerate(data):
|
|
r = i // COLUMNS
|
|
c = i % COLUMNS
|
|
lo = min(row)
|
|
hi = max(row)
|
|
if lo == hi:
|
|
row = [0 for x in row]
|
|
else:
|
|
row = [(x - lo) / float(hi - lo) for x in row]
|
|
path = []
|
|
for j, value in enumerate(row):
|
|
x = (j / len(row) + c * 1.1)
|
|
y = 1-value + r * 1.5
|
|
path.append((x, y))
|
|
paths.append(path)
|
|
d = axi.Drawing(paths)
|
|
|
|
# add title and label and fit to page
|
|
d = d.scale(8.5 / d.width, (12 - 0.5) / d.height)
|
|
d = stack_drawings([d, title()], 0.25)
|
|
d = d.rotate(-90)
|
|
d = d.center(12, 8.5)
|
|
d.add(label())
|
|
|
|
print d.bounds
|
|
|
|
# save outputs
|
|
d.dump('nes/%d.axi' % NUMBER)
|
|
rotated = d.rotate(90).center(8.5, 12)
|
|
rotated.render(scale=109 * 1, line_width=0.3/25.4).write_to_png('nes/%d.png' % NUMBER)
|
|
rotated.dump_svg('nes/%d.svg' % NUMBER)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|