R S4 setMethod'['区分缺失的参数？

Question

为标题道歉,不知道怎么说这个问题.

如果我想setMethod在一个类上创建,我如何区分类似的情况mat[i,]和mat[i]？

我知道前者我可以使用:

setMethod("[",
    signature(x = "foo", j = "missing", drop = "missing"),
    function(x,i,j,drop) return(myFunc(x,i))
)

我如何设置一个方法来区分后者,我不想选择行,而是在base matrix类中选择元素？

查看?[文档,我期望以下内容可行:

setMethod("[",
    signature(x = "foo", i = "numeric"),
    function(x,i,j,drop) return(myFunc(x,i))
)

但它与任何先前定义的方法冲突,其中j和drop缺失.

这个想法是一样的

mat <- matrix(seq(9), 3, 3)

mat[c(1,3),]
1 4 7
3 6 9

mat[c(1,3)]
[1] 1 3

Answer 1

您会在 Matrix 包中找到很多示例。它使用 S4 并为矩阵实现新的类和方法。据我所知，无法在签名中声明您要查找的内容。相反，您必须使用该函数nargs来区分mat[1]和mat[1, ]。这是一个如何做到这一点的示例：

setClass("foo", slot = c(mat = "matrix"))

setMethod(
  "[",
  signature(x = "foo", i = "missing", j = "missing", drop = "missing"),
  function(x, i, j, drop = FALSE) {
    x
  }
)

setMethod(
  "[",
  signature(x = "foo", i = "numeric", j = "missing", drop = "missing"),
  function(x, i, j, ..., drop) {
    if (nargs() == 3) x@mat[i, ]
    else x@mat[i]
  }
)

setMethod(
  "[",
  signature(x = "foo", i = "numeric", j = "numeric", drop = "missing"),
  function(x, i, j, ..., drop) {
    x@mat[i, j]
  }
)

mat <- new("foo", mat = matrix(seq(9), 3, 3))

mat[]
mat[2:5]
mat[1:2, ]
mat[1:2, 2]

但是，如果您直接扩展基类 'matrix'（或 Matrix 包中的 'Matrix'）并执行类似的操作，则会更容易

setClass("Matrix", contains = "matrix")

因为您可以免费获得这些方法。请注意，例如，在上面的实现中，您仍然需要处理参数drop。基本上你必须重新实现已经存在的东西。