Re-apply LiveInterval index dumping patch, with fixes suggested by Bill

and others.


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@78003 91177308-0d34-0410-b5e6-96231b3b80d8

4 files changed, 111 insertions, 6 deletions

diff --git a/include/llvm/CodeGen/Dump.h b/include/llvm/CodeGen/Dump.h
new file mode 100644
index 0000000000..439141cb44
--- /dev/null
+++ b/include/llvm/CodeGen/Dump.h
@@ -0,0 +1,60 @@
+//===- llvm/Support/Dump.h - Easy way to tailor dump output -----*- C++ -*-===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+// This file provides the PrefixPrinter interface to pass to MachineFunction
+// and MachineBasicBlock print methods to output additional information before
+// blocks and instructions are printed.
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef LLVM_CODEGEN_DUMP_H
+#define LLVM_CODEGEN_DUMP_H
+
+#include <iosfwd>
+
+namespace llvm {
+
+class MachineBasicBlock;
+class MachineInstr;
+class raw_ostream;
+
+/// PrefixPrinter - Print some additional information before printing
+/// basic blocks and instructions.
+class PrefixPrinter {
+public:
+  virtual ~PrefixPrinter();
+
+  /// operator() - Print a prefix before each MachineBasicBlock
+  virtual raw_ostream &operator()(raw_ostream &out,
+                                  const MachineBasicBlock &) const {
+    return out; 
+  }
+
+  /// operator() - Print a prefix before each MachineInstr
+  virtual raw_ostream &operator()(raw_ostream &out,
+                                  const MachineInstr &) const {
+    return out; 
+  }
+
+  /// operator() - Print a prefix before each MachineBasicBlock
+  virtual std::ostream &operator()(std::ostream &out,
+                                   const MachineBasicBlock &) const {
+    return out; 
+  }
+
+  /// operator() - Print a prefix before each MachineInstr
+  virtual std::ostream &operator()(std::ostream &out,
+                                   const MachineInstr &) const {
+    return out; 
+  }
+};
+ 
+} // End llvm namespace
+
+#endif
diff --git a/include/llvm/CodeGen/LiveIntervalAnalysis.h b/include/llvm/CodeGen/LiveIntervalAnalysis.h
index 32bf67b8cc..aa5f0ffb01 100644
--- a/include/llvm/CodeGen/LiveIntervalAnalysis.h
+++ b/include/llvm/CodeGen/LiveIntervalAnalysis.h
@@ -538,6 +538,37 @@ namespace llvm {
     void printRegName(unsigned reg) const;
   };
 
+  /// IntervalPrefixPrinter - Print live interval indices before each
+  /// instruction.
+  class IntervalPrefixPrinter : public PrefixPrinter {
+  private:
+    const LiveIntervals &liinfo;
+
+  public:
+    IntervalPrefixPrinter(const LiveIntervals &lii)
+        : liinfo(lii) {};
+
+    // We need null implementations of the other virtual functions to
+    // avoid warnings about hidden virtual functions.
+
+    raw_ostream &operator()(raw_ostream &out,
+                            const MachineBasicBlock &instr) const {
+      return out;
+    }
+    
+    raw_ostream &operator()(raw_ostream &out,
+                            const MachineInstr &instr) const;
+
+    std::ostream &operator()(std::ostream &out,
+                             const MachineBasicBlock &instr) const {
+      return out;
+    }
+    
+    std::ostream &operator()(std::ostream &out,
+                             const MachineInstr &instr) const {
+      return out;
+    }
+  };
 } // End llvm namespace
 
 #endif
diff --git a/include/llvm/CodeGen/MachineBasicBlock.h b/include/llvm/CodeGen/MachineBasicBlock.h
index 90d67e5676..51b2bbf66b 100644
--- a/include/llvm/CodeGen/MachineBasicBlock.h
+++ b/include/llvm/CodeGen/MachineBasicBlock.h
@@ -16,6 +16,7 @@
 
 #include "llvm/CodeGen/MachineInstr.h"
 #include "llvm/ADT/GraphTraits.h"
+#include "llvm/CodeGen/Dump.h"
 
 namespace llvm {
 
@@ -310,10 +311,18 @@ public:
 
   // Debugging methods.
   void dump() const;
-  void print(std::ostream &OS) const;
-  void print(std::ostream *OS) const { if (OS) print(*OS); }
-  void print(raw_ostream &OS) const;
-  void print(raw_ostream *OS) const { if (OS) print(*OS); }
+  void print(std::ostream &OS,
+             const PrefixPrinter &prefix = PrefixPrinter()) const;
+  void print(std::ostream *OS,
+             const PrefixPrinter &prefix = PrefixPrinter()) const {
+    if (OS) print(*OS, prefix); 
+  }
+  void print(raw_ostream &OS,
+             const PrefixPrinter &prefix = PrefixPrinter()) const;
+  void print(raw_ostream *OS,
+             const PrefixPrinter &prefix = PrefixPrinter()) const {
+    if (OS) print(*OS, prefix);
+  }
 
   /// getNumber - MachineBasicBlocks are uniquely numbered at the function
   /// level, unless they're not in a MachineFunction yet, in which case this
diff --git a/include/llvm/CodeGen/MachineFunction.h b/include/llvm/CodeGen/MachineFunction.h
index 13e223ff69..57c946c789 100644
--- a/include/llvm/CodeGen/MachineFunction.h
+++ b/include/llvm/CodeGen/MachineFunction.h
@@ -20,6 +20,7 @@
 
 #include "llvm/ADT/ilist.h"
 #include "llvm/Support/DebugLoc.h"
+#include "llvm/CodeGen/Dump.h"
 #include "llvm/CodeGen/MachineBasicBlock.h"
 #include "llvm/Support/Allocator.h"
 #include "llvm/Support/Recycler.h"
@@ -206,8 +207,12 @@ public:
   /// print - Print out the MachineFunction in a format suitable for debugging
   /// to the specified stream.
   ///
-  void print(std::ostream &OS) const;
-  void print(std::ostream *OS) const { if (OS) print(*OS); }
+  void print(std::ostream &OS, 
+             const PrefixPrinter &prefix = PrefixPrinter()) const;
+  void print(std::ostream *OS,
+             const PrefixPrinter &prefix = PrefixPrinter()) const {
+    if (OS) print(*OS, prefix); 
+  }
 
   /// viewCFG - This function is meant for use from the debugger.  You can just
   /// say 'call F->viewCFG()' and a ghostview window should pop up from the