lib/CodeGen/MachineFunctionPrinterPass.cpp

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