MultipleInputDialog.py 9.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225
  1. """
  2. ***************************************************************************
  3. MultipleInputDialog.py
  4. ---------------------
  5. Date : August 2012
  6. Copyright : (C) 2012 by Victor Olaya
  7. Email : volayaf at gmail dot com
  8. ***************************************************************************
  9. * *
  10. * This program is free software; you can redistribute it and/or modify *
  11. * it under the terms of the GNU General Public License as published by *
  12. * the Free Software Foundation; either version 2 of the License, or *
  13. * (at your option) any later version. *
  14. * *
  15. ***************************************************************************
  16. """
  17. __author__ = 'Victor Olaya'
  18. __date__ = 'August 2012'
  19. __copyright__ = '(C) 2012, Victor Olaya'
  20. import os
  21. import warnings
  22. from pathlib import Path
  23. from qgis.core import (QgsSettings,
  24. QgsProcessing,
  25. QgsVectorFileWriter,
  26. QgsProviderRegistry,
  27. QgsProcessingModelChildParameterSource)
  28. from qgis.PyQt import uic
  29. from qgis.PyQt.QtCore import Qt, QByteArray, QCoreApplication
  30. from qgis.PyQt.QtWidgets import QDialog, QAbstractItemView, QPushButton, QDialogButtonBox, QFileDialog
  31. from qgis.PyQt.QtGui import QStandardItemModel, QStandardItem
  32. pluginPath = os.path.split(os.path.dirname(__file__))[0]
  33. with warnings.catch_warnings():
  34. warnings.filterwarnings("ignore", category=DeprecationWarning)
  35. WIDGET, BASE = uic.loadUiType(
  36. os.path.join(pluginPath, 'ui', 'DlgMultipleSelection.ui'))
  37. class MultipleInputDialog(BASE, WIDGET):
  38. def __init__(self, options, selectedoptions=None, datatype=None):
  39. super().__init__(None)
  40. self.setupUi(self)
  41. self.datatype = datatype
  42. self.model = None
  43. self.options = []
  44. for i, option in enumerate(options):
  45. if option is None or isinstance(option, str):
  46. self.options.append((i, option))
  47. else:
  48. self.options.append((option[0], option[1]))
  49. self.selectedoptions = selectedoptions or []
  50. # Additional buttons
  51. self.btnSelectAll = QPushButton(self.tr('Select All'))
  52. self.buttonBox.addButton(self.btnSelectAll,
  53. QDialogButtonBox.ActionRole)
  54. self.btnClearSelection = QPushButton(self.tr('Clear Selection'))
  55. self.buttonBox.addButton(self.btnClearSelection,
  56. QDialogButtonBox.ActionRole)
  57. self.btnToggleSelection = QPushButton(self.tr('Toggle Selection'))
  58. self.buttonBox.addButton(self.btnToggleSelection,
  59. QDialogButtonBox.ActionRole)
  60. if self.datatype is not None:
  61. btnAddFile = QPushButton(QCoreApplication.translate("MultipleInputDialog", 'Add File(s)…'))
  62. btnAddFile.clicked.connect(self.addFiles)
  63. self.buttonBox.addButton(btnAddFile,
  64. QDialogButtonBox.ActionRole)
  65. btnAddDir = QPushButton(QCoreApplication.translate("MultipleInputDialog", 'Add Directory…'))
  66. btnAddDir.clicked.connect(self.addDirectory)
  67. self.buttonBox.addButton(btnAddDir,
  68. QDialogButtonBox.ActionRole)
  69. self.btnSelectAll.clicked.connect(lambda: self.selectAll(True))
  70. self.btnClearSelection.clicked.connect(lambda: self.selectAll(False))
  71. self.btnToggleSelection.clicked.connect(self.toggleSelection)
  72. self.settings = QgsSettings()
  73. self.restoreGeometry(self.settings.value("/Processing/multipleInputDialogGeometry", QByteArray()))
  74. self.lstLayers.setSelectionMode(QAbstractItemView.ExtendedSelection)
  75. self.lstLayers.setDragDropMode(QAbstractItemView.InternalMove)
  76. self.populateList()
  77. self.finished.connect(self.saveWindowGeometry)
  78. def saveWindowGeometry(self):
  79. self.settings.setValue("/Processing/multipleInputDialogGeometry", self.saveGeometry())
  80. def populateList(self):
  81. self.model = QStandardItemModel()
  82. for value, text in self.options:
  83. item = QStandardItem(text)
  84. item.setData(value, Qt.UserRole)
  85. item.setCheckState(Qt.Checked if value in self.selectedoptions else Qt.Unchecked)
  86. item.setCheckable(True)
  87. item.setDropEnabled(False)
  88. self.model.appendRow(item)
  89. # add extra options (e.g. manually added layers)
  90. for t in [o for o in self.selectedoptions if not isinstance(o, int)]:
  91. if isinstance(t, QgsProcessingModelChildParameterSource):
  92. item = QStandardItem(t.staticValue())
  93. else:
  94. item = QStandardItem(t)
  95. item.setData(item.text(), Qt.UserRole)
  96. item.setCheckState(Qt.Checked)
  97. item.setCheckable(True)
  98. item.setDropEnabled(False)
  99. self.model.appendRow(item)
  100. self.lstLayers.setModel(self.model)
  101. def accept(self):
  102. self.selectedoptions = []
  103. model = self.lstLayers.model()
  104. for i in range(model.rowCount()):
  105. item = model.item(i)
  106. if item.checkState() == Qt.Checked:
  107. self.selectedoptions.append(item.data(Qt.UserRole))
  108. QDialog.accept(self)
  109. def reject(self):
  110. self.selectedoptions = None
  111. QDialog.reject(self)
  112. def getItemsToModify(self):
  113. items = []
  114. if len(self.lstLayers.selectedIndexes()) > 1:
  115. for i in self.lstLayers.selectedIndexes():
  116. items.append(self.model.itemFromIndex(i))
  117. else:
  118. for i in range(self.model.rowCount()):
  119. items.append(self.model.item(i))
  120. return items
  121. def selectAll(self, value):
  122. for item in self.getItemsToModify():
  123. item.setCheckState(Qt.Checked if value else Qt.Unchecked)
  124. def toggleSelection(self):
  125. for item in self.getItemsToModify():
  126. checked = item.checkState() == Qt.Checked
  127. item.setCheckState(Qt.Unchecked if checked else Qt.Checked)
  128. def getFileFilter(self, datatype):
  129. """
  130. Returns a suitable file filter pattern for the specified parameter definition
  131. :param param:
  132. :return:
  133. """
  134. if datatype == QgsProcessing.TypeRaster:
  135. return QgsProviderRegistry.instance().fileRasterFilters()
  136. elif datatype == QgsProcessing.TypeFile:
  137. return self.tr('All files (*.*)')
  138. else:
  139. exts = QgsVectorFileWriter.supportedFormatExtensions()
  140. for i in range(len(exts)):
  141. exts[i] = self.tr('{0} files (*.{1})').format(exts[i].upper(), exts[i].lower())
  142. return self.tr('All files (*.*)') + ';;' + ';;'.join(exts)
  143. def addFiles(self):
  144. filter = self.getFileFilter(self.datatype)
  145. settings = QgsSettings()
  146. path = str(settings.value('/Processing/LastInputPath'))
  147. ret, selected_filter = QFileDialog.getOpenFileNames(self, self.tr('Select File(s)'),
  148. path, filter)
  149. if ret:
  150. files = list(ret)
  151. settings.setValue('/Processing/LastInputPath',
  152. os.path.dirname(str(files[0])))
  153. for filename in files:
  154. item = QStandardItem(filename)
  155. item.setData(filename, Qt.UserRole)
  156. item.setCheckState(Qt.Checked)
  157. item.setCheckable(True)
  158. item.setDropEnabled(False)
  159. self.model.appendRow(item)
  160. def addDirectory(self):
  161. settings = QgsSettings()
  162. path = str(settings.value('/Processing/LastInputPath'))
  163. ret = QFileDialog.getExistingDirectory(self, self.tr('Select File(s)'), path)
  164. if ret:
  165. exts = []
  166. if self.datatype == QgsProcessing.TypeVector:
  167. exts = QgsVectorFileWriter.supportedFormatExtensions()
  168. elif self.datatype == QgsProcessing.TypeRaster:
  169. for t in QgsProviderRegistry.instance().fileRasterFilters().split(';;')[1:]:
  170. for e in t[t.index('(') + 1:-1].split(' '):
  171. if e != "*.*" and e.startswith("*."):
  172. exts.append(e[2:])
  173. files = []
  174. for pp in Path(ret).rglob("*"):
  175. if not pp.is_file():
  176. continue
  177. if exts and pp.suffix[1:] not in exts:
  178. continue
  179. p = pp.as_posix()
  180. files.append(p)
  181. settings.setValue('/Processing/LastInputPath', ret)
  182. for filename in files:
  183. item = QStandardItem(filename)
  184. item.setData(filename, Qt.UserRole)
  185. item.setCheckState(Qt.Checked)
  186. item.setCheckable(True)
  187. item.setDropEnabled(False)
  188. self.model.appendRow(item)