ghc/lib/std/cbits/getCPUTime.lc

   1 %
   2 % (c) The GRASP/AQUA Project, Glasgow University, 1995
   3 %
   4 \subsection[getCPUTime.lc]{getCPUTime Runtime Support}
   5
   6 \begin{code}
   7
   8 #ifndef _AIX
   9 #define NON_POSIX_SOURCE /*needed for solaris2 only?*/
  10 #endif
  11
  12 /* how is this to work given we have not read platform.h yet? */
  13 #ifdef hpux_TARGET_OS
  14 #define _INCLUDE_HPUX_SOURCE
  15 #endif
  16
  17 #include "rtsdefs.h"
  18
  19 #ifdef HAVE_SYS_TYPES_H
  20 #include <sys/types.h>
  21 #endif
  22
  23 #ifdef HAVE_UNISTD_H
  24 #include <unistd.h>
  25 #endif
  26
  27 #ifdef HAVE_SYS_TIMES_H
  28 #include <sys/times.h>
  29 #endif
  30
  31 #ifdef HAVE_SYS_TIME_H
  32 #include <sys/time.h>
  33 #endif
  34
  35 #if defined(HAVE_SYS_RESOURCE_H) && ! irix_TARGET_OS
  36 #include <sys/resource.h>
  37 #endif
  38
  39 #ifdef HAVE_SYS_TIMEB_H
  40 #include <sys/timeb.h>
  41 #endif
  42
  43 #ifdef hpux_TARGET_OS
  44 #include <sys/syscall.h>
  45 #define getrusage(a, b)  syscall(SYS_GETRUSAGE, a, b)
  46 #define HAVE_GETRUSAGE
  47 #endif
  48
  49 StgInt
  50 clockTicks ()
  51 {
  52  return (
  53 #if defined(CLK_TCK)
  54     CLK_TCK
  55 #else
  56     sysconf(_SC_CLK_TCK)
  57 #endif
  58     );
  59 }
  60
  61 /*
  62  * Our caller wants a pointer to four StgInts,
  63  * user seconds, user nanoseconds, system seconds, system nanoseconds.
  64  * Yes, the timerval has unsigned components, but nanoseconds take only
  65  * 30 bits, and our CPU usage would have to be over 68 years for the
  66  * seconds to overflow 31 bits.
  67  */
  68
  69 StgByteArray
  70 getCPUTime(StgByteArray cpuStruct)
  71 {
  72     StgInt *cpu=(StgInt *)cpuStruct;
  73
  74 #if defined(HAVE_GETRUSAGE) && ! irix_TARGET_OS
  75     struct rusage t;
  76
  77     getrusage(RUSAGE_SELF, &t);
  78     cpu[0] = t.ru_utime.tv_sec;
  79     cpu[1] = 1000 * t.ru_utime.tv_usec;
  80     cpu[2] = t.ru_stime.tv_sec;
  81     cpu[3] = 1000 * t.ru_stime.tv_usec;
  82
  83 #else
  84 # if defined(HAVE_TIMES)
  85     struct tms t;
  86 #  if defined(CLK_TCK)
  87 #   define ticks CLK_TCK
  88 #  else
  89     long ticks;
  90     ticks = sysconf(_SC_CLK_TCK);
  91 #  endif
  92
  93     times(&t);
  94     cpu[0] = t.tms_utime / ticks;
  95     cpu[1] = (t.tms_utime - cpu[0] * ticks) * (1000000000 / ticks);
  96     cpu[2] = t.tms_stime / ticks;
  97     cpu[3] = (t.tms_stime - cpu[2] * ticks) * (1000000000 / ticks);
  98
  99 # else
 100     return NULL;
 101 # endif
 102 #endif
 103     return (StgByteArray) cpuStruct;
 104 }
 105
 106 \end{code}
 107