R/combo_heatmap.R

In daranzolin/pmapply: Apply a function across pairs of vectors

#' Plot heatmap result of pmapply
#'
#' @param combo_mat a matrix generated from `pmapply`
#' @param gradient_low color for low values
#' @param gradient_high color for high values
#'
#' @return
#' @export
#'
#' @examples
#'\dontrun{
#' cormat <- pmapply(iris[,c(1:4)], function(x, y) cor(x, y))
#' combo_heatmap(cormat)
#'}
combo_heatmap <- function(combo_mat, gradient_low = "lightgrey", gradient_high = "red") {
  mm <- get_minmax_combos(combo_mat)
  pd <- reshape2::melt(combo_mat)
  p <- ggplot2::ggplot(pd, ggplot2::aes(x = Var2, y = Var1)) +
    ggplot2::geom_raster(ggplot2::aes(fill = value)) +
    ggplot2::scale_fill_gradient(low = gradient_low,
                                 high = gradient_high,
                                 limits = c(mm$min_value, mm$max_value)
                                 )
  print(p)
  invisible(p)
}