Вызывая рекурсию самой функции

У меня есть проблема с моей функцией cxx в R. И я хотел бы вызвать это сам, к сожалению, компилятор выдает мне это сообщение об ошибке:

Matmult2 не объявлен в этой области

Проблема также в том, чтобы позвонить Matmult2 в функции.

Моя первоначальная проблема связана с алгоритмом Штрассена, я хотел бы вызвать рекурсивно с RCPP / Inline. Большое спасибо за вашу помощь!

`

Matmult2<- cxxfunction(signature( a="NumericMatrix",b="NumericMatrix"), plugin="RcppArmadillo",includes = rcpp_inc,
  body='

              Rcpp::NumericMatrix verg(a);
              int n = verg.nrow();

              arma::mat A = Rcpp::as< arma::mat >(a);
              arma::mat B = Rcpp::as< arma::mat >(b);

              arma::mat a11 = A(span(0,(n/2)-1), span(0,(n/2)-1));
              arma::mat a12 = A(span(0,(n/2)-1), span((n/2),n-1));
              arma::mat a21 = A(span((n/2),n-1), span(0,(n/2)-1));
              arma::mat a22 = A(span((n/2),n-1), span((n/2),n-1));

              arma::mat b11 = B(span(0,(n/2)-1), span(0,(n/2)-1));
              arma::mat b12 = B(span(0,(n/2)-1), span((n/2),n-1));
              arma::mat b21 = B(span((n/2),n-1), span(0,(n/2)-1));
              arma::mat b22 = B(span((n/2),n-1), span((n/2),n-1));


              if ( n  < 780) return Rcpp::wrap( A * B );


              arma::mat P1(Matmult2(a11,(b12-b22)));
              arma::mat P2(Matmult2((a11+a12),(b22)));
              arma::mat P3(Matmult2((a21+a22),(b11)));
              arma::mat P4(Matmult2((a22),(b21-b11)));
              arma::mat P5(Matmult2((a11+a22),(b11+b22)));
              arma::mat P6(Matmult2((a12-a22),(b21+b22)));
              arma::mat P7(Matmult2((a11-a21),(b11+b12)));

              arma::mat c11((P5+P4-P2+P6));
              arma::mat c12((P1+P2));
              arma::mat c21((P3+P4));
              arma::mat c22((P5+P1-P3-P7));

              mat C1 = join_rows(c11, c12);
              mat C2 = join_rows(c21, c22);
              mat C = join_cols(C1, C2);

              return Rcpp::wrap( C );

")

`

1 ответ

Решение

Вы не можете выполнить прямую рекурсию, потому что вставляемый нами клеевой код создает функцию-оболочку, которая вызывается --- но вы не хотите, чтобы оболочка вызывалась рекурсивно, вы хотите, чтобы ваша функция вызывалась. Таким образом, вам также понадобятся: обертка, вызываемая из R, и рекурсивная функция только для C(++), которая вызывается из вашей обертки и которая сама вызывает.

Это также работает, если вы переключаетесь на атрибуты Rcpp. Здесь мы используем стандартный носитель рекурсивных функций, последовательность Фибоначчи:

R> library(Rcpp)
R> cppFunction("double fib(double n) { if (n < 2) return n; return(fib(n-1) + fib(n-2)); }")
R> sapply(0:10, fib)
 [1]  0  1  1  2  3  5  8 13 21 34 55
R> 

(А я пользуюсь double вместо int как последний переполняется ранее.)

Другие вопросы по тегам