KJ135
/
ISCE_INSAR
mirror of http://172.16.0.12:5000/WBJY/ISCE_INSAR.git
4
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

stripmapStack/unwrap: adjust indentation

LT1AB
Zhang Yunjun 2019-11-21 13:29:34 -08:00
parent 6e36a73fb0
commit 43aee1dcb3
1 changed files with 47 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

82
contrib/stack/stripmapStack/unwrap.py
View File

@ -258,33 +258,34 @@ def runUnwrapIcu(infile, outfile):
unwImage.finalizeImage() unwImage.finalizeImage()
unwImage.renderHdr() unwImage.renderHdr()
def runUnwrap2Stage(unwrappedIntFilename,connectedComponentsFilename,unwrapped2StageFilename, unwrapper_2stage_name=None, solver_2stage=None): def runUnwrap2Stage(unwrappedIntFilename,connectedComponentsFilename,unwrapped2StageFilename,
unwrapper_2stage_name=None, solver_2stage=None):
if unwrapper_2stage_name is None: if unwrapper_2stage_name is None:
unwrapper_2stage_name = 'REDARC0' unwrapper_2stage_name = 'REDARC0'
if solver_2stage is None: if solver_2stage is None:
# If unwrapper_2state_name is MCF then solver is ignored # If unwrapper_2state_name is MCF then solver is ignored
# and relaxIV MCF solver is used by default # and relaxIV MCF solver is used by default
solver_2stage = 'pulp' solver_2stage = 'pulp'
print('Unwrap 2 Stage Settings:') print('Unwrap 2 Stage Settings:')
print('Name: %s'%unwrapper_2stage_name) print('Name: %s'%unwrapper_2stage_name)
print('Solver: %s'%solver_2stage) print('Solver: %s'%solver_2stage)
inpFile = unwrappedIntFilename inpFile = unwrappedIntFilename
ccFile = connectedComponentsFilename ccFile = connectedComponentsFilename
outFile = unwrapped2StageFilename outFile = unwrapped2StageFilename
# Hand over to 2Stage unwrap # Hand over to 2Stage unwrap
unw = UnwrapComponents() unw = UnwrapComponents()
unw.setInpFile(inpFile) unw.setInpFile(inpFile)
unw.setConnCompFile(ccFile) unw.setConnCompFile(ccFile)
unw.setOutFile(outFile) unw.setOutFile(outFile)
unw.setSolver(solver_2stage) unw.setSolver(solver_2stage)
unw.setRedArcs(unwrapper_2stage_name) unw.setRedArcs(unwrapper_2stage_name)
unw.unwrapComponents() unw.unwrapComponents()
return return
def main(iargs=None): def main(iargs=None):
@ -303,24 +304,26 @@ def main(iargs=None):
if inps.method != 'icu': if inps.method != 'icu':
masterShelveDir = os.path.join(interferogramDir , 'masterShelve') masterShelveDir = os.path.join(interferogramDir , 'masterShelve')
if not os.path.exists(masterShelveDir): if not os.path.exists(masterShelveDir):
os.makedirs(masterShelveDir) os.makedirs(masterShelveDir)
inps.master = os.path.dirname(inps.master)
cpCmd='cp ' + os.path.join(inps.master, 'data*') +' '+masterShelveDir
os.system(cpCmd)
pckfile = os.path.join(masterShelveDir,'data')
print(pckfile)
metadata = extractInfoFromPickle(pckfile, inps)
inps.master = os.path.dirname(inps.master)
cpCmd='cp ' + os.path.join(inps.master, 'data*') +' '+masterShelveDir
os.system(cpCmd)
pckfile = os.path.join(masterShelveDir,'data')
print(pckfile)
metadata = extractInfoFromPickle(pckfile, inps)
######## ########
print ('unwrapping method : ' , inps.method) print ('unwrapping method : ' , inps.method)
if inps.method == 'snaphu': if inps.method == 'snaphu':
if inps.nomcf: if inps.nomcf:
fncall = runUnwrap fncall = runUnwrap
else: else:
fncall = runUnwrapMcf fncall = runUnwrapMcf
fncall(inps.intfile, inps.unwprefix + '_snaphu.unw', inps.cohfile, metadata, defomax=inps.defomax) fncall(inps.intfile, inps.unwprefix + '_snaphu.unw', inps.cohfile, metadata, defomax=inps.defomax)
elif inps.method == 'snaphu2stage': elif inps.method == 'snaphu2stage':
if inps.nomcf: if inps.nomcf:
fncall = runUnwrap fncall = runUnwrap
@ -329,11 +332,12 @@ def main(iargs=None):
fncall(inps.intfile, inps.unwprefix + '_snaphu.unw', inps.cohfile, metadata, defomax=inps.defomax) fncall(inps.intfile, inps.unwprefix + '_snaphu.unw', inps.cohfile, metadata, defomax=inps.defomax)
# adding in the two-stage # adding in the two-stage
runUnwrap2Stage(inps.unwprefix + '_snaphu.unw', inps.unwprefix + '_snaphu.unw.conncomp',inps.unwprefix + '_snaphu2stage.unw') runUnwrap2Stage(inps.unwprefix + '_snaphu.unw',
inps.unwprefix + '_snaphu.unw.conncomp',
inps.unwprefix + '_snaphu2stage.unw')
elif inps.method == 'icu': elif inps.method == 'icu':
runUnwrapIcu(inps.intfile, inps.unwprefix + '_icu.unw') runUnwrapIcu(inps.intfile, inps.unwprefix + '_icu.unw')
if __name__ == '__main__': if __name__ == '__main__':