96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
#
|
|
# Author: Piyush Agram
|
|
# Copyright 2016
|
|
#
|
|
|
|
import logging
|
|
import isceobj
|
|
import mroipac
|
|
import os
|
|
import numpy as np
|
|
from isceobj.Util.decorators import use_api
|
|
|
|
logger = logging.getLogger('isce.insar.VerifyDEM')
|
|
|
|
def runVerifyDEM(self):
|
|
'''
|
|
Make sure that a DEM is available for processing the given data.
|
|
'''
|
|
|
|
self.demStitcher.noFilling = False
|
|
|
|
###If provided in the input XML file
|
|
if self.demFilename not in ['',None]:
|
|
demimg = isceobj.createDemImage()
|
|
demimg.load(self.demFilename + '.xml')
|
|
if not os.path.exists(self.demFilename + '.vrt'):
|
|
demimg.renderVRT()
|
|
|
|
if demimg.reference.upper() == 'EGM96':
|
|
wgsdemname = self.demFilename + '.wgs84'
|
|
|
|
if os.path.exists(wgsdemname) and os.path.exists(wgsdemname + '.xml'):
|
|
demimg = isceobj.createDemImage()
|
|
demimg.load(wgsdemname + '.xml')
|
|
|
|
if demimg.reference.upper() == 'EGM96':
|
|
raise Exception('WGS84 version of dem found by reference set to EGM96')
|
|
|
|
else:
|
|
demimg = self.demStitcher.correct(demimg)
|
|
|
|
elif demimg.reference.upper() != 'WGS84':
|
|
raise Exception('Unknown reference system for DEM: {0}'.format(demimg.reference))
|
|
|
|
else:
|
|
|
|
refPol = self._grd.polarizations[0]
|
|
|
|
master = self._grd.loadProduct( os.path.join(self._grd.outputFolder, 'beta_{0}.xml'.format(refPol)))
|
|
bbox = master.getBbox()
|
|
|
|
####Truncate to integers
|
|
tbox = [np.floor(bbox[0]), np.ceil(bbox[1]),
|
|
np.floor(bbox[2]), np.ceil(bbox[3])]
|
|
|
|
|
|
filename = self.demStitcher.defaultName(tbox)
|
|
wgsfilename = filename + '.wgs84'
|
|
|
|
####Check if WGS84 file exists
|
|
if os.path.exists(wgsfilename) and os.path.exists(wgsfilename + '.xml'):
|
|
demimg = isceobj.createDemImage()
|
|
demimg.load(wgsfilename + '.xml')
|
|
|
|
if not os.path.exists(wgsfilename + '.vrt'):
|
|
demimg.renderVRT()
|
|
|
|
####Check if EGM96 file exists
|
|
elif os.path.exists(filename) and os.path.exists(filename + '.xml'):
|
|
inimg = isceobj.createDemImage()
|
|
inimg.load(filename + '.xml')
|
|
|
|
if not os.path.exists(filename + '.xml'):
|
|
inimg.renderVRT()
|
|
|
|
demimg = self.demStitcher.correct(inimg)
|
|
|
|
else:
|
|
stitchOk = self.demStitcher.stitch(tbox[0:2], tbox[2:4])
|
|
|
|
if not stitchOk:
|
|
logger.error("Cannot form the DEM for the region of interest. If you have one, set the appropriate DEM component in the input file.")
|
|
raise Exception
|
|
|
|
inimg = isceobj.createDemImage()
|
|
inimg.load(filename + '.xml')
|
|
if not os.path.exists(filename):
|
|
inimg.renderVRT()
|
|
|
|
demimg = self.demStitcher.correct(inimg)
|
|
|
|
#get water mask
|
|
# self.runCreateWbdMask(info)
|
|
|
|
return demimg.filename
|