SelectionDAGPrinter.cpp revision e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6 
1//===-- SelectionDAGPrinter.cpp - Implement SelectionDAG::viewGraph() -----===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file was developed by the LLVM research group and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This implements the SelectionDAG::viewGraph method.
11//
12//===----------------------------------------------------------------------===//
13
14#include "llvm/CodeGen/SelectionDAG.h"
15#include "llvm/CodeGen/MachineFunction.h"
16#include "llvm/Function.h"
17#include "llvm/Support/GraphWriter.h"
18#include <fstream>
19using namespace llvm;
20
21namespace llvm {
22  template<>
23  struct DOTGraphTraits<SelectionDAG*> : public DefaultDOTGraphTraits {
24    static std::string getGraphName(const SelectionDAG *G) {
25      return G->getMachineFunction().getFunction()->getName();
26    }
27    static std::string getNodeLabel(const SDNode *Node,
28                                    const SelectionDAG *Graph) {
29      return Node->getOperationName();
30    }
31
32    static std::string getNodeAttributes(const SDNode *N) {
33      return "shape=Mrecord";
34    }
35  };
36}
37
38/// viewGraph - Pop up a ghostview window with the reachable parts of the DAG
39/// rendered using 'dot'.
40///
41void SelectionDAG::viewGraph() {
42  std::string Filename = "/tmp/dag." +
43    getMachineFunction().getFunction()->getName() + ".dot";
44  std::cerr << "Writing '" << Filename << "'... ";
45  std::ofstream F(Filename.c_str());
46
47  if (!F) {
48    std::cerr << "  error opening file for writing!\n";
49    return;
50  }
51
52  WriteGraph(F, this);
53  F.close();
54  std::cerr << "\n";
55
56  std::cerr << "Running 'dot' program... " << std::flush;
57  if (system(("dot -Tps -Nfontname=Courier -Gsize=7.5,10 " + Filename
58              + " > /tmp/dag.tempgraph.ps").c_str())) {
59    std::cerr << "Error running dot: 'dot' not in path?\n";
60  } else {
61    std::cerr << "\n";
62    system("gv /tmp/dag.tempgraph.ps");
63  }
64  system(("rm " + Filename + " /tmp/dag.tempgraph.ps").c_str());
65}
66