From 12be02722f840360451f44841d6345ba8d04a2c2 Mon Sep 17 00:00:00 2001
From: Chris Lattner <sabre@nondot.org>
Date: Mon, 10 Jan 2005 23:26:00 +0000
Subject: Put the operation name in each node, put the function name on the
 graph.

llvm-svn: 19444
---
 llvm/lib/CodeGen/SelectionDAG/SelectionDAGPrinter.cpp | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'llvm/lib')

diff --git a/llvm/lib/CodeGen/SelectionDAG/SelectionDAGPrinter.cpp b/llvm/lib/CodeGen/SelectionDAG/SelectionDAGPrinter.cpp
index 9c483332266..db094a0a120 100644
--- a/llvm/lib/CodeGen/SelectionDAG/SelectionDAGPrinter.cpp
+++ b/llvm/lib/CodeGen/SelectionDAG/SelectionDAGPrinter.cpp
@@ -18,6 +18,23 @@
 #include <fstream>
 using namespace llvm;
 
+namespace llvm {
+  template<>
+  struct DOTGraphTraits<SelectionDAG*> : public DefaultDOTGraphTraits {
+    static std::string getGraphName(const SelectionDAG *G) {
+      return G->getMachineFunction().getFunction()->getName();
+    }
+    static std::string getNodeLabel(const SDNode *Node,
+                                    const SelectionDAG *Graph) {
+      return Node->getOperationName();
+    }
+
+    static std::string getNodeAttributes(const SDNode *N) {
+      return "shape=Mrecord";
+    }
+  };
+}
+
 /// viewGraph - Pop up a ghostview window with the reachable parts of the DAG
 /// rendered using 'dot'.
 ///
-- 
cgit v1.2.3