Add tracker and genesis jobs to setup

When jobs turned on: - Add jobs to rocoto mesh. - Add configs to list of configs to source. - Add jobs to gfsarch dependency list. Refs NOAA-EMC#1988
KateFriedman-NOAA · Nov 7, 2023 · b26dad4 · b26dad4
1 parent 7211e69
commit b26dad4
Show file tree

Hide file tree

Showing 4 changed files with 64 additions and 0 deletions.
diff --git a/workflow/applications/applications.py b/workflow/applications/applications.py
@@ -58,6 +58,9 @@ def __init__(self, conf: Configuration) -> None:
         self.do_verfozn = _base.get('DO_VERFOZN', True)
         self.do_verfrad = _base.get('DO_VERFRAD', True)
         self.do_vminmon = _base.get('DO_VMINMON', True)
+        self.do_tracker = _base.get('DO_TRACKER', True)
+        self.do_genesis = _base.get('DO_GENESIS', True)
+        self.do_genesis_fsu = _base.get('DO_GENESIS_FSU', False)
         self.do_metp = _base.get('DO_METP', False)
 
         self.do_hpssarch = _base.get('HPSSARCH', False)

diff --git a/workflow/applications/gfs_cycled.py b/workflow/applications/gfs_cycled.py
@@ -65,6 +65,15 @@ def _get_app_configs(self):
         if self.do_vminmon:
             configs += ['vminmon']
 
+        if self.do_tracker:
+            configs += ['tracker']
+
+        if self.do_genesis:
+            configs += ['genesis']
+
+        if self.do_genesis_fsu:
+            configs += ['genesis_fsu']
+
         if self.do_metp:
             configs += ['metp']
 
@@ -192,6 +201,15 @@ def get_task_names(self):
         if self.do_vminmon:
             gfs_tasks += ['vminmon']
 
+        if self.do_tracker:
+            gfs_tasks += ['tracker']
+
+        if self.do_genesis:
+            gfs_tasks += ['genesis']
+
+        if self.do_genesis_fsu:
+            gfs_tasks += ['genesis_fsu']
+
         if self.do_metp:
             gfs_tasks += ['metp']
 

diff --git a/workflow/rocoto/gfs_tasks.py b/workflow/rocoto/gfs_tasks.py
@@ -897,6 +897,39 @@ def vminmon(self):
 
         return task
 
+    def tracker(self):
+        deps = []
+        dep_dict = {'type': 'metatask', 'name': f'{self.cdump}post'}
+        deps.append(rocoto.add_dependency(dep_dict))
+        dependencies = rocoto.create_dependency(dep=deps)
+
+        resources = self.get_resource('tracker')
+        task = create_wf_task('tracker', resources, cdump=self.cdump, envar=self.envars, dependency=dependencies)
+
+        return task
+
+    def genesis(self):
+        deps = []
+        dep_dict = {'type': 'metatask', 'name': f'{self.cdump}post'}
+        deps.append(rocoto.add_dependency(dep_dict))
+        dependencies = rocoto.create_dependency(dep=deps)
+
+        resources = self.get_resource('genesis')
+        task = create_wf_task('genesis', resources, cdump=self.cdump, envar=self.envars, dependency=dependencies)
+
+        return task
+
+    def genesis_fsu(self):
+        deps = []
+        dep_dict = {'type': 'metatask', 'name': f'{self.cdump}post'}
+        deps.append(rocoto.add_dependency(dep_dict))
+        dependencies = rocoto.create_dependency(dep=deps)
+
+        resources = self.get_resource('genesis_fsu')
+        task = create_wf_task('genesis_fsu', resources, cdump=self.cdump, envar=self.envars, dependency=dependencies)
+
+        return task
+
     def vrfy(self):
         deps = []
         dep_dict = {'type': 'metatask', 'name': f'{self.cdump}post'}
@@ -973,6 +1006,15 @@ def arch(self):
                     dep_dict = {'type': 'cycleexist', 'condition': 'not', 'offset': '-06:00:00'}
                     dependencies.append(rocoto.add_dependency(dep_dict))
                     dependencies = rocoto.create_dependency(dep_condition='or', dep=dependencies)
+        if self.cdump in ['gfs'] and self.app_config.do_tracker:
+            dep_dict = {'type': 'task', 'name': f'{self.cdump}tracker'}
+            deps.append(rocoto.add_dependency(dep_dict))
+        if self.cdump in ['gfs'] and self.app_config.do_genesis:
+            dep_dict = {'type': 'task', 'name': f'{self.cdump}genesis'}
+            deps.append(rocoto.add_dependency(dep_dict))
+        if self.cdump in ['gfs'] and self.app_config.do_genesis_fsu:
+            dep_dict = {'type': 'task', 'name': f'{self.cdump}genesis_fsu'}
+            deps.append(rocoto.add_dependency(dep_dict))
         if self.app_config.do_vrfy:
             dep_dict = {'type': 'task', 'name': f'{self.cdump}vrfy'}
             deps.append(rocoto.add_dependency(dep_dict))

diff --git a/workflow/rocoto/tasks.py b/workflow/rocoto/tasks.py
@@ -21,6 +21,7 @@ class Tasks:
                    'preplandobs', 'landanl',
                    'fcst', 'post', 'ocnpost',
                    'verfozn', 'verfrad', 'vminmon', 'vrfy', 'metp',
+                   'tracker genesis genesis_fsu'
                    'postsnd', 'awips', 'gempak',
                    'waveawipsbulls', 'waveawipsgridded', 'wavegempak', 'waveinit',
                    'wavepostbndpnt', 'wavepostbndpntbll', 'wavepostpnt', 'wavepostsbs', 'waveprep']