add radiation tendency to updraft

Author: szy21

src/parameterized_tendencies/radiation/radiation.jl

@@ -419,7 +419,19 @@ end
 
 function radiation_tendency!(Yₜ, Y, p, t, ::RRTMGPI.AbstractRRTMGPMode)
     (; ᶠradiation_flux) = p.radiation
+    (; turbconv_model) = p.atmos
     @. Yₜ.c.ρe_tot -= ᶜdivᵥ(ᶠradiation_flux)
+    # Apply radiation tendency to updrafts in prognostic EDMF. We use the
+    # grid-mean radiation as an approximation for updraft radiation.
+    # Note: Radiation is not applied to updrafts in diagnostic EDMF because updrafts 
+    # are typically absent in the stratosphere where radiation is more important.
+    if turbconv_model isa PrognosticEDMFX
+        (; ᶜρʲs) = p.precomputed
+        n = n_mass_flux_subdomains(turbconv_model)
+        for j in 1:n
+            @. Yₜ.c.sgsʲs.:($$j).mse -= ᶜdivᵥ(ᶠradiation_flux) / ᶜρʲs.:($$j)
+        end
+    end
     return nothing
 end