1//===-- ScheduleDAGPrinter.cpp - Implement ScheduleDAG::viewGraph() -------===// 2// 3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4// See https://llvm.org/LICENSE.txt for license information. 5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6// 7//===----------------------------------------------------------------------===// 8// 9// This implements the ScheduleDAG::viewGraph method. 10// 11//===----------------------------------------------------------------------===// 12 13#include "llvm/CodeGen/MachineFunction.h" 14#include "llvm/CodeGen/ScheduleDAG.h" 15#include "llvm/Support/GraphWriter.h" 16#include "llvm/Support/raw_ostream.h" 17using namespace llvm; 18 19namespace llvm { 20 template<> 21 struct DOTGraphTraits<ScheduleDAG*> : public DefaultDOTGraphTraits { 22 23 DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {} 24 25 static std::string getGraphName(const ScheduleDAG *G) { 26 return std::string(G->MF.getName()); 27 } 28 29 static bool renderGraphFromBottomUp() { 30 return true; 31 } 32 33 static bool isNodeHidden(const SUnit *Node, const ScheduleDAG *G) { 34 return (Node->NumPreds > 10 || Node->NumSuccs > 10); 35 } 36 37 static std::string getNodeIdentifierLabel(const SUnit *Node, 38 const ScheduleDAG *Graph) { 39 std::string R; 40 raw_string_ostream OS(R); 41 OS << static_cast<const void *>(Node); 42 return R; 43 } 44 45 /// If you want to override the dot attributes printed for a particular 46 /// edge, override this method. 47 static std::string getEdgeAttributes(const SUnit *Node, 48 SUnitIterator EI, 49 const ScheduleDAG *Graph) { 50 if (EI.isArtificialDep()) 51 return "color=cyan,style=dashed"; 52 if (EI.isCtrlDep()) 53 return "color=blue,style=dashed"; 54 return ""; 55 } 56 57 58 std::string getNodeLabel(const SUnit *SU, const ScheduleDAG *Graph); 59 static std::string getNodeAttributes(const SUnit *N, 60 const ScheduleDAG *Graph) { 61 return "shape=Mrecord"; 62 } 63 64 static void addCustomGraphFeatures(ScheduleDAG *G, 65 GraphWriter<ScheduleDAG*> &GW) { 66 return G->addCustomGraphFeatures(GW); 67 } 68 }; 69} 70 71std::string DOTGraphTraits<ScheduleDAG*>::getNodeLabel(const SUnit *SU, 72 const ScheduleDAG *G) { 73 return G->getGraphNodeLabel(SU); 74} 75 76/// viewGraph - Pop up a ghostview window with the reachable parts of the DAG 77/// rendered using 'dot'. 78/// 79void ScheduleDAG::viewGraph(const Twine &Name, const Twine &Title) { 80 // This code is only for debugging! 81#ifndef NDEBUG 82 ViewGraph(this, Name, false, Title); 83#else 84 errs() << "ScheduleDAG::viewGraph is only available in debug builds on " 85 << "systems with Graphviz or gv!\n"; 86#endif // NDEBUG 87} 88 89/// Out-of-line implementation with no arguments is handy for gdb. 90void ScheduleDAG::viewGraph() { 91 viewGraph(getDAGName(), "Scheduling-Units Graph for " + getDAGName()); 92} 93