Merge pull request #3963 from CliMA/he/rft-conditionally-load

rft: conditionally load microphysics parameters

Author: haakon-e

src/parameters/create_parameters.jl

@@ -15,12 +15,15 @@ import StaticArrays as SA
 Construct the parameter set for any ClimaAtmos configuration.
 """
 ClimaAtmosParameters(config::AtmosConfig) =
-    ClimaAtmosParameters(config.toml_dict)
+    ClimaAtmosParameters(config.toml_dict, config.parsed_args)
 
 ClimaAtmosParameters(::Type{FT}) where {FT <: AbstractFloat} =
     ClimaAtmosParameters(CP.create_toml_dict(FT))
 
-function ClimaAtmosParameters(toml_dict::TD) where {TD <: CP.AbstractTOMLDict}
+function ClimaAtmosParameters(
+    toml_dict::TD,
+    parsed_args = nothing,
+) where {TD <: CP.AbstractTOMLDict}
     FT = CP.float_type(toml_dict)
 
     turbconv_params = TurbulenceConvectionParameters(toml_dict)
@@ -51,6 +54,16 @@ function ClimaAtmosParameters(toml_dict::TD) where {TD <: CP.AbstractTOMLDict}
     microphysics_0m_params = CM.Parameters.Parameters0M(toml_dict)
     microphysics_1m_params = microphys_1m_parameters(toml_dict)
     microphysics_2m_params = microphys_2m_parameters(toml_dict)
+
+    # If parsed_args is provided, we can save parameter space by only loading parameters
+    # needed for the microphysics model that is actually used.
+    if !isnothing(parsed_args)
+        cm_model = get_microphysics_model(parsed_args)
+        cm_model isa Microphysics0Moment || (microphysics_0m_params = nothing)
+        cm_model isa Union{Microphysics1Moment, Microphysics2Moment} ||
+            (microphysics_1m_params = nothing)
+        cm_model isa Microphysics2Moment || (microphysics_2m_params = nothing)
+    end
     MP0M = typeof(microphysics_0m_params)
     MP1M = typeof(microphysics_1m_params)
     MP2M = typeof(microphysics_2m_params)