import os
import numpy as np
import matplotlib.pyplot as plt
import pdb
import glob

# IO: This loads the timings for you 
threads, timings = [], []
files = os.listdir('timings')
files.sort(key= lambda x: int(x.split('_')[0]))
for file in files:
    with open(f'timings/{file}', 'r') as f:
        if 'None' not in file:
            n, t = f.read().strip().split(',')
            threads.append(int(n))
            timings.append(float(t))
threads = np.array(threads)
timings = np.array(timings)

print('This is the data I loaded: threads =', threads, ', timings =',timings)

fig, axs = plt.subplots()

# CREATE YOUR PLOT HERE
plt.plot(threads, timings, '.--')
plt.xlabel('Threads')
plt.ylabel('Timings')
plt.xscale('log')

# Remember to label your axis
# Feel free to make it pretty


plt.savefig('threads_v_timings.png', dpi=300)