if (__name__ == '__main__'): import os from diag_geom import geom_set from diag_rb import rb_open import time geom_set(headpath='../../src/gkvp_header.f90', nmlpath="../../gkvp_namelist.001", mtrpath='../../hst/gkvp.mtr.001') ### Examples of use ### ### trninkxky ### #help(trninkxky) xr_trn = rb_open('../../post/data/trn.*.nc') #print(xr_trn) iss = 0 # Index of species itrn = 10 # Index of outputs in trn.*.nc, see help(trninkxky) print("# Plot trn[ky,kx] at t[it], s[iss].") outdir = '../data/trninkxky/' os.makedirs(outdir, exist_ok=True) for it in range(0, len(xr_trn['t']), 10): trninkxky(it, iss, itrn, xr_trn, flag='savefig', outdir=outdir) print("# Display trn[ky,kx] at t[it], s[iss].") trninkxky(it, iss, itrn, xr_trn, flag='display') print("# Save trn[ky,kx] as text files at t[it], s[iss].") trninkxky(it, iss, itrn, xr_trn, flag='savetxt', outdir=outdir) # In[ ]:
#!/usr/bin/env python # coding: utf-8 # In[ ]: import sys sys.path.append("./src/") import numpy as np import matplotlib.pyplot as plt from diag_rb import rb_open, rb_get_tri_filelist from diag_geom import geom_set ### Read NetCDF data phi.*.nc by xarray ### xr_phi = rb_open('../post/data/phi.*.nc') xr_Al = rb_open('../post/data/Al.*.nc') xr_mom = rb_open('../post/data/mom.*.nc') xr_fxv = rb_open('../post/data/fxv.*.nc') xr_cnt = rb_open('../post/data/cnt.*.nc') xr_trn = rb_open('../post/data/trn.*.nc') tri_filelist = rb_get_tri_filelist('../post/data/tri.*.nc') xr_tri_list = [] for file in tri_filelist: xr_tri = rb_open(file + '.*.nc') xr_tri_list.append(xr_tri) # print("xr_phi:", xr_phi) # print("tri_filelist:", tri_filelist) ### Set geometric constants ### geom_set(headpath='../src/gkvp_header.f90', nmlpath="../gkvp_namelist.001",
if (__name__ == '__main__'): import os from diag_geom import geom_set from diag_rb import rb_open import time geom_set(headpath='../../src/gkvp_header.f90', nmlpath="../../gkvp_namelist.001", mtrpath='../../hst/gkvp.mtr.001') ### Examples of use ### ### fluxinvm_fxv ### #help(phiinz_connect) xr_phi = rb_open('../../post/data/phi.*.nc') xr_fxv = rb_open('../../post/data/fxv.*.nc') #print(xr_phi) from diag_geom import nprocz iss = 0 # Index of species rankz = int(nprocz / 2) zz = float(xr_fxv.zz[rankz]) print("# Plot flux_es[mu,vl] at t[it], s[iss], zz[rankz]. zz=", zz) outdir = '../data/fluxinvm_fxv/' os.makedirs(outdir, exist_ok=True) for it in range(0, len(xr_fxv['t']), 10): fluxinvm_fxv(it, iss, rankz, xr_phi, xr_fxv,
outfile.write('\n') else: # otherwise - return data array return data if (__name__ == '__main__'): #from diag_geom import geom_set from diag_geom import geom_set from diag_rb import rb_open import time global s_time s_time = time.time() xr_phi = rb_open('../../post/data/phi.*.nc') xr_Al = rb_open('../../post/data/Al.*.nc') xr_mom = rb_open('../../post/data/mom.*.nc') #print("\n***** 確認 if (__name__ == '__main__'):・・・ xr_momの属性 >>>\n", xr_mom, '\n') it = 5 iss = 0 imom = 6 geom_set(headpath='../../src/gkvp_header.f90', nmlpath="../../gkvp_namelist.001", mtrpath='../../hst/gkvp.mtr.001') fluidtotaltrans_loop(it, iss, xr_phi, xr_Al,
print('#Current flag=', flag) print( '#flag should be "flux_tube_coord","flux_tube_var","full_torus_coord","full_torus_var"' ) if (__name__ == '__main__'): import os from diag_geom import geom_set from diag_rb import rb_open import time ### Initialization ### geom_set(headpath='../../src/gkvp_header.f90', nmlpath="../../gkvp_namelist.001", mtrpath='../../hst/gkvp.mtr.001') xr_phi = rb_open('../../post/data/phi.*.nc') #print(xr_phi) #help(phiinxmf) from diag_geom import global_nz ### Examples of use ### print("# Output phi[z,y,x] at t[it] in flux_tube XMF format *.xmf") outdir = '../data/xmf_tube/' os.makedirs(outdir, exist_ok=True) it = 0 t1 = time.time() phiinxmf(it, xr_phi, flag="flux_tube_coord", nzw=5 * global_nz,