1 //===-- ScheduleDAGPrinter.cpp - Implement ScheduleDAG::viewGraph() -------===//
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 implements the ScheduleDAG::viewGraph method.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "llvm/Constants.h"
15 #include "llvm/Function.h"
16 #include "llvm/Assembly/Writer.h"
17 #include "llvm/CodeGen/ScheduleDAG.h"
18 #include "llvm/CodeGen/MachineConstantPool.h"
19 #include "llvm/CodeGen/MachineFunction.h"
20 #include "llvm/CodeGen/MachineModuleInfo.h"
21 #include "llvm/Target/TargetRegisterInfo.h"
22 #include "llvm/Target/TargetMachine.h"
23 #include "llvm/Support/Debug.h"
24 #include "llvm/Support/GraphWriter.h"
25 #include "llvm/Support/raw_ostream.h"
26 #include "llvm/ADT/DenseSet.h"
27 #include "llvm/ADT/StringExtras.h"
28 #include "llvm/Config/config.h"
29 #include <fstream>
30 using namespace llvm;
31
32 namespace llvm {
33 template<>
34 struct DOTGraphTraits<ScheduleDAG*> : public DefaultDOTGraphTraits {
35
DOTGraphTraitsllvm::DOTGraphTraits36 DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {}
37
getGraphNamellvm::DOTGraphTraits38 static std::string getGraphName(const ScheduleDAG *G) {
39 return G->MF.getFunction()->getName();
40 }
41
renderGraphFromBottomUpllvm::DOTGraphTraits42 static bool renderGraphFromBottomUp() {
43 return true;
44 }
45
hasNodeAddressLabelllvm::DOTGraphTraits46 static bool hasNodeAddressLabel(const SUnit *Node,
47 const ScheduleDAG *Graph) {
48 return true;
49 }
50
51 /// If you want to override the dot attributes printed for a particular
52 /// edge, override this method.
getEdgeAttributesllvm::DOTGraphTraits53 static std::string getEdgeAttributes(const SUnit *Node,
54 SUnitIterator EI,
55 const ScheduleDAG *Graph) {
56 if (EI.isArtificialDep())
57 return "color=cyan,style=dashed";
58 if (EI.isCtrlDep())
59 return "color=blue,style=dashed";
60 return "";
61 }
62
63
64 std::string getNodeLabel(const SUnit *Node, const ScheduleDAG *Graph);
getNodeAttributesllvm::DOTGraphTraits65 static std::string getNodeAttributes(const SUnit *N,
66 const ScheduleDAG *Graph) {
67 return "shape=Mrecord";
68 }
69
addCustomGraphFeaturesllvm::DOTGraphTraits70 static void addCustomGraphFeatures(ScheduleDAG *G,
71 GraphWriter<ScheduleDAG*> &GW) {
72 return G->addCustomGraphFeatures(GW);
73 }
74 };
75 }
76
getNodeLabel(const SUnit * SU,const ScheduleDAG * G)77 std::string DOTGraphTraits<ScheduleDAG*>::getNodeLabel(const SUnit *SU,
78 const ScheduleDAG *G) {
79 return G->getGraphNodeLabel(SU);
80 }
81
82 /// viewGraph - Pop up a ghostview window with the reachable parts of the DAG
83 /// rendered using 'dot'.
84 ///
viewGraph()85 void ScheduleDAG::viewGraph() {
86 // This code is only for debugging!
87 #ifndef NDEBUG
88 if (BB->getBasicBlock())
89 ViewGraph(this, "dag." + MF.getFunction()->getNameStr(), false,
90 "Scheduling-Units Graph for " + MF.getFunction()->getNameStr() +
91 ":" + BB->getBasicBlock()->getNameStr());
92 else
93 ViewGraph(this, "dag." + MF.getFunction()->getNameStr(), false,
94 "Scheduling-Units Graph for " + MF.getFunction()->getNameStr());
95 #else
96 errs() << "ScheduleDAG::viewGraph is only available in debug builds on "
97 << "systems with Graphviz or gv!\n";
98 #endif // NDEBUG
99 }
100