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