tools to help with Timer.

romainfrancois · romainfrancois · commit a2c3191f0658 · 2014-07-28T18:24:14.000+02:00
diff --git a/inst/include/RcppParallel.h b/inst/include/RcppParallel.h
@@ -17,6 +17,7 @@
 
 #include "RcppParallel/RVector.h"
 #include "RcppParallel/RMatrix.h"
+#include <RcppParallel/Timer.h>
 
 namespace RcppParallel {
 
diff --git a/inst/include/RcppParallel/Timer.h b/inst/include/RcppParallel/Timer.h
@@ -0,0 +1,79 @@
+#ifndef __RCPP_PARALLEL_TIMER__
+#define __RCPP_PARALLEL_TIMER__
+
+namespace RcppParallel {
+    
+    template <typename Timer>
+    class SingleTimer {
+    public:
+        SingleTimer() : timer(){}
+        
+        inline operator SEXP(){
+            Rcpp::List out = Rcpp::List::create(timer) ;
+            out.attr("class") = Rcpp::CharacterVector::create( "SingleTimer", "Timer" );
+            return out ;
+        }
+        
+        inline void step( const char* name ){
+            timer.step(name) ;    
+        }
+        
+    private:
+        Timer timer ;
+    } ;
+    
+    template <typename Timer>
+    class FixedSizeTimers {
+    public:
+        FixedSizeTimers( int n ) : timers(n){}
+        
+        inline Timer& get(int i) {
+            return timers[i] ;
+        }
+        
+        inline operator SEXP(){
+            Rcpp::List out = wrap(timers) ;
+            out.attr("class") = Rcpp::CharacterVector::create("FixedSizeTimers", "Timer") ;
+            return out ;
+        }
+        
+        
+    private:
+        std::vector<Timer> timers ;
+    } ;
+    
+    template <typename Timer, typename Mutex, typename Locker>
+    class TimersList {
+    public:
+        
+        TimersList(): timers(){}
+        
+        Timer& get_new_timer(){
+            Locker lock(mutex) ;
+            if( timers.empty() ) {
+                timers.push_back( Timer() ) ;
+            } else {
+                timers.push_back( Timer( timers.front().origin() ) ) ;
+            }
+            std::cout << timers.size() << " timers" << std::endl ;
+            return timers.back() ;
+        }
+        
+        inline operator SEXP(){
+            Rcpp::List data = wrap(timers) ;
+            data.attr("class") = Rcpp::CharacterVector::create( "TimersList" , "Timer" ) ;
+            return data ;
+        }
+        
+    private:
+        TimersList( const TimersList& ) ;
+        std::list<Timer> timers ;
+        Mutex mutex ;
+        
+    } ;
+    
+    
+} // namespace RcppParallel
+
+
+#endif // __RCPP_PARALLEL_COMMON__
