| 1 | #ifndef _CONSTRAINT_ITERATOR_H_ | 
| 2 | #define _CONSTRAINT_ITERATOR_H_ | 
| 3 | #include <list> | 
| 4 | #include <vector> | 
| 5 |  | 
| 6 | using namespace std; | 
| 7 | //abstract iterator class | 
| 8 | template<class Item> class IteratorBase{ | 
| 9 | public: | 
| 10 | virtual ~IteratorBase(){} | 
| 11 | virtual void first() = 0; | 
| 12 | virtual void next() = 0; | 
| 13 | virtual bool isEnd() = 0; | 
| 14 |  | 
| 15 | virtual Item* currentItem() =0; | 
| 16 |  | 
| 17 | protected: | 
| 18 | IteratorBase() {}; | 
| 19 | }; | 
| 20 |  | 
| 21 | template <class Item> class ConstraintIterator{ | 
| 22 | public: | 
| 23 | ConstraintIterator(list<vector<Item*> >& aContainer) : container(aContainer){} | 
| 24 |  | 
| 25 | virtual void first(){ | 
| 26 | listIter = container.begin(); | 
| 27 | //the construction of the container will guarantee every vector in list will contain at least one element | 
| 28 | //so we don't need to worry about the empty vector problem | 
| 29 | //while (listIter->empty()) | 
| 30 | //  ++listIter; | 
| 31 |  | 
| 32 | vectorIter = listIter->begin(); | 
| 33 | } | 
| 34 |  | 
| 35 | virtual void next(){ | 
| 36 | ++vectorIter; | 
| 37 | if (vectorIter == listIter->end()){ | 
| 38 |  | 
| 39 | //do{ | 
| 40 | //++listIter; | 
| 41 | //}while(listIter->empty()) | 
| 42 | ++listIter; | 
| 43 | if (listIter != container.end()) | 
| 44 | vectorIter = listIter->begin(); | 
| 45 | } | 
| 46 | } | 
| 47 |  | 
| 48 | virtual bool isEnd(){ | 
| 49 | return listIter == container.end(); | 
| 50 | } | 
| 51 |  | 
| 52 | Item* currentItem(){ | 
| 53 | return listIter ==container.end() ? NULL : *vectorIter; | 
| 54 | } | 
| 55 |  | 
| 56 | private: | 
| 57 | list<vector<Item*> >& container; | 
| 58 |  | 
| 59 | list<vector<Item*> >::iterator listIter; | 
| 60 | vector<Item*>::iterator vectorIter; | 
| 61 | }; | 
| 62 |  | 
| 63 | typedef ConstraintIterator<ConstraintPair> ConstraintPairIterator; | 
| 64 | typedef ConstraintIterator<ConstraintElement> ConstraintElementIterator; | 
| 65 |  | 
| 66 |  | 
| 67 | #endif //end ifndef _CONSTRAINT_ITERATOR_H_ |