Retain input factor levels for splitGeneGroups

morinlab · Dec 18, 2024 · 66de5bb · 66de5bb
1 parent 6a6a606
commit 66de5bb
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/R/prettyOncoplot.R b/R/prettyOncoplot.R
@@ -907,10 +907,14 @@ prettyOncoplot = function(
     }
 
 
-    if(missing(splitGeneGroups)){
-        row_split = rep("", length(genes))
-    }else{
-        row_split = factor(splitGeneGroups[genes], levels = unique(splitGeneGroups[genes]))
+    if (missing(splitGeneGroups)) {
+        row_split <- rep("", length(genes))
+    } else {
+        if(is.factor(splitGeneGroups)){
+            row_split <- splitGeneGroups[genes]
+        } else(
+            row_split <- factor(splitGeneGroups[genes], levels = unique(splitGeneGroups[genes]))
+        )
     }
 
     if(!missing(groupNames)){