Bug fix in operator==() and in method fini().
llvm-svn: 4945
diff --git a/llvm/include/Support/SCCIterator.h b/llvm/include/Support/SCCIterator.h
index 42d4bcb..26afdd7 100644
--- a/llvm/include/Support/SCCIterator.h
+++ b/llvm/include/Support/SCCIterator.h
@@ -176,11 +176,12 @@
// Direct loop termination test (I.fini() is more efficient than I == end())
inline bool fini() const {
- return VisitStack.empty();
+ assert(!CurrentSCC.empty() || VisitStack.empty());
+ return CurrentSCC.empty();
}
inline bool operator==(const _Self& x) const {
- return VisitStack == x.VisitStack;
+ return VisitStack == x.VisitStack && CurrentSCC == x.CurrentSCC;
}
inline bool operator!=(const _Self& x) const { return !operator==(x); }
@@ -195,11 +196,11 @@
// Retrieve a pointer to the current SCC. Returns NULL when done.
inline const SccTy* operator*() const {
- assert(!CurrentSCC.empty() || fini());
+ assert(!CurrentSCC.empty() || VisitStack.empty());
return CurrentSCC.empty()? NULL : &CurrentSCC;
}
inline SccTy* operator*() {
- assert(!CurrentSCC.empty() || fini());
+ assert(!CurrentSCC.empty() || VisitStack.empty());
return CurrentSCC.empty()? NULL : &CurrentSCC;
}
};