mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-26 00:02:08 -05:00
194 lines
7.3 KiB
Python
194 lines
7.3 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
"""
|
||
|
|
***************************************************************************
|
||
|
|
BatchAlgorithmDialog.py
|
||
|
|
---------------------
|
||
|
|
Date : August 2012
|
||
|
|
Copyright : (C) 2012 by Victor Olaya
|
||
|
|
Email : volayaf at gmail dot com
|
||
|
|
***************************************************************************
|
||
|
|
* *
|
||
|
|
* This program is free software; you can redistribute it and/or modify *
|
||
|
|
* it under the terms of the GNU General Public License as published by *
|
||
|
|
* the Free Software Foundation; either version 2 of the License, or *
|
||
|
|
* (at your option) any later version. *
|
||
|
|
* *
|
||
|
|
***************************************************************************
|
||
|
|
"""
|
||
|
|
|
||
|
|
__author__ = 'Victor Olaya'
|
||
|
|
__date__ = 'August 2012'
|
||
|
|
__copyright__ = '(C) 2012, Victor Olaya'
|
||
|
|
|
||
|
|
# This will get replaced with a git SHA1 when you do a git archive
|
||
|
|
|
||
|
|
__revision__ = '$Format:%H$'
|
||
|
|
|
||
|
|
from PyQt4.QtGui import *
|
||
|
|
from PyQt4.QtCore import *
|
||
|
|
|
||
|
|
from processing.gui.BatchPanel import BatchPanel
|
||
|
|
from processing.gui.AlgorithmDialogBase import AlgorithmDialogBase
|
||
|
|
from processing.gui.AlgorithmExecutor import runalg
|
||
|
|
from processing.gui.Postprocessing import handleAlgorithmResults
|
||
|
|
|
||
|
|
from processing.core.ProcessingResults import ProcessingResults
|
||
|
|
|
||
|
|
from processing.core.parameters import ParameterFile
|
||
|
|
from processing.core.parameters import ParameterRaster
|
||
|
|
from processing.core.parameters import ParameterTable
|
||
|
|
from processing.core.parameters import ParameterVector
|
||
|
|
from processing.core.parameters import ParameterExtent
|
||
|
|
from processing.core.parameters import ParameterCrs
|
||
|
|
from processing.core.parameters import ParameterBoolean
|
||
|
|
from processing.core.parameters import ParameterSelection
|
||
|
|
from processing.core.parameters import ParameterFixedTable
|
||
|
|
from processing.core.parameters import ParameterMultipleInput
|
||
|
|
from processing.core.outputs import OutputNumber
|
||
|
|
from processing.core.outputs import OutputString
|
||
|
|
from processing.core.outputs import OutputHTML
|
||
|
|
|
||
|
|
class BatchAlgorithmDialog(AlgorithmDialogBase):
|
||
|
|
|
||
|
|
def __init__(self, alg):
|
||
|
|
AlgorithmDialogBase.__init__(self, alg)
|
||
|
|
|
||
|
|
self.alg = alg
|
||
|
|
|
||
|
|
self.setWindowTitle(self.tr('Batch Processing - %s') % self.alg.name)
|
||
|
|
|
||
|
|
self.mainWidget = BatchPanel(self, self.alg)
|
||
|
|
self.setMainWidget()
|
||
|
|
|
||
|
|
def setParamValue(self, param, widget, alg=None):
|
||
|
|
if isinstance(param, (ParameterRaster, ParameterVector, ParameterTable,
|
||
|
|
ParameterMultipleInput)):
|
||
|
|
value = widget.getText()
|
||
|
|
if unicode(value).strip() == '':
|
||
|
|
value = None
|
||
|
|
return param.setValue(value)
|
||
|
|
elif isinstance(param, ParameterBoolean):
|
||
|
|
return param.setValue(widget.currentIndex() == 0)
|
||
|
|
elif isinstance(param, ParameterSelection):
|
||
|
|
return param.setValue(widget.currentIndex())
|
||
|
|
elif isinstance(param, ParameterFixedTable):
|
||
|
|
return param.setValue(widget.table)
|
||
|
|
elif isinstance(param, ParameterExtent):
|
||
|
|
if alg is not None:
|
||
|
|
widget.useNewAlg(alg)
|
||
|
|
return param.setValue(widget.getValue())
|
||
|
|
elif isinstance(param, (ParameterCrs, ParameterFile)):
|
||
|
|
return param.setValue(widget.getValue())
|
||
|
|
else:
|
||
|
|
return param.setValue(widget.text())
|
||
|
|
|
||
|
|
def accept(self):
|
||
|
|
self.algs = []
|
||
|
|
self.load = []
|
||
|
|
self.canceled = False
|
||
|
|
|
||
|
|
for row in range(self.mainWidget.tblParameters.rowCount()):
|
||
|
|
alg = self.alg.getCopy()
|
||
|
|
col = 0
|
||
|
|
for param in alg.parameters:
|
||
|
|
if param.hidden:
|
||
|
|
continue
|
||
|
|
|
||
|
|
widget = self.mainWidget.tblParameters.cellWidget(row, col)
|
||
|
|
if not self.setParamValue(param, widget, alg):
|
||
|
|
self.lblProgress.setText(
|
||
|
|
self.tr('<b>Missing parameter value: %s (row %d)</b>') % (param.description, row + 1))
|
||
|
|
self.algs = None
|
||
|
|
return
|
||
|
|
col += 1
|
||
|
|
|
||
|
|
for out in alg.outputs:
|
||
|
|
if out.hidden:
|
||
|
|
continue
|
||
|
|
|
||
|
|
widget = self.mainWidget.tblParameters.cellWidget(row, col)
|
||
|
|
text = widget.getValue()
|
||
|
|
if text.strip() != '':
|
||
|
|
out.value = text
|
||
|
|
col += 1
|
||
|
|
else:
|
||
|
|
self.lblProgress.setText(
|
||
|
|
self.tr('<b>Wrong or missing parameter value: %s (row %d)</b>') % (out.description, row + 1))
|
||
|
|
self.algs = None
|
||
|
|
return
|
||
|
|
|
||
|
|
self.algs.append(alg)
|
||
|
|
if self.alg.getVisibleOutputsCount():
|
||
|
|
widget = self.mainWidget.tblParameters.cellWidget(row, col)
|
||
|
|
self.load.append(widget.currentIndex() == 0)
|
||
|
|
else:
|
||
|
|
self.load.append(False)
|
||
|
|
|
||
|
|
QApplication.setOverrideCursor(QCursor(Qt.WaitCursor))
|
||
|
|
self.mainWidget.setEnabled(False)
|
||
|
|
|
||
|
|
self.progressBar.setMaximum(len(self.algs))
|
||
|
|
# Make sure the Log tab is visible before executing the algorithm
|
||
|
|
try:
|
||
|
|
self.tabWidget.setCurrentIndex(1)
|
||
|
|
self.repaint()
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
|
||
|
|
for count, alg in enumerate(self.algs):
|
||
|
|
self.setText(self.tr('Processing algorithm %d/%d...') % (count + 1, len(self.algs)))
|
||
|
|
self.setInfo(self.tr('<b>Algorithm %s starting...</b>' % alg.name))
|
||
|
|
if runalg(alg, self) and not self.canceled:
|
||
|
|
if self.load[count]:
|
||
|
|
handleAlgorithmResults(alg, self, False)
|
||
|
|
self.setInfo(self.tr('Algorithm %s correctly executed...') % alg.name)
|
||
|
|
else:
|
||
|
|
QApplication.restoreOverrideCursor()
|
||
|
|
return
|
||
|
|
|
||
|
|
self.finish()
|
||
|
|
|
||
|
|
def finish(self):
|
||
|
|
for count, alg in enumerate(self.algs):
|
||
|
|
self.loadHTMLResults(alg, count)
|
||
|
|
|
||
|
|
self.createSummaryTable()
|
||
|
|
QApplication.restoreOverrideCursor()
|
||
|
|
|
||
|
|
self.mainWidget.setEnabled(True)
|
||
|
|
QMessageBox.information(self, self.tr('Batch processing'),
|
||
|
|
self.tr('Batch processing successfully completed!'))
|
||
|
|
|
||
|
|
def loadHTMLResults(self, alg, num):
|
||
|
|
for out in alg.outputs:
|
||
|
|
if out.hidden or not out.open:
|
||
|
|
continue
|
||
|
|
|
||
|
|
if isinstance(out, OutputHTML):
|
||
|
|
ProcessingResults.addResult(
|
||
|
|
'{} [{}]'.format(out.description, num), out.value)
|
||
|
|
|
||
|
|
def createSummaryTable(self):
|
||
|
|
createTable = False
|
||
|
|
|
||
|
|
for out in self.algs[0].outputs:
|
||
|
|
if isinstance(out, (OutputNumber, OutputString)):
|
||
|
|
createTable = True
|
||
|
|
break
|
||
|
|
|
||
|
|
if not createTable:
|
||
|
|
return
|
||
|
|
|
||
|
|
outputFile = getTempFilename('html')
|
||
|
|
with codecs.open(outputFile, 'w', encoding='utf-8') as f:
|
||
|
|
for alg in self.algs:
|
||
|
|
f.write('<hr>\n')
|
||
|
|
for out in alg.outputs:
|
||
|
|
if isinstance(out, (OutputNumber, OutputString)):
|
||
|
|
f.write('<p>{}: {}</p>\n'.format(out.description, out.value))
|
||
|
|
f.write('<hr>\n')
|
||
|
|
|
||
|
|
ProcessingResults.addResult(
|
||
|
|
'{} [summary]'.format(self.algs[0].name), outputFile)
|