m4: ax_assert

[awesomized/libmemcached] / libtest / timer.cc

diff --git a/libtest/timer.cc b/libtest/timer.cc
index 0a8d69b548c71caabb4e52fd6d27d0f77147ebb9..fae814cad09c61f2e446102aee4659c377ff77a8 100644 (file)
--- a/libtest/timer.cc
+++ b/libtest/timer.cc
@@ -34,20 +34,115 @@
  *
  */
 
-#include <config.h>
+#include "libtest/yatlcon.h"
+
 #include <libtest/timer.hpp>
+
 #include <ctime>
+#include <iomanip>
+
+#ifdef __MACH__
+# include <mach/clock.h>
+# include <mach/mach.h>
+#else
+# include <sys/time.h>
+#endif
 
 namespace libtest {
 
+Timer::Timer()
+{
+  _begin.tv_sec= 0;
+  _begin.tv_nsec= 0;
+  _end.tv_sec= 0;
+  _end.tv_nsec= 0;
+}
+
+void Timer::reset()
+{
+  _end.tv_sec= 0;
+  _end.tv_nsec= 0;
+  _time(_begin);
+}
+
+void Timer::sample()
+{
+  _time(_end);
+}
+
+void Timer::offset(int64_t minutes_arg, int64_t seconds_arg, int64_t nanoseconds)
+{
+  reset();
+  _end= _begin;
+  _end.tv_sec+= (minutes_arg * 60) +seconds_arg;
+  _end.tv_nsec+= nanoseconds;
+}
+
+int64_t Timer::minutes()
+{
+  struct timespec result;
+  difference(result);
+  return int64_t(result.tv_sec / 60);
+}
+
+uint64_t Timer::elapsed_milliseconds() const
+{
+  struct timespec temp;
+  difference(temp);
+
+  return temp.tv_sec*1000 +temp.tv_nsec/1000000;
+}
+
+void Timer::difference(struct timespec& arg) const
+{
+  if ((_end.tv_nsec -_begin.tv_nsec) < 0)
+  {
+    arg.tv_sec= _end.tv_sec -_begin.tv_sec -1;
+    arg.tv_nsec= 1000000000 +_end.tv_nsec -_begin.tv_nsec;
+
+  }
+  else
+  {
+    arg.tv_sec= _end.tv_sec -_begin.tv_sec;
+    arg.tv_nsec= _end.tv_nsec -_begin.tv_nsec;
+  }
+}
+
+void Timer::_time(struct timespec& ts)
+{
+#ifdef __MACH__ // OSX lacks clock_gettime()
+  clock_serv_t _clock_serv;
+  mach_timespec_t _mach_timespec;
+  host_get_clock_service(mach_host_self(), CALENDAR_CLOCK, &_clock_serv);
+  clock_get_time(_clock_serv, &_mach_timespec);
+  mach_port_deallocate(mach_task_self(), _clock_serv);
+  ts.tv_sec= _mach_timespec.tv_sec;
+  ts.tv_nsec= _mach_timespec.tv_nsec;
+#elif defined(_WIN32)
+  ts.tv_sec= time(NULL);
+  ts.tv_nsec= 0;
+#else
+  clock_gettime(CLOCK_REALTIME, &ts);
+#endif
+}
+
 std::ostream& operator<<(std::ostream& output, const libtest::Timer& arg)
 {
   struct timespec temp;
   arg.difference(temp);
 
-  output << temp.tv_sec;
+  if (temp.tv_sec > 60)
+  {
+    output << temp.tv_sec / 60;
+    output << "." << temp.tv_sec % 60;
+  }
+  else
+  {
+    output << temp.tv_sec;
+  }
+
   output << ":";
-  output << temp.tv_nsec;
+  output << std::setfill('0') << std::setw(9) << temp.tv_nsec;
 
   return output;
 }