Blame view

src/common/inc/timer.h 1.58 KB
e41a52bb   Hu Chunming   1.优化数据读取线程;2. 添加A...
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
  /**
   * Copyright 1993-2013 NVIDIA Corporation.  All rights reserved.
   *
   * Please refer to the NVIDIA end user license agreement (EULA) associated
   * with this source code for terms and conditions that govern your use of
   * this software. Any use, reproduction, disclosure, or distribution of
   * this software and related documentation outside the terms of the EULA
   * is strictly prohibited.
   *
   */
  
  #ifndef TIMER_H
  #define TIMER_H
  
  #include <stdlib.h>
  
  #if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
  #define WIN32_LEAN_AND_MEAN
  #include <windows.h>
  #else
  #include <sys/time.h>
  #endif
  
  #if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
  double PCFreq = 0.0;
  __int64 timerStart = 0;
  #else
  struct timeval timerStart;
  #endif
  
  void StartTimer()
  {
  #if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
      LARGE_INTEGER li;
  
      if (!QueryPerformanceFrequency(&li))
      {
          printf("QueryPerformanceFrequency failed!\n");
      }
  
      PCFreq = (double)li.QuadPart/1000.0;
      QueryPerformanceCounter(&li);
      timerStart = li.QuadPart;
  #else
      gettimeofday(&timerStart, NULL);
  #endif
  }
  
  // time elapsed in ms
  double GetTimer()
  {
  #if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
      LARGE_INTEGER li;
      QueryPerformanceCounter(&li);
      return (double)(li.QuadPart-timerStart)/PCFreq;
  #else
      struct timeval timerStop, timerElapsed;
      gettimeofday(&timerStop, NULL);
      timersub(&timerStop, &timerStart, &timerElapsed);
      return timerElapsed.tv_sec*1000.0+timerElapsed.tv_usec/1000.0;
  #endif
  }
  #endif // TIMER_H