PKHeX/PKHeX.Core/Legality/Areas/EncounterArea7b.cs

using System;
using System.Collections.Generic;

namespace PKHeX.Core
{
    /// <inheritdoc cref="EncounterArea" />
    /// <summary>
    /// <see cref="GameVersion.GG"/> encounter area
    /// </summary>
    public sealed record EncounterArea7b : EncounterArea
    {
        public readonly EncounterSlot7b[] Slots;

        protected override IReadOnlyList<EncounterSlot> Raw => Slots;

        public static EncounterArea7b[] GetAreas(byte[][] input, GameVersion game)
        {
            var result = new EncounterArea7b[input.Length];
            for (int i = 0; i < input.Length; i++)
                result[i] = new EncounterArea7b(input[i], game);
            return result;
        }

        private EncounterArea7b(ReadOnlySpan<byte> data, GameVersion game) : base(game)
        {
            Location = data[0] | (data[1] << 8);
            Slots = ReadSlots(data);
        }

        private EncounterSlot7b[] ReadSlots(ReadOnlySpan<byte> data)
        {
            const int size = 4;
            int count = (data.Length - 2) / size;
            var slots = new EncounterSlot7b[count];
            for (int i = 0; i < slots.Length; i++)
            {
                int offset = 2 + (size * i);
                var entry = data.Slice(offset, size);
                slots[i] = ReadSlot(entry);
            }

            return slots;
        }

        private EncounterSlot7b ReadSlot(ReadOnlySpan<byte> entry)
        {
            int species = entry[0]; // always < 255; only original 151
            // form is always 0
            int min = entry[2];
            int max = entry[3];
            return new EncounterSlot7b(this, species, min, max);
        }

        private const int CatchComboBonus = 1;

        public override IEnumerable<EncounterSlot> GetMatchingSlots(PKM pkm, IReadOnlyList<EvoCriteria> chain)
        {
            foreach (var slot in Slots)
            {
                foreach (var evo in chain)
                {
                    if (slot.Species != evo.Species)
                        continue;

                    var met = pkm.Met_Level;
                    if (!slot.IsLevelWithinRange(met, 0, CatchComboBonus))
                        break;
                    if (slot.Form != evo.Form)
                        break;

                    yield return slot;
                    break;
                }
            }
        }
    }
}