+/// CFGOnly flag - This is used to control whether or not the CFG graph printer
+/// prints out the contents of basic blocks or not. This is acceptable because
+/// this code is only really used for debugging purposes.
+///
+static bool CFGOnly = false;
+
+namespace llvm {
+ template<>
+ struct DOTGraphTraits<const MachineFunction*> : public DefaultDOTGraphTraits {
+ static std::string getGraphName(const MachineFunction *F) {
+ return "CFG for '" + F->getFunction()->getName() + "' function";
+ }
+
+ static std::string getNodeLabel(const MachineBasicBlock *Node,
+ const MachineFunction *Graph) {
+ if (CFGOnly && Node->getBasicBlock() &&
+ !Node->getBasicBlock()->getName().empty())
+ return Node->getBasicBlock()->getName() + ":";
+
+ std::ostringstream Out;
+ if (CFGOnly) {
+ Out << Node->getNumber() << ':';
+ return Out.str();
+ }
+
+ Node->print(Out);
+
+ std::string OutStr = Out.str();
+ if (OutStr[0] == '\n') OutStr.erase(OutStr.begin());
+
+ // Process string output to make it nicer...
+ for (unsigned i = 0; i != OutStr.length(); ++i)
+ if (OutStr[i] == '\n') { // Left justify
+ OutStr[i] = '\\';
+ OutStr.insert(OutStr.begin()+i+1, 'l');
+ }
+ return OutStr;
+ }
+ };
+}
+
+void MachineFunction::viewCFG() const
+{
+#ifndef NDEBUG
+ std::string Filename = "/tmp/cfg." + getFunction()->getName() + ".dot";
+ std::cerr << "Writing '" << Filename << "'... ";
+ std::ofstream F(Filename.c_str());
+
+ if (!F) {
+ std::cerr << " error opening file for writing!\n";
+ return;
+ }
+
+ WriteGraph(F, this);
+ F.close();
+ std::cerr << "\n";
+
+#ifdef HAVE_GRAPHVIZ
+ std::cerr << "Running 'Graphviz' program... " << std::flush;
+ if (system((LLVM_PATH_GRAPHVIZ " " + Filename).c_str())) {
+ std::cerr << "Error viewing graph: 'Graphviz' not in path?\n";
+ } else {
+ system(("rm " + Filename).c_str());
+ return;
+ }
+#endif // HAVE_GRAPHVIZ
+
+#ifdef HAVE_GV
+ std::cerr << "Running 'dot' program... " << std::flush;
+ if (system(("dot -Tps -Nfontname=Courier -Gsize=7.5,10 " + Filename
+ + " > /tmp/cfg.tempgraph.ps").c_str())) {
+ std::cerr << "Error running dot: 'dot' not in path?\n";
+ } else {
+ std::cerr << "\n";
+ system("gv /tmp/cfg.tempgraph.ps");
+ }
+ system(("rm " + Filename + " /tmp/cfg.tempgraph.ps").c_str());
+ return;
+#endif // HAVE_GV
+#endif // NDEBUG
+ std::cerr << "MachineFunction::viewCFG is only available in debug builds on "
+ << "systems with Graphviz or gv!\n";
+
+#ifndef NDEBUG
+ system(("rm " + Filename).c_str());
+#endif
+}
+
+void MachineFunction::viewCFGOnly() const
+{
+ CFGOnly = true;
+ viewCFG();
+ CFGOnly = false;
+}
+