Rcpp：如何获得 Rcpp::Nullable NumericVector 的大小

Question

我想把一个R函数翻译成Rcpp，一个简单的测试代码如下，但是我不知道如何处理默认设置为NULL的参数。

test<- function(t=NULL,tmax=NULL,tmin=NULL){
  if(is.null(t)){
    yout=(tmax-tmin)*(tmin+tmax)
  }else{
    yout=2*t
  }
  return(yout)
}

test(tmax=1:3,tmin=0:2)




  // [[Rcpp::export]]
    NumericVector cpptest(Rcpp::Nullable<Rcpp::NumericVector> t=R_NilValue,
                          Rcpp::Nullable<Rcpp::NumericVector> tmax=R_NilValue,
                          Rcpp::Nullable<Rcpp::NumericVector> tmin=R_NilValue){
      int N=0;
      if(t.isNotNull()) {
        N=t.size(); /* which show a error*/
      }else{
        N=tmax.size(); /* which show a error*/
      }
      NumericVector yout=NumericVector(N);

      if(t.isNotNull()) {
        for(i=0;i<N,i++){
          yout[i]=2*t[i]
        }
      }else{
        for(i=0;i<N,i++){
          yout[i]=(tmax[i]-tmin[i])*(tmin[i]+tmax[i])
        }
      }
      return(yout)
    }

Answer 1

而不是打电话，例如.size() 像您在此处所做的那样直接在对象上 -- N = t.size(); -- 您需要将其转换为基础类型。例如，

#include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
int nullable_size(Nullable<NumericVector> x_ = R_NilValue)
{
    if (x_.isNotNull()) {
        NumericVector x(x_.get());
        return x.size();
    }
    warning("argument x_ is NULL");
    return -1;
}

/*** R

nullable_size(rnorm(5))
# [1] 5

nullable_size(NULL)
# [1] -1
# Warning message:
# In .Primitive(".Call")(<pointer: 0x000000006aa417a0>, x_) :
#   argument x_ is NULL

*/

正如德克所指出的，这里使用 .get() 并不是绝对必要的——使用 NumericVector x(x_); 将调用 Nullable<>::operator SEXP() 并且同样有效。

另外，请以后更好地格式化您的代码。

Rcpp：如何获得 Rcpp::Nullable NumericVector 的大小

Rcpp: How can I get the size of a Rcpp::Nullable NumericVector

rcpp