#!/usr/bin/python

import re, sys, os
from pyx import *



def parse_file(name):
	r = re.compile(" +[0-9]+ +[0-9]+ +([0-9.]+) MB/sec +execute" +\
			" +([0-9]+) sec +")
	fp = open(name, "r")
	points = []
	for l in fp:
		m = r.match(l)
		if not m:
			continue
		points.append((float(m.group(1)), int(m.group(2))))
	fp.close()
	filename = os.path.splitext(name)[0]
	return graph.data.list(points, x = 2, y = 1, title = filename)

	

def main():
	g = graph.graphxy(height = 20,
			  y = graph.axis.linear(title="rate MB/s"),
			  x = graph.axis.linear(title="secs elapsed"),
			  key=graph.key.key(pos="br", dist=0.1))
					
	datas = []
	for i in xrange(1, len(sys.argv)):
		datas.append(parse_file(sys.argv[i]))
	g.plot(datas, [graph.style.line(lineattrs = [style.linestyle.solid, color.linearpalette.Rainbow])])
	g.writePDFfile("dbench.pdf")

if __name__ == "__main__":
	main()