PKHeX/PKHeX.Core/Legality/Encounters/EncounterStatic/DreamWorldEntry.cs

using System;

namespace PKHeX.Core;

/// <summary>
/// Intermediary Representation of Dream World Data
/// </summary>
public readonly record struct DreamWorldEntry(ushort Species, byte Level, ushort Move1 = 0, ushort Move2 = 0, ushort Move3 = 0, byte Form = 0, sbyte Gender = -1)
{
    private int EntryCount => Move1 == 0 ? 1 : Move2 == 0 ? 1 : Move3 == 0 ? 2 : 3;

    private void AddTo(GameVersion game, EncounterStatic5[] result, ref int ctr)
    {
        var p = PersonalTable.B2W2[Species];
        var a = p.HasHiddenAbility ? AbilityPermission.OnlyHidden : AbilityPermission.OnlyFirst;
        if (Move1 == 0)
        {
            result[ctr++] = new EncounterStatic5(game)
            {
                Species = Species,
                Form = Form,
                Gender = Gender,
                Level = Level,
                Ability = a,
                Location = 075,
                Shiny = Shiny.Never,
            };
            return;
        }
        result[ctr++] = Create(game, a, Move1);
        if (Move2 == 0)
            return;
        result[ctr++] = Create(game, a, Move2);
        if (Move3 == 0)
            return;
        result[ctr++] = Create(game, a, Move3);
    }

    private EncounterStatic5 Create(GameVersion game, AbilityPermission ability, ushort move) => new(game)
    {
        Species = Species,
        Form = Form,
        Gender = Gender,
        Level = Level,
        Ability = ability,
        Location = 075,
        Shiny = Shiny.Never,
        Moves = new(move),
    };

    public static EncounterStatic5[] GetArray(GameVersion game, ReadOnlySpan<DreamWorldEntry> t)
    {
        // Split encounters with multiple permitted special moves -- a pk can only be obtained with 1 of the special moves!
        int count = 0;
        foreach (var e in t)
            count += e.EntryCount;
        var result = new EncounterStatic5[count];

        int ctr = 0;
        foreach (var s in t)
            s.AddTo(game, result, ref ctr);
        return result;
    }
}