SelectionDAGPrinter.cpp revision 7228aa78686c2920c0f8588628b1278698720d07
166328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner//===-- SelectionDAGPrinter.cpp - Implement SelectionDAG::viewGraph() -----===// 2edf128a7fa90f2b0b7ee24741a04a7ae1ecd6f7eMisha Brukman// 366328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner// The LLVM Compiler Infrastructure 466328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner// 566328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner// This file was developed by the LLVM research group and is distributed under 666328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner// the University of Illinois Open Source License. See LICENSE.TXT for details. 7edf128a7fa90f2b0b7ee24741a04a7ae1ecd6f7eMisha Brukman// 866328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner//===----------------------------------------------------------------------===// 966328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner// 1066328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner// This implements the SelectionDAG::viewGraph method. 1166328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner// 1266328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner//===----------------------------------------------------------------------===// 1366328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner 1466328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner#include "llvm/CodeGen/SelectionDAG.h" 1566328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner#include "llvm/CodeGen/MachineFunction.h" 167228aa78686c2920c0f8588628b1278698720d07Chris Lattner#include "llvm/Target/MRegisterInfo.h" 177228aa78686c2920c0f8588628b1278698720d07Chris Lattner#include "llvm/Target/TargetMachine.h" 1866328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner#include "llvm/Function.h" 1966328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner#include "llvm/Support/GraphWriter.h" 20e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner#include "llvm/ADT/StringExtras.h" 216e741f8842b5570e30f25912dbdaaa2edf36110cChris Lattner#include "llvm/Config/config.h" 2266328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner#include <fstream> 2366328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattnerusing namespace llvm; 2466328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner 25e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattnernamespace llvm { 26e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner template<> 27e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner struct DOTGraphTraits<SelectionDAG*> : public DefaultDOTGraphTraits { 28e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner static std::string getGraphName(const SelectionDAG *G) { 29e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner return G->getMachineFunction().getFunction()->getName(); 30e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner } 31e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner 32e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner static bool renderGraphFromBottomUp() { 33e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner return true; 34e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner } 35e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner 36e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner static std::string getNodeLabel(const SDNode *Node, 37e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner const SelectionDAG *Graph); 38e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner static std::string getNodeAttributes(const SDNode *N) { 39e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner return "shape=Mrecord"; 40e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner } 41fc08d9c789b6698fe5f5904d573bb03fcc52a32dChris Lattner 42fc08d9c789b6698fe5f5904d573bb03fcc52a32dChris Lattner static void addCustomGraphFeatures(SelectionDAG *G, 43fc08d9c789b6698fe5f5904d573bb03fcc52a32dChris Lattner GraphWriter<SelectionDAG*> &GW) { 44fc08d9c789b6698fe5f5904d573bb03fcc52a32dChris Lattner GW.emitSimpleNode(0, "plaintext=circle", "GraphRoot"); 45fc08d9c789b6698fe5f5904d573bb03fcc52a32dChris Lattner GW.emitEdge(0, -1, G->getRoot().Val, -1, ""); 46fc08d9c789b6698fe5f5904d573bb03fcc52a32dChris Lattner } 47e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner }; 48e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner} 49e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner 50e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattnerstd::string DOTGraphTraits<SelectionDAG*>::getNodeLabel(const SDNode *Node, 51e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner const SelectionDAG *G) { 52ad95d6ab201bad53ae155a06d49cc16cfbebe507Chris Lattner std::string Op = Node->getOperationName(G); 53c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner 54ad95d6ab201bad53ae155a06d49cc16cfbebe507Chris Lattner for (unsigned i = 0, e = Node->getNumValues(); i != e; ++i) 55ad95d6ab201bad53ae155a06d49cc16cfbebe507Chris Lattner if (Node->getValueType(i) == MVT::Other) 56ad95d6ab201bad53ae155a06d49cc16cfbebe507Chris Lattner Op += ":ch"; 57ad95d6ab201bad53ae155a06d49cc16cfbebe507Chris Lattner else 58ad95d6ab201bad53ae155a06d49cc16cfbebe507Chris Lattner Op = Op + ":" + MVT::getValueTypeString(Node->getValueType(i)); 59ad95d6ab201bad53ae155a06d49cc16cfbebe507Chris Lattner 60e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner if (const ConstantSDNode *CSDN = dyn_cast<ConstantSDNode>(Node)) { 61e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += ": " + utostr(CSDN->getValue()); 62e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner } else if (const ConstantFPSDNode *CSDN = dyn_cast<ConstantFPSDNode>(Node)) { 63e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += ": " + ftostr(CSDN->getValue()); 64edf128a7fa90f2b0b7ee24741a04a7ae1ecd6f7eMisha Brukman } else if (const GlobalAddressSDNode *GADN = 65e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner dyn_cast<GlobalAddressSDNode>(Node)) { 66e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += ": " + GADN->getGlobal()->getName(); 67dedf2bd5a34dac25e4245f58bb902ced6b64edd9Misha Brukman } else if (const FrameIndexSDNode *FIDN = dyn_cast<FrameIndexSDNode>(Node)) { 68e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += " " + itostr(FIDN->getIndex()); 69e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner } else if (const ConstantPoolSDNode *CP = dyn_cast<ConstantPoolSDNode>(Node)){ 70e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += "<" + utostr(CP->getIndex()) + ">"; 71dedf2bd5a34dac25e4245f58bb902ced6b64edd9Misha Brukman } else if (const BasicBlockSDNode *BBDN = dyn_cast<BasicBlockSDNode>(Node)) { 72e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op = "BB: "; 73e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner const Value *LBB = (const Value*)BBDN->getBasicBlock()->getBasicBlock(); 74e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner if (LBB) 75e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += LBB->getName(); 76e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner //Op += " " + (const void*)BBDN->getBasicBlock(); 77d5d0f9bd20d9df07d6b4d41b7e8ed6d33b6a649dChris Lattner } else if (const RegisterSDNode *R = dyn_cast<RegisterSDNode>(Node)) { 787228aa78686c2920c0f8588628b1278698720d07Chris Lattner if (G && MRegisterInfo::isPhysicalRegister(R->getReg())) { 797228aa78686c2920c0f8588628b1278698720d07Chris Lattner Op = Op + " " + G->getTarget().getRegisterInfo()->getName(R->getReg()); 807228aa78686c2920c0f8588628b1278698720d07Chris Lattner } else { 817228aa78686c2920c0f8588628b1278698720d07Chris Lattner Op += " #" + utostr(R->getReg()); 827228aa78686c2920c0f8588628b1278698720d07Chris Lattner } 83e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner } else if (const ExternalSymbolSDNode *ES = 84e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner dyn_cast<ExternalSymbolSDNode>(Node)) { 85e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += "'" + std::string(ES->getSymbol()) + "'"; 862bf3c26b2bb783aada259f1c70be6dd0798f6126Chris Lattner } else if (const SrcValueSDNode *M = dyn_cast<SrcValueSDNode>(Node)) { 872bf3c26b2bb783aada259f1c70be6dd0798f6126Chris Lattner if (M->getValue()) 882bf3c26b2bb783aada259f1c70be6dd0798f6126Chris Lattner Op += "<" + M->getValue()->getName() + ":" + itostr(M->getOffset()) + ">"; 892bf3c26b2bb783aada259f1c70be6dd0798f6126Chris Lattner else 902bf3c26b2bb783aada259f1c70be6dd0798f6126Chris Lattner Op += "<null:" + itostr(M->getOffset()) + ">"; 91a23e8154dc58a62225074bb4eb1633c363ce331bChris Lattner } else if (const VTSDNode *N = dyn_cast<VTSDNode>(Node)) { 92a23e8154dc58a62225074bb4eb1633c363ce331bChris Lattner std::cerr << ":" << getValueTypeString(N->getVT()); 93e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner } 94e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner return Op; 95e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner} 96edf128a7fa90f2b0b7ee24741a04a7ae1ecd6f7eMisha Brukman 97e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner 9866328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner/// viewGraph - Pop up a ghostview window with the reachable parts of the DAG 9966328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner/// rendered using 'dot'. 10066328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner/// 10166328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattnervoid SelectionDAG::viewGraph() { 102e388b5ea2c599a1db72497bf2d2920895da28f47Chris Lattner// This code is only for debugging! 103c5f44add43bf9ecb7f1f63a320e4440f8f0784c3Chris Lattner#ifndef NDEBUG 10466328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::string Filename = "/tmp/dag." + 10566328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner getMachineFunction().getFunction()->getName() + ".dot"; 10666328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::cerr << "Writing '" << Filename << "'... "; 10766328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::ofstream F(Filename.c_str()); 10866328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner 10966328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner if (!F) { 11066328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::cerr << " error opening file for writing!\n"; 11166328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner return; 11266328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner } 11366328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner 11466328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner WriteGraph(F, this); 11566328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner F.close(); 11666328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::cerr << "\n"; 11766328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner 118f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner#ifdef HAVE_GRAPHVIZ 119f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner std::cerr << "Running 'Graphviz' program... " << std::flush; 120c5f44add43bf9ecb7f1f63a320e4440f8f0784c3Chris Lattner if (system((LLVM_PATH_GRAPHVIZ " " + Filename).c_str())) { 121f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner std::cerr << "Error viewing graph: 'Graphviz' not in path?\n"; 122f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner } else { 1234c64dd7977bc676399df0d5d2bf8016a3242780eChris Lattner system(("rm " + Filename).c_str()); 124f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner return; 125f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner } 126cd33eef1b2dc1e81ab57b4857cb63c47bb62a496Misha Brukman#endif // HAVE_GRAPHVIZ 127f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner 128c5f44add43bf9ecb7f1f63a320e4440f8f0784c3Chris Lattner#ifdef HAVE_GV 12966328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::cerr << "Running 'dot' program... " << std::flush; 13066328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner if (system(("dot -Tps -Nfontname=Courier -Gsize=7.5,10 " + Filename 13166328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner + " > /tmp/dag.tempgraph.ps").c_str())) { 132f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner std::cerr << "Error viewing graph: 'dot' not in path?\n"; 13366328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner } else { 13466328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::cerr << "\n"; 135c5f44add43bf9ecb7f1f63a320e4440f8f0784c3Chris Lattner system(LLVM_PATH_GV " /tmp/dag.tempgraph.ps"); 13666328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner } 13766328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner system(("rm " + Filename + " /tmp/dag.tempgraph.ps").c_str()); 138c5f44add43bf9ecb7f1f63a320e4440f8f0784c3Chris Lattner return; 139cd33eef1b2dc1e81ab57b4857cb63c47bb62a496Misha Brukman#endif // HAVE_GV 140cd33eef1b2dc1e81ab57b4857cb63c47bb62a496Misha Brukman#endif // NDEBUG 141c5f44add43bf9ecb7f1f63a320e4440f8f0784c3Chris Lattner std::cerr << "SelectionDAG::viewGraph is only available in debug builds on " 142c5f44add43bf9ecb7f1f63a320e4440f8f0784c3Chris Lattner << "systems with Graphviz or gv!\n"; 143cd33eef1b2dc1e81ab57b4857cb63c47bb62a496Misha Brukman 144cd33eef1b2dc1e81ab57b4857cb63c47bb62a496Misha Brukman#ifndef NDEBUG 1454c64dd7977bc676399df0d5d2bf8016a3242780eChris Lattner system(("rm " + Filename).c_str()); 146cd33eef1b2dc1e81ab57b4857cb63c47bb62a496Misha Brukman#endif 14766328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner} 148