1//===-- MachineFunctionPrinterPass.cpp ------------------------------------===//
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// MachineFunctionPrinterPass implementation.
11//
12//===----------------------------------------------------------------------===//
13
14#include "llvm/CodeGen/Passes.h"
15#include "llvm/CodeGen/MachineFunction.h"
16#include "llvm/CodeGen/MachineFunctionPass.h"
17#include "llvm/CodeGen/SlotIndexes.h"
18#include "llvm/Support/Debug.h"
19#include "llvm/Support/raw_ostream.h"
20
21using namespace llvm;
22
23namespace {
24/// MachineFunctionPrinterPass - This is a pass to dump the IR of a
25/// MachineFunction.
26///
27struct MachineFunctionPrinterPass : public MachineFunctionPass {
28  static char ID;
29
30  raw_ostream &OS;
31  const std::string Banner;
32
33  MachineFunctionPrinterPass() : MachineFunctionPass(ID), OS(dbgs()) { }
34  MachineFunctionPrinterPass(raw_ostream &os, const std::string &banner)
35      : MachineFunctionPass(ID), OS(os), Banner(banner) {}
36
37  const char *getPassName() const override { return "MachineFunction Printer"; }
38
39  void getAnalysisUsage(AnalysisUsage &AU) const override {
40    AU.setPreservesAll();
41    MachineFunctionPass::getAnalysisUsage(AU);
42  }
43
44  bool runOnMachineFunction(MachineFunction &MF) override {
45    if (!llvm::isFunctionInPrintList(MF.getName()))
46      return false;
47    OS << "# " << Banner << ":\n";
48    MF.print(OS, getAnalysisIfAvailable<SlotIndexes>());
49    return false;
50  }
51};
52
53char MachineFunctionPrinterPass::ID = 0;
54}
55
56char &llvm::MachineFunctionPrinterPassID = MachineFunctionPrinterPass::ID;
57INITIALIZE_PASS(MachineFunctionPrinterPass, "machineinstr-printer",
58                "Machine Function Printer", false, false)
59
60namespace llvm {
61/// Returns a newly-created MachineFunction Printer pass. The
62/// default banner is empty.
63///
64MachineFunctionPass *createMachineFunctionPrinterPass(raw_ostream &OS,
65                                                      const std::string &Banner){
66  return new MachineFunctionPrinterPass(OS, Banner);
67}
68
69}
70