mirror of
https://github.com/RPCSX/llvm.git
synced 2024-12-12 14:17:59 +00:00
1d4f28c6bc
Having two ways to do this doesn't seem terribly helpful and
consistently using the insert version (which we already has) seems like
it'll make the code easier to understand to anyone working with standard
data structures. (I also updated many references to the Entry's
key and value to use first() and second instead of getKey{Data,Length,}
and get/setValue - for similar consistency)
Also removes the GetOrCreateValue functions so there's less surface area
to StringMap to fix/improve/change/accommodate move semantics, etc.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@222319 91177308-0d34-0410-b5e6-96231b3b80d8
116 lines
3.8 KiB
C++
116 lines
3.8 KiB
C++
//===-- ValueSymbolTable.cpp - Implement the ValueSymbolTable class -------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements the ValueSymbolTable class for the IR library.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/IR/ValueSymbolTable.h"
|
|
#include "llvm/ADT/SmallString.h"
|
|
#include "llvm/IR/GlobalValue.h"
|
|
#include "llvm/IR/Type.h"
|
|
#include "llvm/Support/Debug.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
using namespace llvm;
|
|
|
|
#define DEBUG_TYPE "valuesymtab"
|
|
|
|
// Class destructor
|
|
ValueSymbolTable::~ValueSymbolTable() {
|
|
#ifndef NDEBUG // Only do this in -g mode...
|
|
for (iterator VI = vmap.begin(), VE = vmap.end(); VI != VE; ++VI)
|
|
dbgs() << "Value still in symbol table! Type = '"
|
|
<< *VI->getValue()->getType() << "' Name = '"
|
|
<< VI->getKeyData() << "'\n";
|
|
assert(vmap.empty() && "Values remain in symbol table!");
|
|
#endif
|
|
}
|
|
|
|
// Insert a value into the symbol table with the specified name...
|
|
//
|
|
void ValueSymbolTable::reinsertValue(Value* V) {
|
|
assert(V->hasName() && "Can't insert nameless Value into symbol table");
|
|
|
|
// Try inserting the name, assuming it won't conflict.
|
|
if (vmap.insert(V->Name)) {
|
|
//DEBUG(dbgs() << " Inserted value: " << V->Name << ": " << *V << "\n");
|
|
return;
|
|
}
|
|
|
|
// Otherwise, there is a naming conflict. Rename this value.
|
|
SmallString<256> UniqueName(V->getName().begin(), V->getName().end());
|
|
|
|
// The name is too already used, just free it so we can allocate a new name.
|
|
V->Name->Destroy();
|
|
|
|
unsigned BaseSize = UniqueName.size();
|
|
while (1) {
|
|
// Trim any suffix off and append the next number.
|
|
UniqueName.resize(BaseSize);
|
|
raw_svector_ostream(UniqueName) << ++LastUnique;
|
|
|
|
// Try insert the vmap entry with this suffix.
|
|
auto IterBool = vmap.insert(std::make_pair(UniqueName, V));
|
|
if (IterBool.second) {
|
|
// Newly inserted name. Success!
|
|
V->Name = &*IterBool.first;
|
|
//DEBUG(dbgs() << " Inserted value: " << UniqueName << ": " << *V << "\n");
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
void ValueSymbolTable::removeValueName(ValueName *V) {
|
|
//DEBUG(dbgs() << " Removing Value: " << V->getKeyData() << "\n");
|
|
// Remove the value from the symbol table.
|
|
vmap.remove(V);
|
|
}
|
|
|
|
/// createValueName - This method attempts to create a value name and insert
|
|
/// it into the symbol table with the specified name. If it conflicts, it
|
|
/// auto-renames the name and returns that instead.
|
|
ValueName *ValueSymbolTable::createValueName(StringRef Name, Value *V) {
|
|
// In the common case, the name is not already in the symbol table.
|
|
auto IterBool = vmap.insert(std::make_pair(Name, V));
|
|
if (IterBool.second) {
|
|
//DEBUG(dbgs() << " Inserted value: " << Entry.getKeyData() << ": "
|
|
// << *V << "\n");
|
|
return &*IterBool.first;
|
|
}
|
|
|
|
// Otherwise, there is a naming conflict. Rename this value.
|
|
SmallString<256> UniqueName(Name.begin(), Name.end());
|
|
|
|
while (1) {
|
|
// Trim any suffix off and append the next number.
|
|
UniqueName.resize(Name.size());
|
|
raw_svector_ostream(UniqueName) << ++LastUnique;
|
|
|
|
// Try insert the vmap entry with this suffix.
|
|
auto IterBool = vmap.insert(std::make_pair(UniqueName, V));
|
|
if (IterBool.second) {
|
|
// DEBUG(dbgs() << " Inserted value: " << UniqueName << ": " << *V <<
|
|
// "\n");
|
|
return &*IterBool.first;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
// dump - print out the symbol table
|
|
//
|
|
void ValueSymbolTable::dump() const {
|
|
//DEBUG(dbgs() << "ValueSymbolTable:\n");
|
|
for (const_iterator I = begin(), E = end(); I != E; ++I) {
|
|
//DEBUG(dbgs() << " '" << I->getKeyData() << "' = ");
|
|
I->getValue()->dump();
|
|
//DEBUG(dbgs() << "\n");
|
|
}
|
|
}
|