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 class AwakeningUtil
{
    /// <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)
    {
        if (pk is not PB7 pb7)
            return;

        Span<byte> result = stackalloc byte[6];
        GetExpectedMinimumAVs(result, pb7);

        var rnd = Util.Rand;
        for (int i = 0; i < 6; i++)
        {
            var realMin = Math.Max(min, result[i]);
            var realMax = Math.Min(result[i], max);
            result[i] = (byte)rnd.Next(realMin, realMax + 1);
        }
        AwakeningSetVisual(pb7, result);
    }

    /// <summary>
    /// Sets the awakening values according to their displayed order.
    /// </summary>
    /// <param name="pk">Data to set values for</param>
    /// <param name="value"></param>
    public static void AwakeningSetVisual(IAwakened pk, Span<byte> value)
    {
        pk.AV_HP = value[0];
        pk.AV_ATK = value[1];
        pk.AV_DEF = value[2];
        pk.AV_SPA = value[3];
        pk.AV_SPD = value[4];
        pk.AV_SPE = value[5];
    }

    /// <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 byte 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)
    {
        Span<byte> result = stackalloc byte[6];
        GetExpectedMinimumAVs(result, (PB7)a);
        a.AV_HP  = Legal.AwakeningMax;
        a.AV_ATK = pk.IV_ATK == 0 ? result[1] : Legal.AwakeningMax;
        a.AV_DEF = Legal.AwakeningMax;
        a.AV_SPA = Legal.AwakeningMax;
        a.AV_SPD = Legal.AwakeningMax;
        a.AV_SPE = pk.IV_SPE == 0 ? result[5] : Legal.AwakeningMax;
    }

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

    /// <summary>
    /// Checks if the <see cref="current"/> has values greater or equal to the <see cref="initial"/>.
    /// </summary>
    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;
    }

    /// <summary>
    /// Updates the <see cref="result"/> span with the expected minimum values for each <see cref="IAwakened"/> index.
    /// </summary>
    /// <param name="result">Stat results</param>
    /// <param name="pk">Entity to check</param>
    public static void GetExpectedMinimumAVs(Span<byte> result, PB7 pk)
    {
        // go park transfers have 2 AVs for all stats.
        // leveling up in-game applies 1 AV to a "random" index.
        if (pk.Version == (int)GameVersion.GO)
            result.Fill(2);

        var start = pk.Met_Level;
        var end = pk.CurrentLevel;
        if (start == end)
            return;

        // Level up from met level to current level.
        var nature = pk.Nature;
        var character = pk.Characteristic;
        var ec = pk.EncryptionConstant;

        for (int i = start + 1; i <= end; i++)
        {
            var lm10 = i % 10;
            var bits = (ec >> (3 * lm10)) & 7;
            var index = PB7.GetRandomIndex((int)bits, character, nature);
            ++result[index];
        }
    }
}