]>
Commit | Line | Data |
---|---|---|
1 | //----------------------------------------------------------------------------- | |
2 | // Copyright (C) 2010 iZsh <izsh at fail0verflow.com> | |
3 | // | |
4 | // This code is licensed to you under the terms of the GNU GPL, version 2 or, | |
5 | // at your option, any later version. See the LICENSE.txt file for the text of | |
6 | // the license. | |
7 | //----------------------------------------------------------------------------- | |
8 | // utilities requiring Posix library functions | |
9 | //----------------------------------------------------------------------------- | |
10 | ||
11 | #if !defined(_WIN32) | |
12 | #define _POSIX_C_SOURCE 199309L // need nanosleep() | |
13 | #else | |
14 | #include <windows.h> | |
15 | #endif | |
16 | ||
17 | #include "util_posix.h" | |
18 | #include <stdint.h> | |
19 | #include <time.h> | |
20 | ||
21 | ||
22 | // Timer functions | |
23 | #if !defined (_WIN32) | |
24 | #include <errno.h> | |
25 | ||
26 | static void nsleep(uint64_t n) { | |
27 | struct timespec timeout; | |
28 | timeout.tv_sec = n/1000000000; | |
29 | timeout.tv_nsec = n%1000000000; | |
30 | while (nanosleep(&timeout, &timeout) && errno == EINTR); | |
31 | } | |
32 | ||
33 | void msleep(uint32_t n) { | |
34 | nsleep(1000000 * (uint64_t)n); | |
35 | } | |
36 | #endif // _WIN32 | |
37 | ||
38 | #ifdef __APPLE__ | |
39 | ||
40 | #ifndef CLOCK_MONOTONIC | |
41 | #define CLOCK_MONOTONIC (1) | |
42 | #endif | |
43 | #ifndef CLOCK_REALTIME | |
44 | #define CLOCK_REALTIME (2) | |
45 | #endif | |
46 | ||
47 | #include <sys/time.h> | |
48 | #include <mach/clock.h> | |
49 | #include <mach/mach.h> | |
50 | #include <mach/mach_time.h> | |
51 | ||
52 | /* clock_gettime is not implemented on OSX prior to 10.12 */ | |
53 | int _civet_clock_gettime(int clk_id, struct timespec *t); | |
54 | ||
55 | int _civet_clock_gettime(int clk_id, struct timespec *t) | |
56 | { | |
57 | memset(t, 0, sizeof(*t)); | |
58 | if (clk_id == CLOCK_REALTIME) { | |
59 | struct timeval now; | |
60 | int rv = gettimeofday(&now, NULL); | |
61 | if (rv) { | |
62 | return rv; | |
63 | } | |
64 | t->tv_sec = now.tv_sec; | |
65 | t->tv_nsec = now.tv_usec * 1000; | |
66 | return 0; | |
67 | ||
68 | } else if (clk_id == CLOCK_MONOTONIC) { | |
69 | static uint64_t clock_start_time = 0; | |
70 | static mach_timebase_info_data_t timebase_info = {0, 0}; | |
71 | ||
72 | uint64_t now = mach_absolute_time(); | |
73 | ||
74 | if (clock_start_time == 0) { | |
75 | mach_timebase_info(&timebase_info); | |
76 | clock_start_time = now; | |
77 | } | |
78 | ||
79 | now = (uint64_t)((double)(now - clock_start_time) | |
80 | * (double)timebase_info.numer | |
81 | / (double)timebase_info.denom); | |
82 | ||
83 | t->tv_sec = now / 1000000000; | |
84 | t->tv_nsec = now % 1000000000; | |
85 | return 0; | |
86 | } | |
87 | return -1; // EINVAL - Clock ID is unknown | |
88 | } | |
89 | ||
90 | /* if clock_gettime is declared, then __CLOCK_AVAILABILITY will be defined */ | |
91 | #ifdef __CLOCK_AVAILABILITY | |
92 | /* If we compiled with Mac OSX 10.12 or later, then clock_gettime will be declared | |
93 | * but it may be NULL at runtime. So we need to check before using it. */ | |
94 | int _civet_safe_clock_gettime(int clk_id, struct timespec *t); | |
95 | ||
96 | int _civet_safe_clock_gettime(int clk_id, struct timespec *t) { | |
97 | if( clock_gettime ) { | |
98 | return clock_gettime(clk_id, t); | |
99 | } | |
100 | return _civet_clock_gettime(clk_id, t); | |
101 | } | |
102 | #define clock_gettime _civet_safe_clock_gettime | |
103 | #else | |
104 | #define clock_gettime _civet_clock_gettime | |
105 | #endif | |
106 | ||
107 | #endif | |
108 | ||
109 | ||
110 | // a milliseconds timer for performance measurement | |
111 | uint64_t msclock() { | |
112 | #if defined(_WIN32) | |
113 | #include <sys/types.h> | |
114 | ||
115 | // WORKAROUND FOR MinGW (some versions - use if normal code does not compile) | |
116 | // It has no _ftime_s and needs explicit inclusion of timeb.h | |
117 | #include <sys/timeb.h> | |
118 | struct _timeb t; | |
119 | _ftime(&t); | |
120 | return 1000 * t.time + t.millitm; | |
121 | ||
122 | // NORMAL CODE (use _ftime_s) | |
123 | //struct _timeb t; | |
124 | //if (_ftime_s(&t)) { | |
125 | // return 0; | |
126 | //} else { | |
127 | // return 1000 * t.time + t.millitm; | |
128 | //} | |
129 | #else | |
130 | struct timespec t; | |
131 | clock_gettime(CLOCK_MONOTONIC, &t); | |
132 | return (t.tv_sec * 1000 + t.tv_nsec / 1000000); | |
133 | #endif | |
134 | } | |
135 |