R/simple_factor.R

In realgabon/insuRglm: Tools for GLM Modeling in Insurance Context

simple_factor <- function(x, orig_levels, base_level, var_nm) {
  stopifnot(inherits(x, "factor"))
  stopifnot(is.character(orig_levels))
  stopifnot(is.character(base_level) && length(base_level) == 1)

  x <- factor(as.character(x), levels = orig_levels)
  attr(x, "var_nm") <- var_nm
  attr(x, "orig_levels") <- orig_levels
  attr(x, "base_level") <- base_level

  class(x) <- c("simple_factor", class(x))

  x
}