import readPhi
import readF
import matplotlib.pyplot as plt
from matplotlib.colors import Normalize
import glob
import numpy as np
from scipy.constants import e, k

path = '../2024-09-26_11.48.04/'

filesF_i = sorted(glob.glob(path+'time_*_f_i.csv'))
start = 0
end   = len(filesF_i)
every = 50
 
time, x, v, f_i = readF.read(filesF_i[-1])
plt.title('t = {:.3f} ns'.format(time*1e9))
norm = Normalize(vmin=4., vmax=20., clip=False)
plt.imshow(np.log10(f_i), cmap='cividis', interpolation='None', aspect='auto', origin='lower', extent=[v[0],v[-1],x[0],x[-1]], norm=norm)

plt.show()