R/geom_tileglyph.R

Defines functions makeContent.tileglyphtree geom_tileglyph

Documented in geom_tileglyph

#' Add Tile Glyphs as a Scatterplot
#'
#' The tileglyph geom is used to plot multivariate data as tile glyphs similar
#' to 'autoglyph' \insertCite{beddow_shape_1990}{gglyph} or 'stripe glyph'
#' \insertCite{fuchs_evaluation_2013}{gglyph} in a scatterplot.
#'
#' @template fill.gradient-arg
#' @template repel-arg
#' @inheritParams ggplot2::layer
#' @inheritParams starglyphGrob
#' @inheritParams tileglyphGrob
#' @param cols Name of columns specifying the variables to be plotted in the
#'   glyphs as a character vector.
#' @param linewidth The line width of the tile glyphs.
#' @param colour The colour of the tile glyphs.
#' @param ... Other arguments passed on to \code{\link[ggplot2]{layer}()}. These
#'   are often aesthetics, used to set an aesthetic to a fixed value, like
#'   \code{colour = "green"} or \code{size = 3}. They may also be parameters to
#'   the paired geom/stat.
#'
#' @section Aesthetics: \code{geom_tileglyph()} understands the following
#'   aesthetics (required aesthetics are in bold): \itemize{ \item{\strong{x}}
#'   \item{\strong{y}} \item{alpha} \item{group} \item{size} }
#'
#'   See \code{vignette("ggplot2-specs", package = "ggplot2")} for further
#'   details on setting these aesthetics.
#'
#'   The following additional aesthetics are considered if \code{repel = TRUE}:
#'   \itemize{ \item{point.size} \item{segment.linetype} \item{segment.colour}
#'   \item{segment.size} \item{segment.alpha} \item{segment.curvature}
#'   \item{segment.angle} \item{segment.ncp} \item{segment.shape}
#'   \item{segment.square} \item{segment.squareShape} \item{segment.inflect}
#'   \item{segment.debug} }
#'
#'   See \code{ggrepel}
#'   \href{https://ggrepel.slowkow.com/articles/examples.html}{examples} page
#'   for further details on setting these aesthetics.
#'
#' @family geoms
#'
#' @return A \code{geom} layer.
#'
#' @importFrom rlang as_quosures syms
#' @importFrom utils modifyList
#' @importFrom ggplot2 layer ggproto aes
#' @importFrom grid grobTree addGrob makeContent gTree setChildren
#' @importFrom Rdpack reprompt
#' @export
#'
#' @seealso \code{\link[gglyph]{tileglyphGrob}}
#'
#' @references
#'
#' \insertAllCited{}
#'
#' @examples
#'
#' # Scale the data
#' zs <- c("hp", "drat", "wt", "qsec", "vs", "am", "gear", "carb")
#' mtcars[ , zs] <- lapply(mtcars[ , zs], scales::rescale)
#'
#' mtcars$cyl <- as.factor(mtcars$cyl)
#' mtcars$lab <- row.names(mtcars)
#'
#' library(ggplot2)
#' theme_set(theme_bw())
#' options(ggplot2.discrete.colour = RColorBrewer::brewer.pal(8, "Dark2"))
#' options(ggplot2.discrete.fill = RColorBrewer::brewer.pal(8, "Dark2"))
#'
#' ggplot(data = mtcars) +
#'   geom_tileglyph(aes(x = mpg, y = disp),
#'                  cols = zs, size = 2,
#'                  fill.gradient = "Blues",
#'                  alpha =  0.5) +
#'   ylim(c(-0, 550))
#'
#' ggplot(data = mtcars) +
#'   geom_tileglyph(aes(x = mpg, y = disp),
#'                  cols = zs, size = 2,
#'                  nrow = 2,
#'                  fill.gradient = "Greens",
#'                  alpha =  0.5) +
#'   ylim(c(-0, 550))
#'
#' ggplot(data = mtcars) +
#'   geom_tileglyph(aes(x = mpg, y = disp),
#'                  cols = zs, size = 1,
#'                  ratio = 4,
#'                  fill.gradient = "RdYlBu",
#'                  alpha =  0.5) +
#'   ylim(c(-0, 550))
#'
#' ggplot(data = mtcars) +
#'   geom_tileglyph(aes(x = mpg, y = disp),
#'                  cols = zs, size = 1,
#'                  ratio = 4, nrow = 2,
#'                  fill.gradient = "viridis",
#'                  alpha =  0.5) +
#'   ylim(c(-0, 550))
#'
#' # Repel glyphs
#' ggplot(data = mtcars) +
#'   geom_point(aes(x = mpg, y = disp)) +
#'   geom_tileglyph(aes(x = mpg, y = disp),
#'                  cols = zs, size = 2,
#'                  fill.gradient = "Blues",
#'                  alpha = 1, repel = TRUE) +
#'   ylim(c(-0, 550))
#'
#' ggplot(data = mtcars) +
#'   geom_point(aes(x = mpg, y = disp)) +
#'   geom_tileglyph(aes(x = mpg, y = disp),
#'                  cols = zs, size = 1,
#'                  ratio = 4, nrow = 2,
#'                  fill.gradient = "viridis",
#'                  alpha = 1, repel = TRUE) +
#'   ylim(c(-0, 550))
#'
geom_tileglyph <- function(mapping = NULL, data = NULL, stat = "identity",
                           position = "identity", ...,
                           cols = character(0L),
                           colour = "black",
                           # fill = NA,
                           ratio = 1,
                           nrow = 1,
                           linewidth = 1,
                           fill.gradient = NULL,
                           show.legend = NA,
                           repel = FALSE,
                           repel.control = gglyph.repel.control(),
                           inherit.aes = TRUE) {

  # Modify mapping to include cols
  mcols <- rlang::as_quosures(rlang::syms(cols), .GlobalEnv)
  names(mcols) <- cols
  mapping <- modifyList(mapping, mcols)

  params <- list(
    ratio = ratio,
    nrow = nrow,
    colour = colour,
    # fill = fill,
    fill.gradient = fill.gradient,
    repel = repel,
    cols = cols,
    box.padding = unit(repel.control$box.padding, "lines"),
    point.padding = unit(repel.control$point.padding, "lines"),
    min.segment.length = unit(repel.control$min.segment.length, "lines"),
    arrow = repel.control$arrow,
    force = repel.control$force,
    force_pull = repel.control$force_pull,
    max.time = repel.control$max.time,
    max.iter = repel.control$max.iter,
    max.overlaps = repel.control$max.overlaps,
    nudge_x = repel.control$nudge_x,
    nudge_y = repel.control$nudge_y,
    xlim = repel.control$xlim,
    ylim = repel.control$ylim,
    direction = repel.control$direction,
    seed = repel.control$seed,
    verbose = repel.control$verbose,
    ...)

  # Modify geom aesthetics to include cols
  geomout <- GeomTileGlyph
  geomout$required_aes <- c(geomout$required_aes, cols)

  ggplot2::layer(
    data = data,
    mapping = mapping,
    stat = stat,
    geom = geomout,
    position = position,
    show.legend = show.legend,
    inherit.aes = inherit.aes,
    params = params)

}

GeomTileGlyph <- ggplot2::ggproto("GeomTileGlyph", ggplot2::Geom,
                                  required_aes = c("x", "y"),
                                  default_aes = ggplot2::aes(size = 1,
                                                             alpha = 1,
                                                             # repel aes
                                                             point.size = 1,
                                                             segment.linetype = 1,
                                                             segment.colour = NULL,
                                                             segment.size = 0.5,
                                                             segment.alpha = NULL,
                                                             segment.curvature = -1e-20,
                                                             segment.angle = 20,
                                                             segment.ncp = 3,
                                                             segment.shape = 0.5,
                                                             segment.square = TRUE,
                                                             segment.squareShape = 1,
                                                             segment.inflect = FALSE,
                                                             segment.debug = FALSE),

                                  draw_key = ggplot2::draw_key_polygon,

                                  setup_params = function(data, params) {

                                    params
                                  },

                                  setup_data = function(data, params) {

                                    cols <- params$cols

                                    # Check if "cols" exist in data
                                    if (FALSE %in% (cols %in% colnames(data))) {
                                      stop(paste('The following column(s) specified as "cols" are not present in "data":\n',
                                                 paste(cols[!(cols %in% colnames(data))],
                                                       collapse = ", "),
                                                 sep = ""))
                                    }


                                    # Check if cols are numeric or factor
                                    intfactcols <- unlist(lapply(data[, cols],
                                                                 function(x) FALSE %in% (is.vector(x, mode = "integer") |
                                                                                           is.vector(x, mode = "numeric") |
                                                                                           is.factor(x))))
                                    if (TRUE %in% intfactcols) {
                                      stop('The following column(s) specified as "cols" in ',
                                           '"data" are not of type numeric, integer or factor:\n',
                                           paste(names(intfactcols[intfactcols]), collapse = ", "))
                                    }

                                    # Remove rows with missing values in "cols"
                                    # check for missing values
                                    missvcols <- unlist(lapply(data[, cols], function(x) TRUE %in% is.na(x)))
                                    if (TRUE %in% missvcols) {
                                      warning(paste('The following column(s) in "data" have missing values:\n',
                                                    paste(names(missvcols[missvcols]), collapse = ", ")))

                                      data <- remove_missing(df = data, vars = cols)
                                    }

                                    if (is.null(params$fill.gradient)) {
                                      stop('The "fill.gradient" gradient fill palette is missing.')
                                    }

                                    data$linewidth <- params$linewidth
                                    data$colour <- params$colour
                                    # data$fill <- params$fill
                                    data
                                  },

                                  draw_panel = function(data, panel_params,
                                                        coord, cols,
                                                        ratio,
                                                        nrow,
                                                        colour,
                                                        # fill,
                                                        fill.gradient,
                                                        repel,
                                                        box.padding,
                                                        point.padding,
                                                        min.segment.length,
                                                        arrow,
                                                        force,
                                                        force_pull,
                                                        max.time,
                                                        max.iter,
                                                        max.overlaps,
                                                        nudge_x,
                                                        nudge_y,
                                                        xlim,
                                                        ylim,
                                                        direction,
                                                        seed,
                                                        verbose) {

                                    # if needed rename columns using our convention
                                    for (this_dim in c("x", "y")) {
                                      this_orig <- sprintf("%s_orig", this_dim)
                                      this_nudge <- sprintf("nudge_%s", this_dim)
                                      if (!this_nudge %in% colnames(data)) {
                                        data[[this_nudge]] <- data[[this_dim]]
                                        if (this_orig %in% colnames(data)) {
                                          data[[this_dim]] <- data[[this_orig]]
                                          data[[this_orig]] <- NULL
                                        }
                                      }
                                    }

                                    # Transform the nudges to the panel scales.
                                    nudges <- data.frame(x = data$nudge_x, y = data$nudge_y)
                                    nudges <- coord$transform(nudges, panel_params)

                                    data <- coord$transform(data, panel_params)

                                    grid.levels <- NULL

                                    # Convert factor columns to equivalent numeric
                                    fcols <- names(Filter(is.factor, data[, cols]))

                                    if (length(fcols) > 0)  {
                                      data[, fcols] <- lapply(data[, cols], function(f) as.numeric(levels(f))[f])
                                    }

                                    # Gradient colour mapping
                                    gdata <- NULL
                                    if (!is.null(fill.gradient)) {
                                      gdata <- data[, cols]

                                      gdata <- lapply(gdata,
                                                      function(x) scales::col_numeric(palette = fill.gradient,
                                                                                      domain = min(x):max(x))(x))
                                      gdata <- data.frame(gdata)
                                    }

                                    # The nudge is relative to the data.
                                    data$nudge_x <- nudges$x - data$x
                                    data$nudge_y <- nudges$y - data$y

                                    # Transform limits to panel scales.
                                    limits <- data.frame(x = xlim, y = ylim)
                                    limits <- coord$transform(limits, panel_params)

                                    # Allow Inf.
                                    if (length(limits$x) == length(xlim)) {
                                      limits$x[is.infinite(xlim)] <- xlim[is.infinite(xlim)]
                                    }
                                    if (length(limits$y) == length(ylim)) {
                                      limits$y[is.infinite(ylim)] <- ylim[is.infinite(ylim)]
                                    }

                                    # Fill NAs with defaults.
                                    limits$x[is.na(limits$x)] <- c(0, 1)[is.na(limits$x)]
                                    limits$y[is.na(limits$y)] <- c(0, 1)[is.na(limits$y)]

                                    ggname("geom_tileglyph",
                                           grid::gTree(data=data,
                                                       # x = x, y = y,
                                                       cols=cols,
                                                       # fill = fill,
                                                       ratio = ratio,
                                                       nrow = nrow,
                                                       fill.gradient = fill.gradient,
                                                       gdata = gdata,
                                                       colour = colour,
                                                       # alpha = alpha,
                                                       # linejoin = "mitre",
                                                       repel = repel,
                                                       limits = limits,
                                                       box.padding = box.padding,
                                                       point.padding = point.padding,
                                                       min.segment.length = min.segment.length,
                                                       arrow = arrow,
                                                       force = force,
                                                       force_pull = force_pull,
                                                       max.time = max.time,
                                                       max.iter = max.iter,
                                                       max.overlaps = max.overlaps,
                                                       nudge_x = nudge_x,
                                                       nudge_y = nudge_y,
                                                       xlim = xlim,
                                                       ylim = ylim,
                                                       direction = direction,
                                                       seed = seed,
                                                       verbose = verbose,
                                                       cl="tileglyphtree"))

                                    # ggname("geom_tileglyph",
                                    #        grid::gTree(
                                    #          children = grid::gList(
                                    #            grid::pointsGrob(x = data$x,
                                    #                             y = data$y,
                                    #                             default.units = "native",
                                    #                             pch = 20,
                                    #                             gp = grid::gpar(col = data$colour,
                                    #                                             fill = data$fill))
                                    #          )))
                                  }
)

#' grid::makeContent function for the grobTree of tileglyphGrob objects
#' @param g A grid grobTree.
#' @export
#' @noRd
makeContent.tileglyphtree <- function(g) {

  if (g$repel) {

    repel.debug <- getOption("gglyph.repel.debug", default = FALSE)

    # The padding around each bounding box.
    box_padding_x <- grid::convertWidth(g$box.padding, "native", valueOnly = TRUE)
    box_padding_y <- grid::convertHeight(g$box.padding, "native", valueOnly = TRUE)

    # The padding around each point.
    if (is.na(g$point.padding)) {
      g$point.padding = unit(0, "lines")
    }

    # Original glyph grob
    tileg <- lapply(seq_along(g$data$x),
                    function(i) tileglyphGrob(x = g$data$x[i],
                                              y = g$data$y[i],
                                              z = unlist(g$data[i, g$cols]),
                                              size = g$data$size[i],
                                              ratio = g$ratio,
                                              nrow = g$nrow,
                                              fill = unlist(g$gdata[i, ]),
                                              col = g$colour,
                                              lwd = g$data$linewidth[i],
                                              alpha = g$data$alpha[i],
                                              linejoin = "mitre"))

    if (repel.debug) {
      glorg <- lapply(seq_along(g$data$x),
                      function(i) tileglyphGrob(x = g$data$x[i],
                                                y = g$data$y[i],
                                                z = unlist(g$data[i, g$cols]),
                                                size = g$data$size[i],
                                                ratio = g$ratio,
                                                nrow = g$nrow,
                                                fill = "transparent",
                                                col = "grey",
                                                lwd = g$data$linewidth[i],
                                                alpha = g$data$alpha[i],
                                                linejoin = "mitre"))
    }

    # Create a dataframe with x1 y1 x2 y2 - Computed from bounding box
    boxes <- lapply(seq_along(tileg), function(i) {
      x1 <- grid::convertWidth(grid::grobX(tileg[[i]], "west"), "native", TRUE)
      x2 <- grid::convertWidth(grid::grobX(tileg[[i]], "east"), "native", TRUE)
      y1 <- grid::convertHeight(grid::grobY(tileg[[i]], "south"), "native", TRUE)
      y2 <- grid::convertHeight(grid::grobY(tileg[[i]], "north"), "native", TRUE)
      c(
        "x1" = x1 - box_padding_x + g$nudge_x,
        "y1" = y1 - box_padding_y + g$nudge_y,
        "x2" = x2 + box_padding_x + g$nudge_x,
        "y2" = y2 + box_padding_y + g$nudge_y
      )
    })


    if (repel.debug) {
      # Bounding box grob
      boxes2 <- data.frame(do.call(rbind, boxes))
      # bboxg <- lapply(seq_along(boxes2$x1), function(i) {
      #   grid::polylineGrob(x = c(boxes2$x1[i], g$data$x[i],  boxes2$x2[i], g$data$x[i],  boxes2$x1[i]),
      #                      y = c(g$data$y[i],  boxes2$y1[i], g$data$y[i],  boxes2$y2[i], g$data$y[i]),
      #                      gp = gpar(col = "grey"))
      # })
      bboxg <- lapply(seq_along(boxes2$x1), function(i) {
        grid::polylineGrob(x = c(boxes2$x1[i], boxes2$x1[i], boxes2$x2[i], boxes2$x2[i], boxes2$x1[i]),
                           y = c(boxes2$y1[i], boxes2$y2[i], boxes2$y2[i], boxes2$y1[i], boxes2$y1[i]),
                           gp = gpar(col = "gray"))
      })
    }

  # Make the repulsion reproducible if desired.
  if (is.null(g$seed) || !is.na(g$seed)) {
    set.seed(g$seed)
  }

  # Beware the magic numbers. I do not understand them.
  # I just accept them as necessary to get the code to work.
  p_width <- grid::convertWidth(unit(1, "npc"), "inch", TRUE)
  p_height <- grid::convertHeight(unit(1, "npc"), "inch", TRUE)
  p_ratio <- (p_width / p_height)
  if (p_ratio > 1) {
    p_ratio <- p_ratio ^ (1 / (1.15 * p_ratio))
  }
  point_size <- p_ratio * grid::convertWidth(
    grid::unit(g$data$point.size, "lines"), "native", valueOnly = TRUE
  ) / 13
  point_padding <- p_ratio * grid::convertWidth(
    grid::unit(g$point.padding, "lines"), "native", valueOnly = TRUE
  ) / 13

  # Repel overlapping bounding boxes away from each other.
  repel <- repel_boxes2(
    data_points     = as.matrix(g$data[,c("x","y")]),
    point_size      = point_size,
    point_padding_x = point_padding,
    point_padding_y = point_padding,
    boxes           = do.call(rbind, boxes),
    xlim            = range(g$limits$x),
    ylim            = range(g$limits$y),
    hjust           = rep (0.5, nrow(g$data)),
    vjust           = rep (0.5, nrow(g$data)),
    force_push      = g$force * 1e-6,
    force_pull      = g$force_pull * 1e-2,
    max_time        = g$max.time,
    max_iter        = ifelse(is.infinite(g$max.iter), 1e9, g$max.iter),
    max_overlaps    = g$max.overlaps,
    direction       = g$direction,
    verbose         = g$verbose
  )

  if (any(repel$too_many_overlaps)) {
    warning(sum(repel$too_many_overlaps, na.rm = TRUE),
            ' glyphs have too many overlaps.\nConsider increasing "max.overlaps"')
  }

  # if (all(repel$too_many_overlaps)) {
  #   grobs <- list()
  #   class(grobs) <- "gList"
  #   return(setChildren(x, grobs))
  # }

  # create segment grobs
  segg <- lapply(seq_along(g$data$x), function(i) {

    if (!repel$too_many_overlaps[i]) {
      row <- g$data[i, , drop = FALSE]
      grid::curveGrob(x1 = repel[i,]$x, y1 = repel[i,]$y, x2 = row$x, y2 = row$y,
                      default.units = "native",
                      curvature = row$segment.curvature,
                      angle = row$segment.angle,
                      ncp = row$segment.ncp,
                      shape = row$segment.shape,
                      square = row$segment.square,
                      squareShape = row$segment.squareShape,
                      inflect = row$segment.inflect,
                      debug = row$segment.debug,
                      gp = gpar(col = row$segment.colour,
                                lwd = row$segment.size * ggplot2::.pt,
                                lty = row$segment.linetype),
                      arrow = row$arrow)
    } else {
      grid::nullGrob()
    }
  })

  }

  gl <- lapply(seq_along(g$data$x),
               function(i) tileglyphGrob(x = if (g$repel) {
                 repel$x[i]
               } else {
                 g$data$x[i]
               },
               y = if (g$repel) {
                 repel$y[i]
               } else {
                 g$data$y[i]
               },
               z = unlist(g$data[i, g$cols]),
               size = g$data$size[i],
               ratio = g$ratio,
               nrow = g$nrow,
               fill = unlist(g$gdata[i, ]),
               col = g$colour,
               lwd = g$data$linewidth[i],
               alpha = g$data$alpha[i],
               linejoin = "mitre"))

  if (g$repel) {

    gl <- lapply(gl, grobTree)

    if (repel.debug) {

      gl <- lapply(seq_along(gl), function(i) grid::addGrob(gl[[i]], glorg[[i]]))

      gl <- lapply(seq_along(gl), function(i) grid::addGrob(gl[[i]], bboxg[[i]]))

      gl <- lapply(seq_along(gl), function(i) grid::addGrob(gl[[i]], segg[[i]]))

      # reorder grobs
      gl <- lapply(seq_along(gl),
                   function(i) grid::reorderGrob(gl[[i]], c(2:4, 1)))

    } else {

      gl <- lapply(seq_along(gl), function(i) grid::addGrob(gl[[i]], segg[[i]]))

      # reorder grobs
      gl <- lapply(seq_along(gl),
                   function(i) grid::reorderGrob(gl[[i]], c(2, 1)))

    }

  }

  gl <- do.call(grid::gList, gl)

  grid::setChildren(g, gl)
}
aravind-j/gglyph documentation built on Aug. 2, 2022, 12:30 a.m.