mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-27 00:33:48 -05:00
362 lines
15 KiB
Python
362 lines
15 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
BatchProcessingDialog.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 import QtGui
|
|
from PyQt4.QtCore import *
|
|
from PyQt4.QtGui import *
|
|
from processing.core.ProcessingResults import ProcessingResults
|
|
from processing.gui.Postprocessing import handleAlgorithmResults
|
|
from processing.gui.FileSelectionPanel import FileSelectionPanel
|
|
from processing.gui.BatchInputSelectionPanel import BatchInputSelectionPanel
|
|
from processing.gui.AlgorithmExecutionDialog import AlgorithmExecutionDialog
|
|
from processing.gui.CrsSelectionPanel import CrsSelectionPanel
|
|
from processing.gui.ExtentSelectionPanel import ExtentSelectionPanel
|
|
from processing.gui.FixedTablePanel import FixedTablePanel
|
|
from processing.gui.BatchOutputSelectionPanel import BatchOutputSelectionPanel
|
|
from processing.gui.AlgorithmExecutor import runalg
|
|
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
|
|
from processing.tools.system import *
|
|
|
|
|
|
class BatchProcessingDialog(AlgorithmExecutionDialog):
|
|
|
|
def __init__(self, alg):
|
|
self.algs = None
|
|
self.showAdvanced = False
|
|
self.table = QtGui.QTableWidget(None)
|
|
AlgorithmExecutionDialog.__init__(self, alg, self.table)
|
|
self.setWindowModality(1)
|
|
self.resize(800, 500)
|
|
self.setWindowTitle(self.tr('Batch Processing - %s') % self.alg.name)
|
|
for param in self.alg.parameters:
|
|
if param.isAdvanced:
|
|
self.advancedButton = QtGui.QPushButton()
|
|
self.advancedButton.setText(self.tr('Show advanced parameters'))
|
|
self.advancedButton.setMaximumWidth(150)
|
|
self.buttonBox.addButton(self.advancedButton,
|
|
QtGui.QDialogButtonBox.ActionRole)
|
|
self.advancedButton.clicked.connect(
|
|
self.showAdvancedParametersClicked)
|
|
break
|
|
self.addRowButton = QtGui.QPushButton()
|
|
self.addRowButton.setText(self.tr('Add row'))
|
|
self.buttonBox.addButton(self.addRowButton,
|
|
QtGui.QDialogButtonBox.ActionRole)
|
|
self.deleteRowButton = QtGui.QPushButton()
|
|
self.deleteRowButton.setText(self.tr('Delete row'))
|
|
self.buttonBox.addButton(self.addRowButton,
|
|
QtGui.QDialogButtonBox.ActionRole)
|
|
self.buttonBox.addButton(self.deleteRowButton,
|
|
QtGui.QDialogButtonBox.ActionRole)
|
|
|
|
nOutputs = self.alg.getVisibleOutputsCount() + 1
|
|
if nOutputs == 1: nOutputs = 0
|
|
self.table.setColumnCount(self.alg.getVisibleParametersCount()
|
|
+ nOutputs)
|
|
self.setTableContent()
|
|
self.table.horizontalHeader().setResizeMode(QtGui.QHeaderView.Interactive)
|
|
self.table.horizontalHeader().setDefaultSectionSize(250)
|
|
self.table.horizontalHeader().setMinimumSectionSize(150)
|
|
self.table.verticalHeader().setResizeMode(QtGui.QHeaderView.ResizeToContents)
|
|
self.table.horizontalHeader().setStretchLastSection(True)
|
|
self.table.verticalHeader().setVisible(False)
|
|
self.table.setSizePolicy(QtGui.QSizePolicy.Expanding,
|
|
QtGui.QSizePolicy.Expanding)
|
|
self.addRowButton.clicked.connect(self.addRow)
|
|
self.deleteRowButton.clicked.connect(self.deleteRow)
|
|
self.table.horizontalHeader().sectionDoubleClicked.connect(
|
|
self.headerDoubleClicked)
|
|
|
|
def headerDoubleClicked(self, col):
|
|
widget = self.table.cellWidget(0, col)
|
|
if isinstance(widget, QtGui.QComboBox):
|
|
widgetValue = widget.currentIndex()
|
|
for row in range(1, self.table.rowCount()):
|
|
self.table.cellWidget(row, col).setCurrentIndex(widgetValue)
|
|
elif isinstance(widget, ExtentSelectionPanel):
|
|
widgetValue = widget.getValue()
|
|
for row in range(1, self.table.rowCount()):
|
|
if widgetValue is not None:
|
|
self.table.cellWidget(row, col).text.setText(widgetValue)
|
|
else:
|
|
self.table.cellWidget(row, col).text.setText('')
|
|
elif isinstance(widget, CrsSelectionPanel):
|
|
widgetValue = widget.getValue()
|
|
for row in range(1, self.table.rowCount()):
|
|
self.table.cellWidget(row, col).setAuthid(widgetValue)
|
|
elif isinstance(widget, FileSelectionPanel):
|
|
widgetValue = widget.getValue()
|
|
for row in range(1, self.table.rowCount()):
|
|
self.table.cellWidget(row, col).setText(widgetValue)
|
|
elif isinstance(widget, QtGui.QLineEdit):
|
|
widgetValue = widget.text()
|
|
for row in range(1, self.table.rowCount()):
|
|
self.table.cellWidget(row, col).setText(widgetValue)
|
|
elif isinstance(widget, BatchInputSelectionPanel):
|
|
widgetValue = widget.getText()
|
|
for row in range(1, self.table.rowCount()):
|
|
self.table.cellWidget(row, col).setText(widgetValue)
|
|
else:
|
|
pass
|
|
|
|
def setTableContent(self):
|
|
i = 0
|
|
for param in self.alg.parameters:
|
|
self.table.setHorizontalHeaderItem(i,
|
|
QtGui.QTableWidgetItem(param.description))
|
|
if param.isAdvanced:
|
|
self.table.setColumnHidden(i, not self.showAdvanced)
|
|
i += 1
|
|
for out in self.alg.outputs:
|
|
if not out.hidden:
|
|
self.table.setHorizontalHeaderItem(i,
|
|
QtGui.QTableWidgetItem(out.description))
|
|
i += 1
|
|
|
|
if self.alg.getVisibleOutputsCount():
|
|
self.table.setHorizontalHeaderItem(i,
|
|
QtGui.QTableWidgetItem(self.tr('Load in QGIS')))
|
|
|
|
for i in range(3):
|
|
self.addRow()
|
|
|
|
def accept(self):
|
|
self.canceled = False
|
|
self.algs = []
|
|
self.load = []
|
|
for row in range(self.table.rowCount()):
|
|
alg = self.alg.getCopy()
|
|
col = 0
|
|
for param in alg.parameters:
|
|
if param.hidden:
|
|
continue
|
|
widget = self.table.cellWidget(row, col)
|
|
if not self.setParameterValueFromWidget(param, widget, alg):
|
|
self.progressLabel.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.table.cellWidget(row, col)
|
|
text = widget.getValue()
|
|
if text.strip() != '':
|
|
out.value = text
|
|
col += 1
|
|
else:
|
|
self.progressLabel.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.table.cellWidget(row, col)
|
|
self.load.append(widget.currentIndex() == 0)
|
|
else:
|
|
self.load.append(False)
|
|
|
|
QApplication.setOverrideCursor(QCursor(Qt.WaitCursor))
|
|
self.table.setEnabled(False)
|
|
self.tabWidget.setCurrentIndex(1)
|
|
self.progress.setMaximum(len(self.algs))
|
|
# make sure the log tab is visible before executing the algorithm
|
|
try:
|
|
self.repaint()
|
|
except:
|
|
pass
|
|
for (i, alg) in enumerate(self.algs):
|
|
self.setBaseText(self.tr('Processing algorithm %d/%d...') %(i + 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[i]:
|
|
handleAlgorithmResults(alg, self, False)
|
|
self.setInfo(self.tr('Algorithm %s correctly executed...') % alg.name)
|
|
else:
|
|
QApplication.restoreOverrideCursor()
|
|
return
|
|
|
|
self.finishAll()
|
|
|
|
def loadHTMLResults(self, alg, i):
|
|
for out in alg.outputs:
|
|
if out.hidden or not out.open:
|
|
continue
|
|
if isinstance(out, OutputHTML):
|
|
ProcessingResults.addResult(out.description + '[' + str(i)
|
|
+ ']', out.value)
|
|
|
|
def cancel(self):
|
|
self.canceled = True
|
|
self.table.setEnabled(True)
|
|
|
|
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')
|
|
f = open(outputFile, 'w')
|
|
for alg in self.algs:
|
|
f.write('<hr>\n')
|
|
for out in alg.outputs:
|
|
if isinstance(out, (OutputNumber, OutputString)):
|
|
f.write('<p>' + out.description + ': ' + str(out.value)
|
|
+ '</p>\n')
|
|
f.write('<hr>\n')
|
|
f.close()
|
|
ProcessingResults.addResult(self.algs[0].name + '[summary]',
|
|
outputFile)
|
|
|
|
def finishAll(self):
|
|
i = 0
|
|
for alg in self.algs:
|
|
self.loadHTMLResults(alg, i)
|
|
i = i + 1
|
|
self.createSummaryTable()
|
|
QApplication.restoreOverrideCursor()
|
|
self.table.setEnabled(True)
|
|
QMessageBox.information(self, self.tr('Batch processing'),
|
|
self.tr('Batch processing successfully completed!'))
|
|
|
|
def setParameterValueFromWidget(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 getWidgetFromParameter(self, param, row, col):
|
|
if isinstance(param, (ParameterRaster, ParameterVector,
|
|
ParameterTable, ParameterMultipleInput)):
|
|
item = BatchInputSelectionPanel(param, row, col, self)
|
|
elif isinstance(param, ParameterBoolean):
|
|
item = QtGui.QComboBox()
|
|
item.addItem(self.tr('Yes'))
|
|
item.addItem(self.tr('No'))
|
|
if param.default:
|
|
item.setCurrentIndex(0)
|
|
else:
|
|
item.setCurrentIndex(1)
|
|
elif isinstance(param, ParameterSelection):
|
|
item = QtGui.QComboBox()
|
|
item.addItems(param.options)
|
|
elif isinstance(param, ParameterFixedTable):
|
|
item = FixedTablePanel(param)
|
|
elif isinstance(param, ParameterExtent):
|
|
item = ExtentSelectionPanel(self, self.alg, param.default)
|
|
elif isinstance(param, ParameterCrs):
|
|
item = CrsSelectionPanel(param.default)
|
|
elif isinstance(param, ParameterFile):
|
|
item = FileSelectionPanel(param.isFolder)
|
|
else:
|
|
item = QtGui.QLineEdit()
|
|
try:
|
|
item.setText(str(param.default))
|
|
except:
|
|
pass
|
|
|
|
return item
|
|
|
|
def deleteRow(self):
|
|
if self.table.rowCount() > 2:
|
|
self.table.setRowCount(self.table.rowCount() - 1)
|
|
|
|
def addRow(self):
|
|
self.table.setRowCount(self.table.rowCount() + 1)
|
|
i = 0
|
|
for param in self.alg.parameters:
|
|
if param.hidden:
|
|
continue
|
|
self.table.setCellWidget(self.table.rowCount() - 1, i,
|
|
self.getWidgetFromParameter(param,
|
|
self.table.rowCount() - 1, i))
|
|
i += 1
|
|
for out in self.alg.outputs:
|
|
if out.hidden:
|
|
continue
|
|
self.table.setCellWidget(self.table.rowCount() - 1, i,
|
|
BatchOutputSelectionPanel(out, self.alg,
|
|
self.table.rowCount() - 1, i, self))
|
|
i += 1
|
|
|
|
if self.alg.getVisibleOutputsCount():
|
|
item = QtGui.QComboBox()
|
|
item.addItem(self.tr('Yes'))
|
|
item.addItem(self.tr('No'))
|
|
item.setCurrentIndex(0)
|
|
self.table.setCellWidget(self.table.rowCount() - 1, i, item)
|
|
|
|
def showAdvancedParametersClicked(self):
|
|
self.showAdvanced = not self.showAdvanced
|
|
if self.showAdvanced:
|
|
self.advancedButton.setText(self.tr('Hide advanced parameters'))
|
|
else:
|
|
self.advancedButton.setText(self.tr('Show advanced parameters'))
|
|
i = 0
|
|
for param in self.alg.parameters:
|
|
if param.isAdvanced:
|
|
self.table.setColumnHidden(i, not self.showAdvanced)
|
|
i += 1
|
|
|
|
def setText(self, text):
|
|
self.progressLabel.setText(self.baseText + ' --- [' + text + ']')
|
|
|
|
def setBaseText(self, text):
|
|
self.baseText = text
|