Rcpp:构造模板化函数

时间:2017-11-08 16:09:27

标签: c++ rcpp

我有以下MWE。它包含一个包含两个成员的类,一个整数和一个字符串。使用函数first我创建了一个类的对象并返回一个指向它的指针。在second_intsecond_str中,我输出了两个成员。

鉴于second_intsecond_str或多或少具有相同的功能,我似乎很清楚模板化它们。鉴于我的MWE,这可能在Rcpp中做到吗?

class C{
public:
  std::vector<int> i_vec;
  std::vector<std::string> s_vec;

  C(){
    i_vec.push_back(0);
    s_vec.push_back("R");
  }
};

// [[Rcpp::export]]
XPtr<C> first()
{
  XPtr<C> p(new C, true);
  return(p);
}

// [[Rcpp::export]]
void second_int(XPtr<C> p)
{
  Rcout << p->i_vec[0] << "\n";
}

// [[Rcpp::export]]
void second_str(XPtr<C> p)
{
  Rcout << p->s_vec[0] << "\n";
}

1 个答案:

答案 0 :(得分:3)

模板不能直接在这里使用,但我们可以用它来完成我想你想要的!让我们考虑一下你所写的内容,以及你喜欢直接写的内容。

你有:

// [[Rcpp::export]]
void second_int(XPtr<C> p)
{
  Rcout << p->i_vec[0] << "\n";
}

// [[Rcpp::export]]
void second_str(XPtr<C> p)
{
  Rcout << p->s_vec[0] << "\n";
}

你想要的(这不是有效的C ++)是这样的:

// [[Rcpp::export]]
template< membername member_to_extract >
void second(XPtr<C> p)
{
  Rcout << p->member_to_extract[0] << "\n";
}

/**
 * we would like to call it like this:
 * second<i_vec>(p) (to get the int value)
 * second<s_vec>(p) (to get the str value)
 */

为什么这不起作用?好吧,检查http://en.cppreference.com/w/cpp/language/template_parameters,我们发现您无法将变量名称作为模板参数传递!所以这种方法注定要失败。 (注意:您可以使用宏,这可能是您正在寻找的解决方案!)

但是,虽然函数参数和返回类型相同,但它们的行为 仅基于您要使用的类型来决定。如果您要在已经模板化的上下文中使用这些函数,那么需要获得第二个和第二个接口!

拯救模板专业化!

// generic template. We need something that fails to compile
// so that second<int> and second<std::string> are the only
// specializations the compiler will allow us to use.
template< typename T >
void second(XPtr<C> p)
{
  static_assert(false, "second can't be used with this type.");
}

template< >
void second<int>(XPtr<C> p)
{
  Rcout << p->i_vec[0] << "\n";
}

template< >
void second<std::string>(XPtr<C> p)
{
  Rcout << p->s_vec[0] << "\n";
}

我认为这就是你想要的?请注意,如果您的平台不是行缓冲的,您可能希望使用std :: endl而不是&#34; \ n&#34;,因为它强制刷新(这可能是您想要的行为)。

编辑:部分专业化 - &gt;模板专业化