//===--- 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.
  /// void example() {
  ///  RecursiveSharedTimer::Guard guard; // MUST BE AT TOP SCOPE of function to
  ///  work right! if (auto s = getASTContext().Stats) {
  ///    guard =
  ///    ctx.Stats->getFrontendRecursiveSharedTimers().NominalTypeDecl__lookupDirect.getGuard();
  //  }
  ///   ...
  /// }

  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 *recursiveTimerOrNull;

      Guard(RecursiveSharedTimer *rst) : recursiveTimerOrNull(rst) {
        if (recursiveTimerOrNull)
          recursiveTimerOrNull->enterRecursiveFunction();
      }
      ~Guard() {
        if (recursiveTimerOrNull)
          recursiveTimerOrNull->exitRecursiveFunction();
      }

      // All this stuff is to do an RAII object that be moved.
      Guard() : recursiveTimerOrNull(nullptr) {}
      Guard(Guard &&other) {
        recursiveTimerOrNull = other.recursiveTimerOrNull;
        other.recursiveTimerOrNull = nullptr;
      }
      Guard &operator=(Guard &&other) {
        recursiveTimerOrNull = other.recursiveTimerOrNull;
        other.recursiveTimerOrNull = nullptr;
        return *this;
      }
      Guard(const Guard &) = delete;
      Guard &operator=(const Guard &) = delete;
    };

    Guard getGuard() { return Guard(this); }
  };
} // end namespace swift

#endif // SWIFT_BASIC_TIMER_H