contrib/llvm/lib/CodeGen/MIRPrintingPass.cpp

   1 //===- MIRPrintingPass.cpp - Pass that prints out using the MIR format ----===//
   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 // This file implements a pass that prints out the LLVM module using the MIR
  11 // serialization format.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "llvm/CodeGen/MIRPrinter.h"
  16
  17 #include "llvm/CodeGen/MIRYamlMapping.h"
  18 #include "llvm/CodeGen/MachineFunctionPass.h"
  19 #include "llvm/CodeGen/Passes.h"
  20 #include "llvm/Support/Debug.h"
  21 #include "llvm/Support/raw_ostream.h"
  22
  23 using namespace llvm;
  24
  25 namespace {
  26
  27 /// This pass prints out the LLVM IR to an output stream using the MIR
  28 /// serialization format.
  29 struct MIRPrintingPass : public MachineFunctionPass {
  30   static char ID;
  31   raw_ostream &OS;
  32   std::string MachineFunctions;
  33
  34   MIRPrintingPass() : MachineFunctionPass(ID), OS(dbgs()) {}
  35   MIRPrintingPass(raw_ostream &OS) : MachineFunctionPass(ID), OS(OS) {}
  36
  37   StringRef getPassName() const override { return "MIR Printing Pass"; }
  38
  39   void getAnalysisUsage(AnalysisUsage &AU) const override {
  40     AU.setPreservesAll();
  41     MachineFunctionPass::getAnalysisUsage(AU);
  42   }
  43
  44   bool runOnMachineFunction(MachineFunction &MF) override {
  45     std::string Str;
  46     raw_string_ostream StrOS(Str);
  47     printMIR(StrOS, MF);
  48     MachineFunctions.append(StrOS.str());
  49     return false;
  50   }
  51
  52   bool doFinalization(Module &M) override {
  53     printMIR(OS, M);
  54     OS << MachineFunctions;
  55     return false;
  56   }
  57 };
  58
  59 char MIRPrintingPass::ID = 0;
  60
  61 } // end anonymous namespace
  62
  63 char &llvm::MIRPrintingPassID = MIRPrintingPass::ID;
  64 INITIALIZE_PASS(MIRPrintingPass, "mir-printer", "MIR Printer", false, false)
  65
  66 namespace llvm {
  67
  68 MachineFunctionPass *createPrintMIRPass(raw_ostream &OS) {
  69   return new MIRPrintingPass(OS);
  70 }
  71
  72 } // end namespace llvm