133 lines
4.3 KiB
Java
133 lines
4.3 KiB
Java
package mods.tinker.tconstruct.modifiers;
|
|
|
|
import java.util.Iterator;
|
|
import java.util.Map;
|
|
|
|
import mods.tinker.common.ToolMod;
|
|
import mods.tinker.tconstruct.library.Weapon;
|
|
import net.minecraft.enchantment.Enchantment;
|
|
import net.minecraft.enchantment.EnchantmentHelper;
|
|
import net.minecraft.item.ItemStack;
|
|
import net.minecraft.nbt.NBTTagCompound;
|
|
import net.minecraft.nbt.NBTTagList;
|
|
|
|
public class ModLapis extends ToolMod
|
|
{
|
|
String tooltipName;
|
|
int increase;
|
|
int max;
|
|
|
|
public ModLapis(ItemStack[] items, int effect, int inc)
|
|
{
|
|
super(items, effect, "Lapis");
|
|
tooltipName = "\u00a79Luck";
|
|
increase = inc;
|
|
max = 100;
|
|
}
|
|
|
|
@Override
|
|
protected boolean canModify (ItemStack tool, ItemStack[] input)
|
|
{
|
|
NBTTagCompound tags = tool.getTagCompound().getCompoundTag("InfiTool");
|
|
if (!tags.hasKey(key))
|
|
return tags.getInteger("Modifiers") > 0;
|
|
|
|
int keyPair[] = tags.getIntArray(key);
|
|
if (keyPair[0] + increase <= 100)
|
|
return true;
|
|
else
|
|
return false;
|
|
}
|
|
|
|
@Override
|
|
public void modify (ItemStack[] input, ItemStack tool)
|
|
{
|
|
NBTTagCompound tags = tool.getTagCompound().getCompoundTag("InfiTool");
|
|
if (tags.hasKey(key))
|
|
{
|
|
|
|
int keyPair[] = tags.getIntArray(key);
|
|
keyPair[0] += increase;
|
|
tags.setIntArray(key, keyPair);
|
|
if (tool.getItem() instanceof Weapon)
|
|
{
|
|
if (keyPair[0] >= 100)
|
|
addEnchantment(tool, Enchantment.looting, 3);
|
|
else if (keyPair[0] >= 50)
|
|
addEnchantment(tool, Enchantment.looting, 2);
|
|
else if (keyPair[0] >= 20)
|
|
addEnchantment(tool, Enchantment.looting, 1);
|
|
}
|
|
else
|
|
{
|
|
if (keyPair[0] >= 100)
|
|
addEnchantment(tool, Enchantment.fortune, 3);
|
|
else if (keyPair[0] >= 50)
|
|
addEnchantment(tool, Enchantment.fortune, 2);
|
|
else if (keyPair[0] >= 20)
|
|
addEnchantment(tool, Enchantment.fortune, 1);
|
|
}
|
|
|
|
updateModTag(tool, keyPair);
|
|
}
|
|
else
|
|
{
|
|
tags.setBoolean(key, true);
|
|
|
|
String modName = "\u00a79Lapis (0/100)";
|
|
int tooltipIndex = addToolTip(tool, "\u00a79Luck", modName);
|
|
int[] keyPair = new int[] { 0, tooltipIndex };
|
|
tags.setIntArray(key, keyPair);
|
|
|
|
int modifiers = tags.getInteger("Modifiers");
|
|
modifiers -= 1;
|
|
tags.setInteger("Modifiers", modifiers);
|
|
}
|
|
}
|
|
|
|
public void addEnchantment(ItemStack tool, Enchantment enchant, int level)
|
|
{
|
|
NBTTagList tags = new NBTTagList("ench");
|
|
Map enchantMap = EnchantmentHelper.getEnchantments(tool);
|
|
Iterator iterator = enchantMap.keySet().iterator();
|
|
int index;
|
|
int lvl;
|
|
boolean hasEnchant = false;
|
|
while (iterator.hasNext())
|
|
{
|
|
NBTTagCompound enchantTag = new NBTTagCompound();
|
|
index = ((Integer)iterator.next()).intValue();
|
|
lvl = (Integer) enchantMap.get(index);
|
|
if (index == enchant.effectId)
|
|
{
|
|
hasEnchant = true;
|
|
enchantTag.setShort("id", (short)index);
|
|
enchantTag.setShort("lvl", (short)((byte)level));
|
|
tags.appendTag(enchantTag);
|
|
}
|
|
else
|
|
{
|
|
enchantTag.setShort("id", (short)index);
|
|
enchantTag.setShort("lvl", (short)((byte)lvl));
|
|
tags.appendTag(enchantTag);
|
|
}
|
|
}
|
|
if (!hasEnchant)
|
|
{
|
|
NBTTagCompound enchantTag = new NBTTagCompound();
|
|
enchantTag.setShort("id", (short)enchant.effectId);
|
|
enchantTag.setShort("lvl", (short)((byte)level));
|
|
tags.appendTag(enchantTag);
|
|
}
|
|
tool.stackTagCompound.setTag("ench", tags);
|
|
}
|
|
|
|
void updateModTag (ItemStack tool, int[] keys)
|
|
{
|
|
NBTTagCompound tags = tool.getTagCompound().getCompoundTag("InfiTool");
|
|
String tip = "ModifierTip"+keys[1];
|
|
String modName = "\u00a79Lapis ("+keys[0]+"/100)";
|
|
tags.setString(tip, modName);
|
|
}
|
|
}
|