SelectionDAGPrinter.cpp revision edf128a7fa90f2b0b7ee24741a04a7ae1ecd6f7e
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(); 76e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner } else if (const FrameIndexSDNode *FIDN = 77e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner dyn_cast<FrameIndexSDNode>(Node)) { 78e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += " " + itostr(FIDN->getIndex()); 79e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner } else if (const ConstantPoolSDNode *CP = dyn_cast<ConstantPoolSDNode>(Node)){ 80e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += "<" + utostr(CP->getIndex()) + ">"; 81edf128a7fa90f2b0b7ee24741a04a7ae1ecd6f7eMisha Brukman } else if (const BasicBlockSDNode *BBDN = 82e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner dyn_cast<BasicBlockSDNode>(Node)) { 83e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op = "BB: "; 84e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner const Value *LBB = (const Value*)BBDN->getBasicBlock()->getBasicBlock(); 85e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner if (LBB) 86e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += LBB->getName(); 87e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner //Op += " " + (const void*)BBDN->getBasicBlock(); 8818c2f13e0f9d0e5d6227cf6d1881e9ee3d1b6109Chris Lattner } else if (const RegSDNode *C2V = dyn_cast<RegSDNode>(Node)) { 89e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += " #" + utostr(C2V->getReg()); 90e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner } else if (const ExternalSymbolSDNode *ES = 91e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner dyn_cast<ExternalSymbolSDNode>(Node)) { 92e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner Op += "'" + std::string(ES->getSymbol()) + "'"; 938a389bb37603fa99a980475a4f5cdc27fa4014f1Chris Lattner } else if (const MVTSDNode *M = dyn_cast<MVTSDNode>(Node)) { 944b7899343eb492b70868fc73450b708834a3cf99Chris Lattner Op = Op + " ty=" + MVT::getValueTypeString(M->getExtraValueType()); 95e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner } 96e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner return Op; 97e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner} 98edf128a7fa90f2b0b7ee24741a04a7ae1ecd6f7eMisha Brukman 99e9c44cdf18b1235841a5d8bfeb07bee9f5699b9aChris Lattner 10066328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner/// viewGraph - Pop up a ghostview window with the reachable parts of the DAG 10166328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner/// rendered using 'dot'. 10266328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner/// 10366328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattnervoid SelectionDAG::viewGraph() { 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 11866328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::cerr << "Running 'dot' program... " << std::flush; 11966328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner if (system(("dot -Tps -Nfontname=Courier -Gsize=7.5,10 " + Filename 12066328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner + " > /tmp/dag.tempgraph.ps").c_str())) { 12166328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::cerr << "Error running dot: 'dot' not in path?\n"; 12266328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner } else { 12366328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner std::cerr << "\n"; 12466328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner system("gv /tmp/dag.tempgraph.ps"); 12566328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner } 12666328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner system(("rm " + Filename + " /tmp/dag.tempgraph.ps").c_str()); 12766328480bb3eb6aa52e2c155657f3d19d4efea7aChris Lattner} 128