File size: 1,506 Bytes
f6dd1c2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | #ifndef OMPHELPER_H_
#define OMPHELPER_H_
#define USE_OPENMP
#ifdef USE_OPENMP
#include <omp.h>
#ifdef USE_MSVC
#define OMP_PARALLEL __pragma(omp parallel)
#define OMP_FOR __pragma(omp for)
#define OMP_SINGLE __pragma(omp single)
#define OMP_SECTIONS __pragma(omp sections)
#define OMP_SECTION __pragma(omp section)
#else
#define OMP_PARALLEL _Pragma("omp parallel")
#define OMP_FOR _Pragma("omp for")
#define OMP_SINGLE _Pragma("omp single")
#define OMP_SECTIONS _Pragma("omp sections")
#define OMP_SECTION _Pragma("omp section")
#endif
#else
#include <ctime>
#define OMP_PARALLEL
#define OMP_FOR
#define OMP_SINGLE
#define OMP_SECTIONS
#define OMP_SECTION
#endif
#include <cassert>
#include <vector>
class Timer
{
public:
typedef int EventID;
EventID get_time()
{
EventID id = time_values_.size();
#ifdef USE_OPENMP
time_values_.push_back(omp_get_wtime());
#else
time_values_.push_back(clock());
#endif
return id;
}
double elapsed_time(EventID event1, EventID event2)
{
assert(event1 >= 0 && event1 < static_cast<EventID>(time_values_.size()));
assert(event2 >= 0 && event2 < static_cast<EventID>(time_values_.size()));
#ifdef USE_OPENMP
return time_values_[event2] - time_values_[event1];
#else
return double(time_values_[event2] - time_values_[event1]) / CLOCKS_PER_SEC;
#endif
}
void reset()
{
time_values_.clear();
}
private:
#ifdef USE_OPENMP
std::vector<double> time_values_;
#else
std::vector<clock_t> time_values_;
#endif
};
#endif /* OMPHELPER_H_ */
|