lib/VMCore/ValueSymbolTable.cpp

   1 //===-- ValueSymbolTable.cpp - Implement the ValueSymbolTable class -------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file implements the ValueSymbolTable class for the VMCore library.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #define DEBUG_TYPE "valuesymtab"
  15 #include "llvm/GlobalValue.h"
  16 #include "llvm/Type.h"
  17 #include "llvm/ValueSymbolTable.h"
  18 #include "llvm/ADT/SmallString.h"
  19 #include "llvm/Support/Debug.h"
  20 using namespace llvm;
  21
  22 // Class destructor
  23 ValueSymbolTable::~ValueSymbolTable() {
  24 #ifndef NDEBUG   // Only do this in -g mode...
  25   for (iterator VI = vmap.begin(), VE = vmap.end(); VI != VE; ++VI)
  26     cerr << "Value still in symbol table! Type = '"
  27          << VI->getValue()->getType()->getDescription() << "' Name = '"
  28          << VI->getKeyData() << "'\n";
  29   assert(vmap.empty() && "Values remain in symbol table!");
  30 #endif
  31 }
  32
  33 // Insert a value into the symbol table with the specified name...
  34 //
  35 void ValueSymbolTable::reinsertValue(Value* V) {
  36   assert(V->hasName() && "Can't insert nameless Value into symbol table");
  37
  38   // Try inserting the name, assuming it won't conflict.
  39   if (vmap.insert(V->Name)) {
  40     //DOUT << " Inserted value: " << V->Name << ": " << *V << "\n";
  41     return;
  42   }
  43
  44   // Otherwise, there is a naming conflict.  Rename this value.
  45   SmallString<128> UniqueName(V->getNameStart(), V->getNameEnd());
  46
  47   // The name is too already used, just free it so we can allocate a new name.
  48   V->Name->Destroy();
  49
  50   unsigned BaseSize = UniqueName.size();
  51   while (1) {
  52     // Trim any suffix off.
  53     UniqueName.resize(BaseSize);
  54     UniqueName.append_uint_32(++LastUnique);
  55     // Try insert the vmap entry with this suffix.
  56     ValueName &NewName =
  57       vmap.GetOrCreateValue(StringRef(UniqueName.data(),
  58                                       UniqueName.size()));
  59     if (NewName.getValue() == 0) {
  60       // Newly inserted name.  Success!
  61       NewName.setValue(V);
  62       V->Name = &NewName;
  63       //DEBUG(DOUT << " Inserted value: " << UniqueName << ": " << *V << "\n");
  64       return;
  65     }
  66   }
  67 }
  68
  69 void ValueSymbolTable::removeValueName(ValueName *V) {
  70   //DEBUG(DOUT << " Removing Value: " << V->getKeyData() << "\n");
  71   // Remove the value from the symbol table.
  72   vmap.remove(V);
  73 }
  74
  75 /// createValueName - This method attempts to create a value name and insert
  76 /// it into the symbol table with the specified name.  If it conflicts, it
  77 /// auto-renames the name and returns that instead.
  78 ValueName *ValueSymbolTable::createValueName(const StringRef &Name, Value *V) {
  79   // In the common case, the name is not already in the symbol table.
  80   ValueName &Entry = vmap.GetOrCreateValue(Name);
  81   if (Entry.getValue() == 0) {
  82     Entry.setValue(V);
  83     //DEBUG(DOUT << " Inserted value: " << Entry.getKeyData() << ": "
  84     //           << *V << "\n");
  85     return &Entry;
  86   }
  87
  88   // Otherwise, there is a naming conflict.  Rename this value.
  89   SmallString<128> UniqueName(Name.begin(), Name.end());
  90
  91   while (1) {
  92     // Trim any suffix off.
  93     UniqueName.resize(Name.size());
  94     UniqueName.append_uint_32(++LastUnique);
  95
  96     // Try insert the vmap entry with this suffix.
  97     ValueName &NewName =
  98       vmap.GetOrCreateValue(StringRef(UniqueName.data(),
  99                                       UniqueName.size()));
 100     if (NewName.getValue() == 0) {
 101       // Newly inserted name.  Success!
 102       NewName.setValue(V);
 103       //DEBUG(DOUT << " Inserted value: " << UniqueName << ": " << *V << "\n");
 104       return &NewName;
 105     }
 106   }
 107 }
 108
 109
 110 // dump - print out the symbol table
 111 //
 112 void ValueSymbolTable::dump() const {
 113   //DOUT << "ValueSymbolTable:\n";
 114   for (const_iterator I = begin(), E = end(); I != E; ++I) {
 115     //DOUT << "  '" << I->getKeyData() << "' = ";
 116     I->getValue()->dump();
 117     //DOUT << "\n";
 118   }
 119 }