diff options
| -rw-r--r-- | include/llvm/Support/GraphWriter.h | 30 |
1 files changed, 20 insertions, 10 deletions
diff --git a/include/llvm/Support/GraphWriter.h b/include/llvm/Support/GraphWriter.h index 5e91d2543e..adfe5ec6f2 100644 --- a/include/llvm/Support/GraphWriter.h +++ b/include/llvm/Support/GraphWriter.h @@ -92,6 +92,24 @@ public: DTraits = DOTTraits(SN); } + void writeGraph(bool ShortNames = false, + const std::string &Title = "") { + // Start the graph emission process... + GraphWriter<GraphType> W(O, G, ShortNames); + + // Output the header for the graph... + W.writeHeader(Title); + + // Emit all of the nodes in the graph... + W.writeNodes(); + + // Output any customizations on the graph + DOTGraphTraits<GraphType>::addCustomGraphFeatures(G, W); + + // Output the end of the graph + W.writeFooter(); + } + void writeHeader(const std::string &Title) { std::string GraphName = DTraits.getGraphName(G); @@ -286,17 +304,9 @@ raw_ostream &WriteGraph(raw_ostream &O, const GraphType &G, // Start the graph emission process... GraphWriter<GraphType> W(O, G, ShortNames); - // Output the header for the graph... - W.writeHeader(Title); - - // Emit all of the nodes in the graph... - W.writeNodes(); - - // Output any customizations on the graph - DOTGraphTraits<GraphType>::addCustomGraphFeatures(G, W); + // Emit the graph. + W.writeGraph(ShortNames, Title); - // Output the end of the graph - W.writeFooter(); return O; } |