generate_line_graph.py

import matplotlib.pyplot as pyplot
import sys

'''
Generates standard line graph for comparison of runtimes.
Could be improved for better/smoother visualization
'''

xAxis = "Number of States"
yAxis = "Average Minimization Time (ms)"
title = "Comparison of Incremental Algorithms"
scale = "log"

testsfile = sys.argv[1]
xIndex = int(sys.argv[2])
graph_indices = [int(i) for i in sys.argv[3:-1]]
cutoff = int(sys.argv[-1])

def toNum(s):
    if s.isdigit():
        return int(s)
    else:
        return float(s)/1000

def mergeData(data, index):
    data.sort(key = lambda x: x[index])
    new_data = []
    count = 1
    for row in data:
        if new_data and new_data[-1][index] == row[index]:
            old_row = new_data[-1]
            count += 1
            for i in range(0, len(old_row)):
                if i != index:
                    old_row[i] += (row[i] - old_row[i])/float(count) #running average
        else:
            count = 1
            new_data.append(row)
    return new_data
        
with open(testsfile, "r") as f:
    fStrings = f.read().split('\n')

title_row = fStrings[0].split(',')
fData = [[toNum(datum) for datum in row.split(',') if datum]
         for row in fStrings[1:] if row]

average_mins = []
for row in fData:
    if row[0] <= 400 and row[0] != row[1]:
        average_mins.append(row[1]/float(row[0]))
print("average")
print(sum(average_mins)/len(average_mins))

fData = mergeData(fData, xIndex)

lineStyles = ['-','--',':',':'] #Different line style markers

xData = [row[xIndex] for row in fData if row[xIndex] <= cutoff]
style_num = 0
for i in graph_indices:
    assert(style_num < len(lineStyles))
    graphData = [row[i] for row in fData[:len(xData)]]
    pyplot.plot(xData, graphData, label=title_row[i], linestyle=lineStyles[style_num], linewidth=1.0)
    style_num += 1
pyplot.yscale(scale)
pyplot.legend(loc=4)
pyplot.xlabel(xAxis)
pyplot.ylabel(yAxis)
pyplot.title(title)
pyplot.savefig("graph.png", dpi=600)