pkNX/pkNX.Structures/Learnset/Learnset.cs

using System;
using System.Collections.Generic;
using System.Linq;

namespace pkNX.Structures;

public abstract class Learnset
{
    public int Count { get; protected set; }
    public int[] Moves { get; protected set; }
    public int[] Levels { get; protected set; }

    protected Learnset()
    {
        Moves = Array.Empty<int>();
        Levels = Array.Empty<int>();
    }

    public abstract byte[] Write();

    /// <summary>
    /// Returns the moves a Pokémon can learn between the specified level range.
    /// </summary>
    /// <param name="maxLevel">Maximum level</param>
    /// <param name="minLevel">Minimum level</param>
    /// <returns>Array of Move IDs</returns>
    public int[] GetMoves(int maxLevel, int minLevel = 0)
    {
        if (minLevel <= 1 && maxLevel >= 100)
            return Moves;
        if (minLevel > maxLevel)
            return Array.Empty<int>();
        int start = Array.FindIndex(Levels, z => z >= minLevel);
        if (start < 0)
            return Array.Empty<int>();
        int end = Array.FindLastIndex(Levels, z => z <= maxLevel);
        if (end < 0)
            return Array.Empty<int>();
        int[] result = new int[end - start + 1];
        Array.Copy(Moves, start, result, 0, result.Length);
        return result;
    }

    /// <summary>Returns the moves a Pokémon would have if it were encountered at the specified level.</summary>
    /// <remarks>In Generation 1, it is not possible to learn any moves lower than these encounter moves.</remarks>
    /// <param name="level">The level the Pokémon was encountered at.</param>
    /// <returns>Array of Move IDs</returns>
    public int[] GetEncounterMoves(int level)
    {
        const int count = 4;
        IList<int> moves = new int[count];
        int ctr = 0;
        for (int i = 0; i < Moves.Length; i++)
        {
            if (Levels[i] > level)
                break;
            int move = Moves[i];
            if (moves.Contains(move))
                continue;

            moves[ctr++] = move;
            ctr &= 3;
        }
        return (int[])moves;
    }

    /// <summary>Returns the index of the lowest level move if the Pokémon were encountered at the specified level.</summary>
    /// <remarks>Helps determine the minimum level an encounter can be at.</remarks>
    /// <param name="level">The level the Pokémon was encountered at.</param>
    /// <returns>Array of Move IDs</returns>
    public int GetMinMoveLevel(int level)
    {
        if (Levels.Length == 0)
            return 1;

        int end = Array.FindLastIndex(Levels, z => z <= level);
        return Math.Max(end - 4, 1);
    }

    /// <summary>Returns the level that a Pokémon can learn the specified move.</summary>
    /// <param name="move">Move ID</param>
    /// <returns>Level the move is learned at. If the result is below 0, it cannot be learned by levelup.</returns>
    public int GetLevelLearnMove(int move)
    {
        int index = Array.IndexOf(Moves, move);
        return index < 0 ? index : Levels[index];
    }

    public void Update(int[] moves, int[] levels)
    {
        Moves = moves;
        Levels = levels;
        Count = Moves.Length;
    }

    public int[] GetHighPoweredMoves(int count, IReadOnlyList<IMove> movedata)
    {
        var moves = Moves.OrderByDescending(move => movedata[move].Power).Distinct().Take(count).ToArray();
        Array.Resize(ref moves, count);
        return moves;
    }
}