R/check-input.R

In khvorov45/impactflu: Quantification of Population-Level Impact of Vaccination

#' Checks input for simulation and methods
#'
#' @param vaccinations Integer vector counts of vaccinations
#' @param cases Integer vector counts of cases
#' @param ve Numeric vector vaccine effectiveness (0 to 1).
#'
#' @importFrom rlang abort
#' @importFrom glue glue
#'
#' @noRd
check_input_methods <- function(init_pop_size, vaccinations, cases, ve) {
  if (init_pop_size <= 0L) {
    abort("init_pop_size must be greater than 0")
  }
  if (length(cases) == 1) {
    abort("length of cases should be greater than 1")
  }
  if (length(vaccinations) != length(cases)) {
    abort(glue(
      "length of cases ({length(cases)}) should match length of ",
      "vaccinations ({length(vaccinations)})"
    ))
  }
  if (length(ve) == 1) {
    ve <- rep(ve, length(cases))
  }
  if (length(ve) != length(vaccinations)) {
    abort(glue(
      "length of ve ({length(ve)}) should be either 1 or the same as ",
      "the length of vaccinations and cases ({length(cases)})"
    ))
  }
  ve
}