返回Rcpp :: Vector <rtype>元素的类的模板

时间:2017-02-21 09:47:30

标签: c++ rcpp

考虑以下几乎没有做任何事情的类:

class MyNumVec
{
  private:
    const NumericVector& x;
  public:
    MyNumVec(const NumericVector& y) : x(y) {}
    double operator[](int i) const { // here
      return x[i];
    }
    operator NumericVector() const { return x; }
};

我想让它更通用并使用模板,这样我就可以使用任何Vector<RTYPE>而不是数字向量,但问题是注释标记的行,因为我必须声明输出类型。我尝试使用C ++ 11中的auto类型,但它不起作用(“'auto' return without trailing return type”)。如何将其转换为使用任何类型的Vector的模板?

2 个答案:

答案 0 :(得分:5)

如果您需要坚持使用C ++ 98,通常的习惯用法是使用整数模板参数来表示the different SEXPTYPEs。在处理Rcpp::*Vector(或Rcpp::*Matrix)时,您通常只关注其中的5个:

# Integer Value SEXPTYPE  R Vector     Rcpp Vector
#            10   LGLSXP   logical   LogicalVector
#            13   INTSXP   integer   IntegerVector
#            14  REALSXP   numeric   NumericVector
#            15  CPLXSXP   complex   ComplexVector
#            16   STRSXP character CharacterVector

完成此操作后,有Rcpp::traits命名空间中常用变换的标准元编程工具:

  • SEXPTYPE - &gt; POD¹类型:storage_type
    • Rcpp::traits::storage_type<REALSXP>::type - &gt; double
    • Rcpp::traits::storage_type<INTSXP>::type - &gt; int
  • POD¹型 - &gt; SEXPTYPEr_sexptype_traits
    • Rcpp::traits::r_sexptype_traits<double>::rtype - &gt; 14(REALSXP
    • Rcpp::traits::r_sexptype_traits<int>::rtype - &gt; 13(INTSXP

¹storage_type<STRSXP>::type会产生SEXP(特别是CHARSXP),而技术上是POD类型,它与其他简单的矢量类型不同它的原子单位是一个不透明的指针(SEXP),而不是,例如可能会出现const char*std::string

一个无趣的例子,使用RCPP_RETURN_VECTOR来简洁:

#include <Rcpp.h>

template <int RTYPE>
class MyNumVec {
public:
    typedef Rcpp::Vector<RTYPE> vec_t;
    typedef typename Rcpp::traits::storage_type<RTYPE>::type storage_t;

private:
    const vec_t& x;

public:
    MyNumVec(const vec_t& y)
        : x(y)
    {}

    storage_t operator[](int i) const
    { return x[i]; }

    operator vec_t() const
    { return x; }
};

template <int RTYPE>
Rcpp::Vector<RTYPE> get_first_elem_impl(const Rcpp::Vector<RTYPE>& vec)
{
    MyNumVec<RTYPE> tmp(vec);
    return Rcpp::Vector<RTYPE>::create(tmp[0]);
}

// [[Rcpp::export]]
Rcpp::RObject get_first_elem(Rcpp::RObject x) {
    RCPP_RETURN_VECTOR(get_first_elem_impl, x);
}
get_first_elem(c(TRUE, TRUE, FALSE, TRUE, FALSE))
# [1] TRUE

get_first_elem(1L:5L)
# [1] 1

get_first_elem(1:5 + 0.5)
# [1] 1.5

get_first_elem(1:5 + 2i)
# [1] 1+2i

get_first_elem(letters[1:5])
# [1] "a"

答案 1 :(得分:3)

没有尾随返回类型

auto返回类型推导是C ++ 14语言功能。在C ++ 11中,只要添加尾随返回类型,就可以使用auto

auto operator[](int i) const -> decltype(x[i]) { 
    return x[i];
}

在上面的代码中,auto将被替换为x[i]的类型。