Inventory.java

package Roguelike.Items;

import Roguelike.Global.Statistic;
import Roguelike.Items.Item.EquipmentSlot;
import Roguelike.Items.Item.ItemCategory;
import Roguelike.Util.FastEnumMap;
import com.badlogic.gdx.utils.Array;
import com.badlogic.gdx.utils.XmlReader.Element;

import java.util.HashMap;
import java.util.Iterator;

public final class Inventory
{
	public boolean isVariableMapDirty = true;

	public Array<Item> m_items = new Array<Item>();

	private FastEnumMap<EquipmentSlot, Item> m_equipment = new FastEnumMap<EquipmentSlot, Item>( EquipmentSlot.class );

	public int upgradeStones;

	public void load( Element xml )
	{
		Element equipElement = xml.getChildByName( "Equipment" );
		if ( equipElement != null )
		{
			for ( int i = 0; i < equipElement.getChildCount(); i++ )
			{
				Element el = equipElement.getChild( i );

				Item item = null;
				if ( el.getChildCount() > 0 )
				{
					item = Item.load( el );
				}
				else
				{
					item = Item.load( el.getText() );
				}

				item.canDrop = el.getBooleanAttribute( "Drop", false );
				item.dropChanceEqn = el.getAttribute( "DropChance", null );

				addItem( item );
				equip( item );
			}
		}

		for ( Element el : xml.getChildrenByName( "Item" ) )
		{
			Item item = Item.load( el );

			item.canDrop = el.getBooleanAttribute( "Drop", true );
			item.dropChanceEqn = el.getAttribute( "DropChance", null );

			addItem( item );
		}

		for ( Element el : xml.getChildrenByName( "Batch" ) )
		{
			String recipeList = el.get( "Recipes" );
			String[] recipes = recipeList.split( "," );

			for (String recipe : recipes)
			{
				Item item = Item.load( recipe, el );

				item.canDrop = el.getBooleanAttribute( "Drop", false );
				item.dropChanceEqn = el.getAttribute( "DropChance", null );

				addItem( item );
			}
		}
	}

	public void addItem( Item item )
	{
		if ( !m_items.contains( item, true ) )
		{
			if ( item.canStack )
			{
				Item found = null;

				for ( Item i : m_items )
				{
					if ( i.name.equals( item.name ) )
					{
						found = i;
					}
				}

				if ( found == null )
				{
					m_items.add( item );
				}
				else
				{
					found.count += item.count;
				}
			}
			else
			{
				m_items.add( item );
			}
		}
	}

	public void equip( Item item )
	{
		if ( !m_items.contains( item, true ) )
		{
			addItem( item );
		}

		for ( int i = 0; i < item.slots.size; i++ )
		{
			EquipmentSlot slot = item.slots.get( i );
			if ( m_equipment.containsKey( slot ) )
			{
				unequip( m_equipment.get( slot ) );
			}
		}

		for ( int i = 0; i < item.slots.size; i++ )
		{
			EquipmentSlot slot = item.slots.get( i );
			m_equipment.put( slot, item );
			isVariableMapDirty = true;
		}
	}

	public void unequip( Item item )
	{
		for ( int i = 0; i < item.slots.size; i++ )
		{
			EquipmentSlot slot = item.slots.get( i );
			m_equipment.remove( slot );
			isVariableMapDirty = true;
		}
	}

	public void removeItem( Item item )
	{
		removeItem( item.name, 1 );
	}

	public void removeItem( String itemName, int count )
	{
		Item found = null;

		for ( Item i : m_items )
		{
			if ( i.name.equalsIgnoreCase( itemName ) )
			{
				found = i;
				break;
			}
		}

		if ( found != null )
		{
			if ( isEquipped( found ) )
			{
				unequip( found );
			}

			if (count == 0)
			{
				m_items.removeValue( found, true );
			}
			else
			{
				found.count -= count;
				if ( found.count == 0 )
				{
					m_items.removeValue( found, true );
				}
			}
		}
		else
		{
			throw new RuntimeException( "Tried to remove an item that isnt in the inventory! Name: " + itemName );
		}
	}

	public boolean isEquipped( Item item )
	{
		if (item.getMainSlot() == null) { return false; }
		return m_equipment.get( item.getMainSlot() ) == item;
	}

	public int getStatistic( HashMap<String, Integer> variableMap, Statistic stat )
	{
		int val = 0;

		for ( EquipmentSlot slot : EquipmentSlot.values() )
		{
			if ( slot != EquipmentSlot.WEAPON && m_equipment.containsKey( slot ) )
			{
				Item item = m_equipment.get( slot );
				if ( slot == item.getMainSlot() )
				{
					val += item.getStatistic( variableMap, stat );
				}
			}
		}

		return val;
	}

	public void toggleEquip( Item item )
	{
		if ( isEquipped( item ) )
		{
			unequip( item );
		}
		else
		{
			equip( item );
		}
	}

	public Item getEquip( EquipmentSlot slot )
	{
		Item item = m_equipment.get( slot );
		if ( item != null )
		{
			if ( slot == item.getMainSlot() ) { return item; }
		}

		return null;
	}

	public int getItemCount( String name )
	{
		int count = 0;

		for (Item item : m_items)
		{
			if (name.equalsIgnoreCase( item.name ))
			{
				count += item.count;
			}
		}

		return count;
	}

	public Item getItem( String name )
	{
		for (Item item : m_items)
		{
			if (name.equalsIgnoreCase( item.name ))
			{
				return item;
			}
		}

		return null;
	}

	public Iterator<Item> iterator( ItemCategory type )
	{
		return new ItemIterator( type, m_items.iterator() );
	}

	private class ItemIterator implements Iterator<Item>
	{
		private ItemCategory type;
		private Iterator<Item> itr;
		private Item queued;

		public ItemIterator( ItemCategory type, Iterator<Item> itr )
		{
			this.type = type;
			this.itr = itr;
		}

		@Override
		public boolean hasNext()
		{
			while ( itr.hasNext() )
			{
				Item i = itr.next();

				if ( type == ItemCategory.ALL || i.category == type )
				{
					queued = i;
					break;
				}
			}

			return queued != null;
		}

		@Override
		public Item next()
		{
			Item temp = queued;
			queued = null;

			return temp;
		}

		@Override
		public void remove()
		{
		}
	}
}