使用Rcpp在C ++中调用R的optim函数

时间:2017-09-28 16:18:24

标签: c++ r rcpp

我是Rcpp的新手,在我的代码中,我必须从C ++中调用R函数“optim”。我提到了很多例子,但仍然有一个错误:“静态断言失败:无法将类型转换为SEXP”。这是我的代码,问题是最后一个函数:

#include <RcppArmadillo.h>

using namespace Rcpp;
using namespace RcppArmadillo;
using namespace arma;
using namespace std;
// [[Rcpp::depends(RcppArmadillo)]]

// [[Rcpp::plugins(cpp11)]]

// [[Rcpp::export]]
double fr(arma::colvec x){
  double result = 100 * (x(2) - x(1) * x(1)) * (x(2) - x(1) * x(1)) + (1 -     x(1)) * (1 - x(1));
  return result;  
} 


typedef double (*funcPtr)(arma::colvec x);

// [[Rcpp::export]]
XPtr<funcPtr> putFunPtrInXPtr(){
  return(XPtr<funcPtr>(new funcPtr(&fr)));
}


// [[Rcpp::export]]
arma::colvec callOptimFun(SEXP x) {

  RNGScope scope;

  Rcpp::Environment stats("package:stats");
  Rcpp::Function optim = stats["optim"];
  XPtr<funcPtr> xpfun = putFunPtrInXPtr();
  funcPtr fun = *xpfun;
  Rcpp::List y = optim(x, fun);
  arma::colvec r = y["par"];
  return r; 
}

不幸的是,我为我的上一个函数尝试了很多方法,并且所有方法都有相同的错误。这些是我的尝试: 1。

// [[Rcpp::export]]
Rcpp::List callOptimFun(arma::colvec x) {

  \\....
  Rcpp::List y = optim(x, fun);
  return y; 
}

2

// [[Rcpp::export]]
Rcpp::List callOptimFun(arma::colvec x) {

  \\....
  Rcpp::List y = optim(x, fun);
  return y; 
}

3

// [[Rcpp::export]]
Rcpp::List callOptimFun(SEXP x) {

  \\....
  Rcpp::List y = optim(x, fun);
  return y; 
}

我不熟悉C ++。可能是什么问题?谢谢!

1 个答案:

答案 0 :(得分:2)

在这种情况下,使用功能指针是有问题的。 Rcpp 有一种特殊类型的包装器,它要求 C ++ 函数不能导出到名为Rcpp::InternalFunction R 中。在此实现下,您可以轻松地将 R optim C ++ 中的 C ++ 函数合并!

另外,我在fr函数中修改了元素访问索引。这种修改的原因是你遇到越界错误,因为 C ++ 中的索引是基于零的启动系统而不是基于一个系统启动系统,如 R < / em>的。 (例如x(0) R 中的x[1]相同

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

double fr(arma::vec x){
  double result = 100 * (x(1) - x(0) * x(0)) * (x(1) - x(0) * x(0)) + (1 - x(0)) * (1 - x(0));
  return result;  
} 

// [[Rcpp::export]]
arma::vec optim_rcpp(const arma::vec& init_val){

  Rcpp::Environment stats("package:stats"); 
  Rcpp::Function optim = stats["optim"];    

  Rcpp::List opt_results = optim(Rcpp::_["par"]    = init_val,
                                 Rcpp::_["fn"]     = Rcpp::InternalFunction(&fr),
                                 Rcpp::_["method"] = "BFGS");

  // Extract and coerce from list.
  arma::vec out = Rcpp::as<arma::vec>(opt_results[0]);

  return out;
}