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