#include <iter_map.h>

 #include <maps.h>

 #include <iostream>

 using namespace hugo;

 using namespace std;

 const int N = 3;

 typedef StdMap<int,int> BaseMap;

 typedef IterableMap<BaseMap, N> TestMap;

 void print(TestMap const& m) {

   cout << "Size of the map: " << m.size() << endl;

   for(int i=0; i<N; ++i) {

     cout << "  Class " << i << ". (size=" << m.size(i) << "): " << flush;

     cout << "    ";

     for(TestMap::iterator j = m.begin(i); j!=m.end(i); ++j) {

       cout << " " << *j;

     }

     cout << endl;

   }

 }

 int main() {

   BaseMap base(344);

   TestMap test(base);

   print(test);

   cout << "Inserting 12 to class 2...\n";

   test.insert(12,2);

   print(test);

   cout << "Inserting 22 to class 2...\n";

   test.insert(22,2);

   print(test);

   cout << "Testing some map values:\n";

   cout << " 12: " << int(test[12]) << endl;

   cout << "Inserting 10 to class 0...\n";

   test.insert(10,0);

   print(test);

   cout << "Testing some map values:\n";

   cout << " 12: " << int(test[12]) << endl;

   cout << "Inserting 11 to class 1...\n";

   test.insert(11,1);

   print(test);

   cout << "Testing some map values:\n";

   cout << " 12: " << int(test[12]) << endl;

   cout << " 22: " << int(test[22]) << endl;

   cout << " 10: " << int(test[10]) << endl;

   cout << " 11: " << int(test[11]) << endl;

   cout << " 42: " << int(test[42]) << endl;

   cout << "Inserting 21 to class 1...\n";

   test.insert(21,1);

   print(test);

   cout << "Inserting 20 to class 1...\n";

   test.insert(20,0);

   print(test);

   cout << "Testing some map values:\n";

   cout << " 12: " << int(test[12]) << endl;

   cout << " 22: " << int(test[22]) << endl;

   cout << " 10: " << int(test[10]) << endl;

   cout << " 20: " << int(test[20]) << endl;

   cout << " 11: " << int(test[11]) << endl;

   cout << " 21: " << int(test[21]) << endl;

   cout << " 42: " << int(test[42]) << endl;

   cout << "Setting 20 to class 2...\n";

   test.set(20,2);

   print(test);

   cout << "Setting 10 to class 1...\n";

   test.set(10,1);

   print(test);

   cout << "Setting 11 to class 1...\n";

   test.set(11,1);

   print(test);

   cout << "Setting 12 to class 1...\n";

   test.set(12,1);

   print(test);

   cout << "Setting 21 to class 2...\n";

   test.set(21,2);

   print(test);

   cout << "Setting 22 to class 2...\n";

   test.set(22,2);

   print(test);

   cout << "Testing some map values:\n";

   cout << " 12: " << int(test[12]) << endl;

   cout << " 22: " << int(test[22]) << endl;

   cout << " 10: " << int(test[10]) << endl;

   cout << " 20: " << int(test[20]) << endl;

   cout << " 11: " << int(test[11]) << endl;

   cout << " 21: " << int(test[21]) << endl;

   cout << " 42: " << int(test[42]) << endl;

 }