llvm.org GIT mirror llvm / 34239a9 include / llvm / Passes / StandardInstrumentations.h

Tree @34239a9 (Download .tar.gz)

StandardInstrumentations.h @34239a9raw · history · blame

//===- StandardInstrumentations.h ------------------------------*- C++ -*--===//
//                     The LLVM Compiler Infrastructure
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
/// \file
/// This header defines a class that provides bookkeeping for all standard
/// (i.e in-tree) pass instrumentations.


#include "llvm/ADT/SmallVector.h"
#include "llvm/IR/PassInstrumentation.h"
#include "llvm/IR/PassTimingInfo.h"

#include <string>
#include <utility>

namespace llvm {

class Module;

/// Instrumentation to print IR before/after passes.
/// Needs state to be able to print module after pass that invalidates IR unit
/// (typically Loop or SCC).
class PrintIRInstrumentation {
  PrintIRInstrumentation() = default;

  void registerCallbacks(PassInstrumentationCallbacks &PIC);

  bool printBeforePass(StringRef PassID, Any IR);
  void printAfterPass(StringRef PassID, Any IR);
  void printAfterPassInvalidated(StringRef PassID);

  using PrintModuleDesc = std::tuple<const Module *, std::string, StringRef>;

  void pushModuleDesc(StringRef PassID, Any IR);
  PrintModuleDesc popModuleDesc(StringRef PassID);

  /// Stack of Module description, enough to print the module after a given
  /// pass.
  SmallVector<PrintModuleDesc, 2> ModuleDescStack;
  bool StoreModuleDesc = false;

/// This class provides an interface to register all the standard pass
/// instrumentations and manages their state (if any).
class StandardInstrumentations {
  PrintIRInstrumentation PrintIR;
  TimePassesHandler TimePasses;

  StandardInstrumentations() = default;

  void registerCallbacks(PassInstrumentationCallbacks &PIC);
} // namespace llvm