change function name to wb_dims() handle some exceptions

NAMESPACE

@@ -27,7 +27,6 @@ export(create_tablestyle)
 export(current_sheet)
 export(dataframe_to_dims)
 export(delete_data)
-export(dims)
 export(dims_to_dataframe)
 export(dims_to_rowcol)
 export(fmt_txt)
@@ -77,6 +76,7 @@ export(wb_color)
 export(wb_colour)
 export(wb_copy_cells)
 export(wb_data)
+export(wb_dims)
 export(wb_freeze_pane)
 export(wb_get_active_sheet)
 export(wb_get_base_font)

R/utils.R

@@ -261,38 +261,63 @@ rowcol_to_dim <- function(row, col) {
 }
 
 #' @rdname dims_helper
-#' @param ... dims arguments, row/col, rows/cols or objects that can be converted to data frame
+#' @param ... construct dims arguments, from rows/cols vectors or objects that can be coerced to data frame
+#' @examples
+#' # either vectors
+#' wb_dims(rows = 1:10, cols = 1:10)
+#' # or objects
+#' wb_dims(mtcars)
 #' @export
-dims <- function(...) {
+wb_dims <- function(...) {
 
   args <- list(...)
   nams <- names(args)
 
   col_names <- args$col_names
   row_names <- args$row_names
 
-  if (is.null(col_names)) col_names <- FALSE
-  if (is.null(row_names)) row_names <- FALSE
+  has_cnam <- is.null(col_names)
+  has_rnam <- is.null(row_names)
+
+  if (has_cnam) col_names <- FALSE
+  if (has_rnam) row_names <- FALSE
 
   assert_class(col_names, "logical")
   assert_class(row_names, "logical")
 
-  if (any("row" %in% nams) && any("col" %in% nams)) {
-    dims <- rowcol_to_dim(args$row, args$col)
-  } else if (any("rows" %in% nams) && any("cols" %in% nams)) {
-    dims <- rowcol_to_dims(args$rows, args$cols)
+  # wb_dims(rows, cols)
+  if (length(args) == 2 && has_cnam && has_rnam) {
+    rows <- 1L
+    cols <- 2L
+
+    # wb_dims(rows = rows, cols = cols)
+    sel <- pmatch(nams, c("rows", "cols"))
+    valid <- length(sel[!is.na(sel)])
+    if (valid == 2) {
+      rows <- sel[rows]
+      cols <- sel[cols]
+    } else if (valid == 1) {
+      stop("found only one cols/rows argument")
+    }
+
+    rows <- args[[rows]]
+    cols <- args[[cols]]
+
   } else {
 
+    # wb_dims(data.frame())
     x <- as.data.frame(args[[1]])
     rows <- seq_len(nrow(x) + col_names)
     cols <- seq_len(ncol(x) + row_names)
 
-    if (length(rows) == 1L && length(cols) == 1L) {
-      dims <- rowcol_to_dim(rows, cols)
-    } else {
-      dims <- rowcol_to_dims(rows, cols)
-    }
+  }
 
+  if (length(rows) == 1 && length(cols) == 1) {
+    # A1
+    dims <- rowcol_to_dim(rows, cols)
+  } else {
+    # A1:B2
+    dims <- rowcol_to_dims(rows, cols)
   }
 
   dims

tests/testthat/test-utils.R

@@ -46,21 +46,30 @@ test_that("dims to col & row and back", {
 
 })
 
-test_that("dims() works", {
+test_that("wb_dims() works", {
 
   # dim(mtcars)
 
-  expect_equal(dims(mtcars), "A1:K32")
+  expect_equal(wb_dims(mtcars), "A1:K32")
+  expect_equal(wb_dims(mtcars, col_names = TRUE, row_names = TRUE), "A1:L33")
 
-  expect_equal(dims(letters), "A1:A26")
+  expect_equal(wb_dims(letters), "A1:A26")
 
-  expect_equal(dims(t(letters)), "A1:Z1")
+  expect_equal(wb_dims(t(letters)), "A1:Z1")
 
-  expect_equal(dims(1), "A1")
+  expect_equal(wb_dims(1), "A1")
 
-  expect_equal(dims(rows =  1:10, cols = 5:7), "E1:G10")
+  expect_equal(wb_dims(rows =  1:10, cols = 5:7), "E1:G10")
+  expect_equal(wb_dims(cols =  1:10, rows = 5:7), "A5:J7")
+  expect_error(
+    wb_dims(cols =  1:10, col = 5:7),
+    "found only one cols/rows argument"
+  )
+
+  expect_equal(wb_dims(row = 5, col = 7), "G5")
 
-  expect_equal(dims(row = 5, col = 7), "G5")
+  expect_equal(wb_dims(1:10, LETTERS), "A1:Z10")
+  expect_equal(wb_dims(1:10, 1:26), "A1:Z10")
 
 })