253 lines
7.9 KiB
Python
Executable File
253 lines
7.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
#
|
|
# Author: Piyush Agram
|
|
# Copyright 2016
|
|
#
|
|
# Heresh Fattahi: Adopted for stack processing
|
|
|
|
import argparse
|
|
import numpy as np
|
|
import os
|
|
import isce
|
|
import isceobj
|
|
import logging
|
|
import datetime
|
|
from isceobj.Location.Offset import OffsetField, Offset
|
|
from isceobj.Util.decorators import use_api
|
|
import s1a_isce_utils as ut
|
|
|
|
#logger = logging.getLogger('isce.topsinsar.rangecoreg')
|
|
|
|
def createParser():
|
|
parser = argparse.ArgumentParser( description='Estimate range misregistration using overlap bursts')
|
|
|
|
parser.add_argument('-o', '--out_range', type=str, dest='output', default='misreg.txt',
|
|
help='Output textfile with the constant range offset')
|
|
parser.add_argument('-t', '--snr_threshold', type=float, dest='offsetSNRThreshold', default=6.0,
|
|
help='SNR threshold for overlap masking')
|
|
|
|
parser.add_argument('-m','--master', type=str, dest='master', required=True,
|
|
help='Master image')
|
|
parser.add_argument('-s', '--slave',type=str, dest='slave', required=True,
|
|
help='Slave image')
|
|
|
|
return parser
|
|
|
|
def cmdLineParse(iargs=None):
|
|
'''
|
|
Command line parser.
|
|
'''
|
|
parser = createParser()
|
|
return parser.parse_args(args=iargs)
|
|
|
|
|
|
def runAmpcor(master, slave):
|
|
'''
|
|
Run one ampcor process.
|
|
'''
|
|
import isceobj
|
|
from mroipac.ampcor.Ampcor import Ampcor
|
|
|
|
mImg = isceobj.createSlcImage()
|
|
mImg.load(master + '.xml')
|
|
mImg.setAccessMode('READ')
|
|
mImg.createImage()
|
|
|
|
sImg = isceobj.createSlcImage()
|
|
sImg.load(slave + '.xml')
|
|
sImg.setAccessMode('READ')
|
|
sImg.createImage()
|
|
|
|
objAmpcor = Ampcor('ampcor_burst')
|
|
objAmpcor.configure()
|
|
objAmpcor.setImageDataType1('mag')
|
|
objAmpcor.setImageDataType2('mag')
|
|
|
|
|
|
if objAmpcor.acrossGrossOffset is None:
|
|
coarseAcross = 0
|
|
|
|
if objAmpcor.downGrossOffset is None:
|
|
coarseDown = 0
|
|
|
|
objAmpcor.windowSizeWidth = 64
|
|
objAmpcor.windowSizeHeight = 32
|
|
objAmpcor.searchWindowSizeWidth = 16
|
|
objAmpcor.searchWindowSizeHeight = 16
|
|
objAmpcor.oversamplingFactor = 32
|
|
|
|
xMargin = 2*objAmpcor.searchWindowSizeWidth + objAmpcor.windowSizeWidth
|
|
yMargin = 2*objAmpcor.searchWindowSizeHeight + objAmpcor.windowSizeHeight
|
|
|
|
firstAc = 1000
|
|
|
|
#####Compute image positions
|
|
|
|
offDn = objAmpcor.windowSizeHeight//2 + 1
|
|
offAc = firstAc+xMargin
|
|
|
|
offDnmax = mImg.getLength() - objAmpcor.windowSizeHeight//2 - 1
|
|
lastAc = int(mImg.width - 1000 - xMargin)
|
|
|
|
if not objAmpcor.firstSampleAcross:
|
|
objAmpcor.setFirstSampleAcross(offAc)
|
|
|
|
if not objAmpcor.lastSampleAcross:
|
|
objAmpcor.setLastSampleAcross(lastAc)
|
|
|
|
if not objAmpcor.numberLocationAcross:
|
|
objAmpcor.setNumberLocationAcross(80)
|
|
|
|
if not objAmpcor.firstSampleDown:
|
|
objAmpcor.setFirstSampleDown(offDn)
|
|
|
|
if not objAmpcor.lastSampleDown:
|
|
objAmpcor.setLastSampleDown(offDnmax)
|
|
|
|
###Since we are only dealing with overlaps
|
|
objAmpcor.setNumberLocationDown(20)
|
|
|
|
#####Override gross offsets if not provided
|
|
if not objAmpcor.acrossGrossOffset:
|
|
objAmpcor.setAcrossGrossOffset(coarseAcross)
|
|
|
|
if not objAmpcor.downGrossOffset:
|
|
objAmpcor.setDownGrossOffset(coarseDown)
|
|
|
|
|
|
objAmpcor.setImageDataType1('mag')
|
|
objAmpcor.setImageDataType2('mag')
|
|
|
|
objAmpcor.setFirstPRF(1.0)
|
|
objAmpcor.setSecondPRF(1.0)
|
|
objAmpcor.setFirstRangeSpacing(1.0)
|
|
objAmpcor.setSecondRangeSpacing(1.0)
|
|
objAmpcor(mImg, sImg)
|
|
|
|
mImg.finalizeImage()
|
|
sImg.finalizeImage()
|
|
|
|
return objAmpcor.getOffsetField()
|
|
|
|
|
|
def main(iargs=None):
|
|
|
|
inps = cmdLineParse(iargs)
|
|
|
|
'''
|
|
Estimate constant offset in range.
|
|
'''
|
|
|
|
#if not self.doESD:
|
|
# return
|
|
|
|
#catalog = isceobj.Catalog.createCatalog(self._insar.procDoc.name)
|
|
|
|
#swathList = self._insar.getValidSwathList(self.swaths)
|
|
masterSwathList = ut.getSwathList(os.path.join(inps.master, 'overlap'))
|
|
slaveSwathList = ut.getSwathList(os.path.join(inps.slave, 'overlap'))
|
|
swathList = list(sorted(set(masterSwathList+slaveSwathList)))
|
|
|
|
rangeOffsets = []
|
|
snr = []
|
|
|
|
for swath in swathList:
|
|
|
|
#if self._insar.numberOfCommonBursts[swath-1] < 2:
|
|
# print('Skipping range coreg for swath IW{0}'.format(swath))
|
|
# continue
|
|
|
|
#minBurst, maxBurst = self._insar.commonMasterBurstLimits(swath-1)
|
|
|
|
#maxBurst = maxBurst - 1 ###For overlaps
|
|
|
|
#masterTop = self._insar.loadProduct( os.path.join(self._insar.masterSlcOverlapProduct, 'top_IW{0}.xml'.format(swath)))
|
|
#masterBottom = self._insar.loadProduct( os.path.join(self._insar.masterSlcOverlapProduct , 'bottom_IW{0}.xml'.format(swath)))
|
|
masterTop = ut.loadProduct(os.path.join(inps.master , 'overlap','IW{0}_top.xml'.format(swath)))
|
|
masterBottom = ut.loadProduct(os.path.join(inps.master ,'overlap', 'IW{0}_bottom.xml'.format(swath)))
|
|
slaveTop = ut.loadProduct(os.path.join(inps.slave, 'overlap', 'IW{0}_top.xml'.format(swath)))
|
|
slaveBottom = ut.loadProduct(os.path.join(inps.slave, 'overlap', 'IW{0}_bottom.xml'.format(swath)))
|
|
|
|
#slaveTop = self._insar.loadProduct( os.path.join(self._insar.coregOverlapProduct , 'top_IW{0}.xml'.format(swath)))
|
|
#slaveBottom = self._insar.loadProduct( os.path.join(self._insar.coregOverlapProduct, 'bottom_IW{0}.xml'.format(swath)))
|
|
minMaster = masterTop.bursts[0].burstNumber
|
|
maxMaster = masterTop.bursts[-1].burstNumber
|
|
|
|
minSlave = slaveTop.bursts[0].burstNumber
|
|
maxSlave = slaveTop.bursts[-1].burstNumber
|
|
|
|
minBurst = max(minSlave, minMaster)
|
|
maxBurst = min(maxSlave, maxMaster)
|
|
#maxBurst = maxBurst - 1 ###For overlaps
|
|
maxBurst = maxBurst + 1
|
|
|
|
for pair in [(masterTop,slaveTop), (masterBottom,slaveBottom)]:
|
|
for ii in range(minBurst,maxBurst):
|
|
mFile = pair[0].bursts[ii-minMaster].image.filename
|
|
sFile = pair[1].bursts[ii-minSlave].image.filename
|
|
|
|
field = runAmpcor(mFile, sFile)
|
|
|
|
for offset in field:
|
|
rangeOffsets.append(offset.dx)
|
|
snr.append(offset.snr)
|
|
|
|
###Cull
|
|
mask = np.logical_and(np.array(snr) > inps.offsetSNRThreshold, np.abs(rangeOffsets) < 1.2)
|
|
val = np.array(rangeOffsets)[mask]
|
|
|
|
medianval = np.median(val)
|
|
meanval = np.mean(val)
|
|
stdval = np.std(val)
|
|
|
|
# convert the estimations to meters
|
|
medianval = medianval * masterTop.bursts[0].rangePixelSize
|
|
meanval = meanval * masterTop.bursts[0].rangePixelSize
|
|
stdval = stdval * masterTop.bursts[0].rangePixelSize
|
|
|
|
hist, bins = np.histogram(val, 50, normed=1)
|
|
center = 0.5*(bins[:-1] + bins[1:])
|
|
|
|
outputDir = os.path.dirname(inps.output)
|
|
os.makedirs(outputDir, exist_ok=True)
|
|
|
|
try:
|
|
import matplotlib as mpl
|
|
mpl.use('Agg')
|
|
import matplotlib.pyplot as plt
|
|
except:
|
|
print('Matplotlib could not be imported. Skipping debug plot ...')
|
|
debugPlot = False
|
|
|
|
debugPlot = False
|
|
if debugPlot:
|
|
|
|
try:
|
|
####Plotting
|
|
plt.figure()
|
|
plt.bar(center, hist, align='center', width = 0.7*(bins[1] - bins[0]))
|
|
plt.xlabel('Range shift in pixels')
|
|
plt.savefig( os.path.join(outputDir, 'rangeMisregistration.jpg'))
|
|
plt.show()
|
|
plt.close()
|
|
except:
|
|
print('Looks like matplotlib could not save image to JPEG, continuing .....')
|
|
print('Install Pillow to ensure debug plots for Residual range offsets are generated.')
|
|
pass
|
|
|
|
with open(inps.output, 'w') as f:
|
|
f.write('median : '+str(medianval) +'\n')
|
|
f.write('mean : '+str(meanval)+'\n')
|
|
f.write('std : '+str(stdval)+'\n')
|
|
f.write('snr threshold : '+str(inps.offsetSNRThreshold)+'\n')
|
|
f.write('mumber of coherent points : '+str(len(val))+'\n')
|
|
|
|
if __name__ == '__main__':
|
|
'''
|
|
Main driver.
|
|
'''
|
|
main()
|
|
|
|
|
|
|