blob: 8da0a020d99ff0cf20446304e7ec49c4425b15c2 (
plain)
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
|
/* SPDX-License-Identifier: GPL-2.0-or-later */
#pragma once
#include <chrono>
#include <iostream>
#include <string>
#include "BLI_sys_types.h"
namespace blender::timeit {
using Clock = std::chrono::steady_clock;
using TimePoint = Clock::time_point;
using Nanoseconds = std::chrono::nanoseconds;
void print_duration(Nanoseconds duration);
class ScopedTimer {
private:
std::string name_;
TimePoint start_;
public:
ScopedTimer(std::string name) : name_(std::move(name))
{
start_ = Clock::now();
}
~ScopedTimer()
{
const TimePoint end = Clock::now();
const Nanoseconds duration = end - start_;
std::cout << "Timer '" << name_ << "' took ";
print_duration(duration);
std::cout << '\n';
}
};
class ScopedTimerAveraged {
private:
std::string name_;
TimePoint start_;
int64_t &total_count_;
Nanoseconds &total_time_;
Nanoseconds &min_time_;
public:
ScopedTimerAveraged(std::string name,
int64_t &total_count,
Nanoseconds &total_time,
Nanoseconds &min_time)
: name_(std::move(name)),
total_count_(total_count),
total_time_(total_time),
min_time_(min_time)
{
start_ = Clock::now();
}
~ScopedTimerAveraged();
};
} // namespace blender::timeit
#define SCOPED_TIMER(name) blender::timeit::ScopedTimer scoped_timer(name)
/**
* Print the average and minimum runtime of the timer's scope.
* \warning This uses static variables, so it is not thread-safe.
*/
#define SCOPED_TIMER_AVERAGED(name) \
static int64_t total_count_; \
static blender::timeit::Nanoseconds total_time_; \
static blender::timeit::Nanoseconds min_time_ = blender::timeit::Nanoseconds::max(); \
blender::timeit::ScopedTimerAveraged scoped_timer(name, total_count_, total_time_, min_time_)
|