add bucket initial condition

config/ci_configs/amip_default.yml

@@ -6,6 +6,7 @@ dt_save_to_sol: "1days"
 dz_bottom: 30
 dz_top: 3000
 h_elem: 4
+land_initial_condition: "experiments/ClimaEarth/test/bucket_ic_august.nc"
 mode_name: "amip"
 moist: "equil"
 precip_model: "0M"

experiments/ClimaEarth/cli_options.jl

@@ -155,6 +155,10 @@ function argparse_settings()
         help = "Access land surface albedo information from data file. [`map_static` (default), `function`, `map_temporal`]"
         arg_type = String
         default = "map_static" # to be replaced by land config file, when available
+        "--land_initial_condition"
+        help = "A file path for a NetCDF file (read documentation about requirements)"
+        arg_type = String
+        default = ""
         "--land_temperature_anomaly"
         help = "Type of temperature anomaly for bucket model. [`amip`, `aquaplanet` (default)]"
         arg_type = String

experiments/ClimaEarth/components/land/climaland_bucket.jl

@@ -52,6 +52,7 @@ function bucket_init(
     tspan::Tuple{Float64, Float64},
     config::String,
     albedo_type::String,
+    land_initial_condition::String,
     land_temperature_anomaly::String,
     output_dir::String;
     space,
@@ -138,6 +139,46 @@ function bucket_init(
     Y.bucket.Ws .= 0.0
     Y.bucket.σS .= 0.0
 
+    # overwrite initial conditions
+    if !isempty(land_initial_condition)
+        surface_space = domain.space.surface
+        subsurface_space = domain.space.subsurface
+        regridder_type = :InterpolationsRegridder
+        extrapolation_bc = (
+            Interpolations.Periodic(),
+            Interpolations.Flat(),
+            Interpolations.Flat(),
+        )
+        Y.bucket.W .= SpaceVaryingInput(
+            land_initial_condition,
+            "W",
+            surface_space;
+            regridder_type,
+            regridder_kwargs = (; extrapolation_bc,),
+        )
+        Y.bucket.Ws .= SpaceVaryingInput(
+            land_initial_condition,
+            "Ws",
+            surface_space;
+            regridder_type,
+            regridder_kwargs = (; extrapolation_bc,),
+        )
+        Y.bucket.T .= SpaceVaryingInput(
+            land_initial_condition,
+            "T",
+            subsurface_space;
+            regridder_type,
+            regridder_kwargs = (; extrapolation_bc,),
+        )
+        Y.bucket.σS .= SpaceVaryingInput(
+            land_initial_condition,
+            "S",
+            surface_space;
+            regridder_type,
+            regridder_kwargs = (; extrapolation_bc,),
+        )
+    end
+
     # Set initial aux variable values
     set_initial_cache! = CL.make_set_initial_cache(model)
     set_initial_cache!(p, Y, tspan[1])

experiments/ClimaEarth/run_amip.jl

@@ -118,6 +118,7 @@ add_extra_diagnostics!(config_dict)
     turb_flux_partition,
     land_domain_type,
     land_albedo_type,
+    land_initial_condition,
     land_temperature_anomaly,
     energy_check,
     conservation_softfail,
@@ -253,6 +254,7 @@ if mode_name == "amip"
         tspan,
         land_domain_type,
         land_albedo_type,
+        land_initial_condition,
         land_temperature_anomaly,
         land_output_dir;
         dt = component_dt_dict["dt_land"],
@@ -351,6 +353,7 @@ elseif mode_name in ("slabplanet", "slabplanet_aqua", "slabplanet_terra")
         tspan,
         land_domain_type,
         land_albedo_type,
+        land_initial_condition,
         land_temperature_anomaly,
         land_output_dir;
         dt = component_dt_dict["dt_land"],
@@ -401,6 +404,7 @@ elseif mode_name == "slabplanet_eisenman"
         tspan,
         land_domain_type,
         land_albedo_type,
+        land_initial_condition,
         land_temperature_anomaly,
         land_output_dir;
         dt = component_dt_dict["dt_land"],

experiments/ClimaEarth/test/amip_bucket_ic.jl

@@ -0,0 +1,11 @@
+## AMIP testing
+# This script runs a coarse AMIP simulation with the bucket initial condition
+# read from a file. The purpose of this test is to check if the bucket
+# initial condition is read correctly.
+
+# Specify the config file and job id
+push!(ARGS, "--config_file", joinpath(@__DIR__, "amip_bucket_ic.yml"))
+push!(ARGS, "--job_id", "amip_bucket_ic")
+
+# Run the AMIP test
+include("../run_amip.jl")

experiments/ClimaEarth/test/amip_bucket_ic.yml

@@ -0,0 +1,18 @@
+apply_limiter: false
+dt: "150secs"
+dt_cpl: "150secs"
+dt_rad: "1hours"
+dt_save_to_sol: "1days"
+dz_bottom: 30
+dz_top: 3000
+h_elem: 4
+land_initial_condition: "experiments/ClimaEarth/test/bucket_ic_august.nc"
+mode_name: "amip"
+moist: "equil"
+precip_model: "0M"
+rad: "gray"
+rayleigh_sponge: true
+t_end: "300secs"
+vert_diff: "true"
+z_elem: 50
+z_stretch: false

experiments/ClimaEarth/user_io/arg_parsing.jl

@@ -85,6 +85,7 @@ function get_coupler_args(config_dict::Dict)
     # ClimaLand-specific information
     land_domain_type = config_dict["land_domain_type"]
     land_albedo_type = config_dict["land_albedo_type"]
+    land_initial_condition = config_dict["land_initial_condition"]
     land_temperature_anomaly = config_dict["land_temperature_anomaly"]
     use_land_diagnostics = config_dict["use_land_diagnostics"]
 
@@ -116,6 +117,7 @@ function get_coupler_args(config_dict::Dict)
         plot_diagnostics,
         land_domain_type,
         land_albedo_type,
+        land_initial_condition,
         land_temperature_anomaly,
         use_land_diagnostics,
     )