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 
21 using namespace llvm;
22 
23 namespace {
24 
25 /// This pass prints out the LLVM IR to an output stream using the MIR
26 /// serialization format.
27 struct 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 
57 char MIRPrintingPass::ID = 0;
58 
59 } // end anonymous namespace
60 
61 char &llvm::MIRPrintingPassID = MIRPrintingPass::ID;
62 INITIALIZE_PASS(MIRPrintingPass, "mir-printer", "MIR Printer", false, false)
63 
64 namespace llvm {
65 
66 MachineFunctionPass *createPrintMIRPass(raw_ostream &OS) {
67   return new MIRPrintingPass(OS);
68 }
69 
70 } // end namespace llvm
71