SelectionDAGPrinter.cpp revision e388b5ea2c599a1db72497bf2d2920895da28f47
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" 1666328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner#include "llvm/Function.h" 1766328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner#include "llvm/Support/GraphWriter.h" 18e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner#include "llvm/ADT/StringExtras.h" 1966328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner#include <fstream> 2066328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattnerusing namespace llvm; 2166328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner 22e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattnernamespace llvm { 23e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner template<> 24e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner struct DOTGraphTraits<SelectionDAG*> : public DefaultDOTGraphTraits { 25e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner static std::string getGraphName(const SelectionDAG *G) { 26e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner return G->getMachineFunction().getFunction()->getName(); 27e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner } 28e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner 29e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner static bool renderGraphFromBottomUp() { 30e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner return true; 31e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner } 32e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner 33e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner static std::string getNodeLabel(const SDNode *Node, 34e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner const SelectionDAG *Graph); 35e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner static std::string getNodeAttributes(const SDNode *N) { 36e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner return "shape=Mrecord"; 37e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner } 38fc08d9c789b6698fe5f5904d573bb03fcc52a32dChris Lattner 39fc08d9c789b6698fe5f5904d573bb03fcc52a32dChris Lattner static void addCustomGraphFeatures(SelectionDAG *G, 40fc08d9c789b6698fe5f5904d573bb03fcc52a32dChris Lattner GraphWriter<SelectionDAG*> &GW) { 41fc08d9c789b6698fe5f5904d573bb03fcc52a32dChris Lattner GW.emitSimpleNode(0, "plaintext=circle", "GraphRoot"); 42fc08d9c789b6698fe5f5904d573bb03fcc52a32dChris Lattner GW.emitEdge(0, -1, G->getRoot().Val, -1, ""); 43fc08d9c789b6698fe5f5904d573bb03fcc52a32dChris Lattner } 44e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner }; 45e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner} 46e0646b86e3cdc35c5dd0e1c10b7ac564066e3bd6Chris Lattner 47e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattnerstd::string DOTGraphTraits<SelectionDAG*>::getNodeLabel(const SDNode *Node, 48e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner const SelectionDAG *G) { 49e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner std::string Op = Node->getOperationName(); 50c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner 51c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner for (unsigned i = 0, e = Node->getNumValues(); i != e; ++i) { 52c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner switch (Node->getValueType(i)) { 53c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner default: Op += ":unknownvt!"; break; 54c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner case MVT::Other: Op += ":ch"; break; 55c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner case MVT::i1: Op += ":i1"; break; 56c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner case MVT::i8: Op += ":i8"; break; 57c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner case MVT::i16: Op += ":i16"; break; 58c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner case MVT::i32: Op += ":i32"; break; 59c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner case MVT::i64: Op += ":i64"; break; 60c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner case MVT::i128: Op += ":i128"; break; 61c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner case MVT::f32: Op += ":f32"; break; 62c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner case MVT::f64: Op += ":f64"; break; 63c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner case MVT::f80: Op += ":f80"; break; 64c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner case MVT::f128: Op += ":f128"; break; 65c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner case MVT::isVoid: Op += ":void"; break; 66c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner } 67c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner } 68c871e1d56fa9e4c73f01419d4f1fef6d47f24d29Chris Lattner 69e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner if (const ConstantSDNode *CSDN = dyn_cast<ConstantSDNode>(Node)) { 70e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += ": " + utostr(CSDN->getValue()); 71e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner } else if (const ConstantFPSDNode *CSDN = dyn_cast<ConstantFPSDNode>(Node)) { 72e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += ": " + ftostr(CSDN->getValue()); 73edf128a7fa90f2b0b7ee24741a04a7ae1ecd6f7eMisha Brukman } else if (const GlobalAddressSDNode *GADN = 74e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner dyn_cast<GlobalAddressSDNode>(Node)) { 75e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += ": " + GADN->getGlobal()->getName(); 76dedf2bd5a34dac25e4245f58bb902ced6b64edd9Misha Brukman } else if (const FrameIndexSDNode *FIDN = dyn_cast<FrameIndexSDNode>(Node)) { 77e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += " " + itostr(FIDN->getIndex()); 78e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner } else if (const ConstantPoolSDNode *CP = dyn_cast<ConstantPoolSDNode>(Node)){ 79e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += "<" + utostr(CP->getIndex()) + ">"; 80dedf2bd5a34dac25e4245f58bb902ced6b64edd9Misha Brukman } else if (const BasicBlockSDNode *BBDN = dyn_cast<BasicBlockSDNode>(Node)) { 81e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op = "BB: "; 82e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner const Value *LBB = (const Value*)BBDN->getBasicBlock()->getBasicBlock(); 83e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner if (LBB) 84e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += LBB->getName(); 85e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner //Op += " " + (const void*)BBDN->getBasicBlock(); 8618c2f13e0f9d0e5d6227cf6d1881e9ee3d1b6109Chris Lattner } else if (const RegSDNode *C2V = dyn_cast<RegSDNode>(Node)) { 87e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += " #" + utostr(C2V->getReg()); 88e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner } else if (const ExternalSymbolSDNode *ES = 89e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner dyn_cast<ExternalSymbolSDNode>(Node)) { 90e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += "'" + std::string(ES->getSymbol()) + "'"; 912bf3c26b2bb783aada259f1c70be6dd0798f6126Chris Lattner } else if (const SrcValueSDNode *M = dyn_cast<SrcValueSDNode>(Node)) { 922bf3c26b2bb783aada259f1c70be6dd0798f6126Chris Lattner if (M->getValue()) 932bf3c26b2bb783aada259f1c70be6dd0798f6126Chris Lattner Op += "<" + M->getValue()->getName() + ":" + itostr(M->getOffset()) + ">"; 942bf3c26b2bb783aada259f1c70be6dd0798f6126Chris Lattner else 952bf3c26b2bb783aada259f1c70be6dd0798f6126Chris Lattner Op += "<null:" + itostr(M->getOffset()) + ">"; 96e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner } 97e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner return Op; 98e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner} 99edf128a7fa90f2b0b7ee24741a04a7ae1ecd6f7eMisha Brukman 100e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner 10166328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner/// viewGraph - Pop up a ghostview window with the reachable parts of the DAG 10266328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner/// rendered using 'dot'. 10366328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner/// 10466328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattnervoid SelectionDAG::viewGraph() { 105e388b5ea2c599a1db72497bf2d2920895da28f47Chris Lattner// This code is only for debugging! 106e388b5ea2c599a1db72497bf2d2920895da28f47Chris Lattner#ifdef NDEBUG 107e388b5ea2c599a1db72497bf2d2920895da28f47Chris Lattner std::cerr << "SelectionDAG::viewGraph is only available in debug builds!\n"; 108e388b5ea2c599a1db72497bf2d2920895da28f47Chris Lattner#else 10966328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::string Filename = "/tmp/dag." + 11066328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner getMachineFunction().getFunction()->getName() + ".dot"; 11166328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::cerr << "Writing '" << Filename << "'... "; 11266328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::ofstream F(Filename.c_str()); 11366328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner 11466328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner if (!F) { 11566328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::cerr << " error opening file for writing!\n"; 11666328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner return; 11766328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner } 11866328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner 11966328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner WriteGraph(F, this); 12066328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner F.close(); 12166328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::cerr << "\n"; 12266328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner 123f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner#ifdef HAVE_GRAPHVIZ 124f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner std::cerr << "Running 'Graphviz' program... " << std::flush; 125f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner if (system(("Graphviz " + Filename).c_str())) { 126f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner std::cerr << "Error viewing graph: 'Graphviz' not in path?\n"; 127f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner } else { 128f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner return; 129f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner } 130f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner#endif 131f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner 13266328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::cerr << "Running 'dot' program... " << std::flush; 13366328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner if (system(("dot -Tps -Nfontname=Courier -Gsize=7.5,10 " + Filename 13466328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner + " > /tmp/dag.tempgraph.ps").c_str())) { 135f1a2f15ed35c1cf987f6a1a703f855a3c61fdab6Chris Lattner std::cerr << "Error viewing graph: 'dot' not in path?\n"; 13666328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner } else { 13766328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::cerr << "\n"; 13866328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner system("gv /tmp/dag.tempgraph.ps"); 13966328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner } 14066328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner system(("rm " + Filename + " /tmp/dag.tempgraph.ps").c_str()); 141e388b5ea2c599a1db72497bf2d2920895da28f47Chris Lattner#endif 14266328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner} 143