Rqu的反向行为

时间:2018-01-01 21:21:24

标签: r metaprogramming

bquote函数允许评估包含在.()调用中的表达式部分。例如,

a <- 2
b <- 100
bquote(.(2 * a) * x + .(log10(b)))

将返回

4 * x + 2

我想重写此函数来评估除.()调用内部事物之外的所有内容。这是理想的行为:

a <- 2
b <- 100
bquote(2 * a * .(x) + log10(b))

> 4 * x + 2

我明白要这样做,我必须检查abstract syntax tree并评估没有.()调用的早午餐,但我无法处理所有这些递归。

你能帮我写一个这样的功能吗?

1 个答案:

答案 0 :(得分:5)

subst将替换除。(...)中的变量之外的所有变量,simplify函数将简化没有变量的子树 - 如果不需要简化,则省略简化部分。没有包使用。

subst <- function(e) {
   if (typeof(e) == "language") {
      if (identical(e[[1]], as.name("."))) e[[2]]
      else {
        if (length(e) > 1) e[-1] <- lapply(as.list(e[-1]), subst)
        e
      }
   } else {
      eval(e)
   }
}

simplify <- function(e) {
  if (typeof(e) == "language") {
     if (length(all.vars(e))) {
         if (length(e) > 1) {
           e[-1] <- lapply(as.list(e[-1]), simplify)
           e
         } else e
     } else eval(e)
  } else e
}

inverse_bquote <- function(x, SIMPLIFY = TRUE) {
   result <- subst(substitute(x))
   if (SIMPLIFY) simplify(result) else result
}

现在测试一下。

a <- 2
b <- 100

inverse_bquote(2 * a * .(x) + log10(b))
## 4 * x + 2

# without simplification

inverse_bquote(2 * a * .(x) + log10(b), SIMPLIFY = FALSE)
## 2 * 2 * x + log10(100)

更新:添加了简化。选择它。