R数字向量列表 - >带有Rcpp的C ++ 2d数组

时间:2011-11-16 22:48:19

标签: c++ arrays r rcpp

我主要使用R,但最终想使用Rcpp与一些接收和返回2d数字数组的C ++函数接口。因此,为了开始使用C ++和Rcpp,我想我只是做了一个小函数,将我的可变长度数字向量的R列表转换为C ++等价物,然后再回来。

require(inline)
require(Rcpp)

test1 = cxxfunction(signature(x='List'), body = 
'
  using namespace std;
  List xlist(x);
  int xlen = xlist.size();
  vector< vector<int> > xx;
  for(int i=0; i<xlen; i++) {
    vector<int> test = as<vector<int> > (xlist[i]);
    xx.push_back(test);
  }
  return(wrap(xx));
'
, plugin='Rcpp')

这就像我期望的那样:

> test1(list(1:2, 4:6))
[[1]]
[1] 1 2

[[2]]
[1] 4 5 6

不可否认,我只是通过非常详尽的文档的一部分,但有更好的(即更像Rcpp)方式来做R - &gt; C ++转换比使用for循环?我想可能不会,因为文档提到(至少使用内置方法)as“提供的灵活性较低,并且当前处理R对象的转换原始类型“,但我想检查,因为我在这方面非常新手。

1 个答案:

答案 0 :(得分:6)

我会给你一个可重复的例子的奖励积分,当然还有使用Rcpp :)然后我会带走那些没有询问rcpp-devel列表......

至于转换STL类型:你不必,但是当你决定这样做时,as<>()成语是正确的。我能想到的唯一“更好的方法”就是像R本身一样进行名称查找:

require(inline)
require(Rcpp)

set.seed(42)
xl <- list(U=runif(4), N=rnorm(4), T2df=rt(4,2))

fun <- cxxfunction(signature(x="list"), plugin="Rcpp", body = '
  Rcpp::List xl(x);         
  std::vector<double> u = Rcpp::as<std::vector<double> >(xl["U"]);
  std::vector<double> n = Rcpp::as<std::vector<double> >(xl["N"]);
  std::vector<double> t2 = Rcpp::as<std::vector<double> >(xl["T2df"]);
  // do something clever here
  return(R_NilValue);
')

希望有所帮助。否则,列表总是打开...

PS 至于双暗阵列,由于没有本机C ++双暗阵列,因此比较棘手。如果您确实想要进行线性代数,请查看RcppArmadilloRcppEigen