GEFS Staging in exglobal_stage_ic

scripts/exglobal_stage_ic.sh

@@ -8,6 +8,28 @@ GDATE=$(date -d "${PDY} ${cyc} - ${assim_freq} hours" +%Y%m%d%H)
 gPDY="${GDATE:0:8}"
 gcyc="${GDATE:8:2}"
 
+# Determine NMEM_ENS based on the configuration file (e.g., config.base)
+# Assuming NMEM_ENS is defined in config.base as an integer
+
+config_file="${EXPDIR}/config.base"
+
+if [[ -e "${config_file}" ]]; then
+    NMEM_ENS=$(grep "NMEM_ENS" "${config_file}" | awk -F= '{print $2}' || true)
+else
+    echo "Error: ${config_file} does not exist or is not accessible." >&2
+    exit 1
+fi
+
+member_dirs=()
+
+# Populate the member_dirs array based on the value of NMEM_ENS
+for ((i = 0; i < "${NMEM_ENS}"; i++)); do
+    member_dirs+=("mem$(printf "%03d" "${i}")")
+done
+
+# Now, member_dirs contains elements like "mem000," "mem001," ..., "memNNN" where NNN is the value of NMEM_ENS.
+
+
 # Initialize return code
 err=0
 
@@ -16,9 +38,21 @@ error_message(){
 }
 
 ###############################################################
-# Start staging
+# Start staging gefs and gfs here
+
+# Check for command line argument to select gfs or gefs
+if [[ $# -ne 1 ]]; then
+  echo "Usage: ${0} [gfs | gefs]"
+  exit 1
+fi
+
 
+if [[ "${selection}" = "gfs" ]]; then
 # Stage the FV3 initial conditions to ROTDIR (cold start)
+
+member_dir=""
+NMEM_ENS=0
+
 YMD=${PDY} HH=${cyc} generate_com -r COM_ATMOS_INPUT
 [[ ! -d "${COM_ATMOS_INPUT}" ]] && mkdir -p "${COM_ATMOS_INPUT}"
 source="${BASE_CPLIC}/${CPL_ATMIC}/${PDY}${cyc}/${CDUMP}/${CASE}/INPUT/gfs_ctrl.nc"
@@ -27,19 +61,46 @@ ${NCP} "${source}" "${target}"
 rc=$?
 (( rc != 0 )) && error_message "${source}" "${target}" "${rc}"
 err=$((err + rc))
-for ftype in gfs_data sfc_data; do
-  for tt in $(seq 1 6); do
+  for ftype in gfs_data sfc_data; do
+    for tt in $(seq 1 6); do
     source="${BASE_CPLIC}/${CPL_ATMIC}/${PDY}${cyc}/${CDUMP}/${CASE}/INPUT/${ftype}.tile${tt}.nc"
     target="${COM_ATMOS_INPUT}/${ftype}.tile${tt}.nc"
     ${NCP} "${source}" "${target}"
     rc=$?
     (( rc != 0 )) && error_message "${source}" "${target}" "${rc}"
     err=$((err + rc))
+    done
   done
-done
+elif [[ "${selection}" = "gefs" ]]; then
+
+member_dir=""
+YMD=${PDY} HH=${cyc} generate_com -r COM_ATMOS_INPUT
+[[ ! -d "${COM_ATMOS_INPUT}" ]] && mkdir -p "${COM_ATMOS_INPUT}"
+  for member_dir in $(seq -w 0 $((NMEM_ENS - 1))); do
+    source="${BASE_CPLIC}/${CPL_ATMIC}/${YMD}${HH}/${member_dir}/atmos/gfs_ctrl.nc"
+    target="${COM_ATMOS_INPUT}/gfs_ctrl.nc"
+    ${NCP} "${source}" "${target}"
+    rc=$?
+    (( rc != 0 )) && error_message "${source}" "${target}" "${rc}"
+    err=$((err + rc))
+    for ftype in gfs_data sfc_data; do
+      for tt in $(seq 1 6); do
+        source="${BASE_CPLIC}/${CPL_ATMIC}/${YMD}${HH}/${member_dir}/atmos/${ftype}.tile${tt}.nc"
+        target="${COM_ATMOS_INPUT}/${ftype}.tile${tt}.nc"
+        rc=$?
+        (( rc != 0 )) && error_message "${source}" "${target}" "${rc}"
+        err=$((err + rc))
+      done
+    done
+  done
+fi
 
 # Stage ocean initial conditions to ROTDIR (warm start)
-if [[ "${DO_OCN:-}" = "YES" ]]; then
+if [[ "${DO_OCN:-}" = "YES" && "${selection}" = "gfs" ]]; then
+
+member_dir=""
+NMEM_ENS=0
+
   YMD=${gPDY} HH=${gcyc} generate_com -r COM_OCEAN_RESTART
   [[ ! -d "${COM_OCEAN_RESTART}" ]] && mkdir -p "${COM_OCEAN_RESTART}"
   source="${BASE_CPLIC}/${CPL_OCNIC}/${PDY}${cyc}/ocn/${OCNRES}/MOM.res.nc"
@@ -52,27 +113,44 @@ if [[ "${DO_OCN:-}" = "YES" ]]; then
     "500" | "100")  # Only 5 degree or 1 degree ocean does not have MOM.res_[1-4].nc files
     ;;
     "025")  # Only 1/4 degree ocean has MOM.res_[1-4].nc files
-      for nn in $(seq 1 4); do
-        source="${BASE_CPLIC}/${CPL_OCNIC}/${PDY}${cyc}/ocn/${OCNRES}/MOM.res_${nn}.nc"
-        if [[ -f "${source}" ]]; then
+    for nn in $(seq 1 4); do
+    source="${BASE_CPLIC}/${CPL_OCNIC}/${PDY}${cyc}/ocn/${OCNRES}/MOM.res_${nn}.nc"
+      if [[ -f "${source}" ]]; then
           target="${COM_OCEAN_RESTART}/${PDY}.${cyc}0000.MOM.res_${nn}.nc"
           ${NCP} "${source}" "${target}"
           rc=$?
           (( rc != 0 )) && error_message "${source}" "${target}" "${rc}"
           err=$((err + rc))
-        fi
-      done
+      fi
+    done
     ;;
     *)
+    echo "FATAL ERROR: Unsupported ocean resolution ${OCNRES}"
+    rc=1
+    err=$((err + rc))
+    ;;
+  esac
+elif [[ "${DO_OCN:-}" = "YES" && "${selection}" = "gefs" ]]; then
+  member_dir=""
+  YMD=${gPDY} HH=${gcyc} generate_com -r COM_OCEAN_RESTART
+  [[ ! -d "${COM_OCEAN_RESTART}" ]] && mkdir -p "${COM_OCEAN_RESTART}"
+    for member_dir in $(seq -w 0 $((NMEM_ENS - 1))); do
+    source="${BASE_CPLIC}/${CPL_ATMIC}/${YMD}${HH}/${member_dir}/ocean/${PDY}.${cyc}0000.MOM.res.nc"
+    target="${COM_OCEAN_RESTART}/${PDY}.${cyc}0000.MOM.res.nc"
+    ${NCP} "${source}" "${target}"
+    rc=$?
+    (( rc != 0 )) && error_message "${source}" "${target}" "${rc}"
+    err=$((err + rc))
+    done
       echo "FATAL ERROR: Unsupported ocean resolution ${OCNRES}"
       rc=1
       err=$((err + rc))
-    ;;
-  esac
 fi
 
 # Stage ice initial conditions to ROTDIR (warm start)
-if [[ "${DO_ICE:-}" = "YES" ]]; then
+if [[ "${DO_ICE:-}" = "YES" && "${selection}" = "gfs" ]]; then
+  member_dir=""
+  NMEM_ENS=0
   YMD=${gPDY} HH=${gcyc} generate_com -r COM_ICE_RESTART
   [[ ! -d "${COM_ICE_RESTART}" ]] && mkdir -p "${COM_ICE_RESTART}"
   ICERESdec=$(echo "${ICERES}" | awk '{printf "%0.2f", $1/100}')
@@ -82,10 +160,28 @@ if [[ "${DO_ICE:-}" = "YES" ]]; then
   rc=$?
   (( rc != 0 )) && error_message "${source}" "${target}" "${rc}"
   err=$((err + rc))
+
+elif [[ "${DO_ICE:-}" = "YES" && "${selection}" = "gefs" ]]; then
+  member_dir=""
+  YMD=${gPDY} HH=${gcyc} generate_com -r COM_ICE_RESTART
+  [[ ! -d "${COM_ICE_RESTART}" ]] && mkdir -p "${COM_ICE_RESTART}"
+    for member_dir in $(seq -w 0 $((NMEM_ENS - 1))); do
+    source="${BASE_CPLIC}/${CPL_ATMIC}/${YMD}${HH}/${member_dir}/ice/${PDY}.${cyc}0000.cice_model.res.nc"
+    target="${COM_OCEAN_RESTART}/${PDY}.${cyc}0000.cice_model.res.nc"
+    ${NCP} "${source}" "${target}"
+    rc=$?
+    (( rc != 0 )) && error_message "${source}" "${target}" "${rc}"
+    err=$((err + rc))
+    done
+      echo "FATAL ERROR: Unsupported ocean resolution ${OCNRES}"
+      rc=1
+      err=$((err + rc))
 fi
 
 # Stage the WW3 initial conditions to ROTDIR (warm start; TODO: these should be placed in $RUN.$gPDY/$gcyc)
-if [[ "${DO_WAVE:-}" = "YES" ]]; then
+if [[ "${DO_WAVE:-}" = "YES" && "${selection}" = "gfs" ]]; then
+  member_dir=""
+  NMEM_ENS=0
   YMD=${PDY} HH=${cyc} generate_com -r COM_WAVE_RESTART
   [[ ! -d "${COM_WAVE_RESTART}" ]] && mkdir -p "${COM_WAVE_RESTART}"
   for grdID in ${waveGRD}; do  # TODO: check if this is a bash array; if so adjust
@@ -96,6 +192,21 @@ if [[ "${DO_WAVE:-}" = "YES" ]]; then
     (( rc != 0 )) && error_message "${source}" "${target}" "${rc}"
     err=$((err + rc))
   done
+elif [[ "${DO_WAVE:-}" = "YES" && "${selection}" = "gefs" ]]; then
+  member_dir=""
+  YMD=${gPDY} HH=${gcyc} generate_com -r COM_WAVE_RESTART
+  [[ ! -d "${COM_WAVE_RESTART}" ]] && mkdir -p "${COM_WAVE_RESTART}"
+    for member_dir in $(seq -w 0 $((NMEM_ENS - 1))); do
+    source="${BASE_CPLIC}/${CPL_ATMIC}/${YMD}${HH}/${member_dir}/ice/${PDY}.${cyc}0000.restart.gwes_30m"
+    target="${COM_OCEAN_RESTART}/${PDY}.${cyc}0000.restart.gwes_30m"
+    ${NCP} "${source}" "${target}"
+    rc=$?
+    (( rc != 0 )) && error_message "${source}" "${target}" "${rc}"
+    err=$((err + rc))
+    done
+      echo "FATAL ERROR: Unsupported ocean resolution ${OCNRES}"
+      rc=1
+      err=$((err + rc))
 fi
 
 ###############################################################