Remove the need for the custom SPHINXBUILD plugin.
[m6w6/libmemcached] / libtest / timer.hpp
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * Data Differential YATL (i.e. libtest) library
4 *
5 * Copyright (C) 2012 Data Differential, http://datadifferential.com/
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are
9 * met:
10 *
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 *
14 * * Redistributions in binary form must reproduce the above
15 * copyright notice, this list of conditions and the following disclaimer
16 * in the documentation and/or other materials provided with the
17 * distribution.
18 *
19 * * The names of its contributors may not be used to endorse or
20 * promote products derived from this software without specific prior
21 * written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
29 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
30 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
31 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
32 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
33 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34 *
35 */
36
37 #pragma once
38
39 #include <cstdlib>
40 #include <ctime>
41 #include <iostream>
42
43 #ifdef __MACH__
44 # include <mach/clock.h>
45 # include <mach/mach.h>
46 #else
47 # include <sys/time.h>
48 #endif
49
50
51 namespace libtest {
52
53 class Timer {
54 public:
55
56 Timer()
57 {
58 _begin.tv_sec= 0;
59 _begin.tv_nsec= 0;
60 _end.tv_sec= 0;
61 _end.tv_nsec= 0;
62 }
63
64 void reset()
65 {
66 _end.tv_sec= 0;
67 _end.tv_nsec= 0;
68 _time(_begin);
69 }
70
71 void sample()
72 {
73 _time(_end);
74 }
75
76 void offset(int64_t minutes_arg, int64_t seconds_arg, int64_t nanoseconds)
77 {
78 reset();
79 _end= _begin;
80 _end.tv_sec+= (minutes_arg * 60) +seconds_arg;
81 _end.tv_nsec+= nanoseconds;
82 }
83
84 int64_t minutes()
85 {
86 struct timespec result;
87 difference(result);
88 return int64_t(result.tv_sec / 60);
89 }
90
91 uint64_t elapsed_milliseconds() const
92 {
93 struct timespec temp;
94 difference(temp);
95
96 return temp.tv_sec*1000 +temp.tv_nsec/1000000;
97 }
98
99 void difference(struct timespec& arg) const
100 {
101 if ((_end.tv_nsec -_begin.tv_nsec) < 0)
102 {
103 arg.tv_sec= _end.tv_sec -_begin.tv_sec -1;
104 arg.tv_nsec= 1000000000 +_end.tv_nsec -_begin.tv_nsec;
105
106 }
107 else
108 {
109 arg.tv_sec= _end.tv_sec -_begin.tv_sec;
110 arg.tv_nsec= _end.tv_nsec -_begin.tv_nsec;
111 }
112 }
113
114 private:
115 void _time(struct timespec& ts)
116 {
117 #ifdef __MACH__ // OSX lacks clock_gettime()
118 clock_serv_t _clock_serv;
119 mach_timespec_t _mach_timespec;
120 host_get_clock_service(mach_host_self(), CALENDAR_CLOCK, &_clock_serv);
121 clock_get_time(_clock_serv, &_mach_timespec);
122 mach_port_deallocate(mach_task_self(), _clock_serv);
123 ts.tv_sec= _mach_timespec.tv_sec;
124 ts.tv_nsec= _mach_timespec.tv_nsec;
125 #else
126 clock_gettime(CLOCK_REALTIME, &ts);
127 #endif
128 }
129
130 private:
131 struct timespec _begin;
132 struct timespec _end;
133 };
134
135 std::ostream& operator<<(std::ostream& output, const libtest::Timer& arg);
136
137 } // namespace libtest