from __future__ import division import axi import numpy as np import os import sys NUMBER = '?' TITLE = 'Five Seconds of Donkey Kong' LABEL = '#%s' % NUMBER COLUMNS = 6 SECONDS = 5 FRAME_OFFSET = 0 MIN_CHANGES = 1 UNIQUE = False SIMPLIFY = 0 def simplify_sparkline(values, n): if not n: return values result = [] previous = None for x, y in enumerate(values): if result: window = result[-n:] lo = min(window) hi = max(window) if y >= lo and y <= hi: result.append(result[-1]) previous = y continue if previous is not None: result[-1] = previous result.append(y) previous = y return result 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 = int(SECONDS * 60 / 2) mid = int(n // 2) + FRAME_OFFSET a = max(0, mid - m) b = min(n, mid + 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)) > MIN_CHANGES] print '%d series that changed' % len(data) # remove duplicate series if UNIQUE: new_data = [] seen = set() for x in data: k = tuple(x) if k in seen: continue seen.add(k) new_data.append(x) data = new_data print '%d unique series' % len(data) # delete repetitive stuff # del data[136:136+8*14] # 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): row = simplify_sparkline(row, SIMPLIFY) 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 dirname = 'nes/%s' % NUMBER try: os.makedirs(dirname) except Exception: pass d.dump(os.path.join(dirname, 'out.axi')) rotated = d.rotate(90).center(8.5, 12) rotated.dump_svg(os.path.join(dirname, 'out.svg')) im = rotated.render( scale=109 * 1, line_width=0.3/25.4, show_axi_bounds=False, use_axi_bounds=False) im.write_to_png(os.path.join(dirname, 'out.png')) if __name__ == '__main__': main()