R/add_col.R

In jhoffjann/nmmso.R: R Implementation of the NMMSO algorithm

#' @title add_col
#' @description Function that adds new cols into a matrix/vector (depending the case). 
#' It works simulating the behavior of a matrix in Matlab. 
#' Everytime you want to add a col or a full matrix in an out of bounds index, automatically is created with the col given.
#' 
#' @param original Initial variable where the final matrix will be stored, it can be already matrix.
#' @param row_index Numeric index representing the col where the new line should be inserted.
#' @param new_object col or matrix to be inserted in the original one.
#' @return The matrix with the changes done after the function was run.
#' 
#' @export


add_col <- function(original, row_index, new_object){

	nr = nrow(original)
    nc = if(ncol(new_object) == 1 && row_index > ncol(original)) row_index else if(row_index + ncol(new_object) -1 > ncol(original)) row_index + ncol(new_object) -1 else ncol(original)

    i1 = nr * (row_index - 1) + 1
    i2 = i1 + length(new_object) - 1
    original[i1:i2] = new_object

    length(original) = nr * nc
    dim(original) = c(nr, nc)

    return(original)
}