oidn-protoray/sys/timer.h at master · RenderKit/oidn-protoray · GitHub

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
81
82
83
84
85
86
87
88
89
// Copyright 2015 Intel Corporation
// SPDX-License-Identifier: Apache-2.0

#pragma once

#if defined(_WIN32)
#define WIN32_LEAN_AND_MEAN
#define NOMINMAX
#include <Windows.h>
#elif defined(__APPLE__)
#include <mach/mach_time.h>
#else
#include <ctime>
#endif

#include <cassert>

namespace prt {

class Timer
{
private:
#if defined(_WIN32)
  double invCountsPerSec;
  LARGE_INTEGER startCount;
#elif defined(__APPLE__)
  double invCountsPerSec;
  uint64_t startTime;
#else
  timespec startTime;
#endif

public:
  Timer()
  {
#if defined(_WIN32)
    LARGE_INTEGER frequency;

    BOOL result = QueryPerformanceFrequency(&frequency);
    assert(result != 0 && "timer is not supported");

    invCountsPerSec = 1.0 / (double)frequency.QuadPart;
#elif defined(__APPLE__)
    mach_timebase_info_data_t timebaseInfo;
    mach_timebase_info(&timebaseInfo);
    invCountsPerSec = (double)timebaseInfo.numer / (double)timebaseInfo.denom * 1e-9;
#endif

    reset();
  }

  void reset()
  {
#if defined(_WIN32)
    BOOL result = QueryPerformanceCounter(&startCount);
    assert(result != 0 && "could not query counter");
#elif defined(__APPLE__)
    startTime = mach_absolute_time();
#else
    int result = clock_gettime(CLOCK_MONOTONIC, &startTime);
    assert(result == 0 && "could not get time");
#endif
  }

  double query() const
  {
#if defined(_WIN32)
    LARGE_INTEGER currentCount;

    BOOL result = QueryPerformanceCounter(&currentCount);
    assert(result != 0 && "could not query counter");

    return (currentCount.QuadPart - startCount.QuadPart) * invCountsPerSec;
#elif defined(__APPLE__)
    uint64_t endTime = mach_absolute_time();
    uint64_t elapsedTime = endTime - startTime;
    return (double)elapsedTime * invCountsPerSec;
#else
    timespec currentTime;

    int result = clock_gettime(CLOCK_MONOTONIC, &currentTime);
    assert(result == 0 && "could not get time");

    return (double)(currentTime.tv_sec - startTime.tv_sec) + (double)(currentTime.tv_nsec - startTime.tv_nsec) * 1e-9;
#endif
  }
};

} // namespace prt