1 files changed, 814 insertions, 13 deletions

diff --git a/src/Enchantments.cpp b/src/Enchantments.cpp
index 1d8188e96..264878c22 100644
--- a/src/Enchantments.cpp
+++ b/src/Enchantments.cpp
@@ -5,6 +5,7 @@
 #include "Globals.h"
 #include "Enchantments.h"
 #include "WorldStorage/FastNBT.h"
+#include "FastRandom.h"
 
 
 
@@ -28,6 +29,18 @@ cEnchantments::cEnchantments(const AString & a_StringSpec)
 
 
 
+void cEnchantments::Add(const cEnchantments & a_Other)
+{
+	for (cEnchantments::cMap::const_iterator itr = a_Other.m_Enchantments.begin(), end = a_Other.m_Enchantments.end(); itr != end; ++itr)
+	{
+		SetLevel(itr->first, itr->second);
+	}  // for itr - a_Other.m_Enchantments[]
+}
+
+
+
+
+
 void cEnchantments::AddFromString(const AString & a_StringSpec)
 {
 	// Add enchantments in the stringspec; if a specified enchantment already exists, overwrites it
@@ -49,21 +62,17 @@ void cEnchantments::AddFromString(const AString & a_StringSpec)
 			LOG("%s: Malformed enchantment decl: \"%s\", skipping.", __FUNCTION__, itr->c_str());
 			continue;
 		}
-		int id = atoi(Split[0].c_str());
-		if ((id == 0) && (Split[0] != "0"))
+		int id = StringToEnchantmentID(Split[0]);
+		if (id < 0)
 		{
-			id = StringToEnchantmentID(Split[0]);
+			LOG("%s: Failed to parse enchantment \"%s\", skipping.", __FUNCTION__, Split[0].c_str());
+			continue;
 		}
 		int lvl = atoi(Split[1].c_str());
-		if (
-			((id  <= 0) && (Split[0] != "0")) ||
-			((lvl == 0) && (Split[1] != "0"))
-		)
-		{
-			// Numbers failed to parse
-			LOG("%s: Failed to parse enchantment declaration for numbers: \"%s\" and \"%s\", skipping.",
-				__FUNCTION__, Split[0].c_str(), Split[1].c_str()
-			);
+		if ((lvl == 0) && (Split[1] != "0"))
+		{
+			// Level failed to parse
+			LOG("%s: Failed to parse enchantment level \"%s\", skipping.", __FUNCTION__, Split[1].c_str());
 			continue;
 		}
 		SetLevel(id, lvl);
@@ -74,6 +83,15 @@ void cEnchantments::AddFromString(const AString & a_StringSpec)
 
 
 
+size_t cEnchantments::Count(void)
+{
+	return m_Enchantments.size();
+}
+
+
+
+
+
 AString cEnchantments::ToString(void) const
 {
 	// Serialize all the enchantments into a string
@@ -150,7 +168,7 @@ bool cEnchantments::IsEmpty(void) const
 
 int cEnchantments::StringToEnchantmentID(const AString & a_EnchantmentName)
 {
-	struct
+	static const struct
 	{
 		int m_Value;
 		const char * m_Name;
@@ -181,6 +199,15 @@ int cEnchantments::StringToEnchantmentID(const AString & a_EnchantmentName)
 		{ enchLuckOfTheSea,         "LuckOfTheSea"},
 		{ enchLure,                 "Lure"},
 	} ;
+
+	// First try to parse as a number:
+	int id = atoi(a_EnchantmentName.c_str());
+	if ((id != 0) || (a_EnchantmentName == "0"))
+	{
+		return id;
+	}
+
+	// It wasn't a number, do a lookup:
 	for (size_t i = 0; i < ARRAYCOUNT(EnchantmentNames); i++)
 	{
 		if (NoCaseCompare(EnchantmentNames[i].m_Name, a_EnchantmentName) == 0)
@@ -213,7 +240,781 @@ bool cEnchantments::operator !=(const cEnchantments & a_Other) const
 
 
 
+void cEnchantments::AddItemEnchantmentWeights(cWeightedEnchantments & a_Enchantments, short a_ItemType, int a_EnchantmentLevel)
+{
+	if (ItemCategory::IsSword(a_ItemType))
+	{
+		// Sharpness
+		if ((a_EnchantmentLevel >= 34) && (a_EnchantmentLevel <= 54))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchSharpness, 4);
+		}
+		else if ((a_EnchantmentLevel >= 23) && (a_EnchantmentLevel <= 43))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchSharpness, 3);
+		}
+		else if ((a_EnchantmentLevel >= 12) && (a_EnchantmentLevel <= 32))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchSharpness, 2);
+		}
+		else if ((a_EnchantmentLevel >= 1) && (a_EnchantmentLevel <= 21))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchSharpness, 1);
+		}
+
+		// Smite
+		if ((a_EnchantmentLevel >= 29) && (a_EnchantmentLevel <= 49))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchSmite, 4);
+		}
+		else if ((a_EnchantmentLevel >= 21) && (a_EnchantmentLevel <= 41))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchSmite, 3);
+		}
+		else if ((a_EnchantmentLevel >= 13) && (a_EnchantmentLevel <= 33))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchSmite, 2);
+		}
+		else if ((a_EnchantmentLevel >= 5) && (a_EnchantmentLevel <= 25))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchSmite, 1);
+		}
+
+		// Bane of Arthropods
+		if ((a_EnchantmentLevel >= 29) && (a_EnchantmentLevel <= 49))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchBaneOfArthropods, 4);
+		}
+		else if ((a_EnchantmentLevel >= 21) && (a_EnchantmentLevel <= 41))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchBaneOfArthropods, 3);
+		}
+		else if ((a_EnchantmentLevel >= 13) && (a_EnchantmentLevel <= 33))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchBaneOfArthropods, 2);
+		}
+		else if ((a_EnchantmentLevel >= 5) && (a_EnchantmentLevel <= 25))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchBaneOfArthropods, 1);
+		}
+
+		// Knockback
+		if ((a_EnchantmentLevel >= 25) && (a_EnchantmentLevel <= 75))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchKnockback, 2);
+		}
+		else if ((a_EnchantmentLevel >= 5) && (a_EnchantmentLevel <= 55))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchKnockback, 1);
+		}
+
+		// Fire Aspect
+		if ((a_EnchantmentLevel >= 30) && (a_EnchantmentLevel <= 80))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchFireAspect, 2);
+		}
+		else if ((a_EnchantmentLevel >= 10) && (a_EnchantmentLevel <= 60))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchFireAspect, 1);
+		}
+
+		// Looting
+		if ((a_EnchantmentLevel >= 33) && (a_EnchantmentLevel <= 83))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchLooting, 3);
+		}
+		else if ((a_EnchantmentLevel >= 24) && (a_EnchantmentLevel <= 74))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchLooting, 2);
+		}
+		else if ((a_EnchantmentLevel >= 15) && (a_EnchantmentLevel <= 65))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchLooting, 1);
+		}
+	}
+
+	else if (ItemCategory::IsTool(a_ItemType))
+	{
+		// Efficiency
+		if ((a_EnchantmentLevel >= 31) && (a_EnchantmentLevel <= 81))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchEfficiency, 4);
+		}
+		else if ((a_EnchantmentLevel >= 21) && (a_EnchantmentLevel <= 71))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchEfficiency, 3);
+		}
+		else if ((a_EnchantmentLevel >= 11) && (a_EnchantmentLevel <= 61))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchEfficiency, 2);
+		}
+		else if ((a_EnchantmentLevel >= 1) && (a_EnchantmentLevel <= 51))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchEfficiency, 1);
+		}
+
+		// Silk Touch
+		if ((a_EnchantmentLevel >= 15) && (a_EnchantmentLevel <= 65))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchSilkTouch, 1);
+		}
+
+		// Fortune
+		if ((a_EnchantmentLevel >= 33) && (a_EnchantmentLevel <= 83))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchFortune, 3);
+		}
+		else if ((a_EnchantmentLevel >= 24) && (a_EnchantmentLevel <= 74))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchFortune, 2);
+		}
+		else if ((a_EnchantmentLevel >= 15) && (a_EnchantmentLevel <= 65))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchFortune, 1);
+		}
+	}
+
+	else if (ItemCategory::IsArmor(a_ItemType))
+	{
+		// Protection
+		if ((a_EnchantmentLevel >= 34) && (a_EnchantmentLevel <= 54))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchProtection, 4);
+		}
+		else if ((a_EnchantmentLevel >= 23) && (a_EnchantmentLevel <= 43))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchProtection, 3);
+		}
+		else if ((a_EnchantmentLevel >= 12) && (a_EnchantmentLevel <= 32))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchProtection, 2);
+		}
+		else if ((a_EnchantmentLevel >= 1) && (a_EnchantmentLevel <= 21))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchProtection, 1);
+		}
 
+		// Fire Protection
+		if ((a_EnchantmentLevel >= 34) && (a_EnchantmentLevel <= 46))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchFireProtection, 4);
+		}
+		else if ((a_EnchantmentLevel >= 26) && (a_EnchantmentLevel <= 38))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchFireProtection, 3);
+		}
+		else if ((a_EnchantmentLevel >= 18) && (a_EnchantmentLevel <= 30))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchFireProtection, 2);
+		}
+		else if ((a_EnchantmentLevel >= 10) && (a_EnchantmentLevel <= 22))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchFireProtection, 1);
+		}
+
+		// Blast Protection
+		if ((a_EnchantmentLevel >= 29) && (a_EnchantmentLevel <= 41))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchBlastProtection, 4);
+		}
+		else if ((a_EnchantmentLevel >= 21) && (a_EnchantmentLevel <= 33))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchBlastProtection, 3);
+		}
+		else if ((a_EnchantmentLevel >= 13) && (a_EnchantmentLevel <= 25))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchBlastProtection, 2);
+		}
+		else if ((a_EnchantmentLevel >= 5) && (a_EnchantmentLevel <= 17))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchBlastProtection, 1);
+		}
+
+		// Projectile Protection
+		if ((a_EnchantmentLevel >= 21) && (a_EnchantmentLevel <= 36))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchProjectileProtection, 4);
+		}
+		else if ((a_EnchantmentLevel >= 15) && (a_EnchantmentLevel <= 30))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchProjectileProtection, 3);
+		}
+		else if ((a_EnchantmentLevel >= 9) && (a_EnchantmentLevel <= 24))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchProjectileProtection, 2);
+		}
+		else if ((a_EnchantmentLevel >= 3) && (a_EnchantmentLevel <= 18))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchProjectileProtection, 1);
+		}
+
+		// Thorns
+		if ((a_EnchantmentLevel >= 50) && (a_EnchantmentLevel <= 100))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchThorns, 3);
+		}
+		else if ((a_EnchantmentLevel >= 30) && (a_EnchantmentLevel <= 80))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchThorns, 2);
+		}
+		else if ((a_EnchantmentLevel >= 10) && (a_EnchantmentLevel <= 60))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchThorns, 1);
+		}
+
+
+		if (ItemCategory::IsHelmet(a_ItemType))
+		{
+			// Respiration
+			if ((a_EnchantmentLevel >= 30) && (a_EnchantmentLevel <= 60))
+			{
+				AddEnchantmentWeightToVector(a_Enchantments, 2, enchRespiration, 3);
+			}
+			else if ((a_EnchantmentLevel >= 20) && (a_EnchantmentLevel <= 50))
+			{
+				AddEnchantmentWeightToVector(a_Enchantments, 2, enchRespiration, 2);
+			}
+			else if ((a_EnchantmentLevel >= 10) && (a_EnchantmentLevel <= 40))
+			{
+				AddEnchantmentWeightToVector(a_Enchantments, 2, enchRespiration, 1);
+			}
+
+			// Aqua Affinity
+			if ((a_EnchantmentLevel >= 1) && (a_EnchantmentLevel <= 41))
+			{
+				AddEnchantmentWeightToVector(a_Enchantments, 2, enchAquaAffinity, 1);
+			}
+		}
+
+		else if (ItemCategory::IsBoots(a_ItemType))
+		{
+			// Feather Fall
+			if ((a_EnchantmentLevel >= 23) && (a_EnchantmentLevel <= 33))
+			{
+				AddEnchantmentWeightToVector(a_Enchantments, 5, enchFeatherFalling, 4);
+			}
+			else if ((a_EnchantmentLevel >= 17) && (a_EnchantmentLevel <= 27))
+			{
+				AddEnchantmentWeightToVector(a_Enchantments, 5, enchFeatherFalling, 3);
+			}
+			else if ((a_EnchantmentLevel >= 11) && (a_EnchantmentLevel <= 21))
+			{
+				AddEnchantmentWeightToVector(a_Enchantments, 5, enchFeatherFalling, 2);
+			}
+			else if ((a_EnchantmentLevel >= 5) && (a_EnchantmentLevel <= 15))
+			{
+				AddEnchantmentWeightToVector(a_Enchantments, 5, enchFeatherFalling, 1);
+			}
+		}
+	}
+
+	else if (a_ItemType == E_ITEM_BOW)
+	{
+		// Power
+		if ((a_EnchantmentLevel >= 31) && (a_EnchantmentLevel <= 46))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchPower, 4);
+		}
+		else if ((a_EnchantmentLevel >= 21) && (a_EnchantmentLevel <= 36))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchPower, 3);
+		}
+		else if ((a_EnchantmentLevel >= 11) && (a_EnchantmentLevel <= 26))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchPower, 2);
+		}
+		else if ((a_EnchantmentLevel >= 1) && (a_EnchantmentLevel <= 16))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchPower, 1);
+		}
+
+		// Punch
+		if ((a_EnchantmentLevel >= 32) && (a_EnchantmentLevel <= 57))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchPunch, 2);
+		}
+		else if ((a_EnchantmentLevel >= 12) && (a_EnchantmentLevel <= 37))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchPunch, 1);
+		}
+
+		// Flame and Infinity
+		if ((a_EnchantmentLevel >= 20) && (a_EnchantmentLevel <= 50))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchFlame, 1);
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchInfinity, 1);
+		}
+	}
+
+	else if (a_ItemType == E_ITEM_FISHING_ROD)
+	{
+		// Luck of the Sea and Lure
+		if ((a_EnchantmentLevel >= 33) && (a_EnchantmentLevel <= 83))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchLuckOfTheSea, 3);
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchLure, 3);
+		}
+		else if ((a_EnchantmentLevel >= 24) && (a_EnchantmentLevel <= 74))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchLuckOfTheSea, 2);
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchLure, 2);
+		}
+		else if ((a_EnchantmentLevel >= 15) && (a_EnchantmentLevel <= 65))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchLuckOfTheSea, 1);
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchLure, 1);
+		}
+	}
+
+	else if (a_ItemType == E_ITEM_BOOK)
+	{
+		// All Enchantments
+
+		// Sharpness
+		if ((a_EnchantmentLevel >= 34) && (a_EnchantmentLevel <= 54))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchSharpness, 4);
+		}
+		else if ((a_EnchantmentLevel >= 23) && (a_EnchantmentLevel <= 43))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchSharpness, 3);
+		}
+		else if ((a_EnchantmentLevel >= 12) && (a_EnchantmentLevel <= 32))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchSharpness, 2);
+		}
+		else if ((a_EnchantmentLevel >= 1) && (a_EnchantmentLevel <= 21))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchSharpness, 1);
+		}
+
+		// Smite
+		if ((a_EnchantmentLevel >= 29) && (a_EnchantmentLevel <= 49))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchSmite, 4);
+		}
+		else if ((a_EnchantmentLevel >= 21) && (a_EnchantmentLevel <= 41))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchSmite, 3);
+		}
+		else if ((a_EnchantmentLevel >= 13) && (a_EnchantmentLevel <= 33))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchSmite, 2);
+		}
+		else if ((a_EnchantmentLevel >= 5) && (a_EnchantmentLevel <= 25))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchSmite, 1);
+		}
+
+		// Bane of Arthropods
+		if ((a_EnchantmentLevel >= 29) && (a_EnchantmentLevel <= 49))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchBaneOfArthropods, 4);
+		}
+		else if ((a_EnchantmentLevel >= 21) && (a_EnchantmentLevel <= 41))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchBaneOfArthropods, 3);
+		}
+		else if ((a_EnchantmentLevel >= 13) && (a_EnchantmentLevel <= 33))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchBaneOfArthropods, 2);
+		}
+		else if ((a_EnchantmentLevel >= 5) && (a_EnchantmentLevel <= 25))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchBaneOfArthropods, 1);
+		}
+
+		// Knockback
+		if ((a_EnchantmentLevel >= 25) && (a_EnchantmentLevel <= 75))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchKnockback, 2);
+		}
+		else if ((a_EnchantmentLevel >= 5) && (a_EnchantmentLevel <= 55))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchKnockback, 1);
+		}
+
+		// Fire Aspect
+		if ((a_EnchantmentLevel >= 30) && (a_EnchantmentLevel <= 80))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchFireAspect, 2);
+		}
+		else if ((a_EnchantmentLevel >= 10) && (a_EnchantmentLevel <= 60))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchFireAspect, 1);
+		}
+
+		// Looting
+		if ((a_EnchantmentLevel >= 33) && (a_EnchantmentLevel <= 83))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchLooting, 3);
+		}
+		else if ((a_EnchantmentLevel >= 24) && (a_EnchantmentLevel <= 74))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchLooting, 2);
+		}
+		else if ((a_EnchantmentLevel >= 15) && (a_EnchantmentLevel <= 65))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchLooting, 1);
+		}
+
+		// Efficiency
+		if ((a_EnchantmentLevel >= 31) && (a_EnchantmentLevel <= 81))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchEfficiency, 4);
+		}
+		else if ((a_EnchantmentLevel >= 21) && (a_EnchantmentLevel <= 71))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchEfficiency, 3);
+		}
+		else if ((a_EnchantmentLevel >= 11) && (a_EnchantmentLevel <= 61))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchEfficiency, 2);
+		}
+		else if ((a_EnchantmentLevel >= 1) && (a_EnchantmentLevel <= 51))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchEfficiency, 1);
+		}
+
+		// Silk Touch
+		if ((a_EnchantmentLevel >= 15) && (a_EnchantmentLevel <= 65))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchSilkTouch, 1);
+		}
+
+		// Fortune
+		if ((a_EnchantmentLevel >= 33) && (a_EnchantmentLevel <= 83))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchFortune, 3);
+		}
+		else if ((a_EnchantmentLevel >= 24) && (a_EnchantmentLevel <= 74))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchFortune, 2);
+		}
+		else if ((a_EnchantmentLevel >= 15) && (a_EnchantmentLevel <= 65))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchFortune, 1);
+		}
+
+		// Protection
+		if ((a_EnchantmentLevel >= 34) && (a_EnchantmentLevel <= 54))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchProtection, 4);
+		}
+		else if ((a_EnchantmentLevel >= 23) && (a_EnchantmentLevel <= 43))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchProtection, 3);
+		}
+		else if ((a_EnchantmentLevel >= 12) && (a_EnchantmentLevel <= 32))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchProtection, 2);
+		}
+		else if ((a_EnchantmentLevel >= 1) && (a_EnchantmentLevel <= 21))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchProtection, 1);
+		}
+
+		// Fire Protection
+		if ((a_EnchantmentLevel >= 34) && (a_EnchantmentLevel <= 46))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchFireProtection, 4);
+		}
+		else if ((a_EnchantmentLevel >= 26) && (a_EnchantmentLevel <= 38))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchFireProtection, 3);
+		}
+		else if ((a_EnchantmentLevel >= 18) && (a_EnchantmentLevel <= 30))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchFireProtection, 2);
+		}
+		else if ((a_EnchantmentLevel >= 10) && (a_EnchantmentLevel <= 22))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchFireProtection, 1);
+		}
+
+		// Blast Protection
+		if ((a_EnchantmentLevel >= 29) && (a_EnchantmentLevel <= 41))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchBlastProtection, 4);
+		}
+		else if ((a_EnchantmentLevel >= 21) && (a_EnchantmentLevel <= 33))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchBlastProtection, 3);
+		}
+		else if ((a_EnchantmentLevel >= 13) && (a_EnchantmentLevel <= 25))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchBlastProtection, 2);
+		}
+		else if ((a_EnchantmentLevel >= 5) && (a_EnchantmentLevel <= 17))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchBlastProtection, 1);
+		}
+
+		// Projectile Protection
+		if ((a_EnchantmentLevel >= 21) && (a_EnchantmentLevel <= 36))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchProjectileProtection, 4);
+		}
+		else if ((a_EnchantmentLevel >= 15) && (a_EnchantmentLevel <= 30))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchProjectileProtection, 3);
+		}
+		else if ((a_EnchantmentLevel >= 9) && (a_EnchantmentLevel <= 24))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchProjectileProtection, 2);
+		}
+		else if ((a_EnchantmentLevel >= 3) && (a_EnchantmentLevel <= 18))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchProjectileProtection, 1);
+		}
+
+		// Thorns
+		if ((a_EnchantmentLevel >= 50) && (a_EnchantmentLevel <= 100))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchThorns, 3);
+		}
+		else if ((a_EnchantmentLevel >= 30) && (a_EnchantmentLevel <= 80))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchThorns, 2);
+		}
+		else if ((a_EnchantmentLevel >= 10) && (a_EnchantmentLevel <= 60))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchThorns, 1);
+		}
+
+		// Respiration
+		if ((a_EnchantmentLevel >= 30) && (a_EnchantmentLevel <= 60))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchRespiration, 3);
+		}
+		else if ((a_EnchantmentLevel >= 20) && (a_EnchantmentLevel <= 50))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchRespiration, 2);
+		}
+		else if ((a_EnchantmentLevel >= 10) && (a_EnchantmentLevel <= 40))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchRespiration, 1);
+		}
+
+		// Aqua Affinity
+		if ((a_EnchantmentLevel >= 1) && (a_EnchantmentLevel <= 41))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchAquaAffinity, 1);
+		}
+
+		// Feather Fall
+		if ((a_EnchantmentLevel >= 23) && (a_EnchantmentLevel <= 33))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchFeatherFalling, 4);
+		}
+		else if ((a_EnchantmentLevel >= 17) && (a_EnchantmentLevel <= 27))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchFeatherFalling, 3);
+		}
+		else if ((a_EnchantmentLevel >= 11) && (a_EnchantmentLevel <= 21))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchFeatherFalling, 2);
+		}
+		else if ((a_EnchantmentLevel >= 5) && (a_EnchantmentLevel <= 15))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 5, enchFeatherFalling, 1);
+		}
+
+		// Power
+		if ((a_EnchantmentLevel >= 31) && (a_EnchantmentLevel <= 46))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchPower, 4);
+		}
+		else if ((a_EnchantmentLevel >= 21) && (a_EnchantmentLevel <= 36))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchPower, 3);
+		}
+		else if ((a_EnchantmentLevel >= 11) && (a_EnchantmentLevel <= 26))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchPower, 2);
+		}
+		else if ((a_EnchantmentLevel >= 1) && (a_EnchantmentLevel <= 16))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 10, enchPower, 1);
+		}
+
+		// Punch
+		if ((a_EnchantmentLevel >= 32) && (a_EnchantmentLevel <= 57))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchPunch, 2);
+		}
+		else if ((a_EnchantmentLevel >= 12) && (a_EnchantmentLevel <= 37))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchPunch, 1);
+		}
+
+		// Flame and Infinity
+		if ((a_EnchantmentLevel >= 20) && (a_EnchantmentLevel <= 50))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 2, enchFlame, 1);
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchInfinity, 1);
+		}
+
+		// Luck of the Sea and Lure
+		if ((a_EnchantmentLevel >= 33) && (a_EnchantmentLevel <= 83))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchLuckOfTheSea, 3);
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchLure, 3);
+		}
+		else if ((a_EnchantmentLevel >= 24) && (a_EnchantmentLevel <= 74))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchLuckOfTheSea, 2);
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchLure, 2);
+		}
+		else if ((a_EnchantmentLevel >= 15) && (a_EnchantmentLevel <= 65))
+		{
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchLuckOfTheSea, 1);
+			AddEnchantmentWeightToVector(a_Enchantments, 1, enchLure, 1);
+		}
+	}
+
+	// Unbreaking
+	if ((a_EnchantmentLevel >= 21) && (a_EnchantmentLevel <= 71))
+	{
+		AddEnchantmentWeightToVector(a_Enchantments, 5, enchUnbreaking, 3);
+	}
+	else if ((a_EnchantmentLevel >= 13) && (a_EnchantmentLevel <= 63))
+	{
+		AddEnchantmentWeightToVector(a_Enchantments, 5, enchUnbreaking, 2);
+	}
+	else if ((a_EnchantmentLevel >= 5) && (a_EnchantmentLevel <= 55))
+	{
+		AddEnchantmentWeightToVector(a_Enchantments, 5, enchUnbreaking, 1);
+	}
+}
+
+
+
+
+
+void cEnchantments::AddEnchantmentWeightToVector(cWeightedEnchantments & a_Enchantments, int a_Weight, int a_EnchantmentID, int a_EnchantmentLevel)
+{
+	cWeightedEnchantment weightedenchantment;
+	weightedenchantment.m_Weight = a_Weight;
+	cEnchantments enchantment;
+	enchantment.SetLevel(a_EnchantmentID, a_EnchantmentLevel);
+	weightedenchantment.m_Enchantments = enchantment;
+	a_Enchantments.push_back(weightedenchantment);
+}
+
+
+
+
+
+void cEnchantments::RemoveEnchantmentWeightFromVector(cWeightedEnchantments & a_Enchantments, int a_EnchantmentID)
+{
+	for (cWeightedEnchantments::iterator it = a_Enchantments.begin(); it != a_Enchantments.end(); ++it)
+	{
+		if ((*it).m_Enchantments.GetLevel(a_EnchantmentID) > 0)
+		{
+			a_Enchantments.erase(it);
+			break;
+		}
+	}
+}
+
+
+
+
+
+void cEnchantments::RemoveEnchantmentWeightFromVector(cWeightedEnchantments & a_Enchantments, const cEnchantments & a_Enchantment)
+{
+	for (cWeightedEnchantments::iterator it = a_Enchantments.begin(); it != a_Enchantments.end(); ++it)
+	{
+		if ((*it).m_Enchantments == a_Enchantment)
+		{
+			a_Enchantments.erase(it);
+			break;
+		}
+	}
+}
+
+
+
+
+
+void cEnchantments::CheckEnchantmentConflictsFromVector(cWeightedEnchantments & a_Enchantments, cEnchantments a_FirstEnchantment)
+{
+	if (a_FirstEnchantment.GetLevel(cEnchantments::enchProtection) > 0)
+	{
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchFireProtection);
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchBlastProtection);
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchProjectileProtection);
+	}
+	else if (a_FirstEnchantment.GetLevel(cEnchantments::enchFireProtection) > 0)
+	{
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchProtection);
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchBlastProtection);
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchProjectileProtection);
+	}
+	else if (a_FirstEnchantment.GetLevel(cEnchantments::enchBlastProtection) > 0)
+	{
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchProtection);
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchFireProtection);
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchProjectileProtection);
+	}
+	else if (a_FirstEnchantment.GetLevel(cEnchantments::enchProjectileProtection) > 0)
+	{
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchProtection);
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchFireProtection);
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchBlastProtection);
+	}
+
+	else if (a_FirstEnchantment.GetLevel(cEnchantments::enchSharpness) > 0)
+	{
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchSmite);
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchBaneOfArthropods);
+	}
+	else if (a_FirstEnchantment.GetLevel(cEnchantments::enchSmite) > 0)
+	{
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchSharpness);
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchBaneOfArthropods);
+	}
+	else if (a_FirstEnchantment.GetLevel(cEnchantments::enchBaneOfArthropods) > 0)
+	{
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchSharpness);
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchSmite);
+	}
+	else if (a_FirstEnchantment.GetLevel(cEnchantments::enchSilkTouch) > 0)
+	{
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchFortune);
+	}
+	else if (a_FirstEnchantment.GetLevel(cEnchantments::enchFortune) > 0)
+	{
+		RemoveEnchantmentWeightFromVector(a_Enchantments, cEnchantments::enchSilkTouch);
+	}
+}
+
+
+
+
+
+cEnchantments cEnchantments::GetRandomEnchantmentFromVector(cWeightedEnchantments & a_Enchantments)
+{
+	cFastRandom Random;
+
+	int AllWeights = 0;
+	for (cWeightedEnchantments::iterator it = a_Enchantments.begin(); it != a_Enchantments.end(); ++it)
+	{
+		AllWeights += (*it).m_Weight;
+	}
+	int RandomNumber = Random.GenerateRandomInteger(0, AllWeights - 1);
+	for (cWeightedEnchantments::iterator it = a_Enchantments.begin(); it != a_Enchantments.end(); ++it)
+	{
+		RandomNumber -= (*it).m_Weight;
+		if (RandomNumber < 0)
+		{
+			return (*it).m_Enchantments;
+		}
+	}
+
+	return cEnchantments();
+}