Pleiades pipeline and outer command

VHR/vhrbase.py

@@ -228,3 +228,48 @@ class SPOT67RasterPipeline:
             self.types = self.types[:pipe_length]
             self.out_p = self.out_p[:pipe_length]
         return out
+
+class PleiadesOrthoPipeline(SPOT67RasterPipeline):
+    # BEGIN PLEIADES (ORT) VHR PROTOTYPE
+    MS_FOLDER_PATTERN = 'IMG_PHR*_MS_*'
+    PAN_FOLDER_PATTERN = 'IMG_PHR*_P_*'
+    TILE_PATTERN = '*_R*C*.TIF'
+    NDT = 0.0
+    REF_TYPE = otb.ImagePixelType_uint16
+    MS_LABEL = 'MS'
+    OPT_IN = ''
+
+class PleiadesOrthoAutoMosaicPipeline:
+    def __init__(self, mosaic_root, pattern, out_fld):
+        lst = sorted(glob.glob(os.path.join(mosaic_root, pattern)))
+        lst = [x for x in lst if os.path.isdir(x)]
+        self.scene_list = [PleiadesOrthoPipeline(x) for x in lst]
+        self.processed_scenes = []
+        self.out_fld = out_fld
+
+    def process_scenes(self, roi=None, compress=False):
+        for s in self.scene_list:
+            s.to_toa()
+            if roi is not None and os.path.exists(roi):
+                s.clip(roi)
+            s.write_outputs(self.out_fld, update_pipe=True, compress=compress)
+            s.pansharp()
+            self.processed_scenes.extend(s.write_outputs(self.out_fld, compress=compress))
+        return
+    
+    def generate_mosaic(self, vrt=True):
+        if len(self.processed_scenes) > 0:
+            lst = self.processed_scenes
+            fn = lst[0].split('_')
+            fn[-3] = 'MOSAIC'
+            fn = '_'.join(fn)
+            if vrt is True:
+                vrtopt = gdal.BuildVRTOptions()
+                fn = fn.replace('.TIF', '.vrt')
+                gdal.BuildVRT(fn, lst, options=vrtopt)
+            else:
+                print('Not implemented yet.')
+            return fn
+        else:
+            print('Preprocess scenes first.')
+            return None
\ No newline at end of file

Workflows/operations.py

@@ -32,6 +32,24 @@ def preprocess_spot67(in_fld, out_fld, dem_fld, geoid_file, skip_ps, compress,
         sp.rigid_align(align_to, this_band=align_using_band-1, ref_band=align_to_band-1)
     return sp.write_outputs(out_fld, compress=compress)
 
+def preprocess_pleiades(in_fld, out_fld, skip_ps, compress, clip, 
+                        align_to=None, align_to_band=3, align_using_band=1):
+    sp = VHR.vhrbase.PleiadesOrthoPipeline(in_fld)
+    sp.to_toa()
+    if clip is not None and os.path.exists(clip):
+        sp.clip(clip)
+    if not skip_ps:
+        sp.write_outputs(out_fld, update_pipe=True, compress=compress)
+        sp.pansharp()
+    if align_to is not None and os.path.exists(align_to):
+        sp.rigid_align(align_to, this_band=align_using_band-1, ref_band=align_to_band-1)
+    return sp.write_outputs(out_fld, compress=compress)
+
+def preprocess_pleiades_mosaic(in_fld, out_fld, pattern):
+    sp = VHR.vhrbase.PleiadesOrthoAutoMosaicPipeline(in_fld, pattern, out_fld)
+    sp.process_scenes()
+    return sp.generate_mosaic()
+
 def preprocess_s2(in_fld, out_fld, output_dates_file=None, roi=None,
                   align_to=None, align_to_band=3, align_using_band=3, provider='theia'):
     S2Processor = None

moringa.py

@@ -48,6 +48,23 @@ def main(args):
     vhrprep.add_argument("--skip_ps", help="Skip pansharpening step.", action='store_true')
     vhrprep.add_argument("--compress", help="Use lossless compression on outputs.", action='store_true')
 
+    plprep = subpar.add_parser("preprocess_pleiades", help="Perform baseline pre-processing of a Pleaides ortho scene.",
+                                formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+    plprep.add_argument("fld", type=str, help="Path to the root folder containing PAN and MS subfolders.")
+    plprep.add_argument("out_fld", type=str, help="Path to the output folder for preprocessed images.")
+    plprep.add_argument("--clip", type=str, default=None, help="Path to a vector file for clipping.")
+    plprep.add_argument("--align_to", type=str, default=None, help="Path to a reference image to which the image must be aligned (rigid).")
+    plprep.add_argument("--align_to_band", type=int, default=3, help="Band of reference image used for alignment.")
+    plprep.add_argument("--align_using_band", type=int, default=1, help="Band of current image used for alignment.")
+    plprep.add_argument("--skip_ps", help="Skip pansharpening step.", action='store_true')
+    plprep.add_argument("--compress", help="Use lossless compression on outputs.", action='store_true')
+
+    plmprep = subpar.add_parser("preprocess_pleiades_mosaic", help="Perform baseline pre-processing and auto-mosaic of Pleaides ortho scenes.",
+                                formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+    plmprep.add_argument("fld", type=str, help="Path to the root folder containing Pleaides acquisitions.")
+    plmprep.add_argument("pattern", type=str, help="File pattern to collect scene folders within in_fld.")
+    plmprep.add_argument("out_fld", type=str, help="Path to the output folder for preprocessed images.")
+
     s1prepr = subpar.add_parser("preprocess_s1", help="Performs Moringa preset time series preprocessing for Sentinel-1.",
                                 formatter_class=argparse.ArgumentDefaultsHelpFormatter)
     s1prepr.add_argument("in_folder", type=str, help="Path to the folder containing (de-zipped) S1 images.")
@@ -125,6 +142,13 @@ def main(args):
     if arg.cmd == "preprocess_spot67":
         preprocess_spot67(arg.fld, arg.out_fld, arg.dem_fld, arg.geoid, arg.skip_ps, arg.compress,
                           arg.clip, arg.align_to, arg.align_to_band, arg.align_using_band)
+    
+    if arg.cmd == "preprocess_pleiades":
+        preprocess_pleiades(arg.fld, arg.out_fld, arg.skip_ps, arg.compress, arg.clip, 
+                          arg.align_to, arg.align_to_band, arg.align_using_band)
+        
+    if arg.cmd == "preprocess_pleiades_mosaic":
+        preprocess_pleiades_mosaic(arg.fld, arg.out_fld, arg.pattern)
 
     if arg.cmd == "preprocess_s1":
         preprocess_s1(arg.in_folder, arg.roi, arg.out_folder, arg.dem_fld, arg.geoid, arg.direction, arg.satellite,