feat(raster): add RS_Contains, RS_Intersects, RS_Within UDFs

Author: Kontinuation

Cargo.lock

@@ -19,5 +19,5 @@ mod proj;
 mod proj_dyn_bindgen;
 pub mod register;
 pub mod sd_order_lnglat;
-mod st_transform;
+pub mod st_transform;
 pub mod transform;

c/sedona-proj/src/lib.rs

@@ -389,10 +389,7 @@ pub fn configure_global_proj_engine(builder: ProjCrsEngineBuilder) -> Result<()>
 
 /// Do something with the global thread-local PROJ engine, creating it if it has not
 /// already been created.
-pub(crate) fn with_global_proj_engine<
-    R,
-    F: FnMut(&CachingCrsEngine<ProjCrsEngine>) -> Result<R>,
->(
+pub fn with_global_proj_engine<R, F: FnMut(&CachingCrsEngine<ProjCrsEngine>) -> Result<R>>(
     mut func: F,
 ) -> Result<R> {
     PROJ_ENGINE.with(|engine_cell| {

c/sedona-proj/src/st_transform.rs

@@ -41,11 +41,16 @@ sedona-expr = { workspace = true }
 sedona-geometry = { workspace = true }
 sedona-raster = { workspace = true }
 sedona-schema = { workspace = true }
+sedona-tg = { workspace = true }
+sedona-proj = { workspace = true }
 serde_json = { workspace = true }
+wkb = { workspace = true }
 
 [dev-dependencies]
 criterion = { workspace = true }
+geo-traits = { workspace = true }
 sedona-testing = { workspace = true, features = ["criterion"] }
+sedona-proj = { workspace = true, features = ["proj-sys"] }
 rstest = { workspace = true }
 
 [[bench]]

rust/sedona-raster-functions/Cargo.toml

@@ -110,6 +110,55 @@ fn criterion_benchmark(c: &mut Criterion) {
             Float64(-45.0, 45.0),
         ),
     );
+
+    // RS_Intersects(raster, geometry) - point
+    benchmark::scalar(
+        c,
+        &f,
+        "native-raster",
+        "rs_intersects",
+        BenchmarkArgs::ArrayArray(Raster(64, 64), Point),
+    );
+    // RS_Intersects(raster, geometry) - polygon
+    benchmark::scalar(
+        c,
+        &f,
+        "native-raster",
+        "rs_intersects",
+        BenchmarkArgs::ArrayArray(Raster(64, 64), Polygon(4)),
+    );
+    // RS_Intersects(raster, raster)
+    benchmark::scalar(
+        c,
+        &f,
+        "native-raster",
+        "rs_intersects",
+        BenchmarkArgs::ArrayArray(Raster(64, 64), Raster(64, 64)),
+    );
+    // RS_Contains(raster, geometry) - point
+    benchmark::scalar(
+        c,
+        &f,
+        "native-raster",
+        "rs_contains",
+        BenchmarkArgs::ArrayArray(Raster(64, 64), Point),
+    );
+    // RS_Contains(raster, geometry) - polygon
+    benchmark::scalar(
+        c,
+        &f,
+        "native-raster",
+        "rs_contains",
+        BenchmarkArgs::ArrayArray(Raster(64, 64), Polygon(4)),
+    );
+    // RS_Within(raster, geometry) - polygon
+    benchmark::scalar(
+        c,
+        &f,
+        "native-raster",
+        "rs_within",
+        BenchmarkArgs::ArrayArray(Raster(64, 64), Polygon(4)),
+    );
 }
 
 criterion_group!(benches, criterion_benchmark);