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