PKHeX/PKHeX.Core/Legality/Encounters/EncounterSlot/EncounterSlot2.cs

using System.Collections.Generic;

namespace PKHeX.Core
{
    /// <summary>
    /// Encounter Slot found in <see cref="GameVersion.Gen2"/>.
    /// </summary>
    /// <remarks>
    /// Referenced Area object contains Time data which is used for <see cref="GameVersion.C"/> origin data.
    /// </remarks>
    /// <inheritdoc cref="EncounterSlot"/>
    public sealed record EncounterSlot2 : EncounterSlot, INumberedSlot
    {
        public override int Generation => 2;
        public int SlotNumber { get; }

        public EncounterSlot2(EncounterArea2 area, int species, int min, int max, int slot) : base(area, species, 0, min, max)
        {
            SlotNumber = slot;
        }

        protected override void ApplyDetails(ITrainerInfo sav, EncounterCriteria criteria, PKM pk)
        {
            base.ApplyDetails(sav, criteria, pk);

            var pk2 = (PK2)pk;

            if ((Area.Type & SlotType.Headbutt) != 0)
            {
                while (!IsTreeAvailable(pk2.TID))
                    pk2.TID = Util.Rand.Next(ushort.MaxValue);
            }

            if (Version == GameVersion.C)
                pk2.Met_TimeOfDay = ((EncounterArea2)Area).Time.RandomValidTime();
        }

        private static readonly Dictionary<int, int> Trees = new()
        {
            { 02, 0x3FF_3FF }, // Route 29
            { 04, 0x39D_3FF }, // Route 30
            { 05, 0x13D_3FF }, // Route 31
            { 08, 0x2FF_3FF }, // Route 32
            { 11, 0x009_3FF }, // Route 33
            { 12, 0x3DF_3FF }, // Azalea Town
            { 14, 0x3FF_3FF }, // Ilex Forest
            { 15, 0x100_2FF }, // Route 34
            { 18, 0x099_3FF }, // Route 35
            { 20, 0x3FF_3FF }, // Route 36
            { 21, 0x2F6_3FF }, // Route 37
            { 25, 0x3FF_3FF }, // Route 38
            { 26, 0x188_3FF }, // Route 39
            { 34, 0x3FE_3FF }, // Route 42
            { 37, 0x3B7_3FF }, // Route 43
            { 38, 0x3FF_3FF }, // Lake of Rage
            { 39, 0x2FF_3FF }, // Route 44
            { 91, 0x300_3FF }, // Route 26
            { 92, 0x1FE_3FF }, // Route 27
        };

        internal bool IsTreeAvailable(int trainerID)
        {
            if (!Trees.TryGetValue(Location, out var permissions))
                return false;

            var pivot = trainerID % 10;
            var type = Area.Type;
            return type switch
            {
                SlotType.Headbutt => (permissions & (1 << pivot)) != 0,
                /*special*/ _ => (permissions & (1 << (pivot + 12))) != 0,
            };
        }
    }
}