Merge pull request #105 from ilia-kats/fix_factor_plot_colorscale

plot_factors: fix continuous color scale when plotting >2 factors

Author: jan-spr

R/plot_factors.R

@@ -420,14 +420,22 @@ plot_factors <- function(object, factors = c(1, 2), groups = "all",
       legend.text = element_text(size=rel(1.2)),
       legend.title = element_text(size=rel(1.2))
     )
-  if (length(unique(df$color))>1 && isTRUE(legend)) { p <- p + labs(color=color_name) } else { p <- p + guides(color="none") + scale_color_manual(values="black") }
+
+  colorscale <- NULL
+  if (length(unique(df$color))>1 && isTRUE(legend)) {
+    p <- p + labs(color=color_name)
+  } else {
+    p <- p + guides(color="none")
+    colorscale <- scale_color_manual(values="black")
+  }
   if (length(unique(df$color))>1 && isFALSE(legend)) { p <- p + guides(color="none") }
-  if (is.numeric(df$color)) p <- p + scale_color_gradientn(colors=colorRampPalette(rev(brewer.pal(n=5, name="RdYlBu")))(10)) 
+  if (is.numeric(df$color)) colorscale <- scale_color_gradientn(colors=colorRampPalette(rev(brewer.pal(n=5, name="RdYlBu")))(10))
   if (length(unique(df$shape))>1) { p <- p + labs(shape=shape_name) } else { p <- p + guides(shape = "none") }
+  if (!is.null(colorscale)) p <- p + colorscale
   if ((length(unique(df$color))>1 || length(unique(df$shape))>1) && isTRUE(legend)) { legend <- GGally::grab_legend(p) } else { legend <- NULL }
 
   # Generate plot
-  p <- GGally::ggpairs(df, 
+  p <- GGally::ggpairs(df,
     columns = factors,
     lower = list(continuous=GGally::wrap("points", size=dot_size)), 
     diag = list(continuous='densityDiag'), 
@@ -441,7 +449,8 @@ plot_factors <- function(object, factors = c(1, 2), groups = "all",
     axis.ticks = element_blank(),
     axis.text = element_blank()
   )
-  
+  if (!is.null(colorscale)) p <- p + colorscale
+
   return(p)
 }