1 | //===-- MachineFunctionPrinterPass.cpp ------------------------------------===// |
2 | // |
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | // See https://llvm.org/LICENSE.txt for license information. |
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | // |
7 | //===----------------------------------------------------------------------===// |
8 | // |
9 | // MachineFunctionPrinterPass implementation. |
10 | // |
11 | //===----------------------------------------------------------------------===// |
12 | |
13 | #include "llvm/CodeGen/MachineFunction.h" |
14 | #include "llvm/CodeGen/MachineFunctionPass.h" |
15 | #include "llvm/CodeGen/Passes.h" |
16 | #include "llvm/CodeGen/SlotIndexes.h" |
17 | #include "llvm/IR/PrintPasses.h" |
18 | #include "llvm/InitializePasses.h" |
19 | #include "llvm/Support/Debug.h" |
20 | #include "llvm/Support/raw_ostream.h" |
21 | |
22 | using namespace llvm; |
23 | |
24 | namespace { |
25 | /// MachineFunctionPrinterPass - This is a pass to dump the IR of a |
26 | /// MachineFunction. |
27 | /// |
28 | struct MachineFunctionPrinterPass : public MachineFunctionPass { |
29 | static char ID; |
30 | |
31 | raw_ostream &OS; |
32 | const std::string Banner; |
33 | |
34 | MachineFunctionPrinterPass() : MachineFunctionPass(ID), OS(dbgs()) { } |
35 | MachineFunctionPrinterPass(raw_ostream &os, const std::string &banner) |
36 | : MachineFunctionPass(ID), OS(os), Banner(banner) {} |
37 | |
38 | StringRef getPassName() const override { return "MachineFunction Printer" ; } |
39 | |
40 | void getAnalysisUsage(AnalysisUsage &AU) const override { |
41 | AU.setPreservesAll(); |
42 | AU.addUsedIfAvailable<SlotIndexes>(); |
43 | MachineFunctionPass::getAnalysisUsage(AU); |
44 | } |
45 | |
46 | bool runOnMachineFunction(MachineFunction &MF) override { |
47 | if (!isFunctionInPrintList(FunctionName: MF.getName())) |
48 | return false; |
49 | OS << "# " << Banner << ":\n" ; |
50 | MF.print(OS, getAnalysisIfAvailable<SlotIndexes>()); |
51 | return false; |
52 | } |
53 | }; |
54 | |
55 | char MachineFunctionPrinterPass::ID = 0; |
56 | } |
57 | |
58 | char &llvm::MachineFunctionPrinterPassID = MachineFunctionPrinterPass::ID; |
59 | INITIALIZE_PASS(MachineFunctionPrinterPass, "machineinstr-printer" , |
60 | "Machine Function Printer" , false, false) |
61 | |
62 | namespace llvm { |
63 | /// Returns a newly-created MachineFunction Printer pass. The |
64 | /// default banner is empty. |
65 | /// |
66 | MachineFunctionPass *createMachineFunctionPrinterPass(raw_ostream &OS, |
67 | const std::string &Banner){ |
68 | return new MachineFunctionPrinterPass(OS, Banner); |
69 | } |
70 | |
71 | } |
72 | |