R/numeric_blurer.R

In deident: Persistent Data Anonymization Pipeline

numeric_blur.f <- function(cuts){
  function(values){
    cuts <- c(-Inf, cuts, Inf)
    cut(values, cuts)
  }
}

#' Group numeric data into baskets
#' 
#' @export
NumericBlurer <- R6Class(

  "NumericBlurer", list(

    cuts = NA,

    initialize = function(cuts=0){
      self$cuts = cuts
      self$method = function(values) {
        cuts <- c(-Inf, self$cuts, Inf)
        cut(values, cuts)
      }
    },

    transform = function(keys, ...){
      keys <- c(keys, ...)
      self$method(keys)
    },

    serialize = function(){
      super$serialize(cuts = self$cuts)
    }
  ),
  inherit = BaseDeident
)