R/heatmap.R

In CRI-iAtlas/iatlas-app: What the Package Does (One Line, Title Case)

create_heatmap <- function(corr_mat, source_name, scale_colors = F,  legend_title = NULL){
  zmin <- NULL
  zmax <- NULL
  if(scale_colors){
    extreme <- max(abs(min(corr_mat)),
                   abs(max(corr_mat)))
    zmax <- extreme
    zmin <- -extreme
  }

  p <-
    plotly::plot_ly(
      z = corr_mat,
      x = colnames(corr_mat),
      y = rownames(corr_mat),
      type = "heatmap",
      source = source_name,
      colors = rev(RColorBrewer::brewer.pal(8, "RdBu")),
      colorbar = list(title = legend_title),
      zmin = zmin,
      zmax = zmax
    ) %>%
    plotly::layout(xaxis = list(tickangle = 90)) %>%
    format_plotly() %>%
    I
}