Small code cleanup.

[Project_POO.git] / exo2 / ListExtension.java

import java.util.List;
import java.util.ListIterator;

public class ListExtension<E> {
    private List<E> list;

    public ListExtension() {
        setList(null);
    }

    public ListExtension(List<E> l) {
        setList(l);
    }

    /**
     * [setList description]
     * @param l [description]
     */
    public void setList(List<E> l) {
        list = l;
    }

    /**
     * [getList description]
     * @return [description]
     */
    public List<E> getList() {
        return list;
    }

    public boolean add(E e) {
        return list.add(e);
    }

    public int size() {
        return list.size();
    }

    /**
     * Should mimic the List<E> add(int index, E value) method
     * @param index [description]
     * @param value [description]
     * @return      [description]
     */
    public void addIter(int index, E value) {
        ListIterator<E> iter = list.listIterator();
        int i = 0;
        // go to the element at index + 1
        while (iter.hasNext() && i <= index) {
            iter.next();
            i++;
        }
        iter.add(value);
    }

    //FIXME: replace parameter by the list of objects to add
    public void addNelements(int index, int Nelements) {
        ListIterator<E> iter = list.listIterator();
        int i = 0;
        // go to the element at index + 1
        while (iter.hasNext() && i <= index) {
            iter.next();
            i++;
        }
        for (int j = 0; j < Nelements; j++) {
            iter.add((E)new Object());
        }
    }

}