diff --git a/freegs4e/jtor.py b/freegs4e/jtor.py index ba46e3a..6e61e2a 100644 --- a/freegs4e/jtor.py +++ b/freegs4e/jtor.py @@ -973,10 +973,6 @@ def Jtor_part2( psi_bndry = psi[0, 0] self.psi_bndry = psi_bndry - # grid sizes - dR = R[1, 0] - R[0, 0] - dZ = Z[0, 1] - Z[0, 0] - if torefine: R = 1.0 * refineR @@ -1046,6 +1042,10 @@ def Jtor_part2( # if Ip normalisation is required, do it if self.Ip_logic: + # grid sizes + dR = R[1, 0] - R[0, 0] + dZ = Z[0, 1] - Z[0, 0] + jtorIp = np.sum(Jtor) if jtorIp == 0: self.problem_psi = psi