180 lines
6.4 KiB
Python
Executable File
180 lines
6.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
|
||
#Author: Heresh Fattahi
|
||
|
||
import os
|
||
import argparse
|
||
import configparser
|
||
from tkinter.messagebox import NO
|
||
import numpy as np
|
||
import isce
|
||
import isceobj
|
||
from iscesys.DataManager import createManager
|
||
from contrib.demUtils.SWBDStitcher import SWBDStitcher
|
||
|
||
### 扩展现有的方法
|
||
from osgeo import gdal
|
||
###
|
||
|
||
|
||
EXAMPLE = """example:
|
||
createWaterMask.py -b 31 33 130 132
|
||
createWaterMask.py -b 31 33 130 132 -l lat.rdr -L lon.rdr -o waterMask.rdr
|
||
createWaterMask.py -d ../DEM/demLat_N31_N33_Lon_E130_E132.dem.wgs84 -l lat.rdr -L lon.rdr -o waterMask.rdr
|
||
"""
|
||
|
||
def createParser():
|
||
'''
|
||
Create command line parser.
|
||
'''
|
||
|
||
parser = argparse.ArgumentParser(description='Create water body mask in geo and/or radar coordinates',
|
||
formatter_class=argparse.RawTextHelpFormatter,
|
||
epilog=EXAMPLE)
|
||
parser.add_argument('-b', '--bbox', dest='bbox', type=int, default=None, nargs=4, metavar=('S','N','W','E'),
|
||
help = 'Defines the spatial region in the format south north west east.\n'
|
||
'The values should be integers from (-90,90) for latitudes '
|
||
'and (0,360) or (-180,180) for longitudes.')
|
||
parser.add_argument('-d','--dem_file', dest='demName', type=str, default=None,
|
||
help='DEM file in geo coordinates, i.e. demLat*.dem.wgs84.')
|
||
|
||
parser.add_argument('-l', '--lat_file', dest='latName', type=str, default=None,
|
||
help='pixel by pixel lat file in radar coordinate')
|
||
parser.add_argument('-L', '--lon_file', dest='lonName', type=str, default=None,
|
||
help='pixel by pixel lat file in radar coordinate')
|
||
|
||
parser.add_argument('--fill', dest='fillValue', type=int, default=-1, choices={-1,0},
|
||
help='fill value for pixels with missing data. Default: -1.\n'
|
||
'-1 for water body\n'
|
||
' 0 for land')
|
||
parser.add_argument('-o', '--output', dest='outfile', type=str,
|
||
help='output filename of water mask in radar coordinates')
|
||
return parser
|
||
|
||
|
||
def cmdLineParse(iargs = None):
|
||
'''
|
||
Command line parser.
|
||
'''
|
||
|
||
parser = createParser()
|
||
inps = parser.parse_args(args=iargs)
|
||
|
||
if not inps.bbox and not inps.demName:
|
||
parser.print_usage()
|
||
raise SystemExit('ERROR: no --bbox/--dem_file input, at least one is required.')
|
||
|
||
if not inps.outfile and (inps.latName and inps.lonName):
|
||
inps.outfile = os.path.join(os.path.dirname(inps.latName), 'waterMask.rdr')
|
||
|
||
return inps
|
||
|
||
|
||
def dem2bbox(dem_file):
|
||
"""Grab bbox from DEM file in geo coordinates"""
|
||
demImage = isceobj.createDemImage()
|
||
demImage.load(dem_file + '.xml')
|
||
demImage.setAccessMode('read')
|
||
N = demImage.getFirstLatitude()
|
||
W = demImage.getFirstLongitude()
|
||
S = N + demImage.getDeltaLatitude() * demImage.getLength()
|
||
E = W + demImage.getDeltaLongitude() * demImage.getWidth()
|
||
bbox = [np.floor(S).astype(int), np.ceil(N).astype(int),
|
||
np.floor(W).astype(int), np.ceil(E).astype(int)]
|
||
return bbox
|
||
|
||
|
||
def copydem_watermask(dem,out_water):
|
||
|
||
pass
|
||
|
||
|
||
def download_waterMask(bbox, dem_file, fill_value=-1):
|
||
out_dir = os.getcwd()
|
||
# update out_dir and/or bbox if dem_file is input
|
||
if dem_file:
|
||
out_dir = os.path.dirname(dem_file)
|
||
if not bbox:
|
||
bbox = dem2bbox(dem_file)
|
||
print('bounding box in (S, N, W, E): {}'.format(bbox))
|
||
|
||
sw = createManager('wbd')
|
||
sw.configure()
|
||
#inps.waterBodyGeo = sw.defaultName(inps.bbox)
|
||
sw.outputFile = os.path.join(out_dir, sw.defaultName(bbox))
|
||
# 创建waterMask ---- 代码废弃
|
||
|
||
# dem=gdal.Open(dem_file) # 根据输入dem 创建wbd. [0,255] 0: 不是水域，255 水域
|
||
# row_count = dem.RasterYSize # 读取图像的行数
|
||
# col_count = dem.RasterXSize # 读取图像的列数
|
||
# im_proj = dem.GetProjection()
|
||
# im_geotrans = list(dem.GetGeoTransform())
|
||
# gt = im_geotrans # 坐标变换参数
|
||
# gtiff_driver = gdal.GetDriverByName('GTiff')
|
||
# swdb=gtiff_driver.Create(sw.outputFile,col_count,row_count,1,gdal.GDT_Byte)
|
||
# swdb.SetGeoTransform(im_geotrans) # 写入仿射变换参数
|
||
# swdb.SetProjection(im_proj) # 写入投影
|
||
# # 写入tiff
|
||
# sbd=np.zeros((row_count,col_count),dtype=np.uint8)
|
||
# swdb.GetRasterBand(1).WriteArray(sbd)
|
||
# swdb.FlushCache()
|
||
# del swdb # 删除影像
|
||
print("=================================")
|
||
print("isce2/contrib/stack/stripmapStack/createWaterMask.py","代码已经修改")
|
||
print("================================")
|
||
|
||
if os.path.isfile(sw.outputFile):
|
||
print('wbd file already exists at: {}'.format(sw.outputFile))
|
||
print('skip re-downloading and continue')
|
||
return None
|
||
else:
|
||
print("========================")
|
||
print("输出结果：None")
|
||
print("==========================")
|
||
return None
|
||
sw._noFilling = False
|
||
sw._fillingValue = fill_value
|
||
sw.stitch(bbox[0:2], bbox[2:]) # 下载对应的影像
|
||
return sw.outputFile
|
||
|
||
|
||
def geo2radar(geo_file, rdr_file, lat_file, lon_file):
|
||
#inps.waterBodyRadar = inps.waterBodyGeo + '.rdr'
|
||
print('converting water mask file to radar coordinates ...')
|
||
sw = SWBDStitcher()
|
||
sw.toRadar(geo_file, lat_file, lon_file, rdr_file)
|
||
return rdr_file
|
||
|
||
def geo2radarUsr(rdr_file, lat_file, lon_file):
|
||
#inps.waterBodyRadar = inps.waterBodyGeo + '.rdr'
|
||
print('converting water mask file to radar coordinates ...')
|
||
sw = SWBDStitcher()
|
||
sw.toRadarNone( lat_file, lon_file, rdr_file)
|
||
return rdr_file
|
||
#looks.py -i watermask.msk -r 4 -a 14 -o 'waterMask.14alks_4rlks.msk'
|
||
|
||
#imageMath.py -e='a*b' --a=filt_20100911_20101027.int --b=watermask.14alks_4rlks.msk -o filt_20100911_20101027_masked.int -t cfloat -s BIL
|
||
|
||
def main(iargs=None):
|
||
print("============================")
|
||
print("当前影像")
|
||
print("================================")
|
||
inps = cmdLineParse(iargs)
|
||
geo_file = download_waterMask(inps.bbox, inps.demName, inps.fillValue)
|
||
if inps.latName and inps.lonName:
|
||
if geo_file is None:
|
||
geo2radarUsr( inps.outfile, inps.latName, inps.lonName)
|
||
else:
|
||
geo2radar(geo_file, inps.outfile, inps.latName, inps.lonName)
|
||
return
|
||
|
||
|
||
if __name__ == '__main__' :
|
||
'''
|
||
creates a water mask and transforms to radar Coord
|
||
'''
|
||
main()
|
||
|
||
|
||
|