timer.h

Go to the documentation of this file.

#ifndef TIMER_H
#define TIMER_H
 
#include <ctime>
#include <fstream>
#include <iomanip>
#include <iostream>
#include <map>
#include <string>
 
namespace ModuleBase
{
class timer
{
  public:
    struct Timer_One
    {
        double cpu_start;
        double cpu_second = 0.0;
        size_t calls = 0;
        size_t order = n_now++;
        bool start_flag = true;
    };
 
    static std::map<std::string, std::map<std::string, Timer_One>> timer_pool;
 
    static void start(const std::string &class_name_in, const std::string &name_in);
    static void end(const std::string &class_name_in, const std::string &name_in);
 
    static void start(void);
 
    static void finish(std::ofstream &ofs, const bool print_flag = true, const bool check_end = true);
 
    static void enable(void)
    {
        disabled = false;
    }
 
    static void set_nvtx_enabled(bool b)
    {
        enable_nvtx_ = b;
    }
 
    static void disable(void)
    {
        disabled = true;
    }
 
    static void write_to_json(std::string file_name);
 
    static void print_all(std::ofstream &ofs, const bool check_end);
 
    static long double print_until_now(void);
 
  private:
    static bool disabled;
 
    static bool enable_nvtx_;
 
    static size_t n_now;
 
    static double cpu_time(void);
};
 
} // namespace ModuleBase
#endif