feat(rust/sedona-expr): Implement SpatialFilter for overlaps and crosses for geoparquet pruning (#217)

Author: petern48

python/sedonadb/tests/io/test_parquet.py

@@ -164,6 +164,8 @@ def test_read_geoparquet_prune_points(geoarrow_data, name, predicate):
         "contains",
         "covers",
         "touches",
+        "crosses",
+        "overlaps",
     ],
 )
 def test_read_geoparquet_prune_polygons(sedona_testing, predicate):
@@ -176,9 +178,16 @@ def test_read_geoparquet_prune_polygons(sedona_testing, predicate):
     # A point inside of a polygon for contains / covers
     wkt_filter = "POINT (33.60 -5.54)"
 
+    # Use a wkt_filter that will lead to non-empty results
     if predicate == "touches":
         # A point on the boundary of a polygon
         wkt_filter = "POINT (33.90371119710453 -0.9500000000000001)"
+    elif predicate == "overlaps":
+        # A polygon that intersects the polygon but neither contain each other
+        wkt_filter = "POLYGON ((33 -1.9, 33.00 0, 34 0, 34 -1.9, 33 -1.9))"
+    elif predicate == "crosses":
+        # A linestring that intersects the polygon but is not contained by it
+        wkt_filter = "LINESTRING (33 -1.9, 33.00 0, 34 0, 34 -1.9, 33 -1.9)"
 
     poly_filter = shapely.from_wkt(wkt_filter)
 

rust/sedona-expr/src/spatial_filter.rs

@@ -177,7 +177,7 @@ impl SpatialFilter {
         let args = parse_args(raw_args);
         let fun_name = scalar_fun.fun().name();
         match fun_name {
-            "st_intersects" | "st_equals" | "st_touches" => {
+            "st_intersects" | "st_equals" | "st_touches" | "st_crosses" | "st_overlaps" => {
                 if args.len() != 2 {
                     return sedona_internal_err!("unexpected argument count in filter evaluation");
                 }
@@ -576,7 +576,14 @@ mod test {
 
     #[rstest]
     fn predicate_from_expr_commutative_functions(
-        #[values("st_intersects", "st_equals", "st_touches")] func_name: &str,
+        #[values(
+            "st_intersects",
+            "st_equals",
+            "st_touches",
+            "st_crosses",
+            "st_overlaps"
+        )]
+        func_name: &str,
     ) {
         let column: Arc<dyn PhysicalExpr> = Arc::new(Column::new("geometry", 0));
         let storage_field = WKB_GEOMETRY.to_storage_field("", true).unwrap();
@@ -804,7 +811,9 @@ mod test {
             "st_covers",
             "st_within",
             "st_covered_by",
-            "st_coveredby"
+            "st_coveredby",
+            "st_crosses",
+            "st_overlaps"
         )]
         func_name: &str,
     ) {
@@ -846,7 +855,9 @@ mod test {
             "st_covers",
             "st_within",
             "st_covered_by",
-            "st_coveredby"
+            "st_coveredby",
+            "st_crosses",
+            "st_overlaps"
         )]
         func_name: &str,
     ) {