Merge pull request #1458 from CliMA/cc/wx_quest_ic

Add subseasonal mode for WeatherQuest.

Author: costachris

config/atmos_configs/climaatmos_wx_diagedmf.yml

@@ -0,0 +1,48 @@
+dt: 120secs
+t_end: 12hours
+start_date: "20250831"
+initial_condition: "WeatherModel"
+discrete_hydrostatic_balance: false
+topo_smoothing: true
+topography: "Earth"
+rayleigh_sponge: true
+viscous_sponge: true
+hyperdiff: ClimaHyperdiffusion
+divergence_damping_factor: 50.0
+scalar_hyperdiffusion_coefficient: 0.929738
+vorticity_hyperdiffusion_coefficient: 0.1857
+insolation: "timevarying"
+rad: allskywithclear
+dt_cloud_fraction: "1hours"
+dt_rad: "1hours"
+co2_model: fixed
+# prescribe_ozone: true
+# prescribed_aerosols: ["CB1", "CB2", "DST01", "OC1", "OC2", "SO4", "SSLT01"]
+turbconv: diagnostic_edmfx
+implicit_diffusion: true
+approximate_linear_solve_iters: 2
+prognostic_tke: true
+edmfx_upwinding: first_order
+h_elem: 15
+z_max: 48000.0
+z_elem: 63
+dz_bottom: 30.0
+edmfx_entr_model: "Generalized"
+edmfx_detr_model: "Generalized"
+edmfx_nh_pressure: true
+edmfx_sgs_mass_flux: true
+edmfx_sgs_diffusive_flux: true
+
+### 0M Microphysics ###
+precip_model: 0M
+moist: equil
+toml: [toml/diagnostic_edmfx_era5_ic.toml]
+
+### 1M Microphysics ###
+# precip_model: "1M"
+# moist: "nonequil"
+# toml: [toml/diagnostic_edmfx_1M.toml]
+
+cloud_model: "quadrature_sgs"
+use_itime: true
+deep_atmosphere: false

config/atmos_configs/climaatmos_wx_progedmf.yml

@@ -0,0 +1,54 @@
+dt: 120secs
+t_end: 12hours
+start_date: "20250831"
+initial_condition: "WeatherModel"
+discrete_hydrostatic_balance: false
+topo_smoothing: true
+topography: "Earth"
+rayleigh_sponge: true
+viscous_sponge: true
+hyperdiff: ClimaHyperdiffusion
+divergence_damping_factor: 50.0
+scalar_hyperdiffusion_coefficient: 0.929738
+vorticity_hyperdiffusion_coefficient: 0.1857
+insolation: "timevarying"
+rad: allskywithclear
+dt_cloud_fraction: "1hours"
+dt_rad: "1hours"
+co2_model: fixed
+# prescribe_ozone: true
+# prescribed_aerosols: ["CB1", "CB2", "DST01", "OC1", "OC2", "SO4", "SSLT01"]
+turbconv: "prognostic_edmfx"
+implicit_diffusion: true
+implicit_sgs_advection: true
+implicit_sgs_entr_detr: true
+implicit_sgs_mass_flux: true
+implicit_sgs_nh_pressure: true
+edmfx_filter: true
+approximate_linear_solve_iters: 2
+prognostic_tke: true
+edmfx_upwinding: first_order
+h_elem: 15
+z_max: 48000.0
+z_elem: 63
+dz_bottom: 30.0
+edmfx_entr_model: "Generalized"
+edmfx_detr_model: "Generalized"
+edmfx_nh_pressure: true
+edmfx_sgs_mass_flux: true
+edmfx_sgs_diffusive_flux: true
+
+### 0M Microphysics ###
+precip_model: 0M
+moist: equil
+# toml: [toml/longrun_aquaplanet_progedmf.toml]
+
+
+### 1M Microphysics ###
+# precip_model: "1M"
+# moist: "nonequil"
+# toml: [toml/diagnostic_edmfx_1M.toml]
+
+cloud_model: "quadrature_sgs"
+use_itime: true
+deep_atmosphere: false

config/subseasonal_configs/wxquest_diagedmf.yml

@@ -0,0 +1,36 @@
+FLOAT_TYPE: "Float32"
+albedo_model: "CouplerAlbedo"
+atmos_config_file: "config/atmos_configs/climaatmos_wx_diagedmf.yml"
+coupler_toml: ["toml/amip.toml"]
+mode_name: "subseasonal"
+era5_initial_condition_dir: "/net/sampo/data1/wxquest_data/initial_conditions"
+initial_condition: "WeatherModel"
+checkpoint_dt: "7days"
+co2_model: "fixed"
+dt: "30secs"
+dt_cpl: "30secs"
+energy_check: false
+h_elem: 16
+
+### integrated land ###
+# land_model: "integrated"
+
+### bucket land ###
+land_model: "bucket"
+bucket_albedo_type: "map_temporal"
+bucket_initial_condition: "/net/sampo/data1/wxquest_data/initial_conditions/era5_bucket_processed_20250907_0000.nc"
+
+land_spun_up_ic: false
+land_temperature_anomaly: "nothing"
+use_land_diagnostics: true
+radiation_reset_rng_seed: true
+start_date: "20250907"
+surface_setup: "PrescribedSurface"
+topo_smoothing: true
+topography: "Earth"
+t_end: "7days"
+netcdf_output_at_levels: true
+extra_atmos_diagnostics:
+  - short_name: [tas, mslp, pr, ua, va, rhoa, pfull, hur, hus, clw, cli, clivi, clwvi, cl, arup, tke]
+    period: 1hours
+    reduction_time: average

config/subseasonal_configs/wxquest_progedmf.yml

@@ -0,0 +1,36 @@
+FLOAT_TYPE: "Float32"
+albedo_model: "CouplerAlbedo"
+atmos_config_file: "config/atmos_configs/climaatmos_wx_progedmf.yml"
+coupler_toml: ["toml/amip_progedmf.toml"]
+mode_name: "subseasonal"
+era5_initial_condition_dir: "/net/sampo/data1/wxquest_data/initial_conditions"
+initial_condition: "WeatherModel"
+checkpoint_dt: "7days"
+co2_model: "fixed"
+dt: "10secs"
+dt_cpl: "10secs"
+energy_check: false
+h_elem: 32
+
+### integrated land ###
+# land_model: "integrated"
+
+### bucket land ###
+land_model: "bucket"
+bucket_albedo_type: "map_temporal"
+bucket_initial_condition: "/net/sampo/data1/wxquest_data/initial_conditions/era5_bucket_processed_20250907_0000.nc"
+
+land_spun_up_ic: false
+land_temperature_anomaly: "nothing"
+use_land_diagnostics: true
+radiation_reset_rng_seed: true
+start_date: "20250907"
+surface_setup: "PrescribedSurface"
+topo_smoothing: true
+topography: "Earth"
+t_end: "7days"
+netcdf_output_at_levels: true
+extra_atmos_diagnostics:
+  - short_name: [tas, mslp, pr, ua, va, rhoa, pfull, hur, hus, clw, cli, clivi, clwvi, cl, arup, tke]
+    period: 1hours
+    reduction_time: average

docs/src/interfacer.md

@@ -296,6 +296,7 @@ end
     ClimaCoupler.Interfacer.update_field!
     ClimaCoupler.Interfacer.AbstractSlabplanetSimulationMode
     ClimaCoupler.Interfacer.AMIPMode
+    ClimaCoupler.Interfacer.SubseasonalMode
     ClimaCoupler.Interfacer.SlabplanetMode
     ClimaCoupler.Interfacer.SlabplanetAquaMode
     ClimaCoupler.Interfacer.SlabplanetTerraMode

experiments/ClimaEarth/Project.toml

@@ -50,4 +50,4 @@ Insolation = "0.10.2"
 Interpolations = "0.14, 0.15"
 Oceananigans = "0.97.1, 0.98"
 StaticArrays = "1"
-YAML = "0.4"
+YAML = "0.4"

experiments/ClimaEarth/README.md

@@ -93,6 +93,48 @@ only surface model is the ocean, which is evaluated over the entire surface. The
 This configuration is similar to the general "Slabplanet" configuration, except that the
 only surface model is the land, which is evaluated over the entire surface. There are no ocean or sea ice models.
 
+## Subseasonal mode
+
+Generates 3-4 week forecasts from ERA5-derived initial conditions. Similar to `amip` setup. Given a directory containing initial condition files, the filenames associated with current date are inferred from `start_date`. Initial condition files are generated by `https://github.com/CliMA/WeatherQuest`.
+
+- `era5_initial_condition_dir`: directory that contains files named like:
+  - `sst_processed_YYYYMMDD_0000.nc`
+  - `sic_processed_YYYYMMDD_0000.nc`
+  - `era5_land_processed_YYYYMMDD_0000.nc`
+  - `era5_bucket_processed_YYYYMMDD_0000.nc`
+
+- **How filename inference works**: given `start_date` formatted as `YYYYMMDD`, the model constructs paths
+  - `joinpath(era5_initial_condition_dir, "sst_processed_YYYYMMDD_0000.nc")` for the initial conditions. 
+
+- **Contents and expected variables**:
+  - `sst_processed_YYYYMMDD_0000.nc`:
+    - Variable: `SST`
+    - Units: degrees C in file; converted internally to K
+  - `sic_processed_YYYYMMDD_0000.nc`:
+    - Variable: `SEAICE`
+    - Units: percent (0–100) in file; converted internally to fraction (0–1)
+  - `era5_land_processed_YYYYMMDD_0000.nc` (for integrated land model):
+    - Variables used directly: `skt` (skin temperature, K), `tsn` (snow surface temperature, K)
+    - Required fields for land initialization:
+      - `swe` (m): snow water equivalent
+      - `swvl` (m^3/m^3): volumetric fraction of liquid water, dims `(z, lat, lon)`
+      - `si` (m^3/m^3): volumetric fraction of ice, dims `(z, lat, lon)`
+      - `sie` (J/m^3): soil volumetric internal energy, dims `(z, lat, lon)`
+      - `stl` (K): soil temperature, dims `(z, lat, lon)`
+      - `tsn` (K): temperature of snow layer
+      - `skt` (K): skin temperature
+    - era5 land level midpoints: 0.035, 0.175, 0.64, 1.945 (m)
+
+  - `era5_bucket_processed_YYYYMMDD_0000.nc` (for bucket land model):
+    - Required fields for initialization:
+      - `W` (m): subsurface water content, dims `(lat, lon)`
+      - `Ws` (m): surface water content, dims `(lat, lon)`
+      - `S` (m): snow water equivalent, dims `(lat, lon)`
+      - `T` (K): soil temperature profile, dims `(z, lat, lon)`
+      - `tsn` (K): temperature of snow layer, dims `(lat, lon)`
+      - `skt` (K): skin temperature, dims `(lat, lon)`
+    - era5 land level midpoints: 0.035, 0.175, 0.64, 1.945 (m)
+
 ## Configuration files
 We use configuration files to specify all simulation parameters and options. The configuration files are organized hierarchically:
 

experiments/ClimaEarth/cli_options.jl

@@ -17,7 +17,7 @@ function argparse_settings()
         arg_type = Bool
         default = true
         "--mode_name"
-        help = "Mode of coupled simulation. [`cmip`, `amip` (default), `slabplanet`, `slabplanet_aqua`, `slabplanet_terra`]"
+        help = "Mode of coupled simulation. [`cmip`, `amip` (default), `subseasonal`, `slabplanet`, `slabplanet_aqua`, `slabplanet_terra`]"
         arg_type = String
         default = "amip"
         "--coupler_toml"
@@ -151,7 +151,7 @@ function argparse_settings()
         arg_type = String
         default = "bucket"
         "--land_temperature_anomaly"
-        help = "Type of temperature anomaly for land model. [`amip`, `aquaplanet` (default)]"
+        help = "Type of temperature anomaly for land model. [`amip`, `aquaplanet` (default), `nothing`]"
         arg_type = String
         default = "aquaplanet"
         "--use_land_diagnostics"
@@ -171,6 +171,10 @@ function argparse_settings()
         help = "A file path for a NetCDF file (read documentation about requirements)"
         arg_type = String
         default = ""
+        "--era5_initial_condition_dir"
+        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
     end
     return s
 end

experiments/ClimaEarth/components/land/climaland_bucket.jl

@@ -143,25 +143,27 @@ function BucketSimulation(
     # Initial conditions with no moisture
     Y, p, coords = CL.initialize(model)
 
-    # Get temperature anomaly function
-    T_functions = Dict("aquaplanet" => temp_anomaly_aquaplanet, "amip" => temp_anomaly_amip)
-    haskey(T_functions, land_temperature_anomaly) ||
-        error("land temp anomaly function $land_temperature_anomaly not supported")
-    temp_anomaly = T_functions[land_temperature_anomaly]
-
-    # Set temperature IC including anomaly, based on atmospheric setup
-    # Bucket surface temperature is in `p.bucket.T_sfc` (ClimaLand.jl)
-    lapse_rate = FT(6.5e-3)
-    T_sfc_0 = FT(271)
-    @. Y.bucket.T = T_sfc_0 + temp_anomaly(coords.subsurface)
-    # `surface_elevation` is a ClimaCore.Fields.Field(`half` level)
-    orog_adjusted_T_data =
-        CC.Fields.field_values(Y.bucket.T) .-
-        lapse_rate .* CC.Fields.field_values(surface_elevation)
-    orog_adjusted_T = CC.Fields.Field(orog_adjusted_T_data, domain.space.subsurface)
-    # Adjust T based on surface elevation (p.bucket.T_sfc is then set using the
-    # set_initial_cache! function)
-    Y.bucket.T .= orog_adjusted_T
+    if land_temperature_anomaly != "nothing"
+        T_functions =
+            Dict("aquaplanet" => temp_anomaly_aquaplanet, "amip" => temp_anomaly_amip)
+        haskey(T_functions, land_temperature_anomaly) ||
+            error("land temp anomaly function $land_temperature_anomaly not supported")
+        temp_anomaly = T_functions[land_temperature_anomaly]
+
+        # Set temperature IC including anomaly, based on atmospheric setup
+        # Bucket surface temperature is in `p.bucket.T_sfc` (ClimaLand.jl)
+        lapse_rate = FT(6.5e-3)
+        T_sfc_0 = FT(271)
+        @. Y.bucket.T = T_sfc_0 + temp_anomaly(coords.subsurface)
+        # `surface_elevation` is a ClimaCore.Fields.Field(`half` level)
+        orog_adjusted_T_data =
+            CC.Fields.field_values(Y.bucket.T) .-
+            lapse_rate .* CC.Fields.field_values(surface_elevation)
+        orog_adjusted_T = CC.Fields.Field(orog_adjusted_T_data, domain.space.subsurface)
+        # Adjust T based on surface elevation (p.bucket.T_sfc is then set using the
+        # set_initial_cache! function)
+        Y.bucket.T .= orog_adjusted_T
+    end
 
     Y.bucket.W .= 0.15
     Y.bucket.Ws .= 0.0
@@ -175,6 +177,7 @@ function BucketSimulation(
     # - `S`, for snow water equivalent (2D).
 
     if !isempty(bucket_initial_condition)
+        @info "ClimaLand Bucket using land IC file" bucket_initial_condition
         ds = NCDataset(bucket_initial_condition)
         has_all_variables = all(key -> haskey(ds, key), ["W", "Ws", "T", "S"])
         @assert has_all_variables "The land iniital condition file is expected to contain the variables W, Ws, T, and S (read documentation about requirements)."
@@ -185,6 +188,7 @@ function BucketSimulation(
         regridder_type = :InterpolationsRegridder
         extrapolation_bc =
             (Interpolations.Periodic(), Interpolations.Flat(), Interpolations.Flat())
+
         Y.bucket.W .= SpaceVaryingInput(
             bucket_initial_condition,
             "W",