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/TargetMachine.h" 25 #include "llvm/Target/TargetRegisterInfo.h" 26 #include <fstream> 27 using namespace llvm; 28 29 namespace llvm { 30 template<> 31 struct DOTGraphTraits<ScheduleDAG*> : public DefaultDOTGraphTraits { 32 33 DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {} 34 35 static std::string getGraphName(const ScheduleDAG *G) { 36 return G->MF.getName(); 37 } 38 39 static bool renderGraphFromBottomUp() { 40 return true; 41 } 42 43 static bool isNodeHidden(const SUnit *Node) { 44 return (Node->NumPreds > 10 || Node->NumSuccs > 10); 45 } 46 47 static bool hasNodeAddressLabel(const SUnit *Node, 48 const ScheduleDAG *Graph) { 49 return true; 50 } 51 52 /// If you want to override the dot attributes printed for a particular 53 /// edge, override this method. 54 static std::string getEdgeAttributes(const SUnit *Node, 55 SUnitIterator EI, 56 const ScheduleDAG *Graph) { 57 if (EI.isArtificialDep()) 58 return "color=cyan,style=dashed"; 59 if (EI.isCtrlDep()) 60 return "color=blue,style=dashed"; 61 return ""; 62 } 63 64 65 std::string getNodeLabel(const SUnit *Node, const ScheduleDAG *Graph); 66 static std::string getNodeAttributes(const SUnit *N, 67 const ScheduleDAG *Graph) { 68 return "shape=Mrecord"; 69 } 70 71 static void addCustomGraphFeatures(ScheduleDAG *G, 72 GraphWriter<ScheduleDAG*> &GW) { 73 return G->addCustomGraphFeatures(GW); 74 } 75 }; 76 } 77 78 std::string DOTGraphTraits<ScheduleDAG*>::getNodeLabel(const SUnit *SU, 79 const ScheduleDAG *G) { 80 return G->getGraphNodeLabel(SU); 81 } 82 83 /// viewGraph - Pop up a ghostview window with the reachable parts of the DAG 84 /// rendered using 'dot'. 85 /// 86 void ScheduleDAG::viewGraph(const Twine &Name, const Twine &Title) { 87 // This code is only for debugging! 88 #ifndef NDEBUG 89 ViewGraph(this, Name, false, Title); 90 #else 91 errs() << "ScheduleDAG::viewGraph is only available in debug builds on " 92 << "systems with Graphviz or gv!\n"; 93 #endif // NDEBUG 94 } 95 96 /// Out-of-line implementation with no arguments is handy for gdb. 97 void ScheduleDAG::viewGraph() { 98 viewGraph(getDAGName(), "Scheduling-Units Graph for " + getDAGName()); 99 } 100