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

m_i = 1.9712e-25

#  paths = ['../quasiNeutral_fullAblation/','../Poisson_fullAblation/']
#  paths = ['../quasiNeutral_partialAblation/','../Poisson_partialAblation/','../PoissonTi_partialAblation/','../2024-11-05_13.56.23/']
#  paths = ['../2024-11-11_13.58.54/']
#  paths = ['../quasiNeutral_fullAblation/','../Poisson_fullAblation/', '../quasiNeutral_partialAblation/', '../Poisson_partialAblation/']
#  paths = ['../2024-12-02_21.07.52/', '../Poisson_50ns_T30Z11/']
#  paths = ['../2024-12-10_18.45.17/', '../Poisson_50ns_T30Z11/']
#  paths = ['../2024-12-11_12.38.27/', '../Poisson_polytropic_fa_T30Z11/', '../Poisson_fa_T30Z11/']
#  paths = ['../Poisson_partialAblation/','../Poisson_partialAblation_lowerT/','../Poisson_partialAblation_lowT/','../Poisson_partialAblation_highT/']
paths = ['../2025-04-17_14.39.34/']
labels = [path[3:-1] for path in paths]

m2s2_to_eV = m_i*0.5/e

for path, label in zip(paths, labels):
    Zlist      = readZlist.read(path+'ZList.csv')
    filesCum_i = sorted(glob.glob(path+'time_*_fCum_i.csv'))
    _, _, v, _ = readF.read(filesCum_i[-1])
    sumF = np.zeros(len(v))
    E = v**2*m2s2_to_eV

    for Z in Zlist:
        filename='time_*_Z_{:.1f}_fCum_i.csv'.format(Z)
        filesCum_i = sorted(glob.glob(path+filename))
        time, rCum, v, f_i = readF.read(filesCum_i[-1])
        sumF += f_i[0]
        plt.plot(E, 4.0*np.pi*rCum[0]**2*f_i[0]/E, label=Z)
    plt.plot(E, 4.0*np.pi*rCum[0]**2*sumF/E, label='sum', color='k', linestyle='dashed')
    print(time)

plt.yscale('log')
plt.ylim([1e8,5e11])
plt.ylabel('dN / dE (eV^-1)')
plt.xscale('log')
plt.xlim([1e0,1e4])
plt.xlabel('e (eV)')

plt.legend()

#  plt.title('r = {:.1f} mm, time_max={:.1f} ns, '.format(x[0]*1e3, time*1e9) + label)

plt.show()