    001package net.minecraft.inventory;
    002
    003import cpw.mods.fml.relauncher.Side;
    004import cpw.mods.fml.relauncher.SideOnly;
    005import java.util.List;
    006import java.util.Random;
    007import net.minecraft.block.Block;
    008import net.minecraft.enchantment.EnchantmentData;
    009import net.minecraft.enchantment.EnchantmentHelper;
    010import net.minecraft.entity.player.EntityPlayer;
    011import net.minecraft.entity.player.InventoryPlayer;
    012import net.minecraft.item.Item;
    013import net.minecraft.item.ItemStack;
    014import net.minecraft.world.World;
    015
    016public class ContainerEnchantment extends Container
    017{
    018    /** SlotEnchantmentTable object with ItemStack to be enchanted */
    019    public IInventory tableInventory = new SlotEnchantmentTable(this, "Enchant", 1);
    020
    021    /** current world (for bookshelf counting) */
    022    private World worldPointer;
    023    private int posX;
    024    private int posY;
    025    private int posZ;
    026    private Random rand = new Random();
    027
    028    /** used as seed for EnchantmentNameParts (see GuiEnchantment) */
    029    public long nameSeed;
    030
    031    /** 3-member array storing the enchantment levels of each slot */
    032    public int[] enchantLevels = new int[3];
    033
    034    public ContainerEnchantment(InventoryPlayer par1InventoryPlayer, World par2World, int par3, int par4, int par5)
    035    {
    036        this.worldPointer = par2World;
    037        this.posX = par3;
    038        this.posY = par4;
    039        this.posZ = par5;
    040        this.addSlotToContainer(new SlotEnchantment(this, this.tableInventory, 0, 25, 47));
    041        int var6;
    042
    043        for (var6 = 0; var6 < 3; ++var6)
    044        {
    045            for (int var7 = 0; var7 < 9; ++var7)
    046            {
    047                this.addSlotToContainer(new Slot(par1InventoryPlayer, var7 + var6 * 9 + 9, 8 + var7 * 18, 84 + var6 * 18));
    048            }
    049        }
    050
    051        for (var6 = 0; var6 < 9; ++var6)
    052        {
    053            this.addSlotToContainer(new Slot(par1InventoryPlayer, var6, 8 + var6 * 18, 142));
    054        }
    055    }
    056
    057    public void addCraftingToCrafters(ICrafting par1ICrafting)
    058    {
    059        super.addCraftingToCrafters(par1ICrafting);
    060        par1ICrafting.sendProgressBarUpdate(this, 0, this.enchantLevels[0]);
    061        par1ICrafting.sendProgressBarUpdate(this, 1, this.enchantLevels[1]);
    062        par1ICrafting.sendProgressBarUpdate(this, 2, this.enchantLevels[2]);
    063    }
    064
    065    /**
    066     * Looks for changes made in the container, sends them to every listener.
    067     */
    068    public void detectAndSendChanges()
    069    {
    070        super.detectAndSendChanges();
    071
    072        for (int var1 = 0; var1 < this.crafters.size(); ++var1)
    073        {
    074            ICrafting var2 = (ICrafting)this.crafters.get(var1);
    075            var2.sendProgressBarUpdate(this, 0, this.enchantLevels[0]);
    076            var2.sendProgressBarUpdate(this, 1, this.enchantLevels[1]);
    077            var2.sendProgressBarUpdate(this, 2, this.enchantLevels[2]);
    078        }
    079    }
    080
    081    @SideOnly(Side.CLIENT)
    082    public void updateProgressBar(int par1, int par2)
    083    {
    084        if (par1 >= 0 && par1 <= 2)
    085        {
    086            this.enchantLevels[par1] = par2;
    087        }
    088        else
    089        {
    090            super.updateProgressBar(par1, par2);
    091        }
    092    }
    093
    094    /**
    095     * Callback for when the crafting matrix is changed.
    096     */
    097    public void onCraftMatrixChanged(IInventory par1IInventory)
    098    {
    099        if (par1IInventory == this.tableInventory)
    100        {
    101            ItemStack var2 = par1IInventory.getStackInSlot(0);
    102            int var3;
    103
    104            if (var2 != null && var2.isItemEnchantable())
    105            {
    106                this.nameSeed = this.rand.nextLong();
    107
    108                if (!this.worldPointer.isRemote)
    109                {
    110                    var3 = 0;
    111                    int var4;
    112
    113                    for (var4 = -1; var4 <= 1; ++var4)
    114                    {
    115                        for (int var5 = -1; var5 <= 1; ++var5)
    116                        {
    117                            if ((var4 != 0 || var5 != 0) && this.worldPointer.isAirBlock(this.posX + var5, this.posY, this.posZ + var4) && this.worldPointer.isAirBlock(this.posX + var5, this.posY + 1, this.posZ + var4))
    118                            {
    119                                if (this.worldPointer.getBlockId(this.posX + var5 * 2, this.posY, this.posZ + var4 * 2) == Block.bookShelf.blockID)
    120                                {
    121                                    ++var3;
    122                                }
    123
    124                                if (this.worldPointer.getBlockId(this.posX + var5 * 2, this.posY + 1, this.posZ + var4 * 2) == Block.bookShelf.blockID)
    125                                {
    126                                    ++var3;
    127                                }
    128
    129                                if (var5 != 0 && var4 != 0)
    130                                {
    131                                    if (this.worldPointer.getBlockId(this.posX + var5 * 2, this.posY, this.posZ + var4) == Block.bookShelf.blockID)
    132                                    {
    133                                        ++var3;
    134                                    }
    135
    136                                    if (this.worldPointer.getBlockId(this.posX + var5 * 2, this.posY + 1, this.posZ + var4) == Block.bookShelf.blockID)
    137                                    {
    138                                        ++var3;
    139                                    }
    140
    141                                    if (this.worldPointer.getBlockId(this.posX + var5, this.posY, this.posZ + var4 * 2) == Block.bookShelf.blockID)
    142                                    {
    143                                        ++var3;
    144                                    }
    145
    146                                    if (this.worldPointer.getBlockId(this.posX + var5, this.posY + 1, this.posZ + var4 * 2) == Block.bookShelf.blockID)
    147                                    {
    148                                        ++var3;
    149                                    }
    150                                }
    151                            }
    152                        }
    153                    }
    154
    155                    for (var4 = 0; var4 < 3; ++var4)
    156                    {
    157                        this.enchantLevels[var4] = EnchantmentHelper.calcItemStackEnchantability(this.rand, var4, var3, var2);
    158                    }
    159
    160                    this.detectAndSendChanges();
    161                }
    162            }
    163            else
    164            {
    165                for (var3 = 0; var3 < 3; ++var3)
    166                {
    167                    this.enchantLevels[var3] = 0;
    168                }
    169            }
    170        }
    171    }
    172
    173    /**
    174     * enchants the item on the table using the specified slot; also deducts XP from player
    175     */
    176    public boolean enchantItem(EntityPlayer par1EntityPlayer, int par2)
    177    {
    178        ItemStack var3 = this.tableInventory.getStackInSlot(0);
    179
    180        if (this.enchantLevels[par2] > 0 && var3 != null && (par1EntityPlayer.experienceLevel >= this.enchantLevels[par2] || par1EntityPlayer.capabilities.isCreativeMode))
    181        {
    182            if (!this.worldPointer.isRemote)
    183            {
    184                List var4 = EnchantmentHelper.buildEnchantmentList(this.rand, var3, this.enchantLevels[par2]);
    185                boolean var5 = var3.itemID == Item.book.itemID;
    186
    187                if (var4 != null)
    188                {
    189                    par1EntityPlayer.addExperienceLevel(-this.enchantLevels[par2]);
    190
    191                    if (var5)
    192                    {
    193                        var3.itemID = Item.enchantedBook.itemID;
    194                    }
    195
    196                    int var6 = var5 ? this.rand.nextInt(var4.size()) : -1;
    197
    198                    for (int var7 = 0; var7 < var4.size(); ++var7)
    199                    {
    200                        EnchantmentData var8 = (EnchantmentData)var4.get(var7);
    201
    202                        if (!var5 || var7 == var6)
    203                        {
    204                            if (var5)
    205                            {
    206                                Item.enchantedBook.func_92115_a(var3, var8);
    207                            }
    208                            else
    209                            {
    210                                var3.addEnchantment(var8.enchantmentobj, var8.enchantmentLevel);
    211                            }
    212                        }
    213                    }
    214
    215                    this.onCraftMatrixChanged(this.tableInventory);
    216                }
    217            }
    218
    219            return true;
    220        }
    221        else
    222        {
    223            return false;
    224        }
    225    }
    226
    227    /**
    228     * Callback for when the crafting gui is closed.
    229     */
    230    public void onCraftGuiClosed(EntityPlayer par1EntityPlayer)
    231    {
    232        super.onCraftGuiClosed(par1EntityPlayer);
    233
    234        if (!this.worldPointer.isRemote)
    235        {
    236            ItemStack var2 = this.tableInventory.getStackInSlotOnClosing(0);
    237
    238            if (var2 != null)
    239            {
    240                par1EntityPlayer.dropPlayerItem(var2);
    241            }
    242        }
    243    }
    244
    245    public boolean canInteractWith(EntityPlayer par1EntityPlayer)
    246    {
    247        return this.worldPointer.getBlockId(this.posX, this.posY, this.posZ) != Block.enchantmentTable.blockID ? false : par1EntityPlayer.getDistanceSq((double)this.posX + 0.5D, (double)this.posY + 0.5D, (double)this.posZ + 0.5D) <= 64.0D;
    248    }
    249
    250    /**
    251     * Called when a player shift-clicks on a slot. You must override this or you will crash when someone does that.
    252     */
    253    public ItemStack transferStackInSlot(EntityPlayer par1EntityPlayer, int par2)
    254    {
    255        ItemStack var3 = null;
    256        Slot var4 = (Slot)this.inventorySlots.get(par2);
    257
    258        if (var4 != null && var4.getHasStack())
    259        {
    260            ItemStack var5 = var4.getStack();
    261            var3 = var5.copy();
    262
    263            if (par2 == 0)
    264            {
    265                if (!this.mergeItemStack(var5, 1, 37, true))
    266                {
    267                    return null;
    268                }
    269            }
    270            else
    271            {
    272                if (((Slot)this.inventorySlots.get(0)).getHasStack() || !((Slot)this.inventorySlots.get(0)).isItemValid(var5))
    273                {
    274                    return null;
    275                }
    276
    277                if (var5.hasTagCompound() && var5.stackSize == 1)
    278                {
    279                    ((Slot)this.inventorySlots.get(0)).putStack(var5.copy());
    280                    var5.stackSize = 0;
    281                }
    282                else if (var5.stackSize >= 1)
    283                {
    284                    ((Slot)this.inventorySlots.get(0)).putStack(new ItemStack(var5.itemID, 1, var5.getItemDamage()));
    285                    --var5.stackSize;
    286                }
    287            }
    288
    289            if (var5.stackSize == 0)
    290            {
    291                var4.putStack((ItemStack)null);
    292            }
    293            else
    294            {
    295                var4.onSlotChanged();
    296            }
    297
    298            if (var5.stackSize == var3.stackSize)
    299            {
    300                return null;
    301            }
    302
    303            var4.onPickupFromSlot(par1EntityPlayer, var5);
    304        }
    305
    306        return var3;
    307    }
    308}