Move IC variables from base to stage config

- Move ICSDIR and BASE_IC variables to config.stage_ic
- Update workflow/setup_expt.py to fill in templates
in config.stage_ic based on user input and host

Refs #2475

parm/config/gefs/config.base

@@ -38,8 +38,6 @@ export FIXugwd=${FIXgfs}/ugwd
 export PACKAGEROOT="@PACKAGEROOT@"    # TODO: set via prod_envir in Ops
 export COMROOT="@COMROOT@"    # TODO: set via prod_envir in Ops
 export COMINsyn="@COMINsyn@"
-export ICSDIR="@ICSDIR@"
-export BASE_IC="@BASE_IC@"
 
 # USER specific paths
 export HOMEDIR="@HOMEDIR@"

parm/config/gefs/config.stage_ic

@@ -7,6 +7,9 @@ echo "BEGIN: config.stage_ic"
 # Get task specific resources
 source "${EXPDIR}/config.resources" stage_ic
 
+export ICSDIR="@ICSDIR@"
+export BASE_IC="@BASE_IC@"
+
 export STAGE_IC_YAML_TMPL="${PARMgfs}/stage/stage.yaml.j2"
 
 # Set ICSDIR

parm/config/gfs/config.base

@@ -47,8 +47,6 @@ export PACKAGEROOT="@PACKAGEROOT@"    # TODO: set via prod_envir in Ops
 export COMROOT="@COMROOT@"    # TODO: set via prod_envir in Ops
 export COMINsyn="@COMINsyn@"
 export DMPDIR="@DMPDIR@"
-export ICSDIR="@ICSDIR@"
-export BASE_IC="@BASE_IC@"
 
 # Gempak from external models
 # Default locations are to dummy locations for testing

parm/config/gfs/config.stage_ic

@@ -7,6 +7,9 @@ echo "BEGIN: config.stage_ic"
 # Get task specific resources
 source "${EXPDIR}/config.resources" stage_ic
 
+export ICSDIR="@ICSDIR@"
+export BASE_IC="@BASE_IC@"
+
 export STAGE_IC_YAML_TMPL="${PARMgfs}/stage/stage.yaml.j2"
 
 # Set ICSDIR (if not defined)

workflow/setup_expt.py

@@ -65,7 +65,17 @@ def _update_defaults(dict_in: dict) -> dict:
     # First update config.base
     edit_baseconfig(host, inputs, yaml_dict)
 
-    # loop over other configs and update them
+    # Update stage config
+    stage_dict = {
+        "@ICSDIR@": inputs.icsdir
+    }
+    host_dict = get_template_dict(host.info)
+    stage_dict = dict(stage_dict, **host_dict)
+    stage_input = f'{inputs.configdir}/config.stage_ic'
+    stage_output = f'{inputs.expdir}/{inputs.pslot}/config.stage_ic'
+    edit_config(stage_input, stage_output, stage_dict)
+
+    # Loop over other configs and update them with defaults
     for cfg in yaml_dict.keys():
         if cfg == 'base':
             continue
@@ -104,7 +114,6 @@ def edit_baseconfig(host, inputs, yaml_dict):
         "@OCNRES@": f"{int(100.*inputs.resdetocean):03d}",
         "@EXPDIR@": inputs.expdir,
         "@COMROOT@": inputs.comroot,
-        "@ICSDIR@": inputs.icsdir,
         "@EXP_WARM_START@": is_warm_start,
         "@MODE@": inputs.mode,
         "@gfs_cyc@": inputs.gfs_cyc,