xapian-letor/tests/harness/cputimer.cc

   1 /** @file cputimer.cc
   2  * @brief Measure CPU time.
   3  */
   4 /* Copyright (C) 2009,2015 Olly Betts
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License as
   8  * published by the Free Software Foundation; either version 2 of the
   9  * License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  19  */
  20
  21 #include <config.h>
  22
  23 #include "cputimer.h"
  24
  25 #include "testsuite.h"
  26
  27 #ifdef HAVE_GETRUSAGE
  28 # include <sys/time.h>
  29 # include <sys/resource.h>
  30 #elif defined HAVE_TIMES
  31 # include <sys/times.h>
  32 # ifdef HAVE_SYSCONF
  33 #  include "safeunistd.h"
  34 # endif
  35 #elif defined HAVE_FTIME
  36 # include <sys/timeb.h>
  37 #else
  38 # include <ctime>
  39 #endif
  40
  41 #include <cerrno>
  42 #include <cstdlib>
  43 #include <cstring>
  44 #include <string>
  45
  46 using namespace std;
  47
  48 double
  49 CPUTimer::get_current_cputime() const
  50 {
  51     static bool skip = (getenv("AUTOMATED_TESTING") != NULL);
  52     if (skip) {
  53         SKIP_TEST("Skipping timed test because $AUTOMATED_TESTING is set");
  54     }
  55
  56     double t = 0;
  57 #ifdef HAVE_GETRUSAGE
  58     struct rusage r;
  59     if (getrusage(RUSAGE_SELF, &r) == -1) {
  60         FAIL_TEST("Couldn't measure CPU for self: " << strerror(errno));
  61     }
  62
  63     t = r.ru_utime.tv_sec + r.ru_stime.tv_sec;
  64     t += (r.ru_utime.tv_usec + r.ru_stime.tv_usec) * 0.000001;
  65 #elif defined HAVE_TIMES
  66     struct tms b;
  67     if (times(&b) == clock_t(-1)) {
  68         FAIL_TEST("Couldn't measure CPU: " << strerror(errno));
  69     }
  70     t = (double)(b.tms_utime + b.tms_stime);
  71 # ifdef HAVE_SYSCONF
  72     t /= sysconf(_SC_CLK_TCK);
  73 # else
  74     t /= CLK_TCK;
  75 # endif
  76 #else
  77     // FIXME: Fallback to just using wallclock time, which is probably only
  78     // going to be used on Microsoft Windows, where nobody has implemented
  79     // the code required to get the CPU time used by a process.
  80 # ifdef HAVE_FTIME
  81     struct timeb tb;
  82 #  ifdef FTIME_RETURNS_VOID
  83     ftime(&tb);
  84     t = tb.time + (tb.millitm * 0.001);
  85 #  else
  86     if (ftime(&tb) == -1) {
  87         t = time(NULL);
  88     } else {
  89         t = tb.time + (tb.millitm * 0.001);
  90     }
  91 #  endif
  92 # else
  93     t = time(NULL);
  94 # endif
  95 #endif
  96
  97     return t;
  98 }