GraphWriter.h revision 8c836ce4f86713376b626e30765494f0a4f7e4b3
195b923d548ed2e0f0993bb613868c871646f120cChris Lattner//===-- Support/GraphWriter.h - Write a graph to a .dot file ---*- C++ -*--===// 295b923d548ed2e0f0993bb613868c871646f120cChris Lattner// 395b923d548ed2e0f0993bb613868c871646f120cChris Lattner// This file defines a simple interface that can be used to print out generic 495b923d548ed2e0f0993bb613868c871646f120cChris Lattner// LLVM graphs to ".dot" files. "dot" is a tool that is part of the AT&T 595b923d548ed2e0f0993bb613868c871646f120cChris Lattner// graphviz package (http://www.research.att.com/sw/tools/graphviz/) which can 695b923d548ed2e0f0993bb613868c871646f120cChris Lattner// be used to turn the files output by this interface into a variety of 795b923d548ed2e0f0993bb613868c871646f120cChris Lattner// different graphics formats. 895b923d548ed2e0f0993bb613868c871646f120cChris Lattner// 995b923d548ed2e0f0993bb613868c871646f120cChris Lattner// Graphs do not need to implement any interface past what is already required 1095b923d548ed2e0f0993bb613868c871646f120cChris Lattner// by the GraphTraits template, but they can choose to implement specializations 1195b923d548ed2e0f0993bb613868c871646f120cChris Lattner// of the DOTGraphTraits template if they want to customize the graphs output in 1295b923d548ed2e0f0993bb613868c871646f120cChris Lattner// any way. 1395b923d548ed2e0f0993bb613868c871646f120cChris Lattner// 1495b923d548ed2e0f0993bb613868c871646f120cChris Lattner//===----------------------------------------------------------------------===// 1595b923d548ed2e0f0993bb613868c871646f120cChris Lattner 1695b923d548ed2e0f0993bb613868c871646f120cChris Lattner#ifndef SUPPORT_GRAPHWRITER_H 1795b923d548ed2e0f0993bb613868c871646f120cChris Lattner#define SUPPORT_GRAPHWRITER_H 1895b923d548ed2e0f0993bb613868c871646f120cChris Lattner 1995b923d548ed2e0f0993bb613868c871646f120cChris Lattner#include "Support/DOTGraphTraits.h" 20f6516da9b3b39ccdf0efe8f999ecbae31d2497beChris Lattner#include "Support/GraphTraits.h" 2195b923d548ed2e0f0993bb613868c871646f120cChris Lattner#include <ostream> 2295b923d548ed2e0f0993bb613868c871646f120cChris Lattner 2395b923d548ed2e0f0993bb613868c871646f120cChris Lattnernamespace DOT { // Private functions... 2495b923d548ed2e0f0993bb613868c871646f120cChris Lattner inline std::string EscapeString(const std::string &Label) { 2595b923d548ed2e0f0993bb613868c871646f120cChris Lattner std::string Str(Label); 2695b923d548ed2e0f0993bb613868c871646f120cChris Lattner for (unsigned i = 0; i != Str.length(); ++i) 2795b923d548ed2e0f0993bb613868c871646f120cChris Lattner switch (Str[i]) { 2895b923d548ed2e0f0993bb613868c871646f120cChris Lattner case '\n': 2995b923d548ed2e0f0993bb613868c871646f120cChris Lattner Str.insert(Str.begin()+i, '\\'); // Escape character... 3095b923d548ed2e0f0993bb613868c871646f120cChris Lattner ++i; 3195b923d548ed2e0f0993bb613868c871646f120cChris Lattner Str[i] = 'n'; 3295b923d548ed2e0f0993bb613868c871646f120cChris Lattner break; 3395b923d548ed2e0f0993bb613868c871646f120cChris Lattner case '\t': 3495b923d548ed2e0f0993bb613868c871646f120cChris Lattner Str.insert(Str.begin()+i, ' '); // Convert to two spaces 3595b923d548ed2e0f0993bb613868c871646f120cChris Lattner ++i; 3695b923d548ed2e0f0993bb613868c871646f120cChris Lattner Str[i] = ' '; 3795b923d548ed2e0f0993bb613868c871646f120cChris Lattner break; 3895b923d548ed2e0f0993bb613868c871646f120cChris Lattner case '\\': 3995b923d548ed2e0f0993bb613868c871646f120cChris Lattner if (i+1 != Str.length() && Str[i+1] == 'l') 4095b923d548ed2e0f0993bb613868c871646f120cChris Lattner break; // don't disturb \l 4195b923d548ed2e0f0993bb613868c871646f120cChris Lattner case '{': case '}': 4295b923d548ed2e0f0993bb613868c871646f120cChris Lattner case '<': case '>': 4395b923d548ed2e0f0993bb613868c871646f120cChris Lattner Str.insert(Str.begin()+i, '\\'); // Escape character... 4495b923d548ed2e0f0993bb613868c871646f120cChris Lattner ++i; // don't infinite loop 4595b923d548ed2e0f0993bb613868c871646f120cChris Lattner break; 4695b923d548ed2e0f0993bb613868c871646f120cChris Lattner } 4795b923d548ed2e0f0993bb613868c871646f120cChris Lattner return Str; 4895b923d548ed2e0f0993bb613868c871646f120cChris Lattner } 4995b923d548ed2e0f0993bb613868c871646f120cChris Lattner} 5095b923d548ed2e0f0993bb613868c871646f120cChris Lattner 5195b923d548ed2e0f0993bb613868c871646f120cChris Lattnertemplate<typename GraphType> 528c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattnerclass GraphWriter { 538c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner std::ostream &O; 548c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner const GraphType &G; 5595b923d548ed2e0f0993bb613868c871646f120cChris Lattner 568c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner typedef DOTGraphTraits<GraphType> DOTTraits; 578c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner typedef GraphTraits<GraphType> GTraits; 588c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner typedef typename GTraits::NodeType NodeType; 598c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner typedef typename GTraits::nodes_iterator node_iterator; 608c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner typedef typename GTraits::ChildIteratorType child_iterator; 618c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattnerpublic: 628c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner GraphWriter(std::ostream &o, const GraphType &g, 638c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner const std::string &Name) : O(o), G(g) { 648c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner if (Name.empty()) 658c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner O << "digraph foo {\n"; // Graph name doesn't matter 668c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner else 678c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner O << "digraph " << Name << " {\n"; 6895b923d548ed2e0f0993bb613868c871646f120cChris Lattner 698c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner std::string GraphName = DOTTraits::getGraphName(G); 708c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner if (!GraphName.empty()) 718c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner O << "\tlabel=\"" << DOT::EscapeString(GraphName) << "\";\n"; 728c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner O << DOTTraits::getGraphProperties(G); 738c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner O << "\n"; 7495b923d548ed2e0f0993bb613868c871646f120cChris Lattner 758c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner writeNodes(); 768c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner } 778c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner 788c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner ~GraphWriter() { 798c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner // Finish off the graph 808c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner O << "}\n"; 818c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner } 8295b923d548ed2e0f0993bb613868c871646f120cChris Lattner 838c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner void writeNodes() { 848c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner // Loop over the graph, printing it out... 858c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner for (node_iterator I = GTraits::nodes_begin(G), E = GTraits::nodes_end(G); 868c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner I != E; ++I) 878c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner writeNode(&*I); 888c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner } 898c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner 908c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner void writeNode(NodeType *Node) { 918c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner std::string NodeAttributes = DOTTraits::getNodeAttributes(Node); 928c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner 9363a32de77680dd705d3f2788ea60021ecd4ad51eChris Lattner O << "\tNode" << (void*)Node << " [shape=record,"; 9495b923d548ed2e0f0993bb613868c871646f120cChris Lattner if (!NodeAttributes.empty()) O << NodeAttributes << ","; 9563a32de77680dd705d3f2788ea60021ecd4ad51eChris Lattner O << "label=\"{" 9695b923d548ed2e0f0993bb613868c871646f120cChris Lattner << DOT::EscapeString(DOTTraits::getNodeLabel(Node, G)); 9795b923d548ed2e0f0993bb613868c871646f120cChris Lattner 9895b923d548ed2e0f0993bb613868c871646f120cChris Lattner // Print out the fields of the current node... 998c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner child_iterator EI = GTraits::child_begin(Node); 1008c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner child_iterator EE = GTraits::child_end(Node); 10195b923d548ed2e0f0993bb613868c871646f120cChris Lattner if (EI != EE) { 10295b923d548ed2e0f0993bb613868c871646f120cChris Lattner O << "|{"; 1038c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner 10495b923d548ed2e0f0993bb613868c871646f120cChris Lattner for (unsigned i = 0; EI != EE && i != 64; ++EI, ++i) { 10595b923d548ed2e0f0993bb613868c871646f120cChris Lattner if (i) O << "|"; 10695b923d548ed2e0f0993bb613868c871646f120cChris Lattner O << "<g" << i << ">" << DOTTraits::getEdgeSourceLabel(Node, EI); 10795b923d548ed2e0f0993bb613868c871646f120cChris Lattner } 1088c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner 10995b923d548ed2e0f0993bb613868c871646f120cChris Lattner if (EI != EE) 11095b923d548ed2e0f0993bb613868c871646f120cChris Lattner O << "|truncated..."; 11195b923d548ed2e0f0993bb613868c871646f120cChris Lattner O << "}"; 11295b923d548ed2e0f0993bb613868c871646f120cChris Lattner } 11395b923d548ed2e0f0993bb613868c871646f120cChris Lattner O << "}\"];\n"; // Finish printing the "node" line 1148c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner 11595b923d548ed2e0f0993bb613868c871646f120cChris Lattner // Output all of the edges now 11695b923d548ed2e0f0993bb613868c871646f120cChris Lattner EI = GTraits::child_begin(Node); 1178c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner for (unsigned i = 0; EI != EE && i != 64; ++EI, ++i) 1188c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner writeEdge(Node, i, EI); 1198c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner } 1208c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner 1218c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner void writeEdge(NodeType *Node, unsigned edgeidx, child_iterator EI) { 1228c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner if (NodeType *TargetNode = *EI) { 1238c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner O << "\tNode" << (void*)Node << ":g" << edgeidx << " -> Node" 1248c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner << (void*)TargetNode; 1258c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner if (DOTTraits::edgeTargetsEdgeSource(Node, EI)) { 1268c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner child_iterator TargetIt = DOTTraits::getEdgeTarget(Node, EI); 1278c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner 1288c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner // Figure out which edge this targets... 1298c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner unsigned Offset = std::distance(GTraits::child_begin(TargetNode), 1308c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner TargetIt); 1318c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner if (Offset > 64) Offset = 64; // Targetting the truncated part? 1328c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner O << ":g" << Offset; 13395b923d548ed2e0f0993bb613868c871646f120cChris Lattner } 1348c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner 1358c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner std::string EdgeAttributes = DOTTraits::getEdgeAttributes(Node, EI); 1368c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner if (!EdgeAttributes.empty()) 1378c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner O << "[" << EdgeAttributes << "]"; 1388c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner O << ";\n"; 13995b923d548ed2e0f0993bb613868c871646f120cChris Lattner } 14095b923d548ed2e0f0993bb613868c871646f120cChris Lattner } 1418c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner}; 14295b923d548ed2e0f0993bb613868c871646f120cChris Lattner 1438c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattnertemplate<typename GraphType> 1448c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattnerstd::ostream &WriteGraph(std::ostream &O, const GraphType &G, 1458c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner const std::string &Name = "") { 1468c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner // Start the graph emission process... 1478c836ce4f86713376b626e30765494f0a4f7e4b3Chris Lattner GraphWriter<GraphType> W(O, G, Name); 14895b923d548ed2e0f0993bb613868c871646f120cChris Lattner return O; 14995b923d548ed2e0f0993bb613868c871646f120cChris Lattner} 15095b923d548ed2e0f0993bb613868c871646f120cChris Lattner 15195b923d548ed2e0f0993bb613868c871646f120cChris Lattner#endif 152