Merge pull request #3930 from CliMA/zs/updraft_w_viscous_sponge

add viscous sponge to updraft u3

Author: szy21

src/prognostic_equations/remaining_tendency.jl

@@ -139,8 +139,8 @@ NVTX.@annotate function additional_tendency!(Yₜ, Y, p, t)
     ᶜuₕ = Y.c.uₕ
     ᶠu₃ = Y.f.u₃
     ᶜρ = Y.c.ρ
-    (; radiation_mode, moisture_model, rayleigh_sponge, viscous_sponge) =
-        p.atmos
+    (; radiation_mode, moisture_model, turbconv_model) = p.atmos
+    (; rayleigh_sponge, viscous_sponge) = p.atmos
     (; ls_adv, scm_coriolis) = p.atmos
     (; params) = p
     thermo_params = CAP.thermodynamics_params(params)
@@ -182,6 +182,15 @@ NVTX.@annotate function additional_tendency!(Yₜ, Y, p, t)
         end
     end
 
+    if turbconv_model isa PrognosticEDMFX
+        n = n_mass_flux_subdomains(turbconv_model)
+        for j in 1:n
+            ᶠu₃ʲ = Y.f.sgsʲs.:($j).u₃
+            vst_u₃ʲ = viscous_sponge_tendency_u₃(ᶠu₃ʲ, viscous_sponge)
+            @. Yₜ.f.sgsʲs.:($$j).u₃.components.data.:1 += vst_u₃ʲ
+        end
+    end
+
     # Held Suarez tendencies
     @. Yₜ.c.uₕ += hs_tendency_uₕ
     @. Yₜ.c.ρe_tot += hs_tendency_ρe_tot