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 
21 using namespace llvm;
22 
23 namespace {
24 /// MachineFunctionPrinterPass - This is a pass to dump the IR of a
25 /// MachineFunction.
26 ///
27 struct MachineFunctionPrinterPass : public MachineFunctionPass {
28   static char ID;
29 
30   raw_ostream &OS;
31   const std::string Banner;
32 
MachineFunctionPrinterPass__anonc04613ec0111::MachineFunctionPrinterPass33   MachineFunctionPrinterPass() : MachineFunctionPass(ID), OS(dbgs()) { }
MachineFunctionPrinterPass__anonc04613ec0111::MachineFunctionPrinterPass34   MachineFunctionPrinterPass(raw_ostream &os, const std::string &banner)
35       : MachineFunctionPass(ID), OS(os), Banner(banner) {}
36 
getPassName__anonc04613ec0111::MachineFunctionPrinterPass37   const char *getPassName() const override { return "MachineFunction Printer"; }
38 
getAnalysisUsage__anonc04613ec0111::MachineFunctionPrinterPass39   void getAnalysisUsage(AnalysisUsage &AU) const override {
40     AU.setPreservesAll();
41     MachineFunctionPass::getAnalysisUsage(AU);
42   }
43 
runOnMachineFunction__anonc04613ec0111::MachineFunctionPrinterPass44   bool runOnMachineFunction(MachineFunction &MF) override {
45     OS << "# " << Banner << ":\n";
46     MF.print(OS, getAnalysisIfAvailable<SlotIndexes>());
47     return false;
48   }
49 };
50 
51 char MachineFunctionPrinterPass::ID = 0;
52 }
53 
54 char &llvm::MachineFunctionPrinterPassID = MachineFunctionPrinterPass::ID;
55 INITIALIZE_PASS(MachineFunctionPrinterPass, "machineinstr-printer",
56                 "Machine Function Printer", false, false)
57 
58 namespace llvm {
59 /// Returns a newly-created MachineFunction Printer pass. The
60 /// default banner is empty.
61 ///
createMachineFunctionPrinterPass(raw_ostream & OS,const std::string & Banner)62 MachineFunctionPass *createMachineFunctionPrinterPass(raw_ostream &OS,
63                                                       const std::string &Banner){
64   return new MachineFunctionPrinterPass(OS, Banner);
65 }
66 
67 }
68