1 files changed, 231 insertions, 0 deletions

diff --git a/source/Enchantments.cpp b/source/Enchantments.cpp
new file mode 100644
index 000000000..69bf8f7eb
--- /dev/null
+++ b/source/Enchantments.cpp
@@ -0,0 +1,231 @@
+

+// Enchantments.cpp

+

+// Implements the cEnchantments class representing a storage for item enchantments and stored-enchantments

+

+#include "Globals.h"

+#include "Enchantments.h"

+#include "WorldStorage/FastNBT.h"

+

+

+

+

+

+cEnchantments::cEnchantments(void)

+{

+	// Nothing needed yet, but the constructor needs to be declared and impemented in order to be usable

+}

+

+

+

+

+

+cEnchantments::cEnchantments(const AString & a_StringSpec)

+{

+	AddFromString(a_StringSpec);

+}

+

+

+

+

+

+void cEnchantments::AddFromString(const AString & a_StringSpec)

+{

+	// Add enchantments in the stringspec; if a specified enchantment already exists, overwrites it

+	

+	// Split the StringSpec into separate declarations, each in the form "id=lvl":

+	AStringVector Decls = StringSplit(a_StringSpec, ";");

+	for (AStringVector::const_iterator itr = Decls.begin(), end = Decls.end(); itr != end; ++itr)

+	{

+		// Split each declaration into the id and lvl part:

+		if (itr->empty())

+		{

+			// The decl is empty (may happen if there's an extra semicolon at the end), ignore silently

+			continue;

+		}

+		AStringVector Split = StringSplitAndTrim(*itr, "=");

+		if (Split.size() != 2)

+		{

+			// Malformed decl

+			LOG("%s: Malformed enchantment decl: \"%s\", skipping.", __FUNCTION__, itr->c_str());

+			continue;

+		}

+		int id = atoi(Split[0].c_str());

+		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()

+			);

+			continue;

+		}

+		SetLevel(id, lvl);

+	}  // for itr - Decls[]

+}

+

+

+

+

+

+AString cEnchantments::ToString(void) const

+{

+	// Serialize all the enchantments into a string

+	AString res;

+	for (cEnchantments::cMap::const_iterator itr = m_Enchantments.begin(), end = m_Enchantments.end(); itr != end; ++itr)

+	{

+		AppendPrintf(res, "%d=%d;", itr->first, itr->second);

+	}  // for itr - m_Enchantments[]

+	return res;

+}

+

+

+

+

+

+int cEnchantments::GetLevel(int a_EnchantmentID) const

+{

+	// Return the level for the specified enchantment; 0 if not stored

+	cMap::const_iterator itr = m_Enchantments.find(a_EnchantmentID);

+	if (itr != m_Enchantments.end())

+	{

+		return itr->second;

+	}

+

+	// Not stored, return zero

+	return 0;

+}

+

+

+

+

+

+void cEnchantments::SetLevel(int a_EnchantmentID, int a_Level)

+{

+	// Sets the level for the specified enchantment, adding it if not stored before or removing it if level <= 0

+	if (a_Level == 0)

+	{

+		// Delete enchantment, if present:

+		cMap::iterator itr = m_Enchantments.find(a_EnchantmentID);

+		if (itr != m_Enchantments.end())

+		{

+			m_Enchantments.erase(itr);

+		}

+	}

+	else

+	{

+		// Add / overwrite enchantment

+		m_Enchantments[a_EnchantmentID] = a_Level;

+	}

+}

+

+

+

+

+

+

+void cEnchantments::Clear(void)

+{

+	m_Enchantments.clear();

+}

+

+

+

+

+

+bool cEnchantments::IsEmpty(void) const

+{

+	return m_Enchantments.empty();

+}

+

+

+

+

+

+void cEnchantments::WriteToNBTCompound(cFastNBTWriter & a_Writer, const AString & a_ListTagName) const

+{

+	// Write the enchantments into the specified NBT writer

+	// begin with the LIST tag of the specified name ("ench" or "StoredEnchantments")

+	

+	a_Writer.BeginList(a_ListTagName, TAG_Compound);

+	for (cMap::const_iterator itr = m_Enchantments.begin(), end = m_Enchantments.end(); itr != end; ++itr)

+	{

+		a_Writer.BeginCompound("");

+			a_Writer.AddShort("id",  itr->first);

+			a_Writer.AddShort("lvl", itr->second);

+		a_Writer.EndCompound();

+	}  // for itr - m_Enchantments[]

+	a_Writer.EndList();

+}

+

+

+

+

+

+void cEnchantments::ParseFromNBT(cParsedNBT & a_NBT, int a_EnchListTagIdx)

+{

+	// Read the enchantments from the specified NBT list tag (ench or StoredEnchantments)

+

+	// Verify that the tag is a list:

+	if (a_NBT.GetType(a_EnchListTagIdx) != TAG_List)

+	{

+		LOGWARNING("%s: Invalid EnchListTag type: exp %d, got %d. Enchantments not parsed",

+			__FUNCTION__, TAG_List, a_NBT.GetType(a_EnchListTagIdx)

+		);

+		ASSERT(!"Bad EnchListTag type");

+		return;

+	}

+	

+	// Verify that the list is of Compounds:

+	if (a_NBT.GetChildrenType(a_EnchListTagIdx) != TAG_Compound)

+	{

+		LOGWARNING("%s: Invalid NBT list children type: exp %d, got %d. Enchantments not parsed",

+			__FUNCTION__, TAG_Compound, a_NBT.GetChildrenType(a_EnchListTagIdx)

+		);

+		ASSERT(!"Bad EnchListTag children type");

+		return;

+	}

+	

+	Clear();

+	

+	// Iterate over all the compound children, parse an enchantment from each:

+	for (int tag = a_NBT.GetFirstChild(a_EnchListTagIdx); tag >= 0; tag = a_NBT.GetNextSibling(tag))

+	{

+		// tag is the compound inside the "ench" list tag

+		ASSERT(a_NBT.GetType(tag) == TAG_Compound);

+		

+		// Search for the id and lvl tags' values:

+		int id = -1, lvl = -1;

+		for (int ch = a_NBT.GetFirstChild(tag); ch >= 0; ch = a_NBT.GetNextSibling(ch))

+		{

+			if (a_NBT.GetType(ch) != TAG_Short)

+			{

+				continue;

+			}

+			if (a_NBT.GetName(ch) == "id")

+			{

+				id = a_NBT.GetShort(ch);

+			}

+			else if (a_NBT.GetName(ch) == "lvl")

+			{

+				lvl = a_NBT.GetShort(ch);

+			}

+		}  // for ch - children of the compound tag

+		

+		if ((id == -1) || (lvl <= 0))

+		{

+			// Failed to parse either the id or the lvl, skip this compound

+			continue;

+		}

+		

+		// Store the enchantment:

+		m_Enchantments[id] = lvl;

+	}  // for tag - children of the ench list tag

+}

+

+

+

+