/* -*- mode: C++; indent-tabs-mode: nil; -*-

 *

 * This file is a part of LEMON, a generic C++ optimization library.

 *

 * Copyright (C) 2003-2009

 * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport

 * (Egervary Research Group on Combinatorial Optimization, EGRES).

 *

 * Permission to use, modify and distribute this software is granted

 * provided that this copyright notice appears in all copies. For

 * precise terms see the accompanying LICENSE file.

 *

 * This software is provided "AS IS" with no warranty of any kind,

 * express or implied, and with no claim as to its suitability for any

 * purpose.

 *

 */

#ifndef LEMON_TIME_MEASURE_H

#define LEMON_TIME_MEASURE_H

///\ingroup timecount

///\file

///\brief Tools for measuring cpu usage

#ifdef WIN32

#include <lemon/bits/windows.h>

#else

#include <unistd.h>

#include <sys/times.h>

#include <sys/time.h>

#endif

#include <string>

#include <fstream>

#include <iostream>

namespace lemon {

  /// \addtogroup timecount

  /// @{

  /// A class to store (cpu)time instances.

  /// This class stores five time values.

  /// - a real time

  /// - a user cpu time

  /// - a system cpu time

  /// - a user cpu time of children

  /// - a system cpu time of children

  ///

  /// TimeStamp's can be added to or substracted from each other and

  /// they can be pushed to a stream.

  ///

  /// In most cases, perhaps the \ref Timer or the \ref TimeReport

  /// class is what you want to use instead.

  class TimeStamp

  {

    double utime;

    double stime;

    double cutime;

    double cstime;

    double rtime;

    void _reset() {

      utime = stime = cutime = cstime = rtime = 0;

    }

  public:

    ///Read the current time values of the process

    void stamp()

    {

#ifndef WIN32

      timeval tv;

      gettimeofday(&tv, 0);

      rtime=tv.tv_sec+double(tv.tv_usec)/1e6;

      tms ts;

      double tck=sysconf(_SC_CLK_TCK);

      times(&ts);

      utime=ts.tms_utime/tck;

      stime=ts.tms_stime/tck;

      cutime=ts.tms_cutime/tck;

      cstime=ts.tms_cstime/tck;

#else

      bits::getWinProcTimes(rtime, utime, stime, cutime, cstime);

#endif

    }

    /// Constructor initializing with zero

    TimeStamp()

    { _reset(); }

    ///Constructor initializing with the current time values of the process

    TimeStamp(void *) { stamp();}

    ///Set every time value to zero

    TimeStamp &reset() {_reset();return *this;}

    ///\e

    TimeStamp &operator+=(const TimeStamp &b)

    {

      utime+=b.utime;

      stime+=b.stime;

      cutime+=b.cutime;

      cstime+=b.cstime;

      rtime+=b.rtime;

      return *this;

    }

    ///\e

    TimeStamp operator+(const TimeStamp &b) const

    {

      TimeStamp t(*this);

      return t+=b;

    }

    ///\e

    TimeStamp &operator-=(const TimeStamp &b)

    {

      utime-=b.utime;

      stime-=b.stime;

      cutime-=b.cutime;

      cstime-=b.cstime;

      rtime-=b.rtime;

      return *this;

    }

    ///\e

    TimeStamp operator-(const TimeStamp &b) const

    {

      TimeStamp t(*this);

      return t-=b;

    }

    ///\e

    TimeStamp &operator*=(double b)

    {

      utime*=b;

      stime*=b;

      cutime*=b;

      cstime*=b;

      rtime*=b;

      return *this;

    }

    ///\e

    TimeStamp operator*(double b) const

    {

      TimeStamp t(*this);

      return t*=b;

    }

    friend TimeStamp operator*(double b,const TimeStamp &t);

    ///\e

    TimeStamp &operator/=(double b)

    {

      utime/=b;

      stime/=b;

      cutime/=b;

      cstime/=b;

      rtime/=b;

      return *this;

    }

    ///\e

    TimeStamp operator/(double b) const

    {

      TimeStamp t(*this);

      return t/=b;

    }

    ///The time ellapsed since the last call of stamp()

    TimeStamp ellapsed() const

    {

      TimeStamp t(NULL);

      return t-*this;

    }

    friend std::ostream& operator<<(std::ostream& os,const TimeStamp &t);

    ///Gives back the user time of the process

    double userTime() const

    {

      return utime;

    }

    ///Gives back the system time of the process

    double systemTime() const

    {

      return stime;

    }

    ///Gives back the user time of the process' children

    ///\note On <tt>WIN32</tt> platform this value is not calculated.

    ///

    double cUserTime() const

    {

      return cutime;

    }

    ///Gives back the user time of the process' children

    ///\note On <tt>WIN32</tt> platform this value is not calculated.

    ///

    double cSystemTime() const

    {

      return cstime;

    }

    ///Gives back the real time

    double realTime() const {return rtime;}

  };

  inline TimeStamp operator*(double b,const TimeStamp &t)

  {

    return t*b;

  }

  ///Prints the time counters

  ///Prints the time counters in the following form:

  ///

  /// <tt>u: XX.XXs s: XX.XXs cu: XX.XXs cs: XX.XXs real: XX.XXs</tt>

  ///

  /// where the values are the

  /// \li \c u: user cpu time,

  /// \li \c s: system cpu time,

  /// \li \c cu: user cpu time of children,

  /// \li \c cs: system cpu time of children,

  /// \li \c real: real time.

  /// \relates TimeStamp

  /// \note On <tt>WIN32</tt> platform the cummulative values are not

  /// calculated.

  inline std::ostream& operator<<(std::ostream& os,const TimeStamp &t)

  {

    os << "u: " << t.userTime() <<

      "s, s: " << t.systemTime() <<

      "s, cu: " << t.cUserTime() <<

      "s, cs: " << t.cSystemTime() <<

      "s, real: " << t.realTime() << "s";

    return os;

  }

  ///Class for measuring the cpu time and real time usage of the process

  ///Class for measuring the cpu time and real time usage of the process.

  ///It is quite easy-to-use, here is a short example.

  ///\code

  /// #include<lemon/time_measure.h>

  /// #include<iostream>

  ///

  /// int main()

  /// {

  ///

  ///   ...

  ///

  ///   Timer t;

  ///   doSomething();

  ///   std::cout << t << '\n';

  ///   t.restart();

  ///   doSomethingElse();

  ///   std::cout << t << '\n';

  ///

  ///   ...

  ///

  /// }

  ///\endcode

  ///

  ///The \ref Timer can also be \ref stop() "stopped" and

  ///\ref start() "started" again, so it is possible to compute collected

  ///running times.

  ///

  ///\warning Depending on the operation system and its actual configuration

  ///the time counters have a certain (10ms on a typical Linux system)

  ///granularity.

  ///Therefore this tool is not appropriate to measure very short times.

  ///Also, if you start and stop the timer very frequently, it could lead to

  ///distorted results.

  ///

  ///\note If you want to measure the running time of the execution of a certain

  ///function, consider the usage of \ref TimeReport instead.

  ///

  ///\sa TimeReport

  class Timer

  {

    int _running; //Timer is running iff _running>0; (_running>=0 always holds)

    TimeStamp start_time; //This is the relativ start-time if the timer

                          //is _running, the collected _running time otherwise.

    void _reset() {if(_running) start_time.stamp(); else start_time.reset();}

  public:

    ///Constructor.

    ///\param run indicates whether or not the timer starts immediately.

    ///

    Timer(bool run=true) :_running(run) {_reset();}

    ///\name Control the state of the timer

    ///Basically a Timer can be either running or stopped,

    ///but it provides a bit finer control on the execution.

    ///The \ref lemon::Timer "Timer" also counts the number of

    ///\ref lemon::Timer::start() "start()" executions, and it stops

    ///only after the same amount (or more) \ref lemon::Timer::stop()

    ///"stop()"s. This can be useful e.g. to compute the running time

    ///of recursive functions.

    ///@{

    ///Reset and stop the time counters

    ///This function resets and stops the time counters

    ///\sa restart()

    void reset()

    {

      _running=0;

      _reset();

    }

    ///Start the time counters

    ///This function starts the time counters.

    ///

    ///If the timer is started more than ones, it will remain running

    ///until the same amount of \ref stop() is called.

    ///\sa stop()

    void start()

    {

      if(_running) _running++;

      else {

        _running=1;

        TimeStamp t;

        t.stamp();

        start_time=t-start_time;

      }

    }

    ///Stop the time counters

    ///This function stops the time counters. If start() was executed more than

    ///once, then the same number of stop() execution is necessary the really

    ///stop the timer.

    ///

    ///\sa halt()

    ///\sa start()

    ///\sa restart()

    ///\sa reset()

    void stop()

    {

      if(_running && !--_running) {

        TimeStamp t;

        t.stamp();

        start_time=t-start_time;

      }

    }

    ///Halt (i.e stop immediately) the time counters

    ///This function stops immediately the time counters, i.e. <tt>t.halt()</tt>

    ///is a faster

    ///equivalent of the following.

    ///\code

    ///  while(t.running()) t.stop()

    ///\endcode

    ///

    ///

    ///\sa stop()

    ///\sa restart()

    ///\sa reset()

    void halt()

    {

      if(_running) {

        _running=0;

        TimeStamp t;

        t.stamp();

        start_time=t-start_time;

      }

    }

    ///Returns the running state of the timer

    ///This function returns the number of stop() exections that is

    ///necessary to really stop the timer.

    ///For example the timer

    ///is running if and only if the return value is \c true

    ///(i.e. greater than

    ///zero).

    int running()  { return _running; }

    ///Restart the time counters

    ///This function is a shorthand for

    ///a reset() and a start() calls.

    ///

    void restart()

    {

      reset();

      start();

    }

    ///@}

    ///\name Query Functions for the ellapsed time

    ///@{

    ///Gives back the ellapsed user time of the process

    double userTime() const

    {

      return operator TimeStamp().userTime();

    }

    ///Gives back the ellapsed system time of the process

    double systemTime() const

    {

      return operator TimeStamp().systemTime();

    }

    ///Gives back the ellapsed user time of the process' children

    ///\note On <tt>WIN32</tt> platform this value is not calculated.

    ///

    double cUserTime() const

    {

      return operator TimeStamp().cUserTime();

    }

    ///Gives back the ellapsed user time of the process' children

    ///\note On <tt>WIN32</tt> platform this value is not calculated.

    ///

    double cSystemTime() const

    {

      return operator TimeStamp().cSystemTime();

    }

    ///Gives back the ellapsed real time

    double realTime() const

    {

      return operator TimeStamp().realTime();

    }

    ///Computes the ellapsed time

    ///This conversion computes the ellapsed time, therefore you can print

    ///the ellapsed time like this.

    ///\code

    ///  Timer t;

    ///  doSomething();

    ///  std::cout << t << '\n';

    ///\endcode

    operator TimeStamp () const

    {

      TimeStamp t;

      t.stamp();

      return _running?t-start_time:start_time;

    }

    ///@}

  };

  ///Same as Timer but prints a report on destruction.

  ///Same as \ref Timer but prints a report on destruction.

  ///This example shows its usage.

  ///\code

  ///  void myAlg(ListGraph &g,int n)

  ///  {

  ///    TimeReport tr("Running time of myAlg: ");

  ///    ... //Here comes the algorithm

  ///  }

  ///\endcode

  ///

  ///\sa Timer

  ///\sa NoTimeReport

  class TimeReport : public Timer

  {

    std::string _title;

    std::ostream &_os;

  public:

    ///Constructor

    ///Constructor.

    ///\param title This text will be printed before the ellapsed time.

    ///\param os The stream to print the report to.

    ///\param run Sets whether the timer should start immediately.

    TimeReport(std::string title,std::ostream &os=std::cerr,bool run=true)

      : Timer(run), _title(title), _os(os){}

    ///Destructor that prints the ellapsed time

    ~TimeReport()

    {

      _os << _title << *this << std::endl;

    }

  };

  ///'Do nothing' version of TimeReport

  ///\sa TimeReport

  ///

  class NoTimeReport

  {

  public:

    ///\e

    NoTimeReport(std::string,std::ostream &,bool) {}

    ///\e

    NoTimeReport(std::string,std::ostream &) {}

    ///\e

    NoTimeReport(std::string) {}

    ///\e Do nothing.

    ~NoTimeReport() {}

    operator TimeStamp () const { return TimeStamp(); }

    void reset() {}

    void start() {}

    void stop() {}

    void halt() {}

    int running() { return 0; }

    void restart() {}

    double userTime() const { return 0; }

    double systemTime() const { return 0; }

    double cUserTime() const { return 0; }

    double cSystemTime() const { return 0; }

    double realTime() const { return 0; }

  };

  ///Tool to measure the running time more exactly.

  ///This function calls \c f several times and returns the average

  ///running time. The number of the executions will be choosen in such a way

  ///that the full real running time will be roughly between \c min_time

  ///and <tt>2*min_time</tt>.

  ///\param f the function object to be measured.

  ///\param min_time the minimum total running time.

  ///\retval num if it is not \c NULL, then the actual

  ///        number of execution of \c f will be written into <tt>*num</tt>.

  ///\retval full_time if it is not \c NULL, then the actual

  ///        total running time will be written into <tt>*full_time</tt>.

  ///\return The average running time of \c f.

  template<class F>

  TimeStamp runningTimeTest(F f,double min_time=10,unsigned int *num = NULL,

                            TimeStamp *full_time=NULL)

  {

    TimeStamp full;

    unsigned int total=0;

    Timer t;

    for(unsigned int tn=1;tn <= 1U<<31 && full.realTime()<=min_time; tn*=2) {

      for(;total<tn;total++) f();

      full=t;

    }

    if(num) *num=total;

    if(full_time) *full_time=full;

    return full/total;

  }

  /// @}

} //namespace lemon

#endif //LEMON_TIME_MEASURE_H