include/swift/Basic/Timer.h

//===--- Timer.h - Shared timers for compilation phases ---------*- C++ -*-===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

#ifndef SWIFT_BASIC_TIMER_H
#define SWIFT_BASIC_TIMER_H

#include "swift/Basic/LLVM.h"
#include "llvm/ADT/Optional.h"
#include "llvm/Support/Timer.h"

namespace swift {
  /// A convenience class for declaring a timer that's part of the Swift
  /// compilation timers group.
  class SharedTimer {
    enum class State {
      Initial,
      Skipped,
      Enabled
    };
    static State CompilationTimersEnabled;

    Optional<llvm::NamedRegionTimer> Timer;

  public:
    explicit SharedTimer(StringRef name) {
      if (CompilationTimersEnabled == State::Enabled)
        Timer.emplace(name, name, "swift", "Swift compilation");
      else
        CompilationTimersEnabled = State::Skipped;
    }

    /// Must be called before any SharedTimers have been created.
    static void enableCompilationTimers() {
      assert(CompilationTimersEnabled != State::Skipped &&
             "a timer has already been created");
      CompilationTimersEnabled = State::Enabled;
    }
  };

  /// A SharedTimer for recursive routines.
  /// Declare as a static, and use as below:
  /// void example() {
  ///     static RecursiveSharedTimer timer("lookupDirect");
  ///     auto guard = RecursiveSharedTimer::Guard(timer);
  ///     (void)guard;
  ///   ...
  /// }

  class RecursiveSharedTimer {
  private:
    int recursionCount = 0;
    const StringRef name;
    llvm::Optional<SharedTimer> timer;

    void enterRecursiveFunction() {
      assert(recursionCount >= 0  &&  "too many exits");
      if (recursionCount++ == 0)
        timer.emplace(name);
    }
    void exitRecursiveFunction() {
      assert(recursionCount > 0  &&  "too many exits");
      if (--recursionCount == 0)
        timer.reset();
    }

  public:
    RecursiveSharedTimer(StringRef name) : name(name) {}

    struct Guard {
      RecursiveSharedTimer &recursiveTimer;
      Guard &operator=(Guard &) = delete;
      Guard(RecursiveSharedTimer &rst) : recursiveTimer(rst) {
        recursiveTimer.enterRecursiveFunction();
      }
      ~Guard() { recursiveTimer.exitRecursiveFunction(); }
    };
  };
} // end namespace swift

#endif // SWIFT_BASIC_TIMER_H