Now everything compiles, but some tests are failing

Author: Kontinuation

Cargo.lock

@@ -58,6 +58,7 @@ rust-version = "1.82"
 approx = "0.5"
 abi_stable = "0.11.3"
 adbc_core = "0.20.0"
+adbc_ffi = "0.20.0"
 lru = "0.12"
 arrow = { version = "55.1.0", features = ["prettyprint", "ffi", "chrono-tz"] }
 arrow-array = { version = "55.1.0" }
@@ -129,6 +130,7 @@ datafusion-ffi = { git = "https://github.com/paleolimbot/datafusion.git", branch
 datafusion-physical-expr = { git = "https://github.com/paleolimbot/datafusion.git", branch = "local-49-with-patch", package = "datafusion-physical-expr" }
 datafusion-physical-plan = { git = "https://github.com/paleolimbot/datafusion.git", branch = "local-49-with-patch", package = "datafusion-physical-plan" }
 
-# geo-index = { git = "https://github.com/Kontinuation/geo-index.git", branch = "geo-0.31.0" }
-geo-index = { path = "/Users/bopeng/workspace/github/geo-index" }
-wkb = { path = "/Users/bopeng/workspace/github/wkb.worktrees/main" }
+geo-index = { git = "https://github.com/Kontinuation/geo-index.git", branch = "geo-0.31.0" }
+# geo-index = { path = "/Users/bopeng/workspace/github/geo-index" }
+wkb = { git = "https://github.com/Kontinuation/wkb", branch = "expose-more-apis" }
+# wkb = { path = "/Users/bopeng/workspace/github/wkb.worktrees/main" }

Cargo.toml

@@ -31,6 +31,7 @@ criterion = { workspace = true }
 sedona = { path = "../../rust/sedona" }
 sedona-testing = { path = "../../rust/sedona-testing", features = ["criterion"] }
 rstest = { workspace = true }
+geo = { workspace = true }
 
 [dependencies]
 arrow-schema = { workspace = true }
@@ -49,3 +50,7 @@ byteorder = { workspace = true }
 [[bench]]
 harness = false
 name = "geos-functions"
+
+[[bench]]
+harness = false
+name = "wkb_to_geos"

c/sedona-geos/Cargo.toml

@@ -0,0 +1,61 @@
+use criterion::{criterion_group, criterion_main};
+use geo::{LineString, Point};
+use sedona_geos::wkb_to_geos::GEOSWkbFactory;
+use wkb::Endianness;
+
+fn generate_wkb_linestring(num_points: usize, endianness: Endianness) -> Vec<u8> {
+    let mut points = Vec::new();
+    for i in 0..num_points {
+        points.push(Point::new(i as f64, i as f64));
+    }
+    let linestring = LineString::from(points);
+    let mut buffer = Vec::new();
+    wkb::writer::write_geometry(
+        &mut buffer,
+        &linestring,
+        &wkb::writer::WriteOptions { endianness },
+    )
+    .unwrap();
+    buffer
+}
+
+fn bench_parse(c: &mut criterion::Criterion) {
+    for num_points in [4, 10, 100, 500, 1000] {
+        for endianness in [Endianness::BigEndian, Endianness::LittleEndian] {
+            let wkb_buf = generate_wkb_linestring(num_points, endianness);
+            let wkb = wkb::reader::read_wkb(&wkb_buf).unwrap();
+            let endianness_name: &str = match endianness {
+                Endianness::BigEndian => "big endian",
+                Endianness::LittleEndian => "little endian",
+            };
+
+            c.bench_function(
+                &format!(
+                    "convert linestring containing {num_points} points using to_geos ({endianness_name})"
+                ),
+                |b| {
+                    let factory = GEOSWkbFactory::new();
+                    b.iter(|| {
+                        let g = factory.create(&wkb).unwrap();
+                        criterion::black_box(g);
+                    });
+                },
+            );
+
+            c.bench_function(
+                &format!(
+                    "convert linestring containing {num_points} points using geos wkb parser ({endianness_name})"
+                ),
+                |b| {
+                    b.iter(|| {
+                        let g = geos::Geometry::new_from_wkb(wkb.buf()).unwrap();
+                        criterion::black_box(g);
+                    });
+                },
+            );
+        }
+    }
+}
+
+criterion_group!(benches, bench_parse);
+criterion_main!(benches);