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