package appeng.me.storage; import java.util.HashSet; import java.util.Iterator; import net.minecraft.inventory.IInventory; import net.minecraft.item.Item; import net.minecraft.item.ItemStack; import net.minecraft.nbt.NBTBase; import net.minecraft.nbt.NBTTagCompound; import net.minecraft.nbt.NBTTagInt; import net.minecraft.nbt.NBTTagShort; import net.minecraftforge.oredict.OreDictionary; import appeng.api.config.Actionable; import appeng.api.config.FuzzyMode; import appeng.api.exceptions.AppEngException; import appeng.api.implementations.items.IStorageCell; import appeng.api.networking.security.BaseActionSource; import appeng.api.storage.IMEInventory; import appeng.api.storage.IMEInventoryHandler; import appeng.api.storage.StorageChannel; import appeng.api.storage.data.IAEItemStack; import appeng.api.storage.data.IItemList; import appeng.util.Platform; import appeng.util.item.AEItemStack; import appeng.util.item.ItemList; public class CellInventory implements IMEInventory { static final String ITEM_TYPE_TAG = "it"; static final String ITEM_COUNT_TAG = "ic"; static final String ITEM_SLOT = "#"; static final String ITEM_SLOTCOUNT = "@"; static final String ITEM_PRE_FORMATED_COUNT = "PF"; static final String ITEM_PRE_FORMATED_SLOT = "PF#"; static final String ITEM_PRE_FORMATED_NAME = "PN"; static final String ITEM_PRE_FORMATED_FUZZY = "FP"; static protected String[] ITEM_SLOT_ARR; static protected String[] ITEM_SLOTCOUNT_ARR; final protected NBTTagCompound tagCompound; protected int MAX_ITEM_TYPES = 63; protected short storedItems = 0; protected int storedItemCount = 0; protected ItemList cellItems; protected ItemStack i; protected IStorageCell CellType; protected CellInventory(NBTTagCompound data) { tagCompound = data; } protected void loadCellItems() { if ( cellItems == null ) cellItems = new ItemList(); cellItems.resetStatus(); // clears totals and stuff. int types = (int) storedItemTypes(); for (int x = 0; x < types; x++) { ItemStack t = ItemStack.loadItemStackFromNBT( tagCompound.getCompoundTag( ITEM_SLOT_ARR[x] ) ); if ( t != null ) { t.stackSize = tagCompound.getInteger( ITEM_SLOTCOUNT_ARR[x] ); if ( t.stackSize > 0 ) { cellItems.add( AEItemStack.create( t ) ); } } } // cellItems.clean(); } void saveChanges() { // cellItems.clean(); int itemCount = 0; // add new pretty stuff... int x = 0; Iterator i = cellItems.iterator(); while (i.hasNext()) { IAEItemStack v = i.next(); itemCount += v.getStackSize(); NBTBase c = tagCompound.getTag( ITEM_SLOT_ARR[x] ); if ( c instanceof NBTTagCompound ) v.writeToNBT( (NBTTagCompound) c ); else { NBTTagCompound g = new NBTTagCompound(); v.writeToNBT( g ); tagCompound.setCompoundTag( ITEM_SLOT_ARR[x], g ); } NBTBase tagSlotCount = tagCompound.getTag( ITEM_SLOTCOUNT_ARR[x] ); if ( tagSlotCount instanceof NBTTagInt ) ((NBTTagInt) tagSlotCount).data = (int) v.getStackSize(); else tagCompound.setInteger( ITEM_SLOTCOUNT_ARR[x], (int) v.getStackSize() ); x++; } NBTBase tagType = tagCompound.getTag( ITEM_TYPE_TAG ); NBTBase tagCount = tagCompound.getTag( ITEM_COUNT_TAG ); short oldStoreditems = storedItems; if ( tagType instanceof NBTTagShort ) ((NBTTagShort) tagType).data = storedItems = (short) cellItems.size(); else tagCompound.setShort( ITEM_TYPE_TAG, storedItems = (short) cellItems.size() ); if ( tagCount instanceof NBTTagInt ) ((NBTTagInt) tagCount).data = storedItemCount = itemCount; else tagCompound.setInteger( ITEM_COUNT_TAG, storedItemCount = itemCount ); // clean any old crusty stuff... for (; x < oldStoreditems && x < MAX_ITEM_TYPES; x++) { tagCompound.removeTag( ITEM_SLOT_ARR[x] ); tagCompound.removeTag( ITEM_SLOTCOUNT_ARR[x] ); } } protected CellInventory(ItemStack o) throws AppEngException { if ( ITEM_SLOT_ARR == null ) { ITEM_SLOT_ARR = new String[MAX_ITEM_TYPES]; ITEM_SLOTCOUNT_ARR = new String[MAX_ITEM_TYPES]; for (int x = 0; x < MAX_ITEM_TYPES; x++) { ITEM_SLOT_ARR[x] = ITEM_SLOT + x; ITEM_SLOTCOUNT_ARR[x] = ITEM_SLOTCOUNT + x; } } if ( o == null ) { throw new AppEngException( "ItemStack was used as a cell, but was not a cell!" ); } CellType = null; i = o; Item type = i.getItem(); if ( type instanceof IStorageCell ) { CellType = (IStorageCell) i.getItem(); MAX_ITEM_TYPES = CellType.getTotalTypes( i ); } if ( CellType == null ) { throw new AppEngException( "ItemStack was used as a cell, but was not a cell!" ); } if ( !CellType.isStorageCell( i ) ) { throw new AppEngException( "ItemStack was used as a cell, but was not a cell!" ); } if ( MAX_ITEM_TYPES > 63 ) MAX_ITEM_TYPES = 63; if ( MAX_ITEM_TYPES < 1 ) MAX_ITEM_TYPES = 1; tagCompound = Platform.openNbtData( o ); storedItems = tagCompound.getShort( ITEM_TYPE_TAG ); storedItemCount = tagCompound.getInteger( ITEM_COUNT_TAG ); cellItems = null; } ItemList getCellItems() { if ( cellItems == null ) { cellItems = new ItemList(); loadCellItems(); } return cellItems; } public int BytesPerType() { return CellType.BytePerType( i ); } public boolean canHoldNewItem() { long bytesFree = freeBytes(); return (bytesFree > BytesPerType() || (bytesFree == BytesPerType() && unusedItemCount() > 0)) && remainingItemTypes() > 0; } public static IMEInventoryHandler getCell(ItemStack o) { try { return new CellInventoryHandler( new CellInventory( o ) ); } catch (AppEngException e) { return null; } } private static boolean isStorageCell(ItemStack i) { if ( i == null ) { return false; } try { Item type = i.getItem(); if ( type instanceof IStorageCell ) { return !((IStorageCell) type).storableInStorageCell(); } } catch (Throwable err) { return true; } return false; } public static boolean isCell(ItemStack i) { if ( i == null ) { return false; } Item type = i.getItem(); if ( type instanceof IStorageCell ) { return ((IStorageCell) type).isStorageCell( i ); } return false; } public long totalBytes() { return CellType.getBytes( i ); } public long freeBytes() { return totalBytes() - usedBytes(); } public long usedBytes() { long bytesForItemCount = (storedItemCount() + unusedItemCount()) / 8; return storedItemTypes() * BytesPerType() + bytesForItemCount; } public long getTotalItemTypes() { return MAX_ITEM_TYPES; } public long storedItemTypes() { return storedItems; } public long storedItemCount() { return storedItemCount; } private void updateItemCount(long delta) { tagCompound.setInteger( ITEM_COUNT_TAG, storedItemCount = (int) (storedItemCount + delta) ); } public long remainingItemTypes() { long basedOnStorage = freeBytes() / BytesPerType(); long baseOnTotal = getTotalItemTypes() - storedItemTypes(); return basedOnStorage > baseOnTotal ? baseOnTotal : basedOnStorage; } public long remainingItemCount() { long remaining = freeBytes() * 8 + unusedItemCount(); return remaining > 0 ? remaining : 0; } // returns the number of items that can be added without using an additional // byte! public int unusedItemCount() { int div = (int) (storedItemCount() % 8); if ( div == 0 ) { return 0; } return 8 - div; } private static HashSet blackList = new HashSet(); public static void addBasicBlackList(int itemID, int Meta) { blackList.add( (Meta << Platform.DEF_OFFSET) | itemID ); } public static boolean isBlackListed(IAEItemStack input) { if ( blackList.contains( (OreDictionary.WILDCARD_VALUE << Platform.DEF_OFFSET) | input.getItem().itemID ) ) return true; return blackList.contains( (input.getItemDamage() << Platform.DEF_OFFSET) | input.getItem().itemID ); } private boolean isEmpty(IMEInventory meinv) { return meinv.getAvailableItems( new ItemList() ).isEmpty(); } @Override public IAEItemStack injectItems(IAEItemStack input, Actionable mode, BaseActionSource src) { if ( input == null ) return null; if ( input.getStackSize() == 0 ) return null; if ( isBlackListed( input ) || CellType.isBlackListed( i, input ) ) return input; ItemStack sharedItemStack = input.getItemStack(); if ( CellInventory.isStorageCell( sharedItemStack ) ) { IMEInventory meinv = getCell( sharedItemStack ); if ( meinv != null && !isEmpty( meinv ) ) return input; } IAEItemStack l = getCellItems().findPrecise( input ); if ( l != null ) { long remainingItemSlots = remainingItemCount(); if ( remainingItemSlots < 0 ) return input; if ( input.getStackSize() > remainingItemSlots ) { IAEItemStack r = input.copy(); r.setStackSize( r.getStackSize() - remainingItemSlots ); if ( mode == Actionable.MODULATE ) { l.setStackSize( l.getStackSize() + remainingItemSlots ); updateItemCount( remainingItemSlots ); saveChanges(); } return r; } else { if ( mode == Actionable.MODULATE ) { l.setStackSize( l.getStackSize() + input.getStackSize() ); updateItemCount( input.getStackSize() ); saveChanges(); } return null; } } if ( canHoldNewItem() ) // room for new type, and for at least one item! { int remainingItemCount = (int) remainingItemCount() - BytesPerType() * 8; if ( remainingItemCount > 0 ) { if ( input.getStackSize() > remainingItemCount ) { ItemStack toReturn = Platform.cloneItemStack( sharedItemStack ); toReturn.stackSize = sharedItemStack.stackSize - remainingItemCount; if ( mode == Actionable.MODULATE ) { ItemStack toWrite = Platform.cloneItemStack( sharedItemStack ); toWrite.stackSize = remainingItemCount; cellItems.add( AEItemStack.create( toWrite ) ); updateItemCount( toWrite.stackSize ); saveChanges(); } return AEItemStack.create( toReturn ); } if ( mode == Actionable.MODULATE ) { updateItemCount( input.getStackSize() ); cellItems.add( input ); saveChanges(); } return null; } } return input; } @Override public IAEItemStack extractItems(IAEItemStack request, Actionable mode, BaseActionSource src) { if ( request == null ) return null; ItemStack sharedItem = request.getItemStack(); int size = sharedItem.stackSize; IAEItemStack Results = null; IAEItemStack l = getCellItems().findPrecise( request ); if ( l != null ) { Results = l.copy(); if ( l.getStackSize() <= size ) { Results.setStackSize( l.getStackSize() ); if ( mode == Actionable.MODULATE ) { updateItemCount( -l.getStackSize() ); l.setStackSize( 0 ); saveChanges(); } } else { Results.setStackSize( size ); if ( mode == Actionable.MODULATE ) { l.setStackSize( l.getStackSize() - size ); updateItemCount( -size ); saveChanges(); } } } return Results; } @Override public IItemList getAvailableItems(IItemList out) { for (IAEItemStack i : getCellItems()) out.add( i ); return out; } @Override public StorageChannel getChannel() { return StorageChannel.ITEMS; } public double getIdleDrain() { return CellType.getIdleDrain(); } public FuzzyMode getFuzzyMode() { return CellType.getFuzzyMode( this.i ); } public IInventory getConfigInventory() { return CellType.getConfigInventory( this.i ); } public IInventory getUpgradesInventory() { return CellType.getUpgradesInventory( this.i ); } public int getStatusForCell() { if ( canHoldNewItem() ) return 1; if ( remainingItemCount() > 0 ) return 2; return 3; } }