100 lines
3.5 KiB
Python
100 lines
3.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
# Bekaert David
|
||
|
|
|
||
|
|
|
||
|
|
import os
|
||
|
|
import glob
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
|
||
|
|
def createParser():
|
||
|
|
'''
|
||
|
|
Create command line parser.
|
||
|
|
'''
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser(description='Script that attempts to recognize the sensor automatically and then call the correspodning unzips/unpacks command.')
|
||
|
|
parser.add_argument('-i', '--input', dest='input', type=str, required=True,
|
||
|
|
help='directory which has all the slc data.')
|
||
|
|
parser.add_argument('-rmfile', '--rmfile', dest='rmfile',action='store_true', default=False,
|
||
|
|
help='Optional: remove zip/tar/compressed files after unpacking into date structure (default is to keep in archive folder)')
|
||
|
|
parser.add_argument('-o', '--output', dest='output', type=str, required=False,
|
||
|
|
help='Optional: output directory which will be used for unpacking into isce format (run file generation only).')
|
||
|
|
parser.add_argument('-t', '--text_cmd', dest='text_cmd', type=str, default='source ~/.bash_profile;'
|
||
|
|
, help='Optional: text command to be added to the beginning of each line of the run files. Default: source ~/.bash_profile;')
|
||
|
|
|
||
|
|
return parser
|
||
|
|
|
||
|
|
def cmdLineParse(iargs=None):
|
||
|
|
'''
|
||
|
|
Command line parser.
|
||
|
|
'''
|
||
|
|
|
||
|
|
parser = createParser()
|
||
|
|
return parser.parse_args(args = iargs)
|
||
|
|
|
||
|
|
def main(iargs=None):
|
||
|
|
'''
|
||
|
|
The main driver.
|
||
|
|
'''
|
||
|
|
|
||
|
|
inps = cmdLineParse(iargs)
|
||
|
|
## parsing of the required input arguments
|
||
|
|
inputDir = os.path.abspath(inps.input)
|
||
|
|
## parsing of the optional input arguments
|
||
|
|
# output dir for generating runfile
|
||
|
|
if inps.output:
|
||
|
|
outputDir = os.path.abspath(inps.output)
|
||
|
|
outputDir_str = ' -o ' + outputDir
|
||
|
|
else:
|
||
|
|
outputDir_str = ''
|
||
|
|
# textcommand to be added to the runfile start
|
||
|
|
text_str = ' -t "' + inps.text_cmd + '"'
|
||
|
|
# delete zip file option
|
||
|
|
if inps.rmfile:
|
||
|
|
rmfile_str = ' -rmfile'
|
||
|
|
else:
|
||
|
|
rmfile_str = ''
|
||
|
|
|
||
|
|
# search criteria for the different sensors
|
||
|
|
ENV_str = 'ASA*' # Envisat
|
||
|
|
ALOS1_str = 'ALPSRP*' # ALOS-1/ALOS-2 Palsar, zip files and extracted files
|
||
|
|
CSK_str = 'EL*' # CSK, zip files
|
||
|
|
CSK_str2 = 'CSK*.h5' # CSK, extracted files
|
||
|
|
RSAT2_str = 'RS2*SLC*' # RSAT2 zip files
|
||
|
|
RSAT2_str2 = 'imagery_HH.tif' # RSAT2 extracted files
|
||
|
|
TSX_TDX_str = 'dims_op*' # TSX zip files
|
||
|
|
TSX_TDX_str2 = 'T*X*.xml' # TSX extracted files
|
||
|
|
|
||
|
|
# combine together
|
||
|
|
sensor_str_list = (ENV_str,ALOS1_str,CSK_str,CSK_str2,RSAT2_str,RSAT2_str2,TSX_TDX_str,TSX_TDX_str2)
|
||
|
|
sensor_list = ('Envisat','ALOS1','CSK','CSK','RSAT2','RSAT2','TSX/TDX','TSX/TDX')
|
||
|
|
sensor_unpackcommand = ('TODO','TODO','TODO','TODO','prepSlcRSAT2.py','prepSlcRSAT2.py','TODO','TODO')
|
||
|
|
Sensors = dict(zip(sensor_str_list,sensor_list))
|
||
|
|
Sensors_unpack = dict(zip(sensor_str_list,sensor_unpackcommand))
|
||
|
|
|
||
|
|
# Loop over the different sensor strings and try to find them
|
||
|
|
sensor_found = False
|
||
|
|
for sensor_str in Sensors:
|
||
|
|
for file in glob.iglob(os.path.join(inputDir,'**',sensor_str),recursive=True):
|
||
|
|
sensor_found = True
|
||
|
|
sensor_str_keep = sensor_str
|
||
|
|
break
|
||
|
|
|
||
|
|
# report back to user
|
||
|
|
if sensor_found:
|
||
|
|
print("Looks like " + Sensors[sensor_str_keep])
|
||
|
|
cmd = Sensors_unpack[sensor_str_keep] + ' -i ' + inputDir + rmfile_str + outputDir_str + text_str
|
||
|
|
print(cmd)
|
||
|
|
os.system(cmd)
|
||
|
|
|
||
|
|
else:
|
||
|
|
print("Did not find the sensor automatically, unzip and run unpack routines manual")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
|
||
|
|
main()
|
||
|
|
|
||
|
|
|