Reverting #697 (#722)

NAMESPACE

@@ -1,5 +1,7 @@
 # Generated by roxygen2: do not edit by hand
 
+S3method(plot,gtable)
+S3method(print,gtable)
 export("%>%")
 export(.CohortSizeConst)
 export(.CohortSizeDLT)
@@ -565,6 +567,7 @@ importFrom(graphics,lines)
 importFrom(graphics,matlines)
 importFrom(graphics,matplot)
 importFrom(graphics,plot)
+importFrom(grid,grid.draw)
 importFrom(gridExtra,arrangeGrob)
 importFrom(kableExtra,add_footnote)
 importFrom(kableExtra,add_header_above)

R/crmPack-package.R

@@ -30,6 +30,7 @@
 #' @import ggplot2
 #' @import methods
 #' @import tibble
+#' @importFrom grid grid.draw
 #' @importFrom gridExtra arrangeGrob
 #' @importFrom graphics plot hist legend lines matlines matplot
 #' @importFrom stats binomial coef cov2cor gaussian glm lm median model.matrix

R/helpers.R

@@ -219,6 +219,28 @@ crmPackHelp <- function() {
   utils::help(package = "crmPack", help_type = "html")
 }
 
+#' Plot `gtable` Objects
+#'
+#' This is needed because `crmPack` uses [gridExtra::arrangeGrob()] to combine
+#' `ggplot2` plots, and the resulting `gtable` object is not plotted otherwise
+#' when implicitly printing it in the console, e.g.
+#'
+#' @method plot gtable
+#' @param x (`gtable`)\cr object to plot.
+#' @param ... additional parameters for [grid::grid.draw()].
+#'
+#' @export
+plot.gtable <- function(x, ...) {
+  grid::grid.draw(x, ...)
+}
+
+#' @method print gtable
+#' @rdname plot.gtable
+#' @export
+print.gtable <- function(x, ...) {
+  plot(x, ...)
+}
+
 ##' Taken from utils package (print.vignette)
 ##'
 ##' @importFrom tools file_ext

_pkgdown.yaml

@@ -452,6 +452,7 @@ reference:
       - plot,SimulationsSummary,missing-method
       - plotDualResponses
       - plotGain
+      - plot.gtable
       - probit
       - set_seed
       - show,DualSimulationsSummary-method

tests/testthat/test-helpers.R

@@ -724,3 +724,19 @@ test_that("h_group_data works as expected", {
   )
   expect_setequal(combo_data_from_group, combo_data_from_start)
 })
+
+# print.gtable ----
+
+test_that("print for gtable works", {
+  result <- gridExtra::arrangeGrob(grid::rectGrob(), grid::rectGrob())
+  assert_class(result, "gtable")
+  vdiffr::expect_doppelganger("print-gtable", result)
+})
+
+# plot.gtable ----
+
+test_that("plot for gtable works", {
+  result <- gridExtra::arrangeGrob(grid::rectGrob(), grid::rectGrob())
+  assert_class(result, "gtable")
+  vdiffr::expect_doppelganger("plot-gtable", plot(result))
+})