R/plotModules.R

In milescsmith/scWGCNA: scWGCNA

#' @title plotModules
#'
#' @description Use a bubbleplot to display expression levels and percentage for
#' genes identified as functioning within a network module.
#'
#' @param object scRNA-seq data object
#' @param modules Gene expression modules to plot, as determined by
#' seuratClusterWGCNA
#' @param ... Additional arguments to pass to bubbleplot
#'
#' @import patchwork
#' @importFrom ggplot2 labs xlab ylab theme aes
#' @importFrom seuratBubblePlot bubbleplot
#' @importFrom gridExtra grid.arrange
#' @importFrom purrr reduce
#'
#' @return gg
#' @export
#'
#' @examples
plotModules <- function(object, modules, ...) {
  modplots <- lapply(
    (1:length(modules)),
    function(x) {
      bp <- bubbleplot(
        object = object,
        genes.plot = modules[[x]],
        do.return = TRUE,
        ...
      ) +
        labs(title = names(modules)[[x]],
             x = NULL,
             y = NULL) +
        theme(legend.position = "none")
    }
  )

  reduce(modplots, `+`)
}