Skip to content

Conversion functions for input files between InputTJLF and InputCGYRO #2

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Conversion functions for input files between InputTJLF and InputCGYRO #2

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
190 changes: 190 additions & 0 deletions src/utils.jl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -120,3 +120,193 @@ function parse_out_tglf_gbflux(lines::String; outnames::NTuple{4,String}=("Gam/G
end
return out
end


"""
convertInputFromCGYRO(InputCGYRO::InputTJLF)

convert input file from InputCGYRO to InputTJLF
#"""
function convert_input_CGYRO_to_TJLF(inputCGYRO::InputCGYRO, nky::Integer)
#get number of species
ns = inputCGYRO.N_SPECIES
#generate TJLF structure
inputTJLF = InputTJLF{Float64}(ns, nky)

inputTJLF.NS=ns
inputTJLF.RMIN_LOC = inputCGYRO.RMIN
inputTJLF.RMAJ_LOC = inputCGYRO.RMAJ
inputTJLF.Q_LOC = abs(inputCGYRO.Q)
inputTJLF.Q_PRIME_LOC=inputCGYRO.S*(inputTJLF.Q_LOC/inputTJLF.RMIN_LOC)^2

inputTJLF.VEXB_SHEAR=inputCGYRO.GAMMA_E
inputTJLF.BETAE=inputCGYRO.BETAE_UNIT
inputTJLF.XNUE=inputCGYRO.NU_EE

pressure_sum=0.0
pressure_grad_sum=0.0
for i in 1:ns
n=getproperty(inputCGYRO, Symbol("DENS_" * string(i)))
T=getproperty(inputCGYRO, Symbol("TEMP_" * string(i)))
dn=getproperty(inputCGYRO, Symbol("DLNNDR_" * string(i)))
dT=getproperty(inputCGYRO, Symbol("DLNTDR_" * string(i)))
pressure = n*T
pressure_sum += pressure
pressure_grad_sum += dn*T+dT*n
end
#should calculate it properly, but BETAE_UNIT and beta_star are very close with BETA_STAR_SCALE=1
beta_star=inputCGYRO.BETAE_UNIT# *pressure_grad_sum/pressure_sum

inputTJLF.P_PRIME_LOC=(-beta_star/8/pi)*abs(inputTJLF.Q_LOC/inputTJLF.RMIN_LOC)
#inputTJLF.DEBYE=inputCGYRO.LAMBDA_STAR
inputTJLF.ZMAJ_LOC=inputCGYRO.ZMAG
inputTJLF.DZMAJDX_LOC=inputCGYRO.DZMAG
inputTJLF.DRMAJDX_LOC=inputCGYRO.SHIFT
inputTJLF.KAPPA_LOC=inputCGYRO.KAPPA
inputTJLF.S_KAPPA_LOC=inputCGYRO.S_KAPPA
inputTJLF.DELTA_LOC=inputCGYRO.DELTA
inputTJLF.S_DELTA_LOC=inputCGYRO.S_DELTA
inputTJLF.ZETA_LOC=inputCGYRO.ZETA
inputTJLF.S_ZETA_LOC=inputCGYRO.S_ZETA



for i in 1:ns
inputTJLF.VPAR[i]=abs(inputCGYRO.MACH)

end

zeff = 0.0

# do not use electrons to calculate zeff
for i in 1:ns-1
zeff += (getproperty(inputCGYRO, Symbol("Z_" * string(i)))^2* getproperty(inputCGYRO, Symbol("DENS_" * string(i))))/getproperty(inputCGYRO, Symbol("DENS_" * string(ns)))

end

inputTJLF.ZEFF=zeff
geom_array=["SHAPE_SIN", "SHAPE_S_SIN"]
for i in 3:6
for prefix in geom_array
x = Symbol(prefix*string(i))
if x in fieldnames(typeof(inputCGYRO))
cgyro_geom = getproperty(inputCGYRO, x)
setproperty!(inputTJLF, Symbol(prefix*string(i)), cgyro_geom)
end
end
end


geom_array=["SHAPE_COS", "SHAPE_S_COS"]
for i in 0:6
for prefix in geom_array
x = Symbol(prefix*string(i))
if x in fieldnames(typeof(inputCGYRO))
cgyro_geom = getproperty(inputCGYRO, x)
setproperty!(inputTJLF, Symbol(prefix*string(i)), cgyro_geom)
end
end
end

plasma_array=["Z_","MASS_","DENS_","TEMP_","DLNNDR_","DLNTDR_"]
tglf_plasma_array=["ZS","MASS","AS","TAUS","RLNS","RLTS"]
#tglf starts with electrons, while in CGYRO it is the last species
for i in 2:ns
for (k,prefix) in enumerate(plasma_array)

cgyro_plasma = getproperty(inputCGYRO, Symbol(prefix*string(i-1)))
cgyro_plasma_elec=getproperty(inputCGYRO, Symbol(prefix*string(ns)))
tglf_dic=getproperty(inputTJLF, Symbol(tglf_plasma_array[k]))
tglf_dic[i]=cgyro_plasma
tglf_dic[1]=cgyro_plasma_elec
end
end


inputTJLF.SIGN_BT=inputCGYRO.BTCCW
inputTJLF.SIGN_IT=inputCGYRO.IPCCW


return inputTJLF

end

"""
convert InputTJLF into InputCGYRO file. Create the same structure as InputCGYRO function working with dd

convert input file from InputCGYRO to InputTJLF
#"""

function convert_input_TJLF_to_CGYRO(inputTJLF::InputTJLF)
#get number of species
ns=inputTJLF.NS
#generate InputCGYRO structure
inputCGYRO = InputCGYRO()


inputCGYRO.RMIN = inputTJLF.RMIN_LOC
inputCGYRO.RMAJ = inputTJLF.RMAJ_LOC
inputCGYRO.SHIFT = inputTJLF.DRMAJDX_LOC
inputCGYRO.KAPPA = inputTJLF.KAPPA_LOC
inputCGYRO.S_KAPPA = inputTJLF.S_KAPPA_LOC
inputCGYRO.DELTA = inputTJLF.DELTA_LOC
inputCGYRO.S_DELTA = inputTJLF.S_DELTA_LOC
inputCGYRO.ZETA = inputTJLF.ZETA_LOC
inputCGYRO.S_ZETA = inputTJLF.S_ZETA_LOC
inputCGYRO.ZMAG = inputTJLF.ZMAJ_LOC
inputCGYRO.DZMAG = inputTJLF.DZMAJDX_LOC

inputCGYRO.Q= inputTJLF.Q_LOC
inputCGYRO.S = (inputTJLF.Q_LOC/inputTJLF.RMIN_LOC)^2/inputTJLF.Q_PRIME_LOC

inputCGYRO.BTCCW = inputTJLF.SIGN_BT
inputCGYRO.IPCCW = inputTJLF.SIGN_IT
inputCGYRO.BETAE_UNIT = inputTJLF.BETAE
inputCGYRO.N_SPECIES = inputTJLF.NS

inputCGYRO.GAMMA_E = inputTJLF.VEXB_SHEAR

inputCGYRO.NU_EE = inputTJLF.XNUE
inputCGYRO.MACH = inputTJLF.VPAR[1]


geom_array=["SHAPE_SIN", "SHAPE_S_SIN"]
for i in 3:6
for prefix in geom_array
x = Symbol(prefix*string(i))
tglf_value = getproperty(inputTJLF, x)
if (tglf_value !==missing) && (x in fieldnames(typeof(inputCGYRO)))
setproperty!(inputCGYRO, Symbol(prefix*string(i)), tglf_value)
end
end
end


geom_array=["SHAPE_COS", "SHAPE_S_COS"]
for i in 0:6
for prefix in geom_array
x = Symbol(prefix*string(i))
tglf_value = getproperty(inputTJLF, x)
if (tglf_value !==missing) && (x in fieldnames(typeof(inputCGYRO)))
setproperty!(inputCGRYO, Symbol(prefix*string(i)), tglf_value)
end
end
end

plasma_array=["Z_","MASS_","DENS_","TEMP_","DLNNDR_","DLNTDR_"]
tglf_plasma_array=["ZS","MASS","AS","TAUS","RLNS","RLTS"]
#tglf starts with electrons, while in CGYRO it is the last species
for i in 2:ns
for (k,prefix) in enumerate(plasma_array)
tglf_dic=getproperty(inputTJLF, Symbol(tglf_plasma_array[k]))

setproperty!(inputCGYRO, Symbol(prefix*string(i-1)), tglf_dic[i])
setproperty!(inputCGYRO, Symbol(prefix*string(ns)), tglf_dic[1])

end
end


return inputCGYRO

end