Author: kremenek
Date: Wed Jan 2 15:31:48 2008
New Revision: 45503
URL: http://llvm.org/viewvc/llvm-project?rev=45503&view=rev
Log:
Added iterator and profiling (i.e. FoldingSetNodeID) support to ImmutableMap.
Modified:
llvm/trunk/include/llvm/ADT/ImmutableMap.h
Modified: llvm/trunk/include/llvm/ADT/ImmutableMap.h
URL:
http://llvm.org/viewvc/llvm-project/llvm/trunk/include/llvm/ADT/ImmutableMap.h?rev=45503&r1=45502&r2=45503&view=diff
==============================================================================
--- llvm/trunk/include/llvm/ADT/ImmutableMap.h (original)
+++ llvm/trunk/include/llvm/ADT/ImmutableMap.h Wed Jan 2 15:31:48 2008
@@ -154,8 +154,47 @@
//===--------------------------------------------------===//
void verify() const { if (Root) Root->verify(); }
- unsigned getHeight() const { return Root ? Root->getHeight() : 0; }
+ //===--------------------------------------------------===//
+ // Iterators.
+ //===--------------------------------------------------===//
+
+ class iterator {
+ typename TreeTy::iterator itr;
+
+ iterator() {}
+ iterator(TreeTy* t) : itr(t) {}
+ friend class ImmutableSet<ValT,ValInfo>;
+
+ public:
+ inline value_type_ref operator*() const { return itr->getValue(); }
+ inline key_type_ref getKey() const { return itr->getValue().first; }
+ inline data_type_ref getData() const { return itr->getValue().second; }
+
+ inline iterator& operator++() { ++itr; return *this; }
+ inline iterator operator++(int) { iterator tmp(*this); ++itr; return tmp;
}
+ inline iterator& operator--() { --itr; return *this; }
+ inline iterator operator--(int) { iterator tmp(*this); --itr; return tmp;
}
+ inline bool operator==(const iterator& RHS) const { return RHS.itr == itr;
}
+ inline bool operator!=(const iterator& RHS) const { return RHS.itr != itr;
}
+ };
+
+ iterator begin() const { return iterator(Root); }
+ iterator end() const { return iterator(); }
+
+ //===--------------------------------------------------===//
+ // Utility methods.
+ //===--------------------------------------------------===//
+
+ inline unsigned getHeight() const { return Root ? Root->getHeight() : 0; }
+
+ static inline void Profile(const ImmutableMap& M, FoldingSetNodeID& ID) {
+ ID.AddPointer(M.Root);
+ }
+
+ inline void Profile(FoldingSetNodeID& ID) const {
+ return Profile(*this,ID);
+ }
};
} // end namespace llvm
_______________________________________________
llvm-commits mailing list
llvm-commits@cs.uiuc.edu
http://lists.cs.uiuc.edu/mailman/listinfo/llvm-commits
