R/default_ics.R

In Higgs32584/SIR_to_Python: Interactive Plotting for Mathematical Models of Infectious Disease Spread

#' Get default initial conditions
#'
#' This function returns the default parameter vectors for a particular built-in model.
#' @param model name of the model to be solved. Examples include: SIR and SIR vaccination.
#' @return named vector of default initial conditions.
#'
get_ics <- function(model){

    ics <- NULL

    # SIR variants
    if(substr(model, start = 1, stop = 3) == "SIR" ){
        ics <- c(S = 9999, I = 1, R = 0)
    }

    # Other
    if(substr(model, start = 1, stop = 3) == "SIS" ){
        ics <- c(S = 9999, I = 1)
    }
    return(ics)
}