Criando vetores

Otimizando código R com Rcpp

Romain François

Consulting Datactive, ThinkR

Obter um vetor do lado do R

Código C++

// [[Rcpp::export]]
double extract( NumericVector x, int i){
    return x[i] ;}

Chamado a partir do R

x <- c(13.2, 34.1)
extract(x, 0)

x[1]

13.2

13.2

// [[Rcpp::export]]
double extract( NumericVector x, int i){
    return x[i] ;}

Vários casos

# x já é um vetor numérico
extract( c(13.3, 54.2), 0 )

13.3

# x é um vetor de inteiros; primeiro é convertido em numérico
extract( 1:10, 0 )

// [[Rcpp::export]]
double extract( NumericVector x, int i){
    return x[i] ;}

# conversão impossível: erro
extract( letters, 0 )

Error in extract(letters, 0) : 
  Not compatible with requested type: [type=character; target=double].

Criar um vetor de tamanho dado

// [[Rcpp::export]]
NumericVector ones(int n){
    // cria um novo vetor numérico de tamanho n
    NumericVector x(n) ;    
    // manipula o vetor
    for( int i=0; i<n; i++){
        x[i] = 1 ; }
    return x ;
}

Chamando ones do R:

ones(10)

1 1 1 1 1 1 1 1 1 1

Variações do construtor

double value = 42.0 ;
int n = 20 ;

// cria um vetor numérico de tamanho 20 
// com todos os valores iguais a 42
NumericVector x( n, value ) ;

Conjunto dado de valores

NumericVector x = NumericVector::create( 1, 2, 3 );

CharacterVector s = CharacterVector::create( "pink", "blue" );

Conjunto dado de valores com nomes

Nomeando todos os valores

NumericVector x = NumericVector::create( 
    _["a"] = 1, _["b"] = 2, _["c"] = 3
) ;

Nomeando apenas alguns valores

IntegerVector y = IntegerVector::create( 
    _["d"] = 4, 5, 6, _["f"] = 7
) ;

Clonagem de vetores

// [[Rcpp::export]]
NumericVector positives( NumericVector x ){

    // clona x em y
    NumericVector y = clone(x) ;

    for( int i=0; i< y.size(); i++){
        if( y[i] < 0 ) y[i] = 0 ;
    }
    return y ;
}

Vamos praticar!

Otimizando código R com Rcpp