Add function to calculate zeff

Author: orso82

src/physics/profiles.jl

@@ -642,6 +642,28 @@ function lump_ions_as_bulk_and_impurity(cp1d::IMAS.core_profiles__profiles_1d{T}
     return ions2
 end
 
+"""
+    zeff(cp1d::IMAS.core_profiles__profiles_1d; temperature_dependent_ionization_state::Bool=true)
+
+Returns plasma effective charge
+
+`temperature_dependent_ionization_state` evaluates Zeff with average ionization state of an ion at a given temperature
+"""
+function zeff(cp1d::IMAS.core_profiles__profiles_1d; temperature_dependent_ionization_state::Bool=true)
+    num = zero(cp1d.grid.rho_tor_norm)
+    den = zero(cp1d.grid.rho_tor_norm)
+    for ion in cp1d.ion
+        if temperature_dependent_ionization_state
+            Zi = avgZ(ion.element[1].z_n, ion.temperature)
+        else
+            Zi = ion.element[1].z_n
+        end
+        num .+= ion.density .* Zi .^ 2
+        den .+= ion.density .* Zi
+    end
+    return num ./ cp1d.electrons.density
+end
+
 """
     avgZ(Z::Float64,Ti::T)::T