干净的方式在rlang中进行嵌套的懒惰评估

时间:2017-05-17 11:21:59

标签: r lazy-evaluation tidyverse

假设我有一个函数f,它带有一堆参数,以及一个可选的额外参数。

f <- function(..., extra)
{
    arglst <- lapply(quos(...), get_expr)
    if(!missing(extra))
    {
        extra <- get_expr(enquo(extra))
        arglst <- c(arglst, extra=extra)
    }
    arglst
    ## ... do something with argument list ... ##
}

f(a, extra=foo)
# [[1]]
# a
# 
# $extra
# foo

请注意,我不想这样评估参数,但我确实想要获取传入的表达式,以便由其他代码进行评估。

新的rlang软件包(为下一版本的dplyr提供支持,将在CRAN Real Soon Now上发布)为我在f上面使用的延迟评估提供了广泛的工具。例如,quosget_exprenquo都是来自rlang的函数。

f中,我处理extra的部分实际上是样板代码:我想在其他函数中执行此操作,而不仅仅是f。我不想每次都重写它,所以我想我会把它放到它自己的功能中:

doExtra <- function(arglst, extra)
{
    if(!missing(extra))
    {
        extra <- get_expr(enquo(extra))
        arglst <- c(arglst, extra=extra)
    }
    arglst
}

f2 <- function(..., extra)
{
    arglst <- lapply(quos(...), get_expr)
    arglst <- doExtra(arglst, extra)
    arglst
}

问题在于,当我这样做时,extra看到的doExtra的价值是从f2传来的,而不是原来的:

f2(a, extra=foo)
# [[1]]
# a
#
# $extra
# extra

如何修改f以隔离样板代码,而不会得到错误的结果?我可以做一些事情,比如直接操纵doExtra调用框架的环境,但那会非常难看。

1 个答案:

答案 0 :(得分:5)

  • 要将命名参数转发给另一个enquoting函数,您必须引用然后取消引用:!! enquo(arg)。如果你只是传递enquo(arg),那么enquoting函数就会看到:enquo(arg)。如果你传递参数符号,它也会看到它。这就是为什么你需要在它捕获的参数中取消引用。

    !! enquo(arg)触发enquo(arg)的评估,它返回提供给arg参数的表达式。然后它在您的函数捕获的参数内不加引号。

  • 如果您要引用可能缺少的参数,最好引用它,然后使用quo_is_missing()检查缺失情况。 Enquoting缺少参数会通过调用quo()而不带参数来创建返回的同一对象。

  • 如果您不需要提交,则可以使用exprs()enexpr()。然而,你正在失去环境,而你正在进一步评估脆弱。

    如果您以某种其他方式捕获环境以使用base::eval()或类似方式对其进行评估,请注意,quosures可以包含其他quosures。只有eval_tidy()才能理解这些嵌套的状态。

IIUC你的问题,是关于传递一个应该引用另一个函数的论证。一种方法是在第一个函数中捕获,然后通过值传递给第二个函数:

library("purrr")
library("rlang")

f <- function(..., extra) {
  exprs <- exprs(...)

  # Pass the enquoted argument by value
  exprs <- extra_by_value(exprs, enexpr(extra))

  exprs
}
extra_by_value <- function(exprs, extra) {
  if (!is_missing(extra)) {
    c(exprs, extra = extra)
  } else {
    exprs
  }
}

如果第二个函数必须通过表达而不是值(可能是因为它是另一个面向用户的动词),你必须取消引用被引用的表达式:

f <- function(..., extra) {
  exprs <- exprs(...)

  # Since the argument is captured by the function, we need
  # to unquote the relevant expression into the argument:
  exprs <- extra_by_expression(exprs, !! enexpr(extra))

  exprs
}
extra_by_expression <- function(exprs, extra) {
  extra <- enexpr(extra)
  if (!is_missing(extra)) {
    c(exprs, extra = extra)
  } else {
    exprs
  }
}

所有这些概念都适用于quosures。这是等效的代码:

f <- function(..., extra) {
  quos <- quos(...)

  # Since the argument is captured by the function, we need
  # to unquote the relevant expression into the argument:
  quos <- extra_by_expression(quos, !! enquo(extra))

  quos
}
extra_by_expression <- function(quos, extra) {
  extra <- enquo(extra)
  if (!quo_is_missing(extra)) {
    c(quos, extra = extra)
  } else {
    quos
  }
}

使用quosures几乎总是比原始表达式更好,因为它们会跟踪它们的上下文。