R/partition-all-reals.R

In sakrejda/dt: Core modeling tools

#' Partition the real numbers into intervals
#'
#' @param breaks vector of breakpoints to create partitions with
#' @return list of vectors, each vector contains the lower and upper
#'         bound in order
#' @export
partition_all_reals = function(breaks = vector(mode = 'numeric')) {
  N = length(breaks)
  if (N == 0) {
    return(list(c(lb = -Inf, ub = Inf)))
  } else if (N == 1) {
    return(list(c(lb = -Inf, ub = breaks), c(lb = breaks, ub = Inf)))
  } else {
    breaks = sort(breaks)
    lb = c(-Inf, breaks[1:N])
    ub = c(breaks[1:N], Inf)
    partitions = purrr::map2(lb, ub, ~ c(lb = .x, ub = .y))
    return(partitions)
  }
}