add ice_model config arg

Author: juliasloan25

experiments/ClimaEarth/cli_options.jl

@@ -175,6 +175,11 @@ function argparse_settings()
         help = "Directory containing ERA5 initial condition files (subseasonal mode). Filenames inferred from start_date [none (default)]. Generated with `https://github.com/CliMA/WeatherQuest`"
         arg_type = String
         default = nothing
+        # Ice model specific
+        "--ice_model"
+        help = "Sea ice model to use. [`prescribed` (default), `clima_seaice`]"
+        arg_type = String
+        default = "prescribed"
     end
     return s
 end

experiments/ClimaEarth/setup_run.jl

@@ -362,28 +362,24 @@ function CoupledSimulation(config_dict::AbstractDict)
 
         # TODO separate drivers to clean this up
         ## sea ice model
-        ice_sim = PrescribedIceSimulation(
-            FT;
-            tspan = tspan,
-            dt = component_dt_dict["dt_seaice"],
-            saveat = saveat,
-            space = boundary_space,
-            thermo_params = thermo_params,
-            comms_ctx,
-            start_date,
-            land_fraction,
-            sic_path = subseasonal_sic,
-        )
-
-        ## ocean model
-        ice_fraction = Interfacer.get_field(ice_sim, Val(:area_fraction))
-        ocean_fraction = FT(1) .- ice_fraction .- land_fraction
-
+        if ice_model == "prescribed"
+            ice_sim = PrescribedIceSimulation(
+                FT;
+                tspan = tspan,
+                dt = component_dt_dict["dt_seaice"],
+                saveat = saveat,
+                space = boundary_space,
+                thermo_params = thermo_params,
+                comms_ctx,
+                start_date,
+                land_fraction,
+                sic_path = subseasonal_sic,
+            )
+            ice_fraction = Interfacer.get_field(ice_sim, Val(:area_fraction))
+        elseif ice_model == "clima_seaice"
+            @assert sim_mode <: CMIPMode
 
-        if sim_mode <: CMIPMode
-            ########### TODO CLEAN UP ############
             # TODO how should we initialize ocean fraction when using ClimaSeaIce?
-            # Set up prescribed sea ice concentration object
             sic_data = try
                 joinpath(
                     @clima_artifact("historical_sst_sic", comms_ctx),
@@ -409,10 +405,14 @@ function CoupledSimulation(config_dict::AbstractDict)
             # Get initial SIC values and use them to calculate ice fraction
             ice_fraction = CC.Fields.zeros(space)
             evaluate!(ice_fraction, SIC_timevaryinginput, tspan[1])
-            ########### TODO CLEAN UP ############
+        else
+            error("Invalid ice model specified: $(ice_model)")
+        end
 
-            ocean_fraction = FT(1) .- ice_fraction .- land_fraction
+        ## ocean model
+        ocean_fraction = FT(1) .- ice_fraction .- land_fraction
 
+        if sim_mode <: CMIPMode
             stop_date = date(tspan[end] - tspan[begin])
             ocean_sim = OceananigansSimulation(
                 ocean_fraction,
@@ -422,8 +422,9 @@ function CoupledSimulation(config_dict::AbstractDict)
                 comms_ctx,
             )
 
-            ice_sim = ClimaSeaIceSimulation(ice_fraction, ocean_sim; output_dir)
-
+            if ice_model == "clima_seaice"
+                ice_sim = ClimaSeaIceSimulation(ice_fraction, ocean_sim; output_dir)
+            end
         else
             ocean_sim = PrescribedOceanSimulation(
                 FT,

experiments/ClimaEarth/user_io/arg_parsing.jl

@@ -170,8 +170,12 @@ function get_coupler_args(config_dict::Dict)
     bucket_albedo_type = config_dict["bucket_albedo_type"]
     bucket_initial_condition = config_dict["bucket_initial_condition"]
 
+    # Initial condition setting
     era5_initial_condition_dir = config_dict["era5_initial_condition_dir"]
 
+    # Ice model-specific information
+    ice_model = config_dict["ice_model"]
+
     return (;
         job_id,
         sim_mode,
@@ -203,6 +207,7 @@ function get_coupler_args(config_dict::Dict)
         bucket_initial_condition,
         parameter_files,
         era5_initial_condition_dir,
+        ice_model,
     )
 end