add option to read land ICs from a file

Author: juliasloan25

.buildkite/pipeline.yml

@@ -219,6 +219,13 @@ steps:
           slurm_ntasks: 1
           slurm_mem: 20GB
 
+      - label: "AMIP: integrated land non-spun up initial condition test"
+        command: "julia --color=yes --project=experiments/ClimaEarth/ experiments/ClimaEarth/run_amip.jl --config_file $CONFIG_PATH/amip_land_ic.yml --job_id amip_land_ic"
+        artifact_paths: "experiments/ClimaEarth/output/amip_land_ic/artifacts/*"
+        agents:
+          slurm_ntasks: 1
+          slurm_mem: 20GB
+
       - label: "AMIP - Float64 + hourly checkpoint"
         key: "amip"
         command: "julia --color=yes --project=experiments/ClimaEarth/ experiments/ClimaEarth/run_amip.jl --config_file $CONFIG_PATH/amip_coarse_ft64_hourly_checkpoints.yml --job_id amip_coarse_ft64_hourly_checkpoints"

NEWS.md

@@ -6,6 +6,13 @@ ClimaCoupler.jl Release Notes
 
 ### ClimaCoupler features
 
+#### Add option for integrated land spun up IC. PR[#1318](https://github.com/CliMA/ClimaCoupler.jl/pull/1318)
+
+Adds a boolean flag `land_spun_up_ic` that allows the user to request reading integrated land
+initial conditions from a saved simulation, rather than setting them with default values.
+If this option is true, the land/sea mask must be used, since the spun-up ICs are only defined
+over land (i.e. this cannot be used in the `slabplanet_terra` mode). The default is true.
+
 #### Use EN4 dataset for ocean initial conditions and forcing PR[#1425](https://github.com/CliMA/ClimaCoupler.jl/pull/1425)
 
 Use EN4 instead of ECCO for ocean initial conditions and forcing, to avoid

config/ci_configs/amip_land_ic.yml

@@ -0,0 +1,19 @@
+apply_limiter: false
+co2_model: "maunaloa"
+dt: "150secs"
+dt_cpl: "150secs"
+dt_rad: "1hours"
+dz_bottom: 30
+dz_top: 3000
+h_elem: 4
+land_model: "integrated"
+land_spun_up_ic: false
+mode_name: "amip"
+moist: "equil"
+precip_model: "0M"
+rad: "allskywithclear"
+rayleigh_sponge: true
+t_end: "300secs"
+vert_diff: "true"
+z_elem: 50
+z_stretch: false

experiments/ClimaEarth/cli_options.jl

@@ -156,6 +156,10 @@ function argparse_settings()
         help = "Boolean flag indicating whether to compute and output land model diagnostics [`true` (default), `false`]"
         arg_type = Bool
         default = true
+        "--land_spun_up_ic"
+        help = "Boolean flag to indicate whether to use integrated land initial conditions from spun up state [`true` (default), `false`]"
+        arg_type = Bool
+        default = true
         # BucketModel specific
         "--bucket_albedo_type"
         help = "Access bucket surface albedo information from data file. [`map_static` (default), `function`, `map_temporal`]"

experiments/ClimaEarth/components/land/climaland_integrated.jl

@@ -63,10 +63,11 @@ function ClimaLandSimulation(
     start_date::Dates.DateTime,
     output_dir::String,
     area_fraction,
-    nelements::Tuple{Int, Int} = (101, 10),
+    nelements::Tuple{Int, Int} = (101, 15),
     depth::FT = FT(50),
     dz_tuple::Tuple{FT, FT} = FT.((10.0, 0.05)),
     shared_surface_space = nothing,
+    land_spun_up_ic::Bool = true,
     saveat::Vector{TT} = [tspan[1], tspan[2]],
     surface_elevation = nothing,
     land_temperature_anomaly::String = "amip",
@@ -171,19 +172,34 @@ function ClimaLandSimulation(
     orog_adjusted_T = CC.Fields.Field(orog_adjusted_T_data, subsurface_space)
     orog_adjusted_T_surface = CC.Fields.Field(CC.Fields.level(orog_adjusted_T_data, 1), surface_space)
 
-    # Set initial conditions for the state
-    θ_r, ν, ρc_ds = soil_params.θ_r, soil_params.ν, soil_params.ρc_ds
-    @. Y.soil.ϑ_l = θ_r + (ν - θ_r) / 2
-    Y.soil.θ_i .= FT(0.0)
-    ρc_s = CL.Soil.volumetric_heat_capacity.(Y.soil.ϑ_l, Y.soil.θ_i, ρc_ds, earth_param_set)
-    Y.soil.ρe_int .= CL.Soil.volumetric_internal_energy.(Y.soil.θ_i, ρc_s, orog_adjusted_T, earth_param_set)
+    # Set initial conditions that aren't read in from file
     Y.soilco2.C .= FT(0.000412) # set to atmospheric co2, mol co2 per mol air
     Y.canopy.hydraulics.ϑ_l.:1 .= canopy_component_args.hydraulics.parameters.ν
     @. Y.canopy.energy.T = orog_adjusted_T_surface
 
-    Y.snow.S .= FT(0)
-    Y.snow.S_l .= FT(0)
-    Y.snow.U .= FT(0)
+    # Read in initial conditions for snow and soil from file, if requested
+    θ_r, ν, ρc_ds = soil_params.θ_r, soil_params.ν, soil_params.ρc_ds
+    if land_spun_up_ic
+        # Set snow T first to use in computing snow internal energy
+        @. p.snow.T = orog_adjusted_T_surface
+
+        # Read in initial conditions for snow and soil
+        ic_path = CL.Artifacts.soil_ic_2008_50m_path()
+        CL.Simulations.set_snow_initial_conditions!(Y, p, surface_space, ic_path, model.snow.parameters)
+
+        T_bounds = extrema(Y.canopy.energy.T)
+        CL.Simulations.set_soil_initial_conditions!(Y, ν, θ_r, subsurface_space, ic_path, model.soil, T_bounds)
+    else
+        # Set initial conditions for the state
+        @. Y.soil.ϑ_l = θ_r + (ν - θ_r) / 2
+        Y.soil.θ_i .= FT(0.0)
+        ρc_s = CL.Soil.volumetric_heat_capacity.(Y.soil.ϑ_l, Y.soil.θ_i, ρc_ds, earth_param_set)
+        Y.soil.ρe_int .= CL.Soil.volumetric_internal_energy.(Y.soil.θ_i, ρc_s, orog_adjusted_T, earth_param_set)
+
+        Y.snow.S .= FT(0)
+        Y.snow.S_l .= FT(0)
+        Y.snow.U .= FT(0)
+    end
 
     # Initialize the surface temperature so the atmosphere can compute radiation.
     # This cache variable is normally computed using `p.drivers.LW_d`, which is

experiments/ClimaEarth/setup_run.jl

@@ -121,9 +121,10 @@ function CoupledSimulation(config_dict::AbstractDict)
         diagnostics_dt,
         evolving_ocean,
         land_model,
+        land_temperature_anomaly,
+        land_spun_up_ic,
         bucket_albedo_type,
         bucket_initial_condition,
-        land_temperature_anomaly,
         energy_check,
         use_coupler_diagnostics,
         output_dir_root,
@@ -298,6 +299,8 @@ function CoupledSimulation(config_dict::AbstractDict)
                 output_dir = land_output_dir,
                 area_fraction = land_fraction,
                 shared_surface_space,
+                land_spun_up_ic,
+                saveat,
                 surface_elevation,
                 land_temperature_anomaly,
                 use_land_diagnostics,

experiments/ClimaEarth/test/amip_test.yml

@@ -10,6 +10,7 @@ h_elem: 8
 checkpoint_dt: "720hours"
 bucket_albedo_type: "map_temporal"
 land_model: "integrated"
+land_spun_up_ic: false
 mode_name: "amip"
 netcdf_interpolation_num_points: [90, 45, 31]
 netcdf_output_at_levels: true

experiments/ClimaEarth/user_io/arg_parsing.jl

@@ -119,10 +119,11 @@ function get_coupler_args(config_dict::Dict)
 
     # ClimaLand-specific information
     land_model = config_dict["land_model"]
-    bucket_albedo_type = config_dict["bucket_albedo_type"]
-    bucket_initial_condition = config_dict["bucket_initial_condition"]
     land_temperature_anomaly = config_dict["land_temperature_anomaly"]
     use_land_diagnostics = config_dict["use_land_diagnostics"]
+    land_spun_up_ic = config_dict["land_spun_up_ic"]
+    bucket_albedo_type = config_dict["bucket_albedo_type"]
+    bucket_initial_condition = config_dict["bucket_initial_condition"]
 
     return (;
         job_id,
@@ -147,10 +148,11 @@ function get_coupler_args(config_dict::Dict)
         rmse_check,
         output_dir_root,
         land_model,
-        bucket_albedo_type,
-        bucket_initial_condition,
         land_temperature_anomaly,
+        land_spun_up_ic,
         use_land_diagnostics,
+        bucket_albedo_type,
+        bucket_initial_condition,
         parameter_files,
     )
 end