Applied-Energistics-2-tilera-Edition/src/main/java/appeng/me/storage/ItemWatcher.java

/*
 * This file is part of Applied Energistics 2.
 * Copyright (c) 2013 - 2014, AlgorithmX2, All rights reserved.
 *
 * Applied Energistics 2 is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Applied Energistics 2 is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with Applied Energistics 2.  If not, see <http://www.gnu.org/licenses/lgpl>.
 */

package appeng.me.storage;

import java.util.Collection;
import java.util.HashSet;
import java.util.Iterator;

import appeng.api.networking.storage.IStackWatcher;
import appeng.api.networking.storage.IStackWatcherHost;
import appeng.api.storage.data.IAEStack;
import appeng.me.cache.GridStorageCache;

/**
 * Maintain my interests, and a global watch list, they should always be fully
 * synchronized.
 */
public class ItemWatcher implements IStackWatcher {
    private final GridStorageCache gsc;
    private final IStackWatcherHost myObject;
    private final HashSet<IAEStack> myInterests = new HashSet<IAEStack>();

    public ItemWatcher(final GridStorageCache cache, final IStackWatcherHost host) {
        this.gsc = cache;
        this.myObject = host;
    }

    public IStackWatcherHost getHost() {
        return this.myObject;
    }

    @Override
    public int size() {
        return this.myInterests.size();
    }

    @Override
    public boolean isEmpty() {
        return this.myInterests.isEmpty();
    }

    @Override
    public boolean contains(final Object o) {
        return this.myInterests.contains(o);
    }

    @Override
    public Iterator<IAEStack> iterator() {
        return new ItemWatcherIterator(this, this.myInterests.iterator());
    }

    @Override
    public Object[] toArray() {
        return this.myInterests.toArray();
    }

    @Override
    public <T> T[] toArray(final T[] a) {
        return this.myInterests.toArray(a);
    }

    @Override
    public boolean add(final IAEStack e) {
        if (this.myInterests.contains(e)) {
            return false;
        }

        return this.myInterests.add(e.copy())
            && this.gsc.getInterestManager().put(e, this);
    }

    @Override
    public boolean remove(final Object o) {
        return this.myInterests.remove(o)
            && this.gsc.getInterestManager().remove((IAEStack) o, this);
    }

    @Override
    public boolean containsAll(final Collection<?> c) {
        return this.myInterests.containsAll(c);
    }

    @Override
    public boolean addAll(final Collection<? extends IAEStack> c) {
        boolean didChange = false;

        for (final IAEStack o : c) {
            didChange = this.add(o) || didChange;
        }

        return didChange;
    }

    @Override
    public boolean removeAll(final Collection<?> c) {
        boolean didSomething = false;
        for (final Object o : c) {
            didSomething = this.remove(o) || didSomething;
        }
        return didSomething;
    }

    @Override
    public boolean retainAll(final Collection<?> c) {
        boolean changed = false;
        final Iterator<IAEStack> i = this.iterator();

        while (i.hasNext()) {
            if (!c.contains(i.next())) {
                i.remove();
                changed = true;
            }
        }

        return changed;
    }

    @Override
    public void clear() {
        final Iterator<IAEStack> i = this.myInterests.iterator();
        while (i.hasNext()) {
            this.gsc.getInterestManager().remove(i.next(), this);
            i.remove();
        }
    }

    private class ItemWatcherIterator implements Iterator<IAEStack> {
        private final ItemWatcher watcher;
        private final Iterator<IAEStack> interestIterator;
        private IAEStack myLast;

        public ItemWatcherIterator(final ItemWatcher parent, final Iterator<IAEStack> i) {
            this.watcher = parent;
            this.interestIterator = i;
        }

        @Override
        public boolean hasNext() {
            return this.interestIterator.hasNext();
        }

        @Override
        public IAEStack next() {
            return this.myLast = this.interestIterator.next();
        }

        @Override
        public void remove() {
            ItemWatcher.this.gsc.getInterestManager().remove(this.myLast, this.watcher);
            this.interestIterator.remove();
        }
    }
}