commit 710ad9343f32b33b336369b20edad1a21a0b3299
author Ted Kremenek <kremenek@apple.com> Thu Aug 28 03:54:51 2008 +0000
committer Ted Kremenek <kremenek@apple.com> Thu Aug 28 03:54:51 2008 +0000
tree 4a9c1d12f4fb17356cd2c4d93cfdd23277599771
parent 9b111a08c39a4449e3f7fe37b4e504e4993fe2df

Automatically execute ubiviz.
Cleanup ubigraph files when don't drawing the graphs.


git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@55459 91177308-0d34-0410-b5e6-96231b3b80d8

Driver/AnalysisConsumer.cpp

diff --git a/Driver/AnalysisConsumer.cpp b/Driver/AnalysisConsumer.cpp
index 81f1455..a8390e1 100644
--- a/Driver/AnalysisConsumer.cpp
+++ b/Driver/AnalysisConsumer.cpp
@@ -33,6 +33,7 @@
 #include "llvm/Support/Streams.h"
 #include "llvm/Support/raw_ostream.h"
 #include "llvm/System/Path.h"
+#include "llvm/System/Program.h"
 #include <vector>
 
 using namespace clang;
@@ -470,13 +471,19 @@
   
 class UbigraphViz : public ExplodedNodeImpl::Auditor {
   llvm::OwningPtr<llvm::raw_ostream> Out;
+  llvm::sys::Path Dir, Filename;
   unsigned Cntr;
 
   typedef llvm::DenseMap<void*,unsigned> VMap;
   VMap M;
   
 public:
-  UbigraphViz(llvm::raw_ostream* out) : Out(out), Cntr(0) {}  
+  UbigraphViz(llvm::raw_ostream* out, llvm::sys::Path& dir,
+              llvm::sys::Path& filename)
+    : Out(out), Dir(dir), Filename(filename), Cntr(0) {} 
+  
+  ~UbigraphViz();
+  
   virtual void AddEdge(ExplodedNodeImpl* Src, ExplodedNodeImpl* Dst);  
 };
   
@@ -485,10 +492,11 @@
 static ExplodedNodeImpl::Auditor* CreateUbiViz() {
   std::string ErrMsg;
   
-  llvm::sys::Path Filename = llvm::sys::Path::GetTemporaryDirectory(&ErrMsg);
+  llvm::sys::Path Dir = llvm::sys::Path::GetTemporaryDirectory(&ErrMsg);
   if (!ErrMsg.empty())
     return 0;
 
+  llvm::sys::Path Filename = Dir;
   Filename.appendComponent("llvm_ubi");
   Filename.makeUnique(true,&ErrMsg);
 
@@ -504,7 +512,7 @@
   if (!ErrMsg.empty())
     return 0;
   
-  return new UbigraphViz(Stream.take());
+  return new UbigraphViz(Stream.take(), Dir, Filename);
 }
 
 void UbigraphViz::AddEdge(ExplodedNodeImpl* Src, ExplodedNodeImpl* Dst) {
@@ -535,3 +543,20 @@
        << ", ('arrow','true'), ('oriented', 'true'))\n";
 }
 
+UbigraphViz::~UbigraphViz() {
+  Out.reset(0);
+  llvm::cerr << "Running 'ubiviz' program... ";
+  std::string ErrMsg;
+  llvm::sys::Path Ubiviz = llvm::sys::Program::FindProgramByName("ubiviz");
+  std::vector<const char*> args;
+  args.push_back(Ubiviz.c_str());
+  args.push_back(Filename.c_str());
+  args.push_back(0);
+  
+  if (llvm::sys::Program::ExecuteAndWait(Ubiviz, &args[0],0,0,0,0,&ErrMsg)) {
+    llvm::cerr << "Error viewing graph: " << ErrMsg << "\n";
+  }
+  
+  // Delete the directory.
+  Dir.eraseFromDisk(true); 
+}
\ No newline at end of file