[FEATURE] add gdal_fillnodata to GdalTools

Author: alexbruy

python/plugins/GdalTools/GdalTools.py

@@ -180,6 +180,12 @@ def initGui( self ):
       QObject.connect( self.nearBlack, SIGNAL( "triggered()" ), self.doNearBlack )
       self.analysisMenu.addAction( self.nearBlack )
 
+    if self.GdalVersion >= "1.7":
+      self.fillNodata = QAction( QIcon(":/icons/fillnodata.png"), QCoreApplication.translate( "GdalTools", "Fill nodata" ), self.iface.mainWindow() )
+      self.fillNodata.setStatusTip( QCoreApplication.translate( "GdalTools", "Fill raster regions by interpolation from edges") )
+      QObject.connect( self.fillNodata, SIGNAL( "triggered()" ), self.doFillNodata )
+      self.analysisMenu.addAction( self.fillNodata )
+
     if self.GdalVersion >= "1.6":
       self.proximity = QAction( QIcon(":/icons/proximity.png"),  QCoreApplication.translate( "GdalTools", "Proximity (Raster distance)" ), self.iface.mainWindow() )
       self.proximity.setStatusTip( QCoreApplication.translate( "GdalTools", "Produces a raster proximity map") )
@@ -295,6 +301,11 @@ def doNearBlack( self ):
     d = NearBlack( self.iface )
     self.runToolDialog( d )
 
+  def doFillNodata( self ):
+    from tools.doFillNodata import GdalToolsDialog as FillNodata
+    d = FillNodata( self.iface )
+    self.runToolDialog( d )
+
   def doWarp( self ):
     from tools.doWarp import GdalToolsDialog as Warp
     d = Warp( self.iface )

python/plugins/GdalTools/icons/fillnodata.png

@@ -21,6 +21,7 @@
         <file>icons/about.png</file>
         <file>icons/dem.png</file>
         <file>icons/projection-export.png</file>
+        <file>icons/fillnodata.png</file>
         <file>icons/edit.png</file>
         <file>icons/reset.png</file>
     </qresource>

python/plugins/GdalTools/resources.qrc

@@ -0,0 +1,215 @@
+# -*- coding: utf-8 -*-
+from PyQt4.QtCore import *
+from PyQt4.QtGui import *
+from qgis.core import *
+from qgis.gui import *
+
+from ui_widgetFillNodata import Ui_GdalToolsWidget as Ui_Widget
+from widgetBatchBase import GdalToolsBaseBatchWidget as BaseBatchWidget
+import GdalTools_utils as Utils
+
+import os.path
+
+class GdalToolsDialog( QWidget, Ui_Widget, BaseBatchWidget ):
+  def __init__( self, iface ):
+      QWidget.__init__( self )
+      self.iface = iface
+
+      self.setupUi( self )
+      BaseBatchWidget.__init__( self, self.iface, "gdal_fillnodata.py" )
+
+      self.inSelector.setType( self.inSelector.FILE_LAYER )
+      self.outSelector.setType( self.outSelector.FILE )
+      self.maskSelector.setType( self.maskSelector.FILE )
+
+      self.progressBar.setValue(0)
+      self.progressBar.hide()
+      self.formatLabel.hide()
+      self.formatCombo.hide()
+
+      self.outputFormat = Utils.fillRasterOutputFormat()
+
+      self.setParamsStatus(
+        [
+          ( self.inSelector, SIGNAL( "filenameChanged()" ) ),
+          ( self.outSelector, SIGNAL( "filenameChanged()" ) ),
+          ( self.maskSelector, SIGNAL( "filenameChanged()" ), self.maskCheck ),
+          ( self.distanceSpin, SIGNAL( "valueChanged( int )" ), self.distanceCheck ),
+          ( self.smoothSpin, SIGNAL( "valueChanged( int )" ), self.smoothCheck ),
+          ( self.bandSpin, SIGNAL( "valueChanged( int )" ), self.bandCheck ),
+          ( self.nomaskCheck, SIGNAL( "stateChanged( int )" ) )
+        ]
+      )
+
+      self.connect( self.inSelector, SIGNAL( "selectClicked()" ), self.fillInputFile )
+      self.connect( self.outSelector, SIGNAL( "selectClicked()" ), self.fillOutputFile)
+      self.connect( self.maskSelector, SIGNAL( "selectClicked()" ), self.fillMaskFile)
+      self.connect( self.batchCheck, SIGNAL( "stateChanged( int )" ), self.switchToolMode )
+
+      # add raster filters to combo
+      self.formatCombo.addItems( Utils.FileFilter.allRastersFilter().split( ";;" ) )
+
+
+  def switchToolMode( self ):
+      self.setCommandViewerEnabled( not self.batchCheck.isChecked() )
+      self.progressBar.setVisible( self.batchCheck.isChecked() )
+      self.formatLabel.setVisible( self.batchCheck.isChecked() )
+      self.formatCombo.setVisible( self.batchCheck.isChecked() )
+
+      self.inSelector.setType( self.inSelector.FILE if self.batchCheck.isChecked() else self.inSelector.FILE_LAYER )
+      self.outSelector.clear()
+
+      if self.batchCheck.isChecked():
+        self.inFileLabel = self.label.text()
+        self.outFileLabel = self.label_1.text()
+        self.label.setText( QCoreApplication.translate( "GdalTools", "&Input directory" ) )
+        self.label_1.setText( QCoreApplication.translate( "GdalTools", "&Output directory" ) )
+
+        QObject.disconnect( self.inSelector, SIGNAL( "selectClicked()" ), self.fillInputFile )
+        QObject.disconnect( self.outSelector, SIGNAL( "selectClicked()" ), self.fillOutputFile )
+
+        QObject.connect( self.inSelector, SIGNAL( "selectClicked()" ), self. fillInputDir )
+        QObject.connect( self.outSelector, SIGNAL( "selectClicked()" ), self.fillOutputDir )
+      else:
+        self.label.setText( self.inFileLabel )
+        self.label_1.setText( self.outFileLabel )
+
+        QObject.disconnect( self.inSelector, SIGNAL( "selectClicked()" ), self.fillInputDir )
+        QObject.disconnect( self.outSelector, SIGNAL( "selectClicked()" ), self.fillOutputDir )
+
+        QObject.connect( self.inSelector, SIGNAL( "selectClicked()" ), self.fillInputFile )
+        QObject.connect( self.outSelector, SIGNAL( "selectClicked()" ), self.fillOutputFile )
+
+  def fillInputFile( self ):
+      lastUsedFilter = Utils.FileFilter.lastUsedRasterFilter()
+      inputFile = Utils.FileDialog.getOpenFileName( self,
+                                                    self.tr( "Select the files to analyse" ),
+                                                    Utils.FileFilter.allRastersFilter(),
+                                                    lastUsedFilter )
+      if inputFile.isEmpty():
+        return
+      Utils.FileFilter.setLastUsedRasterFilter( lastUsedFilter )
+      self.inSelector.setFilename( inputFile )
+
+  def fillOutputFile( self ):
+      lastUsedFilter = Utils.FileFilter.lastUsedRasterFilter()
+      outputFile = Utils.FileDialog.getSaveFileName( self, self.tr( "Select the raster file to save the results to" ), Utils.FileFilter.allRastersFilter(), lastUsedFilter )
+      if outputFile.isEmpty():
+        return
+      Utils.FileFilter.setLastUsedRasterFilter( lastUsedFilter )
+
+      self.outputFormat = Utils.fillRasterOutputFormat( lastUsedFilter, outputFile )
+      self.outSelector.setFilename( outputFile )
+
+  def fillMaskFile( self ):
+      lastUsedFilter = Utils.FileFilter.lastUsedRasterFilter()
+      inputFile = Utils.FileDialog.getOpenFileName( self,
+                                                    self.tr( "Select the files to analyse" ),
+                                                    Utils.FileFilter.allRastersFilter(),
+                                                    lastUsedFilter )
+      if inputFile.isEmpty():
+        return
+      Utils.FileFilter.setLastUsedRasterFilter( lastUsedFilter )
+      self.maskSelector.setFilename( inputFile )
+
+  def fillInputDir( self ):
+      inputDir = Utils.FileDialog.getExistingDirectory( self, self.tr( "Select the input directory with files" ))
+      if inputDir.isEmpty():
+        return
+      self.inSelector.setFilename( inputDir )
+
+  def fillOutputDir( self ):
+      outputDir = Utils.FileDialog.getExistingDirectory( self, self.tr( "Select the output directory to save the results to" ) )
+      if outputDir.isEmpty():
+        return
+      self.outSelector.setFilename( outputDir )
+
+  def getArguments(self):
+      arguments = QStringList()
+      maskFile = self.maskSelector.filename()
+      if self.distanceCheck.isChecked() and self.distanceSpin.value() != 0:
+        arguments << "-md"
+        arguments << self.distanceSpin.text()
+      if self.smoothCheck.isChecked() and self.smoothSpin.value() != 0:
+        arguments << "-si"
+        arguments << str( self.smoothSpin.value() )
+      if self.bandCheck.isChecked() and self.bandSpin.value() != 0:
+        arguments << "-b"
+        arguments << str( self.bandSpin.value() )
+      if self.maskCheck.isChecked() and not maskFile.isEmpty():
+        arguments << "-mask"
+        arguments << maskFile
+      if self.nomaskCheck.isChecked():
+        arguments << "-nomask"
+      if self.isBatchEnabled():
+        if self.formatCombo.currentIndex() != 0:
+          arguments << "-of"
+          arguments << Utils.fillRasterOutputFormat( self.formatCombo.currentText() )
+        return arguments
+      else:
+        outputFn = self.getOutputFileName()
+        if not outputFn.isEmpty():
+          arguments << "-of"
+          arguments << self.outputFormat
+        arguments << self.getInputFileName()
+        arguments << outputFn
+        return arguments
+
+  def onLayersChanged( self ):
+      self.inSelector.setLayers( Utils.LayerRegistry.instance().getRasterLayers() )
+
+  def getInputFileName(self):
+      return self.inSelector.filename()
+
+  def getOutputFileName(self):
+      return self.outSelector.filename()
+
+  def addLayerIntoCanvas(self, fileInfo):
+      self.iface.addRasterLayer(fileInfo.filePath())
+
+  def isBatchEnabled(self):
+      return self.batchCheck.isChecked()
+
+  def setProgressRange(self, maximum):
+      self.progressBar.setRange(0, maximum)
+
+  def updateProgress(self, index, total):
+      if index < total:
+        self.progressBar.setValue( index + 1 )
+      else:
+        self.progressBar.setValue( 0 )
+
+  def batchRun(self):
+      exts = self.formatCombo.currentText().remove( QRegExp('^.*\(') ).remove( QRegExp('\).*$') ).split( " " )
+      if not exts.isEmpty() and exts != "*" and exts != "*.*":
+        outExt = exts[ 0 ].remove( "*" )
+      else:
+        outExt = ".tif"
+
+      self.base.enableRun( False )
+      self.base.setCursor( Qt.WaitCursor )
+
+      inDir = self.getInputFileName()
+      outDir = self.getOutputFileName()
+
+      extensions = Utils.getRasterExtensions()
+      workDir = QDir( inDir )
+      workDir.setFilter( QDir.Files | QDir.NoSymLinks | QDir.NoDotAndDotDot )
+      workDir.setNameFilters( extensions )
+      files = workDir.entryList()
+
+      self.inFiles = []
+      self.outFiles = []
+
+      for f in files:
+        self.inFiles.append( inDir + "/" + f )
+        if outDir != None:
+          outFile = f.replace( QRegExp( "\.[a-zA-Z0-9]{2,4}" ), outExt )
+          self.outFiles.append( outDir + "/" + outFile )
+
+      self.errors = QStringList()
+      self.batchIndex = 0
+      self.batchTotal = len( self.inFiles )
+      self.setProgressRange( self.batchTotal )
+
+      self.runItem( self.batchIndex, self.batchTotal )