diff options
Diffstat (limited to 'snappy-test.cc')
-rw-r--r-- | snappy-test.cc | 39 |
1 files changed, 36 insertions, 3 deletions
diff --git a/snappy-test.cc b/snappy-test.cc index 4fb3cf4..e43a556 100644 --- a/snappy-test.cc +++ b/snappy-test.cc @@ -30,6 +30,10 @@ #include "snappy-test.h" +#ifdef HAVE_WINDOWS_H +#include <windows.h> +#endif + #include <algorithm> DEFINE_bool(run_microbenchmarks, true, @@ -64,20 +68,31 @@ int64 benchmark_cpu_time_us = 0; string *benchmark_label = NULL; int64 benchmark_bytes_processed = 0; -struct timeval benchmark_start_real; -struct rusage benchmark_start_cpu; - void ResetBenchmarkTiming() { benchmark_real_time_us = 0; benchmark_cpu_time_us = 0; } +struct timeval benchmark_start_real; + +#ifdef WIN32 +FILETIME benchmark_start_cpu; +#else // WIN32 +struct rusage benchmark_start_cpu; +#endif // WIN32 + void StartBenchmarkTiming() { gettimeofday(&benchmark_start_real, NULL); +#ifdef WIN32 + FILETIME dummy; + CHECK(GetProcessTimes( + GetCurrentProcess(), &dummy, &dummy, &dummy, &benchmark_start_cpu)); +#else if (getrusage(RUSAGE_SELF, &benchmark_start_cpu) == -1) { perror("getrusage(RUSAGE_SELF)"); exit(1); } +#endif benchmark_running = true; } @@ -92,6 +107,22 @@ void StopBenchmarkTiming() { benchmark_real_time_us += (benchmark_stop_real.tv_usec - benchmark_start_real.tv_usec); +#ifdef WIN32 + FILETIME benchmark_stop_cpu, dummy; + CHECK(GetProcessTimes( + GetCurrentProcess(), &dummy, &dummy, &dummy, &benchmark_stop_cpu)); + + ULARGE_INTEGER start_ulargeint; + start_ulargeint.LowPart = benchmark_start_cpu.dwLowDateTime; + start_ulargeint.HighPart = benchmark_start_cpu.dwHighDateTime; + + ULARGE_INTEGER stop_ulargeint; + stop_ulargeint.LowPart = benchmark_stop_cpu.dwLowDateTime; + stop_ulargeint.HighPart = benchmark_stop_cpu.dwHighDateTime; + + benchmark_cpu_time_us += + (stop_ulargeint.QuadPart - start_ulargeint.QuadPart + 5) / 10; +#else // WIN32 struct rusage benchmark_stop_cpu; if (getrusage(RUSAGE_SELF, &benchmark_stop_cpu) == -1) { perror("getrusage(RUSAGE_SELF)"); @@ -101,6 +132,8 @@ void StopBenchmarkTiming() { benchmark_start_cpu.ru_utime.tv_sec); benchmark_cpu_time_us += (benchmark_stop_cpu.ru_utime.tv_usec - benchmark_start_cpu.ru_utime.tv_usec); +#endif // WIN32 + benchmark_running = false; } |