1 //===-- GraphWriter.cpp - Implements GraphWriter support routines ---------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // This file implements misc. GraphWriter support routines. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #include "llvm/Support/GraphWriter.h" 15 #include "llvm/System/Path.h" 16 #include "llvm/System/Program.h" 17 #include "llvm/Config/config.h" 18 using namespace llvm; 19 20 std::string llvm::DOT::EscapeString(const std::string &Label) { 21 std::string Str(Label); 22 for (unsigned i = 0; i != Str.length(); ++i) 23 switch (Str[i]) { 24 case '\n': 25 Str.insert(Str.begin()+i, '\\'); // Escape character... 26 ++i; 27 Str[i] = 'n'; 28 break; 29 case '\t': 30 Str.insert(Str.begin()+i, ' '); // Convert to two spaces 31 ++i; 32 Str[i] = ' '; 33 break; 34 case '\\': 35 if (i+1 != Str.length()) 36 switch (Str[i+1]) { 37 case 'l': continue; // don't disturb \l 38 case '|': case '{': case '}': 39 Str.erase(Str.begin()+i); continue; 40 default: break; 41 } 42 case '{': case '}': 43 case '<': case '>': 44 case '|': case '"': 45 Str.insert(Str.begin()+i, '\\'); // Escape character... 46 ++i; // don't infinite loop 47 break; 48 } 49 return Str; 50 } 51 52 53 54 void llvm::DisplayGraph(const sys::Path &Filename, bool wait, 55 GraphProgram::Name program) { 56 std::string ErrMsg; 57 #if HAVE_GRAPHVIZ 58 sys::Path Graphviz(LLVM_PATH_GRAPHVIZ); 59 60 std::vector<const char*> args; 61 args.push_back(Graphviz.c_str()); 62 args.push_back(Filename.c_str()); 63 args.push_back(0); 64 65 errs() << "Running 'Graphviz' program... "; 66 if (sys::Program::ExecuteAndWait(Graphviz, &args[0],0,0,0,0,&ErrMsg)) { 67 errs() << "Error viewing graph " << Filename.str() << ": " << ErrMsg 68 << "\n"; 69 return; 70 } 71 Filename.eraseFromDisk(); 72 errs() << " done. \n"; 73 74 #elif HAVE_XDOT_PY 75 std::vector<const char*> args; 76 args.push_back(LLVM_PATH_XDOT_PY); 77 args.push_back(Filename.c_str()); 78 79 switch (program) { 80 case GraphProgram::DOT: args.push_back("-f"); args.push_back("dot"); break; 81 case GraphProgram::FDP: args.push_back("-f"); args.push_back("fdp"); break; 82 case GraphProgram::NEATO: args.push_back("-f"); args.push_back("neato");break; 83 case GraphProgram::TWOPI: args.push_back("-f"); args.push_back("twopi");break; 84 case GraphProgram::CIRCO: args.push_back("-f"); args.push_back("circo");break; 85 default: errs() << "Unknown graph layout name; using default.\n"; 86 } 87 88 args.push_back(0); 89 90 errs() << "Running 'xdot.py' program... "; 91 if (sys::Program::ExecuteAndWait(sys::Path(LLVM_PATH_XDOT_PY), 92 &args[0],0,0,0,0,&ErrMsg)) { 93 errs() << "Error viewing graph " << Filename.str() << ": " << ErrMsg 94 << "\n"; 95 return; 96 } 97 Filename.eraseFromDisk(); 98 errs() << " done. \n"; 99 100 #elif (HAVE_GV && (HAVE_DOT || HAVE_FDP || HAVE_NEATO || \ 101 HAVE_TWOPI || HAVE_CIRCO)) 102 sys::Path PSFilename = Filename; 103 PSFilename.appendSuffix("ps"); 104 105 sys::Path prog; 106 107 // Set default grapher 108 #if HAVE_CIRCO 109 prog = sys::Path(LLVM_PATH_CIRCO); 110 #endif 111 #if HAVE_TWOPI 112 prog = sys::Path(LLVM_PATH_TWOPI); 113 #endif 114 #if HAVE_NEATO 115 prog = sys::Path(LLVM_PATH_NEATO); 116 #endif 117 #if HAVE_FDP 118 prog = sys::Path(LLVM_PATH_FDP); 119 #endif 120 #if HAVE_DOT 121 prog = sys::Path(LLVM_PATH_DOT); 122 #endif 123 124 // Find which program the user wants 125 #if HAVE_DOT 126 if (program == GraphProgram::DOT) 127 prog = sys::Path(LLVM_PATH_DOT); 128 #endif 129 #if (HAVE_FDP) 130 if (program == GraphProgram::FDP) 131 prog = sys::Path(LLVM_PATH_FDP); 132 #endif 133 #if (HAVE_NEATO) 134 if (program == GraphProgram::NEATO) 135 prog = sys::Path(LLVM_PATH_NEATO); 136 #endif 137 #if (HAVE_TWOPI) 138 if (program == GraphProgram::TWOPI) 139 prog = sys::Path(LLVM_PATH_TWOPI); 140 #endif 141 #if (HAVE_CIRCO) 142 if (program == GraphProgram::CIRCO) 143 prog = sys::Path(LLVM_PATH_CIRCO); 144 #endif 145 146 std::vector<const char*> args; 147 args.push_back(prog.c_str()); 148 args.push_back("-Tps"); 149 args.push_back("-Nfontname=Courier"); 150 args.push_back("-Gsize=7.5,10"); 151 args.push_back(Filename.c_str()); 152 args.push_back("-o"); 153 args.push_back(PSFilename.c_str()); 154 args.push_back(0); 155 156 errs() << "Running '" << prog.str() << "' program... "; 157 158 if (sys::Program::ExecuteAndWait(prog, &args[0], 0, 0, 0, 0, &ErrMsg)) { 159 errs() << "Error viewing graph " << Filename.str() << ": '" 160 << ErrMsg << "\n"; 161 return; 162 } 163 errs() << " done. \n"; 164 165 sys::Path gv(LLVM_PATH_GV); 166 args.clear(); 167 args.push_back(gv.c_str()); 168 args.push_back(PSFilename.c_str()); 169 args.push_back("--spartan"); 170 args.push_back(0); 171 172 ErrMsg.clear(); 173 if (wait) { 174 if (sys::Program::ExecuteAndWait(gv, &args[0],0,0,0,0,&ErrMsg)) 175 errs() << "Error viewing graph: " << ErrMsg << "\n"; 176 Filename.eraseFromDisk(); 177 PSFilename.eraseFromDisk(); 178 } 179 else { 180 sys::Program::ExecuteNoWait(gv, &args[0],0,0,0,&ErrMsg); 181 errs() << "Remember to erase graph files: " << Filename.str() << " " 182 << PSFilename.str() << "\n"; 183 } 184 #elif HAVE_DOTTY 185 sys::Path dotty(LLVM_PATH_DOTTY); 186 187 std::vector<const char*> args; 188 args.push_back(dotty.c_str()); 189 args.push_back(Filename.c_str()); 190 args.push_back(0); 191 192 errs() << "Running 'dotty' program... "; 193 if (sys::Program::ExecuteAndWait(dotty, &args[0],0,0,0,0,&ErrMsg)) { 194 errs() << "Error viewing graph " << Filename.str() << ": " 195 << ErrMsg << "\n"; 196 } else { 197 // Dotty spawns another app and doesn't wait until it returns 198 #if defined (__MINGW32__) || defined (_WINDOWS) 199 return; 200 #endif 201 Filename.eraseFromDisk(); 202 } 203 #endif 204 } 205