Skip to content

Commit

Permalink
print psi_rzq_omp
Browse files Browse the repository at this point in the history
  • Loading branch information
EJCT22 committed Oct 29, 2024
1 parent f06120f commit cdb0f6d
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion source/heatfluxClass.py
Original file line number Diff line number Diff line change
Expand Up @@ -578,9 +578,11 @@ def from_rzq_profile(self, rzq_data, PFC):
psi_rzq = PFC.ep.psiFunc.ev(R,Z) #convert (r,z) coordinate to psi
psi_rzq_omp = self.map_R_psi(psi_rzq, PFC) #map the psi_rzq to OMP

print("Q is:", Q)
print("psi_rzq is:", psi_rzq)
print("psi_rzq_omp is", psi_rzq_omp)

q_data = np.stack(Q, psi_rzq, psi_rzq_omp)
np.savetxt("q_data", q_data, delimiter=",",header="q,psi_rzq,psi_omp")
q_interp = scinter.UnivariateSpline(psi_rzq_omp, Q, s = None, ext = 'const') #interpolate the value of q at OMP
psi = self.map_R_psi(PFC.psimin, PFC) #map PFC centers to OMP
q1 = q_interp(psi) #calculate the value of q basesd on mapped to OMP PFC centers
Expand Down

0 comments on commit cdb0f6d

Please sign in to comment.