lib/VMCore/PrintModulePass.cpp

   1 //===--- VMCore/PrintModulePass.cpp - Module/Function Printer -------------===//
   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 // PrintModulePass and PrintFunctionPass implementations.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "llvm/Assembly/PrintModulePass.h"
  15
  16 #include "llvm/Function.h"
  17 #include "llvm/Module.h"
  18 #include "llvm/Pass.h"
  19 #include "llvm/Support/Compiler.h"
  20 #include "llvm/Support/raw_ostream.h"
  21 using namespace llvm;
  22
  23 namespace {
  24
  25   class VISIBILITY_HIDDEN PrintModulePass : public ModulePass {
  26     raw_ostream *Out;       // raw_ostream to print on
  27     bool DeleteStream;      // Delete the ostream in our dtor?
  28   public:
  29     static char ID;
  30     PrintModulePass() : ModulePass(&ID), Out(&errs()),
  31       DeleteStream(false) {}
  32     PrintModulePass(raw_ostream *o, bool DS)
  33       : ModulePass(&ID), Out(o), DeleteStream(DS) {}
  34
  35     ~PrintModulePass() {
  36       if (DeleteStream) delete Out;
  37     }
  38
  39     bool runOnModule(Module &M) {
  40       (*Out) << M;
  41       return false;
  42     }
  43
  44     virtual void getAnalysisUsage(AnalysisUsage &AU) const {
  45       AU.setPreservesAll();
  46     }
  47   };
  48
  49   class PrintFunctionPass : public FunctionPass {
  50     std::string Banner;     // String to print before each function
  51     raw_ostream *Out;       // raw_ostream to print on
  52     bool DeleteStream;      // Delete the ostream in our dtor?
  53   public:
  54     static char ID;
  55     PrintFunctionPass() : FunctionPass(&ID), Banner(""), Out(&errs()),
  56                           DeleteStream(false) {}
  57     PrintFunctionPass(const std::string &B, raw_ostream *o, bool DS)
  58       : FunctionPass(&ID), Banner(B), Out(o), DeleteStream(DS) {}
  59
  60     inline ~PrintFunctionPass() {
  61       if (DeleteStream) delete Out;
  62     }
  63
  64     // runOnFunction - This pass just prints a banner followed by the
  65     // function as it's processed.
  66     //
  67     bool runOnFunction(Function &F) {
  68       (*Out) << Banner << static_cast<Value&>(F);
  69       return false;
  70     }
  71
  72     virtual void getAnalysisUsage(AnalysisUsage &AU) const {
  73       AU.setPreservesAll();
  74     }
  75   };
  76 }
  77
  78 char PrintModulePass::ID = 0;
  79 static RegisterPass<PrintModulePass>
  80 X("print-module", "Print module to stderr");
  81 char PrintFunctionPass::ID = 0;
  82 static RegisterPass<PrintFunctionPass>
  83 Y("print-function","Print function to stderr");
  84
  85 /// createPrintModulePass - Create and return a pass that writes the
  86 /// module to the specified raw_ostream.
  87 ModulePass *llvm::createPrintModulePass(llvm::raw_ostream *OS,
  88                                         bool DeleteStream) {
  89   return new PrintModulePass(OS, DeleteStream);
  90 }
  91
  92 /// createPrintFunctionPass - Create and return a pass that prints
  93 /// functions to the specified raw_ostream as they are processed.
  94 FunctionPass *llvm::createPrintFunctionPass(const std::string &Banner,
  95                                             llvm::raw_ostream *OS,
  96                                             bool DeleteStream) {
  97   return new PrintFunctionPass(Banner, OS, DeleteStream);
  98 }
  99