package set;

import java.util.NoSuchElementException;
import java.util.Iterator;

public class MaxSet<E extends Comparable<E>> extends ArraySet<E> {
    private E maxElement = null;

    /**
     * Constructs a new empty set.
     */
    public MaxSet() {
        super();

    }

    /**
     * Returns the currently largest element in this set. pre: the set is not
     * empty post: the set is unchanged
     *
     * @return the currently largest element in this set
     * @throws NoSuchElementException
     *             if this set is empty
     */
    public E getMax() {
        if (!super.isEmpty()) {
            return maxElement;
        } else {
            throw new NoSuchElementException("set empty");
        }
        // System.out.println(maxElement);
    }

    /**
     * Adds the specified element to this set, if it is not already present.
     * post: x is added to the set if it is not already present
     *
     * @param x
     *            the element to be added
     * @return true if the specified element was added
     */
    public boolean add(E x) {
        if (maxElement == null) {
            maxElement = x;
        } else {
            if (x.compareTo(maxElement) > 0) {
                maxElement = x;
            }
        }
        return super.add(x);
    }

    /**
     * post: x is removed if it was present
     *
     * @param x
     *            the element to remove - if present
     * @return true if the set contained the specified element
     */
    public boolean remove(Object x) { // varför object
        if (super.isEmpty()) {
            return false;
        }

        if (super.remove(x)) {

            if (maxElement.equals(x))
                findMax();
            return true;
        }

        return false;

    }

    private void findMax() {
        // hittar nya maxElement

        maxElement = null;
        Iterator<? extends E> itr = super.iterator();
        for (int i = 0; i < super.size(); i++) {
            if (itr.hasNext()) {
                E temp = itr.next();
                if (maxElement == null) {
                    maxElement = temp;
                } else if (temp.compareTo(maxElement) > 0) {
                    maxElement = temp;
                }
            }
        }
    }

    /**
     * Adds all of the elements in the specified set, for which it is possible,
     * to this set. post: all elements, for which it is possible, in the
     * specified set are added to this set.
     *
     * @return true if this set changed as a result of the call
     */
    public boolean addAll(SimpleSet<? extends E> c) {
        return super.addAll(c);
    }
}