class RCMRD_LandDegrDialogTest(unittest.TestCase):
"""Test dialog works."""
def setUp(self):
"""Runs before each test."""
self.dialog = RCMRD_LandDegrDialog(None)
def tearDown(self):
"""Runs after each test."""
self.dialog = None
def test_dialog_ok(self):
"""Test we can click OK."""
button = self.dialog.button_box.button(QDialogButtonBox.Ok)
button.click()
result = self.dialog.result()
self.assertEqual(result, QDialog.Accepted)
def test_dialog_cancel(self):
"""Test we can click cancel."""
button = self.dialog.button_box.button(QDialogButtonBox.Cancel)
button.click()
result = self.dialog.result()
self.assertEqual(result, QDialog.Rejected)
def __init__(self, iface):
"""Constructor.
:param iface: An interface instance that will be passed to this class
which provides the hook by which you can manipulate the QGIS
application at run time.
:type iface: QgsInterface
"""
# Save reference to the QGIS interface
self.iface = iface
# initialize plugin directory
self.plugin_dir = os.path.dirname(__file__)
# initialize locale
locale = QSettings().value('locale/userLocale')[0:2]
locale_path = os.path.join(
self.plugin_dir,
'i18n',
'RCMRD_LandDegr_{}.qm'.format(locale))
if os.path.exists(locale_path):
self.translator = QTranslator()
self.translator.load(locale_path)
if qVersion() > '4.3.3':
QCoreApplication.installTranslator(self.translator)
# Create the dialog (after translation) and keep reference
self.dlg = RCMRD_LandDegrDialog()
# Declare instance attributes
self.actions = []
self.menu = self.tr(u'&RCRMD: land degradation')
# TODO: We are going to let the user set this up in a future iteration
self.toolbar = self.iface.addToolBar(u'RCMRD: land degradation')
self.toolbar.setObjectName(u'RCMRD: land degradation')
# Declare user instance variables
self.roiDefinitions = None
self.roiDefinitions = [{"name":"IGAD", "roiXY":[21.8094, -4.6775, 51.417, 22.227]},
{"name":"Djibouti", "roiXY":[41.749110148,10.929824931, 43.418711785,12.707912502]},
{"name":"Eritrea", "roiXY":[36.423647095,12.360021871, 43.123871290,18.004828192]},
{"name":"Ethiopia", "roiXY":[32.989799845,3.403333435, 47.979169149,14.879532166]},
{"name":"Kenya", "roiXY":[33.890468384,-4.677504165, 41.885019165,5.030375823]},
{"name":"Sudan", "roiXY":[42.647246541,7.996515605, 48.93911199991,11.498928127]},
{"name":"South Sudan", "roiXY":[24.121555623,3.490201518, 35.920835409,12.216154684]},
{"name":"Somali Land", "roiXY":[42.647246541,7.996515605, 48.93911199991,11.498928127]},
{"name":"Somalia", "roiXY":[40.965385376,-1.69628316498, 51.417037811,11.989118646]},
{"name":"Uganda", "roiXY":[29.548459513,-1.475205994, 35.006472615,4.219691875]}]
#self.selectedRoi = None
self.raster_list = [] # list all files open in QGis, contains file objects
self.dictReproj = None # dictionary: 'input filename' -> reprojected filename
self.clipLayer = None # store the clipping vector layer
self.listIDInputs={} # dictionary: key -> layer name. See keys definition in the code ('VGT', 'RFE', ...)
self.listIDWeightsPotential={}
self.listIDWeightsActual={}
SettingsOrganisation='RCMRD_QGIS'
SettingsApplication='RCMRD_LandDegr'
class RCMRD_LandDegr:
"""QGIS Plugin Implementation."""
def __init__(self, iface):
"""Constructor.
:param iface: An interface instance that will be passed to this class
which provides the hook by which you can manipulate the QGIS
application at run time.
:type iface: QgsInterface
"""
# Save reference to the QGIS interface
self.iface = iface
# initialize plugin directory
self.plugin_dir = os.path.dirname(__file__)
# initialize locale
locale = QSettings().value('locale/userLocale')[0:2]
locale_path = os.path.join(
self.plugin_dir,
'i18n',
'RCMRD_LandDegr_{}.qm'.format(locale))
if os.path.exists(locale_path):
self.translator = QTranslator()
self.translator.load(locale_path)
if qVersion() > '4.3.3':
QCoreApplication.installTranslator(self.translator)
# Create the dialog (after translation) and keep reference
self.dlg = RCMRD_LandDegrDialog()
# Declare instance attributes
self.actions = []
self.menu = self.tr(u'&RCRMD: land degradation')
# TODO: We are going to let the user set this up in a future iteration
self.toolbar = self.iface.addToolBar(u'RCMRD: land degradation')
self.toolbar.setObjectName(u'RCMRD: land degradation')
# Declare user instance variables
self.roiDefinitions = None
self.roiDefinitions = [{"name":"IGAD", "roiXY":[21.8094, -4.6775, 51.417, 22.227]},
{"name":"Djibouti", "roiXY":[41.749110148,10.929824931, 43.418711785,12.707912502]},
{"name":"Eritrea", "roiXY":[36.423647095,12.360021871, 43.123871290,18.004828192]},
{"name":"Ethiopia", "roiXY":[32.989799845,3.403333435, 47.979169149,14.879532166]},
{"name":"Kenya", "roiXY":[33.890468384,-4.677504165, 41.885019165,5.030375823]},
{"name":"Sudan", "roiXY":[42.647246541,7.996515605, 48.93911199991,11.498928127]},
{"name":"South Sudan", "roiXY":[24.121555623,3.490201518, 35.920835409,12.216154684]},
{"name":"Somali Land", "roiXY":[42.647246541,7.996515605, 48.93911199991,11.498928127]},
{"name":"Somalia", "roiXY":[40.965385376,-1.69628316498, 51.417037811,11.989118646]},
{"name":"Uganda", "roiXY":[29.548459513,-1.475205994, 35.006472615,4.219691875]}]
#self.selectedRoi = None
self.raster_list = [] # list all files open in QGis, contains file objects
self.dictReproj = None # dictionary: 'input filename' -> reprojected filename
self.clipLayer = None # store the clipping vector layer
self.listIDInputs={} # dictionary: key -> layer name. See keys definition in the code ('VGT', 'RFE', ...)
self.listIDWeightsPotential={}
self.listIDWeightsActual={}
SettingsOrganisation='RCMRD_QGIS'
SettingsApplication='RCMRD_LandDegr'
# noinspection PyMethodMayBeStatic
def tr(self, message):
"""Get the translation for a string using Qt translation API.
We implement this ourselves since we do not inherit QObject.
:param message: String for translation.
:type message: str, QString
:returns: Translated version of message.
:rtype: QString
"""
# noinspection PyTypeChecker,PyArgumentList,PyCallByClass
return QCoreApplication.translate('RCMRD_LandDegr', message)
def add_action(
self,
icon_path,
text,
callback,
enabled_flag=True,
add_to_menu=True,
add_to_toolbar=True,
status_tip=None,
whats_this=None,
parent=None):
"""Add a toolbar icon to the toolbar.
:param icon_path: Path to the icon for this action. Can be a resource
path (e.g. ':/plugins/foo/bar.png') or a normal file system path.
:type icon_path: str
:param text: Text that should be shown in menu items for this action.
:type text: str
:param callback: Function to be called when the action is triggered.
:type callback: function
:param enabled_flag: A flag indicating if the action should be enabled
by default. Defaults to True.
:type enabled_flag: bool
:param add_to_menu: Flag indicating whether the action should also
be added to the menu. Defaults to True.
:type add_to_menu: bool
:param add_to_toolbar: Flag indicating whether the action should also
be added to the toolbar. Defaults to True.
:type add_to_toolbar: bool
:param status_tip: Optional text to show in a popup when mouse pointer
hovers over the action.
:type status_tip: str
:param parent: Parent widget for the new action. Defaults None.
:type parent: QWidget
:param whats_this: Optional text to show in the status bar when the
mouse pointer hovers over the action.
:returns: The action that was created. Note that the action is also
added to self.actions list.
:rtype: QAction
"""
icon = QIcon(icon_path)
action = QAction(icon, text, parent)
action.triggered.connect(callback)
action.setEnabled(enabled_flag)
if status_tip is not None:
action.setStatusTip(status_tip)
if whats_this is not None:
action.setWhatsThis(whats_this)
if add_to_toolbar:
self.toolbar.addAction(action)
if add_to_menu:
self.iface.addPluginToMenu(
self.menu,
action)
self.actions.append(action)
return action
def initGui(self):
"""Create the menu entries and toolbar icons inside the QGIS GUI."""
icon_path = ':/plugins/RCMRD_LandDegr/icon.png'
self.add_action(
icon_path,
text=self.tr(u'Land Degradation Indices'),
callback=self.run,
whats_this=self.tr(u'Compute RCMRD LDIM'),
parent=self.iface.mainWindow())
def unload(self):
"""Removes the plugin menu item and icon from QGIS GUI."""
for action in self.actions:
self.iface.removePluginMenu(
self.tr(u'&Land Degradation (RCMRD)'),
action)
self.iface.removeToolBarIcon(action)
# remove the toolbar
del self.toolbar
# ____________________
def logMsg(self, msg, errorLvl = QgsMessageLog.INFO):
QgsMessageLog.logMessage( msg, tag='RCMRD Land Degradation', level=errorLvl)
prepend=''
if errorLvl==QgsMessageLog.WARNING:
self.iface.messageBar().pushMessage("WARNING", msg)
prepend="Warning! "
if errorLvl==QgsMessageLog.CRITICAL:
self.iface.messageBar().pushMessage("CRITICAL",msg)
prepend="Critical error! "
self.dlg.logTextDump.append(prepend + msg)
# ____________________
# Business logic functions and methods
# ____________________
# returns uniq of an array, preserve order
def uniquify(self, seq):
noDupes = []
[noDupes.append(i) for i in seq if not noDupes.count(i)]
return noDupes
# ____________________
# Get target extent, in geographical coordinates
#def getTE(self):
# xmin=float(self.dlg.RoiWestEdit.text())
# xmax=float(self.dlg.RoiEastEdit.text())
# ymin=float(self.dlg.RoiSouthEdit.text())
# ymax=float(self.dlg.RoiNorthEdit.text())
# return [xmin, ymin, xmax, ymax]
# _____________________
def ParseType(self, type):
if type == 'Byte':
return GDT_Byte
elif type == 'Int16':
return GDT_Int16
elif type == 'UInt16':
return GDT_UInt16
elif type == 'Int32':
return GDT_Int32
elif type == 'UInt32':
return GDT_UInt32
elif type == 'Float32':
return GDT_Float32
elif type == 'Float64':
return GDT_Float64
elif type == 'CInt16':
return GDT_CInt16
elif type == 'CInt32':
return GDT_CInt32
elif type == 'CFloat32':
return GDT_CFloat32
elif type == 'CFloat64':
return GDT_CFloat64
else:
return GDT_Float32
# _____________________
# doInitDir: if the directory does not exist: create it
def doInitDir(self, thisDir):
if os.path.isdir(thisDir):
return True
os.makedirs(thisDir)
# _____________________
def doTempFile(self, prefix='', suffix='', dir=''):
idTag = random.randint(10000, 99999)
return os.path.join(dir, prefix + '_' + str(idTag) + suffix)
# ____________________
# search for a layer name, return filename
def retrieveFromName(self, name):
for ii in self.raster_list:
if ii.name()==name:
return ii.source()
return False
# ____________________
# return an array of input files (file paths)
# For each type of file (listed in listIDInputs), tries macthing corresponding name from rasters_layer
# note: do not trust sorting order
def getInputFiles(self):
inFiles=[]
for kk, ii in self.listIDInputs.iteritems():
thisFile = self.retrieveFromName( ii )
if thisFile==False:
self.logMsg("Could no retrieve file {} associated to key {}".format( ii, kk ))
return False
else:
inFiles.append(thisFile)
self.dictInput[ii]=thisFile
return inFiles
# _____________________
# doResample: resample the inputs to a common projection and resolution, will crop images
# images saved into the working directory
# note: to read documentation about an algorithm: import processing; processing.alghelp("gdalogr:warpreproject")
def doResample(self, inFileName, output):
self.doInitDir( self.dlg.editWrkDir.text() )
# for now, dst is hard coded
#inCRS = QgsCoordinateReferenceSystem(4326) # actually, must be read from the input file
newCRS = QgsCoordinateReferenceSystem()
#newCRS.createFromProj4('+proj=aea +lat_1=18 +lat_2=-4 +lat_0=11 +lon_0=25 +x_0=37.5 +y_0=11')
newCRS.createFromSrid(4326)
if not newCRS.isValid():
self.logMsg("The projection definition is not valid. Exit", QgsMessageLog.CRITICAL)
return False
# see: https://docs.qgis.org/2.6/en/docs/user_manual/processing_algs/gdalogr/gdal_projections/warpreproject.html
# inFileName = self.raster_list[self.dlg.comboVegetationIndex.currentIndex()].source()
inFID = gdal.Open(inFileName, GA_ReadOnly)
inCRS = inFID.GetProjection()
inTrans = inFID.GetGeoTransform()
inDataType = inFID.GetRasterBand(1).DataType
# for now, we only resample 1 dataset, to be placed in a loop later on
TR = self.dlg.spinTR_deg.value()
extraParam=''
if self.clipLayer is not None:
ext = self.clipLayer.extent()
bb = [ ext.xMinimum(), ext.yMinimum(), ext.xMaximum(), ext.yMaximum() ]
extraParam = '-dstnodata 0 -te {} {} {} {} -cutline "{}" -crop_to_cutline '.format(bb[0], bb[1], bb[2], bb[3], self.dlg.editClipShp.text() )
method = 0 # because we resample thematic layers
# rtype 0: Byte, 1: int16, 2: uint16, 3:uint32, 4: int32, 5: Float32, 6: Float54
# processing.runalg("gdalogr:warpreproject")
#ALGORITHM: Warp (reproject)
#INPUT <ParameterRaster>
#SOURCE_SRS <ParameterCrs>
#DEST_SRS <ParameterCrs>
#NO_DATA <ParameterString>
#TR <ParameterNumber>
#METHOD <ParameterSelection>
#RTYPE <ParameterSelection>
#COMPRESS <ParameterSelection>
#JPEGCOMPRESSION <ParameterNumber>
#ZLEVEL <ParameterNumber>
#PREDICTOR <ParameterNumber>
#TILED <ParameterBoolean>
#BIGTIFF <ParameterSelection>
#TFW <ParameterBoolean>
#EXTRA <ParameterString>
#OUTPUT <OutputRaster>
#METHOD(Resampling method): 0 - near, 1 - bilinear, 2 - cubic, 3 - cubicspline, 4 - lanczos
#RTYPE(Output raster type): 0 - Byte, 1 - Int16, 2 - UInt16, 3 - UInt32, 4 - Int32, 5 - Float32, 6 - Float64
#COMPRESS(GeoTIFF options. Compression type): 0 - NONE, 1 - JPEG, 2 - LZW, 3 - PACKBITS, 4 - DEFLATE
#BIGTIFF(Control whether the created file is a BigTIFF or a classic TIFF): 0 - , 1 - YES, 2 - NO, 3 - IF_NEEDED, 4 - IF_SAFER
testproc = processing.runalg('gdalogr:warpreproject',
inFileName, # input
inCRS, # source ss
newCRS.toProj4(), # dest srs
'', # no data, <parameterString>
TR, # target resolution: 0=unchanged
0, # method: thematic layers, use 0
0, # output raster type
2, # compression
None, # jpeg compression
None, # zlevel
None, # predictor
None, # tiled
None, # bigtiff
None, # TFW
extraParam, # extra
output)
if not testproc:
self.logMsg("Reprojection failed for file {}".format(inFileName))
return False
return True
# ____________________
def doIndices(self):
self.logMsg("In do indices")
# open reprojected files
fid={}
noData={}
for id, layerName in self.listIDInputs.iteritems():
thisFile = self.dictReproj[ self.retrieveFromName( self.listIDInputs[id] )]
fid[id] = gdal.Open(thisFile, GA_ReadOnly)
if fid[id] is None:
self.logMsg('Error: could not open file {}'.format(thisFile), QgsMessageLog.CRITICAL)
return False
noData[id]=fid[id].GetRasterBand(1).GetNoDataValue()
self.logMsg("dataset {}, no data is {}".format(id, noData[id]))
# create ouputs from info read in VGT
# this implies that all files were reprojected to the same footprint and resolution
ns = fid['VGT'].RasterXSize
nl = fid['VGT'].RasterYSize
projection = fid['VGT'].GetProjection()
geotrans = fid['VGT'].GetGeoTransform()
outDrv = gdal.GetDriverByName('GTiff')
outFID = {}
outFID['Actual'] = outDrv.Create( self.dlg.editOutALDI.text(), ns, nl, 1, GDT_Float32 )
outFID['Potential'] = outDrv.Create( self.dlg.editOutPLDI.text(), ns, nl, 1, GDT_Float32 )
# define outputs projections and geotransformations
for ii in ['Actual', 'Potential']:
outFID[ii].SetProjection(projection)
outFID[ii].SetGeoTransform(geotrans)
# infos
self.logMsg("Weights Actual")
for ii, ww in self.listIDWeightsActual.iteritems():
self.logMsg("IDWeightsActual[{}]={}".format(ii,ww))
self.logMsg("Weights Potential")
for ii, ww in self.listIDWeightsPotential.iteritems():
self.logMsg("IDWeightsPotential[{}]={}".format(ii,ww))
# compute indicators for each line
for il in range(nl):
data={}
outData = {}
for id in self.listIDInputs:
data[ id ] = numpy.ravel( fid[id].GetRasterBand(1).ReadAsArray(0,il, ns, 1) )
# let's compute actual LDIM
dataActual=numpy.zeros(ns)
for ii, ww in self.listIDWeightsActual.iteritems():
if noData[ii] is not None:
wdata = ( data[ii] != noData[ii] )
if wdata.any():
dataActual[wdata] += ww * data[ii][wdata]
else:
dataActual += ww * data[ii]
dataPotential = numpy.zeros(ns)
for ii, ww in self.listIDWeightsPotential.iteritems():
if noData[ii] is not None:
wdata = data[ii] != noData[ii]
if wdata.any():
dataPotential[wdata] += ww * data[ii][wdata]
else:
dataPotential += ww * data[ii]
# save lines
outFID['Actual'].GetRasterBand(1).WriteArray( dataActual.reshape(1,ns), 0, il)
outFID['Potential'].GetRasterBand(1).WriteArray( dataPotential.reshape(1,ns), 0, il)
return True
# ____________________
# Run business logic
# 1./ reproject/resample input files, save in tmp files
# 2./ compute indices, per pixel, using weights, save in output files
# Will exit on any error, reports for errors
def doProcessing(self):
# Inputs are reprojected and resampled
listInput = self.getInputFiles()
if listInput == False:
self.logMsg("Missing input files, processing impossible")
return False
for thisFName in listInput:
self.logMsg("Resampling and reprojection file {}".format(thisFName))
output = self.doTempFile('reproject', '.tif', self.dlg.editWrkDir.text() )
self.dictReproj[ thisFName ] = output
if self.doResample(thisFName, output) == False:
self.logMsg("Reprojection failed for {}".format(thisFName))
return False
# now compute indices
if self.doIndices() == False:
self.logMsg("Error in processing indices")
return False
return True
# ____________________
def doCheckToGo(self):
#if (self.selectedRoi) is None:
# self.logMsg("Please choose a region, in the 'Settings' tab.")
# self.dlg.logTextDump.append("Please choose a region, in the 'Settings' tab.")
# self.dlg.tabWidget.setCurrentWidget(self.dlg.logTextDump)
# return False
self.logMsg("Checking inputs")
# clipping vector is now mandatory
if self.dlg.editClipShp.text()=='':
self.logMsg("You must define a clipping vector (for example a shapefile)", QgsMessageLog.CRITICAL)
self.dlg.tabWidget.setCurrentWidget(self.dlg.logTextDump)
return False
# check all files are different
listFName=[ii.source() for ii in self.raster_list]
if self.uniquify(listFName) != listFName:
self.logMsg("Input files must be different. Please revise inputs in 'Input files' tab.", QgsMessageLog.CRITICAL)
tabWidget.setCurrentWidget(self.dlg.logTextDump)
self.listIDInputs={'VGT': self.dlg.comboVegetationIndex.currentText(),
'RFE':self.dlg.comboRainfallErosivity.currentText(),
'PopDens':self.dlg.comboPopDensity.currentText(),
'SoilErod':self.dlg.comboSoilErodibility.currentText(),
'SlopeLF':self.dlg.comboSlopeLF.currentText() }
self.listIDWeightsPotential={'RFE': self.dlg.spinPotRFE.value(),
'PopDens': self.dlg.spinPotPopDens.value(),
'SoilErod': self.dlg.spinPotSoilErod.value(),
'SlopeLF': self.dlg.spinPotSlopeLF.value()}
self.listIDWeightsActual={'VGT': self.dlg.spinActVGT.value(),
'RFE': self.dlg.spinActRFE.value(),
'PopDens': self.dlg.spinActPopDens.value(),
'SoilErod': self.dlg.spinActSoilErod.value(),
'SlopeLF': self.dlg.spinActSlopeLF.value()}
return True
# _____________________
#def displayRoiValues(self):
# if self.dlg.comboChooseArea.currentIndex() is None:
# return
# elif self.dlg.comboChooseArea.currentIndex() == -1:
# self.dlg.RoiWestEdit.clear()
# self.dlg.RoiEastEdit.clear()
# self.dlg.RoiSouthEdit.clear()
# self.dlg.RoiNorthEdit.clear()
# else:
# listRoiNames = []
# listRoiNames = [ ii["name"] for ii in self.roiDefinitions ]
# self.selectedRoi = listRoiNames[self.dlg.comboChooseArea.currentIndex()]
# # West, xmin, roiXY[0]
# self.dlg.RoiWestEdit.setText( "%12.7f" % [ f["roiXY"][0] for f in self.roiDefinitions if (f["name"]==self.selectedRoi) ][0] )
# self.dlg.RoiWestEdit.displayText
# # East, xmax, roiXY[2]
# self.dlg.RoiEastEdit.setText( "%12.7f" % [ f["roiXY"][2] for f in self.roiDefinitions if (f["name"]==self.selectedRoi) ][0] )
# self.dlg.RoiEastEdit.displayText
# # South, ymin, roiXY[1]
# self.dlg.RoiSouthEdit.setText( "%12.7f" % [ f["roiXY"][1] for f in self.roiDefinitions if (f["name"]==self.selectedRoi) ][0] )
# self.dlg.RoiEastEdit.displayText
# # North ymax, roiXY[3]
# self.dlg.RoiNorthEdit.setText( "%12.7f" % [ f["roiXY"][3] for f in self.roiDefinitions if (f["name"]==self.selectedRoi) ][0] )
# self.dlg.RoiEastEdit.displayText
#
# return True
# _____________________
# add a filename to the list of opened files
# anf move the index to the last opened file
def openFile(self, name):
text={'SHPCLIP':'clipping vector file','VGT':'Vegetation', 'RFE':'Rainfall erosivity', 'PopDens': 'Population density', 'SoilErod':'Soil Erodibility','SlopeLF':'Slope length'}
ltype={'SHPCLIP':'vector','VGT':'raster', 'RFE':'raster', 'PopDens': 'raster', 'SoilErod':'raster','SlopeLF':'raster'}
dialog = QFileDialog()
fname = dialog.getOpenFileName(self.dlg, self.tr("Open raster file"))
if fname=='':
return True
if ltype[name]=='raster':
layer = QgsRasterLayer(fname, name)
if not layer.isValid():
self.logMsg("Could not open this file")
self.dlg.logTextDump.append( "Could not open file {}".format(name) )
self.dlg.tabWidget.setCurrentWidget(self.dlg.logTextDump)
return False
self.raster_list.append(layer)
if ltype[name]=='vector':
layer = QgsVectorLayer( fname, "Clip", 'ogr')
if not layer.isValid():
self.logMsg( "Could not load vector layer {} with {}".format(layer.name(), fname), QgsMessageLog.WARNING)
self.dlg.logTextDump.append( "Could not load vector layer {}".format(fname) )
self.dlg.setCurrentWidget(self.logTextDump)
return False
self.clipLayer = layer
# now put it in the correct list and make it the current selection
if name=='VGT':
self.dlg.comboVegetationIndex.addItem(name)
self.dlg.comboVegetationIndex.setCurrentIndex( self.dlg.comboVegetationIndex.count()-1 )
elif name=='RFE':
self.dlg.comboRainfallErosivity.addItem(fname)
self.dlg.comboRainfallErosivity.setCurrentIndex( self.dlg.comboRainfallErosivity.count()-1 )
elif name=='PopDens':
self.dlg.comboPopDensity.addItem(fname)
self.dlg.comboPopDensity.setCurrentIndex( self.dlg.comboPopDensity.count()-1 )
elif name=='SoilErod':
self.dlg.comboSoilErodibility.addItem(fname)
self.dlg.comboSoilErodibility.setCurrentIndex( self.dlg.comboSoilErodibility.count()-1 )
elif name=='SlopeLF':
self.dlg.comboSlopeLF.addItem( fname)
self.dlg.comboSlopeLF.setCurrentIndex( self.dlg.comboSlopeLF.count()-1 )
elif name=='SHPCLIP':
self.dlg.editClipShp.setText( fname )
return True
# ____________________
def saveFile(self, selector):
text={'PotLDIM':'Potential LDIM', 'ActLDIM':'Actual LDIM'}
dialog = QFileDialog()
fname = dialog.getSaveFileName(self.dlg, self.tr("Define a file name to save {}".format(text[selector])), os.path.expanduser("~"))
if fname:
# be sure to append '.tif'
pathname, extension = os.path.splitext(fname)
if extension!='.tif':
fname = pathname + '.tif'
if selector=='PotLDIM':
self.dlg.editOutPLDI.setText(fname)
if selector=='ActLDIM':
self.dlg.editOutALDI.setText(fname)
return True
# ___________________
def TR_degEdited(self):
# spot VGT has pixel centered coordinates and considers 112 pixels per degrees
# for pixel corners coordinates, one considers 111 pixels per degrees
# convert into meters
self.dlg.spinTR_m.setValue( self.dlg.spinTR_deg.value()* 111.0 * 1000.0 )
def TR_mEdited(self):
# convert into degrees
self.dlg.spinTR_deg.setValue( self.dlg.spinTR_m.value() / ( 1000.0 * 111.0) )
#____________________
def saveDir(self, selector):
text={'WrkDir':'intermediate processing'}
dialog = QFileDialog()
dname = dialog.getExistingDirectory(self.dlg, self.tr("Choose a directory to save {}".format(text[selector])), os.path.expanduser("~"))
if dname:
if selector=='WrkDir':
self.dlg.editWrkDir.setText(dname)
# ___________________
# def doClipShpWidgetsUpdate(self):
# if self.dlg.checkClipShp.isChecked():
# self.dlg.editClipShp.setEnabled(True)
# self.dlg.buttonClipShp.setEnabled(True)
# else:
# self.dlg.editClipShp.setEnabled(False)
# self.dlg.buttonClipShp.setEnabled(False)
# ____________________
def doInitGUI(self):
self.raster_list = []
self.dictInput = {}
self.dictReproj = {}
# clear files selectors
#self.dlg.comboChooseArea.clear()
self.dlg.comboVegetationIndex.clear()
self.dlg.comboPopDensity.clear()
self.dlg.comboRainfallErosivity.clear()
self.dlg.comboSlopeLF.clear()
self.dlg.comboSoilErodibility.clear()
# do not clear messages tab: so we can read messages from the former run
# if wrkDir not set, then initialise, else leave it as it is
if self.dlg.editWrkDir.text()=='':
self.dlg.editWrkDir.setText(self.dlg.editWrkDir.text())
# force opening on the "Help" tab
self.dlg.tabWidget.setCurrentWidget(self.dlg.tabHelp)
# setup "settings" tools. ROI are defined with xMin, yMin, xMax, yMax
#self.dlg.comboChooseArea.addItems( [ ii["name"] for ii in self.roiDefinitions ] )
#for ii in self.roiDefinitions:
# self.dlg.logTextDump.append( ii["name"] )
#self.dlg.comboChooseArea.currentIndexChanged.connect(self.displayRoiValues)
#self.dlg.comboChooseArea.setCurrentIndex(-1)
#self.displayRoiValues()
# connect the file chooser function to the buttons
self.dlg.buttonVegetationIndex.clicked.connect(lambda: self.openFile('VGT'))
self.dlg.buttonRainfallErosivity.clicked.connect(lambda: self.openFile('RFE'))
self.dlg.buttonPopDensity.clicked.connect(lambda: self.openFile('PopDens'))
self.dlg.buttonErodibility.clicked.connect(lambda: self.openFile('SoilErod'))
self.dlg.buttonSlopeLF.clicked.connect(lambda: self.openFile('SlopeLF'))
# connect buttons to saveFile functions
self.dlg.buttonPotLDIM.clicked.connect(lambda: self.saveFile('PotLDIM') )
self.dlg.buttonActLDIM.clicked.connect(lambda: self.saveFile('ActLDIM') )
# define temp directory
self.dlg.editWrkDir.setText( os.path.join( os.path.expanduser("~"), "qgis_rcmrdplugin" ) )
self.dlg.buttonWrkDir.clicked.connect(lambda: self.saveDir('WrkDir'))
# pixel resolution control
self.dlg.spinTR_m.setValue(100.0)
self.dlg.spinTR_deg.setValue(100.0 / (1000.0 * 111.0) )
self.dlg.spinTR_m.valueChanged.connect(self.TR_mEdited)
self.dlg.spinTR_deg.valueChanged.connect(self.TR_degEdited)
# signals for clipShp widgets
#self.dlg.checkClipShp.stateChanged.connect( self.doClipShpWidgetsUpdate )
self.dlg.buttonClipShp.clicked.connect( (lambda: self.openFile('SHPCLIP') ) )
# ____________________
def run(self):
"""Set up the interface content and call business-logic functions"""
self.doInitGUI()
# setup input files
self.dlg.logTextDump.append("Scanning loaded files")
layers = self.iface.legendInterface().layers()
vector_list = []
self.raster_list=[] # reset to 0 each time the plugin is called
# detect vectors and rasters already loaded in QGis
for layer in layers:
layerType = layer.type()
if (layerType == QgsMapLayer.VectorLayer) and (layer.wkbType()==QGis.WKBPolygon):
vector_list.append(layer.name())
QgsMessageLog.logMessage("--- Vector")
#QgsMessageLog.logMessage(str(dir(layer)))
print layer.geometryType()
QgsMessageLog.logMessage('Vector: '+str(layer.name())+ ' '+str(layer.geometryType()))
if layerType == QgsMapLayer.RasterLayer:
self.raster_list.append(layer)
# Assign the pre-loaded rasters and vectors to the interface objects
for ii in self.raster_list:
self.dlg.comboVegetationIndex.addItem(ii.name())
self.dlg.comboPopDensity.addItem(ii.name())
self.dlg.comboRainfallErosivity.addItem(ii.name())
self.dlg.comboSlopeLF.addItem(ii.name())
self.dlg.comboSoilErodibility.addItem(ii.name())
# show the dialog
self.dlg.show()
self.dlg.logTextDump.append("Waiting for settings")
# Run the dialog event loop, and exit only if all check are ok
checkToGo = False
result = False
while not checkToGo:
result = self.dlg.exec_()
if result:
checkToGo = self.doCheckToGo()
else:
checkToGo = True
#self.dlg.logTextDump.append("Application running.")
#self.iface.messageBar().pushMessage("Info","Running")
# See if OK was pressed
self.logMsg("Result {} checktogo {}".format(result, checkToGo))
if result:
# then run processings
if self.doProcessing():
self.iface.addRasterLayer(unicode(self.dlg.editOutPLDI.text()), 'Potential LDIM')
self.iface.addRasterLayer(unicode(self.dlg.editOutALDI.text()), 'Actual LDIM')
def __init__(self, iface):
"""Constructor.
:param iface: An interface instance that will be passed to this class
which provides the hook by which you can manipulate the QGIS
application at run time.
:type iface: QgsInterface
"""
# Save reference to the QGIS interface
self.iface = iface
# initialize plugin directory
self.plugin_dir = os.path.dirname(__file__)
# initialize locale
locale = QSettings().value('locale/userLocale')[0:2]
locale_path = os.path.join(self.plugin_dir, 'i18n',
'RCMRD_LandDegr_{}.qm'.format(locale))
if os.path.exists(locale_path):
self.translator = QTranslator()
self.translator.load(locale_path)
if qVersion() > '4.3.3':
QCoreApplication.installTranslator(self.translator)
# Create the dialog (after translation) and keep reference
self.dlg = RCMRD_LandDegrDialog()
# Declare instance attributes
self.actions = []
self.menu = self.tr(u'&RCRMD: land degradation')
# TODO: We are going to let the user set this up in a future iteration
self.toolbar = self.iface.addToolBar(u'RCMRD: land degradation')
self.toolbar.setObjectName(u'RCMRD: land degradation')
# Declare user instance variables
self.roiDefinitions = None
self.roiDefinitions = [{
"name": "IGAD",
"roiXY": [21.8094, -4.6775, 51.417, 22.227]
}, {
"name":
"Djibouti",
"roiXY": [41.749110148, 10.929824931, 43.418711785, 12.707912502]
}, {
"name":
"Eritrea",
"roiXY": [36.423647095, 12.360021871, 43.123871290, 18.004828192]
}, {
"name":
"Ethiopia",
"roiXY": [32.989799845, 3.403333435, 47.979169149, 14.879532166]
}, {
"name":
"Kenya",
"roiXY": [33.890468384, -4.677504165, 41.885019165, 5.030375823]
}, {
"name":
"Sudan",
"roiXY": [42.647246541, 7.996515605, 48.93911199991, 11.498928127]
}, {
"name":
"South Sudan",
"roiXY": [24.121555623, 3.490201518, 35.920835409, 12.216154684]
}, {
"name":
"Somali Land",
"roiXY": [42.647246541, 7.996515605, 48.93911199991, 11.498928127]
}, {
"name":
"Somalia",
"roiXY":
[40.965385376, -1.69628316498, 51.417037811, 11.989118646]
}, {
"name":
"Uganda",
"roiXY": [29.548459513, -1.475205994, 35.006472615, 4.219691875]
}]
#self.selectedRoi = None
self.raster_list = [
] # list all files open in QGis, contains file objects
self.dictReproj = None # dictionary: 'input filename' -> reprojected filename
self.clipLayer = None # store the clipping vector layer
self.listIDInputs = {
} # dictionary: key -> layer name. See keys definition in the code ('VGT', 'RFE', ...)
self.listIDWeightsPotential = {}
self.listIDWeightsActual = {}
SettingsOrganisation = 'RCMRD_QGIS'
SettingsApplication = 'RCMRD_LandDegr'
class RCMRD_LandDegr:
"""QGIS Plugin Implementation."""
def __init__(self, iface):
"""Constructor.
:param iface: An interface instance that will be passed to this class
which provides the hook by which you can manipulate the QGIS
application at run time.
:type iface: QgsInterface
"""
# Save reference to the QGIS interface
self.iface = iface
# initialize plugin directory
self.plugin_dir = os.path.dirname(__file__)
# initialize locale
locale = QSettings().value('locale/userLocale')[0:2]
locale_path = os.path.join(self.plugin_dir, 'i18n',
'RCMRD_LandDegr_{}.qm'.format(locale))
if os.path.exists(locale_path):
self.translator = QTranslator()
self.translator.load(locale_path)
if qVersion() > '4.3.3':
QCoreApplication.installTranslator(self.translator)
# Create the dialog (after translation) and keep reference
self.dlg = RCMRD_LandDegrDialog()
# Declare instance attributes
self.actions = []
self.menu = self.tr(u'&RCRMD: land degradation')
# TODO: We are going to let the user set this up in a future iteration
self.toolbar = self.iface.addToolBar(u'RCMRD: land degradation')
self.toolbar.setObjectName(u'RCMRD: land degradation')
# Declare user instance variables
self.roiDefinitions = None
self.roiDefinitions = [{
"name": "IGAD",
"roiXY": [21.8094, -4.6775, 51.417, 22.227]
}, {
"name":
"Djibouti",
"roiXY": [41.749110148, 10.929824931, 43.418711785, 12.707912502]
}, {
"name":
"Eritrea",
"roiXY": [36.423647095, 12.360021871, 43.123871290, 18.004828192]
}, {
"name":
"Ethiopia",
"roiXY": [32.989799845, 3.403333435, 47.979169149, 14.879532166]
}, {
"name":
"Kenya",
"roiXY": [33.890468384, -4.677504165, 41.885019165, 5.030375823]
}, {
"name":
"Sudan",
"roiXY": [42.647246541, 7.996515605, 48.93911199991, 11.498928127]
}, {
"name":
"South Sudan",
"roiXY": [24.121555623, 3.490201518, 35.920835409, 12.216154684]
}, {
"name":
"Somali Land",
"roiXY": [42.647246541, 7.996515605, 48.93911199991, 11.498928127]
}, {
"name":
"Somalia",
"roiXY":
[40.965385376, -1.69628316498, 51.417037811, 11.989118646]
}, {
"name":
"Uganda",
"roiXY": [29.548459513, -1.475205994, 35.006472615, 4.219691875]
}]
#self.selectedRoi = None
self.raster_list = [
] # list all files open in QGis, contains file objects
self.dictReproj = None # dictionary: 'input filename' -> reprojected filename
self.clipLayer = None # store the clipping vector layer
self.listIDInputs = {
} # dictionary: key -> layer name. See keys definition in the code ('VGT', 'RFE', ...)
self.listIDWeightsPotential = {}
self.listIDWeightsActual = {}
SettingsOrganisation = 'RCMRD_QGIS'
SettingsApplication = 'RCMRD_LandDegr'
# noinspection PyMethodMayBeStatic
def tr(self, message):
"""Get the translation for a string using Qt translation API.
We implement this ourselves since we do not inherit QObject.
:param message: String for translation.
:type message: str, QString
:returns: Translated version of message.
:rtype: QString
"""
# noinspection PyTypeChecker,PyArgumentList,PyCallByClass
return QCoreApplication.translate('RCMRD_LandDegr', message)
def add_action(self,
icon_path,
text,
callback,
enabled_flag=True,
add_to_menu=True,
add_to_toolbar=True,
status_tip=None,
whats_this=None,
parent=None):
"""Add a toolbar icon to the toolbar.
:param icon_path: Path to the icon for this action. Can be a resource
path (e.g. ':/plugins/foo/bar.png') or a normal file system path.
:type icon_path: str
:param text: Text that should be shown in menu items for this action.
:type text: str
:param callback: Function to be called when the action is triggered.
:type callback: function
:param enabled_flag: A flag indicating if the action should be enabled
by default. Defaults to True.
:type enabled_flag: bool
:param add_to_menu: Flag indicating whether the action should also
be added to the menu. Defaults to True.
:type add_to_menu: bool
:param add_to_toolbar: Flag indicating whether the action should also
be added to the toolbar. Defaults to True.
:type add_to_toolbar: bool
:param status_tip: Optional text to show in a popup when mouse pointer
hovers over the action.
:type status_tip: str
:param parent: Parent widget for the new action. Defaults None.
:type parent: QWidget
:param whats_this: Optional text to show in the status bar when the
mouse pointer hovers over the action.
:returns: The action that was created. Note that the action is also
added to self.actions list.
:rtype: QAction
"""
icon = QIcon(icon_path)
action = QAction(icon, text, parent)
action.triggered.connect(callback)
action.setEnabled(enabled_flag)
if status_tip is not None:
action.setStatusTip(status_tip)
if whats_this is not None:
action.setWhatsThis(whats_this)
if add_to_toolbar:
self.toolbar.addAction(action)
if add_to_menu:
self.iface.addPluginToMenu(self.menu, action)
self.actions.append(action)
return action
def initGui(self):
"""Create the menu entries and toolbar icons inside the QGIS GUI."""
icon_path = ':/plugins/RCMRD_LandDegr/icon.png'
self.add_action(icon_path,
text=self.tr(u'Land Degradation Indices'),
callback=self.run,
whats_this=self.tr(u'Compute RCMRD LDIM'),
parent=self.iface.mainWindow())
def unload(self):
"""Removes the plugin menu item and icon from QGIS GUI."""
for action in self.actions:
self.iface.removePluginMenu(self.tr(u'&Land Degradation (RCMRD)'),
action)
self.iface.removeToolBarIcon(action)
# remove the toolbar
del self.toolbar
# ____________________
def logMsg(self, msg, errorLvl=QgsMessageLog.INFO):
QgsMessageLog.logMessage(msg,
tag='RCMRD Land Degradation',
level=errorLvl)
prepend = ''
if errorLvl == QgsMessageLog.WARNING:
self.iface.messageBar().pushMessage("WARNING", msg)
prepend = "Warning! "
if errorLvl == QgsMessageLog.CRITICAL:
self.iface.messageBar().pushMessage("CRITICAL", msg)
prepend = "Critical error! "
self.dlg.logTextDump.append(prepend + msg)
# ____________________
# Business logic functions and methods
# ____________________
# returns uniq of an array, preserve order
def uniquify(self, seq):
noDupes = []
[noDupes.append(i) for i in seq if not noDupes.count(i)]
return noDupes
# ____________________
# Get target extent, in geographical coordinates
#def getTE(self):
# xmin=float(self.dlg.RoiWestEdit.text())
# xmax=float(self.dlg.RoiEastEdit.text())
# ymin=float(self.dlg.RoiSouthEdit.text())
# ymax=float(self.dlg.RoiNorthEdit.text())
# return [xmin, ymin, xmax, ymax]
# _____________________
def ParseType(self, type):
if type == 'Byte':
return GDT_Byte
elif type == 'Int16':
return GDT_Int16
elif type == 'UInt16':
return GDT_UInt16
elif type == 'Int32':
return GDT_Int32
elif type == 'UInt32':
return GDT_UInt32
elif type == 'Float32':
return GDT_Float32
elif type == 'Float64':
return GDT_Float64
elif type == 'CInt16':
return GDT_CInt16
elif type == 'CInt32':
return GDT_CInt32
elif type == 'CFloat32':
return GDT_CFloat32
elif type == 'CFloat64':
return GDT_CFloat64
else:
return GDT_Float32
# _____________________
# doInitDir: if the directory does not exist: create it
def doInitDir(self, thisDir):
if os.path.isdir(thisDir):
return True
os.makedirs(thisDir)
# _____________________
def doTempFile(self, prefix='', suffix='', dir=''):
idTag = random.randint(10000, 99999)
return os.path.join(dir, prefix + '_' + str(idTag) + suffix)
# ____________________
# search for a layer name, return filename
def retrieveFromName(self, name):
for ii in self.raster_list:
if ii.name() == name:
return ii.source()
return False
# ____________________
# return an array of input files (file paths)
# For each type of file (listed in listIDInputs), tries macthing corresponding name from rasters_layer
# note: do not trust sorting order
def getInputFiles(self):
inFiles = []
for kk, ii in self.listIDInputs.iteritems():
thisFile = self.retrieveFromName(ii)
if thisFile == False:
self.logMsg(
"Could no retrieve file {} associated to key {}".format(
ii, kk))
return False
else:
inFiles.append(thisFile)
self.dictInput[ii] = thisFile
return inFiles
# _____________________
# doResample: resample the inputs to a common projection and resolution, will crop images
# images saved into the working directory
# note: to read documentation about an algorithm: import processing; processing.alghelp("gdalogr:warpreproject")
def doResample(self, inFileName, output):
self.doInitDir(self.dlg.editWrkDir.text())
# for now, dst is hard coded
#inCRS = QgsCoordinateReferenceSystem(4326) # actually, must be read from the input file
newCRS = QgsCoordinateReferenceSystem()
#newCRS.createFromProj4('+proj=aea +lat_1=18 +lat_2=-4 +lat_0=11 +lon_0=25 +x_0=37.5 +y_0=11')
newCRS.createFromSrid(4326)
if not newCRS.isValid():
self.logMsg("The projection definition is not valid. Exit",
QgsMessageLog.CRITICAL)
return False
# see: https://docs.qgis.org/2.6/en/docs/user_manual/processing_algs/gdalogr/gdal_projections/warpreproject.html
# inFileName = self.raster_list[self.dlg.comboVegetationIndex.currentIndex()].source()
inFID = gdal.Open(inFileName, GA_ReadOnly)
inCRS = inFID.GetProjection()
inTrans = inFID.GetGeoTransform()
inDataType = inFID.GetRasterBand(1).DataType
# for now, we only resample 1 dataset, to be placed in a loop later on
TR = self.dlg.spinTR_deg.value()
extraParam = ''
if self.clipLayer is not None:
ext = self.clipLayer.extent()
bb = [
ext.xMinimum(),
ext.yMinimum(),
ext.xMaximum(),
ext.yMaximum()
]
extraParam = '-dstnodata 0 -te {} {} {} {} -cutline "{}" -crop_to_cutline '.format(
bb[0], bb[1], bb[2], bb[3], self.dlg.editClipShp.text())
method = 0 # because we resample thematic layers
# rtype 0: Byte, 1: int16, 2: uint16, 3:uint32, 4: int32, 5: Float32, 6: Float54
# processing.runalg("gdalogr:warpreproject")
#ALGORITHM: Warp (reproject)
#INPUT <ParameterRaster>
#SOURCE_SRS <ParameterCrs>
#DEST_SRS <ParameterCrs>
#NO_DATA <ParameterString>
#TR <ParameterNumber>
#METHOD <ParameterSelection>
#RTYPE <ParameterSelection>
#COMPRESS <ParameterSelection>
#JPEGCOMPRESSION <ParameterNumber>
#ZLEVEL <ParameterNumber>
#PREDICTOR <ParameterNumber>
#TILED <ParameterBoolean>
#BIGTIFF <ParameterSelection>
#TFW <ParameterBoolean>
#EXTRA <ParameterString>
#OUTPUT <OutputRaster>
#METHOD(Resampling method): 0 - near, 1 - bilinear, 2 - cubic, 3 - cubicspline, 4 - lanczos
#RTYPE(Output raster type): 0 - Byte, 1 - Int16, 2 - UInt16, 3 - UInt32, 4 - Int32, 5 - Float32, 6 - Float64
#COMPRESS(GeoTIFF options. Compression type): 0 - NONE, 1 - JPEG, 2 - LZW, 3 - PACKBITS, 4 - DEFLATE
#BIGTIFF(Control whether the created file is a BigTIFF or a classic TIFF): 0 - , 1 - YES, 2 - NO, 3 - IF_NEEDED, 4 - IF_SAFER
testproc = processing.runalg(
'gdalogr:warpreproject',
inFileName, # input
inCRS, # source ss
newCRS.toProj4(), # dest srs
'', # no data, <parameterString>
TR, # target resolution: 0=unchanged
0, # method: thematic layers, use 0
0, # output raster type
2, # compression
None, # jpeg compression
None, # zlevel
None, # predictor
None, # tiled
None, # bigtiff
None, # TFW
extraParam, # extra
output)
if not testproc:
self.logMsg("Reprojection failed for file {}".format(inFileName))
return False
return True
# ____________________
def doIndices(self):
self.logMsg("In do indices")
# open reprojected files
fid = {}
noData = {}
for id, layerName in self.listIDInputs.iteritems():
thisFile = self.dictReproj[self.retrieveFromName(
self.listIDInputs[id])]
fid[id] = gdal.Open(thisFile, GA_ReadOnly)
if fid[id] is None:
self.logMsg('Error: could not open file {}'.format(thisFile),
QgsMessageLog.CRITICAL)
return False
noData[id] = fid[id].GetRasterBand(1).GetNoDataValue()
self.logMsg("dataset {}, no data is {}".format(id, noData[id]))
# create ouputs from info read in VGT
# this implies that all files were reprojected to the same footprint and resolution
ns = fid['VGT'].RasterXSize
nl = fid['VGT'].RasterYSize
projection = fid['VGT'].GetProjection()
geotrans = fid['VGT'].GetGeoTransform()
outDrv = gdal.GetDriverByName('GTiff')
outFID = {}
outFID['Actual'] = outDrv.Create(self.dlg.editOutALDI.text(), ns, nl,
1, GDT_Float32)
outFID['Potential'] = outDrv.Create(self.dlg.editOutPLDI.text(), ns,
nl, 1, GDT_Float32)
# define outputs projections and geotransformations
for ii in ['Actual', 'Potential']:
outFID[ii].SetProjection(projection)
outFID[ii].SetGeoTransform(geotrans)
# infos
self.logMsg("Weights Actual")
for ii, ww in self.listIDWeightsActual.iteritems():
self.logMsg("IDWeightsActual[{}]={}".format(ii, ww))
self.logMsg("Weights Potential")
for ii, ww in self.listIDWeightsPotential.iteritems():
self.logMsg("IDWeightsPotential[{}]={}".format(ii, ww))
# compute indicators for each line
for il in range(nl):
data = {}
outData = {}
for id in self.listIDInputs:
data[id] = numpy.ravel(fid[id].GetRasterBand(1).ReadAsArray(
0, il, ns, 1))
# let's compute actual LDIM
dataActual = numpy.zeros(ns)
for ii, ww in self.listIDWeightsActual.iteritems():
if noData[ii] is not None:
wdata = (data[ii] != noData[ii])
if wdata.any():
dataActual[wdata] += ww * data[ii][wdata]
else:
dataActual += ww * data[ii]
dataPotential = numpy.zeros(ns)
for ii, ww in self.listIDWeightsPotential.iteritems():
if noData[ii] is not None:
wdata = data[ii] != noData[ii]
if wdata.any():
dataPotential[wdata] += ww * data[ii][wdata]
else:
dataPotential += ww * data[ii]
# save lines
outFID['Actual'].GetRasterBand(1).WriteArray(
dataActual.reshape(1, ns), 0, il)
outFID['Potential'].GetRasterBand(1).WriteArray(
dataPotential.reshape(1, ns), 0, il)
return True
# ____________________
# Run business logic
# 1./ reproject/resample input files, save in tmp files
# 2./ compute indices, per pixel, using weights, save in output files
# Will exit on any error, reports for errors
def doProcessing(self):
# Inputs are reprojected and resampled
listInput = self.getInputFiles()
if listInput == False:
self.logMsg("Missing input files, processing impossible")
return False
for thisFName in listInput:
self.logMsg(
"Resampling and reprojection file {}".format(thisFName))
output = self.doTempFile('reproject', '.tif',
self.dlg.editWrkDir.text())
self.dictReproj[thisFName] = output
if self.doResample(thisFName, output) == False:
self.logMsg("Reprojection failed for {}".format(thisFName))
return False
# now compute indices
if self.doIndices() == False:
self.logMsg("Error in processing indices")
return False
return True
# ____________________
def doCheckToGo(self):
#if (self.selectedRoi) is None:
# self.logMsg("Please choose a region, in the 'Settings' tab.")
# self.dlg.logTextDump.append("Please choose a region, in the 'Settings' tab.")
# self.dlg.tabWidget.setCurrentWidget(self.dlg.logTextDump)
# return False
self.logMsg("Checking inputs")
# clipping vector is now mandatory
if self.dlg.editClipShp.text() == '':
self.logMsg(
"You must define a clipping vector (for example a shapefile)",
QgsMessageLog.CRITICAL)
self.dlg.tabWidget.setCurrentWidget(self.dlg.logTextDump)
return False
# check all files are different
listFName = [ii.source() for ii in self.raster_list]
if self.uniquify(listFName) != listFName:
self.logMsg(
"Input files must be different. Please revise inputs in 'Input files' tab.",
QgsMessageLog.CRITICAL)
tabWidget.setCurrentWidget(self.dlg.logTextDump)
self.listIDInputs = {
'VGT': self.dlg.comboVegetationIndex.currentText(),
'RFE': self.dlg.comboRainfallErosivity.currentText(),
'PopDens': self.dlg.comboPopDensity.currentText(),
'SoilErod': self.dlg.comboSoilErodibility.currentText(),
'SlopeLF': self.dlg.comboSlopeLF.currentText()
}
self.listIDWeightsPotential = {
'RFE': self.dlg.spinPotRFE.value(),
'PopDens': self.dlg.spinPotPopDens.value(),
'SoilErod': self.dlg.spinPotSoilErod.value(),
'SlopeLF': self.dlg.spinPotSlopeLF.value()
}
self.listIDWeightsActual = {
'VGT': self.dlg.spinActVGT.value(),
'RFE': self.dlg.spinActRFE.value(),
'PopDens': self.dlg.spinActPopDens.value(),
'SoilErod': self.dlg.spinActSoilErod.value(),
'SlopeLF': self.dlg.spinActSlopeLF.value()
}
return True
# _____________________
#def displayRoiValues(self):
# if self.dlg.comboChooseArea.currentIndex() is None:
# return
# elif self.dlg.comboChooseArea.currentIndex() == -1:
# self.dlg.RoiWestEdit.clear()
# self.dlg.RoiEastEdit.clear()
# self.dlg.RoiSouthEdit.clear()
# self.dlg.RoiNorthEdit.clear()
# else:
# listRoiNames = []
# listRoiNames = [ ii["name"] for ii in self.roiDefinitions ]
# self.selectedRoi = listRoiNames[self.dlg.comboChooseArea.currentIndex()]
# # West, xmin, roiXY[0]
# self.dlg.RoiWestEdit.setText( "%12.7f" % [ f["roiXY"][0] for f in self.roiDefinitions if (f["name"]==self.selectedRoi) ][0] )
# self.dlg.RoiWestEdit.displayText
# # East, xmax, roiXY[2]
# self.dlg.RoiEastEdit.setText( "%12.7f" % [ f["roiXY"][2] for f in self.roiDefinitions if (f["name"]==self.selectedRoi) ][0] )
# self.dlg.RoiEastEdit.displayText
# # South, ymin, roiXY[1]
# self.dlg.RoiSouthEdit.setText( "%12.7f" % [ f["roiXY"][1] for f in self.roiDefinitions if (f["name"]==self.selectedRoi) ][0] )
# self.dlg.RoiEastEdit.displayText
# # North ymax, roiXY[3]
# self.dlg.RoiNorthEdit.setText( "%12.7f" % [ f["roiXY"][3] for f in self.roiDefinitions if (f["name"]==self.selectedRoi) ][0] )
# self.dlg.RoiEastEdit.displayText
#
# return True
# _____________________
# add a filename to the list of opened files
# anf move the index to the last opened file
def openFile(self, name):
text = {
'SHPCLIP': 'clipping vector file',
'VGT': 'Vegetation',
'RFE': 'Rainfall erosivity',
'PopDens': 'Population density',
'SoilErod': 'Soil Erodibility',
'SlopeLF': 'Slope length'
}
ltype = {
'SHPCLIP': 'vector',
'VGT': 'raster',
'RFE': 'raster',
'PopDens': 'raster',
'SoilErod': 'raster',
'SlopeLF': 'raster'
}
dialog = QFileDialog()
fname = dialog.getOpenFileName(self.dlg, self.tr("Open raster file"))
if fname == '':
return True
if ltype[name] == 'raster':
layer = QgsRasterLayer(fname, name)
if not layer.isValid():
self.logMsg("Could not open this file")
self.dlg.logTextDump.append(
"Could not open file {}".format(name))
self.dlg.tabWidget.setCurrentWidget(self.dlg.logTextDump)
return False
self.raster_list.append(layer)
if ltype[name] == 'vector':
layer = QgsVectorLayer(fname, "Clip", 'ogr')
if not layer.isValid():
self.logMsg(
"Could not load vector layer {} with {}".format(
layer.name(), fname), QgsMessageLog.WARNING)
self.dlg.logTextDump.append(
"Could not load vector layer {}".format(fname))
self.dlg.setCurrentWidget(self.logTextDump)
return False
self.clipLayer = layer
# now put it in the correct list and make it the current selection
if name == 'VGT':
self.dlg.comboVegetationIndex.addItem(name)
self.dlg.comboVegetationIndex.setCurrentIndex(
self.dlg.comboVegetationIndex.count() - 1)
elif name == 'RFE':
self.dlg.comboRainfallErosivity.addItem(fname)
self.dlg.comboRainfallErosivity.setCurrentIndex(
self.dlg.comboRainfallErosivity.count() - 1)
elif name == 'PopDens':
self.dlg.comboPopDensity.addItem(fname)
self.dlg.comboPopDensity.setCurrentIndex(
self.dlg.comboPopDensity.count() - 1)
elif name == 'SoilErod':
self.dlg.comboSoilErodibility.addItem(fname)
self.dlg.comboSoilErodibility.setCurrentIndex(
self.dlg.comboSoilErodibility.count() - 1)
elif name == 'SlopeLF':
self.dlg.comboSlopeLF.addItem(fname)
self.dlg.comboSlopeLF.setCurrentIndex(
self.dlg.comboSlopeLF.count() - 1)
elif name == 'SHPCLIP':
self.dlg.editClipShp.setText(fname)
return True
# ____________________
def saveFile(self, selector):
text = {'PotLDIM': 'Potential LDIM', 'ActLDIM': 'Actual LDIM'}
dialog = QFileDialog()
fname = dialog.getSaveFileName(
self.dlg,
self.tr("Define a file name to save {}".format(text[selector])),
os.path.expanduser("~"))
if fname:
# be sure to append '.tif'
pathname, extension = os.path.splitext(fname)
if extension != '.tif':
fname = pathname + '.tif'
if selector == 'PotLDIM':
self.dlg.editOutPLDI.setText(fname)
if selector == 'ActLDIM':
self.dlg.editOutALDI.setText(fname)
return True
# ___________________
def TR_degEdited(self):
# spot VGT has pixel centered coordinates and considers 112 pixels per degrees
# for pixel corners coordinates, one considers 111 pixels per degrees
# convert into meters
self.dlg.spinTR_m.setValue(self.dlg.spinTR_deg.value() * 111.0 *
1000.0)
def TR_mEdited(self):
# convert into degrees
self.dlg.spinTR_deg.setValue(self.dlg.spinTR_m.value() /
(1000.0 * 111.0))
#____________________
def saveDir(self, selector):
text = {'WrkDir': 'intermediate processing'}
dialog = QFileDialog()
dname = dialog.getExistingDirectory(
self.dlg,
self.tr("Choose a directory to save {}".format(text[selector])),
os.path.expanduser("~"))
if dname:
if selector == 'WrkDir':
self.dlg.editWrkDir.setText(dname)
# ___________________
# def doClipShpWidgetsUpdate(self):
# if self.dlg.checkClipShp.isChecked():
# self.dlg.editClipShp.setEnabled(True)
# self.dlg.buttonClipShp.setEnabled(True)
# else:
# self.dlg.editClipShp.setEnabled(False)
# self.dlg.buttonClipShp.setEnabled(False)
# ____________________
def doInitGUI(self):
self.raster_list = []
self.dictInput = {}
self.dictReproj = {}
# clear files selectors
#self.dlg.comboChooseArea.clear()
self.dlg.comboVegetationIndex.clear()
self.dlg.comboPopDensity.clear()
self.dlg.comboRainfallErosivity.clear()
self.dlg.comboSlopeLF.clear()
self.dlg.comboSoilErodibility.clear()
# do not clear messages tab: so we can read messages from the former run
# if wrkDir not set, then initialise, else leave it as it is
if self.dlg.editWrkDir.text() == '':
self.dlg.editWrkDir.setText(self.dlg.editWrkDir.text())
# force opening on the "Help" tab
self.dlg.tabWidget.setCurrentWidget(self.dlg.tabHelp)
# setup "settings" tools. ROI are defined with xMin, yMin, xMax, yMax
#self.dlg.comboChooseArea.addItems( [ ii["name"] for ii in self.roiDefinitions ] )
#for ii in self.roiDefinitions:
# self.dlg.logTextDump.append( ii["name"] )
#self.dlg.comboChooseArea.currentIndexChanged.connect(self.displayRoiValues)
#self.dlg.comboChooseArea.setCurrentIndex(-1)
#self.displayRoiValues()
# connect the file chooser function to the buttons
self.dlg.buttonVegetationIndex.clicked.connect(
lambda: self.openFile('VGT'))
self.dlg.buttonRainfallErosivity.clicked.connect(
lambda: self.openFile('RFE'))
self.dlg.buttonPopDensity.clicked.connect(
lambda: self.openFile('PopDens'))
self.dlg.buttonErodibility.clicked.connect(
lambda: self.openFile('SoilErod'))
self.dlg.buttonSlopeLF.clicked.connect(
lambda: self.openFile('SlopeLF'))
# connect buttons to saveFile functions
self.dlg.buttonPotLDIM.clicked.connect(
lambda: self.saveFile('PotLDIM'))
self.dlg.buttonActLDIM.clicked.connect(
lambda: self.saveFile('ActLDIM'))
# define temp directory
self.dlg.editWrkDir.setText(
os.path.join(os.path.expanduser("~"), "qgis_rcmrdplugin"))
self.dlg.buttonWrkDir.clicked.connect(lambda: self.saveDir('WrkDir'))
# pixel resolution control
self.dlg.spinTR_m.setValue(100.0)
self.dlg.spinTR_deg.setValue(100.0 / (1000.0 * 111.0))
self.dlg.spinTR_m.valueChanged.connect(self.TR_mEdited)
self.dlg.spinTR_deg.valueChanged.connect(self.TR_degEdited)
# signals for clipShp widgets
#self.dlg.checkClipShp.stateChanged.connect( self.doClipShpWidgetsUpdate )
self.dlg.buttonClipShp.clicked.connect(
(lambda: self.openFile('SHPCLIP')))
# ____________________
def run(self):
"""Set up the interface content and call business-logic functions"""
self.doInitGUI()
# setup input files
self.dlg.logTextDump.append("Scanning loaded files")
layers = self.iface.legendInterface().layers()
vector_list = []
self.raster_list = [] # reset to 0 each time the plugin is called
# detect vectors and rasters already loaded in QGis
for layer in layers:
layerType = layer.type()
if (layerType == QgsMapLayer.VectorLayer) and (layer.wkbType()
== QGis.WKBPolygon):
vector_list.append(layer.name())
QgsMessageLog.logMessage("--- Vector")
#QgsMessageLog.logMessage(str(dir(layer)))
print layer.geometryType()
QgsMessageLog.logMessage('Vector: ' + str(layer.name()) + ' ' +
str(layer.geometryType()))
if layerType == QgsMapLayer.RasterLayer:
self.raster_list.append(layer)
# Assign the pre-loaded rasters and vectors to the interface objects
for ii in self.raster_list:
self.dlg.comboVegetationIndex.addItem(ii.name())
self.dlg.comboPopDensity.addItem(ii.name())
self.dlg.comboRainfallErosivity.addItem(ii.name())
self.dlg.comboSlopeLF.addItem(ii.name())
self.dlg.comboSoilErodibility.addItem(ii.name())
# show the dialog
self.dlg.show()
self.dlg.logTextDump.append("Waiting for settings")
# Run the dialog event loop, and exit only if all check are ok
checkToGo = False
result = False
while not checkToGo:
result = self.dlg.exec_()
if result:
checkToGo = self.doCheckToGo()
else:
checkToGo = True
#self.dlg.logTextDump.append("Application running.")
#self.iface.messageBar().pushMessage("Info","Running")
# See if OK was pressed
self.logMsg("Result {} checktogo {}".format(result, checkToGo))
if result:
# then run processings
if self.doProcessing():
self.iface.addRasterLayer(unicode(self.dlg.editOutPLDI.text()),
'Potential LDIM')
self.iface.addRasterLayer(unicode(self.dlg.editOutALDI.text()),
'Actual LDIM')
def setUp(self):
"""Runs before each test."""
self.dialog = RCMRD_LandDegrDialog(None)