Update python scripts.

This commit is contained in:
Jorge Gonzalez 2024-10-09 16:31:25 +02:00
commit e82d4378c6
2 changed files with 20 additions and 17 deletions

View file

@ -9,22 +9,23 @@ from scipy.constants import e, k
m_i = 1.9712e-25 m_i = 1.9712e-25
# paths = ['../quasiNeutral_fullAblation/','../Poisson_fullAblation/'] # paths = ['../quasiNeutral_fullAblation/','../Poisson_fullAblation/']
# paths = ['../quasiNeutral_partialAblation/','../Poisson_partialAblation/'] # paths = ['../quasiNeutral_partialAblation/','../quasiNeutral_partialAblation_cs/','../Poisson_partialAblation/']
paths = ['../2024-09-27_17.41.21/'] paths = ['../2024-10-04_12.10.38/']
for path in paths: labels = [path[3:-1] for path in paths]
for path, label in zip(paths, labels):
filesCum_i = sorted(glob.glob(path+'time_*_fCum_i.csv')) filesCum_i = sorted(glob.glob(path+'time_*_fCum_i.csv'))
start = 0 start = 0
end = len(filesCum_i) end = len(filesCum_i)
every = 100 every = 100
for fileCum_i in filesCum_i[start:end:every]: for fileCum_i in filesCum_i[start:end+1:every]:
time, x, v, f_i = readF.read(fileCum_i) time, x, v, f_i = readF.read(fileCum_i)
plt.plot(v**2*m_i*0.5/e, f_i[0]*e/m_i/v, label='{:.3f} ns'.format(time*1e9)) plt.plot(v**2*m_i*0.5/e, f_i[0]*e/m_i/v, label='{:.3f} ns'.format(time*1e9))
time, x, v, f_i = readF.read(filesCum_i[-1]) time, x, v, f_i = readF.read(filesCum_i[-1])
plt.plot(v**2*m_i*0.5/e, f_i[0]*e/m_i/v, label='time_max', color='k')
plt.plot(v**2*m_i*0.5/e, f_i[0]*e/m_i/v, label='{:.3f} ns'.format(time*1e9), color='k')
plt.yscale('log') plt.yscale('log')
plt.ylim([1e18,1e24]) plt.ylim([1e18,1e24])
@ -35,7 +36,7 @@ for path in paths:
plt.legend() plt.legend()
plt.title('r = {:.1f} mm, time_max={:.1f} ns '.format(x[0]*1e3, time*1e9) + path) plt.title('r = {:.1f} mm, time_max={:.1f} ns, '.format(x[0]*1e3, time*1e9) + label)
plt.show() plt.show()

View file

@ -9,31 +9,33 @@ from scipy.constants import e, k
# paths = ['../quasiNeutral_fullAblation/','../2024-09-26_11.48.04/'] # paths = ['../quasiNeutral_fullAblation/','../2024-09-26_11.48.04/']
# paths = ['../2024-09-26_12.47.11/'] # paths = ['../2024-09-26_12.47.11/']
# paths = ['../quasiNeutral_partialAblation/','../2024-09-26_13.58.24/'] # paths = ['../quasiNeutral_partialAblation/','../2024-09-26_13.58.24/']
# path = '../quasiNeutral_fullAblation/' # paths = ['../2024-10-02_14.30.44/']
# path = '../quasiNeutral_partialAblatio/' # paths = ['../quasiNeutral_fullAblation/','../Poisson_fullAblation/']
paths = ['../2024-09-27_17.41.21/'] paths = ['../2024-10-03_19.28.22/']
labels = [path[3:-1] for path in paths]
for path in paths: for path, label in zip(paths, labels):
filesPhi = sorted(glob.glob(path+'time_*_phi.csv')) filesPhi = sorted(glob.glob(path+'time_*_phi.csv'))
filesMom_i = sorted(glob.glob(path+'time_*_mom_i.csv')) filesMom_i = sorted(glob.glob(path+'time_*_mom_i.csv'))
start = 0 start = 100
end = len(filesMom_i) end = len(filesMom_i)
every = 50 every = 100
fig, ax = plt.subplots(4, sharex='all') fig, ax = plt.subplots(4, sharex='all')
for fileMom_i, filePhi in zip(filesMom_i[start:end+1:every], filesPhi[start:end+1:every]): for fileMom_i, filePhi in zip(filesMom_i[start:end+1:every], filesPhi[start:end+1:every]):
time, r, phi, E, n_e = readPhi.read(filePhi) time, r, phi, E, n_e = readPhi.read(filePhi)
time, r, n_i, u_i, T_i, Zave = readMom.read(fileMom_i) time, r, n_i, u_i, T_i, Zave = readMom.read(fileMom_i)
# ax[0].plot(r, phi, label='t = {:.3f} ns'.format(time*1e9)) ax[0].plot(r, phi, label='t = {:.1f} ns'.format(time*1e9))
ax[0].plot(r, E, label='t = {:.3f} ns'.format(time*1e9)) # ax[0].plot(r, E, label='t = {:.1f} ns'.format(time*1e9))
# ax[0].plot(r, (Zave*n_i - n_e)/n_e, label='t = {:.1f} ns'.format(time*1e9))
ax[1].set_yscale('log') ax[1].set_yscale('log')
ax[1].set_ylim([1e14,2e25]) ax[1].set_ylim([1e20,2e25])
ax[1].plot(r, Zave*n_i) ax[1].plot(r, Zave*n_i)
ax[1].plot(r, n_e, color='k', linestyle='dashed') ax[1].plot(r, n_e, color='k', linestyle='dashed')
ax[2].plot(r, u_i) ax[2].plot(r, u_i)
ax[3].plot(r, T_i) ax[3].plot(r, T_i)
ax[0].set_title(path) ax[0].set_title(label)
ax[0].legend() ax[0].legend()
plt.show() plt.show()