R/create_heatmap.r

In SKDE-Felles/SKDEresultater: R-package to produce shiny SKDE portal

#' Title
#'
#' @param data Data to be plotted
#' @param type R-package to generate the plot. c("plotly", "heatmaply")
#'
#' @return A plotly heatmap
#' @importFrom rlang .data
#' @export
#'
create_heatmap <- function(data = NULL, type = "plotly") {
    if ("level3_name" %in% colnames(data)) {
        tmp <- tidyr::unite(data, .data$level1_name, .data$level2_name, .data$level3_name, col = "combined", sep = ": ")
    } else if ("level2_name" %in% colnames(data)) {
        tmp <- tidyr::unite(data, .data$level1_name, .data$level2_name, col = "combined", sep = ": ")
    } else if ("level1_name" %in% colnames(data)) {
        tmp <- data
        tmp$combined <- tmp$level1_name
    }
    spread_data <- tidyr::spread(tmp[, c("area", "area_name", "combined", "value")],
                                 key = .data$combined,
                                 value = .data$value
    )
    spread_data <- dplyr::filter(spread_data, .data$area != 8888)
    row.names(spread_data) <- spread_data$area_name
    spread_data$area_name <- NULL
    spread_data$area <- NULL
    if (type == "plotly") {
        heatmap <- plotly::plot_ly(
            y = row.names(spread_data),
            x = names(spread_data),
            z = abs(scale(spread_data)),
            type = "heatmap",
            colors = c("white", SKDEr::skde_colors(num = 7)),
            showscale = FALSE
        ) %>%
            plotly::config(displayModeBar = FALSE)
    } else if (type == "heatmaply") {
        heatmap <-  heatmaply::heatmaply(
            spread_data,
            plot_method = "plotly",
            dendrogram = "none",
            scale = "column",
            column_text_angle = 45,
            colors = c("white", SKDEr::skde_colors(num = 7)),
            grid_gap = 1,
            hide_colorbar = TRUE
        ) %>%
            plotly::config(displayModeBar = FALSE)
    }
    return(heatmap)
}