Change references from Method to Function
[oota-llvm.git] / lib / VMCore / SymbolTable.cpp
index c32cec936564c45b656b65464669ae26e2eeeb59..bd19291fa5843b6cf4c4aeee5f4334192d041c2b 100644 (file)
@@ -8,8 +8,16 @@
 #include "llvm/InstrTypes.h"
 #include "llvm/DerivedTypes.h"
 #include "llvm/Module.h"
-#include "llvm/Method.h"
+#include "llvm/Function.h"
 #include "Support/StringExtras.h"
+#include <iostream>
+
+using std::string;
+using std::pair;
+using std::make_pair;
+using std::map;
+using std::cerr;
+using std::cout;
 
 #define DEBUG_SYMBOL_TABLE 0
 #define DEBUG_ABSTYPE 0
@@ -26,16 +34,17 @@ SymbolTable::~SymbolTable() {
     }
   }
 
 // TODO: FIXME: BIG ONE: This doesn't unreference abstract types for the planes
 // that could still have entries!
+ // TODO: FIXME: BIG ONE: This doesn't unreference abstract types for the planes
+ // that could still have entries!
 
 #ifndef NDEBUG   // Only do this in -g mode...
   bool LeftoverValues = true;
   for (iterator i = begin(); i != end(); ++i) {
     for (type_iterator I = i->second.begin(); I != i->second.end(); ++I)
-      if (!isa<ConstPoolVal>(I->second) && !isa<Type>(I->second)) {
+      if (!isa<Constant>(I->second) && !isa<Type>(I->second)) {
        cerr << "Value still in symbol table! Type = '"
-            << i->first->getDescription() << "' Name = '" << I->first << "'\n";
+             << i->first->getDescription() << "' Name = '"
+             << I->first << "'\n";
        LeftoverValues = false;
       }
   }
@@ -64,7 +73,7 @@ string SymbolTable::getUniqueName(const Type *Ty, const string &BaseName) {
 
 
 // lookup - Returns null on failure...
-Value *SymbolTable::lookup(const Type *Ty, const string &Name) {
+Value *SymbolTable::localLookup(const Type *Ty, const string &Name) {
   iterator I = find(Ty);
   if (I != end()) {                      // We have symbols in that plane...
     type_iterator J = I->second.find(Name);
@@ -72,13 +81,23 @@ Value *SymbolTable::lookup(const Type *Ty, const string &Name) {
       return J->second;
   }
 
+  return 0;
+}
+
+// lookup - Returns null on failure...
+Value *SymbolTable::lookup(const Type *Ty, const string &Name) {
+  Value *LV = localLookup(Ty, Name);
+  if (LV) return LV;
   return ParentSymTab ? ParentSymTab->lookup(Ty, Name) : 0;
 }
 
 void SymbolTable::remove(Value *N) {
   assert(N->hasName() && "Value doesn't have name!");
+  if (InternallyInconsistent) return;
 
   iterator I = find(N->getType());
+  assert(I != end() &&
+         "Trying to remove a type that doesn't have a plane yet!");
   removeEntry(I, I->second.find(N->getName()));
 }
 
@@ -133,9 +152,11 @@ Value *SymbolTable::removeEntry(iterator Plane, type_iterator Entry) {
 // name...
 //
 void SymbolTable::insertEntry(const string &Name, const Type *VTy, Value *V) {
+
   // Check to see if there is a naming conflict.  If so, rename this value!
-  if (lookup(VTy, Name)) {
+  if (localLookup(VTy, Name)) {
     string UniqueName = getUniqueName(VTy, Name);
+    assert(InternallyInconsistent == false && "Infinite loop inserting entry!");
     InternallyInconsistent = true;
     V->setName(UniqueName, this);
     InternallyInconsistent = false;
@@ -199,8 +220,8 @@ void SymbolTable::refineAbstractType(const DerivedType *OldType,
 #if DEBUG_ABSTYPE
         cerr << "[Added] refined to abstype: "<<NewType->getDescription()<<endl;
 #endif
+      }
     }
-  }
 
     VarMap &NewPlane = NewTypeIt->second;
     VarMap &OldPlane = TPI->second;
@@ -217,8 +238,8 @@ void SymbolTable::refineAbstractType(const DerivedType *OldType,
         // The only thing we are allowing for now is two method prototypes being
         // folded into one.
         //
-        Method *ExistM = dyn_cast<Method>(TI->second);
-        Method *NewM = dyn_cast<Method>(V.second);
+        Function *ExistM = dyn_cast<Function>(TI->second);
+        Function *NewM = dyn_cast<Function>(V.second);
 
         if (ExistM && NewM && ExistM->isExternal() && NewM->isExternal()) {
           // Ok we have two external methods.  Make all uses of the new one
@@ -234,6 +255,8 @@ void SymbolTable::refineAbstractType(const DerivedType *OldType,
           // that turns remove into a noop.  Thus the name will get null'd
           // out, but the symbol table won't get upset.
           //
+          assert(InternallyInconsistent == false &&
+                 "Symbol table already inconsistent!");
           InternallyInconsistent = true;
 
           // Remove newM from the symtab
@@ -241,7 +264,7 @@ void SymbolTable::refineAbstractType(const DerivedType *OldType,
           InternallyInconsistent = false;
 
           // Now we can remove this method from the module entirely...
-          NewM->getParent()->getMethodList().remove(NewM);
+          NewM->getParent()->getFunctionList().remove(NewM);
           delete NewM;
 
         } else {
@@ -274,29 +297,29 @@ void SymbolTable::refineAbstractType(const DerivedType *OldType,
   }
 
   TPI = find(Type::TypeTy);
-  assert(TPI != end() &&"Type plane not in symbol table but we contain types!");
-
-  // Loop over all of the types in the symbol table, replacing any references to
-  // OldType with references to NewType.  Note that there may be multiple
-  // occurances, and although we only need to remove one at a time, it's faster
-  // to remove them all in one pass.
-  //
-  VarMap &TyPlane = TPI->second;
-  for (VarMap::iterator I = TyPlane.begin(), E = TyPlane.end(); I != E; ++I)
-    if (I->second == (Value*)OldType) {  // FIXME when Types aren't const.
+  if (TPI != end()) {  
+    // Loop over all of the types in the symbol table, replacing any references to
+    // OldType with references to NewType.  Note that there may be multiple
+    // occurances, and although we only need to remove one at a time, it's faster
+    // to remove them all in one pass.
+    //
+    VarMap &TyPlane = TPI->second;
+    for (VarMap::iterator I = TyPlane.begin(), E = TyPlane.end(); I != E; ++I)
+      if (I->second == (Value*)OldType) {  // FIXME when Types aren't const.
 #if DEBUG_ABSTYPE
-      cerr << "Removing type " << OldType->getDescription() << endl;
+        cerr << "Removing type " << OldType->getDescription() << endl;
 #endif
-      OldType->removeAbstractTypeUser(this);
-
-      I->second = (Value*)NewType;  // TODO FIXME when types aren't const
-      if (NewType->isAbstract()) {
+        OldType->removeAbstractTypeUser(this);
+        
+        I->second = (Value*)NewType;  // TODO FIXME when types aren't const
+        if (NewType->isAbstract()) {
 #if DEBUG_ABSTYPE
-        cerr << "Added type " << NewType->getDescription() << endl;
+          cerr << "Added type " << NewType->getDescription() << endl;
 #endif
-       cast<const DerivedType>(NewType)->addAbstractTypeUser(this);
+          cast<const DerivedType>(NewType)->addAbstractTypeUser(this);
+        }
       }
-    }
+  }
 }
 
 
@@ -305,11 +328,11 @@ void SymbolTable::refineAbstractType(const DerivedType *OldType,
 #include <algorithm>
 
 static void DumpVal(const pair<const string, Value *> &V) {
-  cout << "  '" << V.first << "' = " << V.second << endl;
+  cout << "  '" << V.first << "' = " << V.second << "\n";
 }
 
 static void DumpPlane(const pair<const Type *, map<const string, Value *> >&P) {
-  cout << "  Plane: " << P.first << endl;
+  cout << "  Plane: " << P.first << "\n";
   for_each(P.second.begin(), P.second.end(), DumpVal);
 }