Applied-Energistics-2-tiler.../util/inv/AdaptorIInventory.java
2014-02-08 23:59:38 -06:00

363 lines
7.4 KiB
Java

package appeng.util.inv;
import java.util.Iterator;
import net.minecraft.inventory.IInventory;
import net.minecraft.item.ItemStack;
import appeng.api.config.FuzzyMode;
import appeng.util.InventoryAdaptor;
import appeng.util.Platform;
public class AdaptorIInventory extends InventoryAdaptor
{
private IInventory i;
private boolean wrapperEnabled;
public AdaptorIInventory(IInventory s) {
i = s;
wrapperEnabled = s instanceof IInventoryWrapper;
}
boolean canRemoveStackFromSlot(int x, ItemStack is)
{
if ( wrapperEnabled )
return ((IInventoryWrapper) i).canRemoveItemFromSlot( x, is );
return true;
}
@Override
public boolean containsItems()
{
int s = i.getSizeInventory();
for (int x = 0; x < s; x++)
{
if ( i.getStackInSlot( x ) != null )
return true;
}
return false;
}
@Override
public ItemStack removeSimilarItems(int how_many, ItemStack filter, FuzzyMode fuzzyMode, IInventoryDestination destination)
{
int s = i.getSizeInventory();
for (int x = 0; x < s; x++)
{
ItemStack is = i.getStackInSlot( x );
if ( is != null && canRemoveStackFromSlot( x, is ) && (filter == null || Platform.isSameItemFuzzy( is, filter, fuzzyMode )) )
{
int lhow_many = how_many;
if ( lhow_many > is.stackSize )
lhow_many = is.stackSize;
if ( destination != null && !destination.canInsert( is ) )
lhow_many = 0;
ItemStack rv = null;
if ( lhow_many > 0 )
{
rv = is.copy();
rv.stackSize = lhow_many;
if ( is.stackSize == rv.stackSize )
{
i.setInventorySlotContents( x, null );
i.markDirty();
}
else
{
ItemStack po = is.copy();
po.stackSize -= rv.stackSize;
i.setInventorySlotContents( x, po );
i.markDirty();
}
}
if ( rv != null )
{
// i.markDirty();
return rv;
}
}
}
return null;
}
@Override
public ItemStack simulateSimilarRemove(int how_many, ItemStack filter, FuzzyMode fuzzyMode, IInventoryDestination destination)
{
int s = i.getSizeInventory();
for (int x = 0; x < s; x++)
{
ItemStack is = i.getStackInSlot( x );
if ( is != null && canRemoveStackFromSlot( x, is ) && (filter == null || Platform.isSameItemFuzzy( is, filter, fuzzyMode )) )
{
int lhow_many = how_many;
if ( lhow_many > is.stackSize )
lhow_many = is.stackSize;
if ( destination != null && !destination.canInsert( is ) )
lhow_many = 0;
if ( lhow_many > 0 )
{
ItemStack rv = is.copy();
rv.stackSize = lhow_many;
return rv;
}
}
}
return null;
}
@Override
public ItemStack removeItems(int how_many, ItemStack filter, IInventoryDestination destination)
{
int s = i.getSizeInventory();
ItemStack rv = null;
for (int x = 0; x < s && how_many > 0; x++)
{
ItemStack is = i.getStackInSlot( x );
if ( is != null && canRemoveStackFromSlot( x, is ) && (filter == null || Platform.isSameItemPrecise( is, filter )) )
{
int lhow_many = how_many;
if ( lhow_many > is.stackSize )
lhow_many = is.stackSize;
if ( destination != null && !destination.canInsert( is ) )
lhow_many = 0;
if ( lhow_many > 0 )
{
if ( rv == null )
{
rv = is.copy();
filter = rv;
rv.stackSize = lhow_many;
how_many -= lhow_many;
}
else
{
rv.stackSize += lhow_many;
how_many -= lhow_many;
}
if ( is.stackSize == lhow_many )
{
i.setInventorySlotContents( x, null );
i.markDirty();
}
else
{
ItemStack po = is.copy();
po.stackSize -= lhow_many;
i.setInventorySlotContents( x, po );
i.markDirty();
}
}
}
}
// if ( rv != null )
// i.markDirty();
return rv;
}
@Override
public ItemStack simulateRemove(int how_many, ItemStack filter, IInventoryDestination destination)
{
int s = i.getSizeInventory();
ItemStack rv = null;
for (int x = 0; x < s && how_many > 0; x++)
{
ItemStack is = i.getStackInSlot( x );
if ( is != null && canRemoveStackFromSlot( x, is ) && (filter == null || Platform.isSameItemPrecise( is, filter )) )
{
int lhow_many = how_many;
if ( lhow_many > is.stackSize )
lhow_many = is.stackSize;
if ( destination != null && !destination.canInsert( is ) )
lhow_many = 0;
if ( lhow_many > 0 )
{
if ( rv == null )
{
rv = is.copy();
rv.stackSize = lhow_many;
how_many -= lhow_many;
}
else
{
rv.stackSize += lhow_many;
how_many -= lhow_many;
}
}
}
}
return rv;
}
@Override
public ItemStack addItems(ItemStack A)
{
if ( A == null )
return null;
if ( A.stackSize == 0 )
return null;
ItemStack left = A.copy();
int stack_limit = A.getMaxStackSize();
if ( stack_limit > i.getInventoryStackLimit() )
stack_limit = i.getInventoryStackLimit();
int s = i.getSizeInventory();
for (int pass = 0; pass < 2; pass++)
{
for (int x = 0; x < s; x++)
{
if ( i.isItemValidForSlot( x, A ) )
{
ItemStack is = i.getStackInSlot( x );
if ( is == null && pass != 0 )
{
ItemStack thisSlot = left.copy();
if ( thisSlot.stackSize > stack_limit )
thisSlot.stackSize = stack_limit;
left.stackSize -= thisSlot.stackSize;
i.setInventorySlotContents( x, thisSlot );
if ( left.stackSize <= 0 )
{
// i.markDirty();
return null;
}
}
else if ( is != null )
{
if ( Platform.isSameItem( is, left ) )
{
if ( is.stackSize < stack_limit )
{
int room = stack_limit - is.stackSize;
int used = left.stackSize;
if ( used > room )
used = room;
is.stackSize += used;
i.setInventorySlotContents( x, is );
i.markDirty();
left.stackSize -= used;
if ( left.stackSize <= 0 )
{
// i.markDirty();
return null;
}
}
}
}
}
}
}
// if ( left.stackSize != A.stackSize )
// i.markDirty();
return left;
}
@Override
public ItemStack simulateAdd(ItemStack A)
{
if ( A == null )
return A;
ItemStack left = A.copy();
int stack_limit = A.getMaxStackSize();
if ( stack_limit > i.getInventoryStackLimit() )
stack_limit = i.getInventoryStackLimit();
int s = i.getSizeInventory();
for (int x = 0; x < s; x++)
{
if ( i.isItemValidForSlot( x, A ) )
{
ItemStack is = i.getStackInSlot( x );
if ( is == null )
{
ItemStack thisSlot = left.copy();
if ( thisSlot.stackSize > stack_limit )
thisSlot.stackSize = stack_limit;
left.stackSize -= thisSlot.stackSize;
if ( left.stackSize <= 0 )
{
return null;
}
}
else
{
if ( Platform.isSameItem( is, left ) )
{
if ( is.stackSize < stack_limit )
{
int room = stack_limit - is.stackSize;
int used = left.stackSize;
if ( used > room )
used = room;
left.stackSize -= used;
if ( left.stackSize <= 0 )
{
return null;
}
}
}
}
}
}
return left;
}
class InvIterator implements Iterator<ItemSlot>
{
final ItemSlot is = new ItemSlot();
int x = 0;
@Override
public boolean hasNext()
{
return x < i.getSizeInventory();
}
@Override
public ItemSlot next()
{
is.slot = x;
is.itemStack = i.getStackInSlot( x++ );
return is;
}
@Override
public void remove()
{
// nothing!
}
};
@Override
public Iterator<ItemSlot> iterator()
{
return new InvIterator();
}
}