R/adjust_numeric_range.R

In orbital: Predict with 'tidymodels' Workflows in Databases

#' @export
orbital.numeric_range <- function(x, tailor, ...) {
  lower <- x$arguments$lower_limit
  upper <- x$arguments$upper_limit

  estimate <- tailor$columns$estimate

  if (!is.finite(lower) && !is.finite(upper)) {
    return(NULL)
  }

  lower_fmt <- format_numeric(lower)
  upper_fmt <- format_numeric(upper)

  out <- "dplyr::case_when("

  if (is.finite(lower)) {
    out <- paste0(out, "{estimate} < {lower_fmt} ~ {lower_fmt},")
  }
  if (is.finite(upper)) {
    out <- paste0(out, "{estimate} > {upper_fmt} ~ {upper_fmt},")
  }

  out <- paste0(out, "TRUE ~ {estimate})")
  out <- glue::glue(out)
  names(out) <- estimate
  out
}

# dplyr::case_when with 2-3 branches ~= 80 chars
#' @exportS3Method
estimate_adj_chars.numeric_range <- function(x, ...) {
  lower <- x$arguments$lower_limit
  upper <- x$arguments$upper_limit

  if (!is.finite(lower) && !is.finite(upper)) {
    return(0L)
  }

  # Base case_when overhead + TRUE branch
  chars <- 40L
  if (is.finite(lower)) {
    chars <- chars + 30L
  }

  if (is.finite(upper)) {
    chars <- chars + 30L
  }

  chars
}