308 lines
9.5 KiB
Python
308 lines
9.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
||
|
|
||
|
#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
# Copyright 2016 California Institute of Technology. ALL RIGHTS RESERVED.
|
||
|
#
|
||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
# you may not use this file except in compliance with the License.
|
||
|
# You may obtain a copy of the License at
|
||
|
#
|
||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
#
|
||
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
# See the License for the specific language governing permissions and
|
||
|
# limitations under the License.
|
||
|
#
|
||
|
# United States Government Sponsorship acknowledged. This software is subject to
|
||
|
# U.S. export control laws and regulations and has been classified as 'EAR99 NLR'
|
||
|
# (No [Export] License Required except when exporting to an embargoed country,
|
||
|
# end user, or in support of a prohibited end use). By downloading this software,
|
||
|
# the user agrees to comply with all applicable U.S. export laws and regulations.
|
||
|
# The user has the responsibility to obtain export licenses, or other export
|
||
|
# authority as may be required before exporting this software to any 'EAR99'
|
||
|
# embargoed foreign country or citizen of those countries.
|
||
|
#
|
||
|
# Authors: Piyush Agram, Heresh Fattahi, David Bekaert
|
||
|
#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
import isce
|
||
|
import sys
|
||
|
import isceobj
|
||
|
from contrib.Snaphu.Snaphu import Snaphu
|
||
|
from isceobj.Constants import SPEED_OF_LIGHT
|
||
|
import argparse
|
||
|
import os
|
||
|
import pickle
|
||
|
import sys
|
||
|
import shelve
|
||
|
from contrib.UnwrapComp.unwrapComponents import UnwrapComponents
|
||
|
|
||
|
def createParser():
|
||
|
'''
|
||
|
Create command line parser.
|
||
|
'''
|
||
|
|
||
|
parser = argparse.ArgumentParser(description='Unwrap interferogram using snaphu')
|
||
|
parser.add_argument('-i', '--ifg', dest='intfile', type=str, required=True,
|
||
|
help='Input interferogram')
|
||
|
parser.add_argument('-c', '--coh', dest='cohfile', type=str, required=True,
|
||
|
help='Coherence file')
|
||
|
parser.add_argument('-u', '--unwprefix', dest='unwprefix', type=str, required=False,
|
||
|
help='Output unwrapped file prefix')
|
||
|
parser.add_argument('--nomcf', action='store_true', default=False,
|
||
|
help='Run full snaphu and not in MCF mode')
|
||
|
parser.add_argument('-a','--alks', dest='azlooks', type=int, default=1,
|
||
|
help='Number of azimuth looks')
|
||
|
parser.add_argument('-r', '--rlks', dest='rglooks', type=int, default=1,
|
||
|
help='Number of range looks')
|
||
|
parser.add_argument('-d', '--defomax', dest='defomax', type=float, default=2.0,
|
||
|
help='Max cycles of deformation')
|
||
|
parser.add_argument('-m', '--method', dest='method', type=str, default='snaphu2stage',
|
||
|
help='unwrapping method (snaphu, snaphu2stage, icu)')
|
||
|
|
||
|
return parser
|
||
|
|
||
|
|
||
|
def cmdLineParse(iargs=None):
|
||
|
'''
|
||
|
Command line parser.
|
||
|
'''
|
||
|
|
||
|
parser = createParser()
|
||
|
return parser.parse_args(args = iargs)
|
||
|
|
||
|
|
||
|
def extractInfo(inps):
|
||
|
'''
|
||
|
Extract required information
|
||
|
'''
|
||
|
from isceobj.Planet.Planet import Planet
|
||
|
from isceobj.Util.geo.ellipsoid import Ellipsoid
|
||
|
|
||
|
|
||
|
planet = Planet(pname='Earth')
|
||
|
elp = Ellipsoid(planet.ellipsoid.a, planet.ellipsoid.e2, 'WGS84')
|
||
|
|
||
|
# for now hard-code default..., can be set to zero for deformation but for height related work needs to be actuals...
|
||
|
hgt = 0
|
||
|
lat = 0
|
||
|
hdg = 0
|
||
|
wavelength = 0.056
|
||
|
data ={}
|
||
|
data['wavelength'] = wavelength
|
||
|
data['altitude'] = hgt
|
||
|
data['earthRadius'] = elp.local_radius_of_curvature(lat, hdg)
|
||
|
data['rglooks'] = inps.rglooks
|
||
|
data['azlooks'] = inps.azlooks
|
||
|
|
||
|
return data
|
||
|
|
||
|
def runUnwrap(infile, outfile, corfile, config, costMode = None,initMethod = None, defomax = None, initOnly = None):
|
||
|
|
||
|
if costMode is None:
|
||
|
costMode = 'DEFO'
|
||
|
|
||
|
if initMethod is None:
|
||
|
initMethod = 'MST'
|
||
|
|
||
|
if defomax is None:
|
||
|
defomax = 4.0
|
||
|
|
||
|
if initOnly is None:
|
||
|
initOnly = False
|
||
|
|
||
|
wrapName = infile
|
||
|
unwrapName = outfile
|
||
|
|
||
|
img = isceobj.createImage()
|
||
|
img.load(infile + '.xml')
|
||
|
|
||
|
|
||
|
wavelength = config['wavelength']
|
||
|
width = img.getWidth()
|
||
|
length = img.getLength()
|
||
|
earthRadius = config['earthRadius']
|
||
|
altitude = config['altitude']
|
||
|
rangeLooks = config['rglooks']
|
||
|
azimuthLooks = config['azlooks']
|
||
|
#corrLooks = config['corrlooks']
|
||
|
maxComponents = 20
|
||
|
|
||
|
snp = Snaphu()
|
||
|
snp.setInitOnly(initOnly)
|
||
|
snp.setInput(wrapName)
|
||
|
snp.setOutput(unwrapName)
|
||
|
snp.setWidth(width)
|
||
|
snp.setCostMode(costMode)
|
||
|
snp.setEarthRadius(earthRadius)
|
||
|
snp.setWavelength(wavelength)
|
||
|
snp.setAltitude(altitude)
|
||
|
snp.setCorrfile(corfile)
|
||
|
snp.setInitMethod(initMethod)
|
||
|
# snp.setCorrLooks(corrLooks)
|
||
|
snp.setMaxComponents(maxComponents)
|
||
|
snp.setDefoMaxCycles(defomax)
|
||
|
snp.setRangeLooks(rangeLooks)
|
||
|
snp.setAzimuthLooks(azimuthLooks)
|
||
|
snp.setCorFileFormat('FLOAT_DATA')
|
||
|
snp.prepare()
|
||
|
snp.unwrap()
|
||
|
|
||
|
######Render XML
|
||
|
outImage = isceobj.Image.createUnwImage()
|
||
|
outImage.setFilename(unwrapName)
|
||
|
outImage.setWidth(width)
|
||
|
outImage.setLength(length)
|
||
|
outImage.setAccessMode('read')
|
||
|
#outImage.createImage()
|
||
|
outImage.renderHdr()
|
||
|
outImage.renderVRT()
|
||
|
#outImage.finalizeImage()
|
||
|
|
||
|
#####Check if connected components was created
|
||
|
if snp.dumpConnectedComponents:
|
||
|
connImage = isceobj.Image.createImage()
|
||
|
connImage.setFilename(unwrapName+'.conncomp')
|
||
|
#At least one can query for the name used
|
||
|
connImage.setWidth(width)
|
||
|
connImage.setLength(length)
|
||
|
connImage.setAccessMode('read')
|
||
|
connImage.setDataType('BYTE')
|
||
|
# connImage.createImage()
|
||
|
connImage.renderHdr()
|
||
|
connImage.renderVRT()
|
||
|
# connImage.finalizeImage()
|
||
|
|
||
|
return
|
||
|
|
||
|
|
||
|
def runUnwrapMcf(infile, outfile, corfile, config, defomax=2):
|
||
|
runUnwrap(infile, outfile, corfile, config, costMode = 'SMOOTH',initMethod = 'MCF', defomax = defomax, initOnly = True)
|
||
|
return
|
||
|
|
||
|
|
||
|
def runUnwrapIcu(infile, outfile):
|
||
|
from mroipac.icu.Icu import Icu
|
||
|
#Setup images
|
||
|
#ampImage
|
||
|
# ampImage = obj.insar.resampAmpImage.copy(access_mode='read')
|
||
|
# width = self.ampImage.getWidth()
|
||
|
|
||
|
img = isceobj.createImage()
|
||
|
img.load(infile + '.xml')
|
||
|
|
||
|
|
||
|
width = img.getWidth()
|
||
|
|
||
|
#intImage
|
||
|
intImage = isceobj.createIntImage()
|
||
|
intImage.initImage(infile, 'read', width)
|
||
|
intImage.createImage()
|
||
|
|
||
|
|
||
|
#unwImage
|
||
|
unwImage = isceobj.Image.createImage()
|
||
|
unwImage.setFilename(outfile)
|
||
|
unwImage.setWidth(width)
|
||
|
unwImage.imageType = 'unw'
|
||
|
unwImage.bands = 2
|
||
|
unwImage.scheme = 'BIL'
|
||
|
unwImage.dataType = 'FLOAT'
|
||
|
unwImage.setAccessMode('write')
|
||
|
unwImage.createImage()
|
||
|
|
||
|
#unwrap with icu
|
||
|
icuObj = Icu()
|
||
|
icuObj.filteringFlag = False
|
||
|
icuObj.useAmplitudeFlag = False
|
||
|
icuObj.singlePatch = True
|
||
|
icuObj.initCorrThreshold = 0.1
|
||
|
icuObj.icu(intImage=intImage, unwImage = unwImage)
|
||
|
|
||
|
#ampImage.finalizeImage()
|
||
|
intImage.finalizeImage()
|
||
|
unwImage.finalizeImage()
|
||
|
unwImage.renderHdr()
|
||
|
|
||
|
def runUnwrap2Stage(unwrappedIntFilename,connectedComponentsFilename,unwrapped2StageFilename, unwrapper_2stage_name=None, solver_2stage=None):
|
||
|
|
||
|
if unwrapper_2stage_name is None:
|
||
|
unwrapper_2stage_name = 'REDARC0'
|
||
|
|
||
|
if solver_2stage is None:
|
||
|
# If unwrapper_2state_name is MCF then solver is ignored
|
||
|
# and relaxIV MCF solver is used by default
|
||
|
solver_2stage = 'pulp'
|
||
|
|
||
|
print('Unwrap 2 Stage Settings:')
|
||
|
print('Name: %s'%unwrapper_2stage_name)
|
||
|
print('Solver: %s'%solver_2stage)
|
||
|
|
||
|
inpFile = unwrappedIntFilename
|
||
|
ccFile = connectedComponentsFilename
|
||
|
outFile = unwrapped2StageFilename
|
||
|
|
||
|
# Hand over to 2Stage unwrap
|
||
|
unw = UnwrapComponents()
|
||
|
unw.setInpFile(inpFile)
|
||
|
unw.setConnCompFile(ccFile)
|
||
|
unw.setOutFile(outFile)
|
||
|
unw.setSolver(solver_2stage)
|
||
|
unw.setRedArcs(unwrapper_2stage_name)
|
||
|
unw.unwrapComponents()
|
||
|
return
|
||
|
|
||
|
|
||
|
def main(iargs=None):
|
||
|
'''
|
||
|
The main driver.
|
||
|
'''
|
||
|
|
||
|
inps = cmdLineParse(iargs)
|
||
|
if (inps.method != 'icu') and (inps.method != 'snaphu') and (inps.method != 'snaphu2stage'):
|
||
|
raise Exception("Unwrapping method needs to be either icu, snaphu or snaphu2stage")
|
||
|
|
||
|
########
|
||
|
interferogramDir = os.path.dirname(inps.intfile)
|
||
|
|
||
|
if not inps.unwprefix:
|
||
|
intfile = (os.path.basename(inps.intfile))
|
||
|
vals = intfile.split('.int')
|
||
|
inps.unwprefix = vals[0]
|
||
|
|
||
|
if inps.method != 'icu':
|
||
|
metadata = extractInfo(inps)
|
||
|
|
||
|
########
|
||
|
print ('unwrapping method : ' , inps.method)
|
||
|
if inps.method == 'snaphu':
|
||
|
if inps.nomcf:
|
||
|
fncall = runUnwrap
|
||
|
else:
|
||
|
fncall = runUnwrapMcf
|
||
|
fncall(inps.intfile, inps.unwprefix + '_snaphu.unw', inps.cohfile, metadata, defomax=inps.defomax)
|
||
|
elif inps.method == 'snaphu2stage':
|
||
|
if inps.nomcf:
|
||
|
fncall = runUnwrap
|
||
|
else:
|
||
|
fncall = runUnwrapMcf
|
||
|
fncall(inps.intfile, inps.unwprefix + '_snaphu.unw', inps.cohfile, metadata, defomax=inps.defomax)
|
||
|
|
||
|
# adding in the two-stage
|
||
|
runUnwrap2Stage(inps.unwprefix + '_snaphu.unw', inps.unwprefix + '_snaphu.unw.conncomp',inps.unwprefix + '_snaphu2stage.unw')
|
||
|
|
||
|
|
||
|
elif inps.method == 'icu':
|
||
|
runUnwrapIcu(inps.intfile, inps.unwprefix + '_icu.unw')
|
||
|
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
|
||
|
main()
|