added regrid operator for workflow

rook/operator.py

@@ -12,6 +12,7 @@
 )
 from rook.utils.subset_utils import run_subset
 from rook.utils.concat_utils import run_concat
+from rook.utils.regrid_utils import run_regrid
 from roocs_utils.utils.file_utils import is_file_list, FileMapper
 
 
@@ -89,6 +90,13 @@ def _get_runner(self):
         return run_weighted_average
 
 
+class Regrid(Operator):
+    prefix = "regrid"
+
+    def _get_runner(self):
+        return run_regrid
+
+
 class Concat(Operator):
     prefix = "concat"
 

rook/workflow.py

@@ -11,6 +11,7 @@
     AverageByDimension,
     WeightedAverage,
     Subset,
+    Regrid,
     Concat,
 )
 from .provenance import Provenance
@@ -86,6 +87,7 @@ def __init__(self, output_dir):
         self.average_time_op = AverageByTime(output_dir)
         self.average_dim_op = AverageByDimension(output_dir)
         self.weighted_average_op = WeightedAverage(output_dir)
+        self.regrid_op = Regrid(output_dir)
         self.prov = Provenance(output_dir)
 
     def validate(self, wfdoc):
@@ -154,6 +156,10 @@ def _run_step(self, step_id, step, inputs=None):
             collection = step["in"]["collection"]
             result = self.weighted_average_op.call(step["in"])
             self.prov.add_operator(step_id, step["in"], collection, result)
+        elif "regrid" == step["run"]:
+            collection = step["in"]["collection"]
+            result = self.regrid_op.call(step["in"])
+            self.prov.add_operator(step_id, step["in"], collection, result)
         elif "concat" == step["run"]:
             collection = step["in"]["collection"]
             result = self.concat_op.call(step["in"])