CustomNPC-Plus/src/main/java/noppes/npcs/containers/ContainerNPCCompanion.java

package noppes.npcs.containers;

import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.inventory.Slot;
import net.minecraft.item.ItemStack;
import noppes.npcs.constants.EnumCompanionTalent;
import noppes.npcs.entity.EntityNPCInterface;
import noppes.npcs.roles.RoleCompanion;

public class ContainerNPCCompanion extends ContainerNpcInterface {
    public InventoryNPC currencyMatrix;
    public RoleCompanion role;

    public ContainerNPCCompanion(EntityNPCInterface npc, EntityPlayer player) {
        super(player);
        role = (RoleCompanion) npc.roleInterface;
        for (int k = 0; k < 3; k++) {
            for (int j1 = 0; j1 < 9; j1++) {
                addSlotToContainer(new Slot(player.inventory, j1 + k * 9 + 9, 6 + j1 * 18, 87 + k * 18));
            }
        }

        for (int l = 0; l < 9; l++) {
            addSlotToContainer(new Slot(player.inventory, l, 6 + l * 18, 145));
        }

        if (role.talents.containsKey(EnumCompanionTalent.INVENTORY)) {
            int size = (role.getTalentLevel(EnumCompanionTalent.INVENTORY) + 1) * 2;
            for (int i = 0; i < size; i++) {
                addSlotToContainer(new Slot(role.inventory, i, 114 + i % 3 * 18, 8 + i / 3 * 18));
            }
        }
        if (role.getTalentLevel(EnumCompanionTalent.ARMOR) > 0) {
            for (int l = 0; l < 4; l++) {
                addSlotToContainer(new SlotCompanionArmor(role, npc.inventory, l, 6, 8 + l * 18, l));
            }
        }
        if (role.getTalentLevel(EnumCompanionTalent.SWORD) > 0) {
            addSlotToContainer(new SlotCompanionWeapon(role, npc.inventory, 4, 79, 17));
        }
    }

    @Override
    public ItemStack transferStackInSlot(EntityPlayer par1EntityPlayer, int i) {
        return null;
    }

    @Override
    public void onContainerClosed(EntityPlayer entityplayer) {
        super.onContainerClosed(entityplayer);
        role.setStats();
    }
}