[llvm-commits] [poolalloc] r107348 - in /poolalloc/trunk: include/dsa/DSGraph.h lib/DSA/DataStructure.cpp

[Will Dietz]

Author: wdietz2
Date: Wed Jun 30 17:59:39 2010
New Revision: 107348

URL: http://llvm.org/viewvc/llvm-project?rev=107348&view=rev
Log:
expose VANode mapping to clients, make use of it in copyValue()

Modified:
    poolalloc/trunk/include/dsa/DSGraph.h
    poolalloc/trunk/lib/DSA/DataStructure.cpp

Modified: poolalloc/trunk/include/dsa/DSGraph.h
URL: http://llvm.org/viewvc/llvm-project/poolalloc/trunk/include/dsa/DSGraph.h?rev=107348&r1=107347&r2=107348&view=diff
==============================================================================
--- poolalloc/trunk/include/dsa/DSGraph.h (original)
+++ poolalloc/trunk/include/dsa/DSGraph.h Wed Jun 30 17:59:39 2010
@@ -409,6 +409,12 @@
     return I->second;
   }
 
+  /// getVANodes - Return the mapping of functions to their var-arg nodes for
+  /// this graph.
+  ///
+  const VANodesTy &getVANodes() const { return VANodes; }
+        VANodesTy &getVANodes()       { return VANodes; }
+
   /// getVANodeFor - Return the var-arg node for the specified function.
   ///
   DSNodeHandle &getVANodeFor(const Function &F) {

Modified: poolalloc/trunk/lib/DSA/DataStructure.cpp
URL: http://llvm.org/viewvc/llvm-project/poolalloc/trunk/lib/DSA/DataStructure.cpp?rev=107348&r1=107347&r2=107348&view=diff
==============================================================================
--- poolalloc/trunk/lib/DSA/DataStructure.cpp (original)
+++ poolalloc/trunk/lib/DSA/DataStructure.cpp Wed Jun 30 17:59:39 2010
@@ -1122,7 +1122,7 @@
     // Change the Function* in the vanodes map to the ToF
     DSNodeHandle VA = NG->vanodes_begin()->second;
     NG->getVANodes().clear();
-    NG->getOrCreateVANodeFor(ToF) = VA;
+    NG->getVANodes()[ToF] = VA;
     return;
   }