如何为修改后的ggplot对象添加scale_x_*操作警告？

要给修改后的ggplot对象添加告警，提示或阻止用户执行scale_x_*类操作，可以通过自定义S3类+重载+操作符实现，具体步骤如下：

1. 修改原函数，标记自定义绘图对象 #

给修改后的ggplot对象添加一个自定义S3类，方便后续识别并拦截操作：

library(ggplot2)

p = ggplot(mtcars, aes(x=cyl, y=mpg)) + geom_point()

mod_gg <- function(plot) {
  plot$labels$x = "CYL"
  # 给对象添加自定义类，保留原有ggplot类以兼容其他操作
  class(plot) <- c("modified_gg", class(plot))
  return(plot)
}

pnew = mod_gg(p)

2. 重载+操作符，拦截X轴刻度操作 #

自定义+.modified_gg方法，检测用户是否添加X轴相关刻度，并触发告警：

`+.modified_gg` <- function(e1, e2) {
  # 精准判断是否为X轴刻度操作（匹配scale_x_discrete/scale_x_continuous等）
  if (inherits(e2, c("ScaleDiscreteX", "ScaleContinuousX"))) {
    # 弹出警告（如果要直接禁止操作，将warning替换为stop）
    warning("已对该绘图的X轴标签进行自定义修改，添加X轴刻度会覆盖原有设置，请谨慎操作。", call. = FALSE)
  }
  # 调用ggplot原生的+逻辑，保证其他操作正常执行
  NextMethod()
}

3. 测试效果 #

当用户尝试添加X轴刻度时，会自动弹出告警：

> pnew = pnew + scale_x_discrete(limits = c(5,7))
Warning message:
已对该绘图的X轴标签进行自定义修改，添加X轴刻度会覆盖原有设置，请谨慎操作。 

如果需要直接禁止该操作，将代码中的warning替换为stop即可，此时用户执行scale_x_*操作会直接报错终止。

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