在R

时间:2016-10-17 13:45:10

标签: r r-s3

所以我正在改变我在R中构建的一些函数的类,以便添加一个描述属性,因为我想使用S3泛型来为我处理所有事情。基本上,我有一个像

这样的结构
foo <- function(x) x + 1

addFunction <- function(f, description) {
    class(f) <- c("addFunction", "function")
    attr(f, "description") <- description
    f
}

foo <- addFunction(foo, "Add one")

然后我做了像

这样的事情
description <- function(x) UseMethod("description")
description.default <- function(x) deparse(substitute(x))
description.addFunction <- function(x) attr(x, "description")

这很好用,但不是那么优雅。我想知道是否可以定义一个新的函数类,以便可以使用类似于function语法的语法定义此类的实例。换句话说,是否可以定义addFunction以便以下列方式生成foo

foo <- addFunction(description = "Add one", x) {
    x + 1
}

(或者类似的东西,我对属性应该添加到函数的位置没有强烈的感觉)?

感谢阅读!

更新:我已经尝试了更多这个想法,但还没有真正达到任何具体结果 - 所以这只是我对当前(更新)主题的概述:

我尝试了只复制function() - 函数的想法,给它一个不同的名字,然后再操作它。但是,这不起作用,我想对这里发生的事情有任何意见:

> function2 <- `function`
> identical(`function`, function2)
[1] TRUE
> function(x) x
function(x) x
> function2(x) x
Error: unexpected symbol in "function2(x) x"
> function2(x)
Error: incorrect number of arguments to "function"

由于function()是一个原始函数,我试着查看定义它的C代码以获得更多线索。我对来自function2(x)电话的错误消息特别感兴趣。 function()底层的C代码是

/* Declared with a variable number of args in names.c */
  SEXP attribute_hidden do_function(SEXP call, SEXP op, SEXP args, SEXP rho)
{
  SEXP rval, srcref;

  if (TYPEOF(op) == PROMSXP) {
    op = forcePromise(op);
    SET_NAMED(op, 2);
  }
  if (length(args) < 2) WrongArgCount("function");
  CheckFormals(CAR(args));
  rval = mkCLOSXP(CAR(args), CADR(args), rho);
  srcref = CADDR(args);
  if (!isNull(srcref)) setAttrib(rval, R_SrcrefSymbol, srcref);
  return rval;
  }

从此我得出结论,由于某种原因,现在需要四个参数callopargsrho中的至少两个。从do_function()的签名我猜测传递给do_function的四个参数应该是一个调用,一个promise,一个参数列表,然后可能是一个环境。我为function2尝试了很多不同的组合(包括将这些参数中的两个设置为NULL),但我不断收到相同的(新的)错误消息:

> function2(call("sum", 2, 1), NULL, list(x=NULL), baseenv())
Error: invalid formal argument list for "function"
> function2(call("sum", 2, 1), NULL, list(x=NULL), NULL)
Error: invalid formal argument list for "function"

从C函数CheckFormals()返回此错误消息,我也查了一下:

/* used in coerce.c */
  void attribute_hidden CheckFormals(SEXP ls)
{
  if (isList(ls)) {
    for (; ls != R_NilValue; ls = CDR(ls))
      if (TYPEOF(TAG(ls)) != SYMSXP)
        goto err;
    return;
  }
  err:
    error(_("invalid formal argument list for \"function\""));
  }

我根本不会说流利的C语言,所以从现在开始我不太确定下一步该怎么做。

所以这些是我更新的问题:

  • 为什么functionfunction2的行为方式不同?为什么 我是否需要使用不同的语法来调用function2 在R中被视为相同?
  • function2的正确论据是什么 这样function2([arguments])实际上会定义一个函数吗?

1 个答案:

答案 0 :(得分:2)

R中的某些关键字(如iffunction)在调用基础函数的方式中具有特殊语法。如果需要,可以很容易地使用if作为函数,例如

`if`(1 == 1, "True", "False")

相当于

if (1 == 1) {
  "True"
} else {
  "False"
}

function比较棘手。在a previous question处有一些帮助。

对于您目前的问题,这里有一个解决方案:

# Your S3 methods
description <- function(x) UseMethod("description")
description.default <- function(x) deparse(substitute(x))
description.addFunction <- function(x) attr(x, "description")

# Creates the pairlist for arguments, handling arguments with no defaults
# properly. Also brings in the description
addFunction <- function(description, ...) {
  args <- eval(substitute(alist(...)))
  tmp <- names(args)
  if (is.null(tmp)) tmp <- rep("", length(args))
  names(args)[tmp==""] <- args[tmp==""]
  args[tmp==""] <- list(alist(x=)$x)
  list(args = as.pairlist(args), description = description)
}

# Actually creates the function using the structure created by addFunction and the body
`%{%` <- function(args, body) {
  stopifnot(is.pairlist(args$args), class(substitute(body)) == "{")
  f <- eval(call("function", args$args, substitute(body), parent.frame()))
  class(f) <- c("addFunction", "function")
  attr(f, "description") <- args$description
  f
}

# Example. Note that the braces {} are mandatory even for one line functions

foo <- addFunction(description = "Add one", x) %{% {
  x + 1
}

foo(1)
#[1] 2