Apply current workspace changes
This commit is contained in:
+319
@@ -0,0 +1,319 @@
|
||||
#! /usr/bin/env python
|
||||
#################################################################
|
||||
### This program is part of PyINT v2.1 ###
|
||||
### Copy Right (c): 2017-2019, Yunmeng Cao ###
|
||||
### Author: Yunmeng Cao ###
|
||||
### Contact : ymcmrs@gmail.com ###
|
||||
#################################################################
|
||||
import os
|
||||
import sys
|
||||
import argparse
|
||||
|
||||
from pyint import _utils as ut
|
||||
|
||||
|
||||
INTRODUCTION = '''
|
||||
-------------------------------------------------------------------
|
||||
将 geocode_gamma.py 已地理编码的 GAMMA 二进制产品转换为
|
||||
LiCSBAS 所需的 GeoTIFF 格式, 并组织到 GEOC 目录结构。
|
||||
|
||||
本脚本 **不做任何地理编码或计算**, 仅执行格式转换:
|
||||
1. data2geotiff : GAMMA 二进制 (EQA) → GeoTIFF
|
||||
2. GMT grdmath : 从 look_vector 计算 E/N/U 分量
|
||||
3. 文件复制 : 组织到 GEOC/{Pair}/ 目录
|
||||
|
||||
所有地理编码产品由 geocode_gamma.py 提供:
|
||||
amp, corr, unw, diff_filt, diff, dem, lv_theta, lv_phi
|
||||
'''
|
||||
|
||||
EXAMPLE = '''
|
||||
Usage:
|
||||
gamma2licsbas_gamma.py projectName Mdate-Sdate
|
||||
gamma2licsbas_gamma.py PacayaT163TsxHhA 20150102-20150601
|
||||
-------------------------------------------------------------------
|
||||
'''
|
||||
|
||||
def cmdLineParse():
|
||||
parser = argparse.ArgumentParser(description='Convert geocoded GAMMA binaries to LiCSBAS GeoTIFF format.',\
|
||||
formatter_class=argparse.RawTextHelpFormatter,\
|
||||
epilog=INTRODUCTION+'\n'+EXAMPLE)
|
||||
|
||||
parser.add_argument('projectName',help='projectName for processing.')
|
||||
parser.add_argument('pair',help='Master-Slave, e.g., 20150101-20150106.')
|
||||
|
||||
inps = parser.parse_args()
|
||||
return inps
|
||||
|
||||
|
||||
def main(argv):
|
||||
|
||||
inps = cmdLineParse()
|
||||
projectName = inps.projectName
|
||||
Pair = inps.pair
|
||||
|
||||
scratchDir = os.getenv('SCRATCHDIR')
|
||||
ifgDir = scratchDir + '/' + projectName + "/ifgrams"
|
||||
demDir = scratchDir + '/' + projectName + "/DEM"
|
||||
workDir = ifgDir + '/' + Pair
|
||||
|
||||
templateDir = os.getenv('TEMPLATEDIR')
|
||||
templateFile = templateDir + "/" + projectName + ".template"
|
||||
templateDict=ut.update_template(templateFile)
|
||||
rlks = templateDict['range_looks']
|
||||
azlks = templateDict['azimuth_looks']
|
||||
masterDate = templateDict['masterDate']
|
||||
|
||||
#################################################################
|
||||
# DEM 参数 (直接引用, 不复制不删除)
|
||||
#################################################################
|
||||
DEM_par = demDir + '/' + masterDate + '_' + rlks + 'rlks.utm.dem.par'
|
||||
DEM = demDir + '/' + masterDate + '_' + rlks + 'rlks.utm.dem'
|
||||
DEM_tif = demDir + '/' + masterDate + '_' + rlks + 'rlks.utm.dem.tif'
|
||||
|
||||
# look_vector 文件 (由 geocode_gamma.py 生成)
|
||||
lv_theta = workDir + '/' + 'lv_theta'
|
||||
lv_phi = workDir + '/' + 'lv_phi'
|
||||
lv_theta_tif = workDir + '/' + 'lv_theta.geo.tif'
|
||||
lv_phi_tif = workDir + '/' + 'lv_phi.geo.tif'
|
||||
lv_theta_grd = workDir + '/' + 'lv_theta.grd'
|
||||
lv_theta_grd_final = workDir + '/' + 'inc_deg.nc'
|
||||
lv_phi_grd = workDir + '/' + 'lv_phi.grd'
|
||||
lv_phi_grd_final = workDir + '/' + 'azi_deg.nc'
|
||||
UE_grd = workDir + '/' + 'ue.grd'
|
||||
UN_grd = workDir + '/' + 'un.grd'
|
||||
UU_grd = workDir + '/' + 'uu.grd'
|
||||
UE_tif = workDir + '/' + 'ue.geo.E.tif'
|
||||
UN_tif = workDir + '/' + 'un.geo.N.tif'
|
||||
UU_tif = workDir + '/' + 'uu.geo.U.tif'
|
||||
|
||||
if not os.path.exists(lv_theta) or not os.path.exists(lv_phi):
|
||||
print(f"WARNING: look_vector 文件不存在, 请先运行 geocode_gamma.py {projectName} {Pair}")
|
||||
print(f" 缺少: lv_theta={os.path.exists(lv_theta)}, lv_phi={os.path.exists(lv_phi)}")
|
||||
|
||||
GEOC = scratchDir + '/' + projectName + '/GEOC'
|
||||
if not os.path.isdir(GEOC):
|
||||
os.mkdir(GEOC)
|
||||
Store_dir = GEOC + '/' + Pair
|
||||
if not os.path.isdir(Store_dir):
|
||||
os.mkdir(Store_dir)
|
||||
|
||||
# 判断DEM_tif是否存在
|
||||
if not os.path.exists(DEM_tif):
|
||||
call_str = 'data2geotiff ' + DEM_par + ' ' + DEM + ' 2 ' + DEM_tif
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {DEM_tif} already exists, skipping generation.")
|
||||
|
||||
# 判断lv_theta_tif是否存在
|
||||
if not os.path.exists(lv_theta_tif):
|
||||
call_str = 'data2geotiff ' + DEM_par + ' ' + lv_theta + ' 2 ' + lv_theta_tif
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {lv_theta_tif} already exists, skipping generation.")
|
||||
|
||||
# 判断lv_phi_tif是否存在
|
||||
if not os.path.exists(lv_phi_tif):
|
||||
call_str = 'data2geotiff ' + DEM_par + ' ' + lv_phi + ' 2 ' + lv_phi_tif
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {lv_phi_tif} already exists, skipping generation.")
|
||||
|
||||
# 判断lv_theta_grd是否存在
|
||||
if not os.path.exists(lv_theta_grd):
|
||||
call_str = 'gdal_translate -of GSBG ' + lv_theta_tif + ' ' + lv_theta_grd
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {lv_theta_grd} already exists, skipping conversion.")
|
||||
|
||||
# 判断lv_phi_grd是否存在
|
||||
if not os.path.exists(lv_phi_grd):
|
||||
call_str = 'gdal_translate -of GSBG ' + lv_phi_tif + ' ' + lv_phi_grd
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {lv_phi_grd} already exists, skipping conversion.")
|
||||
|
||||
# 判断lv_theta_grd_final是否存在
|
||||
if not os.path.exists(lv_theta_grd_final):
|
||||
call_str = 'gmt grdmath 90 ' + lv_theta_grd + ' 3.1415926 DIV 180 MUL SUB = ' + lv_theta_grd_final
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {lv_theta_grd_final} already exists, skipping calculation.")
|
||||
|
||||
# 判断lv_phi_grd_final是否存在
|
||||
if not os.path.exists(lv_phi_grd_final):
|
||||
call_str = 'gmt grdmath -180 ' + lv_phi_grd + ' 3.1415926 DIV 180 MUL SUB = ' + lv_phi_grd_final
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {lv_phi_grd_final} already exists, skipping calculation.")
|
||||
|
||||
# 判断UE_grd是否存在
|
||||
if not os.path.exists(UE_grd):
|
||||
call_str = 'gmt grdmath ' + lv_phi_grd_final + ' COSD ' + lv_theta_grd_final + ' SIND MUL NEG = ' + UE_grd
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {UE_grd} already exists, skipping calculation.")
|
||||
|
||||
# 判断UN_grd是否存在
|
||||
if not os.path.exists(UN_grd):
|
||||
call_str = 'gmt grdmath ' + lv_phi_grd_final + ' SIND ' + lv_theta_grd_final + ' SIND MUL = ' + UN_grd
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {UN_grd} already exists, skipping calculation.")
|
||||
|
||||
# 判断UU_grd是否存在
|
||||
if not os.path.exists(UU_grd):
|
||||
call_str = 'gmt grdmath ' + lv_theta_grd_final + ' COSD = ' + UU_grd
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {UU_grd} already exists, skipping calculation.")
|
||||
|
||||
# 判断UE_tif是否存在
|
||||
if not os.path.exists(UE_tif):
|
||||
call_str = 'gdal_translate -of GTiff ' + UE_grd + ' ' + UE_tif
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {UE_tif} already exists, skipping conversion.")
|
||||
|
||||
# 判断UN_tif是否存在
|
||||
if not os.path.exists(UN_tif):
|
||||
call_str = 'gdal_translate -of GTiff ' + UN_grd + ' ' + UN_tif
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {UN_tif} already exists, skipping conversion.")
|
||||
|
||||
# 判断UU_tif是否存在
|
||||
if not os.path.exists(UU_tif):
|
||||
call_str = 'gdal_translate -of GTiff ' + UU_grd + ' ' + UU_tif
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {UU_tif} already exists, skipping conversion.")
|
||||
|
||||
##### get unw.tif
|
||||
GeoMamp = workDir + '/geo_' + masterDate + '_' + rlks + 'rlks.amp'
|
||||
GeoMamp_tif = workDir + '/geo_' + masterDate + '_' + rlks + 'rlks.amp.tif'
|
||||
GeoCOR = workDir + '/geo_' + masterDate + '_' + rlks + 'rlks.diff_filt.cor'
|
||||
GeoCOR_tif = workDir + '/geo_' + Pair + '_' + rlks + 'rlks.diff_filt.cor.tif'
|
||||
GeoUNW_tif = workDir + '/geo_' + Pair + '_' + rlks + 'rlks.diff_filt.unw.tif'
|
||||
GeoUNW = workDir + '/geo_' + Pair + '_' + rlks + 'rlks.diff_filt.unw'
|
||||
GeoDIFF = workDir + '/geo_' + Pair + '_' + rlks + 'rlks.diff_filt'
|
||||
GeoDIFF_tif = workDir + '/geo_' + Pair + '_' + rlks + 'rlks.diff_filt.tif'
|
||||
Geodiff = workDir + '/geo_' + Pair + '_' + rlks + 'rlks.diff'
|
||||
Geodiff_tif = workDir + '/geo_' + Pair + '_' + rlks + 'rlks.diff.tif'
|
||||
GeoATMCOR_UNW = workDir + '/geo_' + Pair + '_' + rlks + 'rlks.diff_filt.unw.gacos'
|
||||
GeoATMCOR_UNW_tif = workDir + '/geo_' + Pair + '_' + rlks + 'rlks.diff_filt.unw.gacos.tif'
|
||||
Geodem = workDir + '/geo_' + masterDate + '_' + rlks + 'rlks.hgt'
|
||||
Geodem_tif = workDir + '/geo_' + masterDate + '_' + rlks + 'rlks.hgt.tif'
|
||||
|
||||
# 判断GeoUNW_tif是否存在
|
||||
if not os.path.exists(GeoUNW_tif):
|
||||
call_str = 'data2geotiff ' + DEM_par + ' ' + GeoUNW + ' ' + ' 2 ' + GeoUNW_tif
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {GeoUNW_tif} already exists, skipping conversion.")
|
||||
|
||||
# 判断Geodem_tif是否存在
|
||||
if not os.path.exists(Geodem_tif):
|
||||
call_str = 'data2geotiff ' + DEM_par + ' ' + Geodem + ' ' + ' 2 ' + Geodem_tif
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {Geodem_tif} already exists, skipping conversion.")
|
||||
|
||||
# 判断GeoCOR_tif是否存在
|
||||
if not os.path.exists(GeoCOR_tif):
|
||||
call_str = 'data2geotiff ' + DEM_par + ' ' + GeoCOR + ' ' + ' 2 ' + GeoCOR_tif
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {GeoCOR_tif} already exists, skipping conversion.")
|
||||
|
||||
# 判断GeoMamp_tif是否存在
|
||||
if not os.path.exists(GeoMamp_tif):
|
||||
call_str = 'data2geotiff ' + DEM_par + ' ' + GeoMamp + ' ' + ' 2 ' + GeoMamp_tif
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {GeoMamp_tif} already exists, skipping conversion.")
|
||||
|
||||
# 判断GeoDIFF_tif是否存在
|
||||
if not os.path.exists(GeoDIFF_tif):
|
||||
call_str = 'data2geotiff ' + DEM_par + ' ' + GeoDIFF + ' ' + ' 2 ' + GeoDIFF_tif
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {GeoDIFF_tif} already exists, skipping conversion.")
|
||||
|
||||
# 判断Geodiff_tif是否存在
|
||||
if not os.path.exists(Geodiff_tif):
|
||||
call_str = 'data2geotiff ' + DEM_par + ' ' +Geodiff + ' ' + ' 2 ' + Geodiff_tif
|
||||
os.system(call_str)
|
||||
else:
|
||||
print(f"File {Geodiff_tif} already exists, skipping conversion.")
|
||||
|
||||
# 判断GeoATMCOR_UNW_tif是否存在（GACOS校正后的解缠结果）
|
||||
if os.path.exists(GeoATMCOR_UNW) and not os.path.exists(GeoATMCOR_UNW_tif):
|
||||
call_str = 'data2geotiff ' + DEM_par + ' ' + GeoATMCOR_UNW + ' ' + ' 2 ' + GeoATMCOR_UNW_tif
|
||||
os.system(call_str)
|
||||
elif os.path.exists(GeoATMCOR_UNW_tif):
|
||||
print(f"File {GeoATMCOR_UNW_tif} already exists, skipping conversion.")
|
||||
|
||||
GeoCOR_tif0 = Store_dir + '/' + Pair + '.geo.cc.tif'
|
||||
GeoUNW_tif0 = Store_dir + '/' + Pair + '.geo.unw.tif'
|
||||
UU_tif0 = GEOC + '/' + 'uu.geo.U.tif'
|
||||
UE_tif0 = GEOC + '/' + 'ue.geo.E.tif'
|
||||
UN_tif0 = GEOC + '/' + 'un.geo.N.tif'
|
||||
dem_tif0 = GEOC + '/' + masterDate + '.geo.hgt.tif'
|
||||
amp_tif0 = GEOC + '/' + masterDate + '.geo.mli.tif'
|
||||
GeoDIFF_tif0 = Store_dir + '/' + Pair + '.geo.diff_pha.tif'
|
||||
Geodiff_tif0 = Store_dir + '/' + Pair + '.geo.diff_unfiltered_pha.tif'
|
||||
GeoATMCOR_UNW_tif0 = Store_dir + '/' + Pair + '.geo.unw.gacos.tif'
|
||||
# 判断并复制文件，如果目标文件已存在则不复制
|
||||
if not os.path.exists(GeoUNW_tif0):
|
||||
ut.copy_file(GeoUNW_tif, GeoUNW_tif0)
|
||||
else:
|
||||
print(f"File {GeoUNW_tif0} already exists, skipping copy.")
|
||||
|
||||
if not os.path.exists(GeoCOR_tif0):
|
||||
ut.copy_file(GeoCOR_tif, GeoCOR_tif0)
|
||||
else:
|
||||
print(f"File {GeoCOR_tif0} already exists, skipping copy.")
|
||||
|
||||
if not os.path.exists(UU_tif0):
|
||||
ut.copy_file(UU_tif, UU_tif0)
|
||||
else:
|
||||
print(f"File {UU_tif0} already exists, skipping copy.")
|
||||
|
||||
if not os.path.exists(UE_tif0):
|
||||
ut.copy_file(UE_tif, UE_tif0)
|
||||
else:
|
||||
print(f"File {UE_tif0} already exists, skipping copy.")
|
||||
|
||||
if not os.path.exists(UN_tif0):
|
||||
ut.copy_file(UN_tif, UN_tif0)
|
||||
else:
|
||||
print(f"File {UN_tif0} already exists, skipping copy.")
|
||||
if not os.path.exists(GeoDIFF_tif0):
|
||||
ut.copy_file(GeoDIFF_tif, GeoDIFF_tif0)
|
||||
else:
|
||||
print(f"File {GeoDIFF_tif0} already exists, skipping copy.")
|
||||
if not os.path.exists(Geodiff_tif0):
|
||||
ut.copy_file(Geodiff_tif, Geodiff_tif0)
|
||||
else:
|
||||
print(f"File {Geodiff_tif0} already exists, skipping copy.")
|
||||
|
||||
if os.path.exists(GeoATMCOR_UNW_tif) and not os.path.exists(GeoATMCOR_UNW_tif0):
|
||||
ut.copy_file(GeoATMCOR_UNW_tif, GeoATMCOR_UNW_tif0)
|
||||
elif os.path.exists(GeoATMCOR_UNW_tif0):
|
||||
print(f"File {GeoATMCOR_UNW_tif0} already exists, skipping copy.")
|
||||
|
||||
if not os.path.exists(dem_tif0):
|
||||
ut.copy_file(Geodem_tif, dem_tif0)
|
||||
else:
|
||||
print(f"File {dem_tif0} already exists, skipping copy.")
|
||||
|
||||
if not os.path.exists(amp_tif0):
|
||||
ut.copy_file(GeoMamp_tif, amp_tif0)
|
||||
else:
|
||||
print(f"File {amp_tif0} already exists, skipping copy.")
|
||||
|
||||
print("Convert to LiCSBAS format is done!")
|
||||
sys.exit(0)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main(sys.argv[:])
|
||||
Reference in New Issue
Block a user