mirror of
https://github.com/Ellpeck/PrettyPipes.git
synced 2024-05-27 02:03:37 +02:00
95 lines
3.7 KiB
Java
95 lines
3.7 KiB
Java
package de.ellpeck.prettypipes.pipe.modules.extraction;
|
|
|
|
import de.ellpeck.prettypipes.Registry;
|
|
import de.ellpeck.prettypipes.items.IModule;
|
|
import de.ellpeck.prettypipes.items.ModuleItem;
|
|
import de.ellpeck.prettypipes.items.ModuleTier;
|
|
import de.ellpeck.prettypipes.misc.DirectionSelector;
|
|
import de.ellpeck.prettypipes.misc.ItemFilter;
|
|
import de.ellpeck.prettypipes.network.PipeNetwork;
|
|
import de.ellpeck.prettypipes.pipe.PipeBlockEntity;
|
|
import de.ellpeck.prettypipes.pipe.containers.AbstractPipeContainer;
|
|
import net.minecraft.core.Direction;
|
|
import net.minecraft.world.entity.player.Inventory;
|
|
import net.minecraft.world.entity.player.Player;
|
|
import net.minecraft.world.item.ItemStack;
|
|
import net.neoforged.neoforge.items.IItemHandler;
|
|
|
|
public class ExtractionModuleItem extends ModuleItem {
|
|
|
|
private final int maxExtraction;
|
|
private final int speed;
|
|
private final boolean preventOversending;
|
|
public final int filterSlots;
|
|
|
|
public ExtractionModuleItem(String name, ModuleTier tier) {
|
|
super(name);
|
|
this.maxExtraction = tier.forTier(1, 8, 64);
|
|
this.speed = tier.forTier(20, 15, 10);
|
|
this.filterSlots = tier.forTier(3, 6, 9);
|
|
this.preventOversending = tier.forTier(false, true, true);
|
|
}
|
|
|
|
@Override
|
|
public void tick(ItemStack module, PipeBlockEntity tile) {
|
|
if (!tile.shouldWorkNow(this.speed) || !tile.canWork())
|
|
return;
|
|
var filter = this.getItemFilter(module, tile);
|
|
var dirSelector = this.getDirectionSelector(module, tile);
|
|
for (var dir : dirSelector.directions()) {
|
|
var handler = tile.getItemHandler(dir);
|
|
if (handler == null)
|
|
continue;
|
|
var network = PipeNetwork.get(tile.getLevel());
|
|
for (var j = 0; j < handler.getSlots(); j++) {
|
|
var stack = handler.extractItem(j, this.maxExtraction, true);
|
|
if (stack.isEmpty())
|
|
continue;
|
|
if (!filter.isAllowed(stack, dir))
|
|
continue;
|
|
var remain = network.routeItem(tile.getBlockPos(), tile.getBlockPos().relative(dir), stack, this.preventOversending);
|
|
if (remain.getCount() != stack.getCount()) {
|
|
handler.extractItem(j, stack.getCount() - remain.getCount(), false);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public boolean canNetworkSee(ItemStack module, PipeBlockEntity tile, Direction direction, IItemHandler handler) {
|
|
return !this.getDirectionSelector(module, tile).has(direction);
|
|
}
|
|
|
|
@Override
|
|
public boolean canAcceptItem(ItemStack module, PipeBlockEntity tile, ItemStack stack, Direction direction, IItemHandler destination) {
|
|
return !this.getDirectionSelector(module, tile).has(direction);
|
|
}
|
|
|
|
@Override
|
|
public boolean isCompatible(ItemStack module, PipeBlockEntity tile, IModule other) {
|
|
return !(other instanceof ExtractionModuleItem);
|
|
}
|
|
|
|
@Override
|
|
public boolean hasContainer(ItemStack module, PipeBlockEntity tile) {
|
|
return true;
|
|
}
|
|
|
|
@Override
|
|
public AbstractPipeContainer<?> getContainer(ItemStack module, PipeBlockEntity tile, int windowId, Inventory inv, Player player, int moduleIndex) {
|
|
return new ExtractionModuleContainer(Registry.extractionModuleContainer, windowId, player, tile.getBlockPos(), moduleIndex);
|
|
}
|
|
|
|
@Override
|
|
public ItemFilter getItemFilter(ItemStack module, PipeBlockEntity tile) {
|
|
return new ItemFilter(this.filterSlots, module, tile);
|
|
}
|
|
|
|
@Override
|
|
public DirectionSelector getDirectionSelector(ItemStack module, PipeBlockEntity tile) {
|
|
return new DirectionSelector(module, tile);
|
|
}
|
|
|
|
}
|