rework point path to use GeoSeries.within()

wildlife-dynamics · Aug 26, 2024 · 7fd296c · 7fd296c
1 parent 74e544a
commit 7fd296c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/ecoscope/analysis/feature_density.py b/ecoscope/analysis/feature_density.py
@@ -3,11 +3,13 @@
 
 def calculate_feature_density(selection, grid, geometry_type="point"):
     def clip_density(cell):
-        result = selection.clip_by_rect(*cell.bounds)
-        result = result[~result.is_empty]
         if geometry_type == "point":
-            return result.geometry.count()
+            result = selection.geometry.within(cell)
+            result = result[result]
+            return result.count()
         elif geometry_type == "line":
+            result = selection.clip_by_rect(*cell.bounds)
+            result = result[~result.is_empty]
             return result.geometry.length.sum()
         else:
             raise ValueError("Unsupported geometry type")