import argparse
import numpy as np
import matplotlib.pyplot as plt
from itertools import izip

parser = argparse.ArgumentParser(description="plot single trace")
parser.add_argument("filenames", metavar="<file>")
parser.add_argument("--title", "-t",  default="")
args = parser.parse_args()

def k_stats(costs, times):
    ks = np.array([time/cost for cost, time in izip(costs, times)])
    print "Mean: " + str(np.mean(ks)) + " STD DEV: " + str(np.std(ks))
    
costs = []
times = []
with open(args.filenames, 'r') as f:
    for line in f:
        split = line.split()
        times.append(float(split[0]))
        costs.append(int(split[1]))


coeffs = np.polyfit(costs, times, 1)
fit_fn = np.poly1d(coeffs)
k_stats(costs, times)
print fit_fn
print "rsquared"
plt.ylabel("Execution time ($\mu s$)")
plt.xlabel("Cost")
plt.title(args.title)
plt.plot( costs, times,  'xg' , costs, fit_fn(costs), '-b')
plt.show()