Ticket #965: testcollections.d

import tango.group.collection;
import tango.io.Stdout;

void outputView(T)(View!(T) v)
{
  Stdout("[");
  foreach(x; v)
    Stdout(" ")(x);
  Stdout("]").newline;
}

void outputMap(K, V)(MapView!(K, V) m)
{
  Stdout("[");
  foreach(k, v; m)
    Stdout(" ")(k)(":")(v);
  Stdout("]").newline;
}

void removeEveryX(T)(Dispenser!(T) v, int X)
{
  int x = X;
  auto iter = v.dispensableElements;
  /*
  while(iter.more())
  {
    auto n = iter.get();
    if(--x == 0)
    {
      Stdout("removing element ")(n).newline;
      iter.remove();
      x = X;
    }
  }*/
  foreach(n; iter)
  {
    if(--x == 0)
    {
      Stdout("removing element ")(n).newline;
      iter.remove();
      x = X;
    }
  }
}

int main(char[][] args)
{
  int[] keys = [34789, 346,7, 3677,11, 85824, 986,1234,78,17890];
  int[] values = [1,2,3,4,5,6,7,8,9,10];

  void doView(Dispenser!(int) view)
  {
    outputView!(int)(view);
    removeEveryX!(int)(view, 3);
    outputView!(int)(view);
  }

  void doBag(char[] name, Bag!(int) bag)
  {
    Stdout(name ~ ":").newline;
    bag.add(new ArrayIterator!(int)(keys));
    doView(bag);
  }

  void doSeq(char[] name, Seq!(int) seq)
  {
    Stdout(name ~ ":").newline;
    seq.append(new ArrayIterator!(int)(keys));
    doView(seq);
  }

  void doSet(char[] name, Set!(int) set)
  {
    Stdout(name ~ ":").newline;
    set.add(new ArrayIterator!(int)(keys));
    doView(set);
  }

  void doMap(char[] name, Map!(int, int) map)
  {
    Stdout(name ~ ":").newline;
    foreach(i, v; keys)
      map[v] = values[i];
    outputMap!(int, int)(map);
    removeEveryX!(int)(map, 3);
    outputMap!(int, int)(map);
  }


  doBag("ArrayBag", new ArrayBag!(int));
  doBag("TreeBag", new TreeBag!(int));
  doSeq("ArraySeq", new ArraySeq!(int));
  doSeq("CircularSeq", new CircularSeq!(int));
  doSeq("LinkSeq", new LinkSeq!(int));
  doMap("HashMap", new HashMap!(int, int));
  doMap("LinkMap", new LinkMap!(int, int));
  doMap("TreeMap", new TreeMap!(int, int));
  doSet("HashSet", new HashSet!(int));
  return 0;
}