Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'master' of github.com:qgis/Quantum-GIS
- Loading branch information
Showing
6 changed files
with
412 additions
and
7 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -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 ) |
Oops, something went wrong.