使用犰狳数据类型在 RcppArmadillo 中调用 R 函数

Question

我正在将我的 R 代码和一些准备好的函数翻译成 RcppArmadillo。我想直接在我的 Rcpp 代码中使用其中一些功能，而不是翻译。比如我要调用sigma2函数：

sigma2<- function(xi.vec,w.vec,log10lambda,n,q){
  lambda <- 10^log10lambda
  (1/(n-q))*sum((lambda*xi.vec*(w.vec^2))/(lambda*xi.vec+1))
}

典型的Rcpp代码如下：

#include <RcppArmadillo.h>
// [[Rcpp::depends(RcppArmadillo)]]


// [[Rcpp::export]]
SEXP myS(){

  Rcpp::Environment myEnv = Rcpp::Environment::global_env();
  Rcpp::Function myS = myEnv["sigma2"];
  arma::vec xvec = myEnv["xi.vec"];
  arma::vec wvec = myEnv["w.vec"];
  double l = myEnv["log10lambda"];
  int n = myEnv["n"];
  int q = myEnv["q"];

  return myS(Rcpp::Named("xi.vec",xvec),
             Rcpp::Named("w.vec",wvec),
             Rcpp::Named("l",l),
             Rcpp::Named("n",n),
             Rcpp::Named("q",q));
}

当然可以。但我的问题是，在我的例子中，sigma2 函数的参数应该在 RcppArmadillo 中定义为另一个函数（比如 func1）的输出，并且它们具有犰狳数据类型。例如，xi.vec 和 w.vec 具有 vec 类型。现在我想知道如何修改这段代码来调用 sigma2？我需要改变我的环境吗？

Answer 1

首先，对将 R 函数和环境嵌入到 C++ 例程中说否。在这种情况下没有加速；只有相当大的放缓。此外，如果无法在 global.env 范围内检索到变量，则事情发生混乱的可能性更大。

在您的情况下，您似乎是从 myS() 内调用 myS()，没有终止条件。因此，您的函数永远不会 end。

例如

SEXP myS(){

Rcpp::Function myS = myEnv["sigma2"];

return myS(Rcpp::Named("xi.vec",xvec),
           Rcpp::Named("w.vec",wvec),
           Rcpp::Named("l",l),
           Rcpp::Named("n",n),
           Rcpp::Named("q",q));
}

将一个切换为 myS_R 和 myS_cpp。

关于环境劫持，您需要将值传递给 C++。您无法进入 R 函数以获取在调用之前传递给它的特定值。

例如

SEXP myS_cpp(arma::vec xvec, arma::vec wvec, double l, int n, int q){
    // code here
}

使用犰狳数据类型在 RcppArmadillo 中调用 R 函数

Call R functions in RcppArmadillo with armadillo data type

r

armadillo

rcpp