Align: new method to add schemes

Yunuuuu · Dec 11, 2024 · 05a94c8 · 05a94c8
1 parent 285fe63
commit 05a94c8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/R/align-.R b/R/align-.R
@@ -272,7 +272,7 @@ Align <- ggproto("Align", AlignProto,
             object_name
         )
     },
-    build = function(self, plot, direction, position, schemes,
+    build = function(self, plot, direction, position,
                      coords, extra_coords, previous_coords = NULL) {
         panel <- .subset2(coords, "panel")
         index <- .subset2(coords, "index")
@@ -333,6 +333,8 @@ Align <- ggproto("Align", AlignProto,
         )
     },
 
+    add_schemes = function(plot, schemes) plot_add_schemes(plot, schemes),
+
     # Most parameters for the `Align` are taken automatically from `compute()`,
     # `layout()` and `draw()`. However, some additional parameters may be
     # removed in `setup_params`. You should put these paramters here, otherwise,

diff --git a/R/plot-align-.R b/R/plot-align-.R
@@ -115,16 +115,15 @@ plot_build.ggalign_align_plot <- function(plot, ..., direction, schemes) {
     ans <- align$build(
         plot = plot@plot,
         ...,
-        direction = direction,
-        schemes = schemes
+        direction = direction
     )
 
     # remove axis titles, text, ticks used for alignment
     if (isTRUE(plot@no_axes)) {
         schemes$scheme_theme <- .subset2(schemes, "scheme_theme") +
             theme_no_axes(switch_direction(direction, "y", "x"))
     }
-    plot_add_schemes(ans, schemes)
+    align$add_schemes(ans, schemes)
 }
 
 ggproto_formals <- function(x) formals(environment(x)$f)