Applied-Energistics-2/src/main/java/appeng/util/item/ItemList.java

/*
 * This file is part of Applied Energistics 2.
 * Copyright (c) 2013 - 2020, 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.util.item;

import java.util.Collection;
import java.util.Collections;
import java.util.IdentityHashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.NoSuchElementException;

import net.minecraft.item.Item;

import appeng.api.config.FuzzyMode;
import appeng.api.storage.data.IAEItemStack;
import appeng.api.storage.data.IItemList;

public final class ItemList implements IItemList<IAEItemStack> {

    private final static IItemList<IAEItemStack> NULL_ITEMLIST = new NullItemList();

    private final Map<Item, IItemList<IAEItemStack>> records = new IdentityHashMap<>();

    @Override
    public IAEItemStack findPrecise(final IAEItemStack itemStack) {
        if (itemStack == null) {
            return null;
        }

        return this.getRecord(itemStack.getItem()).findPrecise(itemStack);
    }

    @Override
    public Collection<IAEItemStack> findFuzzy(final IAEItemStack filter, final FuzzyMode fuzzy) {
        if (filter == null) {
            return Collections.emptyList();
        }

        return this.getRecord(filter.getItem()).findFuzzy(filter, fuzzy);
    }

    @Override
    public boolean isEmpty() {
        return !this.iterator().hasNext();
    }

    @Override
    public void add(final IAEItemStack itemStack) {
        if (itemStack == null) {
            return;
        }

        this.getOrCreateRecord(itemStack.getItem()).add(itemStack);
    }

    @Override
    public void addStorage(final IAEItemStack itemStack) {
        if (itemStack == null) {
            return;
        }

        this.getOrCreateRecord(itemStack.getItem()).addStorage(itemStack);
    }

    @Override
    public void addCrafting(final IAEItemStack itemStack) {
        if (itemStack == null) {
            return;
        }

        this.getOrCreateRecord(itemStack.getItem()).addCrafting(itemStack);
    }

    @Override
    public void addRequestable(final IAEItemStack itemStack) {
        if (itemStack == null) {
            return;
        }

        this.getOrCreateRecord(itemStack.getItem()).addRequestable(itemStack);
    }

    @Override
    public IAEItemStack getFirstItem() {
        for (final IAEItemStack stackType : this) {
            return stackType;
        }

        return null;
    }

    @Override
    public int size() {
        int size = 0;
        for (IItemList<IAEItemStack> entry : records.values()) {
            size += entry.size();
        }

        return size;
    }

    @Override
    public Iterator<IAEItemStack> iterator() {
        return new ChainedIterator(this.records.values().iterator());
    }

    @Override
    public void resetStatus() {
        for (final IAEItemStack i : this) {
            i.reset();
        }
    }

    private IItemList<IAEItemStack> getRecord(Item item) {
        return this.records.getOrDefault(item, NULL_ITEMLIST);
    }

    private IItemList<IAEItemStack> getOrCreateRecord(Item item) {
        return this.records.computeIfAbsent(item, this::makeRecordMap);
    }

    private IItemList<IAEItemStack> makeRecordMap(Item item) {
        if (item.isDamageable()) {
            return new FuzzyItemList();
        } else {
            return new StrictItemList();
        }
    }

    private class ChainedIterator implements Iterator<IAEItemStack> {

        private final Iterator<IItemList<IAEItemStack>> parent;
        private Iterator<IAEItemStack> next;

        public ChainedIterator(Iterator<IItemList<IAEItemStack>> iterator) {
            this.parent = iterator;
            if (this.parent.hasNext()) {
                this.next = this.parent.next().iterator();
            }
        }

        @Override
        public boolean hasNext() {
            while (this.next != null) {
                if (this.next.hasNext()) {
                    return true;
                }

                if (this.parent.hasNext()) {
                    this.next = this.parent.next().iterator();
                } else {
                    this.next = null;
                }
            }

            return false;
        }

        @Override
        public IAEItemStack next() {
            if (this.next == null) {
                throw new NoSuchElementException();
            }

            return this.next.next();
        }

    }
}