diff options
Diffstat (limited to 'lib/CodeGen/MachineFunctionPrinterPass.cpp')
| -rw-r--r-- | lib/CodeGen/MachineFunctionPrinterPass.cpp | 60 | 
1 files changed, 60 insertions, 0 deletions
diff --git a/lib/CodeGen/MachineFunctionPrinterPass.cpp b/lib/CodeGen/MachineFunctionPrinterPass.cpp new file mode 100644 index 000000000000..547c4febc8da --- /dev/null +++ b/lib/CodeGen/MachineFunctionPrinterPass.cpp @@ -0,0 +1,60 @@ +//===-- MachineFunctionPrinterPass.cpp ------------------------------------===// +// +//                     The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// +// +// MachineFunctionPrinterPass implementation. +// +//===----------------------------------------------------------------------===// + +#include "llvm/CodeGen/Passes.h" +#include "llvm/CodeGen/MachineFunctionPass.h" +#include "llvm/CodeGen/MachineFunction.h" +#include "llvm/Support/raw_ostream.h" + +using namespace llvm; + +namespace { +/// MachineFunctionPrinterPass - This is a pass to dump the IR of a +/// MachineFunction. +/// +struct MachineFunctionPrinterPass : public MachineFunctionPass { +  static char ID; + +  raw_ostream &OS; +  const std::string Banner; + +  MachineFunctionPrinterPass(raw_ostream &os, const std::string &banner)  +      : MachineFunctionPass(&ID), OS(os), Banner(banner) {} + +  const char *getPassName() const { return "MachineFunction Printer"; } + +  virtual void getAnalysisUsage(AnalysisUsage &AU) const { +    AU.setPreservesAll(); +    MachineFunctionPass::getAnalysisUsage(AU); +  } + +  bool runOnMachineFunction(MachineFunction &MF) { +    OS << "# " << Banner << ":\n"; +    MF.print(OS); +    return false; +  } +}; + +char MachineFunctionPrinterPass::ID = 0; +} + +namespace llvm { +/// Returns a newly-created MachineFunction Printer pass. The +/// default banner is empty. +/// +MachineFunctionPass *createMachineFunctionPrinterPass(raw_ostream &OS, +                                                      const std::string &Banner){ +  return new MachineFunctionPrinterPass(OS, Banner); +} + +}  | 
