WFS, OGC API Features support; MultiPolygon

__init__.py

@@ -0,0 +1,35 @@
+"""
+/***************************************************************************
+ SpatialFilter
+                                 A QGIS plugin
+ This plugin applies spatial filters to layers for performance and efficiency.
+
+ Base was generated by Plugin Builder: http://g-sherman.github.io/Qgis-Plugin-Builder/
+                              -------------------
+        begin                : 2022-10-16
+        copyright            : (C) 2022 Wheregroup GmbH
+        email                : info@wheregroup.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.                                   *
+ *                                                                         *
+ ***************************************************************************/
+ This script initializes the plugin, making it known to QGIS.
+"""
+
+
+# noinspection PyPep8Naming
+def classFactory(iface):  # pylint: disable=invalid-name
+    """Load MapFilter class from file MapFilter.
+
+    :param iface: A QGIS interface instance.
+    :type iface: QgsInterface
+    """
+    #
+    from .spatial_filter import SpatialFilter
+    return SpatialFilter(iface)

controller.py

@@ -0,0 +1,142 @@
+from typing import Iterable, Optional, List
+
+from qgis.PyQt.QtCore import pyqtSignal, QObject
+from qgis.core import QgsProject, QgsMapLayer, QgsMapLayerType, QgsWkbTypes, QgsGeometry
+from qgis.gui import QgsRubberBand
+from qgis.utils import iface
+
+from .maptool import PolygonTool
+from .filters import FilterDefinition, Predicate, RemoteFilterStrategy
+from .helpers import getSupportedLayers, removeFilterFromLayer, addFilterToLayer, refreshLayerTree, hasLayerException, \
+    warnAboutCurveGeoms
+from .settings import FILTER_COMMENT_START, LOCALIZED_PLUGIN_NAME
+
+
+class FilterController(QObject):
+    currentFilter: Optional[FilterDefinition]
+    rubberBands: Optional[List[QgsRubberBand]]
+
+    filterChanged = pyqtSignal(object)
+
+    def __init__(self, parent: Optional[QObject] = None) -> None:
+        super().__init__(parent=parent)
+        self.currentFilter = None
+        self.rubberBands = []
+        self.connectSignals()
+
+    def connectSignals(self):
+        # Many thanks to *Thomas B* for suggesting that the layersAdded signal fires so quickly that editing
+        # the layer filter will happen before the data is actually requested, at least for PostGIS
+        QgsProject.instance().layersAdded.connect(self.onLayersAdded)
+
+        # to clear the filter when the project is cleared (e.g. a new project is loaded or created)
+        QgsProject.instance().cleared.connect(self.onProjectCleared)
+
+    def disconnectSignals(self):
+        # We shouldn't have to explicitly disconnect slots that vanish with the plugin (e.g. self.foo)
+        # but it does not hurt either
+        QgsProject.instance().layersAdded.disconnect(self.onLayersAdded)
+        QgsProject.instance().cleared.disconnect(self.onProjectCleared)
+
+    def removeFilter(self):
+        self.currentFilter = None
+        self.refreshFilter()
+
+    def onLayersAdded(self, layers: Iterable[QgsMapLayer]):
+        warnAboutCurveGeoms(layers)
+        if self.hasValidFilter():
+            # Apply the filter to added layers or loaded project
+            for layer in getSupportedLayers(layers):
+                addFilterToLayer(layer, self.currentFilter)
+        else:
+            # Look for saved filters to use with the plugin (possible when project was loaded)
+            for layer in getSupportedLayers(layers):
+                if FILTER_COMMENT_START in layer.subsetString():
+                    self.setFilterFromLayer(layer)
+                    return
+
+    def onProjectCleared(self):
+        """Removes the filter if one is active.
+
+        This prevents filters from staying active when the user loads another project or creates a new one.
+        """
+        self.removeFilter()
+
+    def setFilterFromLayer(self, layer):
+        filterDefinition = FilterDefinition.fromFilterString(layer.subsetString())
+        if not filterDefinition.crs.isValid():
+            filterDefinition.crs = layer.crs()
+        self.currentFilter = filterDefinition
+        self.refreshFilter()
+
+    def updateLayerFilters(self):
+        for layer in getSupportedLayers(QgsProject.instance().mapLayers().values()):
+            if self.hasValidFilter() and not hasLayerException(layer):
+                addFilterToLayer(layer, self.currentFilter)
+            else:
+                removeFilterFromLayer(layer)
+        refreshLayerTree()
+
+    def updateProjectLayers(self):
+        self.updateLayerFilters()
+
+    def refreshFilter(self):
+        self.filterChanged.emit(self.currentFilter)
+        self.updateProjectLayers()
+
+    def setFilterFromSelection(self):
+        layer = iface.activeLayer()
+        if not layer or not layer.type() == QgsMapLayerType.VectorLayer:
+            iface.messageBar().pushInfo(LOCALIZED_PLUGIN_NAME, self.tr('Select a polygon layer'))
+            return
+        if not layer.geometryType() == QgsWkbTypes.GeometryType.PolygonGeometry:
+            iface.messageBar().pushInfo(LOCALIZED_PLUGIN_NAME, self.tr('Select a polygon layer'))
+            return
+        if not layer.selectedFeatureCount():
+            iface.messageBar().pushInfo(LOCALIZED_PLUGIN_NAME, self.tr('No features selected'))
+            return
+        crs = iface.activeLayer().crs()
+        geom = QgsGeometry().unaryUnion([feature.geometry() for feature in layer.selectedFeatures()])
+        self.initFilter()
+        self.currentFilter.name = self.tr('New filter from selection')
+        self.currentFilter.crs = crs
+        self.currentFilter.wkt = geom.asWkt()
+        self.refreshFilter()
+
+    def setFilterPredicate(self, predicate: Predicate):
+        self.currentFilter.predicate = predicate.value
+        self.refreshFilter()
+
+    def setFilterBbox(self, bbox: bool):
+        self.currentFilter.bbox = bbox
+        self.refreshFilter()
+
+    def setRemoteFilterStrategy(self, strategy: RemoteFilterStrategy):
+        self.currentFilter.remote_strategy = strategy.value
+        self.refreshFilter()
+
+    def initFilter(self):
+        self.currentFilter = FilterDefinition.defaultFilter()
+
+    def hasValidFilter(self):
+        return self.currentFilter and self.currentFilter.isValid
+
+    def startSketchingTool(self):
+        self.mapTool = PolygonTool()
+        self.mapTool.sketchFinished.connect(self.onSketchFinished)
+        iface.mapCanvas().setMapTool(self.mapTool)
+
+    def stopSketchingTool(self):
+        iface.mapCanvas().unsetMapTool(self.mapTool)
+        self.mapTool.deactivate()
+
+    def onSketchFinished(self, geometry: QgsGeometry):
+        self.stopSketchingTool()
+        if not geometry.isGeosValid():
+            iface.messageBar().pushWarning(LOCALIZED_PLUGIN_NAME, self.tr("Geometry is not valid"))
+            return
+        self.initFilter()
+        self.currentFilter.name = self.tr('New filter from sketch')
+        self.currentFilter.wkt = geometry.asWkt()
+        self.currentFilter.crs = QgsProject.instance().crs()
+        self.refreshFilter()