是否可以让 `dput` 返回在封闭环境之外运行的源代码？

Question

假设我有一个闭包add_y(y) ，它返回一个 function ，它将y添加到其输入中。

add_y <- function(y) {
  function(x) {
    x + y
  }
}
add_4 <- add_y(4)

所以add_4的值是一个 function ，它的输入加 4。 这行得通。 我想使用 dput 将dput的定义add_4为

function(x) {
  x + 4
}

但这不是 dput 返回的。

add_y <- function(y) {
  function(x) {
    x + y
  }
}
add_4 <- add_y(4)
dput(add_4)
#> function (x) 
#> {
#>     x + y
#> }

有没有办法获得可以在封闭环境之外运行的源代码？

Answer 1

这可以工作，但它涉及更改add_y的内容。

library(rlang)
library(magrittr)
library(stringr)

add_y <- function(y) {
  fn <- expr(function(x) {
    x+!!y
  })
  fn <- deparse(fn) %>% str_c(collapse = "")
  fn <- eval(parse(text = fn))
}

add_4 <- add_y(4)

dput(add_4)
#> function (x) 
#> {
#>     x + 4
#> }

^{由代表 package (v2.0.1) 于 2021 年 12 月 24 日创建}

Answer 2

您可以构建一个dput替换，该替换生成的代码创建一个 function 就像add_4一样，但它不会按照您想要的方式进行解析：

dput_with_env <- function(f) {
  fn <- deparse(f, control = c("keepNA", "keepInteger", 
                               "niceNames", "showAttributes"))
  env <- as.list(environment(f))
  cat("local({ f =\n")
  cat(fn, sep = "\n")
  cat("\nenvironment(f) <- list2env(\n")
  dput(env)
  cat(")\nf})")
}

add_y <- function(y) {
  function(x) {
    x + y
  }
}
add_4 <- add_y(4)

dput_with_env(add_4)
#> local({ f =
#> function (x) 
#> {
#>     x + y
#> }
#> 
#> environment(f) <- list2env(
#> list(y = 4)
#> )
#> f})

^{由代表 package (v2.0.1) 于 2021 年 12 月 24 日创建}

这假设add_4的环境非常简单，因此其环境的父级可以是您评估代码时所在的环境。 我们可以试一试：

newfn <- local({ f =
function (x) 
{
   x + y
}
environment(f) <- list2env(
list(y = 4)
)
f})

newfn
#> function (x) 
#> {
#>    x + y
#> }
#> <environment: 0x7f9a1b5e2318>
newfn(1)
#> [1] 5

^{由代表 package (v2.0.1) 于 2021 年 12 月 24 日创建}

Answer 3

没有dput() ，不。 dput() function 不会创建环境的文本表示。

如果你想保存 function，你可以这样做

save(add_4, file="add4.Rdata")

然后在另一个 R session

load("add4.Rdata")

这将捕获所有封闭的值，并且您的 function 将像以前一样运行

Answer 4

如果您控制 add_y 那么解决方法是将 y 的值直接注入内部 function 的主体中，或者将其注入正式参数列表中。 这消除了环境的使用，因此问题不再存在。 这涉及到命名匿名内部 function 并且只有一行额外的行来执行注入加上一行来返回结果。

# 1. inject into body
add_y2 <- function(y) {
  inner <- function(x) {
    x + y
  }
  body(inner) <- do.call("substitute", list(body(inner)))
  inner
}
# test
add_4 <- add_y2(4)
dput(add_4)
## function (x) 
## {
##     x + 4
## }

# 2. inject into formal arguments
add_y3 <- function(y) {
  inner <- function(x) {
    x + y
  }
  formals(inner)$y <- y
  inner
}
# test
add_4 <- add_y3(4)
dput(add_4)
## function (x, y = 4) 
## {
##     x + y
## }