change ldas_setup for ensemble forcing

src/Applications/LDAS_App/GEOSldas_LDAS.rc

@@ -219,3 +219,7 @@ MAPL_ENABLE_BOOTSTRAP: NO
 #----- Write restart or checkpoint by oserver
 #
 WRITE_RESTART_BY_OSERVER: NO
+
+#----- Ensemble forcing
+#  MET_PATH should not contain the ensemble IDs. Default is NO 
+ENSEMBLE_FORCING: NO

src/Applications/LDAS_App/ldas_setup

@@ -628,13 +628,26 @@ class LDASsetup:
         os.symlink(self.blddir, self.blddirLn)
 
         # met forcing dir
-        if 'MET_PATH' in self.rqdExeInp:
-            metpath = self.rqdExeInp['MET_PATH'].rstrip('/')
-            myMetDir = self.inpdir + '/met_forcing'
-            myMetPath = myMetDir + '/' + metpath.split('/')[-1]
-            os.symlink(metpath, myMetPath)
-            # update 'met_path' to use relative path from outdir
-            self.rqdExeInp['MET_PATH'] = os.path.relpath(myMetPath, self.rundir)
+        self.ensemble_forcing = True if self.rqdExeInp.get('ENSEMBLE_FORCING', 'NO').upper() == 'YES' else False
+
+        myMetPath =''
+        for _i in range(1, _nens+1) :
+           if (_nens == 1 or not self.ensemble_forcing):
+              str_ens = ''
+           else:
+              str_ens = '%03d'%(_i)
+           metpath = self.rqdExeInp['MET_PATH'].rstrip('/')+str_ens
+           myMetDir = self.inpdir + '/met_forcing'
+           myMetPath = myMetDir + '/' + metpath.split('/')[-1]
+           os.symlink(metpath, myMetPath)
+           # update 'met_path' to use relative path from outdir
+           if ( not self.ensemble_forcing):
+              break
+        if ( _nens !=1 and self.ensemble_forcing) :
+           # replace last three character with '%s"
+           self.rqdExeInp['MET_PATH'] = os.path.relpath(myMetPath, self.rundir)[:-3]+'%s'
+        else:
+           self.rqdExeInp['MET_PATH'] = os.path.relpath(myMetPath, self.rundir)
 
         # update tile file
         tile= self.rqdExeInp['TILING_FILE']

src/Components/GEOSldas_GridComp/GEOS_LdasGridComp.F90

@@ -179,6 +179,7 @@ subroutine SetServices(gc, rc)
 
     do i=1,NUM_ENSEMBLE
        ens_id(i) = i-1 + FIRST_ENS_ID ! id start form FIRST_ENS_ID
+       ens_id(i) = ens_id(i)+1 ! ensemble forcing is starting with 1 ( 1 off with GEOSldas)
        if(NUM_ENSEMBLE == 1 .or. .not. ensemble_forcing) then
           id_string=''
        else