R/strict.R

In egenn/rtemis: Machine Learning and Visualization

# strict.R
# ::rtemis::
# 2016 E.D. Gennatas www.lambdamd.org

#' Strict assignment by class or type
#'
#' Allow assignment only if input is of correct class and/or type
#'
#' @param x Value to be assigned if `type` matches
#' @param accept.class Required class of value
#' @param accept.type Required type of value
#' @author E.D. Gennatas
#' @export

strict <- function(x, accept.class = NULL, accept.type = NULL) {
  input.class <- class(x)
  input.type <- typeof(x)
  if (!is.null(accept.class)) {
    if (input.class != accept.class) {
      stop(
        "Incorrect class: Expected ", accept.class, ", got ",
        input.class, " instead"
      )
    }
  }

  if (!is.null(accept.type)) {
    if (input.type != accept.type) {
      stop(
        "Incorrect type: Expected ", accept.type, ", got ",
        input.type, " instead"
      )
    }
  }

  return(x)
} # rtemis::strict