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 22using namespace llvm; 23 24namespace { 25/// MachineFunctionPrinterPass - This is a pass to dump the IR of a 26/// MachineFunction. 27/// 28struct 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 55char MachineFunctionPrinterPass::ID = 0; 56} 57 58char &llvm::MachineFunctionPrinterPassID = MachineFunctionPrinterPass::ID; 59INITIALIZE_PASS(MachineFunctionPrinterPass, "machineinstr-printer", 60 "Machine Function Printer", false, false) 61 62namespace llvm { 63/// Returns a newly-created MachineFunction Printer pass. The 64/// default banner is empty. 65/// 66MachineFunctionPass *createMachineFunctionPrinterPass(raw_ostream &OS, 67 const std::string &Banner){ 68 return new MachineFunctionPrinterPass(OS, Banner); 69} 70 71} 72