make stat_function() work with empty input data. fixes tidyverse#3983.

Author: clauswilke

R/geom-function.R

@@ -18,7 +18,7 @@
 #'   geom_function(fun = dnorm, colour = "red")
 #'
 #' # To plot functions without data, specify range of x-axis
-#' base <- ggplot(data.frame(x = c(-5, 5)), aes(x))
+#' base <- ggplot() + xlim(-5, 5)
 #' base + geom_function(fun = dnorm)
 #' base + geom_function(fun = dnorm, args = list(mean = 2, sd = .5))
 #'
@@ -45,8 +45,11 @@ geom_function <- function(mapping = NULL, data = NULL, stat = "function",
                           position = "identity", ..., na.rm = FALSE,
                           show.legend = NA, inherit.aes = TRUE) {
   # Warn if supplied data is going to be overwritten
-  if (!is.null(data) && identical(stat, "function")) {
-    warn("`data` is not used by stat_function()")
+  if (identical(stat, "function")) {
+    if (!is.null(data)) {
+      warn("`data` is not used by stat_function()")
+    }
+    data <- ensure_nonempty_data
   }
 
   layer(

R/stat-function.r

@@ -29,6 +29,7 @@ stat_function <- function(mapping = NULL, data = NULL,
   if (!is.null(data)) {
     warn("`data` is not used by stat_function()")
   }
+  data <- ensure_nonempty_data
 
   layer(
     data = data,
@@ -57,15 +58,21 @@ StatFunction <- ggproto("StatFunction", Stat,
   default_aes = aes(y = after_scale(y)),
 
   compute_group = function(data, scales, fun, xlim = NULL, n = 101, args = list()) {
-    range <- xlim %||% scales$x$dimension()
-    xseq <- seq(range[1], range[2], length.out = n)
-
-    if (scales$x$is_discrete()) {
+    if (is.null(scales$x)) {
+      range <- xlim %||% c(0, 1)
+      xseq <- seq(range[1], range[2], length.out = n)
       x_trans <- xseq
     } else {
-      # For continuous scales, need to back transform from transformed range
-      # to original values
-      x_trans <- scales$x$trans$inverse(xseq)
+      range <- xlim %||% scales$x$dimension()
+      xseq <- seq(range[1], range[2], length.out = n)
+
+      if (scales$x$is_discrete()) {
+        x_trans <- xseq
+      } else {
+        # For continuous scales, need to back transform from transformed range
+        # to original values
+        x_trans <- scales$x$trans$inverse(xseq)
+      }
     }
 
     if (is.formula(fun)) fun <- as_function(fun)
@@ -82,3 +89,15 @@ StatFunction <- ggproto("StatFunction", Stat,
     ))
   }
 )
+
+# Convenience function used by `stat_function()` and
+# `geom_function()` to convert empty input data into
+# non-empty input data without touching any non-empty
+# input data that may have been provided.
+ensure_nonempty_data <- function(data) {
+  if (empty(data)) {
+    new_data_frame(list(group = 1), n = 1)
+  } else {
+    data
+  }
+}