ourMELONS/R/cell.R
2021-03-31 10:26:56 +02:00

20 lines
No EOL
681 B
R

#' @title Cell array
#' @description Creates an array of zeros
#' @param n a the first dimension (or both, if sz is not passed)
#' @param sz the second dimension (or 1st and 2nd, if not passed)
#' @param expandable if TRUE, output is a list (so it can take different
#' lengths)
#' @param ... Other dimensions
#' @return An array of zeroes with the dimensions passed on call
cell <- function(n, sz = c(n, n), expandable=FALSE, ...) {
if (expandable) {
return(vector("list", length = n))
}
if (length(sz) == 1 & missing(...)) {
return(array(0, dim = c(n, sz)))
} else if (length(sz) == 2) {
return(array(0, dim = sz))
} else {
return(array(0, dim = c(n, sz, ...)))
}
}