diff options
| -rw-r--r-- | include/llvm/ADT/EquivalenceClasses.h | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/include/llvm/ADT/EquivalenceClasses.h b/include/llvm/ADT/EquivalenceClasses.h index eb6bfa2cad..e318919421 100644 --- a/include/llvm/ADT/EquivalenceClasses.h +++ b/include/llvm/ADT/EquivalenceClasses.h @@ -112,6 +112,19 @@ class EquivalenceClasses { std::set<ECValue> TheMapping; public: + EquivalenceClasses() {} + EquivalenceClasses(const EquivalenceClasses &RHS) { + operator=(RHS); + } + + const EquivalenceClasses &operator=(const EquivalenceClasses &RHS) { + for (iterator I = RHS.begin(), E = RHS.end(); I != E; ++I) + if (I->isLeader()) + insert(I->getData()); + else + unionSets(I->getData(), *RHS.findLeader(I)); + return *this; + } //===--------------------------------------------------------------------===// // Inspection methods @@ -133,6 +146,17 @@ public: return member_iterator(0); } + /// getNumClasses - Return the number of equivalence classes in this set. + /// Note that this is a linear time operation. + unsigned getNumClasses() const { + unsigned NC = 0; + for (iterator I = begin(), E = end(); I != E; ++I) + if (I->isLeader()) ++NC; + return NC; + } + + + //===--------------------------------------------------------------------===// // Mutation methods |