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
diff --git a/include/llvm/CodeGen/Dump.h b/include/llvm/CodeGen/Dump.h
new file mode 100644
index 0000000..439141c
--- /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 32bf67b..aa5f0ff 100644
--- a/include/llvm/CodeGen/LiveIntervalAnalysis.h
+++ b/include/llvm/CodeGen/LiveIntervalAnalysis.h
@@ -538,6 +538,37 @@
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 90d67e5..51b2bbf 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 @@
// 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 13e223f..57c946c 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 @@
/// 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
diff --git a/lib/CodeGen/Dump.cpp b/lib/CodeGen/Dump.cpp
new file mode 100644
index 0000000..2e70d5f
--- /dev/null
+++ b/lib/CodeGen/Dump.cpp
@@ -0,0 +1,18 @@
+//===- lib/Support/Dump.h - Virtual function homes --------------*- 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 virtual function homes.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/CodeGen/Dump.h"
+
+using namespace llvm;
+
+PrefixPrinter::~PrefixPrinter() {}
diff --git a/lib/CodeGen/LiveIntervalAnalysis.cpp b/lib/CodeGen/LiveIntervalAnalysis.cpp
index 2c7cc76..63a391a 100644
--- a/lib/CodeGen/LiveIntervalAnalysis.cpp
+++ b/lib/CodeGen/LiveIntervalAnalysis.cpp
@@ -2501,3 +2501,9 @@
return LR;
}
+
+raw_ostream &
+IntervalPrefixPrinter::operator()(raw_ostream &out,
+ const MachineInstr &instr) const {
+ return out << liinfo.getInstructionIndex(&instr);
+}
diff --git a/lib/CodeGen/MachineBasicBlock.cpp b/lib/CodeGen/MachineBasicBlock.cpp
index 80ca872..b30a3fc 100644
--- a/lib/CodeGen/MachineBasicBlock.cpp
+++ b/lib/CodeGen/MachineBasicBlock.cpp
@@ -153,12 +153,14 @@
os << " %reg" << RegNo;
}
-void MachineBasicBlock::print(std::ostream &OS) const {
+void MachineBasicBlock::print(std::ostream &OS,
+ const PrefixPrinter &prefix) const {
raw_os_ostream RawOS(OS);
- print(RawOS);
+ print(RawOS, prefix);
}
-void MachineBasicBlock::print(raw_ostream &OS) const {
+void MachineBasicBlock::print(raw_ostream &OS,
+ const PrefixPrinter &prefix) const {
const MachineFunction *MF = getParent();
if(!MF) {
OS << "Can't print out MachineBasicBlock because parent MachineFunction"
@@ -191,7 +193,7 @@
}
for (const_iterator I = begin(); I != end(); ++I) {
- OS << "\t";
+ prefix(OS, *I) << "\t";
I->print(OS, &getParent()->getTarget());
}
diff --git a/lib/CodeGen/MachineFunction.cpp b/lib/CodeGen/MachineFunction.cpp
index 858f403..2e9303a 100644
--- a/lib/CodeGen/MachineFunction.cpp
+++ b/lib/CodeGen/MachineFunction.cpp
@@ -218,7 +218,8 @@
print(*cerr.stream());
}
-void MachineFunction::print(std::ostream &OS) const {
+void MachineFunction::print(std::ostream &OS,
+ const PrefixPrinter &prefix) const {
OS << "# Machine code for " << Fn->getNameStr () << "():\n";
// Print Frame Information
@@ -260,8 +261,10 @@
OS << "\n";
}
- for (const_iterator BB = begin(); BB != end(); ++BB)
- BB->print(OS);
+ for (const_iterator BB = begin(); BB != end(); ++BB) {
+ prefix(OS, *BB);
+ BB->print(OS, prefix);
+ }
OS << "\n# End machine code for " << Fn->getNameStr () << "().\n\n";
}