From d672ab936f32a6eef0d7a099e6ef6ceb1a131e49 Mon Sep 17 00:00:00 2001
From: Chris Lattner <sabre@nondot.org>
Date: Tue, 15 Feb 2005 18:40:55 +0000
Subject: [PATCH] Add a new method to make it easy to update graphs.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@20194 91177308-0d34-0410-b5e6-96231b3b80d8
---
 include/llvm/Analysis/DataStructure/DSGraph.h | 12 ++++++++--
 lib/Analysis/DataStructure/DataStructure.cpp  | 24 ++++++++++++++++++-
 2 files changed, 33 insertions(+), 3 deletions(-)

diff --git a/include/llvm/Analysis/DataStructure/DSGraph.h b/include/llvm/Analysis/DataStructure/DSGraph.h
index 760bf5783c2..44b3fac1121 100644
--- a/include/llvm/Analysis/DataStructure/DSGraph.h
+++ b/include/llvm/Analysis/DataStructure/DSGraph.h
@@ -245,8 +245,6 @@ public:
   afc_iterator afc_begin() const { return AuxFunctionCalls.begin(); }
   afc_iterator afc_end() const { return AuxFunctionCalls.end(); }
 
-
-
   /// getInlinedGlobals - Get the set of globals that are have been inlined
   /// (from callees in BU or from callers in TD) into the current graph.
   ///
@@ -298,6 +296,16 @@ public:
     return Nodes.size();
   }
 
+  /// addObjectToGraph - This method can be used to add global, stack, and heap
+  /// objects to the graph.  This can be used when updating DSGraphs due to the
+  /// introduction of new temporary objects.  The new object is not pointed to
+  /// and does not point to any other objects in the graph.  Note that this
+  /// method initializes the type of the DSNode to the declared type of the
+  /// object if UseDeclaredType is true, otherwise it leaves the node type as
+  /// void.
+  DSNode *addObjectToGraph(Value *Ptr, bool UseDeclaredType = true);
+
+
   /// print - Print a dot graph to the specified ostream...
   ///
   void print(std::ostream &O) const;
diff --git a/lib/Analysis/DataStructure/DataStructure.cpp b/lib/Analysis/DataStructure/DataStructure.cpp
index df5530c15df..c0e3ef46b21 100644
--- a/lib/Analysis/DataStructure/DataStructure.cpp
+++ b/lib/Analysis/DataStructure/DataStructure.cpp
@@ -147,7 +147,6 @@ void DSNode::addGlobal(GlobalValue *GV) {
     std::lower_bound(Globals.begin(), Globals.end(), GV);
 
   if (I == Globals.end() || *I != GV) {
-    //assert(GV->getType()->getElementType() == Ty);
     Globals.insert(I, GV);
     NodeType |= GlobalNode;
   }
@@ -1141,6 +1140,29 @@ void DSGraph::updateFromGlobalGraph() {
     }
 }
 
+/// addObjectToGraph - This method can be used to add global, stack, and heap
+/// objects to the graph.  This can be used when updating DSGraphs due to the
+/// introduction of new temporary objects.  The new object is not pointed to
+/// and does not point to any other objects in the graph.
+DSNode *DSGraph::addObjectToGraph(Value *Ptr, bool UseDeclaredType) {
+  assert(isa<PointerType>(Ptr->getType()) && "Ptr is not a pointer!");
+  const Type *Ty = cast<PointerType>(Ptr->getType())->getElementType();
+  DSNode *N = new DSNode(UseDeclaredType ? Ty : 0, this);
+  ScalarMap[Ptr] = N;
+
+  if (GlobalValue *GV = dyn_cast<GlobalValue>(Ptr)) {
+    N->addGlobal(GV);
+  } else if (MallocInst *MI = dyn_cast<MallocInst>(Ptr)) {
+    N->setHeapNodeMarker();
+  } else if (AllocaInst *AI = dyn_cast<AllocaInst>(Ptr)) {
+    N->setAllocaNodeMarker();
+  } else {
+    assert(0 && "Illegal memory object input!");
+  }
+  return N;
+}
+
+
 /// cloneInto - Clone the specified DSGraph into the current graph.  The
 /// translated ScalarMap for the old function is filled into the OldValMap
 /// member, and the translated ReturnNodes map is returned into ReturnNodes.