Add viscous sponge for updraft quantities

Author: costachris

src/parameterized_tendencies/sponge/viscous_sponge.jl

@@ -46,3 +46,10 @@ function viscous_sponge_tendency_tracer(ᶜρ, ᶜχ, s)
     zmax = z_max(axes(ᶠz))
     return @. lazy(β_viscous(s, ᶜz, zmax) * wdivₕ(ᶜρ * gradₕ(ᶜχ)))
 end
+
+function viscous_sponge_tendency_sgs(ᶜvar, s)
+    s isa Nothing && return NullBroadcasted()
+    (; ᶜz, ᶠz) = z_coordinate_fields(axes(ᶜvar))
+    zmax = z_max(axes(ᶠz))
+    return @. lazy(β_viscous(s, ᶜz, zmax) * wdivₕ(gradₕ(ᶜvar)))
+end

src/prognostic_equations/remaining_tendency.jl

@@ -182,6 +182,12 @@ NVTX.@annotate function additional_tendency!(Yₜ, Y, p, t)
         end
     end
 
+    vst_mse = viscous_sponge_tendency_sgs(Y.c.sgsʲs.:(1).mse, viscous_sponge)
+    vst_q_tot = viscous_sponge_tendency_sgs(Y.c.sgsʲs.:(1).q_tot, viscous_sponge)
+
+    @. Yₜ.c.sgsʲs.:(1).mse += vst_mse
+    @. Yₜ.c.sgsʲs.:(1).q_tot += vst_q_tot
+
     # Held Suarez tendencies
     @. Yₜ.c.uₕ += hs_tendency_uₕ
     @. Yₜ.c.ρe_tot += hs_tendency_ρe_tot