ProfileClock.h

Go to the documentation of this file.

#ifndef AMPGEN_PROFILECLOCK_H
#define AMPGEN_PROFILECLOCK_H 1
#include <chrono>
#include <cmath>
#include "AmpGen/MsgService.h"
#include "AmpGen/Utilities.h"
 
namespace AmpGen{
  struct ProfileClock {
    std::chrono::time_point<std::chrono::high_resolution_clock>       t_start;
    std::chrono::time_point<std::chrono::high_resolution_clock>       t_end;
    double t_duration = {0}; 
    
    ProfileClock() : t_start(std::chrono::high_resolution_clock::now()) {}
    void stop() 
    { 
      t_end      = std::chrono::high_resolution_clock::now() ; 
      t_duration += std::chrono::duration<double, std::milli>( t_end - t_start ).count();
    }
    double count() const 
    {
      auto now = std::chrono::high_resolution_clock::now() ; 
      return std::chrono::duration<double, std::milli>(now - t_start ).count();
    }
    void start(){ t_start = std::chrono::high_resolution_clock::now() ; }
    operator double() const { return t_duration; } ; 
  };
 
  template <int N, class FCN> 
    double  Profile( const FCN& fcn, const std::string& name ="" ){
      ProfileClock t; 
      for( size_t i = 0 ; i < N; ++i ) fcn();
      t.stop();
      INFO( (name == "" ? type_string<FCN>() : name ) << " " << t/double(N) << "[ms] per iteration" );
      return t;
    }
  template <int N, class FCN> 
    double  ProfileWithStat( const FCN& fcn, const std::string& name ="" ){
      double t = 0;
      double t2 = 0;
      double tmin = 1e9;
      double tmax = 0;
      for( size_t i = 0 ; i < N; ++i ){
        ProfileClock pi;
        fcn();
        pi.stop();
        t  += pi;
        t2 += pi*pi;
        tmin = pi < tmin ? pi : tmin;
        tmax = pi > tmax ? pi : tmax;
      }
      t /= double(N);
      t2 = std::sqrt( t2 / double(N) - t*t);
      INFO( (name == "" ? type_string<FCN>() : name ) << " " << t << " ± " << t2 << "[ms] per iteration << [" << tmin << ", " << tmax << "]" );
      return t;
    }
  
  template <int N, class FCN> 
    double  Profile2( const FCN& fcn ){
      ProfileClock t;
      auto z = 0 ; 
      for( size_t i = 0 ; i < N; ++i ) z += fcn();
      t.stop();
      INFO( type_string<FCN>() << " " << t/double(N) << "[ms] per iteration; " << z );
      return t;
    }
}
#endif