Profiler.h

Go to the documentation of this file.

// $Id: Profiler.h 1013 2011-05-26 14:39:16Z kmo $
//==============================================================================
//==============================================================================

#ifndef _PROFILER_H
#define _PROFILER_H

#include <iostream>
#include <string>
#include <map>
#include <sys/time.h>

namespace LR {

class Profiler
{
public:
  Profiler(const std::string& name);
  ~Profiler();

  void start(const std::string& funcName);
  void stop(const std::string& funcName);

  void report(std::ostream& os) const;
  void clear() { myTimers.clear(); allCPU = allWall = 0.0; nRunners = 0; }

private:
  struct Profile
  {
    clock_t startCPU;  
    double  startWall; 
    double  totalCPU;  
    double  totalWall; 
    size_t  nCalls;    
    bool    running;   

    Profile() : nCalls(0), running(false) { totalCPU = totalWall = 0.0; }
    bool haveTime() const { return totalCPU >= 0.005 || totalWall >= 0.005; }
  };

  friend std::ostream& operator<<(std::ostream& os, const Profile& p);

  std::string myName; 

  std::map<std::string,Profile> myTimers; 

  double allCPU;  
  double allWall; 

  size_t nRunners; 
};


namespace utl
{
  extern Profiler* profiler; 

  class prof
  {
    const char* name; 
  public:
    prof(const char* tag) : name(tag) { if (profiler) profiler->start(name); }
    ~prof() { if (profiler) profiler->stop(name); }
  };
}


#define PROFILE(label) utl::prof _prof(label)

#if PROFILE_LEVEL >= 1
#define PROFILE1(label) PROFILE(label)
#else

#define PROFILE1(label)
#endif

#if PROFILE_LEVEL >= 2
#define PROFILE2(label) PROFILE(label)
#else

#define PROFILE2(label)
#endif

#if PROFILE_LEVEL >= 3
#define PROFILE3(label) PROFILE(label)
#else

#define PROFILE3(label)
#endif

#if PROFILE_LEVEL >= 4
#define PROFILE4(label) PROFILE(label)
#else

#define PROFILE4(label)
#endif

} // end namespace LR

#endif