Simplify MDNode::getFunction() and assertLocalFunction() by avoiding extra Function* variable and smallptrset since function-local metadata cannot be cyclic

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@93762 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Victor Hernandez 2010-01-18 20:36:54 +00:00
parent eb231ce51d
commit 54630e1cef

View File

@ -122,66 +122,59 @@ MDNode::~MDNode() {
}
#ifndef NDEBUG
static Function *assertLocalFunction(const MDNode *N,
SmallPtrSet<const MDNode *, 32> &Visited) {
Function *F = NULL;
// Only visit each MDNode once.
if (!Visited.insert(N)) return F;
for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) {
Value *V = N->getOperand(i);
Function *NewF = NULL;
if (!V) continue;
if (Instruction *I = dyn_cast<Instruction>(V))
NewF = I->getParent()->getParent();
else if (BasicBlock *BB = dyn_cast<BasicBlock>(V))
NewF = BB->getParent();
else if (Argument *A = dyn_cast<Argument>(V))
NewF = A->getParent();
else if (MDNode *MD = dyn_cast<MDNode>(V))
if (MD->isFunctionLocal())
NewF = assertLocalFunction(MD, Visited);
if (F && NewF) assert(F == NewF && "inconsistent function-local metadata");
if (!F) F = NewF;
}
return F;
}
#endif
static Function *assertLocalFunction(const MDNode *N) {
if (!N->isFunctionLocal()) return NULL;
static Function *getFunctionHelper(const MDNode *N,
SmallPtrSet<const MDNode *, 32> &Visited) {
assert(N->isFunctionLocal() && "Should only be called on function-local MD");
#ifndef NDEBUG
return assertLocalFunction(N, Visited);
#endif
Function *F = NULL;
// Only visit each MDNode once.
if (!Visited.insert(N)) return F;
for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) {
Value *V = N->getOperand(i);
if (!V) continue;
if (Instruction *I = dyn_cast<Instruction>(V))
F = I->getParent()->getParent();
else if (BasicBlock *BB = dyn_cast<BasicBlock>(V))
F = BB->getParent();
else if (Argument *A = dyn_cast<Argument>(V))
F = A->getParent();
else if (MDNode *MD = dyn_cast<MDNode>(V))
if (MD->isFunctionLocal())
F = getFunctionHelper(MD, Visited);
if (F) break;
if (Instruction *I = dyn_cast<Instruction>(V)) {
if (F) assert(F == I->getParent()->getParent() &&
"inconsistent function-local metadata");
else F = I->getParent()->getParent();
} else if (BasicBlock *BB = dyn_cast<BasicBlock>(V)) {
if (F) assert(F == BB->getParent() &&
"inconsistent function-local metadata");
else F = BB->getParent();
} else if (Argument *A = dyn_cast<Argument>(V)) {
if (F) assert(F == A->getParent() &&
"inconsistent function-local metadata");
else F = A->getParent();
} else if (MDNode *MD = dyn_cast<MDNode>(V)) {
if (Function *NewF = assertLocalFunction(MD)) {
if (F) assert(F == NewF && "inconsistent function-local metadata");
else F = NewF;
}
}
}
return F;
}
#endif
// getFunction - If this metadata is function-local and recursively has a
// function-local operand, return the first such operand's parent function.
// Otherwise, return null.
Function *MDNode::getFunction() const {
#ifndef NDEBUG
return assertLocalFunction(this);
#endif
if (!isFunctionLocal()) return NULL;
SmallPtrSet<const MDNode *, 32> Visited;
return getFunctionHelper(this, Visited);
for (unsigned i = 0, e = getNumOperands(); i != e; ++i) {
Value *V = getOperand(i);
if (!V) continue;
if (Instruction *I = dyn_cast<Instruction>(V))
return I->getParent()->getParent();
if (BasicBlock *BB = dyn_cast<BasicBlock>(V))
return BB->getParent();
if (Argument *A = dyn_cast<Argument>(V))
return A->getParent();
if (MDNode *MD = dyn_cast<MDNode>(V))
if (Function *F = MD->getFunction()) return F;
}
return NULL;
}
// destroy - Delete this node. Only when there are no uses.