[NewPM][CodeGen] Port MIRPrinter to NewPM

[aeubanks]

Should we get the llc testing infra working with the NPM before porting passes?

llvm/include/llvm/CodeGen/MIRPrinter.h

38

I thought this was only for analysis passes

[ychen]

In D83607#2438923, @aeubanks wrote:

Should we get the llc testing infra working with the NPM before porting passes?

Yes, we could. I thought it was a straightforward porting and D67687 depends on it. Somehow I could make D67687 not depend on this. Yeah, let's do this later.

llvm/include/llvm/CodeGen/MIRPrinter.h

 //===- MIRPrinter.h - MIR serialization format printer ----------*- C++ -*-===//
 //
 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 // See https://llvm.org/LICENSE.txt for license information.
 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 //
 //===----------------------------------------------------------------------===//
 //
 // This file declares the functions that print out the LLVM IR and the machine
 // functions using the MIR serialization format.
 //
 //===----------------------------------------------------------------------===//
 
 #ifndef LLVM_LIB_CODEGEN_MIRPRINTER_H
 #define LLVM_LIB_CODEGEN_MIRPRINTER_H
 
+#include "llvm/CodeGen/MachinePassManager.h"
+
 namespace llvm {
 
 class MachineBasicBlock;
 class MachineFunction;
 class Module;
 class raw_ostream;
 template <typename T> class SmallVectorImpl;
 
+class PrintMIRPass : public PassInfoMixin<PrintMIRPass> {
+  raw_ostream &OS;
+  std::string MachineFunctions;
+
+public:
+  PrintMIRPass() : OS(dbgs()) {}
+  PrintMIRPass(raw_ostream &OS) : OS(OS) {}
+
+  PreservedAnalyses run(MachineFunction &MF, MachineFunctionAnalysisManager &);
+  Error doFinalization(Module &M, MachineFunctionAnalysisManager &);
+
+  static AnalysisKey Key;
+};
+
 /// Print LLVM IR using the MIR serialization format to the given output stream.
 void printMIR(raw_ostream &OS, const Module &M);
 
 /// Print a machine function using the MIR serialization format to the given
 /// output stream.
 void printMIR(raw_ostream &OS, const MachineFunction &MF);
 
 /// Determine a possible list of successors of a basic block based on the

llvm/lib/CodeGen/MIRPrintingPass.cpp

 //
 //===----------------------------------------------------------------------===//
 
 #include "llvm/CodeGen/MIRPrinter.h"
 #include "llvm/CodeGen/MachineFunctionPass.h"
 #include "llvm/CodeGen/Passes.h"
 #include "llvm/InitializePasses.h"
 #include "llvm/Support/Debug.h"
+#include "llvm/Support/Error.h"
 #include "llvm/Support/raw_ostream.h"
 
 using namespace llvm;
 
+PreservedAnalyses PrintMIRPass::run(MachineFunction &MF,
+                                    MachineFunctionAnalysisManager &) {
+  std::string Str;
+  raw_string_ostream StrOS(Str);
+  printMIR(StrOS, MF);
+  MachineFunctions.append(StrOS.str());
+  return PreservedAnalyses::all();
+}
+
+Error PrintMIRPass::doFinalization(Module &M,
+                                   MachineFunctionAnalysisManager &) {
+  printMIR(OS, M);
+  OS << MachineFunctions;
+  return Error::success();
+}
+
+AnalysisKey PrintMIRPass::Key;
+
 namespace {
 
 /// This pass prints out the LLVM IR to an output stream using the MIR
 /// serialization format.
 struct MIRPrintingPass : public MachineFunctionPass {
   static char ID;
   raw_ostream &OS;
   std::string MachineFunctions;