prep_aria: support for ARIA product v3 correction layers (#1247)

* added loading of correction layers

* Add support for updated epoch stacks

* Mask out 0s

* updating correction func to be consistent w others

* Adjust sign error and handle multiple tropo files

* Update cli examples in prep_aria

* Reuse write_ifgram_stack and iono_split_spectrum to reduce maintainance burden

* add ARIA-tools example

* change 'write_model_stack' to 'write_timeseries' and correct indentations

* fix pre-commit hooks

* Update prep_aria.py

* remove unused variable

* remove unused arguments and comments

---------

Co-authored-by: mgovorcin <[email protected]>
Co-authored-by: Simran S Sangha <[email protected]>
Co-authored-by: Zhang Yunjun <[email protected]>
Co-authored-by: rzinke <[email protected]>
Co-authored-by: ehavazli <[email protected]>

Author: 6 people

src/mintpy/cli/prep_aria.py

@@ -49,11 +49,12 @@
   prep_aria.py -t SanFranSenDT42.txt
   prep_aria.py -s ../stack/ -d ../DEM/SRTM_3arcsec.dem -i '../incidenceAngle/*.vrt'
   prep_aria.py -s ../stack/ -d ../DEM/SRTM_3arcsec.dem -i '../incidenceAngle/*.vrt' -a '../azimuthAngle/*.vrt' -w ../mask/watermask.msk
+  prep_aria.py -s ../stack/ -d ../DEM/SRTM_3arcsec.dem -i '../incidenceAngle/*.vrt' --set '../stack/setStack.vrt' --tropo '../stack/troposphereTotal/HRRR_stack.vrt' --iono '../stack/ionStack.vrt'
 
   # download / extract / prepare inteferograms stack from ARIA using ARIA-tools:
-  # reference: https://github.com/aria-tools/ARIA-tools
-  ariaDownload.py -b '37.25 38.1 -122.6 -121.75' --track 42
-  ariaTSsetup.py -f 'products/*.nc' -b '37.25 38.1 -122.6 -121.75' --mask Download --num_threads 4 --verbose
+  ariaDownload.py --track 71 --bbox "34.21 34.31 -118.60 -118.43" --start 20240101 --end 20240301
+  ariaTSsetup.py -f "products/*.nc" --bbox "34.21 34.31 -118.60 -118.43" --mask Download --num_threads 4 --layers "solidEarthTide, troposphereTotal, ionosphere" --tropo_models HRRR
+  prep_aria.py -s stack -d DEM/glo_90.dem -i incidenceAngle/*.vrt -a azimuthAngle/*.vrt -w mask/esa_world_cover_2021.msk --set stack/setStack.vrt --tropo stack/troposphereTotal/HRRRStack.vrt --iono stack/ionoStack.vrt
 """
 
 def create_parser(subparsers=None):
@@ -106,6 +107,16 @@ def create_parser(subparsers=None):
                       help='Name of the azimuth angle file.')
     geom.add_argument('-w','--water-mask', dest='waterMaskFile', type=str,
                       help='Name of the water mask file')
+
+    # correction layers: troposphereTotal, ionosphere, solidEarthTides
+    corr = parser.add_argument_group('corrections')
+    corr.add_argument('-ct', '--tropo', dest='tropoFile', type=str,
+                      help='Name of the Troposhere Delay stack file', default=None)
+    corr.add_argument('-ci', '--iono', dest='ionoFile', type=str,
+                    help='Name of the Ionosphere Delay stack file', default=None)
+    corr.add_argument('-cs', '--set', dest='setFile', type=str,
+                    help='Name of the Solid Earth Tides stack file', default=None)
+
     return parser
 
 
@@ -146,10 +157,14 @@ def cmd_line_parse(iargs = None):
         # search for wildcard pattern
         fnames = glob.glob(iDict[key]) if iDict[key] else []
 
-        # user the first element if more than one exist
+        # return the first element if more than one exist
+        # except for tropo, for which multiple inputs could be passed
         if len(fnames) > 0:
-            iDict[key] = fnames[0]
-            print('{k:<{w}} : {f}'.format(k=key, w=max_digit, f=fnames[0]))
+            if 'tropo' not in key:
+                iDict[key] = fnames[0]
+            else:
+                iDict[key] = fnames
+            print('{k:<{w}} : {f}'.format(k=key, w=max_digit, f=iDict[key]))
 
         elif key in required_ds_keys:
             # raise exception if any required DS is missing