ourMELONS/R/repmat.R

#' @title Repeat matrix
#' @description Repeats a matrix over n columns and rows
#' @details This function was created to replicate the behavior of a homonymous
#' function on Matlab
#' @param mx matrix
#' @param n either a scalar with the number of replications in both rows and columns or a 2-length vector with individual repetitions.
#' @return matrix replicated over `ncol(mx) * n` columns and `nrow(mx) * n` rows
#' @note The Matlab implementation of this function accepts `n` with length > 2.
#' 
#' It should also be noted that a concatenated vector in R, e.g. `c(5, 2)`, becomes a column vector when coerced to matrix, even though it may look like a row vector at first glance. This is important to keep in mind when considering the expected output of this function. Vectors in R make sense to be seen as column vectors, given R's Statistics-oriented paradigm where variables are usually disposed as columns in a dataset.
#' @export
repmat <- function (mx, n) {
    # Validation
    if (length(n) > 2) warning("Extra dimensions of n ignored")
    if (length(n) == 1) n <- rep(n, 2)
    if (class(mx) != "matrix") mx <- as.matrix(mx)

    # Replicating cols
    out <- mx_col <- matrix(rep(mx, n[2]), nrow(mx))

    # Replicating rows
    if (n[1] > 1) {
        for (i in seq(n[1] - 1)) out <- rbind(out, mx_col)
    }

    # Output
    return(unname(as.matrix(out)))
}