timer.hh

// -*- coding: utf-8 -*-
// Copyright (C) by the Spot authors, see the AUTHORS file for details.
//
// This file is part of Spot, a model checking library.
//
// Spot is free software; you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 3 of the License, or
// (at your option) any later version.
//
// Spot is distributed in the hope that it will be useful, but WITHOUT
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
// or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public
// License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
#pragma once
 
#include <spot/misc/common.hh>
#include <spot/misc/_config.h>
#include <cassert>
#include <iosfwd>
#include <string>
#include <map>
#include <chrono>
#if __has_include(<sys/times.h>)
# include <sys/times.h>
#endif
#include <ctime>
#include <chrono>
 
namespace spot
{
 
 
  struct stopwatch
  {
  protected:
    typedef std::chrono::steady_clock clock;
    clock::time_point start_; 
  public:
    void start()
    {
      start_ = clock::now();
    }
 
    double stop()
    {
      auto t = clock::now();
      typedef std::chrono::duration<double> seconds;
      return std::chrono::duration_cast<seconds>(t - start_).count();
    }
  };
 
  struct time_info
  {
    time_info()
      : utime(0), stime(0), cutime(0), cstime(0)
    {
    }
    clock_t utime;   
    clock_t stime;   
    clock_t cutime;  
    clock_t cstime;  
  };
 
  class timer
  {
  public:
    timer()
      : running(false)
    {
    }
 
    void
    start()
    {
      SPOT_ASSERT(!running);
      running = true;
      wall_start_ = std::chrono::steady_clock::now();
#ifdef SPOT_HAVE_TIMES
      struct tms tmp;
      times(&tmp);
      start_.utime = tmp.tms_utime;
      start_.cutime = tmp.tms_cutime;
      start_.stime = tmp.tms_stime;
      start_.cstime = tmp.tms_cstime;
#else
      start_.utime = clock();
#endif
    }
 
    void
    stop()
    {
      auto end = std::chrono::steady_clock::now();
      wall_cumul_ += std::chrono::duration_cast
        <std::chrono::milliseconds>(end - wall_start_).count();
#ifdef SPOT_HAVE_TIMES
      struct tms tmp;
      times(&tmp);
      total_.utime += tmp.tms_utime - start_.utime;
      total_.cutime += tmp.tms_cutime - start_.cutime;
      total_.stime += tmp.tms_stime - start_.stime;
      total_.cstime += tmp.tms_cstime - start_.cstime;
#else
      total_.utime += clock() - start_.utime;
#endif
      SPOT_ASSERT(running);
      running = false;
    }
 
    clock_t
    utime() const
    {
      return total_.utime;
    }
 
    clock_t
    cutime() const
    {
      return total_.cutime;
    }
 
    clock_t
    stime() const
    {
      return total_.stime;
    }
 
    clock_t
    cstime() const
    {
      return total_.cstime;
    }
 
    clock_t get_uscp(bool user, bool system, bool children, bool parent) const
    {
      clock_t res = 0;
 
      if (user && parent)
        res += utime();
 
      if (user && children)
        res += cutime();
 
      if (system && parent)
        res += stime();
 
      if (system && children)
        res += cstime();
 
      return res;
    }
 
    bool
    is_running() const
    {
      return running;
    }
 
    std::chrono::milliseconds::rep
    walltime() const
    {
      return wall_cumul_;
    }
 
  protected:
    time_info start_; 
    time_info total_; 
    bool running;     
    std::chrono::steady_clock::time_point wall_start_;
    std::chrono::milliseconds::rep wall_cumul_ = 0; 
  };
 
  // This function declared here must be implemented in each file
  // that includes this header, well, only if this operator is needed!
  inline std::ostream& operator<<(std::ostream& os, const timer& dt);
 
  class timer_map
  {
  public:
 
    void
    start(const std::string& name)
    {
      item_type& it = tm[name];
      it.first.start();
      ++it.second;
    }
 
    void
    stop(const std::string& name)
    {
      tm[name].first.stop();
    }
 
    void
    cancel(const std::string& name)
    {
      tm_type::iterator i = tm.find(name);
      if (SPOT_UNLIKELY(i == tm.end()))
        throw std::invalid_argument("timer_map::cancel(): unknown name");
      SPOT_ASSERT(0 < i->second.second);
      if (0 == --i->second.second)
        tm.erase(i);
    }
 
    const spot::timer&
    timer(const std::string& name) const
    {
      tm_type::const_iterator i = tm.find(name);
      if (SPOT_UNLIKELY(i == tm.end()))
        throw std::invalid_argument("timer_map::timer(): unknown name");
      return i->second.first;
    }
 
    bool
    empty() const
    {
      return tm.empty();
    }
 
    SPOT_API std::ostream&
    print(std::ostream& os) const;
 
    void
    reset_all()
    {
      tm.clear();
    }
 
  protected:
    typedef std::pair<spot::timer, int> item_type; 
    typedef std::map<std::string, item_type> tm_type; 
    tm_type tm; 
  };
 
  typedef struct process_timer
  {
    void start()
    {
      walltimer.start();
      cputimer.start();
    }
    // sw.stop() --> It always returns the duration since the last call to
    // start(). Therefore, it won't stop timing, moreover, it can be called
    // multiple times.
    void stop()
    {
      walltime_lap_ = walltimer.stop();
      cputimer.stop();
    }
 
    double walltime() const
    {
      return walltime_lap_;
    }
 
    clock_t cputime(bool user, bool system, bool children, bool parent) const
    {
      return cputimer.get_uscp(user, system, children, parent);
    }
 
  private:
    spot::timer cputimer;
    spot::stopwatch walltimer;
    double walltime_lap_ = 0;
  } process_timer;
 
}