lib/Support/GraphWriter.cpp

   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"
  19 using namespace llvm;
  20
  21 void 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... ";
  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... ";
 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... ";
 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 }