def load(dimension,tstart,tend,tstep,full_load,fluc_only,eq_only): time_steps = np.arange(tstart,tend+1,tstep) if dimension == 3: xgc_nstx_139047 = xgc.XGC_Loader(xgc_path,grid3D,time_steps,dn_amplifier = 1,n_cross_section = 16, Equilibrium_Only = eq_only,Full_Load = full_load, Fluc_Only = fluc_only) elif dimension == 2: xgc_nstx_139047 = xgc.XGC_Loader(xgc_path,grid2D,time_steps,dn_amplifier = 1,n_cross_section = 1, Full_Load = full_load, Fluc_Only = fluc_only) return xgc_nstx_139047
def single_load(t): try: import sys sys.path.append('/p/gkp/lshi/') import sdp.plasma.xgc.loader as xgc import sdp.geometry.grid as Grid grid3D = Grid.Cartesian3D(Xmin=1.25, Xmax=1.6, Ymin=-0.5, Ymax=0.5, Zmin=-0.35, Zmax=0.35, NX=100, NY=300, NZ=100) xgc_nstx = xgc.XGC_Loader(xgc_path, grid3D, t, t + 1, 2, dn_amplifier=1, n_cross_section=16, Equilibrium_Only=False, Full_Load=True, Fluc_Only=False) xgc_nstx.cdf_output(output_path, eq_file='eqfile{0}.cdf'.format(t)) except Exception as e: return str(e) return 0
def single_load_2d(t): try: import sys sys.path.append('/p/gkp/lshi/') import sdp.plasma.xgc.loader as xgc import sdp.geometry.grid as Grid grid2D = Grid.Cartesian2D(DownLeft = (-0.5,1.25),UpRight = (0.5,1.6),NR = 256, NZ = 512) xgc_nstx = xgc.XGC_Loader(xgc_path,grid2D,t,t+1,2,dn_amplifier = 1,n_cross_section = 32, Equilibrium_Only = False,Full_Load = True, Fluc_Only = False) xgc_nstx.cdf_output(output_path,eq_file = 'eqfile{0}.cdf'.format(t)) except Exception as e: return str(e) return 0