import pickle
from matplotlib import pyplot
from matplotlib.legend_handler import HandlerLine2D

file = open("cholesky_data", "rb")
m_list = pickle.load(file)
SetupTime = pickle.load(file)
FactorisationTime = pickle.load(file)
TotalTime = pickle.load(file)
ApproximateSolution = pickle.load(file)
TrueSolution = 0.199268  # hardcoded
AbsoluteError = pickle.load(file)
file.close()

file = open("cholesky_banded_data", "rb")
bm_list = pickle.load(file)
bSetupTime = pickle.load(file)
bFactorisationTime = pickle.load(file)
bTotalTime = pickle.load(file)
bApproximateSolution = pickle.load(file)
bTrueSolution = 0.199268  # hardcoded
bAbsoluteError = pickle.load(file)
file.close()

O_n3 = [pow(i, 2) for i in m_list]
true_solution = [TrueSolution for i in bm_list]

pyplot.plot(m_list, FactorisationTime)
pyplot.plot(bm_list, bFactorisationTime)
pyplot.ylabel('Factorisation time')

# line1, = pyplot.plot(m_list, ApproximateSolution, label='Non-banded Approximate solution')
# pyplot.plot(bm_list, bApproximateSolution, label='Banded Approximate solution')
# pyplot.plot(bm_list, true_solution, label='True solution')
# pyplot.legend(handler_map={line1: HandlerLine2D(numpoints=4)})
pyplot.xlabel('m')
pyplot.show()