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/MachineFunctionPass.h" 16#include "llvm/CodeGen/MachineFunction.h" 17#include "llvm/CodeGen/SlotIndexes.h" 18#include "llvm/Support/raw_ostream.h" 19#include "llvm/Support/Debug.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 { return "MachineFunction Printer"; } 38 39 virtual void getAnalysisUsage(AnalysisUsage &AU) const { 40 AU.setPreservesAll(); 41 MachineFunctionPass::getAnalysisUsage(AU); 42 } 43 44 bool runOnMachineFunction(MachineFunction &MF) { 45 OS << "# " << Banner << ":\n"; 46 MF.print(OS, getAnalysisIfAvailable<SlotIndexes>()); 47 return false; 48 } 49}; 50 51char MachineFunctionPrinterPass::ID = 0; 52} 53 54char &MachineFunctionPrinterPassID = MachineFunctionPrinterPass::ID; 55INITIALIZE_PASS(MachineFunctionPrinterPass, "print-machineinstrs", 56 "Machine Function Printer", false, false) 57 58namespace llvm { 59/// Returns a newly-created MachineFunction Printer pass. The 60/// default banner is empty. 61/// 62MachineFunctionPass *createMachineFunctionPrinterPass(raw_ostream &OS, 63 const std::string &Banner){ 64 return new MachineFunctionPrinterPass(OS, Banner); 65} 66 67} 68