R/map.R

In arulesViz: Visualizing Association Rules and Frequent Itemsets

#######################################################################
# arulesViz - Visualizing Association Rules and Frequent Itemsets
# Copyright (C) 2021 Michael Hahsler
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

## mapping helper

map <- function(
    x,
    range = c(0, 1),
    from.range = NA) {
  if (is.null(from.range) || any(is.na(from.range))) {
    from.range <- range(x, na.rm = TRUE)
  }

  ## check if all values are the same
  if (!diff(from.range)) {
    if (is.matrix(x)) {
      return(matrix(
        mean(range),
        ncol = ncol(x),
        nrow = nrow(x),
        dimnames = dimnames(x)
      ))
    } else {
      return(structure(rep(mean(range), length(x)), names = names(x)))
    }
  }

  ## map to [0,1]
  x <- (x - from.range[1])
  x <- x / diff(from.range)
  ## handle single values
  if (diff(from.range) == 0) {
    x <- 0
  }

  ## map from [0,1] to [range]
  if (range[1] > range[2]) {
    x <- 1 - x
  }
  x <- x * (abs(diff(range))) + min(range)

  x[x < min(range) | x > max(range)] <- NA

  x
}


map_int <- function(
    x,
    range = c(1, 100),
    from.range = NA) {
  floor(map(x, c(range[1], range[2] + .9), from.range))
}