1 files changed, 185 insertions, 0 deletions
diff --git a/Minecraft.World/ModifiableAttributeInstance.cpp b/Minecraft.World/ModifiableAttributeInstance.cpp
new file mode 100644
index 00000000..50aa6504
--- /dev/null
+++ b/Minecraft.World/ModifiableAttributeInstance.cpp
@@ -0,0 +1,185 @@
+#include "stdafx.h"
+#include "net.minecraft.world.entity.ai.attributes.h"
+#include "ModifiableAttributeInstance.h"
+
+ModifiableAttributeInstance::ModifiableAttributeInstance(BaseAttributeMap *attributeMap, Attribute *attribute)
+{
+	this->attributeMap = attributeMap;
+	this->attribute = attribute;
+
+	dirty = true;
+	cachedValue = 0.0;
+
+	baseValue = attribute->getDefaultValue();
+}
+
+ModifiableAttributeInstance::~ModifiableAttributeInstance()
+{
+	for (int i = 0; i < AttributeModifier::TOTAL_OPERATIONS; i++)
+	{
+		for (AUTO_VAR(it, modifiers[i].begin()); it != modifiers[i].end(); ++it)
+		{
+			// Delete all modifiers
+			delete *it;
+		}
+	}
+}
+
+Attribute *ModifiableAttributeInstance::getAttribute()
+{
+	return attribute;
+}
+
+double ModifiableAttributeInstance::getBaseValue()
+{
+	return baseValue;
+}
+
+void ModifiableAttributeInstance::setBaseValue(double baseValue)
+{
+	if (baseValue == this->getBaseValue()) return;
+	this->baseValue = baseValue;
+	setDirty();
+}
+
+// Returns a pointer to an internally managed vector of modifers by operation
+unordered_set<AttributeModifier *> *ModifiableAttributeInstance::getModifiers(int operation)
+{
+	return &modifiers[operation];
+}
+
+// Returns a pointer to a new vector of all modifiers
+void ModifiableAttributeInstance::getModifiers(unordered_set<AttributeModifier *>& result)
+{
+	for (int i = 0; i < AttributeModifier::TOTAL_OPERATIONS; i++)
+	{
+		unordered_set<AttributeModifier *> *opModifiers = &modifiers[i];
+
+		for (AUTO_VAR(it, opModifiers->begin()); it != opModifiers->end(); ++it)
+		{
+			result.insert(*it);
+		}		
+	}
+}
+
+AttributeModifier *ModifiableAttributeInstance::getModifier(eMODIFIER_ID id)
+{
+	AttributeModifier *modifier = NULL;
+
+	AUTO_VAR(it, modifierById.find(id));
+	if(it != modifierById.end())
+	{
+		modifier = it->second;
+	}
+
+	return modifier; 
+}
+
+void ModifiableAttributeInstance::addModifiers(unordered_set<AttributeModifier *> *modifiers)
+{
+	for (AUTO_VAR(it, modifiers->begin()); it != modifiers->end(); ++it)
+	{
+		addModifier(*it);
+	}
+}
+
+// Add new modifier to attribute instance (takes ownership of modifier)
+void ModifiableAttributeInstance::addModifier(AttributeModifier *modifier)
+{
+	// Can't add modifiers with the same ID (unless the modifier is anonymous)
+	if (modifier->getId() != eModifierId_ANONYMOUS && getModifier(modifier->getId()) != NULL) 
+	{
+		assert(0);
+		// throw new IllegalArgumentException("Modifier is already applied on this attribute!");
+		return;
+	}
+
+	modifiers[modifier->getOperation()].insert(modifier);
+	modifierById[modifier->getId()] = modifier;
+
+	setDirty();
+}
+
+void ModifiableAttributeInstance::setDirty()
+{
+	dirty = true;
+	attributeMap->onAttributeModified(this);
+}
+
+void ModifiableAttributeInstance::removeModifier(AttributeModifier *modifier)
+{
+	for (int i = 0; i < AttributeModifier::TOTAL_OPERATIONS; i++)
+	{
+		for (AUTO_VAR(it, modifiers[i].begin()); it != modifiers[i].end(); ++it)
+		{
+			if (modifier->equals(*it))
+			{
+				modifiers[i].erase(it);
+				break;
+			}
+		}
+	}
+
+	modifierById.erase(modifier->getId());
+
+	setDirty();
+}
+
+void ModifiableAttributeInstance::removeModifier(eMODIFIER_ID id)
+{
+	AttributeModifier *modifier = getModifier(id);
+	if (modifier != NULL) removeModifier(modifier);
+}
+
+void ModifiableAttributeInstance::removeModifiers()
+{
+	unordered_set<AttributeModifier *> removingModifiers;
+	getModifiers(removingModifiers);
+
+	for (AUTO_VAR(it, removingModifiers.begin()); it != removingModifiers.end(); ++it)
+	{
+		removeModifier(*it);
+	}
+}
+
+double ModifiableAttributeInstance::getValue()
+{
+	if (dirty)
+	{
+		cachedValue = calculateValue();
+		dirty = false;
+	}
+
+	return cachedValue;
+}
+
+double ModifiableAttributeInstance::calculateValue()
+{
+	double base = getBaseValue();
+	unordered_set<AttributeModifier *> *modifiers;
+
+	modifiers = getModifiers(AttributeModifier::OPERATION_ADDITION);
+	for (AUTO_VAR(it, modifiers->begin()); it != modifiers->end(); ++it)
+	{
+		AttributeModifier *modifier = *it;
+		base += modifier->getAmount();
+	}
+
+	double result = base;
+
+	modifiers = getModifiers(AttributeModifier::OPERATION_MULTIPLY_BASE);
+	for (AUTO_VAR(it, modifiers->begin()); it != modifiers->end(); ++it)
+	{
+		AttributeModifier *modifier = *it;
+		result += base * modifier->getAmount();
+	}
+	
+	modifiers = getModifiers(AttributeModifier::OPERATION_MULTIPLY_TOTAL);
+	for (AUTO_VAR(it, modifiers->begin()); it != modifiers->end(); ++it)
+	{
+		AttributeModifier *modifier = *it;
+		result *= 1 + modifier->getAmount();
+	}
+
+	return attribute->sanitizeValue(result);
+}
+\ No newline at end of file