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,
