R/plot.visualisation_recipe.R

In easystats/see: Model Visualisation Toolbox for 'easystats' and 'ggplot2'

#' @export
plot.see_visualisation_recipe <- function(x, ...) {
  if ("ggraph" %in% names(attributes(x))) {
    insight::check_if_installed("ggraph")
    if (!"ggraph" %in% .packages()) {
      attachNamespace("ggraph") # Needs to be attached
    }
    suppressWarnings(
      ggraph::ggraph(attributes(x)$data, layout = attributes(x)$layout) +
        geoms_from_list(x)
    )
  } else {
    global_aes <- attributes(x)$global_aes
    if (!is.null(global_aes) && length(global_aes)) {
      global_aes <- do.call(
        ggplot2::aes,
        args = lapply(global_aes, .str_to_sym)
      )
    }
    suppressWarnings(
      do.call(
        ggplot2::ggplot,
        insight::compact_list(list(
          data = attributes(x)$data,
          mapping = global_aes
        ))
      ) +
        geoms_from_list(x, ...)
    )
  }
}

# Example
# vr1 <- datawizard::visualisation_recipe(correlation::correlation(iris))
# vr2 <- datawizard::visualisation_recipe(summary(correlation::correlation(iris)))
# x <- list(p1 = vr1, p2 = vr2)
# class(x) <- "see_visualisation_recipes"
# plot(x)
#' @export
plot.see_visualisation_recipes <- function(x, ...) {
  the_plots <- list()
  for (i in names(x)) {
    the_plots[[i]] <- suppressWarnings(graphics::plot(x[[i]]))
  }
  pw <- plots(the_plots, ...)
  .safe_print_plots(pw)
  invisible(pw)
}