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 36 DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {} 37 38 static std::string getGraphName(const ScheduleDAG *G) { 39 return G->MF.getFunction()->getName(); 40 } 41 42 static bool renderGraphFromBottomUp() { 43 return true; 44 } 45 46 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. 53 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); 65 static std::string getNodeAttributes(const SUnit *N, 66 const ScheduleDAG *Graph) { 67 return "shape=Mrecord"; 68 } 69 70 static void addCustomGraphFeatures(ScheduleDAG *G, 71 GraphWriter<ScheduleDAG*> &GW) { 72 return G->addCustomGraphFeatures(GW); 73 } 74 }; 75 } 76 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 /// 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
