add qt and tracer tendency in viscous sponge

Author: szy21

src/parameterized_tendencies/sponge/viscous_sponge.jl

@@ -25,17 +25,10 @@ function viscous_sponge_cache(Y, viscous_sponge::ViscousSponge)
     return (; ᶜβ_viscous, ᶠβ_viscous)
 end
 
-function add_viscous_sponge_energy_tendency!(Yₜ, Y, p, t)
-    (; ᶜβ_viscous) = p.viscous_sponge
-    (; ᶜh_tot) = p.precomputed
-    ᶜρ = Y.c.ρ
-    @. Yₜ.c.ρe_tot += ᶜβ_viscous * wdivₕ(ᶜρ * gradₕ(ᶜh_tot))
-end
-
 function viscous_sponge_tendency!(Yₜ, Y, p, t, ::ViscousSponge)
     (; ᶜβ_viscous, ᶠβ_viscous) = p.viscous_sponge
+    (; ᶜh_tot, ᶜspecific) = p.precomputed
     ᶜuₕ = Y.c.uₕ
-    add_viscous_sponge_energy_tendency!(Yₜ, Y, p, t)
     @. Yₜ.c.uₕ +=
         ᶜβ_viscous * (
             wgradₕ(divₕ(ᶜuₕ)) - Geometry.project(
@@ -45,4 +38,11 @@ function viscous_sponge_tendency!(Yₜ, Y, p, t, ::ViscousSponge)
         )
     @. Yₜ.f.u₃.components.data.:1 +=
         ᶠβ_viscous * wdivₕ(gradₕ(Y.f.u₃.components.data.:1))
+
+    @. Yₜ.c.ρe_tot += ᶜβ_viscous * wdivₕ(Y.c.ρ * gradₕ(ᶜh_tot))
+    for (ᶜρχₜ, ᶜχ, χ_name) in matching_subfields(Yₜ.c, ᶜspecific)
+        χ_name == :e_tot && continue
+        @. ᶜρχₜ += ᶜβ_viscous * wdivₕ(Y.c.ρ * gradₕ(ᶜχ))
+        @. Yₜ.c.ρ += ᶜβ_viscous * wdivₕ(Y.c.ρ * gradₕ(ᶜχ))
+    end
 end