R/misc.R

In ravetools: Signal and Image Processing Toolbox for Analyzing Intracranial Electroencephalography Data

detrend_naive <- function(x, y) {
  if (missing(y)) {
    y <- x
    x <- seq_along(y)
  } else {
    stopifnot2(length(x) == length(y), msg = "x and y must have the same length.")
  }
  n <- length(y)
  b <- (y[n] - y[1]) / (x[n] - x[1])
  a <- y[1] - b * x[1]
  list(Y = y - (a + b * x), a = a, b = b)
}

# x is either a vector or a column-major matrix
postpad <- function(x, n) {
  if (is.matrix(x)) {
    x_len <- nrow(x)
    if (n > x_len) {
      nc <- ncol(x)
      return(rbind(x, array(0, dim = c(n - x_len, nc))))
    } else {
      return(x[seq_len(n), , drop = FALSE])
    }
  } else {
    x_len <- length(x)
    if (n > x_len) {
      return(c(x, rep(0, n - x_len)))
    } else {
      return(x[seq_len(n)])
    }
  }
}