Moving to source sets

src/main/java/appeng/util/item/ItemList.java

@@ -0,0 +1,185 @@
+/*
+ * 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();
+        }
+
+    }
+}