NotEnoughItems/src/main/java/codechicken/nei/ItemStackSet.java

package codechicken.nei;

import net.minecraft.block.Block;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;

import codechicken.nei.api.ItemFilter;

public class ItemStackSet extends ItemStackMap<ItemStack> implements ItemFilter {

    public ItemStackSet with(ItemStack... items) {
        for (ItemStack item : items) add(item);
        return this;
    }

    public ItemStackSet with(Item... items) {
        for (Item item : items) add(ItemStackMap.wildcard(item));
        return this;
    }

    public ItemStackSet with(Block... blocks) {
        for (Block block : blocks) add(ItemStackMap.wildcard(Item.getItemFromBlock(block)));
        return this;
    }

    public void add(ItemStack item) {
        put(item, item);
    }

    public ItemStackSet addAll(Iterable<ItemStack> items) {
        for (ItemStack item : items) add(item);
        return this;
    }

    public ItemStackSet removeAll(Iterable<ItemStack> items) {
        for (ItemStack item : items) remove(item);
        return this;
    }

    public boolean contains(ItemStack item) {
        return get(item) != null;
    }

    public boolean containsAll(Item item) {
        return get(ItemStackMap.wildcard(item)) != null;
    }

    @Override
    public boolean matches(ItemStack item) {
        return contains(item);
    }

    public static ItemStackSet of(Block... blocks) {
        return new ItemStackSet().with(blocks);
    }

    public static ItemStackSet of(Item... items) {
        return new ItemStackSet().with(items);
    }

    public static ItemStackSet of(ItemStack... items) {
        return new ItemStackSet().with(items);
    }
}