add callbacks to read in prescribed data daily

Author: juliasloan25

experiments/ClimaEarth/components/ocean/prescr_ocean.jl

@@ -3,6 +3,7 @@ import ClimaUtilities.ClimaArtifacts: @clima_artifact
 import Interpolations # triggers InterpolationsExt in ClimaUtilities
 import Thermodynamics as TD
 import ClimaCoupler: Checkpointer, FieldExchanger, Interfacer
+import ClimaDiagnostics.Schedules: EveryCalendarDtSchedule
 
 """
     PrescribedOceanSimulation{C}
@@ -84,6 +85,8 @@ function PrescribedOceanSimulation(
     SST_init = zeros(space)
     evaluate!(SST_init, SST_timevaryinginput, t_start)
 
+    SST_schedule = CD.Schedules.EveryCalendarDtSchedule(TimeManager.time_to_period("1days"); start_date = date0)
+
     # Create the cache
     cache = (;
         T_sfc = SST_init,
@@ -97,6 +100,7 @@ function PrescribedOceanSimulation(
         phase = TD.Liquid(),
         thermo_params = thermo_params,
         SST_timevaryinginput = SST_timevaryinginput,
+        SST_schedule = SST_schedule,
     )
     return PrescribedOceanSimulation(cache)
 end
@@ -107,10 +111,11 @@ end
     Interfacer.step!(sim::PrescribedOceanSimulation, t)
 
 Update the cached surface temperature field using the prescribed data
-at each timestep.
+at each timestep. This doesn't happen at every timestep,
+but only when the SST data is scheduled to be updated.
 """
 function Interfacer.step!(sim::PrescribedOceanSimulation, t)
-    evaluate!(sim.cache.T_sfc, sim.cache.SST_timevaryinginput, t)
+    sim.cache.SST_schedule(t) && evaluate!(sim.cache.T_sfc, sim.cache.SST_timevaryinginput, t)
 end
 
 function Checkpointer.get_model_cache(sim::PrescribedOceanSimulation)

experiments/ClimaEarth/components/ocean/prescr_seaice.jl

@@ -1,5 +1,6 @@
 import SciMLBase
 import ClimaCore as CC
+import ClimaDiagnostics as CD
 import ClimaTimeSteppers as CTS
 import ClimaUtilities.TimeVaryingInputs: TimeVaryingInput, evaluate!
 import ClimaUtilities.ClimaArtifacts: @clima_artifact
@@ -143,8 +144,20 @@ function PrescribedIceSimulation(
         tspan = Float64.(tspan)
         saveat = Float64.(saveat)
     end
+
+    # Set up a callback to read in SST data daily
+    SIC_schedule = CD.Schedules.EveryCalendarDtSchedule(TimeManager.time_to_period("1days"); start_date = date0)
+    SIC_update_cb = TimeManager.Callback(SIC_schedule, read_sic_data!)
+
     problem = SciMLBase.ODEProblem(ode_function, Y, tspan, (; cache..., params = params))
-    integrator = SciMLBase.init(problem, ode_algo, dt = dt, saveat = saveat, adaptive = false)
+    integrator = SciMLBase.init(
+        problem,
+        ode_algo,
+        dt = dt,
+        saveat = saveat,
+        adaptive = false,
+        callback = SciMLBase.CallbackSet(SIC_update_cb),
+    )
 
     sim = PrescribedIceSimulation(params, space, integrator)
 
@@ -153,6 +166,16 @@ function PrescribedIceSimulation(
     return sim
 end
 
+"""
+    read_sic_data!(integrator)
+
+Read in the sea ice concentration data at the current time step.
+This function is intended to be used within a callback
+"""
+function read_sic_data!(integrator)
+    evaluate!(integrator.p.area_fraction, integrator.p.SIC_timevaryinginput, integrator.t)
+end
+
 # extensions required by Interfacer
 Interfacer.get_field(sim::PrescribedIceSimulation, ::Val{:area_fraction}) = sim.integrator.p.area_fraction
 Interfacer.get_field(sim::PrescribedIceSimulation, ::Val{:roughness_buoyancy}) = sim.integrator.p.params.z0b
@@ -237,9 +260,6 @@ function ice_rhs!(dY, Y, p, t)
     FT = eltype(Y)
     params = p.params
 
-    # Update the cached area fraction with the current SIC
-    evaluate!(p.area_fraction, p.SIC_timevaryinginput, t)
-
     # Overwrite ice fraction with the static land area fraction anywhere we have nonzero land area
     #  max needed to avoid Float32 errors (see issue #271; Heisenbug on HPC)
     @. p.area_fraction = max(min(p.area_fraction, FT(1) - p.land_fraction), FT(0))

experiments/ClimaEarth/setup_run.jl

@@ -527,7 +527,7 @@ function setup_and_run(config_dict::AbstractDict)
     NB: Eventually, we will call all of radiation from the coupler, in addition to the albedo calculation.
     =#
     schedule_checkpoint = EveryCalendarDtSchedule(TimeManager.time_to_period(checkpoint_dt); start_date = date0)
-    checkpoint_cb = TimeManager.TimeManager.Callback(schedule_checkpoint, Checkpointer.checkpoint_sims)
+    checkpoint_cb = TimeManager.Callback(schedule_checkpoint, Checkpointer.checkpoint_sims)
 
     if sim_mode <: AMIPMode
         schedule_albedo = EveryCalendarDtSchedule(TimeManager.time_to_period(dt_rad); start_date = date0)

experiments/ClimaEarth/user_io/arg_parsing.jl

@@ -236,7 +236,7 @@ function parse_component_dts!(config_dict)
     @assert all(key -> !isnothing(config_dict[key]), component_dt_names) || haskey(config_dict, "dt") "all model-specific timesteps (dt_atmos, dt_land, dt_ocean, and dt_seaice) or a generic timestep (dt) must be specified"
 
     for key in component_dt_names
-        if haskey(config_dict, key)
+        if !isnothing(config_dict[key])
             # Check if the component timestep is specified
             component_dt = Float64(Utilities.time_to_seconds(config_dict[key]))
             @assert isapprox(Δt_cpl % component_dt, 0.0) "Coupler dt must be divisible by all component dt's\n dt_cpl = $Δt_cpl\n $key = $component_dt"