如何在Rcpp中的函数参数中将默认值设置为Rcpp :: Function?

时间:2017-12-07 16:03:42

标签: rcpp

我想在函数的参数中设置默认值为Rcpp :: Function参数。

简单的作业Rcpp::Function func = mean是不可能的。它返回错误:no viable conversion from '<overloaded function type>' to 'Rcpp::Function' (aka 'Function_Impl<PreserveStorage>')

或者,我尝试过这样的事情:Rcpp::Function func = Function("mean"),但同样,它不起作用。它会返回警告消息:Unable to parse C++ default value 'Function("mean")' for argument func of function

例如,我有自己的函数,最大名为maxC

// [[Rcpp::export]]
double maxC(NumericVector x) {
  double max;
  max = *std::max_element(x.begin(), x.end());

  return max;
}

现在,我想将它(maxC)用作另一个函数的默认参数,例如:

// [[Rcpp::export]]
double aggregate(NumericVector x, Rcpp::Function func = maxC) {
  double agg;
  agg = Rcpp::as<double>(func(x));

  return agg;   
}

但它不起作用。有什么建议?谢谢。

1 个答案:

答案 0 :(得分:2)

我不相信你可以用这种方式设置默认函数...最好的办法是将函数设置为NULL值,然后让代码稍后执行相应的默认值。例如......

#include <Rcpp.h>

// [[Rcpp::export]]
Rcpp::NumericVector func_defaults(Rcpp::NumericVector x,
                                  Rcpp::Nullable<Rcpp::Function> f = R_NilValue) {

  if (f.isNotNull()) {
    Rcpp::NumericVector res = Rcpp::as<Rcpp::Function>(f)(x);
    return res;
  }

  Rcpp::Environment global_funcs = Rcpp::Environment::global_env();
  Rcpp::Function mean_r = global_funcs["mean"];
  return mean_r(x);
}

测试:

func_defaults(c(2.5,3,1))
# [1] 2.166667
func_defaults(c(2.5,3,1), mean)
# [1] 2.166667
func_defaults(c(2.5,3,1), median)
# [1] 2.5