equivalent-exchange-3/src/main/java/com/pahimar/ee3/inventory/ContainerAlchemicalChest.java

package com.pahimar.ee3.inventory;

import com.pahimar.ee3.tileentity.TileEntityAlchemicalChest;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.entity.player.InventoryPlayer;
import net.minecraft.inventory.Slot;
import net.minecraft.item.ItemStack;

public class ContainerAlchemicalChest extends ContainerEE
{
    // Small Chest
    public static final int SMALL_CHEST_INVENTORY_ROWS = 4;
    public static final int SMALL_CHEST_INVENTORY_COLUMNS = 12;
    public static final int SMALL_INVENTORY_SIZE = SMALL_CHEST_INVENTORY_ROWS * SMALL_CHEST_INVENTORY_COLUMNS;
    // Medium Chest
    public static final int MEDIUM_CHEST_INVENTORY_ROWS = 7;
    public static final int MEDIUM_CHEST_INVENTORY_COLUMNS = 12;
    public static final int MEDIUM_INVENTORY_SIZE = MEDIUM_CHEST_INVENTORY_ROWS * MEDIUM_CHEST_INVENTORY_COLUMNS;
    // Large Chest
    public static final int LARGE_CHEST_INVENTORY_ROWS = 9;
    public static final int LARGE_CHEST_INVENTORY_COLUMNS = 13;
    public static final int LARGE_INVENTORY_SIZE = LARGE_CHEST_INVENTORY_ROWS * LARGE_CHEST_INVENTORY_COLUMNS;

    private TileEntityAlchemicalChest tileEntityAlchemicalChest;
    private int chestInventoryRows;
    private int chestInventoryColumns;

    public ContainerAlchemicalChest(InventoryPlayer inventoryPlayer, TileEntityAlchemicalChest tileEntityAlchemicalChest)
    {
        this.tileEntityAlchemicalChest = tileEntityAlchemicalChest;
        tileEntityAlchemicalChest.openInventory();

        if (this.tileEntityAlchemicalChest.getState() == 0)
        {
            chestInventoryRows = SMALL_CHEST_INVENTORY_ROWS;
            chestInventoryColumns = SMALL_CHEST_INVENTORY_COLUMNS;
        }
        else if (this.tileEntityAlchemicalChest.getState() == 1)
        {
            chestInventoryRows = MEDIUM_CHEST_INVENTORY_ROWS;
            chestInventoryColumns = MEDIUM_CHEST_INVENTORY_COLUMNS;
        }
        else if (this.tileEntityAlchemicalChest.getState() == 2)
        {
            chestInventoryRows = LARGE_CHEST_INVENTORY_ROWS;
            chestInventoryColumns = LARGE_CHEST_INVENTORY_COLUMNS;
        }

        // Add the Alchemical Chest slots to the container
        for (int chestRowIndex = 0; chestRowIndex < chestInventoryRows; ++chestRowIndex)
        {
            for (int chestColumnIndex = 0; chestColumnIndex < chestInventoryColumns; ++chestColumnIndex)
            {
                if (this.tileEntityAlchemicalChest.getState() == 0)
                {
                    this.addSlotToContainer(new Slot(tileEntityAlchemicalChest, chestColumnIndex + chestRowIndex * chestInventoryColumns, 8 + chestColumnIndex * 18, 18 + chestRowIndex * 18));
                }
                else if (this.tileEntityAlchemicalChest.getState() == 1)
                {
                    this.addSlotToContainer(new Slot(tileEntityAlchemicalChest, chestColumnIndex + chestRowIndex * chestInventoryColumns, 8 + chestColumnIndex * 18, 18 + chestRowIndex * 18));
                }
                else if (this.tileEntityAlchemicalChest.getState() == 2)
                {
                    this.addSlotToContainer(new Slot(tileEntityAlchemicalChest, chestColumnIndex + chestRowIndex * chestInventoryColumns, 8 + chestColumnIndex * 18, 8 + chestRowIndex * 18));
                }
            }
        }

        // Add the player's inventory slots to the container
        for (int inventoryRowIndex = 0; inventoryRowIndex < PLAYER_INVENTORY_ROWS; ++inventoryRowIndex)
        {
            for (int inventoryColumnIndex = 0; inventoryColumnIndex < PLAYER_INVENTORY_COLUMNS; ++inventoryColumnIndex)
            {
                if (this.tileEntityAlchemicalChest.getState() == 0)
                {
                    this.addSlotToContainer(new Slot(inventoryPlayer, inventoryColumnIndex + inventoryRowIndex * 9 + 9, 35 + inventoryColumnIndex * 18, 104 + inventoryRowIndex * 18));
                }
                else if (this.tileEntityAlchemicalChest.getState() == 1)
                {
                    this.addSlotToContainer(new Slot(inventoryPlayer, inventoryColumnIndex + inventoryRowIndex * 9 + 9, 35 + inventoryColumnIndex * 18, 158 + inventoryRowIndex * 18));
                }
                else if (this.tileEntityAlchemicalChest.getState() == 2)
                {
                    this.addSlotToContainer(new Slot(inventoryPlayer, inventoryColumnIndex + inventoryRowIndex * 9 + 9, 44 + inventoryColumnIndex * 18, 174 + inventoryRowIndex * 18));
                }
            }
        }

        // Add the player's action bar slots to the container
        for (int actionBarSlotIndex = 0; actionBarSlotIndex < PLAYER_INVENTORY_COLUMNS; ++actionBarSlotIndex)
        {
            if (this.tileEntityAlchemicalChest.getState() == 0)
            {
                this.addSlotToContainer(new Slot(inventoryPlayer, actionBarSlotIndex, 35 + actionBarSlotIndex * 18, 162));
            }
            else if (this.tileEntityAlchemicalChest.getState() == 1)
            {
                this.addSlotToContainer(new Slot(inventoryPlayer, actionBarSlotIndex, 35 + actionBarSlotIndex * 18, 216));
            }
            else if (this.tileEntityAlchemicalChest.getState() == 2)
            {
                this.addSlotToContainer(new Slot(inventoryPlayer, actionBarSlotIndex, 44 + actionBarSlotIndex * 18, 232));
            }
        }
    }

    /**
     * Callback for when the crafting gui is closed.
     */
    @Override
    public void onContainerClosed(EntityPlayer entityPlayer)
    {
        super.onContainerClosed(entityPlayer);
        tileEntityAlchemicalChest.closeInventory();
    }

    @Override
    public boolean canInteractWith(EntityPlayer entityPlayer)
    {
        return this.tileEntityAlchemicalChest.isUseableByPlayer(entityPlayer);
    }

    @Override
    public ItemStack transferStackInSlot(EntityPlayer entityPlayer, int slotIndex)
    {
        ItemStack newItemStack = null;
        Slot slot = (Slot) inventorySlots.get(slotIndex);

        if (slot != null && slot.getHasStack())
        {
            ItemStack itemStack = slot.getStack();
            newItemStack = itemStack.copy();

            if (slotIndex < chestInventoryRows * chestInventoryColumns)
            {
                if (!this.mergeItemStack(itemStack, chestInventoryRows * chestInventoryColumns, inventorySlots.size(), false))
                {
                    return null;
                }
            }
            else if (!this.mergeItemStack(itemStack, 0, chestInventoryRows * chestInventoryColumns, false))
            {
                return null;
            }

            if (itemStack.stackSize == 0)
            {
                slot.putStack(null);
            }
            else
            {
                slot.onSlotChanged();
            }
        }

        return newItemStack;
    }
}