Statistic.h

Go to the documentation of this file.

#ifndef GALOIS_STATISTIC_H
#define GALOIS_STATISTIC_H

#include "Galois/Runtime/Support.h"
#include "Galois/Runtime/PerThreadStorage.h"
#include "Galois/Runtime/Sampling.h"
#include "Galois/Timer.h"

#include "boost/utility.hpp"

#include <list>

namespace Galois {

class Statistic {
  std::string statname;
  std::string loopname;
  GaloisRuntime::PerThreadStorage<unsigned long> val;
  bool valid;

public:
  Statistic(const std::string& _sn, unsigned long v, const std::string& _ln = "(NULL)"): statname(_sn), loopname(_ln), valid(true) {
    *val.getLocal() = v;
  }
  
  Statistic(const std::string& _sn, const std::string& _ln = "(NULL)"): statname(_sn), loopname(_ln), valid(true) { }

  ~Statistic() {
    report();
  }

  void report() {
    if (valid)
      GaloisRuntime::reportStat(this);
    valid = false;
  }

  unsigned long getValue(unsigned tid) {
    return *val.getRemote(tid);
  }

  std::string& getLoopname() {
    return loopname;
  }

  std::string& getStatname() {
    return statname;
  }

  Statistic& operator+=(unsigned long v) {
    *val.getLocal() += v;
    return *this;
  }
};

class StatManager: private boost::noncopyable {
  std::list<Statistic*> stats;

public:
  ~StatManager() {
    for (std::list<Statistic*>::iterator ii = stats.begin(), ei = stats.end(); ii != ei; ++ii) {
      (*ii)->report();
    }
    GaloisRuntime::printStats();
  }

  void push(Statistic& s) {
    stats.push_back(&s);
  }
};

class StatTimer : public Timer {
  const char* name;
  const char* loopname;
  bool main;

public:
  StatTimer(): name("Time"), loopname(0), main(true) { }
  StatTimer(const char* n, const char* l = 0): name(n), loopname(l), main(false) { }
  ~StatTimer() {
    GaloisRuntime::reportStat(loopname, name, get());
    if (main)
      GaloisRuntime::reportSampling(loopname);
  }

  void start() {
    if (main)
      GaloisRuntime::beginSampling();
    Timer::start();
  }

  void stop() {
    Timer::stop();
    if (main)
      GaloisRuntime::endSampling();
  }
};

}

#endif

---