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)
}