1 files changed, 62 insertions, 19 deletions

diff --git a/src/Enchantments.h b/src/Enchantments.h
index f77b535d8..85a316414 100644
--- a/src/Enchantments.h
+++ b/src/Enchantments.h
@@ -8,6 +8,7 @@
 
 #pragma once
 
+#include "Defines.h"
 #include "WorldStorage/EnchantmentSerializer.h"
 
 
@@ -18,6 +19,11 @@ class cFastNBTWriter;
 class cParsedNBT;
 
 
+// fwd:
+struct cWeightedEnchantment;
+
+typedef std::vector<cWeightedEnchantment> cWeightedEnchantments;
+
 
 
 
@@ -28,11 +34,14 @@ mapping each enchantment's id onto its level. ID may be either a number or the e
 Level value of 0 means no such enchantment, and it will not be stored in the m_Enchantments.
 Serialization will never put zero-level enchantments into the stringspec and will always use numeric IDs.
 */
+
+
 // tolua_begin
 class cEnchantments
 {
 public:
-	/// Individual enchantment IDs, corresponding to their NBT IDs ( http://www.minecraftwiki.net/wiki/Data_Values#Enchantment_IDs )
+	/** Individual enchantment IDs, corresponding to their NBT IDs ( http://www.minecraftwiki.net/wiki/Data_Values#Enchantment_IDs )
+	*/
 	
 	enum
 	{
@@ -61,56 +70,90 @@ public:
 		enchLuckOfTheSea         = 61,
 		enchLure                 = 62,
 	} ;
-	
-	/// Creates an empty enchantments container
+
+	/** Creates an empty enchantments container */
 	cEnchantments(void);
 	
-	/// Creates an enchantments container filled with enchantments parsed from stringspec
+	/** Creates an enchantments container filled with enchantments parsed from stringspec */
 	cEnchantments(const AString & a_StringSpec);
 	
-	/// Adds enchantments in the stringspec; if a specified enchantment already exists, overwrites it
+	/** Adds the enchantments contained in a_Other into this object.
+	Existing enchantments are preserved, unless a_Other specifies a different level, in which case the level is changed. */
+	void Add(const cEnchantments & a_Other);
+	
+	/** Adds enchantments in the stringspec; if a specified enchantment already exists, overwrites it */
 	void AddFromString(const AString & a_StringSpec);
 	
-	/// Serializes all the enchantments into a string
+	/** Get the count of enchantments */
+	size_t Count(void);
+	
+	/** Serializes all the enchantments into a string */
 	AString ToString(void) const;
 	
-	/// Returns the level for the specified enchantment; 0 if not stored
+	/** Returns the level for the specified enchantment; 0 if not stored */
 	int GetLevel(int a_EnchantmentID) const;
 	
-	/// Sets the level for the specified enchantment, adding it if not stored before or removing it if level <= 0
+	/** Sets the level for the specified enchantment, adding it if not stored before or removing it if level <= 0 */
 	void SetLevel(int a_EnchantmentID, int a_Level);
 	
-	/// Removes all enchantments
+	/** Removes all enchantments */
 	void Clear(void);
 	
-	/// Returns true if there are no enchantments
+	/** Returns true if there are no enchantments */
 	bool IsEmpty(void) const;
 	
-	/// Converts enchantment name to the numeric representation; returns -1 if enchantment name not found; case insensitive
+	/** Converts enchantment name or ID (number in string) to the numeric representation; returns -1 if enchantment name not found; case insensitive */
 	static int StringToEnchantmentID(const AString & a_EnchantmentName);
 	
-	/// Returns true if a_Other contains exactly the same enchantments and levels
+	/** Returns true if a_Other contains exactly the same enchantments and levels */
 	bool operator ==(const cEnchantments & a_Other) const;
-	
+
 	// tolua_end
+
+	/** Add enchantment weights from item to the vector */
+	static void AddItemEnchantmentWeights(cWeightedEnchantments & a_Enchantments, short a_ItemType, int a_EnchantmentLevel);
+
+	/** Add a enchantment with weight to the vector */
+	static void AddEnchantmentWeightToVector(cWeightedEnchantments & a_Enchantments, int a_Weight, int a_EnchantmentID, int a_EnchantmentLevel);
+	
+	/** Remove the entire enchantment (with weight) from the vector */
+	static void RemoveEnchantmentWeightFromVector(cWeightedEnchantments & a_Enchantments, int a_EnchantmentID);
 	
-	/// Returns true if a_Other doesn't contain exactly the same enchantments and levels
+	/** Remove the entire enchantment (with weight) from the vector */
+	static void RemoveEnchantmentWeightFromVector(cWeightedEnchantments & a_Enchantments, const cEnchantments & a_Enchantment);
+
+	/** Check enchantment conflicts from enchantments from the vector */
+	static void CheckEnchantmentConflictsFromVector(cWeightedEnchantments & a_Enchantments, cEnchantments a_FirstEnchantment);
+
+	/** Gets random enchantment from Vector and returns it */
+	static cEnchantments GetRandomEnchantmentFromVector(cWeightedEnchantments & a_Enchantments);
+
+	/** Returns true if a_Other doesn't contain exactly the same enchantments and levels */
 	bool operator !=(const cEnchantments & a_Other) const;
 	
-	/// Writes the enchantments into the specified NBT writer; begins with the LIST tag of the specified name ("ench" or "StoredEnchantments")
+	/** Writes the enchantments into the specified NBT writer; begins with the LIST tag of the specified name ("ench" or "StoredEnchantments") */
 	friend void EnchantmentSerializer::WriteToNBTCompound(cEnchantments const& a_Enchantments, cFastNBTWriter & a_Writer, const AString & a_ListTagName);
 	
-	/// Reads the enchantments from the specified NBT list tag (ench or StoredEnchantments)
+	/** Reads the enchantments from the specified NBT list tag (ench or StoredEnchantments) */
 	friend void EnchantmentSerializer::ParseFromNBT(cEnchantments& a_Enchantments, const cParsedNBT & a_NBT, int a_EnchListTagIdx);
-	
+
 protected:
-	/// Maps enchantment ID -> enchantment level
+	/** Maps enchantment ID -> enchantment level */
 	typedef std::map<int, int> cMap;
 	
-	/// Currently stored enchantments
+	/** Currently stored enchantments */
 	cMap m_Enchantments;
 } ;  // tolua_export
 
 
 
 
+// Define the cWeightedEnchantment struct for the Enchanting System to store the EnchantmentWeights:
+struct cWeightedEnchantment
+{
+	int m_Weight;
+	cEnchantments m_Enchantments;
+};
+
+
+