Timer.cpp revision 86026cd24eb8bdffdd6cf94669f98baeb87d8ef3 
1//===-- Timer.cpp - Interval Timing Support -------------------------------===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// Interval Timing implementation.
11//
12//===----------------------------------------------------------------------===//
13
14#include "llvm/Support/Timer.h"
15#include "llvm/Support/CommandLine.h"
16#include "llvm/Support/Debug.h"
17#include "llvm/Support/ManagedStatic.h"
18#include "llvm/Support/raw_ostream.h"
19#include "llvm/Support/Format.h"
20#include "llvm/System/Mutex.h"
21#include "llvm/System/Process.h"
22#include "llvm/ADT/OwningPtr.h"
23#include "llvm/ADT/StringMap.h"
24using namespace llvm;
25
26// CreateInfoOutputFile - Return a file stream to print our output on.
27namespace llvm { extern raw_ostream *CreateInfoOutputFile(); }
28
29// getLibSupportInfoOutputFilename - This ugly hack is brought to you courtesy
30// of constructor/destructor ordering being unspecified by C++.  Basically the
31// problem is that a Statistic object gets destroyed, which ends up calling
32// 'GetLibSupportInfoOutputFile()' (below), which calls this function.
33// LibSupportInfoOutputFilename used to be a global variable, but sometimes it
34// would get destroyed before the Statistic, causing havoc to ensue.  We "fix"
35// this by creating the string the first time it is needed and never destroying
36// it.
37static ManagedStatic<std::string> LibSupportInfoOutputFilename;
38static std::string &getLibSupportInfoOutputFilename() {
39  return *LibSupportInfoOutputFilename;
40}
41
42static ManagedStatic<sys::SmartMutex<true> > TimerLock;
43
44namespace {
45  static cl::opt<bool>
46  TrackSpace("track-memory", cl::desc("Enable -time-passes memory "
47                                      "tracking (this may be slow)"),
48             cl::Hidden);
49
50  static cl::opt<std::string, true>
51  InfoOutputFilename("info-output-file", cl::value_desc("filename"),
52                     cl::desc("File to append -stats and -timer output to"),
53                   cl::Hidden, cl::location(getLibSupportInfoOutputFilename()));
54}
55
56// CreateInfoOutputFile - Return a file stream to print our output on.
57raw_ostream *llvm::CreateInfoOutputFile() {
58  const std::string &OutputFilename = getLibSupportInfoOutputFilename();
59  if (OutputFilename.empty())
60    return new raw_fd_ostream(2, false); // stderr.
61  if (OutputFilename == "-")
62    return new raw_fd_ostream(1, false); // stdout.
63
64  // Append mode is used because the info output file is opened and closed
65  // each time -stats or -time-passes wants to print output to it. To
66  // compensate for this, the test-suite Makefiles have code to delete the
67  // info output file before running commands which write to it.
68  std::string Error;
69  raw_ostream *Result = new raw_fd_ostream(OutputFilename.c_str(),
70                                           Error, raw_fd_ostream::F_Append);
71  if (Error.empty())
72    return Result;
73
74  errs() << "Error opening info-output-file '"
75    << OutputFilename << " for appending!\n";
76  delete Result;
77  return new raw_fd_ostream(2, false); // stderr.
78}
79
80
81static TimerGroup *DefaultTimerGroup = 0;
82static TimerGroup *getDefaultTimerGroup() {
83  TimerGroup *tmp = DefaultTimerGroup;
84  sys::MemoryFence();
85  if (tmp) return tmp;
86
87  llvm_acquire_global_lock();
88  tmp = DefaultTimerGroup;
89  if (!tmp) {
90    tmp = new TimerGroup("Miscellaneous Ungrouped Timers");
91    sys::MemoryFence();
92    DefaultTimerGroup = tmp;
93  }
94  llvm_release_global_lock();
95
96  return tmp;
97}
98
99//===----------------------------------------------------------------------===//
100// Timer Implementation
101//===----------------------------------------------------------------------===//
102
103void Timer::init(StringRef N) {
104  assert(TG == 0 && "Timer already initialized");
105  Name.assign(N.begin(), N.end());
106  Started = false;
107  TG = getDefaultTimerGroup();
108  TG->addTimer(*this);
109}
110
111void Timer::init(StringRef N, TimerGroup &tg) {
112  assert(TG == 0 && "Timer already initialized");
113  Name.assign(N.begin(), N.end());
114  Started = false;
115  TG = &tg;
116  TG->addTimer(*this);
117}
118
119Timer::~Timer() {
120  if (!TG) return;  // Never initialized, or already cleared.
121  TG->removeTimer(*this);
122}
123
124static inline size_t getMemUsage() {
125  if (!TrackSpace) return 0;
126  return sys::Process::GetMallocUsage();
127}
128
129TimeRecord TimeRecord::getCurrentTime(bool Start) {
130  TimeRecord Result;
131  sys::TimeValue now(0,0), user(0,0), sys(0,0);
132
133  if (Start) {
134    Result.MemUsed = getMemUsage();
135    sys::Process::GetTimeUsage(now, user, sys);
136  } else {
137    sys::Process::GetTimeUsage(now, user, sys);
138    Result.MemUsed = getMemUsage();
139  }
140
141  Result.WallTime   =  now.seconds() +  now.microseconds() / 1000000.0;
142  Result.UserTime   = user.seconds() + user.microseconds() / 1000000.0;
143  Result.SystemTime =  sys.seconds() +  sys.microseconds() / 1000000.0;
144  return Result;
145}
146
147static ManagedStatic<std::vector<Timer*> > ActiveTimers;
148
149void Timer::startTimer() {
150  Started = true;
151  ActiveTimers->push_back(this);
152  Time -= TimeRecord::getCurrentTime(true);
153}
154
155void Timer::stopTimer() {
156  Time += TimeRecord::getCurrentTime(false);
157
158  if (ActiveTimers->back() == this) {
159    ActiveTimers->pop_back();
160  } else {
161    std::vector<Timer*>::iterator I =
162      std::find(ActiveTimers->begin(), ActiveTimers->end(), this);
163    assert(I != ActiveTimers->end() && "stop but no startTimer?");
164    ActiveTimers->erase(I);
165  }
166}
167
168static void printVal(double Val, double Total, raw_ostream &OS) {
169  if (Total < 1e-7)   // Avoid dividing by zero.
170    OS << "        -----     ";
171  else {
172    OS << "  " << format("%7.4f", Val) << " (";
173    OS << format("%5.1f", Val*100/Total) << "%)";
174  }
175}
176
177void TimeRecord::print(const TimeRecord &Total, raw_ostream &OS) const {
178  if (Total.getUserTime())
179    printVal(getUserTime(), Total.getUserTime(), OS);
180  if (Total.getSystemTime())
181    printVal(getSystemTime(), Total.getSystemTime(), OS);
182  if (Total.getProcessTime())
183    printVal(getProcessTime(), Total.getProcessTime(), OS);
184  printVal(getWallTime(), Total.getWallTime(), OS);
185
186  OS << "  ";
187
188  if (Total.getMemUsed())
189    OS << format("%9lld", (long long)getMemUsed()) << "  ";
190}
191
192
193//===----------------------------------------------------------------------===//
194//   NamedRegionTimer Implementation
195//===----------------------------------------------------------------------===//
196
197namespace {
198
199typedef StringMap<Timer> Name2TimerMap;
200
201class Name2PairMap {
202  StringMap<std::pair<TimerGroup*, Name2TimerMap> > Map;
203public:
204  ~Name2PairMap() {
205    for (StringMap<std::pair<TimerGroup*, Name2TimerMap> >::iterator
206         I = Map.begin(), E = Map.end(); I != E; ++I)
207      delete I->second.first;
208  }
209
210  Timer &get(StringRef Name, StringRef GroupName) {
211    sys::SmartScopedLock<true> L(*TimerLock);
212
213    std::pair<TimerGroup*, Name2TimerMap> &GroupEntry = Map[GroupName];
214
215    if (!GroupEntry.first)
216      GroupEntry.first = new TimerGroup(GroupName);
217
218    Timer &T = GroupEntry.second[Name];
219    if (!T.isInitialized())
220      T.init(Name, *GroupEntry.first);
221    return T;
222  }
223};
224
225}
226
227static ManagedStatic<Name2TimerMap> NamedTimers;
228static ManagedStatic<Name2PairMap> NamedGroupedTimers;
229
230static Timer &getNamedRegionTimer(StringRef Name) {
231  sys::SmartScopedLock<true> L(*TimerLock);
232
233  Timer &T = (*NamedTimers)[Name];
234  if (!T.isInitialized())
235    T.init(Name);
236  return T;
237}
238
239NamedRegionTimer::NamedRegionTimer(StringRef Name)
240  : TimeRegion(getNamedRegionTimer(Name)) {}
241
242NamedRegionTimer::NamedRegionTimer(StringRef Name, StringRef GroupName)
243  : TimeRegion(NamedGroupedTimers->get(Name, GroupName)) {}
244
245//===----------------------------------------------------------------------===//
246//   TimerGroup Implementation
247//===----------------------------------------------------------------------===//
248
249/// TimerGroupList - This is the global list of TimerGroups, maintained by the
250/// TimerGroup ctor/dtor and is protected by the TimerLock lock.
251static TimerGroup *TimerGroupList = 0;
252
253TimerGroup::TimerGroup(StringRef name)
254  : Name(name.begin(), name.end()), FirstTimer(0) {
255
256  // Add the group to TimerGroupList.
257  sys::SmartScopedLock<true> L(*TimerLock);
258  if (TimerGroupList)
259    TimerGroupList->Prev = &Next;
260  Next = TimerGroupList;
261  Prev = &TimerGroupList;
262  TimerGroupList = this;
263}
264
265TimerGroup::~TimerGroup() {
266  // If the timer group is destroyed before the timers it owns, accumulate and
267  // print the timing data.
268  while (FirstTimer != 0)
269    removeTimer(*FirstTimer);
270
271  // Remove the group from the TimerGroupList.
272  sys::SmartScopedLock<true> L(*TimerLock);
273  *Prev = Next;
274  if (Next)
275    Next->Prev = Prev;
276}
277
278
279void TimerGroup::removeTimer(Timer &T) {
280  sys::SmartScopedLock<true> L(*TimerLock);
281
282  // If the timer was started, move its data to TimersToPrint.
283  if (T.Started)
284    TimersToPrint.push_back(std::make_pair(T.Time, T.Name));
285
286  T.TG = 0;
287
288  // Unlink the timer from our list.
289  *T.Prev = T.Next;
290  if (T.Next)
291    T.Next->Prev = T.Prev;
292
293  // Print the report when all timers in this group are destroyed if some of
294  // them were started.
295  if (FirstTimer != 0 || TimersToPrint.empty())
296    return;
297
298  raw_ostream *OutStream = CreateInfoOutputFile();
299  PrintQueuedTimers(*OutStream);
300  delete OutStream;   // Close the file.
301}
302
303void TimerGroup::addTimer(Timer &T) {
304  sys::SmartScopedLock<true> L(*TimerLock);
305
306  // Add the timer to our list.
307  if (FirstTimer)
308    FirstTimer->Prev = &T.Next;
309  T.Next = FirstTimer;
310  T.Prev = &FirstTimer;
311  FirstTimer = &T;
312}
313
314void TimerGroup::PrintQueuedTimers(raw_ostream &OS) {
315  // Sort the timers in descending order by amount of time taken.
316  std::sort(TimersToPrint.begin(), TimersToPrint.end());
317
318  TimeRecord Total;
319  for (unsigned i = 0, e = TimersToPrint.size(); i != e; ++i)
320    Total += TimersToPrint[i].first;
321
322  // Print out timing header.
323  OS << "===" << std::string(73, '-') << "===\n";
324  // Figure out how many spaces to indent TimerGroup name.
325  unsigned Padding = (80-Name.length())/2;
326  if (Padding > 80) Padding = 0;         // Don't allow "negative" numbers
327  OS.indent(Padding) << Name << '\n';
328  OS << "===" << std::string(73, '-') << "===\n";
329
330  // If this is not an collection of ungrouped times, print the total time.
331  // Ungrouped timers don't really make sense to add up.  We still print the
332  // TOTAL line to make the percentages make sense.
333  if (this != DefaultTimerGroup) {
334    OS << "  Total Execution Time: ";
335    OS << format("%5.4f", Total.getProcessTime()) << " seconds (";
336    OS << format("%5.4f", Total.getWallTime()) << " wall clock)\n";
337  }
338  OS << '\n';
339
340  if (Total.getUserTime())
341    OS << "   ---User Time---";
342  if (Total.getSystemTime())
343    OS << "   --System Time--";
344  if (Total.getProcessTime())
345    OS << "   --User+System--";
346  OS << "   ---Wall Time---";
347  if (Total.getMemUsed())
348    OS << "  ---Mem---";
349  OS << "  --- Name ---\n";
350
351  // Loop through all of the timing data, printing it out.
352  for (unsigned i = 0, e = TimersToPrint.size(); i != e; ++i) {
353    const std::pair<TimeRecord, std::string> &Entry = TimersToPrint[e-i-1];
354    Entry.first.print(Total, OS);
355    OS << Entry.second << '\n';
356  }
357
358  Total.print(Total, OS);
359  OS << "Total\n\n";
360  OS.flush();
361
362  TimersToPrint.clear();
363}
364
365/// print - Print any started timers in this group and zero them.
366void TimerGroup::print(raw_ostream &OS) {
367  sys::SmartScopedLock<true> L(*TimerLock);
368
369  // See if any of our timers were started, if so add them to TimersToPrint and
370  // reset them.
371  for (Timer *T = FirstTimer; T; T = T->Next) {
372    if (!T->Started) continue;
373    TimersToPrint.push_back(std::make_pair(T->Time, T->Name));
374
375    // Clear out the time.
376    T->Started = 0;
377    T->Time = TimeRecord();
378  }
379
380  // If any timers were started, print the group.
381  if (!TimersToPrint.empty())
382    PrintQueuedTimers(OS);
383}
384
385/// printAll - This static method prints all timers and clears them all out.
386void TimerGroup::printAll(raw_ostream &OS) {
387  sys::SmartScopedLock<true> L(*TimerLock);
388
389  for (TimerGroup *TG = TimerGroupList; TG; TG = TG->Next)
390    TG->print(OS);
391}
392