422 lines
11 KiB
Java
422 lines
11 KiB
Java
package appeng.tile.storage;
|
|
|
|
import net.minecraft.inventory.IInventory;
|
|
import net.minecraft.item.ItemStack;
|
|
import net.minecraft.nbt.NBTTagCompound;
|
|
import net.minecraftforge.common.util.ForgeDirection;
|
|
import appeng.api.AEApi;
|
|
import appeng.api.config.Actionable;
|
|
import appeng.api.config.FullnessMode;
|
|
import appeng.api.config.OperationMode;
|
|
import appeng.api.config.RedstoneMode;
|
|
import appeng.api.config.Settings;
|
|
import appeng.api.config.Upgrades;
|
|
import appeng.api.config.YesNo;
|
|
import appeng.api.implementations.IUpgradeableHost;
|
|
import appeng.api.networking.GridFlags;
|
|
import appeng.api.networking.IGridNode;
|
|
import appeng.api.networking.energy.IEnergySource;
|
|
import appeng.api.networking.security.BaseActionSource;
|
|
import appeng.api.networking.security.MachineSource;
|
|
import appeng.api.networking.ticking.IGridTickable;
|
|
import appeng.api.networking.ticking.TickRateModulation;
|
|
import appeng.api.networking.ticking.TickingRequest;
|
|
import appeng.api.storage.IMEInventory;
|
|
import appeng.api.storage.IMEMonitor;
|
|
import appeng.api.storage.StorageChannel;
|
|
import appeng.api.storage.data.IAEFluidStack;
|
|
import appeng.api.storage.data.IAEItemStack;
|
|
import appeng.api.storage.data.IAEStack;
|
|
import appeng.api.storage.data.IItemList;
|
|
import appeng.api.util.AECableType;
|
|
import appeng.api.util.DimensionalCoord;
|
|
import appeng.api.util.IConfigManager;
|
|
import appeng.core.settings.TickRates;
|
|
import appeng.me.GridAccessException;
|
|
import appeng.parts.automation.UpgradeInventory;
|
|
import appeng.tile.TileEvent;
|
|
import appeng.tile.events.TileEventType;
|
|
import appeng.tile.grid.AENetworkInvTile;
|
|
import appeng.tile.inventory.AppEngInternalInventory;
|
|
import appeng.tile.inventory.InvOperation;
|
|
import appeng.util.ConfigManager;
|
|
import appeng.util.IConfigManagerHost;
|
|
import appeng.util.InventoryAdaptor;
|
|
import appeng.util.Platform;
|
|
import appeng.util.inv.WrapperInventoryRange;
|
|
|
|
public class TileIOPort extends AENetworkInvTile implements IUpgradeableHost, IConfigManagerHost, IGridTickable
|
|
{
|
|
|
|
ConfigManager cm = new ConfigManager( this );
|
|
|
|
final int input[] = { 0, 1, 2, 3, 4, 5 };
|
|
final int output[] = { 6, 7, 8, 9, 10, 11 };
|
|
|
|
final int outputSlots[] = { 6, 7, 8, 9, 10, 11 };
|
|
|
|
AppEngInternalInventory cells = new AppEngInternalInventory( this, 12 );
|
|
UpgradeInventory upgrades = new UpgradeInventory( AEApi.instance().blocks().blockIOPort.block(), this, 3 );
|
|
|
|
BaseActionSource mySrc = new MachineSource( this );
|
|
|
|
YesNo lastRedstoneState = YesNo.UNDECIDED;
|
|
|
|
@TileEvent(TileEventType.WORLD_NBT_WRITE)
|
|
public void writeToNBT_TileIOPort(NBTTagCompound data)
|
|
{
|
|
cm.writeToNBT( data );
|
|
cells.writeToNBT( data, "cells" );
|
|
upgrades.writeToNBT( data, "upgrades" );
|
|
data.setInteger( "lastRedstoneState", lastRedstoneState.ordinal() );
|
|
}
|
|
|
|
@TileEvent(TileEventType.WORLD_NBT_READ)
|
|
public void readFromNBT_TileIOPort(NBTTagCompound data)
|
|
{
|
|
cm.readFromNBT( data );
|
|
cells.readFromNBT( data, "cells" );
|
|
upgrades.readFromNBT( data, "upgrades" );
|
|
if ( data.hasKey( "lastRedstoneState" ) )
|
|
lastRedstoneState = YesNo.values()[data.getInteger( "lastRedstoneState" )];
|
|
}
|
|
|
|
public TileIOPort() {
|
|
gridProxy.setFlags( GridFlags.REQUIRE_CHANNEL );
|
|
cm.registerSetting( Settings.REDSTONE_CONTROLLED, RedstoneMode.IGNORE );
|
|
cm.registerSetting( Settings.FULLNESS_MODE, FullnessMode.EMPTY );
|
|
cm.registerSetting( Settings.OPERATION_MODE, OperationMode.EMPTY );
|
|
}
|
|
|
|
@Override
|
|
public AECableType getCableConnectionType(ForgeDirection dir)
|
|
{
|
|
return AECableType.SMART;
|
|
}
|
|
|
|
@Override
|
|
public DimensionalCoord getLocation()
|
|
{
|
|
return new DimensionalCoord( this );
|
|
}
|
|
|
|
@Override
|
|
public IInventory getInternalInventory()
|
|
{
|
|
return cells;
|
|
}
|
|
|
|
@Override
|
|
public void onChangeInventory(IInventory inv, int slot, InvOperation mc, ItemStack removed, ItemStack added)
|
|
{
|
|
if ( cells == inv )
|
|
{
|
|
updateTask();
|
|
}
|
|
}
|
|
|
|
private void updateTask()
|
|
{
|
|
try
|
|
{
|
|
if ( hasWork() )
|
|
gridProxy.getTick().wakeDevice( gridProxy.getNode() );
|
|
else
|
|
gridProxy.getTick().sleepDevice( gridProxy.getNode() );
|
|
}
|
|
catch (GridAccessException e)
|
|
{
|
|
// :P
|
|
}
|
|
}
|
|
|
|
public void updateRedstoneState()
|
|
{
|
|
YesNo currentState = worldObj.isBlockIndirectlyGettingPowered( xCoord, yCoord, zCoord ) ? YesNo.YES : YesNo.NO;
|
|
if ( lastRedstoneState != currentState )
|
|
{
|
|
lastRedstoneState = currentState;
|
|
updateTask();
|
|
}
|
|
}
|
|
|
|
public boolean getRedstoneState()
|
|
{
|
|
if ( lastRedstoneState == YesNo.UNDECIDED )
|
|
updateRedstoneState();
|
|
|
|
return lastRedstoneState == YesNo.YES;
|
|
}
|
|
|
|
private boolean isEnabled()
|
|
{
|
|
if ( getInstalledUpgrades( Upgrades.REDSTONE ) == 0 )
|
|
return true;
|
|
|
|
RedstoneMode rs = (RedstoneMode) cm.getSetting( Settings.REDSTONE_CONTROLLED );
|
|
if ( rs == RedstoneMode.HIGH_SIGNAL )
|
|
return getRedstoneState();
|
|
return !getRedstoneState();
|
|
}
|
|
|
|
@Override
|
|
public int[] getAccessibleSlotsBySide(ForgeDirection d)
|
|
{
|
|
if ( d == ForgeDirection.UP || d == ForgeDirection.DOWN )
|
|
return input;
|
|
|
|
return output;
|
|
}
|
|
|
|
@Override
|
|
public IConfigManager getConfigManager()
|
|
{
|
|
return cm;
|
|
}
|
|
|
|
@Override
|
|
public IInventory getInventoryByName(String name)
|
|
{
|
|
if ( name.equals( "upgrades" ) )
|
|
return upgrades;
|
|
|
|
if ( name.equals( "cells" ) )
|
|
return cells;
|
|
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public int getInstalledUpgrades(Upgrades u)
|
|
{
|
|
return upgrades.getInstalledUpgrades( u );
|
|
}
|
|
|
|
@Override
|
|
public void updateSetting(IConfigManager manager, Enum settingName, Enum newValue)
|
|
{
|
|
updateTask();
|
|
}
|
|
|
|
boolean hasWork()
|
|
{
|
|
if ( isEnabled() )
|
|
{
|
|
for (int x = 0; x < 6; x++)
|
|
if ( cells.getStackInSlot( x ) != null )
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
@Override
|
|
public TickingRequest getTickingRequest(IGridNode node)
|
|
{
|
|
return new TickingRequest( TickRates.IOPort.min, TickRates.IOPort.max, hasWork(), false );
|
|
}
|
|
|
|
@Override
|
|
public TickRateModulation tickingRequest(IGridNode node, int TicksSinceLastCall)
|
|
{
|
|
if ( !gridProxy.isActive() )
|
|
return TickRateModulation.IDLE;
|
|
|
|
long ItemsToMove = 256;
|
|
|
|
switch (getInstalledUpgrades( Upgrades.SPEED ))
|
|
{
|
|
case 1:
|
|
ItemsToMove *= 2;
|
|
break;
|
|
case 2:
|
|
ItemsToMove *= 4;
|
|
break;
|
|
case 3:
|
|
ItemsToMove *= 8;
|
|
break;
|
|
}
|
|
|
|
try
|
|
{
|
|
IMEInventory<IAEItemStack> itemNet = gridProxy.getStorage().getItemInventory();
|
|
IMEInventory<IAEFluidStack> fluidNet = gridProxy.getStorage().getFluidInventory();
|
|
IEnergySource energy = gridProxy.getEnergy();
|
|
for (int x = 0; x < 6; x++)
|
|
{
|
|
ItemStack is = cells.getStackInSlot( x );
|
|
if ( is != null )
|
|
{
|
|
if ( ItemsToMove > 0 )
|
|
{
|
|
IMEInventory<IAEItemStack> itemInv = getInv( is, StorageChannel.ITEMS );
|
|
IMEInventory<IAEFluidStack> fluidInv = getInv( is, StorageChannel.FLUIDS );
|
|
|
|
if ( cm.getSetting( Settings.OPERATION_MODE ) == OperationMode.EMPTY )
|
|
{
|
|
if ( itemInv != null )
|
|
ItemsToMove = transferContents( energy, itemInv, itemNet, ItemsToMove, StorageChannel.ITEMS );
|
|
if ( fluidInv != null )
|
|
ItemsToMove = transferContents( energy, fluidInv, fluidNet, ItemsToMove, StorageChannel.FLUIDS );
|
|
}
|
|
else
|
|
{
|
|
if ( itemInv != null )
|
|
ItemsToMove = transferContents( energy, itemNet, itemInv, ItemsToMove, StorageChannel.ITEMS );
|
|
if ( fluidInv != null )
|
|
ItemsToMove = transferContents( energy, fluidNet, fluidInv, ItemsToMove, StorageChannel.FLUIDS );
|
|
}
|
|
|
|
if ( ItemsToMove > 0 && shouldMove( itemInv, fluidInv ) && !moveSlot( x ) )
|
|
return TickRateModulation.IDLE;
|
|
|
|
return TickRateModulation.URGENT;
|
|
}
|
|
else
|
|
return TickRateModulation.URGENT;
|
|
}
|
|
}
|
|
|
|
}
|
|
catch (GridAccessException e)
|
|
{
|
|
return TickRateModulation.IDLE;
|
|
}
|
|
|
|
// nothing left to do...
|
|
return TickRateModulation.SLEEP;
|
|
}
|
|
|
|
private boolean shouldMove(IMEInventory<IAEItemStack> itemInv, IMEInventory<IAEFluidStack> fluidInv)
|
|
{
|
|
FullnessMode fm = (FullnessMode) cm.getSetting( Settings.FULLNESS_MODE );
|
|
|
|
if ( itemInv != null && fluidInv != null )
|
|
return matches( fm, itemInv ) && matches( fm, fluidInv );
|
|
else if ( itemInv != null )
|
|
return matches( fm, itemInv );
|
|
else if ( fluidInv != null )
|
|
return matches( fm, fluidInv );
|
|
|
|
return true;
|
|
}
|
|
|
|
private boolean matches(FullnessMode fm, IMEInventory src)
|
|
{
|
|
if ( fm == FullnessMode.HALF )
|
|
return true;
|
|
|
|
IItemList<? extends IAEStack> myList;
|
|
|
|
if ( src instanceof IMEMonitor )
|
|
myList = ((IMEMonitor) src).getStorageList();
|
|
else
|
|
myList = src.getAvailableItems( src.getChannel().createList() );
|
|
|
|
if ( fm == FullnessMode.EMPTY )
|
|
return myList.isEmpty();
|
|
|
|
IAEStack test = myList.getFirstItem();
|
|
if ( test != null )
|
|
{
|
|
test.setStackSize( 1 );
|
|
return src.injectItems( test, Actionable.SIMULATE, mySrc ) != null;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
ItemStack currentCell;
|
|
IMEInventory<IAEFluidStack> cachedFluid;
|
|
IMEInventory<IAEItemStack> cachedItem;
|
|
|
|
private IMEInventory getInv(ItemStack is, StorageChannel chan)
|
|
{
|
|
if ( currentCell != is )
|
|
{
|
|
currentCell = is;
|
|
cachedFluid = AEApi.instance().registries().cell().getCellInventory( is, null, StorageChannel.FLUIDS );
|
|
cachedItem = AEApi.instance().registries().cell().getCellInventory( is, null, StorageChannel.ITEMS );
|
|
}
|
|
|
|
if ( StorageChannel.ITEMS == chan )
|
|
return cachedItem;
|
|
|
|
return cachedFluid;
|
|
}
|
|
|
|
private long transferContents(IEnergySource energy, IMEInventory src, IMEInventory dest, long itemsToMove, StorageChannel chan)
|
|
{
|
|
IItemList<? extends IAEStack> myList;
|
|
if ( src instanceof IMEMonitor )
|
|
myList = ((IMEMonitor) src).getStorageList();
|
|
else
|
|
myList = src.getAvailableItems( src.getChannel().createList() );
|
|
|
|
boolean didStuff;
|
|
|
|
do
|
|
{
|
|
didStuff = false;
|
|
|
|
for (IAEStack s : myList)
|
|
{
|
|
long totalStackSize = s.getStackSize();
|
|
if ( totalStackSize > 0 )
|
|
{
|
|
IAEStack stack = dest.injectItems( s, Actionable.SIMULATE, mySrc );
|
|
|
|
long possible = 0;
|
|
if ( stack == null )
|
|
possible = totalStackSize;
|
|
else
|
|
possible = totalStackSize - stack.getStackSize();
|
|
|
|
if ( possible > 0 )
|
|
{
|
|
possible = Math.min( possible, itemsToMove );
|
|
s.setStackSize( possible );
|
|
|
|
IAEStack extracted = src.extractItems( s, Actionable.MODULATE, mySrc );
|
|
if ( extracted != null )
|
|
{
|
|
possible = extracted.getStackSize();
|
|
IAEStack failed = Platform.poweredInsert( energy, dest, extracted, mySrc );
|
|
|
|
if ( failed != null )
|
|
{
|
|
possible -= failed.getStackSize();
|
|
src.injectItems( failed, Actionable.MODULATE, mySrc );
|
|
}
|
|
|
|
if ( possible > 0 )
|
|
{
|
|
itemsToMove -= possible;
|
|
didStuff = true;
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
while (itemsToMove > 0 && didStuff);
|
|
|
|
return itemsToMove;
|
|
}
|
|
|
|
private boolean moveSlot(int x)
|
|
{
|
|
WrapperInventoryRange wir = new WrapperInventoryRange( this, outputSlots, true );
|
|
ItemStack result = InventoryAdaptor.getAdaptor( wir, ForgeDirection.UNKNOWN ).addItems( getStackInSlot( x ) );
|
|
|
|
if ( result == null )
|
|
{
|
|
setInventorySlotContents( x, null );
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
}
|