Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line | 
|---|---|---|---|
| 14 | pmbaty | 1 | //===- GraphPrinter.h - Create a DOT output describing the Scop. ----------===// | 
| 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 | // Create a DOT output describing the Scop. | ||
| 10 | // | ||
| 11 | // For each function a dot file is created that shows the control flow graph of | ||
| 12 | // the function and highlights the detected Scops. | ||
| 13 | // | ||
| 14 | //===----------------------------------------------------------------------===// | ||
| 15 | |||
| 16 | #ifndef POLLY_SCOP_GRAPH_PRINTER_H | ||
| 17 | #define POLLY_SCOP_GRAPH_PRINTER_H | ||
| 18 | |||
| 19 | #include "polly/ScopDetection.h" | ||
| 20 | #include "polly/Support/ScopLocation.h" | ||
| 21 | #include "llvm/Analysis/DOTGraphTraitsPass.h" | ||
| 22 | #include "llvm/Analysis/RegionInfo.h" | ||
| 23 | #include "llvm/Analysis/RegionIterator.h" | ||
| 24 | #include "llvm/Analysis/RegionPrinter.h" | ||
| 25 | #include "llvm/IR/PassManager.h" | ||
| 26 | |||
| 27 | namespace llvm { | ||
| 28 | |||
| 29 | template <> | ||
| 30 | struct GraphTraits<polly::ScopDetection *> : GraphTraits<RegionInfo *> { | ||
| 31 | static NodeRef getEntryNode(polly::ScopDetection *SD) { | ||
| 32 | return GraphTraits<RegionInfo *>::getEntryNode(SD->getRI()); | ||
| 33 |   } | ||
| 34 | static nodes_iterator nodes_begin(polly::ScopDetection *SD) { | ||
| 35 | return nodes_iterator::begin(getEntryNode(SD)); | ||
| 36 |   } | ||
| 37 | static nodes_iterator nodes_end(polly::ScopDetection *SD) { | ||
| 38 | return nodes_iterator::end(getEntryNode(SD)); | ||
| 39 |   } | ||
| 40 | }; | ||
| 41 | |||
| 42 | template <> | ||
| 43 | struct DOTGraphTraits<polly::ScopDetection *> : DOTGraphTraits<RegionNode *> { | ||
| 44 | DOTGraphTraits(bool isSimple = false) | ||
| 45 | : DOTGraphTraits<RegionNode *>(isSimple) {} | ||
| 46 | static std::string getGraphName(polly::ScopDetection *SD) { | ||
| 47 | return "Scop Graph"; | ||
| 48 |   } | ||
| 49 | |||
| 50 | std::string getEdgeAttributes(RegionNode *srcNode, | ||
| 51 | GraphTraits<RegionInfo *>::ChildIteratorType CI, | ||
| 52 | polly::ScopDetection *SD); | ||
| 53 | |||
| 54 | std::string getNodeLabel(RegionNode *Node, polly::ScopDetection *SD) { | ||
| 55 | return DOTGraphTraits<RegionNode *>::getNodeLabel( | ||
| 56 | Node, reinterpret_cast<RegionNode *>(SD->getRI()->getTopLevelRegion())); | ||
| 57 |   } | ||
| 58 | |||
| 59 | static std::string escapeString(llvm::StringRef String); | ||
| 60 | |||
| 61 |   /// Print the cluster of the subregions. This groups the single basic blocks | ||
| 62 |   /// and adds a different background color for each group. | ||
| 63 | static void printRegionCluster(polly::ScopDetection *SD, const Region *R, | ||
| 64 | raw_ostream &O, unsigned depth = 0); | ||
| 65 | |||
| 66 | static void addCustomGraphFeatures(polly::ScopDetection *SD, | ||
| 67 | GraphWriter<polly::ScopDetection *> &GW); | ||
| 68 | }; | ||
| 69 | } // end namespace llvm | ||
| 70 | |||
| 71 | namespace polly { | ||
| 72 | |||
| 73 | struct ScopViewer final : llvm::DOTGraphTraitsViewer<ScopAnalysis, false> { | ||
| 74 | ScopViewer() : llvm::DOTGraphTraitsViewer<ScopAnalysis, false>("scops") {} | ||
| 75 | |||
| 76 | bool processFunction(Function &F, const ScopDetection &SD) override; | ||
| 77 | }; | ||
| 78 | |||
| 79 | struct ScopOnlyViewer final : llvm::DOTGraphTraitsViewer<ScopAnalysis, false> { | ||
| 80 | ScopOnlyViewer() | ||
| 81 | : llvm::DOTGraphTraitsViewer<ScopAnalysis, false>("scops-only") {} | ||
| 82 | }; | ||
| 83 | |||
| 84 | struct ScopPrinter final : llvm::DOTGraphTraitsPrinter<ScopAnalysis, false> { | ||
| 85 | ScopPrinter() : llvm::DOTGraphTraitsPrinter<ScopAnalysis, false>("scops") {} | ||
| 86 | }; | ||
| 87 | |||
| 88 | struct ScopOnlyPrinter final : llvm::DOTGraphTraitsPrinter<ScopAnalysis, true> { | ||
| 89 | ScopOnlyPrinter() | ||
| 90 | : llvm::DOTGraphTraitsPrinter<ScopAnalysis, true>("scopsonly") {} | ||
| 91 | }; | ||
| 92 | |||
| 93 | } // end namespace polly | ||
| 94 | |||
| 95 | #endif /* POLLY_SCOP_GRAPH_PRINTER_H */ |