GraphWriter.cpp revision 00ad26ff5760ff2d1b24acb18718e63541088923 
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/Support/Streams.h"
16#include "llvm/System/Path.h"
17#include "llvm/System/Program.h"
18#include "llvm/Config/config.h"
19using namespace llvm;
20
21void llvm::DisplayGraph(const sys::Path &Filename, bool wait,
22                        GraphProgram::Name program) {
23  std::string ErrMsg;
24#if HAVE_GRAPHVIZ
25  sys::Path Graphviz(LLVM_PATH_GRAPHVIZ);
26
27  std::vector<const char*> args;
28  args.push_back(Graphviz.c_str());
29  args.push_back(Filename.c_str());
30  args.push_back(0);
31
32  cerr << "Running 'Graphviz' program... " << std::flush;
33  if (sys::Program::ExecuteAndWait(Graphviz, &args[0],0,0,0,0,&ErrMsg)) {
34     cerr << "Error viewing graph " << Filename << ": " << ErrMsg << "\n";
35  }
36  else {
37     Filename.eraseFromDisk();
38  }
39
40#elif (HAVE_GV && (HAVE_DOT || HAVE_FDP || HAVE_NEATO || \
41                   HAVE_TWOPI || HAVE_CIRCO))
42  sys::Path PSFilename = Filename;
43  PSFilename.appendSuffix("ps");
44
45  sys::Path prog;
46
47  // Set default grapher
48#if HAVE_CIRCO
49  prog = sys::Path(LLVM_PATH_CIRCO);
50#endif
51#if HAVE_TWOPI
52  prog = sys::Path(LLVM_PATH_TWOPI);
53#endif
54#if HAVE_NEATO
55  prog = sys::Path(LLVM_PATH_NEATO);
56#endif
57#if HAVE_FDP
58  prog = sys::Path(LLVM_PATH_FDP);
59#endif
60#if HAVE_DOT
61  prog = sys::Path(LLVM_PATH_DOT);
62#endif
63
64  // Find which program the user wants
65#if HAVE_DOT
66  if (program == GraphProgram::DOT) {
67    prog = sys::Path(LLVM_PATH_DOT);
68  }
69#endif
70#if (HAVE_FDP)
71  if (program == GraphProgram::FDP) {
72    prog = sys::Path(LLVM_PATH_FDP);
73  }
74#endif
75#if (HAVE_NEATO)
76  if (program == GraphProgram::NEATO) {
77    prog = sys::Path(LLVM_PATH_NEATO);
78    }
79#endif
80#if (HAVE_TWOPI)
81  if (program == GraphProgram::TWOPI) {
82    prog = sys::Path(LLVM_PATH_TWOPI);
83  }
84#endif
85#if (HAVE_CIRCO)
86  if (program == GraphProgram::CIRCO) {
87    prog = sys::Path(LLVM_PATH_CIRCO);
88  }
89#endif
90
91  std::vector<const char*> args;
92  args.push_back(prog.c_str());
93  args.push_back("-Tps");
94  args.push_back("-Nfontname=Courier");
95  args.push_back("-Gsize=7.5,10");
96  args.push_back(Filename.c_str());
97  args.push_back("-o");
98  args.push_back(PSFilename.c_str());
99  args.push_back(0);
100
101  cerr << "Running '" << prog << "' program... " << std::flush;
102
103  if (sys::Program::ExecuteAndWait(prog, &args[0],0,0,0,0,&ErrMsg)) {
104     cerr << "Error viewing graph " << Filename << ": '" << ErrMsg << "\n";
105  } else {
106    cerr << " done. \n";
107
108    sys::Path gv(LLVM_PATH_GV);
109    args.clear();
110    args.push_back(gv.c_str());
111    args.push_back(PSFilename.c_str());
112    args.push_back("-spartan");
113    args.push_back(0);
114
115    ErrMsg.clear();
116    if (wait) {
117       if (sys::Program::ExecuteAndWait(gv, &args[0],0,0,0,0,&ErrMsg)) {
118          cerr << "Error viewing graph: " << ErrMsg << "\n";
119       }
120       Filename.eraseFromDisk();
121       PSFilename.eraseFromDisk();
122    }
123    else {
124       sys::Program::ExecuteNoWait(gv, &args[0],0,0,0,&ErrMsg);
125       cerr << "Remember to erase graph files: " << Filename << " " << PSFilename << "\n";
126    }
127  }
128#elif HAVE_DOTTY
129  sys::Path dotty(LLVM_PATH_DOTTY);
130
131  std::vector<const char*> args;
132  args.push_back(dotty.c_str());
133  args.push_back(Filename.c_str());
134  args.push_back(0);
135
136  cerr << "Running 'dotty' program... " << std::flush;
137  if (sys::Program::ExecuteAndWait(dotty, &args[0],0,0,0,0,&ErrMsg)) {
138     cerr << "Error viewing graph " << Filename << ": " << ErrMsg << "\n";
139  } else {
140#ifdef __MINGW32__ // Dotty spawns another app and doesn't wait until it returns
141    return;
142#endif
143    Filename.eraseFromDisk();
144  }
145#endif
146}
147