def run(self):
# index = 0
arrmat = np.empty((1, 8))
# #Check OS and dep
# if sys.platform == 'darwin':
# gdalwarp_os_dep = '/Library/Frameworks/GDAL.framework/Versions/Current/Programs/gdalwarp'
# else:
# gdalwarp_os_dep = 'gdalwarp'
ret = 0
imp_point = 0
# Allt arbete en trad ska utforas maste goras i en try-sats
try:
# j = 0
# Loop som utfor det arbete som annars hade "hangt" anvandargranssnittet i Qgis
pre = self.dlg.textOutput_prefix.text()
header = ' Wd pai fai zH zHmax zHstd zd z0'
numformat = '%3d %4.3f %4.3f %5.3f %5.3f %5.3f %5.3f %5.3f'
header2 = ' id pai fai zH zHmax zHstd zd z0'
numformat2 = '%3d %4.3f %4.3f %5.3f %5.3f %5.3f %5.3f %5.3f'
# temporary fix for mac, ISSUE #15
pf = sys.platform
if pf == 'darwin' or pf == 'linux2':
if not os.path.exists(self.folderPath[0] + '/' + pre):
os.makedirs(self.folderPath[0] + '/' + pre)
for f in self.vlayer.getFeatures(
): # looping through each grid polygon
# Kollar sa att traden inte har avbrutits, ifall den har det sa slutar loopning.
if self.killed is True:
break
# pydevd.settrace('localhost', port=53100, stdoutToServer=True, stderrToServer=True) #used for debugging
attributes = f.attributes()
geometry = f.geometry()
feature = QgsFeature()
feature.setAttributes(attributes)
feature.setGeometry(geometry)
if self.imid == 1: # use center point
r = self.radius
y = f.geometry().centroid().asPoint().y()
x = f.geometry().centroid().asPoint().x()
else:
r = 0 # Uses as info to separate from IMP point to grid
writer = QgsVectorFileWriter(self.dir_poly, "CP1250",
self.fields,
self.prov.geometryType(),
self.prov.crs(),
"ESRI shapefile")
if writer.hasError() != QgsVectorFileWriter.NoError:
self.iface.messageBar().pushMessage(
"Error when creating shapefile: ",
str(writer.hasError()))
writer.addFeature(feature)
del writer
if self.dlg.checkBoxOnlyBuilding.isChecked(
): # Only building heights
provider = self.dsm_build.dataProvider()
filePath_dsm_build = str(provider.dataSourceUri())
if self.imid == 1:
bbox = (x - r, y + r, x + r, y - r)
# gdalruntextdsm_build = gdalwarp_os_dep + ' -dstnodata -9999 -q -overwrite -te ' + str(x - r) + ' ' + str(y - r) + \
# ' ' + str(x + r) + ' ' + str(y + r) + ' -of GTiff "' + \
# filePath_dsm_build + '" "' + self.plugin_dir + '/data/clipdsm.tif"'
else:
# Remove gdalwarp cuttoline with gdal.Translate. Cut to envelope of polygon feature
VectorDriver = ogr.GetDriverByName("ESRI Shapefile")
Vector = VectorDriver.Open(self.dir_poly, 0)
layer = Vector.GetLayer()
feature = layer.GetFeature(0)
geom = feature.GetGeometryRef()
minX, maxX, minY, maxY = geom.GetEnvelope()
bbox = (minX, maxY, maxX, minY
) # Reorder bbox to use with gdal_translate
Vector.Destroy()
# gdalruntextdsm_build = gdalwarp_os_dep + ' -dstnodata -9999 -q -overwrite -cutline ' + self.dir_poly + \
# ' -crop_to_cutline -of GTiff "' + filePath_dsm_build + '" "' + \
# self.plugin_dir + '/data/clipdsm.tif"'
# if sys.platform == 'win32':
# si = subprocess.STARTUPINFO()
# si.dwFlags |= subprocess.STARTF_USESHOWWINDOW
# subprocess.call(gdalruntextdsm_build, startupinfo=si)
# else:
# os.system(gdalruntextdsm_build)
# Remove gdalwarp with gdal.Translate
bigraster = gdal.Open(filePath_dsm_build)
gdal.Translate(self.plugin_dir + '/data/clipdsm.tif',
bigraster,
projWin=bbox)
bigraster = None
# os.system(gdalruntextdsm_build)
dataset = gdal.Open(self.plugin_dir + '/data/clipdsm.tif')
dsm_array = dataset.ReadAsArray().astype(np.float)
sizex = dsm_array.shape[0]
sizey = dsm_array.shape[1]
dem_array = np.zeros((sizex, sizey))
else: # Both building ground heights
provider = self.dsm.dataProvider()
filePath_dsm = str(provider.dataSourceUri())
provider = self.dem.dataProvider()
filePath_dem = str(provider.dataSourceUri())
# # get raster source - gdalwarp
if self.imid == 1:
bbox = (x - r, y + r, x + r, y - r)
# gdalruntextdsm = gdalwarp_os_dep + ' -dstnodata -9999 -q -overwrite -te ' + str(x - r) + ' ' + str(y - r) + \
# ' ' + str(x + r) + ' ' + str(y + r) + ' -of GTiff "' + \
# filePath_dsm + '" "' + self.plugin_dir + '/data/clipdsm.tif"'
# gdalruntextdem = gdalwarp_os_dep + ' -dstnodata -9999 -q -overwrite -te ' + str(x - r) + ' ' + str(y - r) + \
# ' ' + str(x + r) + ' ' + str(y + r) + ' -of GTiff "' + \
# filePath_dem + '" "' + self.plugin_dir + '/data/clipdem.tif"'
else:
# Remove gdalwarp cuttoline with gdal.Translate. Cut to envelope of polygon feature
VectorDriver = ogr.GetDriverByName("ESRI Shapefile")
Vector = VectorDriver.Open(self.dir_poly, 0)
layer = Vector.GetLayer()
feature = layer.GetFeature(0)
geom = feature.GetGeometryRef()
minX, maxX, minY, maxY = geom.GetEnvelope()
bbox = (minX, maxY, maxX, minY
) # Reorder bbox to use with gdal_translate
Vector.Destroy()
# gdalruntextdsm = gdalwarp_os_dep + ' -dstnodata -9999 -q -overwrite -cutline ' + self.dir_poly + \
# ' -crop_to_cutline -of GTiff "' + filePath_dsm + \
# '" "' + self.plugin_dir + '/data/clipdsm.tif"'
# gdalruntextdem = gdalwarp_os_dep + ' -dstnodata -9999 -q -overwrite -cutline ' + self.dir_poly + \
# ' -crop_to_cutline -of GTiff "' + filePath_dem + \
# '" "' + self.plugin_dir + '/data/clipdem.tif"'
# if sys.platform == 'win32':
# si = subprocess.STARTUPINFO()
# si.dwFlags |= subprocess.STARTF_USESHOWWINDOW
# subprocess.call(gdalruntextdsm, startupinfo=si)
# subprocess.call(gdalruntextdem, startupinfo=si)
# else:
# os.system(gdalruntextdsm)
# os.system(gdalruntextdem)
# Remove gdalwarp with gdal.Translate
bigraster = gdal.Open(filePath_dsm)
gdal.Translate(self.plugin_dir + '/data/clipdsm.tif',
bigraster,
projWin=bbox)
bigraster = None
bigraster = gdal.Open(filePath_dem)
gdal.Translate(self.plugin_dir + '/data/clipdem.tif',
bigraster,
projWin=bbox)
bigraster = None
time.sleep(0.05)
dataset = gdal.Open(self.plugin_dir + '/data/clipdsm.tif')
dsm_array = dataset.ReadAsArray().astype(np.float)
dataset2 = gdal.Open(self.plugin_dir + '/data/clipdem.tif')
dem_array = dataset2.ReadAsArray().astype(np.float)
if not (dsm_array.shape[0] == dem_array.shape[0]) & (
dsm_array.shape[1] == dem_array.shape[1]):
QMessageBox.critical(
None, "Error",
"All grids must be of same pixel resolution")
return
geotransform = dataset.GetGeoTransform()
scale = 1 / geotransform[1]
nd = dataset.GetRasterBand(1).GetNoDataValue()
nodata_test = (dsm_array == nd)
if self.dlg.checkBoxNoData.isChecked():
if np.sum(dsm_array) == (dsm_array.shape[0] *
dsm_array.shape[1] * nd):
QgsMessageLog.logMessage(
"Grid " + str(f.attributes()[self.idx]) +
" not calculated. Includes Only NoData Pixels",
level=QgsMessageLog.CRITICAL)
cal = 0
else:
dsm_array[dsm_array == nd] = np.mean(dem_array)
dem_array[dem_array == nd] = np.mean(dem_array)
cal = 1
else:
if nodata_test.any(): # == True
QgsMessageLog.logMessage(
"Grid " + str(f.attributes()[self.idx]) +
" not calculated. Includes NoData Pixels",
level=QgsMessageLog.CRITICAL)
cal = 0
else:
cal = 1
if cal == 1:
# arr = np.array([f.attributes()[self.idx], -99, -99, -99, -99, -99, -99, -99])
# arrmat = np.vstack([arrmat, arr])
# else:
immorphresult = imagemorphparam_v2(dsm_array, dem_array,
scale, self.imid,
self.degree, self.dlg,
imp_point)
zH = immorphresult["zH"]
fai = immorphresult["fai"]
pai = immorphresult["pai"]
zMax = immorphresult["zHmax"]
zSdev = immorphresult["zH_sd"]
zd, z0 = rg.RoughnessCalcMany(self.rm, zH, fai, pai, zMax,
zSdev)
# save to file
# header = ' Wd pai fai zH zHmax zHstd zd z0'
# numformat = '%3d %4.3f %4.3f %5.3f %5.3f %5.3f %5.3f %5.3f'
arr = np.concatenate(
(immorphresult["deg"], immorphresult["pai"],
immorphresult["fai"], immorphresult["zH"],
immorphresult["zHmax"], immorphresult["zH_sd"], zd,
z0),
axis=1)
np.savetxt(self.folderPath[0] + '/' + pre + '_' +
'IMPGrid_anisotropic_' +
str(f.attributes()[self.idx]) + '.txt',
arr,
fmt=numformat,
delimiter=' ',
header=header,
comments='')
del arr
zHall = immorphresult["zH_all"]
faiall = immorphresult["fai_all"]
paiall = immorphresult["pai_all"]
zMaxall = immorphresult["zHmax_all"]
zSdevall = immorphresult["zH_sd_all"]
zdall, z0all = rg.RoughnessCalc(self.rm, zHall, faiall,
paiall, zMaxall, zSdevall)
# If zd and z0 are lower than open country, set to open country
if zdall == 0.0:
zdall = 0.1
if z0all == 0.0:
z0all = 0.03
arr2 = np.array([[
f.attributes()[self.idx], immorphresult["pai_all"],
immorphresult["fai_all"], immorphresult["zH_all"],
immorphresult["zHmax_all"], immorphresult["zH_sd_all"],
zdall, z0all
]])
arrmat = np.vstack([arrmat, arr2])
dataset = None
dataset2 = None
dataset3 = None
self.progress.emit()
# header2 = ' id pai fai zH zHmax zHstd zd z0'
# numformat2 = '%3d %4.3f %4.3f %5.3f %5.3f %5.3f %5.3f %5.3f'
arrmatsave = arrmat[1:arrmat.shape[0], :]
np.savetxt(self.folderPath[0] + '/' + pre + '_' +
'IMPGrid_isotropic.txt',
arrmatsave,
fmt=numformat2,
delimiter=' ',
header=header2,
comments='')
if self.dlg.addResultToGrid.isChecked():
self.addattributes(self.vlayer, arrmatsave, header, pre)
# Nas om hela loopen utforts, kan anvandas for att tilldela ret-variabeln resultatet av arbetet som ska
# ska skickas tillbaka till image_morph_param.py
if self.killed is False:
self.progress.emit()
ret = 1
except Exception:
# forward the exception upstream
ret = 0
errorstring = self.print_exception()
#self.error.emit(e, traceback.format_exc())
self.error.emit(errorstring)
self.finished.emit(ret)
