Merge pull request #3856 from CliMA/sa/add_2M_number_adjustment

add 2M number adjustment tendencies

Author: sajjadazimi

NEWS.md

@@ -4,6 +4,8 @@ ClimaAtmos.jl Release Notes
 main
 -------
 
+PR [#3856](https://github.com/CliMA/ClimaAtmos.jl/pull/3856) adds number adjustment tendencies to the two-moment microphysics scheme.
+
 v0.30.3
 -------
 

Project.toml

@@ -44,10 +44,10 @@ AtmosphericProfilesLibrary = "0.1.7"
 ClimaComms = "0.6.8"
 ClimaCore = "0.14.34"
 ClimaDiagnostics = "0.2.12"
-ClimaParams = "0.10.31"
+ClimaParams = "0.10.32"
 ClimaTimeSteppers = "0.8.2"
 ClimaUtilities = "0.1.22"
-CloudMicrophysics = "0.24.0"
+CloudMicrophysics = "0.24.1"
 Dates = "1"
 ForwardDiff = ">= 0.10"
 Insolation = "0.9.2"

src/parameterized_tendencies/microphysics/microphysics_wrappers.jl

@@ -558,4 +558,48 @@ function compute_warm_precipitation_sources_2M!(
         )
     @. Snᵣᵖ += Sᵖ
 
+    # cloud liquid number adjustment for mass limits
+    # TODO: Once CCN number becomes a prognostic variable, these number adjustment tendencies
+    #       should be linked to it. Any increase in droplet number (source here) would imply 
+    #       a corresponding sink in CCN, and vice versa.
+    @. Sᵖ = CM2.number_increase_for_mass_limit(
+        mp.sb.numadj,
+        mp.sb.pdf_c.xc_max,
+        qₗ,
+        ρ,
+        ρ * nₗ,
+    )
+    @. S₂ᵖ =
+        -triangle_inequality_limiter(
+            -CM2.number_decrease_for_mass_limit(
+                mp.sb.numadj,
+                mp.sb.pdf_c.xc_min,
+                qₗ,
+                ρ,
+                ρ * nₗ,
+            ),
+            limit(nₗ, dt, 5),
+        )
+    @. Snₗᵖ = Sᵖ + S₂ᵖ
+    # rain number adjustment for mass limits
+    @. Sᵖ = CM2.number_increase_for_mass_limit(
+        mp.sb.numadj,
+        mp.sb.pdf_r.xr_max,
+        qᵣ,
+        ρ,
+        ρ * nᵣ,
+    )
+    @. S₂ᵖ =
+        -triangle_inequality_limiter(
+            -CM2.number_decrease_for_mass_limit(
+                mp.sb.numadj,
+                mp.sb.pdf_r.xr_min,
+                qᵣ,
+                ρ,
+                ρ * nᵣ,
+            ),
+            limit(nᵣ, dt, 5),
+        )
+    @. Snᵣᵖ += Sᵖ + S₂ᵖ
+
 end

src/prognostic_equations/implicit/implicit_tendency.jl

@@ -182,6 +182,29 @@ function implicit_vertical_advection_tendency!(Yₜ, Y, p, t)
             ),
         )
     end
+    if precip_model isa Microphysics2Moment
+        (; ᶜwnₗ, ᶜwnᵣ, ᶜwᵣ, ᶜwₛ) = p.precomputed
+        @. Yₜ.c.ρn_liq -= ᶜprecipdivᵥ(
+            ᶠinterp(Y.c.ρ * ᶜJ) / ᶠJ * ᶠright_bias(
+                Geometry.WVector(-(ᶜwnₗ)) * specific(Y.c.ρn_liq, Y.c.ρ),
+            ),
+        )
+        @. Yₜ.c.ρn_rai -= ᶜprecipdivᵥ(
+            ᶠinterp(Y.c.ρ * ᶜJ) / ᶠJ * ᶠright_bias(
+                Geometry.WVector(-(ᶜwnᵣ)) * specific(Y.c.ρn_rai, Y.c.ρ),
+            ),
+        )
+        @. Yₜ.c.ρq_rai -= ᶜprecipdivᵥ(
+            ᶠinterp(Y.c.ρ * ᶜJ) / ᶠJ * ᶠright_bias(
+                Geometry.WVector(-(ᶜwᵣ)) * specific(Y.c.ρq_rai, Y.c.ρ),
+            ),
+        )
+        @. Yₜ.c.ρq_sno -= ᶜprecipdivᵥ(
+            ᶠinterp(Y.c.ρ * ᶜJ) / ᶠJ * ᶠright_bias(
+                Geometry.WVector(-(ᶜwₛ)) * specific(Y.c.ρq_sno, Y.c.ρ),
+            ),
+        )
+    end
 
     # TODO - decide if this needs to be explicit or implicit
     #vertical_advection_of_water_tendency!(Yₜ, Y, p, t)