R/align_x_list.R

In seasonal: R Interface to X-13-ARIMA-SEATS

align_x_list <- function(x, list = NULL) {
  if (is.null(x)) {
    # x specified in list
    n_series <- length(list)
    series.names <- paste0("ser_", seq(n_series))
    xs <- lapply(seq(n_series), function(i) NULL)
  } else if (is.list(x)) {
    stopifnot(all(sapply(x, inherits, "ts")))
    n_series <- length(x)
    series.names <- if (is.null(names(x))) paste0("ser_", seq(n_series)) else valid_names(names(x))
    xs <- x
  } else {
    # x specified as mts
    stopifnot(inherits(x, "mts"))
    n_series <- ncol(x)
    series.names <- valid_names(colnames(x))
    xs <- lapply(seq(n_series), function(i) x[, i])
  }

  if (is.null(list)) list <- list()
  # expand lists
  if (all(sapply(list, inherits, "list")) && length(list) == n_series) {
    # one list for each series
    lists <- list
  } else {
    # one list for all series
    lists <- rep(list(list), n_series)
  }

  list(
    xs = xs,
    lists = lists,
    series.names = series.names
  )
}