// -*- c++ -*- //

 #ifndef HUGO_ITER_MAP

 #define HUGO_ITER_MAP

 #include <vector>

 #include <algorithm>

 // for uint8_t

 #include <stdint.h>

 // for memset

 #include <cstring>

 #include <invalid.h>

 namespace hugo {

   /// \todo Decide whether we need all the range checkings!!!

   template<typename KeyIntMap, uint8_t N, typename Val = uint8_t>

   class IterableMap {

   public:

     typedef typename KeyIntMap::KeyType KeyType;

     typedef Val ValueType;

     typedef typename std::vector<KeyType>::const_iterator iterator;

   protected:

     KeyIntMap &base;

     std::vector<KeyType> data;

     size_t bounds[N];

     Val def_val;

     Val find(size_t a) const {

       for(uint8_t n=0; n<N; ++n) {

 	if(bounds[n] > a)

 	  return n;

       }

       return def_val;

     }

     void half_swap(size_t &a, size_t b) {

       if(a != b) {

 	base.set(data[b],a);

 	data[a] = data[b];

 	a = b;

       }

     }

     size_t move(size_t a, uint8_t m, uint8_t n) {

       if(m != n) {

 	size_t orig_a = a;

 	KeyType orig_key = data[a];

 	while(m > n) {

 	  --m;

 	  half_swap(a, bounds[m]++);

 	}

 	// FIXME: range check ide?

 	while(m < n) {

 	  half_swap(a, --bounds[m]);

 	  ++m;

 	}

 	if(a != orig_a) {

 	  base.set(orig_key, a);

 	  data[a]=orig_key;

 	}

       }

       return a;

     }

   public:

     IterableMap(KeyIntMap &_base, Val d = N+1) : base(_base), def_val(d) {

       memset(bounds, 0, sizeof(bounds));

       //    for(int i=0; i<N; ++i) { bounds[i]=0; }

     }

     Val operator[](const KeyType& k) const {

       return find(base[k]);

     }

     void set(const KeyType& k, Val n) {

       // FIXME: range check?

       size_t a = base[k];

       if(a < bounds[N-1]) {

 	base.set(k, move(a, find(a), n));

       }

       else {

 	insert(k, n);

       }

     }

     void insert(const KeyType& k, Val n) {

       data.push_back(k);

       base.set(k, move(bounds[N-1]++, N-1, n));

     }

     iterator begin(Val n) const {

       return data.begin() + (n ? bounds[n-1] : 0);

     }

     iterator end(Val n) const {

       return data.begin() + bounds[n];

     }

     size_t size(Val n) const {

       return bounds[n] - (n ? bounds[n-1] : 0);

     }

     size_t size() const {

       // assert(bounds[N-1] == data.size());

       return bounds[N-1];

     }

     /// For use as an iterator...

     KeyType& first(KeyType &k, Val n) {

       size_t i = (n ? bounds[n-1] : 0);

       if( i < bounds[n] ) {

 	k = data[i];

       }

       else {

 	k = INVALID;

       }

       return k;

     }

     /// For use as an iterator...

     KeyType& next(KeyType &k) {

       size_t i = base[k];

       uint8_t n = find(i);

       ++i;

       if( i < bounds[n] ) {

 	k = data[i];

       }

       else {

 	k = INVALID;

       }

       return k;

     }

   };

   template<typename KeyIntMap>

   class IterableBoolMap : public IterableMap<KeyIntMap, 2, bool> {

     typedef IterableMap<KeyIntMap, 2, bool> Parent;

   public:

     IterableBoolMap(KeyIntMap &_base, bool d = false) : Parent(_base, d) {}

   };

 }

 #endif