有没有办法将ALTREP与Rcpp一起使用?

时间:2018-11-13 23:57:39

标签: r rcpp

示例:

// [[Rcpp::export]]
size_t z1(SEXP x) {
  return Rf_xlength(x);
}

// [[Rcpp::export]]
size_t z2(NumericVector x) {
  // do anything or nothing
  return 1;
}

R:

x <- seq(1,1e10)
z1(x)
[1] 1e+10

z2(x)
# computer hard crash

假设,我只是想采用ALTREP向量的前10个元素。用C ++做到这一点的最佳方法是什么?

1 个答案:

答案 0 :(得分:1)

Rcpp当前不识别ALTREP。从ALTREP SEXPNumericVector的转换将成为现实。在我的情况下,R不再告诉我它无法分配所需的内存,但这是一个很小的差异。

目前,您必须处理没有Rcpp细节的ALTREP对象,例如您的最后一个问题:

#include <Rcpp.h>

// [[Rcpp::export]]
SEXP get_region(SEXP x, R_xlen_t i, R_xlen_t n) {
  SEXP result;
  switch (TYPEOF(x)) {
    case INTSXP: {
      result = PROTECT(Rf_allocVector(INTSXP, n));
      INTEGER_GET_REGION(x, i, n, INTEGER(result));
      UNPROTECT(1);
      break;
    }
    case REALSXP: {
      result = PROTECT(Rf_allocVector(REALSXP, n));
      REAL_GET_REGION(x, i, n, REAL(result));
      UNPROTECT(1);
      break;
    }
    default: {
      Rcpp::stop("Invalid SEXPTYPE %d (%s).\n", TYPEOF(x), Rcpp::type2name(x));
    }
  }
  return result;
}

/*** R
x <- seq(1,1e10)
.Internal(inspect(x)) # @5623ba0b65f0 14 REALSXP g0c0 [NAM(3)]  1 : 10000000000 (compact)
get_region(x, 0, 10)  # [1]  1  2  3  4  5  6  7  8  9 10

x <- seq(1,1e9)
.Internal(inspect(x)) # @5623ba143ff0 13 INTSXP g0c0 [NAM(3)]  1 : 1000000000 (compact)
get_region(x, 0, 10)  # [1]  1  2  3  4  5  6  7  8  9 10
*/

这使用了http://gallery.rcpp.org/articles/rcpp-wrap-and-recurse/中的运行时多态性。