Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 1 | /* Copyright 2013-2014 Pierre Ossman <ossman@cendio.se> for Cendio AB |
| 2 | * |
| 3 | * This is free software; you can redistribute it and/or modify |
| 4 | * it under the terms of the GNU General Public License as published by |
| 5 | * the Free Software Foundation; either version 2 of the License, or |
| 6 | * (at your option) any later version. |
| 7 | * |
| 8 | * This software is distributed in the hope that it will be useful, |
| 9 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 10 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 11 | * GNU General Public License for more details. |
| 12 | * |
| 13 | * You should have received a copy of the GNU General Public License |
| 14 | * along with this software; if not, write to the Free Software |
| 15 | * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, |
| 16 | * USA. |
| 17 | */ |
| 18 | |
| 19 | #include <stdint.h> |
Pierre Ossman | 8ac3111 | 2015-02-06 13:50:36 +0100 | [diff] [blame] | 20 | #include <stdlib.h> |
| 21 | #include <string.h> |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 22 | |
| 23 | #ifdef WIN32 |
| 24 | #include <windows.h> |
| 25 | #else |
| 26 | #include <sys/resource.h> |
| 27 | #endif |
| 28 | |
Pierre Ossman | 8ac3111 | 2015-02-06 13:50:36 +0100 | [diff] [blame] | 29 | #include "util.h" |
| 30 | |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 31 | #ifdef WIN32 |
Pierre Ossman | b591c9d | 2015-11-09 15:29:30 +0100 | [diff] [blame] | 32 | typedef struct { |
| 33 | FILETIME kernelTime; |
| 34 | FILETIME userTime; |
| 35 | } syscounter_t; |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 36 | #else |
Pierre Ossman | 8ac3111 | 2015-02-06 13:50:36 +0100 | [diff] [blame] | 37 | typedef struct rusage syscounter_t; |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 38 | #endif |
| 39 | |
Pierre Ossman | 8ac3111 | 2015-02-06 13:50:36 +0100 | [diff] [blame] | 40 | static syscounter_t _globalCounter[2]; |
| 41 | static cpucounter_t globalCounter = _globalCounter; |
| 42 | |
| 43 | void startCpuCounter(void) |
| 44 | { |
| 45 | startCpuCounter(globalCounter); |
| 46 | } |
| 47 | |
| 48 | void endCpuCounter(void) |
| 49 | { |
| 50 | endCpuCounter(globalCounter); |
| 51 | } |
| 52 | |
| 53 | double getCpuCounter(void) |
| 54 | { |
| 55 | return getCpuCounter(globalCounter); |
| 56 | } |
| 57 | |
| 58 | cpucounter_t newCpuCounter(void) |
| 59 | { |
| 60 | syscounter_t *c; |
| 61 | |
| 62 | c = (syscounter_t*)malloc(sizeof(syscounter_t) * 2); |
| 63 | if (c == NULL) |
| 64 | return NULL; |
| 65 | |
| 66 | memset(c, 0, sizeof(syscounter_t) * 2); |
| 67 | |
| 68 | return c; |
| 69 | } |
| 70 | |
| 71 | void freeCpuCounter(cpucounter_t c) |
| 72 | { |
| 73 | free(c); |
| 74 | } |
| 75 | |
| 76 | static void measureCpu(syscounter_t *counter) |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 77 | { |
| 78 | #ifdef WIN32 |
Pierre Ossman | b591c9d | 2015-11-09 15:29:30 +0100 | [diff] [blame] | 79 | FILETIME dummy1, dummy2; |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 80 | |
| 81 | GetProcessTimes(GetCurrentProcess(), &dummy1, &dummy2, |
Pierre Ossman | b591c9d | 2015-11-09 15:29:30 +0100 | [diff] [blame] | 82 | &counter->kernelTime, &counter->userTime); |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 83 | #else |
Pierre Ossman | 8ac3111 | 2015-02-06 13:50:36 +0100 | [diff] [blame] | 84 | getrusage(RUSAGE_SELF, counter); |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 85 | #endif |
| 86 | } |
| 87 | |
Pierre Ossman | 8ac3111 | 2015-02-06 13:50:36 +0100 | [diff] [blame] | 88 | void startCpuCounter(cpucounter_t c) |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 89 | { |
Pierre Ossman | 8ac3111 | 2015-02-06 13:50:36 +0100 | [diff] [blame] | 90 | syscounter_t *s = (syscounter_t*)c; |
| 91 | measureCpu(&s[0]); |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 92 | } |
| 93 | |
Pierre Ossman | 8ac3111 | 2015-02-06 13:50:36 +0100 | [diff] [blame] | 94 | void endCpuCounter(cpucounter_t c) |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 95 | { |
Pierre Ossman | 8ac3111 | 2015-02-06 13:50:36 +0100 | [diff] [blame] | 96 | syscounter_t *s = (syscounter_t*)c; |
| 97 | measureCpu(&s[1]); |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 98 | } |
| 99 | |
Pierre Ossman | 8ac3111 | 2015-02-06 13:50:36 +0100 | [diff] [blame] | 100 | double getCpuCounter(cpucounter_t c) |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 101 | { |
Pierre Ossman | 8ac3111 | 2015-02-06 13:50:36 +0100 | [diff] [blame] | 102 | syscounter_t *s = (syscounter_t*)c; |
Pierre Ossman | b591c9d | 2015-11-09 15:29:30 +0100 | [diff] [blame] | 103 | double sysSeconds, userSeconds; |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 104 | |
| 105 | #ifdef WIN32 |
| 106 | uint64_t counters[2]; |
| 107 | |
Pierre Ossman | b591c9d | 2015-11-09 15:29:30 +0100 | [diff] [blame] | 108 | counters[0] = (uint64_t)s[0].kernelTime.dwHighDateTime << 32 | |
| 109 | s[0].kernelTime.dwLowDateTime; |
| 110 | counters[1] = (uint64_t)s[1].kernelTime.dwHighDateTime << 32 | |
| 111 | s[1].kernelTime.dwLowDateTime; |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 112 | |
Pierre Ossman | b591c9d | 2015-11-09 15:29:30 +0100 | [diff] [blame] | 113 | sysSeconds = (double)(counters[1] - counters[0]) / 10000000.0; |
| 114 | |
| 115 | counters[0] = (uint64_t)s[0].userTime.dwHighDateTime << 32 | |
| 116 | s[0].userTime.dwLowDateTime; |
| 117 | counters[1] = (uint64_t)s[1].userTime.dwHighDateTime << 32 | |
| 118 | s[1].userTime.dwLowDateTime; |
| 119 | |
| 120 | userSeconds = (double)(counters[1] - counters[0]) / 10000000.0; |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 121 | #else |
Pierre Ossman | b591c9d | 2015-11-09 15:29:30 +0100 | [diff] [blame] | 122 | sysSeconds = (double)(s[1].ru_stime.tv_sec - |
| 123 | s[0].ru_stime.tv_sec); |
| 124 | sysSeconds += (double)(s[1].ru_stime.tv_usec - |
| 125 | s[0].ru_stime.tv_usec) / 1000000.0; |
| 126 | |
| 127 | userSeconds = (double)(s[1].ru_utime.tv_sec - |
| 128 | s[0].ru_utime.tv_sec); |
| 129 | userSeconds += (double)(s[1].ru_utime.tv_usec - |
| 130 | s[0].ru_utime.tv_usec) / 1000000.0; |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 131 | #endif |
| 132 | |
Pierre Ossman | b591c9d | 2015-11-09 15:29:30 +0100 | [diff] [blame] | 133 | return sysSeconds + userSeconds; |
Pierre Ossman | 236c03c | 2014-07-04 14:12:49 +0200 | [diff] [blame] | 134 | } |