timer_wrapper.h

Go to the documentation of this file.

#ifndef TIMER_WRAPPER_H
#define TIMER_WRAPPER_H
 
#include <chrono>
 
#ifdef __MPI
#include <mpi.h>
#endif
 
namespace ModuleBase {
 
typedef double TimePoint;
 
inline TimePoint get_time()
{
#ifdef __MPI
    int is_initialized = 0;
    MPI_Initialized(&is_initialized);
    if (is_initialized)
    {
        return MPI_Wtime();
    }
    else
    {
        return std::chrono::duration_cast<std::chrono::microseconds>(
            std::chrono::system_clock::now().time_since_epoch()).count() / 1e6;
    }
#else
    return std::chrono::duration_cast<std::chrono::microseconds>(
        std::chrono::system_clock::now().time_since_epoch()).count() / 1e6;
#endif
}
 
inline double get_duration(const TimePoint& start, const TimePoint& end)
{
    return end - start;
}
 
}
 
#endif // TIMER_WRAPPER_H