6

我刚开始学习 R,因此陷入了看似琐碎的问题。我试图弄清楚值是如何在 R 中输出的。在 C++ 中,我们只是使用返回变量,但 R 似乎不是这种情况。假设我有一个接受 4 个输入参数的函数,将这些参数传递给执行所需计算的 c++ 函数,现在如果我想在 R 中加载这个 myfun 并从 c++ 函数获取输出,我需要做什么?以下是我尝试使用的模板。

extern "C" {
   SEXP myfun(SEXP S, SEXP A, SEXP B, SEXP C) { //will call this function from R.
     SEXP rate, dir, list, list_names; //declare variables

     PROTECT( rate = allocMatrix(REALSXP, 10, 2) ); //allocate 10x2 matrix of double type?
     PROTECT( dir = allocVector(INTSXP, 10) ); //allocated vector(10) of int type?

     double* p_rate = REAL(rate); //why do I need pointers? 
     int* p_dir = INTEGER(dir);

//这里我调用一个计算vector<vector<double> > someVal和的C++函数vector<int> someVal2

现在我想将这些值传递给速率和目录。

 for(int i =0; i < 10; i++){
      rate[i][0] = someVal1[i][0];
      rate[i][1] = someVal1[i][1];
      dir[i] = someVal2[i]; 
  }

但编译器不喜欢它。在这种情况下,我不确定如何使用 p_rate 和 p_dir 指针。

另外,我不太明白为什么下面的代码是这样的。它似乎将字符串、rate 和 dir 传递给 list_names,并将值传递给 rate 和 dir。为什么我们不能简单地做 cout 并打印所需的值?

   char *tag[2] = {"rate","dir"};
   PROTECT(list_names = allocVector(STRSXP ,2));
   SET_STRING_ELT(list_names, 0, mkChar(tag[0]));
   SET_STRING_ELT(list_names, 1, mkChar(tag[1]));
   PROTECT(list = allocVector(VECSXP ,2));
   SET_VECTOR_ELT(list, 0, rate);
   SET_VECTOR_ELT(list, 1, dir);
   setAttrib(list, R_NamesSymbol , list_names);

   UNPROTECT(8);
   return (list);
}

我没有得到在 R 中返回变量的基本想法,这似乎很混乱。如果有人能指出我解释这些东西的适当资源,我将不胜感激。

提前感谢您的帮助!

4

2 回答 2

9

如果您已经熟悉 C++,您可能会发现我们用于无缝 R 和 C++集成的Rcpp项目很有用。它包含大量文档和示例,此外,CRAN 现在有超过 80 个使用它的包,这提供了另一个大型示例语料库。

您可以从与我们的JSS 论文相对应的“简介”小插图开始,并在[rcpp]此处仔细阅读标签。

否则,您将回到更简单且仅基于 C 的 R API。

于 2012-11-11T18:09:35.807 回答
5

正如我们在回答您之前的问题时所说的那样,给Rcpp一个机会,并消除代码中的所有问题。

#include <Rcpp.h>
using namespace Rcpp ;

void yourCode( std::vector< std::vector<double> >& someVal, 
               std::vector<double>& someVal2 ){ ... }

extern "C" SEXP myfun() { 
    NumericMatrix rate(10, 2); 
    NumericVector dir(10) ;

    std::vector< std::vector<double> > someVal ;
    std::vector< double > someVal2 ;

    // call whatever function that fills someVal and someVal2
    yourCode( someVal, someVal2 ) ;

    // fills rate and dir
    for(int i =0; i < 10; i++){
      rate(i, 0) = someVal1[i][0];
      rate(i, 1) = someVal1[i][1];
      dir[i] = someVal2[i]; 
    }

    // structure the output
    List result = List::create( 
        _["rate"] = rate,
        _["dir"] = dir
    ) ;

    return result ;

}
于 2012-11-12T08:40:53.383 回答