r - 有没有办法将 ALTREP 与 Rcpp 一起使用？

Question

例子:

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

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

回复:

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

z2(x)
# computer hard crash

假设我只想获取 ALTREP 向量的前 10 个元素。在 C++ 中这样做的最佳方法是什么？

Answer 1

Rcpp 目前无法识别 ALTREP。因此，从 ALTREP SEXP 到 NumericVector 的转换将实现它。在我的例子中，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/ 的运行时多态性.