Handle distance within in python provider

qgis · Aug 5, 2021 · a1ba106 · a1ba106
1 parent d208d17
commit a1ba106
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/src/python/provider_python.py b/tests/src/python/provider_python.py
@@ -14,6 +14,7 @@
 __copyright__ = 'Copyright 2018, The QGIS Project'
 
 from qgis.core import (
+    Qgis,
     QgsField,
     QgsFields,
     QgsLayerDefinition,
@@ -73,6 +74,15 @@ def __init__(self, source, request):
         else:
             self._select_rect_engine = None
             self._select_rect_geom = None
+
+        if self._request.spatialFilterType() == Qgis.SpatialFilterType.DistanceWithin and not self._request.referenceGeometry().isEmpty():
+            self._select_distance_within_geom = self._request.referenceGeometry()
+            self._select_distance_within_engine = QgsGeometry.createGeometryEngine(self._select_distance_within_geom.constGet())
+            self._select_distance_within_engine.prepareGeometry()
+        else:
+            self._select_distance_within_geom = None
+            self._select_distance_within_engine = None
+
         self._feature_id_list = None
         if self._filter_rect is not None and self._source._provider._spatialindex is not None:
             self._feature_id_list = self._source._provider._spatialindex.intersects(self._filter_rect)
@@ -122,6 +132,10 @@ def fetchFeature(self, f):
                         continue
                 f.setGeometry(_f.geometry())
                 self.geometryToDestinationCrs(f, self._transform)
+
+                if self._select_distance_within_engine and self._select_distance_within_engine.distance(f.geometry().constGet()) > self._request.distanceWithin():
+                    continue
+
                 f.setFields(_f.fields())
                 f.setAttributes(_f.attributes())
                 f.setValid(_f.isValid())