package iteratorExamples;

import java.lang.Iterable;
import java.util.Iterator;
import java.util.NoSuchElementException;

/**
 * a correct implementation of an Iterator of an array,
 *  based upon
 * http://www.java-forums.org/new-java/2747-implementing-iterator.html
 */
public class MyArray2<AnotherClass> implements Iterable<AnotherClass>
{
	private AnotherClass[] collection; // collection.size() change to collection.length throughout
	
	public MyArray2(){
		// details added by HMW
		collection = (AnotherClass []) new Object[10];
	}
 
	//constructors and other stuff here
	// details added by HMW
	AnotherClass get (int i) {
		if (i < 0 || i >= collection.length)
			throw (new NoSuchElementException());
		return collection[i];
	}
		
	void put (int i, AnotherClass ac){
		if (i < 0 || i >= collection.length)
			throw (new NoSuchElementException());
				collection[i] = ac;
	}
		 
	// Iterator material starts here
	public Iterator<AnotherClass> iterator() {
		return new LocalIterator ();
	}
	
	private class LocalIterator implements Iterator<AnotherClass> {
		private int count;
		
		public LocalIterator () {
			count=0;
		}

		public boolean hasNext() {
			return count < collection.length;
		}
		 
		public AnotherClass next() {
			if(count == collection.length){
				throw new NoSuchElementException();
			}
			return (AnotherClass) collection[count++]; // ++ fix added according to Forum
		}	

		public void remove() {
			throw new UnsupportedOperationException();
		}
		
		// method added per Forum's discussion
		public void reset() {
			count = 0;
		}
		
	}
	public static void main (String args[]) {
		MyArray2<Integer> mc = new MyArray2<Integer>();
			
		int i;
			
		for (i = 0; i < 10; i++)
			mc.put(i, 2*i);
			
		// print out using Iterator
                System.out.println ("First output segment:");
		for (Integer item : mc) 
                    {
                        System.out.print("\t" + item);
                    }
		System.out.println();
		
		// print out using Iterator
                System.out.println ("Second output segment:");
		for (Integer item1 : mc) 
                    {   System.out.print("\tout " + item1);	
                        for (Integer item2 : mc) 
                            {   System.out.print("\tin  " + item2);	
                            }			
                        System.out.println();
                    }			
		System.out.println();

		// print a third time using Iterator
                System.out.println ("Third output segment:");
		Iterator <Integer> it = mc.iterator();
		while (it.hasNext())
			System.out.print("\t" + it.next());
                System.out.println();
	}

}