Rcpparmadillo С++ создает логический вектор

Я пытаюсь передать вектор bool в качестве аргумента функции, используя Rcpparmadillo. Глупый пример выглядит так:

// [[Rcpp::depends(RcppArmadillo)]]
#include <RcppArmadillo.h>

// [[Rcpp::export]]
arma::mat myfun(arma::mat A, arma::vec mybool)
{
    int n = A.n_rows;
    arma::vec B(n);

    for(unsigned int i = 0; i < n; ++i)
    {
        if(mybool.row(i) && i < 10) // mybool.row(i) && throws the error
        {
            B.row(i) = arma::accu(A.row(i));
        }
        else
        {
            B.row(i) = pow(arma::accu(A.row(i)), 0.5);
        }
    }

    return B;
}

Здесь предлагается тип mat<unsigned char> но не работает для меня. Я также пробовал uvec и std::vector<bool>, но тоже не работает. Как лучше всего передать логический вектор в качестве аргумента, используя Rcpparmadillo?


r c++ vector rcpp armadillo
person nopeva    schedule 31.08.2015    source источник
comment
Попробуйте отформатировать свой код более стандартным способом.   -  person Dirk Eddelbuettel    schedule 31.08.2015
comment
может быть, эта помощь: stackoverflow.com/questions/18899665/   -  person    schedule 31.08.2015
comment
Функция-член строки возвращает вектор-строку из матрицы. В этом случае у вас есть вектор, поэтому вы хотите, чтобы mybool(i) возвращал элемент с проверкой границ или mybool[i] без проверки границ.   -  person Manos Nikolaidis    schedule 31.08.2015

Ответы (1)


arrow_upward
4
arrow_downward

Вы хотите uvec от Armadillo — у него нет типа bool. Вот переформатированная версия вашего кода, которая * использовала uvec * напрямую индексирует векторы

// [[Rcpp::depends(RcppArmadillo)]]
#include <RcppArmadillo.h>

// [[Rcpp::export]]
arma::mat myfun(arma::mat A, arma::uvec mybool) {
    unsigned int n = A.n_rows;
    arma::vec B(n);
    for (unsigned int i=0; i<n; ++i) {
        if (mybool[i] && i < 10) {
           B[i] = arma::accu(A.row(i)) ;
        } else {
           B[i] = pow(arma::accu(A.row(i)), 0.5);
        }
    } //end loop
    return B;
}

/*** R
A <- matrix(1:16,4,4)
mybool <- c(FALSE, TRUE, TRUE, FALSE)
myfun(A, mybool)
*/

Если мы sourceCpp() это, он запускает R внизу для нас:

R> sourceCpp("/tmp/ap13.cpp")

R> A <- matrix(1:16,4,4)

R> mybool <- c(FALSE, TRUE, TRUE, FALSE)

R> myfun(A, mybool)
         [,1]
[1,]  5.29150
[2,] 32.00000
[3,] 36.00000
[4,]  6.32456
R>
person Dirk Eddelbuettel    schedule 31.08.2015
comment
Пара моментов на заметку. (1). функцию можно сделать более эффективной, избегая копирования A и mybook в начале. Должно быть arma::mat myfun(const arma::mat& A, const arma::uvec& mybool). (2) Данные в A следует хранить и обращаться к ним по столбцам, а не по строкам, поскольку Armadillo хранит данные по столбцам. - person mtall; 31.08.2015
comment
@mtall: У вас внимательный взгляд программиста на C/C++, но я приглашаю вас рассказать об этом. Наши объекты на самом деле являются тонкими прокси вокруг базовых SEXP объектов, которые являются указателями. Значит, копий нет. Теперь вы также можете писать через const &, но производительность будет (примерно) такой же. - person Dirk Eddelbuettel; 31.08.2015