ARROW-4560: [R] array() needs to take single input, not ...

r/DESCRIPTION

@@ -22,14 +22,13 @@ Imports:
     assertthat,
     glue,
     R6,
-    vctrs (>= 0.1.0.9000),
+    vctrs (>= 0.1.0),
     fs,
     tibble,
     crayon,
     withr,
     bit64
 Remotes:
-    r-lib/vctrs, 
     r-lib/withr
 Roxygen: list(markdown = TRUE)
 RoxygenNote: 6.1.1

r/NAMESPACE

@@ -68,6 +68,10 @@ S3method(read_table,"arrow::ipc::RecordBatchStreamReader")
 S3method(read_table,character)
 S3method(read_table,fs_path)
 S3method(read_table,raw)
+S3method(type,"arrow::Array")
+S3method(type,"arrow::ChunkedArray")
+S3method(type,"arrow::Column")
+S3method(type,default)
 S3method(write_arrow,"arrow::ipc::RecordBatchWriter")
 S3method(write_arrow,character)
 S3method(write_arrow,fs_path)
@@ -148,6 +152,7 @@ export(table)
 export(time32)
 export(time64)
 export(timestamp)
+export(type)
 export(uint16)
 export(uint32)
 export(uint64)
@@ -165,6 +170,7 @@ importFrom(glue,glue)
 importFrom(purrr,map)
 importFrom(purrr,map2)
 importFrom(purrr,map_int)
+importFrom(rlang,"%||%")
 importFrom(rlang,abort)
 importFrom(rlang,dots_n)
 importFrom(rlang,list2)

r/R/ArrayData.R

@@ -26,7 +26,7 @@
 #' @section Usage:
 #'
 #' ```
-#' data <- array(...)$data()
+#' data <- array(x)$data()
 #'
 #' data$type()
 #' data$length()

r/R/ChunkedArray.R

@@ -60,11 +60,8 @@
 #' @param \dots Vectors to coerce
 #' @param type currently ignored
 #'
-#' @importFrom rlang list2
+#' @importFrom rlang list2 %||%
 #' @export
-chunked_array <- function(..., type){
-  if (!missing(type)) {
-    warn("The `type` argument is currently ignored")
-  }
-  shared_ptr(`arrow::ChunkedArray`, ChunkedArray__from_list(list2(...)))
+chunked_array <- function(..., type = NULL){
+  shared_ptr(`arrow::ChunkedArray`, ChunkedArray__from_list(list2(...), type))
 }

r/R/R6.R

@@ -130,6 +130,31 @@ unique_ptr <- function(class, xp) {
   shared_ptr(`arrow::DataType`, xp)$..dispatch()
 }
 
+#' infer the arrow Array type from an R vector
+#'
+#' @param x an R vector
+#'
+#' @return an arrow logical type
+#' @export
+type <- function(x) {
+  UseMethod("type")
+}
+
+#' @export
+type.default <- function(x) {
+  `arrow::DataType`$dispatch(Array__infer_type(x))
+}
+
+#' @export
+`type.arrow::Array` <- function(x) x$type
+
+#' @export
+`type.arrow::ChunkedArray` <- function(x) x$type
+
+#' @export
+`type.arrow::Column` <- function(x) x$type
+
+
 #----- metadata
 
 #' @title class arrow::FixedWidthType

r/R/array.R

@@ -28,7 +28,7 @@
 #' @section Usage:
 #'
 #' ```
-#' a <- array(...)
+#' a <- array(x)
 #'
 #' a$IsNull(i)
 #' a$IsValid(i)
@@ -119,16 +119,13 @@
 
 #' create an [arrow::Array][arrow__Array] from an R vector
 #'
-#' @param \dots Vectors to coerce
-#' @param type currently ignored
+#' @param x R object
+#' @param type Explicit [type][arrow__DataType], or NULL (the default) to infer from the data
 #'
 #' @importFrom rlang warn
 #' @export
-array <- function(..., type){
-  if (!missing(type)) {
-    warn("The `type` argument is currently ignored")
-  }
-  `arrow::Array`$dispatch(Array__from_vector(vctrs::vec_c(...)))
+array <- function(x, type = NULL){
+  `arrow::Array`$dispatch(Array__from_vector(x, type))
 }
 
 `arrow::DictionaryArray` <- R6Class("arrow::DictionaryArray", inherit = `arrow::Array`,