fix entrainment and detrainment in TKE equation

CliMA · Dec 21, 2023 · 1bb83b6 · 1bb83b6
1 parent c621bcb
commit 1bb83b6
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/prognostic_equations/edmfx_tke.jl b/src/prognostic_equations/edmfx_tke.jl
@@ -50,11 +50,14 @@ function edmfx_tke_tendency!(
         # entrainment and detraiment
         # using ᶜu⁰ and local geometry results in allocation
         for j in 1:n
+            ᶜρaʲ_colidx =
+                turbconv_model isa PrognosticEDMFX ?
+                Y.c.sgsʲs.:($j).ρa[colidx] : p.precomputed.ᶜρaʲs.:($j)[colidx]
             @. Yₜ.c.sgs⁰.ρatke[colidx] +=
-                ᶜρa⁰[colidx] * (
-                    ᶜentrʲs.:($$j)[colidx] * 1 / 2 * norm_sqr(
+                ᶜρaʲ_colidx * (
+                    ᶜdetrʲs.:($$j)[colidx] * 1 / 2 * norm_sqr(
                         ᶜinterp(ᶠu³⁰[colidx]) - ᶜinterp(ᶠu³ʲs.:($$j)[colidx]),
-                    ) - ᶜdetrʲs.:($$j)[colidx] * ᶜtke⁰[colidx]
+                    ) - ᶜentrʲs.:($$j)[colidx] * ᶜtke⁰[colidx]
                 )
         end
         # pressure work