group argument added to geom_linerange #45

kassambara · Oct 30, 2017 · d452881 · d452881
1 parent 5ede873
commit d452881
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/R/ggdotchart.R b/R/ggdotchart.R
@@ -122,12 +122,16 @@ ggdotchart_core <- function(data, x, y, group = NULL,
                        ...)
 {
   add <- match.arg(add)
-
   if(!is.null(group)){
     if(group == 1)
       group <- NULL
   }
 
+  if(is.null(group) & color[1] %in% names(data)){
+    group <- color[1]
+  }
+
+
   .dots <- list(...)
   sorting <- match.arg(sorting)
   decreasing <- ifelse(sorting == "descending", FALSE, TRUE)
@@ -157,7 +161,7 @@ ggdotchart_core <- function(data, x, y, group = NULL,
                           size = size, position = position)
 
     mapping <- seg.opts$mapping %>%
-      .add_item(x = x, ymin = 0, ymax = y)
+      .add_item(x = x, ymin = 0, ymax = y, group = group)
     option <- seg.opts$option
 
     # mapping <- seg.opts$mapping %>%