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/IRPrintingPasses.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 (!llvm::isFunctionInPrintList(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