[llvm-commits] [llvm] r45503 - /llvm/trunk/include/llvm/ADT/ImmutableMap.h

[Ted Kremenek]

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