fluid terminal

src/main/java/appeng/fluids/container/ContainerMEPortableFluidCell.java

@@ -0,0 +1,518 @@
+package appeng.fluids.container;
+
+import appeng.api.AEApi;
+import appeng.api.config.*;
+import appeng.api.implementations.guiobjects.IPortableCell;
+import appeng.api.networking.IGridHost;
+import appeng.api.networking.IGridNode;
+import appeng.api.networking.energy.IEnergySource;
+import appeng.api.networking.security.IActionHost;
+import appeng.api.networking.security.IActionSource;
+import appeng.api.networking.storage.IBaseMonitor;
+import appeng.api.storage.IMEMonitor;
+import appeng.api.storage.IMEMonitorHandlerReceiver;
+import appeng.api.storage.ITerminalHost;
+import appeng.api.storage.channels.IFluidStorageChannel;
+import appeng.api.storage.data.IAEFluidStack;
+import appeng.api.storage.data.IItemList;
+import appeng.api.util.AEPartLocation;
+import appeng.api.util.IConfigManager;
+import appeng.api.util.IConfigurableObject;
+import appeng.container.AEBaseContainer;
+import appeng.container.guisync.GuiSync;
+import appeng.container.interfaces.IInventorySlotAware;
+import appeng.container.slot.AppEngSlot;
+import appeng.container.slot.SlotPlayerHotBar;
+import appeng.container.slot.SlotPlayerInv;
+import appeng.core.AEConfig;
+import appeng.core.AELog;
+import appeng.core.localization.PlayerMessages;
+import appeng.core.sync.network.NetworkHandler;
+import appeng.core.sync.packets.PacketMEFluidInventoryUpdate;
+import appeng.core.sync.packets.PacketTargetFluidStack;
+import appeng.core.sync.packets.PacketValueConfig;
+import appeng.fluids.util.AEFluidStack;
+import appeng.helpers.InventoryAction;
+import appeng.helpers.WirelessTerminalGuiObject;
+import appeng.util.ConfigManager;
+import appeng.util.IConfigManagerHost;
+import appeng.util.Platform;
+import net.minecraft.entity.player.EntityPlayer;
+import net.minecraft.entity.player.EntityPlayerMP;
+import net.minecraft.entity.player.InventoryPlayer;
+import net.minecraft.inventory.IContainerListener;
+import net.minecraft.item.ItemStack;
+import net.minecraftforge.fluids.FluidStack;
+import net.minecraftforge.fluids.FluidUtil;
+import net.minecraftforge.fluids.capability.IFluidHandlerItem;
+
+import javax.annotation.Nonnull;
+import java.io.IOException;
+import java.nio.BufferOverflowException;
+
+public class ContainerMEPortableFluidCell extends AEBaseContainer implements IConfigManagerHost, IConfigurableObject, IMEMonitorHandlerReceiver<IAEFluidStack> {
+
+    private final WirelessTerminalGuiObject wirelessTerminalGUIObject;
+
+    private final IConfigManager clientCM;
+    private final IMEMonitor<IAEFluidStack> monitor;
+    private final IItemList<IAEFluidStack> fluids = AEApi.instance().storage().getStorageChannel(IFluidStorageChannel.class).createList();
+    @GuiSync(99)
+    public boolean hasPower = false;
+    private final ITerminalHost terminal;
+    private IConfigManager serverCM;
+    private IConfigManagerHost gui;
+    private IGridNode networkNode;
+    // Holds the fluid the client wishes to extract, or null for insert
+    private IAEFluidStack clientRequestedTargetFluid = null;
+    private double powerMultiplier = 0.5;
+
+    protected final IPortableCell civ;
+    private int ticks = 0;
+    private final int slot;
+
+    public ContainerMEPortableFluidCell(final InventoryPlayer ip, final IPortableCell monitorable) {
+        this(ip, monitorable, null, true);
+    }
+
+    public ContainerMEPortableFluidCell(final InventoryPlayer ip, final IPortableCell monitorable, WirelessTerminalGuiObject iGuiItemObject) {
+        this(ip, monitorable, iGuiItemObject, true);
+    }
+
+    public ContainerMEPortableFluidCell(InventoryPlayer ip, IPortableCell monitorable, WirelessTerminalGuiObject iGuiItemObject, boolean bindInventory) {
+        super(ip, monitorable);
+
+        this.terminal = monitorable;
+        this.wirelessTerminalGUIObject = (WirelessTerminalGuiObject) monitorable;
+
+        this.clientCM = new ConfigManager(this);
+
+        this.clientCM.registerSetting(Settings.SORT_BY, SortOrder.NAME);
+        this.clientCM.registerSetting(Settings.SORT_DIRECTION, SortDir.ASCENDING);
+        this.clientCM.registerSetting(Settings.VIEW_MODE, ViewItems.ALL);
+        if (Platform.isServer()) {
+            this.serverCM = terminal.getConfigManager();
+            this.monitor = terminal.getInventory(AEApi.instance().storage().getStorageChannel(IFluidStorageChannel.class));
+
+            if (this.monitor != null) {
+                this.monitor.addListener(this, null);
+
+                this.setPowerSource((IEnergySource) terminal);
+                if (terminal instanceof IGridHost || terminal instanceof IActionHost) {
+                    final IGridNode node;
+                    if (terminal instanceof IGridHost) {
+                        node = ((IGridHost) terminal).getGridNode(AEPartLocation.INTERNAL);
+                    } else {
+                        node = ((IActionHost) terminal).getActionableNode();
+                    }
+
+                    if (node != null) {
+                        this.networkNode = node;
+                    }
+                }
+            }
+        } else {
+            this.monitor = null;
+        }
+
+        if (monitorable instanceof IInventorySlotAware) {
+            final int slotIndex = ((IInventorySlotAware) monitorable).getInventorySlot();
+            this.lockPlayerInventorySlot(slotIndex);
+            this.slot = slotIndex;
+        } else {
+            this.slot = -1;
+            this.lockPlayerInventorySlot(ip.currentItem);
+        }
+        this.civ = monitorable;
+        if (bindInventory) {
+            this.bindPlayerInventory(ip, 0, 140);
+        }
+        hasPower = this.civ.extractAEPower(this.getPowerMultiplier(), Actionable.SIMULATE, PowerMultiplier.CONFIG) > 0.001;
+
+    }
+
+    @Override
+    public void detectAndSendChanges() {
+        final ItemStack currentItem = this.slot < 0 ? this.getPlayerInv().getCurrentItem() : this.getPlayerInv().getStackInSlot(this.slot);
+
+        if (this.civ == null || currentItem.isEmpty()) {
+            this.setValidContainer(false);
+        } else if (this.civ != null && !this.civ.getItemStack().isEmpty() && currentItem != this.civ.getItemStack()) {
+            if (ItemStack.areItemsEqual(this.civ.getItemStack(), currentItem)) {
+                this.getPlayerInv().setInventorySlotContents(this.getPlayerInv().currentItem, this.civ.getItemStack());
+            } else {
+                this.setValidContainer(false);
+            }
+        }
+
+        // drain 1 ae t
+        this.ticks++;
+        if (this.ticks > 10) {
+            double power = this.civ.extractAEPower(this.getPowerMultiplier() * this.ticks, Actionable.MODULATE, PowerMultiplier.CONFIG);
+            this.ticks = 0;
+            this.hasPower = power > 0.001;
+        }
+
+        if (!this.wirelessTerminalGUIObject.rangeCheck()) {
+            if (Platform.isServer() && this.isValidContainer()) {
+                this.getPlayerInv().player.sendMessage(PlayerMessages.OutOfRange.get());
+            }
+
+            this.setValidContainer(false);
+        } else {
+            this.setPowerMultiplier(AEConfig.instance().wireless_getDrainRate(this.wirelessTerminalGUIObject.getRange()));
+        }
+        if (Platform.isServer()) {
+            if (this.monitor != this.terminal.getInventory(AEApi.instance().storage().getStorageChannel(IFluidStorageChannel.class))) {
+                this.setValidContainer(false);
+            }
+
+            for (final Settings set : this.serverCM.getSettings()) {
+                final Enum<?> sideLocal = this.serverCM.getSetting(set);
+                final Enum<?> sideRemote = this.clientCM.getSetting(set);
+
+                if (sideLocal != sideRemote) {
+                    this.clientCM.putSetting(set, sideLocal);
+                    for (final IContainerListener crafter : this.listeners) {
+                        if (crafter instanceof EntityPlayerMP) {
+                            try {
+                                NetworkHandler.instance().sendTo(new PacketValueConfig(set.name(), sideLocal.name()), (EntityPlayerMP) crafter);
+                            } catch (final IOException e) {
+                                AELog.debug(e);
+                            }
+                        }
+                    }
+                }
+            }
+
+            if (!this.fluids.isEmpty()) {
+                try {
+                    final IItemList<IAEFluidStack> monitorCache = this.monitor.getStorageList();
+
+                    final PacketMEFluidInventoryUpdate piu = new PacketMEFluidInventoryUpdate();
+
+                    for (final IAEFluidStack is : this.fluids) {
+                        final IAEFluidStack send = monitorCache.findPrecise(is);
+                        if (send == null) {
+                            is.setStackSize(0);
+                            piu.appendFluid(is);
+                        } else {
+                            piu.appendFluid(send);
+                        }
+                    }
+
+                    if (!piu.isEmpty()) {
+                        this.fluids.resetStatus();
+
+                        for (final Object c : this.listeners) {
+                            if (c instanceof EntityPlayer) {
+                                NetworkHandler.instance().sendTo(piu, (EntityPlayerMP) c);
+                            }
+                        }
+                    }
+                } catch (final IOException e) {
+                    AELog.debug(e);
+                }
+            }
+            //this.updatePowerStatus();
+
+        }
+        super.detectAndSendChanges();
+    }
+
+    @Override
+    public ItemStack transferStackInSlot(final EntityPlayer p, final int idx) {
+        if (Platform.isClient()) {
+            return ItemStack.EMPTY;
+        }
+        EntityPlayerMP player = (EntityPlayerMP) p;
+        if (this.inventorySlots.get(idx) instanceof SlotPlayerInv || this.inventorySlots.get(idx) instanceof SlotPlayerHotBar) {
+            final AppEngSlot clickSlot = (AppEngSlot) this.inventorySlots.get(idx); // require AE SLots!
+            ItemStack itemStack = clickSlot.getStack();
+
+            ItemStack copy = itemStack.copy();
+            copy.setCount(1);
+            IFluidHandlerItem fh = FluidUtil.getFluidHandler(copy);
+            if (fh == null) {
+                // only fluid handlers items
+                return ItemStack.EMPTY;
+            }
+
+            int heldAmount = itemStack.getCount();
+            for (int i = 0; i < heldAmount; i++) {
+                copy = itemStack.copy();
+                copy.setCount(1);
+                fh = FluidUtil.getFluidHandler(copy);
+
+                final FluidStack extract = fh.drain(Integer.MAX_VALUE, false);
+                if (extract == null || extract.amount < 1) {
+                    return ItemStack.EMPTY;
+                }
+
+                // Check if we can push into the system
+                final IAEFluidStack notStorable = Platform.poweredInsert(this.getPowerSource(), this.monitor, AEFluidStack.fromFluidStack(extract), this.getActionSource(), Actionable.SIMULATE);
+
+                if (notStorable != null && notStorable.getStackSize() > 0) {
+                    final int toStore = (int) (extract.amount - notStorable.getStackSize());
+                    final FluidStack storable = fh.drain(toStore, false);
+
+                    if (storable == null || storable.amount == 0) {
+                        return ItemStack.EMPTY;
+                    } else {
+                        extract.amount = storable.amount;
+                    }
+                }
+
+                // Actually drain
+                final FluidStack drained = fh.drain(extract, true);
+                extract.amount = drained.amount;
+
+                final IAEFluidStack notInserted = Platform.poweredInsert(this.getPowerSource(), this.monitor, AEFluidStack.fromFluidStack(extract), this.getActionSource());
+
+                if (notInserted != null && notInserted.getStackSize() > 0) {
+                    IAEFluidStack spill = this.monitor.injectItems(notInserted, Actionable.MODULATE, this.getActionSource());
+                    if (spill != null && spill.getStackSize() > 0) {
+                        fh.fill(spill.getFluidStack(), true);
+                    }
+                }
+
+                if (notInserted == null || notInserted.getStackSize() == 0) {
+                    if (!player.inventory.addItemStackToInventory(fh.getContainer())) {
+                        player.dropItem(fh.getContainer(), false);
+                    }
+                    clickSlot.decrStackSize(1);
+                }
+            }
+            this.detectAndSendChanges();
+            return ItemStack.EMPTY;
+        }
+        return super.transferStackInSlot(p, idx);
+    }
+
+    @Override
+    public void doAction(EntityPlayerMP player, InventoryAction action, int slot, long id) {
+        if (action != InventoryAction.FILL_ITEM && action != InventoryAction.EMPTY_ITEM) {
+            super.doAction(player, action, slot, id);
+            return;
+        }
+
+        final ItemStack held = player.inventory.getItemStack();
+        ItemStack heldCopy = held.copy();
+        heldCopy.setCount(1);
+        IFluidHandlerItem fh = FluidUtil.getFluidHandler(heldCopy);
+        if (fh == null) {
+            // only fluid handlers items
+            return;
+        }
+
+        if (action == InventoryAction.FILL_ITEM && this.clientRequestedTargetFluid != null) {
+            final IAEFluidStack stack = this.clientRequestedTargetFluid.copy();
+
+            // Check how much we can store in the item
+            stack.setStackSize(Integer.MAX_VALUE);
+            int amountAllowed = fh.fill(stack.getFluidStack(), false);
+            int heldAmount = held.getCount();
+            for (int i = 0; i < heldAmount; i++) {
+                ItemStack copiedFluidContainer = held.copy();
+                copiedFluidContainer.setCount(1);
+                fh = FluidUtil.getFluidHandler(copiedFluidContainer);
+
+                // Check if we can pull out of the system
+                final IAEFluidStack canPull = Platform.poweredExtraction(this.getPowerSource(), this.monitor, stack.setStackSize(amountAllowed), this.getActionSource(), Actionable.SIMULATE);
+                if (canPull == null || canPull.getStackSize() < 1) {
+                    return;
+                }
+
+                // How much could fit into the container
+                final int canFill = fh.fill(canPull.getFluidStack(), false);
+                if (canFill == 0) {
+                    return;
+                }
+
+                // Now actually pull out of the system
+                final IAEFluidStack pulled = Platform.poweredExtraction(this.getPowerSource(), this.monitor, stack.setStackSize(canFill), this.getActionSource());
+                if (pulled == null || pulled.getStackSize() < 1) {
+                    // Something went wrong
+                    AELog.error("Unable to pull fluid out of the ME system even though the simulation said yes ");
+                    return;
+                }
+
+                // Actually fill
+                final int used = fh.fill(pulled.getFluidStack(), true);
+
+                if (used != canFill) {
+                    AELog.error("Fluid item [%s] reported a different possible amount than it actually accepted.", held.getDisplayName());
+                }
+
+                if (held.getCount() == 1) {
+                    player.inventory.setItemStack(fh.getContainer());
+                } else {
+                    player.inventory.getItemStack().shrink(1);
+                    if (!player.inventory.addItemStackToInventory(fh.getContainer())) {
+                        player.dropItem(fh.getContainer(), false);
+                    }
+                }
+            }
+            this.updateHeld(player);
+
+        } else if (action == InventoryAction.EMPTY_ITEM) {
+            int heldAmount = held.getCount();
+            for (int i = 0; i < heldAmount; i++) {
+                ItemStack copiedFluidContainer = held.copy();
+                copiedFluidContainer.setCount(1);
+                fh = FluidUtil.getFluidHandler(copiedFluidContainer);
+
+                // See how much we can drain from the item
+                final FluidStack extract = fh.drain(Integer.MAX_VALUE, false);
+                if (extract == null || extract.amount < 1) {
+                    return;
+                }
+
+                // Check if we can push into the system
+                final IAEFluidStack notStorable = Platform.poweredInsert(this.getPowerSource(), this.monitor, AEFluidStack.fromFluidStack(extract), this.getActionSource(), Actionable.SIMULATE);
+
+                if (notStorable != null && notStorable.getStackSize() > 0) {
+                    final int toStore = (int) (extract.amount - notStorable.getStackSize());
+                    final FluidStack storable = fh.drain(toStore, false);
+
+                    if (storable == null || storable.amount == 0) {
+                        return;
+                    } else {
+                        extract.amount = storable.amount;
+                    }
+                }
+
+                // Actually drain
+                final FluidStack drained = fh.drain(extract, true);
+                extract.amount = drained.amount;
+
+                final IAEFluidStack notInserted = Platform.poweredInsert(this.getPowerSource(), this.monitor, AEFluidStack.fromFluidStack(extract), this.getActionSource());
+
+                if (notInserted != null && notInserted.getStackSize() > 0) {
+                    IAEFluidStack spill = this.monitor.injectItems(notInserted, Actionable.MODULATE, this.getActionSource());
+                    if (spill != null && spill.getStackSize() > 0) {
+                        fh.fill(spill.getFluidStack(), true);
+                    }
+                }
+
+                if (held.getCount() == 1) {
+                    player.inventory.setItemStack(fh.getContainer());
+                } else {
+                    player.inventory.getItemStack().shrink(1);
+                    if (!player.inventory.addItemStackToInventory(fh.getContainer())) {
+                        player.dropItem(fh.getContainer(), false);
+                    }
+                }
+            }
+            this.updateHeld(player);
+        }
+    }
+
+    private double getPowerMultiplier() {
+        return this.powerMultiplier;
+    }
+
+    void setPowerMultiplier(final double powerMultiplier) {
+        this.powerMultiplier = powerMultiplier;
+    }
+
+    public boolean isValid(Object verificationToken) {
+        return true;
+    }
+
+
+    @Override
+    public void postChange(IBaseMonitor<IAEFluidStack> monitor, Iterable<IAEFluidStack> change, IActionSource actionSource) {
+        for (final IAEFluidStack is : change) {
+            this.fluids.add(is);
+        }
+    }
+
+    @Override
+    public void onListUpdate() {
+        for (final IContainerListener c : this.listeners) {
+            this.queueInventory(c);
+        }
+    }
+
+    @Override
+    public void addListener(IContainerListener listener) {
+        super.addListener(listener);
+
+        this.queueInventory(listener);
+    }
+
+    @Override
+    public void onContainerClosed(final EntityPlayer player) {
+        super.onContainerClosed(player);
+        if (this.monitor != null) {
+            this.monitor.removeListener(this);
+        }
+    }
+
+    private void queueInventory(final IContainerListener c) {
+        if (Platform.isServer() && c instanceof EntityPlayer && this.monitor != null) {
+            try {
+                PacketMEFluidInventoryUpdate piu = new PacketMEFluidInventoryUpdate();
+                final IItemList<IAEFluidStack> monitorCache = this.monitor.getStorageList();
+
+                for (final IAEFluidStack send : monitorCache) {
+                    try {
+                        piu.appendFluid(send);
+                    } catch (final BufferOverflowException boe) {
+                        NetworkHandler.instance().sendTo(piu, (EntityPlayerMP) c);
+
+                        piu = new PacketMEFluidInventoryUpdate();
+                        piu.appendFluid(send);
+                    }
+                }
+
+                NetworkHandler.instance().sendTo(piu, (EntityPlayerMP) c);
+            } catch (final IOException e) {
+                AELog.debug(e);
+            }
+        }
+    }
+
+    @Override
+    public IConfigManager getConfigManager() {
+        if (Platform.isServer()) {
+            return this.serverCM;
+        }
+        return this.clientCM;
+    }
+
+    @Override
+    public void updateSetting(IConfigManager manager, Enum settingName, Enum newValue) {
+        if (this.getGui() != null) {
+            this.getGui().updateSetting(manager, settingName, newValue);
+        }
+    }
+
+    public void setTargetStack(final IAEFluidStack stack) {
+        if (Platform.isClient()) {
+            if (stack == null && this.clientRequestedTargetFluid == null) {
+                return;
+            }
+            if (stack != null && this.clientRequestedTargetFluid != null && stack.getFluidStack()
+                    .isFluidEqual(this.clientRequestedTargetFluid.getFluidStack())) {
+                return;
+            }
+            NetworkHandler.instance().sendToServer(new PacketTargetFluidStack((AEFluidStack) stack));
+        }
+
+        this.clientRequestedTargetFluid = stack == null ? null : stack.copy();
+    }
+
+    private IConfigManagerHost getGui() {
+        return this.gui;
+    }
+
+    public boolean isPowered() {
+        return this.hasPower;
+    }
+
+    public void setGui(@Nonnull final IConfigManagerHost gui) {
+        this.gui = gui;
+    }
+}