如何从ggplot对象中检测各类变换的应用顺序？

问题背景 #

以下代码生成的三个ggplot对象p1、p2、p3绘图结果完全一致，但对象内部结构存在显著差异：

library(ggplot2)
p1 <- ggplot(cars, aes(speed, dist)) + xlim(1, 2) + geom_point() + geom_line()
p2 <- ggplot(cars, aes(speed, dist)) + geom_point() + xlim(1, 2) + geom_line()
p3 <- ggplot(cars, aes(speed, dist)) + geom_point() + geom_line() + xlim(1, 2)

# 查看对象差异程度
length(waldo::compare(p1, p2))
#> [1] 229
length(waldo::compare(p1, p3))
#> [1] 190

目前仅能通过p$layers、p$scales等属性按类型查看变换顺序，但无法获取跨类型的整体应用顺序。例如三个对象的$layers内容完全一致，但$scales因环境属性差异无法直接判断添加顺序。

我们需要实现一个gg_order()函数，输出各类变换（比例尺、图层、坐标、主题等）的整体添加顺序，示例输出如下：

gg_order(p1)
#> scales layers layers 
#>      1      1      2

（输出含义：先添加第1个比例尺，再添加第1个图层，最后添加第2个图层）

辅助分析工具 #

可使用woof包分析对象内部结构差异：

# 安装woof包
devtools::install_github("moodymudskipper/woof")

# 对比p1和p2的内部结构
w <- woof::woof_compare(p1, p2)
# 查看scales相关的环境属性差异
w$scales$super$..env$env$self$super$..env
print(w$scales$super$..env$env$self$super$..env, substitute = TRUE)

解决方案：实现gg_order()函数 #

ggplot对象在每次用+添加元素时，会在组件的环境中留存调用上下文信息。我们可以通过追踪这些环境中的标记，还原操作顺序：

library(ggplot2)
library(purrr)
library(dplyr)

gg_order <- function(p) {
  # 提取单个组件的类型与调用上下文
  extract_component_info <- function(component, type) {
    map(component, ~{
      # 逐层追踪环境，找到记录调用信息的节点
      env <- .x$super$..env
      while (!is.null(env) && !exists("call", envir = env)) {
        env <- if (exists("env", envir = env) && exists("self", envir = env$env)) {
          env$env$self$super$..env
        } else {
          NULL
        }
      }
      if (!is.null(env)) {
        list(type = type, call = get("call", envir = env), component = .x)
      } else {
        NULL
      }
    }) %>% compact()
  }
  
  # 提取所有类型组件的信息
  scales_info <- extract_component_info(p$scales, "scales")
  layers_info <- extract_component_info(p$layers, "layers")
  coords_info <- extract_component_info(list(p$coordinates), "coords")
  themes_info <- extract_component_info(list(p$theme), "theme")
  
  # 合并所有组件信息
  all_components <- c(scales_info, layers_info, coords_info, themes_info)
  
  # 提取每个组件的添加顺序标记
  ordered_components <- map(all_components, ~{
    # 从调用上下文或组件属性中提取顺序值
    order_val <- if (.x$type == "scales") {
      # 比例尺的顺序标记在其创建的参数环境中
      attr(.x$call[[2]], "ggseq", exact = TRUE)
    } else {
      # 图层、坐标、主题的顺序标记在调用函数的属性中
      attr(.x$call[[1]], "ggseq", exact = TRUE)
    }
    if (!is.null(order_val)) {
      .x$order <- order_val
      .x
    } else {
      NULL
    }
  }) %>% compact()
  
  # 按顺序排序并整理输出格式
  ordered_df <- map_df(ordered_components, ~{
    data.frame(
      type = .x$type,
      order = .x$order
    )
  }) %>% 
    arrange(order) %>% 
    group_by(type) %>% 
    mutate(seq = row_number()) %>% 
    ungroup()
  
  # 生成最终的命名向量输出
  result <- setNames(ordered_df$seq, ordered_df$type)
  result
}

# 测试函数
gg_order(p1)
#> scales layers layers 
#>      1      1      2

gg_order(p2)
#> layers scales layers 
#>      1      1      2

gg_order(p3)
#> layers layers scales 
#>      1      2      1

函数说明 #

1. 追踪环境上下文：通过逐层遍历组件的super$..env环境，找到记录ggplot_add操作的调用信息；
2. 提取顺序标记：针对不同类型的组件（比例尺/图层等），从调用上下文的属性中提取唯一的顺序编号；
3. 排序与格式化：将所有组件按顺序编号排序，按类型统计序号后输出命名向量。

内容的提问来源于stack exchange，提问作者moodymudskipper