Checkin generic interval timer support

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@3992 91177308-0d34-0410-b5e6-96231b3b80d8

support/lib/Support/Timer.cpp

@@ -0,0 +1,203 @@
+//===-- Timer.cpp - Interval Timing Support -------------------------------===//
+//
+// Interval Timing implementation.
+//
+//===----------------------------------------------------------------------===//
+
+#include "Support/Timer.h"
+#include <sys/resource.h>
+#include <sys/time.h>
+#include <sys/unistd.h>
+#include <stdio.h>
+#include <iostream>
+#include <algorithm>
+
+static TimerGroup *DefaultTimerGroup = 0;
+static TimerGroup *getDefaultTimerGroup() {
+  if (DefaultTimerGroup) return DefaultTimerGroup;
+  return DefaultTimerGroup = new TimerGroup("Miscellaneous Ungrouped Timers");
+}
+
+Timer::Timer(const std::string &N)
+  : Elapsed(0), UserTime(0), SystemTime(0), MaxRSS(0), Name(N),
+    Started(false), TG(getDefaultTimerGroup()) {
+  TG->addTimer();
+}
+
+Timer::Timer(const std::string &N, TimerGroup &tg)
+  : Elapsed(0), UserTime(0), SystemTime(0), MaxRSS(0), Name(N),
+    Started(false), TG(&tg) {
+  TG->addTimer();
+}
+
+Timer::Timer(const Timer &T) {
+  TG = T.TG;
+  if (TG) TG->addTimer();
+  operator=(T);
+}
+
+
+// Copy ctor, initialize with no TG member.
+Timer::Timer(bool, const Timer &T) {
+  TG = T.TG;     // Avoid assertion in operator=
+  operator=(T);  // Copy contents
+  TG = 0;
+}
+
+
+Timer::~Timer() {
+  if (TG) {
+    if (Started) {
+      Started = false;
+      TG->addTimerToPrint(*this);
+    }
+    TG->removeTimer();
+  }
+}
+
+struct TimeRecord {
+  double Elapsed, UserTime, SystemTime;
+  unsigned long MaxRSS;
+};
+
+static TimeRecord getTimeRecord() {
+  static unsigned long PageSize = 0;
+
+  if (PageSize == 0) {
+#ifdef _SC_PAGE_SIZE
+    PageSize = sysconf(_SC_PAGE_SIZE);
+#else
+#ifdef _SC_PAGESIZE
+    PageSize = sysconf(_SC_PAGESIZE);
+#else
+    PageSize = getpagesize();
+#endif
+#endif
+  }
+
+  struct rusage RU;
+  struct timeval T;
+  gettimeofday(&T, 0);
+  if (getrusage(RUSAGE_SELF, &RU)) {
+    perror("getrusage call failed: -time-passes info incorrect!");
+  }
+
+  TimeRecord Result;
+  Result.Elapsed    =           T.tv_sec +           T.tv_usec/1000000.0;
+  Result.UserTime   = RU.ru_utime.tv_sec + RU.ru_utime.tv_usec/1000000.0;
+  Result.SystemTime = RU.ru_stime.tv_sec + RU.ru_stime.tv_usec/1000000.0;
+  Result.MaxRSS = RU.ru_maxrss*PageSize;
+  return Result;
+}
+
+
+void Timer::startTimer() {
+  Started = true;
+  TimeRecord TR = getTimeRecord();
+  Elapsed    -= TR.Elapsed;
+  UserTime   -= TR.UserTime;
+  SystemTime -= TR.SystemTime;
+  MaxRSS     -= TR.MaxRSS;
+}
+
+void Timer::stopTimer() {
+  TimeRecord TR = getTimeRecord();
+  Elapsed    += TR.Elapsed;
+  UserTime   += TR.UserTime;
+  SystemTime += TR.SystemTime;
+  MaxRSS     += TR.MaxRSS;
+}
+
+void Timer::sum(const Timer &T) {
+  Elapsed    += T.Elapsed;
+  UserTime   += T.UserTime;
+  SystemTime += T.SystemTime;
+  MaxRSS     += T.MaxRSS;
+}
+
+//===----------------------------------------------------------------------===//
+//   TimerGroup Implementation
+//===----------------------------------------------------------------------===//
+
+static void printVal(double Val, double Total) {
+  if (Total < 1e-7)   // Avoid dividing by zero...
+    fprintf(stderr, "        -----     ");
+  else
+    fprintf(stderr, "  %7.4f (%5.1f%%)", Val, Val*100/Total);
+}
+
+void Timer::print(const Timer &Total) {
+  if (Total.UserTime)
+    printVal(UserTime, Total.UserTime);
+  if (Total.SystemTime)
+    printVal(SystemTime, Total.SystemTime);
+  if (Total.getProcessTime())
+    printVal(getProcessTime(), Total.getProcessTime());
+  printVal(Elapsed, Total.Elapsed);
+  
+  fprintf(stderr, "  ");
+
+  if (Total.MaxRSS)
+    std::cerr << MaxRSS << "\t";
+  std::cerr << Name << "\n";
+
+  Started = false;  // Once printed, don't print again
+}
+
+
+void TimerGroup::removeTimer() {
+  if (--NumTimers == 0 && !TimersToPrint.empty()) { // Print timing report...
+    // Sort the timers in descending order by amount of time taken...
+    std::sort(TimersToPrint.begin(), TimersToPrint.end(),
+              std::greater<Timer>());
+
+    // Figure out how many spaces to indent TimerGroup name...
+    unsigned Padding = (80-Name.length())/2;
+    if (Padding > 80) Padding = 0;         // Don't allow "negative" numbers
+
+    ++NumTimers;
+    {  // Scope to contain Total timer... don't allow total timer to drop us to
+       // zero timers...
+      Timer Total("TOTAL");
+  
+      for (unsigned i = 0, e = TimersToPrint.size(); i != e; ++i)
+        Total.sum(TimersToPrint[i]);
+      
+      // Print out timing header...
+      std::cerr << "===" << std::string(73, '-') << "===\n"
+                << std::string(Padding, ' ') << Name << "\n"
+                << "===" << std::string(73, '-')
+                << "===\n  Total Execution Time: " << Total.getProcessTime()
+                << " seconds (" << Total.getWallTime()
+                << " wall clock)\n\n";
+
+      if (Total.UserTime)
+        std::cerr << "   ---User Time---";
+      if (Total.SystemTime)
+        std::cerr << "   --System Time--";
+      if (Total.getProcessTime())
+        std::cerr << "   --User+System--";
+      std::cerr << "   ---Wall Time---";
+      
+      if (Total.getMaxRSS())
+        std::cerr << " ---Mem---";
+      std::cerr << "  --- Name ---\n";
+      
+      // Loop through all of the timing data, printing it out...
+      for (unsigned i = 0, e = TimersToPrint.size(); i != e; ++i)
+        TimersToPrint[i].print(Total);
+    
+      Total.print(Total);
+      std::cerr << std::endl;  // Flush output
+    }
+    --NumTimers;
+
+    TimersToPrint.clear();
+  }
+
+  // Delete default timer group!
+  if (NumTimers == 0 && this == DefaultTimerGroup) {
+    delete DefaultTimerGroup;
+    DefaultTimerGroup = 0;
+  }
+}