R/impulseplot.R

In control: A Control Systems Toolbox

# Impulse Response Plotting utility

#' @export
impulseplot <- function(sys, t = NULL, input = 1) {
  res <- impulse(sys, t, input[1])
  y <- res$y
  t <- res$t

  if (length(input) == 1) {
    # Plot the results for all rows of y against t
    if (nrow(y) == 1) {
      graphics::plot(t, y, type = "l", col = "blue", xlab = "Time, sec", ylab = paste("y"), main = "Impulse response")
      graphics::grid(5, 5)
    } else {
      graphics::par(mfrow = c(1, nrow(y)))
      for (i in 1:nrow(y)) {
        graphics::plot(t, y[i, ], type = "l", col = "blue", xlab = "Time, sec", ylab = paste("y", i), main = "Impulse response")
        graphics::grid(5, 5)
      }
    }
  }
  if (length(input) > 1) {
    for(i in 1:length(input)){
      impulseplot(sys, t, i)
    }
  }
  graphics::par(mfrow = c(1,1))
}