R/transition_function.R

In IxPopDyMod: Framework for Tick Population and Infection Modeling

#' Constructor for transition functions
#'
#' @param fun A function
#'
#' Transition functions must return a numeric vector. See
#'   \code{\link{constant_fun}}, \code{\link{expo_fun}} and
#'   \code{\link{infect_fun}} for examples for how to write custom functions.
new_transition_function <- function(fun) {

  checkmate::assert_function(fun)

  structure(
    fun,
    class = "transition_function"
  )
}

# Return whether a transition function is custom or defined in the package
transition_function_is_custom <- function(fun) {
  env <- environmentName(environment(fun))
  env != "IxPopDyMod"
}