equivalent-exchange-3/src/main/java/com/pahimar/ee3/util/FilterUtils.java

package com.pahimar.ee3.util;

import com.pahimar.ee3.api.EnergyValue;
import com.pahimar.ee3.api.EnergyValueRegistryProxy;
import net.minecraft.item.ItemBlock;
import net.minecraft.item.ItemStack;

import java.util.*;

public class FilterUtils
{
    public static Set<ItemStack> filterForItemBlock(Set<ItemStack> unfilteredItemStackSet)
    {
        return filterForItemBlock(unfilteredItemStackSet, ItemHelper.idComparator);
    }

    public static Set<ItemStack> filterForItemBlock(Set<ItemStack> unfilteredItemStackSet, Comparator comparator)
    {
        Set<ItemStack> itemBlockOnlySet = new TreeSet<ItemStack>(comparator);

        for (ItemStack itemStack : unfilteredItemStackSet)
        {
            if (itemStack.getItem() instanceof ItemBlock)
            {
                itemBlockOnlySet.add(itemStack);
            }
        }

        return itemBlockOnlySet;
    }

    public static Set<ItemStack> filterByNameStartsWith(Set<ItemStack> unfilteredItemStackSet, String filterString)
    {
        return filterByNameStartsWith(unfilteredItemStackSet, filterString, ItemHelper.idComparator);
    }

    public static Set<ItemStack> filterByNameStartsWith(Set<ItemStack> unfilteredItemStackSet, String filterString, Comparator comparator)
    {
        Set<ItemStack> nameSortedSet = new TreeSet<ItemStack>(comparator);

        for (ItemStack itemStack : unfilteredItemStackSet)
        {
            String itemDisplayName = itemStack.getDisplayName().toLowerCase();

            if (filterString != null)
            {
                if (itemDisplayName.startsWith(filterString.toLowerCase()))
                {
                    nameSortedSet.add(itemStack);
                }
            }
            else
            {
                nameSortedSet.add(itemStack);
            }
        }

        return nameSortedSet;
    }

    public static Set<ItemStack> filterByNameContains(Collection<ItemStack> unfilteredCollection, String filterString)
    {
        Set<ItemStack> nameSortedSet = new HashSet<ItemStack>();

        for (ItemStack itemStack : unfilteredCollection)
        {
            String itemDisplayName = itemStack.getDisplayName().toLowerCase();

            if (filterString != null)
            {
                if (itemDisplayName.contains(filterString.toLowerCase()))
                {
                    nameSortedSet.add(itemStack);
                }
            }
            else
            {
                nameSortedSet.add(itemStack);
            }
        }

        return nameSortedSet;
    }

    public static Set<ItemStack> filterByNameContains(Collection<ItemStack> unfilteredCollection, String filterString, Comparator comparator)
    {
        Set<ItemStack> nameSortedSet = new TreeSet<ItemStack>(comparator);

        for (ItemStack itemStack : unfilteredCollection)
        {
            String itemDisplayName = itemStack.getDisplayName().toLowerCase();

            if (filterString != null)
            {
                if (itemDisplayName.contains(filterString.toLowerCase()))
                {
                    nameSortedSet.add(itemStack);
                }
            }
            else
            {
                nameSortedSet.add(itemStack);
            }
        }

        return nameSortedSet;
    }

    public static Set<ItemStack> filterByEnergyValue(Collection<ItemStack> unfilteredCollection, EnergyValue energyValue)
    {
        return filterByEnergyValue(unfilteredCollection, energyValue.getValue());
    }

    public static Set<ItemStack> filterByEnergyValue(Collection<ItemStack> unfilteredCollection, EnergyValue energyValue, Comparator<ItemStack> comparator)
    {
        return filterByEnergyValue(unfilteredCollection, energyValue.getValue(), comparator);
    }

    public static Set<ItemStack> filterByEnergyValue(Collection<ItemStack> unfilteredCollection, float energyValue)
    {
        Set<ItemStack> sortedSet = new HashSet<ItemStack>();

        for (ItemStack itemStack : unfilteredCollection)
        {
            if (energyValue > 0f && EnergyValueRegistryProxy.hasEnergyValue(itemStack))
            {
                if (EnergyValueRegistryProxy.getEnergyValue(itemStack).getValue() <= energyValue)
                {
                    sortedSet.add(itemStack);
                }
            }
        }

        return sortedSet;
    }

    public static Set<ItemStack> filterByEnergyValue(Collection<ItemStack> unfilteredCollection, float energyValue, Comparator<ItemStack> comparator)
    {
        Set<ItemStack> sortedSet = new TreeSet<ItemStack>(comparator);

        for (ItemStack itemStack : unfilteredCollection)
        {
            if (energyValue > 0f && EnergyValueRegistryProxy.hasEnergyValue(itemStack))
            {
                if (EnergyValueRegistryProxy.getEnergyValue(itemStack).getValue() <= energyValue)
                {
                    sortedSet.add(itemStack);
                }
            }
        }

        return sortedSet;
    }
}