#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 (auto& modifier : modifiers)
	{
		for (auto it : modifier)
		{
			// 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 (auto& modifier : modifiers)
	{
		for (auto& opModifier : modifier)
		{
			result.insert(opModifier);
		}
	}
}

AttributeModifier *ModifiableAttributeInstance::getModifier(eMODIFIER_ID id)
{
	AttributeModifier *modifier = nullptr;

    auto it = modifierById.find(id);
    if(it != modifierById.end())
	{
		modifier = it->second;
	}

	return modifier;
}

void ModifiableAttributeInstance::addModifiers(unordered_set<AttributeModifier *> *modifiers)
{
	for (auto& modifier : *modifiers)
	{
		addModifier(modifier);
	}
}

// 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()) != nullptr)
	{
		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 (auto& mod : modifiers)
	{
        for (auto& it : mod )
        {
			if (modifier->equals(it))
			{
				mod.erase(it);
				break;
			}
		}
	}

	modifierById.erase(modifier->getId());

	setDirty();
}

void ModifiableAttributeInstance::removeModifier(eMODIFIER_ID id)
{
	AttributeModifier *modifier = getModifier(id);
	if (modifier != nullptr) removeModifier(modifier);
}

void ModifiableAttributeInstance::removeModifiers()
{
	unordered_set<AttributeModifier *> removingModifiers;
	getModifiers(removingModifiers);

	for (auto& it : removingModifiers)
	{
		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& modifier : *modifiers)
	{
		base += modifier->getAmount();
	}

	double result = base;

	modifiers = getModifiers(AttributeModifier::OPERATION_MULTIPLY_BASE);
	for (auto& modifier : *modifiers)
	{
		result += base * modifier->getAmount();
	}

	modifiers = getModifiers(AttributeModifier::OPERATION_MULTIPLY_TOTAL);
	for (auto& modifier : *modifiers)
	{
		result *= 1 + modifier->getAmount();
	}

	return attribute->sanitizeValue(result);
}