Harmon/insar-management-system-v2
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add Sentinel-1 asset management and PyINT pipeline support

Browse Source
This commit is contained in:
Harmon
2026-05-14 03:02:56 +08:00
parent 508ca5641a
commit 729c5c2a7f
61 changed files with 7800 additions and 885 deletions
Download Patch File Download Diff File Expand all files Collapse all files
third_party/PyINT/pyint/down2slc_sen.py
Vendored
+61 -28
View File
@@ -13,6 +13,7 @@ import getopt
import time
import glob
import argparse
from pathlib import Path
from pyint import _utils as ut
@@ -23,10 +24,11 @@ def get_s1_date(raw_file):
return date
def get_satellite(raw_file):
if 'S1A_IW_SLC_' in raw_file:
s0 = 'A'
name = os.path.basename(str(raw_file))
if name.startswith('S1') and len(name) >= 3:
s0 = name[2]
else:
s0 = 'B'
s0 = 'A'
return s0
@@ -88,16 +90,13 @@ def main(argv):
# os.mkdir(opod_dir)
call_str = " eof --save-dir " + opod_dir + " -p " + down_dir
os.system(call_str)
if opod_dir and not os.path.isdir(opod_dir):
os.makedirs(opod_dir, exist_ok=True)
os.chdir(work_dir)
t_date = 't_' + date
call_str = 'ls ' + down_dir + '/S1*' + date + '*.zip > ' + t_date
os.system(call_str)
start_swath = templateDict['start_swath']
end_swath = templateDict['end_swath']
@@ -115,53 +114,87 @@ def main(argv):
# k_swath = '-'
k_swath = ut.get_sardata_swath(start_swath,end_swath)
raw_files = ut.read_txt2list(t_date)
raw_files = sorted(glob.glob(down_dir + '/S1*' + date + '*.zip') + glob.glob(down_dir + '/S1*' + date + '*.SAFE'))
if not raw_files:
raise FileNotFoundError('No Sentinel-1 scenes found for date ' + date)
with open(t_date, 'w', encoding='utf-8') as f:
for raw_file in raw_files:
f.write(str(raw_file) + '\n')
satellite = get_satellite(str(raw_files[0]))
#orbit_file = ut.download_s1_orbit(date,opod_dir,satellite=satellite)
zipfile_ref=str(raw_files[0])
outfile_name=zipfile_ref.split('/')[-1].split('.')[0]
burst_number_table_ref=outfile_name + '.BURST_tab'
call_str = 'S1_BURST_tab_from_zipfile.py 3 --zip_ref_list ' + t_date + ' --zip_list ' + t_date
os.system(call_str)
if raw_files and os.path.isdir(str(raw_files[0])):
call_str = 'S1_BURST_tab_from_zipfile.py 3 --dir_ref_list ' + t_date + ' --dir_list ' + t_date
else:
call_str = 'S1_BURST_tab_from_zipfile.py 3 --zip_ref_list ' + t_date + ' --zip_list ' + t_date
rc = os.system(call_str)
if rc != 0:
raise RuntimeError('S1_BURST_tab_from_zipfile.py failed with rc=' + str(rc))
# call_str = 'S1_import_SLC_from_zipfiles ' + t_date + ' ' + burst_number_table_ref + ' vv 0 ' + k_swath
call_str = 'read_S1_TOPS_SLC.py ' + zipfile_ref + ' --burst_sel ' + burst_number_table_ref + ' --pol vv --root_name ' + date + ' --sw_start ' + start_swath + ' --swn ' + end_swath + ' --OPOD_dir ' + opod_dir
os.system(call_str)
rc = os.system(call_str)
if rc != 0:
raise RuntimeError('read_S1_TOPS_SLC.py failed with rc=' + str(rc))
os.chdir(work_dir)
call_str = "rename 's/vv.iw1.slc/IW1.slc/g' *"
#call_str = "rename vv.slc.iw1 IW1.slc * "
os.system(call_str)
call_str = "rename 's/vv.iw2.slc/IW2.slc/g' *"
#call_str = "rename vv.slc.iw2 IW2.slc * "
os.system(call_str)
call_str = "rename 's/vv.iw3.slc/IW3.slc/g' *"
#call_str = "rename vv.slc.iw3 IW3.slc * "
os.system(call_str)
call_str = "rename 's/tops_par/TOPS_par/g' *.tops_par "
os.system(call_str)
rename_rules = [
('vv.iw1.slc', 'IW1.slc'),
('vv.iw2.slc', 'IW2.slc'),
('vv.iw3.slc', 'IW3.slc'),
('vv.SLC_tab', 'SLC_tab'),
('vv.slc', 'slc'),
('tops_par', 'TOPS_par'),
]
for src_token, dst_token in rename_rules:
for candidate in list(Path(work_dir).iterdir()):
if src_token not in candidate.name:
continue
target = candidate.with_name(candidate.name.replace(src_token, dst_token))
if target != candidate and not target.exists():
candidate.rename(target)
SLC_Tab = work_dir + '/' + date+'_SLC_Tab'
SLC_list = sorted(glob.glob(work_dir + '/*IW*.slc'))
SLC_par_list = sorted(glob.glob(work_dir + '/*IW*.slc.par'))
TOP_par_list = sorted(glob.glob(work_dir + '/*IW*.slc.TOPS_par'))
if not TOP_par_list:
TOP_par_list = sorted(glob.glob(work_dir + '/*IW*.slc.tops_par'))
if (not SLC_list or not SLC_par_list or not TOP_par_list
or len(SLC_list) != len(SLC_par_list)
or len(SLC_list) != len(TOP_par_list)):
raise RuntimeError('Sentinel-1 SLC generation did not produce expected IW SLC products for ' + date)
if os.path.isfile(SLC_Tab):
os.remove(SLC_Tab)
for kk in range(len(SLC_list)):
call_str = 'echo ' + SLC_list[kk] + ' ' + SLC_par_list[kk] + ' ' + TOP_par_list[kk] + ' >> ' + SLC_Tab
os.system(call_str)
rc = os.system(call_str)
if rc != 0:
raise RuntimeError('failed to append SLC tab with rc=' + str(rc))
BURST = SLC_par_list[kk].replace('slc.par','burst.par')
call_str = 'SLC_burst_corners ' + SLC_par_list[kk] + ' ' + TOP_par_list[kk] + ' > ' +BURST
os.system(call_str)
rc = os.system(call_str)
if rc != 0:
raise RuntimeError('SLC_burst_corners failed with rc=' + str(rc))
call_str = "echo 'SLC has already down' >down2slc.dat"
os.system(call_str)
TSLC = work_dir + '/' + date + '.slc'
TSLCPar = work_dir + '/' + date + '.slc.par'
if not os.path.isfile(TSLC) or not os.path.isfile(TSLCPar):
call_str = 'SLC_mosaic_ScanSAR ' + SLC_Tab + ' ' + TSLC + ' ' + TSLCPar + ' 10 2'
rc = os.system(call_str)
if rc != 0:
raise RuntimeError('SLC_mosaic_ScanSAR failed with rc=' + str(rc))
print("Down to SLC for %s is done! " % date)
sys.exit(1)
sys.exit(0)
if __name__ == '__main__':
main(sys.argv[:])