mirror of
https://github.com/qgis/QGIS.git
synced 2025-03-01 00:46:20 -05:00
8ceb6fc007
if they are unique in the current project Instead of always using the layer id, which is generally gibberish and meaningless for users, instead prefer to use the layer name as an input in the batch processing dialog. This is done only if the name is unique within the current project's loaded layers. This change makes the dialog more user-friendly, but more importantly it means that autofilling output values based on an input layer parameter generates more meaningful automatic output file names.
195 lines
8.0 KiB
Python
195 lines
8.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
BatchInputSelectionPanel.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$'
|
|
|
|
import os
|
|
|
|
from qgis.PyQt.QtCore import pyqtSignal
|
|
from qgis.PyQt.QtWidgets import QWidget, QHBoxLayout, QMenu, QPushButton, QLineEdit, QSizePolicy, QAction, QFileDialog
|
|
from qgis.PyQt.QtGui import QCursor
|
|
|
|
from qgis.core import (QgsMapLayer,
|
|
QgsSettings,
|
|
QgsProject,
|
|
QgsProcessing,
|
|
QgsProcessingUtils,
|
|
QgsProcessingParameterMultipleLayers,
|
|
QgsProcessingParameterRasterLayer,
|
|
QgsProcessingParameterDefinition,
|
|
QgsProcessingParameterVectorLayer,
|
|
QgsProcessingParameterFeatureSource)
|
|
|
|
from processing.gui.MultipleInputDialog import MultipleInputDialog
|
|
|
|
from processing.gui.ParameterGuiUtils import getFileFilter
|
|
from processing.tools import dataobjects
|
|
|
|
|
|
class BatchInputSelectionPanel(QWidget):
|
|
|
|
valueChanged = pyqtSignal()
|
|
|
|
def __init__(self, param, row, col, dialog):
|
|
super(BatchInputSelectionPanel, self).__init__(None)
|
|
self.param = param
|
|
self.dialog = dialog
|
|
self.row = row
|
|
self.col = col
|
|
self.horizontalLayout = QHBoxLayout(self)
|
|
self.horizontalLayout.setSpacing(0)
|
|
self.horizontalLayout.setMargin(0)
|
|
self.text = QLineEdit()
|
|
self.text.setObjectName('text')
|
|
self.text.setMinimumWidth(300)
|
|
self.setValue('')
|
|
self.text.editingFinished.connect(self.textEditingFinished)
|
|
self.text.setSizePolicy(QSizePolicy.Expanding,
|
|
QSizePolicy.Expanding)
|
|
self.horizontalLayout.addWidget(self.text)
|
|
self.pushButton = QPushButton()
|
|
self.pushButton.setText('…')
|
|
self.pushButton.clicked.connect(self.showPopupMenu)
|
|
self.horizontalLayout.addWidget(self.pushButton)
|
|
self.setLayout(self.horizontalLayout)
|
|
|
|
def _panel(self):
|
|
return self.dialog.mainWidget()
|
|
|
|
def _table(self):
|
|
return self._panel().tblParameters
|
|
|
|
def showPopupMenu(self):
|
|
popupmenu = QMenu()
|
|
|
|
if not (isinstance(self.param, QgsProcessingParameterMultipleLayers) and
|
|
self.param.layerType == dataobjects.TYPE_FILE):
|
|
selectLayerAction = QAction(
|
|
self.tr('Select from open layers'), self.pushButton)
|
|
selectLayerAction.triggered.connect(self.showLayerSelectionDialog)
|
|
popupmenu.addAction(selectLayerAction)
|
|
|
|
selectFileAction = QAction(
|
|
self.tr('Select from file system'), self.pushButton)
|
|
selectFileAction.triggered.connect(self.showFileSelectionDialog)
|
|
popupmenu.addAction(selectFileAction)
|
|
|
|
popupmenu.exec_(QCursor.pos())
|
|
|
|
def showLayerSelectionDialog(self):
|
|
layers = []
|
|
if (isinstance(self.param, QgsProcessingParameterRasterLayer) or
|
|
(isinstance(self.param, QgsProcessingParameterMultipleLayers) and
|
|
self.param.layerType() == QgsProcessing.TypeRaster)):
|
|
layers = QgsProcessingUtils.compatibleRasterLayers(QgsProject.instance())
|
|
elif isinstance(self.param, QgsProcessingParameterVectorLayer):
|
|
layers = QgsProcessingUtils.compatibleVectorLayers(QgsProject.instance())
|
|
else:
|
|
datatypes = [QgsProcessing.TypeVectorAnyGeometry]
|
|
if isinstance(self.param, QgsProcessingParameterFeatureSource):
|
|
datatypes = self.param.dataTypes()
|
|
elif isinstance(self.param, QgsProcessingParameterMultipleLayers):
|
|
datatypes = [self.param.layerType()]
|
|
|
|
if QgsProcessing.TypeVectorAnyGeometry not in datatypes:
|
|
layers = QgsProcessingUtils.compatibleVectorLayers(QgsProject.instance(), datatypes)
|
|
else:
|
|
layers = QgsProcessingUtils.compatibleVectorLayers(QgsProject.instance())
|
|
|
|
dlg = MultipleInputDialog([layer.name() for layer in layers])
|
|
dlg.exec_()
|
|
|
|
def generate_layer_id(layer):
|
|
# prefer layer name if unique
|
|
if len([l for l in layers if l.name().lower() == layer.name().lower()]) == 1:
|
|
return layer.name()
|
|
else:
|
|
# otherwise fall back to layer id
|
|
return layer.id()
|
|
|
|
if dlg.selectedoptions is not None:
|
|
selected = dlg.selectedoptions
|
|
if len(selected) == 1:
|
|
self.setValue(generate_layer_id(layers[selected[0]]))
|
|
else:
|
|
if isinstance(self.param, QgsProcessingParameterMultipleLayers):
|
|
self.text.setText(';'.join(layers[idx].id() for idx in selected))
|
|
else:
|
|
rowdif = len(selected) - (self._table().rowCount() - self.row)
|
|
for i in range(rowdif):
|
|
self._panel().addRow()
|
|
for i, layeridx in enumerate(selected):
|
|
self._table().cellWidget(i + self.row,
|
|
self.col).setValue(generate_layer_id(layers[layeridx]))
|
|
|
|
def showFileSelectionDialog(self):
|
|
settings = QgsSettings()
|
|
text = str(self.text.text())
|
|
if os.path.isdir(text):
|
|
path = text
|
|
elif os.path.isdir(os.path.dirname(text)):
|
|
path = os.path.dirname(text)
|
|
elif settings.contains('/Processing/LastInputPath'):
|
|
path = str(settings.value('/Processing/LastInputPath'))
|
|
else:
|
|
path = ''
|
|
|
|
ret, selected_filter = QFileDialog.getOpenFileNames(self, self.tr('Open file'), path,
|
|
getFileFilter(self.param))
|
|
if ret:
|
|
files = list(ret)
|
|
settings.setValue('/Processing/LastInputPath',
|
|
os.path.dirname(str(files[0])))
|
|
for i, filename in enumerate(files):
|
|
files[i] = dataobjects.getRasterSublayer(filename, self.param)
|
|
if len(files) == 1:
|
|
self.text.setText(files[0])
|
|
self.textEditingFinished()
|
|
else:
|
|
if isinstance(self.param, QgsProcessingParameterMultipleLayers):
|
|
self.text.setText(';'.join(str(f) for f in files))
|
|
else:
|
|
rowdif = len(files) - (self._table().rowCount() - self.row)
|
|
for i in range(rowdif):
|
|
self._panel().addRow()
|
|
for i, f in enumerate(files):
|
|
self._table().cellWidget(i + self.row,
|
|
self.col).setValue(f)
|
|
|
|
def textEditingFinished(self):
|
|
self._value = self.text.text()
|
|
self.valueChanged.emit()
|
|
|
|
def value(self):
|
|
return self._value
|
|
|
|
def setValue(self, value):
|
|
self._value = value
|
|
if isinstance(value, QgsMapLayer):
|
|
self.text.setText(value.name())
|
|
else: # should be basestring
|
|
self.text.setText(value)
|
|
self.valueChanged.emit()
|