MIRPrintingPass.cpp revision 360784 
1//===- MIRPrintingPass.cpp - Pass that prints out using the MIR format ----===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This file implements a pass that prints out the LLVM module using the MIR
10// serialization format.
11//
12//===----------------------------------------------------------------------===//
13
14#include "llvm/CodeGen/MIRPrinter.h"
15#include "llvm/CodeGen/MachineFunctionPass.h"
16#include "llvm/CodeGen/Passes.h"
17#include "llvm/InitializePasses.h"
18#include "llvm/Support/Debug.h"
19#include "llvm/Support/raw_ostream.h"
20
21using namespace llvm;
22
23namespace {
24
25/// This pass prints out the LLVM IR to an output stream using the MIR
26/// serialization format.
27struct MIRPrintingPass : public MachineFunctionPass {
28  static char ID;
29  raw_ostream &OS;
30  std::string MachineFunctions;
31
32  MIRPrintingPass() : MachineFunctionPass(ID), OS(dbgs()) {}
33  MIRPrintingPass(raw_ostream &OS) : MachineFunctionPass(ID), OS(OS) {}
34
35  StringRef getPassName() const override { return "MIR Printing Pass"; }
36
37  void getAnalysisUsage(AnalysisUsage &AU) const override {
38    AU.setPreservesAll();
39    MachineFunctionPass::getAnalysisUsage(AU);
40  }
41
42  bool runOnMachineFunction(MachineFunction &MF) override {
43    std::string Str;
44    raw_string_ostream StrOS(Str);
45    printMIR(StrOS, MF);
46    MachineFunctions.append(StrOS.str());
47    return false;
48  }
49
50  bool doFinalization(Module &M) override {
51    printMIR(OS, M);
52    OS << MachineFunctions;
53    return false;
54  }
55};
56
57char MIRPrintingPass::ID = 0;
58
59} // end anonymous namespace
60
61char &llvm::MIRPrintingPassID = MIRPrintingPass::ID;
62INITIALIZE_PASS(MIRPrintingPass, "mir-printer", "MIR Printer", false, false)
63
64namespace llvm {
65
66MachineFunctionPass *createPrintMIRPass(raw_ostream &OS) {
67  return new MIRPrintingPass(OS);
68}
69
70} // end namespace llvm
71