import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import make_axes_locatable
import seaborn as sns
plt.rcParams.update({'figure.autolayout': True})
mpl.rcParams['axes.unicode_minus']=False
#local
from run import Run
import plot_style
import field_helper as field
plot_style.white()
pal = sns.color_palette('deep')
run = Run(sys.argv[1])
run.read_ntot()
os.system('mkdir -p ' + run.run_dir + 'analysis/amplitude')
max_amp_i = np.empty(run.nt, dtype=float)
max_amp_e = np.empty(run.nt, dtype=float)
for it in range(run.nt):
max_amp_i[it] = np.max(run.ntot_i[it, :, :])
max_amp_e[it] = np.max(run.ntot_e[it, :, :])
np.savetxt(run.run_dir + 'analysis/amplitude/max_amp.csv',
np.transpose((range(run.nt), max_amp_i, max_amp_e)),
delimiter=',', fmt=['%d', '%.5f', '%.5f'],
header='t_index,max_amp_i,max_amp_e')
plt.clf()
run.ntot_e = ntot_e_tmp[:]
run.phi = phi_tmp[:]
run.q_i = f_q_i(t_reg)
run.t = t_reg[:]
run.nt = len(run.t)
if __name__ == '__main__':
run = Run(sys.argv[1])
if sys.argv[2] == 'False':
lab_frame = False
else:
lab_frame = True
version = int(sys.argv[3])
run.read_ntot(lab_frame=lab_frame)
run.read_phi(lab_frame=lab_frame)
run.read_q()
interp_fac = 1
interpolate_time(run, interp_fac)
# Heat flux to SI
hflux_SI = run.q_i * (run.nref*run.tref*run.vth*run.rhoref**2/run.amin**2)
# Write
os.system('mkdir -p ' + run.run_dir + 'analysis/exp_bes_write')
nc_file = Dataset(run.run_dir + 'analysis/exp_bes_write/gs2_27268_0.25_' +
str(np.around(run.tprim_1, 2)) + '_' +
str(np.around(run.g_exb, 2)) + '_' + str(version) +
'.cdf', 'w')
