PKHeX/PKHeX.Core/PKM/Interfaces/IAwakened.cs

using System;

namespace PKHeX.Core;

public interface IAwakened
{
    byte AV_HP { get; set; }
    byte AV_ATK { get; set; }
    byte AV_DEF { get; set; }
    byte AV_SPE { get; set; }
    byte AV_SPA { get; set; }
    byte AV_SPD { get; set; }
}

public static partial class Extensions
{
    /// <summary>
    /// Sums all values.
    /// </summary>
    /// <param name="pk">Data to sum with</param>
    public static int AwakeningSum(this IAwakened pk) => pk.AV_HP + pk.AV_ATK + pk.AV_DEF + pk.AV_SPE + pk.AV_SPA + pk.AV_SPD;

    /// <summary>
    /// Clears all values.
    /// </summary>
    /// <param name="pk">Data to clear from</param>
    public static void AwakeningClear(this IAwakened pk) => pk.AwakeningSetAllTo(0);

    /// <summary>
    /// Sets all values to the maximum value.
    /// </summary>
    /// <param name="pk">Data to set values for</param>
    public static void AwakeningMax(this IAwakened pk) => pk.AwakeningSetAllTo(Legal.AwakeningMax);

    /// <summary>
    /// Sets all values to the specified value.
    /// </summary>
    /// <param name="pk">Data to set values for</param>
    /// <param name="value">Value to set all to</param>
    public static void AwakeningSetAllTo(this IAwakened pk, byte value) => pk.AV_HP = pk.AV_ATK = pk.AV_DEF = pk.AV_SPE = pk.AV_SPA = pk.AV_SPD = value;

    /// <summary>
    /// Sets all values to the specified value.
    /// </summary>
    /// <param name="pk">Data to set values for</param>
    /// <param name="min">Minimum value to set</param>
    /// <param name="max">Maximum value to set</param>
    public static void AwakeningSetRandom(this IAwakened pk, byte min = 0, int max = Legal.AwakeningMax)
    {
        var rnd = Util.Rand;
        int randClamp = max + 1;
        for (int index = 0; index < 6; index++)
            pk.SetAV(index, (byte)rnd.Next(min, randClamp));
    }

    /// <summary>
    /// Gets if all values are within legal limits.
    /// </summary>
    /// <param name="pk">Data to check</param>
    public static bool AwakeningAllValid(this IAwakened pk)
    {
        if (pk.AV_HP > Legal.AwakeningMax)
            return false;
        if (pk.AV_ATK > Legal.AwakeningMax)
            return false;
        if (pk.AV_DEF > Legal.AwakeningMax)
            return false;
        if (pk.AV_SPE > Legal.AwakeningMax)
            return false;
        if (pk.AV_SPA > Legal.AwakeningMax)
            return false;
        if (pk.AV_SPD > Legal.AwakeningMax)
            return false;
        return true;
    }

    /// <summary>
    /// Sets one of the <see cref="IAwakened"/> values based on its index within the array.
    /// </summary>
    /// <param name="pk">Pokémon to modify.</param>
    /// <param name="index">Index to set to</param>
    /// <param name="value">Value to set</param>
    public static byte SetAV(this IAwakened pk, int index, byte value) => index switch
    {
        0 => pk.AV_HP = value,
        1 => pk.AV_ATK = value,
        2 => pk.AV_DEF = value,
        3 => pk.AV_SPE = value,
        4 => pk.AV_SPA = value,
        5 => pk.AV_SPD = value,
        _ => throw new ArgumentOutOfRangeException(nameof(index)),
    };

    /// <summary>
    /// Sets one of the <see cref="IAwakened"/> values based on its index within the array.
    /// </summary>
    /// <param name="pk">Pokémon to check.</param>
    /// <param name="index">Index to get</param>
    public static int GetAV(this IAwakened pk, int index) => index switch
    {
        0 => pk.AV_HP,
        1 => pk.AV_ATK,
        2 => pk.AV_DEF,
        3 => pk.AV_SPE,
        4 => pk.AV_SPA,
        5 => pk.AV_SPD,
        _ => throw new ArgumentOutOfRangeException(nameof(index)),
    };

    /// <summary>
    /// Loads the <see cref="IAwakened"/> values to the input span.
    /// </summary>
    public static void GetAVs(this IAwakened pk, Span<byte> value)
    {
        if (value.Length != 6)
            return;
        value[0] = pk.AV_HP;
        value[1] = pk.AV_ATK;
        value[2] = pk.AV_DEF;
        value[3] = pk.AV_SPE;
        value[4] = pk.AV_SPA;
        value[5] = pk.AV_SPD;
    }

    /// <summary>
    /// Sets the values based on the current <see cref="PKM.IVs"/>.
    /// </summary>
    /// <param name="a">Accessor for setting the values</param>
    /// <param name="pk">Retriever for IVs</param>
    public static void SetSuggestedAwakenedValues(this IAwakened a, PKM pk)
    {
        a.AV_HP  = Legal.AwakeningMax;
        a.AV_ATK = pk.IV_ATK == 0 ? (byte)0 : Legal.AwakeningMax;
        a.AV_DEF = Legal.AwakeningMax;
        a.AV_SPE = pk.IV_SPE == 0 ? (byte)0 : Legal.AwakeningMax;
        a.AV_SPA = Legal.AwakeningMax;
        a.AV_SPD = Legal.AwakeningMax;
    }

    public static bool IsAwakeningBelow(this IAwakened current, IAwakened initial) => !current.IsAwakeningAboveOrEqual(initial);

    public static bool IsAwakeningAboveOrEqual(this IAwakened current, IAwakened initial)
    {
        if (current.AV_HP < initial.AV_HP)
            return false;
        if (current.AV_ATK < initial.AV_ATK)
            return false;
        if (current.AV_DEF < initial.AV_DEF)
            return false;
        if (current.AV_SPA < initial.AV_SPA)
            return false;
        if (current.AV_SPD < initial.AV_SPD)
            return false;
        if (current.AV_SPE < initial.AV_SPE)
            return false;
        return true;
    }
}