mirror of
https://github.com/Ellpeck/PrettyPipes.git
synced 2024-06-10 06:40:25 +02:00
174 lines
6.6 KiB
Java
174 lines
6.6 KiB
Java
package de.ellpeck.prettypipes.misc;
|
|
|
|
import com.mojang.blaze3d.matrix.MatrixStack;
|
|
import de.ellpeck.prettypipes.PrettyPipes;
|
|
import de.ellpeck.prettypipes.packets.PacketButton;
|
|
import de.ellpeck.prettypipes.pipe.PipeTileEntity;
|
|
import de.ellpeck.prettypipes.pipe.modules.FilterModifierModuleItem;
|
|
import de.ellpeck.prettypipes.pipe.modules.filter.FilterIncreaseModuleItem;
|
|
import net.minecraft.client.gui.screen.Screen;
|
|
import net.minecraft.client.gui.widget.Widget;
|
|
import net.minecraft.client.gui.widget.button.Button;
|
|
import net.minecraft.inventory.container.Slot;
|
|
import net.minecraft.item.ItemStack;
|
|
import net.minecraft.nbt.CompoundNBT;
|
|
import net.minecraft.util.Direction;
|
|
import net.minecraft.util.text.TextFormatting;
|
|
import net.minecraft.util.text.TranslationTextComponent;
|
|
import net.minecraftforge.api.distmarker.Dist;
|
|
import net.minecraftforge.api.distmarker.OnlyIn;
|
|
import net.minecraftforge.items.IItemHandler;
|
|
import net.minecraftforge.items.ItemHandlerHelper;
|
|
import net.minecraftforge.items.ItemStackHandler;
|
|
import org.apache.commons.lang3.tuple.Pair;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.function.Supplier;
|
|
import java.util.stream.Collectors;
|
|
|
|
public class ItemFilter extends ItemStackHandler {
|
|
|
|
private final ItemStack stack;
|
|
private final PipeTileEntity pipe;
|
|
public boolean isWhitelist;
|
|
|
|
public boolean canPopulateFromInventories;
|
|
public boolean canModifyWhitelist = true;
|
|
private boolean modified;
|
|
|
|
public ItemFilter(int size, ItemStack stack, PipeTileEntity pipe) {
|
|
super(size);
|
|
this.stack = stack;
|
|
this.pipe = pipe;
|
|
if (stack.hasTag())
|
|
this.deserializeNBT(stack.getTag().getCompound("filter"));
|
|
}
|
|
|
|
public List<Slot> getSlots(int x, int y) {
|
|
List<Slot> slots = new ArrayList<>();
|
|
for (int i = 0; i < this.getSlots(); i++)
|
|
slots.add(new FilterSlot(this, i, x + i % 9 * 18, y + i / 9 * 18, true));
|
|
return slots;
|
|
}
|
|
|
|
@OnlyIn(Dist.CLIENT)
|
|
public List<Widget> getButtons(Screen gui, int x, int y) {
|
|
List<Widget> buttons = new ArrayList<>();
|
|
if (this.canModifyWhitelist) {
|
|
Supplier<TranslationTextComponent> whitelistText = () -> new TranslationTextComponent("info." + PrettyPipes.ID + "." + (this.isWhitelist ? "whitelist" : "blacklist"));
|
|
buttons.add(new Button(x, y, 70, 20, whitelistText.get(), button -> {
|
|
PacketButton.sendAndExecute(this.pipe.getPos(), PacketButton.ButtonResult.FILTER_CHANGE, 0);
|
|
button.setMessage(whitelistText.get());
|
|
}));
|
|
}
|
|
if (this.canPopulateFromInventories) {
|
|
buttons.add(new Button(x + 72, y, 70, 20, new TranslationTextComponent("info." + PrettyPipes.ID + ".populate"), button -> PacketButton.sendAndExecute(this.pipe.getPos(), PacketButton.ButtonResult.FILTER_CHANGE, 1)) {
|
|
@Override
|
|
public void renderToolTip(MatrixStack matrix, int x, int y) {
|
|
gui.renderTooltip(matrix, new TranslationTextComponent("info." + PrettyPipes.ID + ".populate.description").mergeStyle(TextFormatting.GRAY), x, y);
|
|
}
|
|
});
|
|
}
|
|
return buttons;
|
|
}
|
|
|
|
public void onButtonPacket(int id) {
|
|
if (id == 0 && this.canModifyWhitelist) {
|
|
this.isWhitelist = !this.isWhitelist;
|
|
this.modified = true;
|
|
this.save();
|
|
} else if (id == 1 && this.canPopulateFromInventories) {
|
|
// populate filter from inventories
|
|
List<ItemFilter> filters = this.getAllFilters();
|
|
for (Direction direction : Direction.values()) {
|
|
IItemHandler handler = this.pipe.getItemHandler(direction, null);
|
|
if (handler == null)
|
|
continue;
|
|
for (int i = 0; i < handler.getSlots(); i++) {
|
|
ItemStack stack = handler.getStackInSlot(i);
|
|
if (stack.isEmpty() || this.isFiltered(stack))
|
|
continue;
|
|
ItemStack copy = stack.copy();
|
|
copy.setCount(1);
|
|
// try inserting into ourselves and any filter increase modifiers
|
|
for (ItemFilter filter : filters) {
|
|
if (ItemHandlerHelper.insertItem(filter, copy, false).isEmpty()) {
|
|
filter.save();
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public boolean isAllowed(ItemStack stack) {
|
|
return this.isFiltered(stack) == this.isWhitelist;
|
|
}
|
|
|
|
private boolean isFiltered(ItemStack stack) {
|
|
ItemEqualityType[] types = this.getEqualityTypes();
|
|
// also check if any filter increase modules have the item we need
|
|
for (ItemStackHandler handler : this.getAllFilters()) {
|
|
for (int i = 0; i < handler.getSlots(); i++) {
|
|
ItemStack filter = handler.getStackInSlot(i);
|
|
if (filter.isEmpty())
|
|
continue;
|
|
if (ItemEqualityType.compareItems(stack, filter, types))
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private List<ItemFilter> getAllFilters() {
|
|
List<ItemFilter> filters = this.pipe.streamModules()
|
|
.filter(p -> p.getRight() instanceof FilterIncreaseModuleItem)
|
|
.map(p -> new ItemFilter(18, p.getLeft(), this.pipe))
|
|
.collect(Collectors.toList());
|
|
// add ourselves to the front
|
|
filters.add(0, this);
|
|
return filters;
|
|
}
|
|
|
|
public ItemEqualityType[] getEqualityTypes() {
|
|
return this.pipe.streamModules()
|
|
.map(Pair::getRight)
|
|
.filter(m -> m instanceof FilterModifierModuleItem)
|
|
.map(m -> ((FilterModifierModuleItem) m).type)
|
|
.toArray(ItemEqualityType[]::new);
|
|
}
|
|
|
|
public void save() {
|
|
if (this.modified) {
|
|
this.stack.getOrCreateTag().put("filter", this.serializeNBT());
|
|
this.modified = false;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public CompoundNBT serializeNBT() {
|
|
CompoundNBT nbt = super.serializeNBT();
|
|
if (this.canModifyWhitelist)
|
|
nbt.putBoolean("whitelist", this.isWhitelist);
|
|
return nbt;
|
|
}
|
|
|
|
@Override
|
|
public void deserializeNBT(CompoundNBT nbt) {
|
|
super.deserializeNBT(nbt);
|
|
if (this.canModifyWhitelist)
|
|
this.isWhitelist = nbt.getBoolean("whitelist");
|
|
}
|
|
|
|
@Override
|
|
protected void onContentsChanged(int slot) {
|
|
this.modified = true;
|
|
}
|
|
|
|
public interface IFilteredContainer {
|
|
ItemFilter getFilter();
|
|
}
|
|
}
|