stub geoarrow array stream for arrow

geoarrow · Dec 6, 2023 · 5dcf5d5 · 5dcf5d5
1 parent 0ff86c6
commit 5dcf5d5
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 1 deletion.
diff --git a/r/geoarrow/NAMESPACE b/r/geoarrow/NAMESPACE
@@ -11,6 +11,8 @@ S3method(as_geoarrow_array,sfc)
 S3method(as_geoarrow_array,wk_wkb)
 S3method(as_geoarrow_array,wk_wkt)
 S3method(as_geoarrow_array,wk_xy)
+S3method(as_geoarrow_array_stream,Array)
+S3method(as_geoarrow_array_stream,ChunkedArray)
 S3method(as_geoarrow_array_stream,default)
 S3method(as_geoarrow_array_stream,geoarrow_vctr)
 S3method(as_geoarrow_array_stream,nanoarrow_array_stream)

diff --git a/r/geoarrow/R/arrow-compat.R b/r/geoarrow/R/arrow-compat.R
@@ -30,3 +30,25 @@ as_chunked_array.geoarrow_vctr <- function(x, ..., type = NULL) {
 
   arrow::ChunkedArray$create(!!!arrays, type = type)
 }
+
+#' @export
+as_geoarrow_array_stream.ChunkedArray <- function(x, ..., schema = NULL) {
+  stream <- nanoarrow::basic_array_stream(
+    lapply(x$chunks, as_nanoarrow_array),
+    schema = as_nanoarrow_schema(x$type),
+    validate = FALSE
+  )
+
+  as_geoarrow_array_stream(stream, schema = schema)
+}
+
+#' @export
+as_geoarrow_array_stream.Array <- function(x, ..., schema = NULL) {
+  stream <- nanoarrow::basic_array_stream(
+    list(as_nanoarrow_array(x)),
+    schema = as_nanoarrow_schema(x$type),
+    validate = FALSE
+  )
+
+  as_geoarrow_array_stream(stream, schema = schema)
+}
diff --git a/r/geoarrow/tests/testthat/test-arrow-compat.R b/r/geoarrow/tests/testthat/test-arrow-compat.R
@@ -33,5 +33,5 @@ test_that("as_arrow_array() works for geoarrow_vctr", {
   expect_equal(array$length(), 2)
 
   # Check with a requested type
-  chunked <- arrow::as_arrow_array(vctr2, type = na_extension_wkb())
+  array <- arrow::as_arrow_array(vctr2, type = na_extension_wkb())
 })