1 files changed, 156 insertions, 0 deletions
diff --git a/lib/Analysis/DataStructure/Printer.cpp b/lib/Analysis/DataStructure/Printer.cpp
new file mode 100644
index 0000000000..c0caaa07d7
--- /dev/null
+++ b/lib/Analysis/DataStructure/Printer.cpp
@@ -0,0 +1,156 @@
+//===- Printer.cpp - Code for printing data structure graphs nicely -------===//
+//
+// This file implements the 'dot' graph printer.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/Analysis/DataStructure.h"
+#include "llvm/Module.h"
+#include "llvm/Assembly/Writer.h"
+#include <fstream>
+#include <sstream>
+
+void DSNode::dump() const { print(std::cerr, 0); }
+
+std::string DSNode::getCaption(Function *F) const {
+  std::stringstream OS;
+  WriteTypeSymbolic(OS, getType(), F ? F->getParent() : 0);
+
+  OS << " ";
+  if (NodeType & ScalarNode) OS << "S";
+  if (NodeType & AllocaNode) OS << "A";
+  if (NodeType & NewNode   ) OS << "N";
+  if (NodeType & GlobalNode) OS << "G";
+  if (NodeType & SubElement) OS << "E";
+  if (NodeType & CastNode  ) OS << "C";
+
+  return OS.str();
+}
+
+static std::string getValueName(Value *V, Function &F) {
+  std::stringstream OS;
+  WriteAsOperand(OS, V, true, true, F.getParent());
+  return OS.str();
+}
+
+
+
+static void replaceIn(std::string &S, char From, const std::string &To) {
+  for (unsigned i = 0; i < S.size(); )
+    if (S[i] == From) {
+      S.replace(S.begin()+i, S.begin()+i+1,
+                To.begin(), To.end());
+      i += To.size();
+    } else {
+      ++i;
+    }
+}
+
+static string escapeLabel(const string &In) {
+  string Label(In);
+  replaceIn(Label, '\\', "\\\\");  // Escape caption...
+  replaceIn(Label, ' ', "\\ ");
+  replaceIn(Label, '{', "\\{");
+  replaceIn(Label, '}', "\\}");
+  return Label;
+}
+
+static void writeEdge(std::ostream &O, const void *SrcNode,
+                      const char *SrcNodePortName, int SrcNodeIdx,
+                      const DSNode *VS, const string &EdgeAttr = "") {
+  O << "\tNode" << SrcNode << SrcNodePortName;
+  if (SrcNodeIdx != -1) O << SrcNodeIdx;
+  O << " -> Node" << (void*)VS;
+
+  if (!EdgeAttr.empty())
+    O << "[" << EdgeAttr << "]";
+  O << ";\n";
+}
+
+void DSNode::print(std::ostream &O, Function *F) const {
+  string Caption = escapeLabel(getCaption(F));
+
+  O << "\tNode" << (void*)this << " [ label =\"{" << Caption;
+
+  if (!Links.empty()) {
+    O << "|{";
+    for (unsigned i = 0; i < Links.size(); ++i) {
+      if (i) O << "|";
+      O << "<g" << i << ">";
+    }
+    O << "}";
+  }
+  O << "}\"];\n";
+
+  for (unsigned i = 0; i < Links.size(); ++i)
+    if (Links[i])
+      writeEdge(O, this, ":g", i, Links[i]);
+}
+
+void DSGraph::print(std::ostream &O) const {
+  O << "digraph DataStructures {\n"
+    << "\tnode [shape=Mrecord];\n"
+    << "\tedge [arrowtail=\"dot\"];\n"
+    << "\tsize=\"10,7.5\";\n"
+    << "\trotate=\"90\";\n"
+    << "\tlabel=\"Function\\ " << Func.getName() << "\";\n\n";
+
+  // Output all of the nodes...
+  for (unsigned i = 0, e = Nodes.size(); i != e; ++i)
+    Nodes[i]->print(O, &Func);
+
+  O << "\n";
+  // Output all of the nodes edges for scalar labels
+  for (std::map<Value*, DSNodeHandle>::const_iterator I = ValueMap.begin(),
+         E = ValueMap.end(); I != E; ++I) {
+    O << "\tNode" << (void*)I->first << "[ shape=circle, label =\""
+      << escapeLabel(getValueName(I->first, Func)) << "\",style=dotted];\n";
+    writeEdge(O, I->first, "",-1, I->second.get(),"arrowtail=tee,style=dotted");
+  }
+
+  // Output the returned value pointer...
+  if (RetNode != 0) {
+    O << "\tNode0x1" << "[ shape=circle, label =\""
+      << escapeLabel("Return") << "\"];\n";
+    writeEdge(O, (void*)1, "", -1, RetNode, "arrowtail=tee,style=dotted");
+  }    
+
+  // Output all of the call nodes...
+  for (unsigned i = 0, e = FunctionCalls.size(); i != e; ++i) {
+    const std::vector<DSNodeHandle> &Call = FunctionCalls[i];
+    O << "\tNode" << (void*)&Call << " [shape=record,label=\"{call|{";
+    for (unsigned j = 0, e = Call.size(); j != e; ++j) {
+      if (j) O << "|";
+      O << "<g" << j << ">";
+    }
+    O << "}}\"];\n";
+
+    for (unsigned j = 0, e = Call.size(); j != e; ++j)
+      if (Call[j])
+        writeEdge(O, &Call, ":g", j, Call[j]);
+  }
+
+
+  O << "}\n";
+}
+
+
+
+
+// print - Print out the analysis results...
+void LocalDataStructures::print(std::ostream &O, Module *M) const {
+  for (Module::iterator I = M->begin(), E = M->end(); I != E; ++I)
+    if (!I->isExternal()) {
+      std::string Filename = "ds." + I->getName() + ".dot";
+      O << "Writing '" << Filename << "'...";
+      std::ofstream F(Filename.c_str());
+
+      if (F.good()) {
+        DSGraph &Graph = getDSGraph(*I);
+        Graph.print(F);
+        O << " [" << Graph.getGraphSize() << "]\n";
+      } else {
+        O << "  error opening file for writing!\n";
+      }
+    }
+}