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/CodeGen/ScheduleDAG.h"
15 #include "llvm/ADT/DenseSet.h"
16 #include "llvm/ADT/StringExtras.h"
17 #include "llvm/CodeGen/MachineConstantPool.h"
18 #include "llvm/CodeGen/MachineFunction.h"
19 #include "llvm/CodeGen/MachineModuleInfo.h"
20 #include "llvm/IR/Constants.h"
21 #include "llvm/Support/Debug.h"
22 #include "llvm/Support/GraphWriter.h"
23 #include "llvm/Support/raw_ostream.h"
24 #include "llvm/Target/TargetRegisterInfo.h"
25 #include <fstream>
26 using namespace llvm;
27 
28 namespace llvm {
29   template<>
30   struct DOTGraphTraits<ScheduleDAG*> : public DefaultDOTGraphTraits {
31 
DOTGraphTraitsllvm::DOTGraphTraits32   DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {}
33 
getGraphNamellvm::DOTGraphTraits34     static std::string getGraphName(const ScheduleDAG *G) {
35       return G->MF.getName();
36     }
37 
renderGraphFromBottomUpllvm::DOTGraphTraits38     static bool renderGraphFromBottomUp() {
39       return true;
40     }
41 
isNodeHiddenllvm::DOTGraphTraits42     static bool isNodeHidden(const SUnit *Node) {
43       return (Node->NumPreds > 10 || Node->NumSuccs > 10);
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(const Twine & Name,const Twine & Title)85 void ScheduleDAG::viewGraph(const Twine &Name, const Twine &Title) {
86   // This code is only for debugging!
87 #ifndef NDEBUG
88   ViewGraph(this, Name, false, Title);
89 #else
90   errs() << "ScheduleDAG::viewGraph is only available in debug builds on "
91          << "systems with Graphviz or gv!\n";
92 #endif  // NDEBUG
93 }
94 
95 /// Out-of-line implementation with no arguments is handy for gdb.
viewGraph()96 void ScheduleDAG::viewGraph() {
97   viewGraph(getDAGName(), "Scheduling-Units Graph for " + getDAGName());
98 }
99