Cog/Plugins/CogAbility/Source/CogAbility/Private/CogAbilityConfig_Alignment.cpp

#include "CogAbilityConfig_Alignment.h"

#include "AbilitySystemComponent.h"
#include "CogAbilityDataAsset.h"
#include "GameplayEffect.h"

//--------------------------------------------------------------------------------------------------------------------------
FVector4f UCogAbilityConfig_Alignment::GetAttributeColor(const UAbilitySystemComponent& AbilitySystemComponent, const FGameplayAttribute& Attribute) const
{
    const float BaseValue = AbilitySystemComponent.GetNumericAttributeBase(Attribute);
    const float CurrentValue = AbilitySystemComponent.GetNumericAttribute(Attribute);

    if (CurrentValue > BaseValue)
    {
        return PositiveColor;
    }

    if (CurrentValue < BaseValue)
    {
        return NegativeColor;
    }

    return NeutralColor;
}

//--------------------------------------------------------------------------------------------------------------------------
bool UCogAbilityConfig_Alignment::GetAbilityColor(const UCogAbilityDataAsset* Asset, const UGameplayAbility& Ability, FLinearColor& OutColor) const
{
    if (Asset == nullptr)
    {
        OutColor = NeutralColor;
        return false;
    }

    const FGameplayTagContainer& Tags = Ability.GetAssetTags();
    if (Tags.HasTag(Asset->NegativeAbilityTag))
    {
        OutColor = NegativeColor;
        return true;
    }

    if (Tags.HasTag(Asset->PositiveAbilityTag))
    {
        OutColor = PositiveColor;
        return true;
    }

    OutColor = NeutralColor;
    return true;
}

//--------------------------------------------------------------------------------------------------------------------------
bool UCogAbilityConfig_Alignment::GetEffectColor(const UCogAbilityDataAsset* Asset, const UGameplayEffect& Effect, FLinearColor& OutColor) const
{
    if (Asset == nullptr)
    {
        OutColor = NeutralColor;
        return false;
    }

    const FGameplayTagContainer& Tags = Effect.GetAssetTags();
    if (Tags.HasTag(Asset->NegativeEffectTag))
    {
        OutColor = NegativeColor;
        return true;
    }

    if (Tags.HasTag(Asset->PositiveEffectTag))
    {
        OutColor = PositiveColor;
        return true;
    }

    OutColor = NeutralColor;
    return true;
}

//--------------------------------------------------------------------------------------------------------------------------
FVector4f UCogAbilityConfig_Alignment::GetEffectModifierColor(const FModifierSpec& ModSpec, const FGameplayModifierInfo& ModInfo, float BaseValue) const
{
    return GetEffectModifierColor(ModSpec.GetEvaluatedMagnitude(), ModInfo.ModifierOp, BaseValue);
}

//--------------------------------------------------------------------------------------------------------------------------
FVector4f UCogAbilityConfig_Alignment::GetEffectModifierColor(float ModifierValue, EGameplayModOp::Type ModifierOp, float BaseValue) const
{
    switch (ModifierOp)
    {
        case EGameplayModOp::AddBase:
        case EGameplayModOp::AddFinal:
        {
            if (ModifierValue > 0.0f)
            { return PositiveColor; }

            if (ModifierValue < 0.0f)
            { return NegativeColor; }
                
            return NeutralColor;
        }

        case EGameplayModOp::MultiplyAdditive:
        case EGameplayModOp::MultiplyCompound:
        {
            if (ModifierValue > 1.0f)
            { return PositiveColor; }

            if (ModifierValue < 1.0f)
            { return NegativeColor; }
                
            return NeutralColor;
        }

        case EGameplayModOp::DivideAdditive:
        {
            if (ModifierValue < 1.0f)
            { return PositiveColor; }

            if (ModifierValue > 1.0f)
            { return NegativeColor; }
                
            return NeutralColor;
        }

        case EGameplayModOp::Override:
        {
            if (ModifierValue > BaseValue)
            { return PositiveColor; }

            if (ModifierValue < BaseValue)
            { return NegativeColor; }

            return NeutralColor;
        }

        default: return NeutralColor;
    }
}