#!/usr/bin/python

#
# Copyright (C) 2000-2005 by Yasushi Saito (yasushi.saito@gmail.com)
# 
# Pychart is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation; either version 2, or (at your option) any
# later version.
#
# Pychart is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
# for more details.
#

import sys
from pychart import *

def get_timing(filename):
	file_ = open(filename, 'r')
	list_ = []
	for line in file_:
		fields = line.split()
		set_size = float(fields[0])
		duration = float(fields[1])
		list_.append( ( set_size, duration ) )
	file_.close()
	return list_

def get_array_timing():
	return get_timing('array.txt')

def get_seek_timing():
	return get_timing('seek.txt')

def desired_y_max(*list_):
	maximum = 0.0
	for element in list_:
		for set_size, duration in element:
			maximum = max(duration, maximum)
	return maximum

def main():
	theme.get_options()
	theme.output_format = 'pdf'
	theme.use_color = 1
	theme.reinitialize()

	width = 800
	height = width * 4 // 5
	size = (width, height)

	array_timing_data = get_array_timing()
	seek_timing_data = get_seek_timing()

	y_max = desired_y_max(array_timing_data, seek_timing_data)

	can = canvas.default_canvas()

	ar = area.T(
		size = size,
		legend=legend.T(),
		x_range = (1, None),
		y_range = (0.0001, y_max + 100),
		x_coord = log_coord.T(),
		y_coord = log_coord.T(),
		x_axis = axis.X(format="%d", label="Set size"),
		y_axis = axis.Y(format="%d", label="Time"),
		)

	lp = line_plot.T(data=array_timing_data, label="Array")
	ar.add_plot(lp)
					
	lp = line_plot.T(data=seek_timing_data, label="Seek")
	ar.add_plot(lp)
					
	ar.draw()

	#can.show(ar.x_pos(4), ar.y_pos(970), "/a50{}seek")

main()