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')