我想复制以下R函数 Rcpp：
fR = function(x) x[1:2]
fR(c(1,2,3))
#[1] 1 2
fR(c('a','b','c'))
#[1] "a" "b"
我可以这样做固定输出类型：
library(inline)
library(Rcpp)
fint = cxxfunction(signature(x = "SEXP"), '
List xin(x);
IntegerVector xout;
for (int i = 0; i < 2; ++i) xout.push_back(xin[i]);
return xout;', plugin = "Rcpp")
但这只适用于整数，如果我尝试更换 xout 输入 List （要么 GenericVector，它是相同的） - 它适用于任何输入类型，但我回来了 list 而不是矢量。
什么是正确的 Rcpp 这样做的方式？
不要用 push_back 上 Rcpp 类型。当前实现Rcpp向量的方式需要每次都复制所有数据。这是一项非常昂贵的操作。
我们有 RCPP_RETURN_VECTOR 对于调度，这需要您编写一个模板函数，将Vector作为输入。
#include <Rcpp.h>
using namespace Rcpp ;
template <int RTYPE>
Vector<RTYPE> first_two_impl( Vector<RTYPE> xin){
Vector<RTYPE> xout(2) ;
for( int i=0; i<2; i++ ){
xout[i] = xin[i] ;
}
return xout ;
}
// [[Rcpp::export]]
SEXP first_two( SEXP xin ){
RCPP_RETURN_VECTOR(first_two_impl, xin) ;
}
/*** R
first_two( 1:3 )
first_two( letters )
*/
只是 sourceCpp 这个文件，这也将运行调用这两个函数的R代码。实际上，模板可能更简单，这也可以工作：
template <typename T>
T first_two_impl( T xin){
T xout(2) ;
for( int i=0; i<2; i++ ){
xout[i] = xin[i] ;
}
return xout ;
}
模板参数 T 只需要：
int operator[](int)或者，这可能是一项工作 dplyr 矢量访客。
#include <dplyr.h>
// [[Rcpp::depends(dplyr,BH)]]
using namespace dplyr ;
using namespace Rcpp ;
// [[Rcpp::export]]
SEXP first_two( SEXP data ){
VectorVisitor* v = visitor(data) ;
IntegerVector idx = seq( 0, 1 ) ;
Shield<SEXP> out( v->subset(idx) ) ;
delete v ;
return out ;
}
访问者允许您在向量上执行一组操作，而不管它包含的数据类型。
> first_two(letters)
[1] "a" "b"
> first_two(1:10)
[1] 1 2
> first_two(rnorm(10))
[1] 0.4647190 0.9790888
不要用 push_back 上 Rcpp 类型。当前实现Rcpp向量的方式需要每次都复制所有数据。这是一项非常昂贵的操作。
我们有 RCPP_RETURN_VECTOR 对于调度，这需要您编写一个模板函数，将Vector作为输入。
#include <Rcpp.h>
using namespace Rcpp ;
template <int RTYPE>
Vector<RTYPE> first_two_impl( Vector<RTYPE> xin){
Vector<RTYPE> xout(2) ;
for( int i=0; i<2; i++ ){
xout[i] = xin[i] ;
}
return xout ;
}
// [[Rcpp::export]]
SEXP first_two( SEXP xin ){
RCPP_RETURN_VECTOR(first_two_impl, xin) ;
}
/*** R
first_two( 1:3 )
first_two( letters )
*/
只是 sourceCpp 这个文件，这也将运行调用这两个函数的R代码。实际上，模板可能更简单，这也可以工作：
template <typename T>
T first_two_impl( T xin){
T xout(2) ;
for( int i=0; i<2; i++ ){
xout[i] = xin[i] ;
}
return xout ;
}
模板参数 T 只需要：
int operator[](int)或者，这可能是一项工作 dplyr 矢量访客。
#include <dplyr.h>
// [[Rcpp::depends(dplyr,BH)]]
using namespace dplyr ;
using namespace Rcpp ;
// [[Rcpp::export]]
SEXP first_two( SEXP data ){
VectorVisitor* v = visitor(data) ;
IntegerVector idx = seq( 0, 1 ) ;
Shield<SEXP> out( v->subset(idx) ) ;
delete v ;
return out ;
}
访问者允许您在向量上执行一组操作，而不管它包含的数据类型。
> first_two(letters)
[1] "a" "b"
> first_two(1:10)
[1] 1 2
> first_two(rnorm(10))
[1] 0.4647190 0.9790888
你需要选择一种类型（即不要使用 signature="SEXP" [哦，无论如何你应该看看属性]）。
或者你保持 SEXP 类型，并在内部发送。例如，参见 这篇文章在Rcpp画廊。
编辑： C当然是静态类型的。这些非常依赖于类型的开关也遍布R源。这里没有免费午餐。