r - 使用 R 的 C API 更改向量的大小

Question

我正在使用从 RallocVector调用的 C 函数中分配 R 向量.Call。是否可以在分配向量后更改向量的大小/长度？即，类似于realloc在 C 中的工作方式。

在代码中，我正在寻找一个函数reallocVector，以便以下函数执行相同的操作。

SEXP my_function1(SEXP input){
    SEXP R_output = PROTECT(allocVector(REALSXP, 100));

    // Do some work on R_output

    // Keep only the first 50 items
    reallocVector(R_output, 50); 

    UNPROTECT(1);
    return R_output;
}

SEXP my_function1(SEXP input){
    SEXP tmp_output = PROTECT(allocVector(REALSXP, 100));

    // Do the same work on tmp_output

    // Keep only the first 50 items
    SEXP R_output = PROTECT(allocVector(REALSXP, 50));
    for (int i = 0; i < 50; ++i) {
        REAL(R_output)[i] = REAL(tmp_output)[i];
    }

    UNPROTECT(2);
    return R_output;
}

Answer 1

似乎SETLENGTH在头文件中定义的宏Rinternals.h是解决这个问题的最佳选择。IE：

SEXP my_function1(SEXP input){
    SEXP R_output = PROTECT(allocVector(REALSXP, 100));

    // Do some work on R_output

    // Keep only the first 50 items
    SETLENGTH(R_output, 50);

    UNPROTECT(1);
    return R_output;
}