diff --git a/include/llvm/Analysis/DSGraph.h b/include/llvm/Analysis/DSGraph.h index c08d404cea9..81979cf74ec 100644 --- a/include/llvm/Analysis/DSGraph.h +++ b/include/llvm/Analysis/DSGraph.h @@ -80,6 +80,10 @@ public: const std::vector &getNodes() const { return Nodes; } std::vector &getNodes() { return Nodes; } + /// getFunctionNames - Return a space separated list of the name of the + /// functions in this graph (if any) + std::string getFunctionNames() const; + /// addNode - Add a new node to the graph. /// void addNode(DSNode *N) { Nodes.push_back(N); } diff --git a/include/llvm/Analysis/DataStructure/DSGraph.h b/include/llvm/Analysis/DataStructure/DSGraph.h index c08d404cea9..81979cf74ec 100644 --- a/include/llvm/Analysis/DataStructure/DSGraph.h +++ b/include/llvm/Analysis/DataStructure/DSGraph.h @@ -80,6 +80,10 @@ public: const std::vector &getNodes() const { return Nodes; } std::vector &getNodes() { return Nodes; } + /// getFunctionNames - Return a space separated list of the name of the + /// functions in this graph (if any) + std::string getFunctionNames() const; + /// addNode - Add a new node to the graph. /// void addNode(DSNode *N) { Nodes.push_back(N); } diff --git a/lib/Analysis/DataStructure/DataStructure.cpp b/lib/Analysis/DataStructure/DataStructure.cpp index fbd223f71b0..cd73acbc7c7 100644 --- a/lib/Analysis/DataStructure/DataStructure.cpp +++ b/lib/Analysis/DataStructure/DataStructure.cpp @@ -703,6 +703,23 @@ Function &DSCallSite::getCaller() const { // DSGraph Implementation //===----------------------------------------------------------------------===// +/// getFunctionNames - Return a space separated list of the name of the +/// functions in this graph (if any) +std::string DSGraph::getFunctionNames() const { + switch (getReturnNodes().size()) { + case 0: return "Globals graph"; + case 1: return getReturnNodes().begin()->first->getName(); + default: + std::string Return; + for (DSGraph::ReturnNodesTy::const_iterator I = getReturnNodes().begin(); + I != getReturnNodes().end(); ++I) + Return += I->first->getName() + " "; + Return.erase(Return.end()-1, Return.end()); // Remove last space character + return Return; + } +} + + DSGraph::DSGraph(const DSGraph &G) : GlobalsGraph(0) { PrintAuxCalls = false; NodeMapTy NodeMap;