Prep some things for using OrdinaryDiffEq.jl

integration_tests/utils/initial_conditions.jl

@@ -40,6 +40,7 @@ function initialize_updrafts(edmf, grid, state, up::TC.UpdraftVariables, gm::TC.
     aux_gm = TC.center_aux_grid_mean(state)
     prog_up = TC.center_prog_updrafts(state)
     prog_up_f = TC.face_prog_updrafts(state)
+    ρ0_c = TC.center_ref_state(state).ρ0
     @inbounds for i in 1:(up.n_updrafts)
         @inbounds for k in TC.real_face_indices(grid)
             aux_up_f[i].w[k] = 0
@@ -62,6 +63,7 @@ function initialize_updrafts(edmf, grid, state, up::TC.UpdraftVariables, gm::TC.
         end
 
         aux_up[i].area[kc_surf] = up.updraft_fraction / up.n_updrafts
+        prog_up[i].ρarea[kc_surf] = ρ0_c[kc_surf] * aux_up[i].area[kc_surf]
     end
     return
 end

integration_tests/utils/main.jl

@@ -2,12 +2,17 @@ import Logging
 import TerminalLoggers
 Logging.global_logger(TerminalLoggers.TerminalLogger())
 
+import UnPack
 import JSON
 import ArgParse
 import TurbulenceConvection
+
 import ClimaCore
 const CC = ClimaCore
 
+import OrdinaryDiffEq
+const ODE = OrdinaryDiffEq
+
 include("initial_conditions.jl")
 include("parameter_set.jl")
 include("Cases.jl")

src/TurbulenceConvection.jl

@@ -18,6 +18,7 @@ import CLIMAParameters
 import OrdinaryDiffEq
 import CloudMicrophysics
 import SurfaceFluxes
+import UnPack
 
 const ODE = OrdinaryDiffEq
 const CC = ClimaCore