Support conversion for geom_vline()

Author: mkcor

R/ggplotly.R

@@ -98,6 +98,10 @@ gg2list <- function(p){
   xrange <- sapply(ggranges, `[[`, "x.range", simplify=FALSE, USE.NAMES=FALSE)
   ggxmin <- min(sapply(xrange, min))
   ggxmax <- max(sapply(xrange, max))
+  # Extract y.range
+  yrange <- sapply(ggranges, `[[`, "y.range", simplify=FALSE, USE.NAMES=FALSE)
+  ggymin <- min(sapply(yrange, min))
+  ggymax <- max(sapply(yrange, max))
 
   # Get global size range because we need some of its info in layer2traces
   if ("size.name" %in% name.names) {
@@ -168,6 +172,9 @@ gg2list <- function(p){
     # Add global x-range info
     misc$prestats.data$globxmin <- ggxmin
     misc$prestats.data$globxmax <- ggxmax
+    # Add global y-range info
+    misc$prestats.data$globymin <- ggymin
+    misc$prestats.data$globymax <- ggymax
 
     # Add global size info if relevant
     if ("size.name" %in% name.names) {

R/trace_generation.R

@@ -121,7 +121,7 @@ layer2traces <- function(l, d, misc) {
   }
   ## Then split on visual characteristics that will get different
   ## legend entries.
-  data.list <- if(basic$geom %in% names(markLegends)){
+  data.list <- if (basic$geom %in% names(markLegends)) {
     mark.names <- markLegends[[basic$geom]]
     ## However, continuously colored points are an exception: they do
     ## not need a legend entry, and they can be efficiently rendered
@@ -152,11 +152,24 @@ layer2traces <- function(l, d, misc) {
     }
   }
 
+  # Split hline and vline when multiple
+  if (g$geom == "hline" || g$geom == "vline") {
+    if (nrow(g$data) > 1) {
+      df.list <- split(basic$data, rep(1:nrow(g$data))) #, drop=TRUE)
+      data.list <- lapply(df.list, function(df) {
+        params <- basic$params
+        list(data=df,
+             params=params)
+      })
+    }
+  }
+  
   ## case of no legend, if either of the two ifs above failed.
   if(is.null(data.list)){
     data.list <- structure(list(list(data=basic$data, params=basic$params)),
                            names=basic$params$name)
   }
+  
   getTrace <- geom2trace[[basic$geom]]
   if(is.null(getTrace)){
     warning("Conversion not implemented for geom_",
@@ -167,7 +180,8 @@ layer2traces <- function(l, d, misc) {
   }
   traces <- NULL
   names.in.legend <- NULL
-  for(data.i in seq_along(data.list)){
+  
+  for (data.i in seq_along(data.list)) {
     data.params <- data.list[[data.i]]
     data.params$params$stat.type <- l$stat$objname
     tr <- do.call(getTrace, data.params)
@@ -303,6 +317,11 @@ toBasic <- list(
     g$params$xend <- max(g$prestats.data$globxmax)
     g
   },
+  vline=function(g) {
+    g$params$ystart <- min(g$prestats.data$globymin)
+    g$params$yend <- max(g$prestats.data$globymax)
+    g
+  },
   point=function(g) {
     if ("size" %in% names(g$data)) {
       g$params$sizemin <- min(g$prestats.data$globsizemin)
@@ -527,5 +546,13 @@ geom2trace <- list(
          type="scatter",
          mode="lines",
          line=paramORdefault(params, aes2line, line.defaults))
+  },
+  vline=function(data, params) {
+    list(x=c(data$xintercept, data$xintercept),
+         y=c(params$ystart, params$yend),
+         name=params$name,
+         type="scatter",
+         mode="lines",
+         line=paramORdefault(params, aes2line, line.defaults))
   }
 )