Я начал пытаться использовать rcpp
повысить скорость цикла for в R, где каждая итерация зависит от предыдущей (то есть нет легкой векторизации). Мой текущий код (ниже) немного быстрее, чем R, но далеко не так быстро, как я думал. Какие-нибудь явные недостатки в коде ниже, которые кто-то может заметить? Любой общий (или конкретный) совет будет полезен.
UpdateInfections <- cxxfunction(signature(pop ="data.frame",inds="integer",alpha="numeric",t="numeric"), '
DataFrame DF(pop);
IntegerVector xinds(inds);
NumericVector inf_time = DF["inf.time"];
IntegerVector loc = DF["loc"] ;
IntegerVector Rind = DF["R.indiv"] ;
NumericVector infector = DF["infector"] ;
IntegerVector vac = DF["vac"] ;
NumericVector wts(loc.size());
double xt = Rcpp::as<double>(t);
double xalpha = Rcpp::as<double>(alpha);RNGScope scope; // Initialize Random number generator
Environment base("package:base");
Function sample = base["sample"];
int n = loc.size();
int i;int j;int k;
int infsize = xinds.size();
for (i=0;i<infsize;i++) {
int infpoint = xinds[i]-1;
NumericVector inf_times_prop(Rind[infpoint]);
NumericVector inf_me(Rind[infpoint]);
for (j=0; j<n;j++){
if (j == infpoint){
wts[j] = 0.0;
} else if (loc[j] == loc[infpoint]){
wts[j] = 1.0;
} else {
wts[j] = xalpha;
}
}
inf_me = sample(n,Named("size",Rind[infpoint]),Named("prob",wts));
//Note that these will be shifted by one
for (k=0;k<Rind[infpoint];k++){
inf_times_prop[k] = floor(::Rf_rlnorm(1.6,.6) + 0.5 + xt);
if (inf_times_prop[k] < inf_time[inf_me[k]-1] && vac[inf_me[k]-1] == 0){
inf_time[inf_me[k]-1] = inf_times_prop[k];
infector[inf_me[k]-1] = inf_me[k];
}
}
}
// create a new data frame
Rcpp::DataFrame NDF =
Rcpp::DataFrame::create(Rcpp::Named("inf.time")=inf_time,
Rcpp::Named("loc")=loc,
Rcpp::Named("R.indiv")=Rind,
Rcpp::Named("infector")=infector,
Rcpp::Named("vac")=vac);
return(NDF);
' , plugin = "Rcpp" )
На самом деле мы сейчас работаем над чистой функцией примера C ++ для RcppArmadillo. Посмотрите здесь http://permalink.gmane.org/gmane.comp.lang.r.rcpp/4179 или здесь http://permalink.gmane.org/gmane.comp.lang.r.rcpp для обновлений.
Вы перезваниваете в R. Это не может быть так быстро, как чистое решение C ++.
Ваш пример тоже длинный, слишком длинный. Я рекомендую профилировать и оптимизировать отдельные части. Увы, до сих пор нет полностью бесплатного обеда.