New x_limits argument

DESCRIPTION

@@ -1,7 +1,7 @@
 Type: Package
 Package: see
 Title: Model Visualisation Toolbox for 'easystats' and 'ggplot2'
-Version: 0.8.0.4
+Version: 0.8.0.5
 Authors@R: 
     c(person(given = "Daniel",
              family = "Lüdecke",

NEWS.md

@@ -20,6 +20,8 @@
   QQ-plot. Previous behavior can be restored by setting `type = "density"`.
 * Plot method for `binned_residuals()` gains a `show_smooth` argument, to show
   or hide the smooth line.
+* Plot method for `check_predictions()` gains a `x_limits` argument, to limit
+  the x-axis-range. This can be useful to "zoom in" certain parts of the plot.
 
 # see 0.8.0
 

R/plot.check_predictions.R

@@ -20,12 +20,12 @@
   dataplot$grp <- rep(seq_len(ncol(x)), each = nrow(x))
 
   attr(dataplot, "info") <- list(
-    "xlab" = attr(x, "response_name"),
-    "ylab" = ifelse(identical(type, "density"), "Density", "Counts"),
-    "title" = "Posterior Predictive Check",
-    "check_range" = attr(x, "check_range"),
-    "bandwidth" = attr(x, "bandwidth"),
-    "model_info" = attr(x, "model_info")
+    xlab = attr(x, "response_name"),
+    ylab = ifelse(identical(type, "density"), "Density", "Counts"),
+    title = "Posterior Predictive Check",
+    check_range = attr(x, "check_range"),
+    bandwidth = attr(x, "bandwidth"),
+    model_info = attr(x, "model_info")
   )
 
   class(dataplot) <- unique(c("data_plot", "see_performance_pp_check", class(dataplot)))
@@ -45,13 +45,15 @@
 #' (default), `"discrete_dots"`, `"discrete_interval"` or `"discrete_both"` (the
 #' `discrete_*` options are appropriate for models with discrete - binary, integer
 #' or ordinal etc. - outcomes).
+#' @param x_limits Numeric vector of length 2 specifying the limits of the x-axis.
+#' If not `NULL`, will zoom in the x-axis to the specified limits.
 #' @inheritParams data_plot
 #' @inheritParams plot.see_check_normality
 #' @inheritParams plot.see_parameters_distribution
 #'
 #' @return A ggplot2-object.
 #'
 #' @seealso See also the vignette about [`check_model()`](https://easystats.github.io/performance/articles/check_model.html).
 #'
 #' @examplesIf require("performance")
 #' model <- lm(Sepal.Length ~ Species * Petal.Width + Petal.Length, data = iris)
@@ -75,12 +77,13 @@
                                            style = theme_lucid,
                                            colors = unname(social_colors(c("green", "blue"))),
                                            type = c("density", "discrete_dots", "discrete_interval", "discrete_both"),
+                                           x_limits = NULL,
                                            ...) {
   orig_x <- x
   check_range <- isTRUE(attributes(x)$check_range)
   plot_type <- attributes(x)$type
 
   if (missing(type) && !is.null(plot_type) && plot_type %in% c("density", "discrete_dots", "discrete_interval", "discrete_both")) {
     type <- plot_type
   } else {
     type <- match.arg(type)
@@ -90,7 +93,16 @@
     x <- data_plot(x, type)
   }
 
-  p1 <- .plot_pp_check(x, size_line, line_alpha, theme_style = style, colors = colors, type = type, ...)
+  p1 <- .plot_pp_check(
+    x,
+    size_line,
+    line_alpha,
+    theme_style = style,
+    colors = colors,
+    type = type,
+    x_limits = x_limits,
+    ...
+  )
 
   if (isTRUE(check_range)) {
     p2 <- .plot_pp_check_range(orig_x, size_bar, colors = colors)
@@ -112,12 +124,13 @@
                                           style = theme_lucid,
                                           colors = unname(social_colors(c("green", "blue"))),
                                           type = c("density", "discrete_dots", "discrete_interval", "discrete_both"),
+                                          x_limits = NULL,
                                           ...) {
   orig_x <- x
   check_range <- isTRUE(attributes(x)$check_range)
   plot_type <- attributes(x)$type
 
   if (missing(type) && !is.null(plot_type) && plot_type %in% c("density", "discrete_dots", "discrete_interval", "discrete_both")) {
     type <- plot_type
   } else {
     type <- match.arg(type)
@@ -127,7 +140,16 @@
     x <- data_plot(x, type)
   }
 
-  p1 <- .plot_pp_check(x, size_line, line_alpha, theme_style = style, colors = colors, type = type, ...)
+  p1 <- .plot_pp_check(
+    x,
+    size_line,
+    line_alpha,
+    theme_style = style,
+    colors = colors,
+    type = type,
+    x_limits = x_limits,
+    ...
+  )
 
   if (isTRUE(check_range)) {
     p2 <- .plot_pp_check_range(orig_x, size_bar, colors = colors)
@@ -139,7 +161,7 @@
 
 
 
-.plot_pp_check <- function(x, size_line, line_alpha, theme_style, colors, type = "density", ...) {
+.plot_pp_check <- function(x, size_line, line_alpha, theme_style, colors, type = "density", x_limits = NULL, ...) {
   info <- attr(x, "info")
 
   # default bandwidth, for smooting
@@ -182,6 +204,10 @@
     )
   }
 
+  if (!is.null(x_limits)) {
+    out <- out + ggplot2::coord_cartesian(xlim = x_limits)
+  }
+
   out
 }
 
@@ -393,13 +419,13 @@
 
   replicated <- rbind(
     data.frame(
-      x = sapply(x[which(names(x) != "y")], min),
+      x = vapply(x[which(names(x) != "y")], min, numeric(1)),
       group = "Minimum",
       color = "Model-predicted data",
       stringsAsFactors = FALSE
     ),
     data.frame(
-      x = sapply(x[which(names(x) != "y")], max),
+      x = vapply(x[which(names(x) != "y")], max, numeric(1)),
       group = "Maximum",
       color = "Model-predicted data",
       stringsAsFactors = FALSE