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

namespace PKHeX.Core;

/// <summary>
/// Encounter Slot found in <see cref="GameVersion.Gen4"/>.
/// </summary>
/// <inheritdoc cref="EncounterSlot"/>
public sealed record EncounterSlot4 : EncounterSlot, IMagnetStatic, INumberedSlot, IGroundTypeTile, ISlotRNGType
{
    public override int Generation => 4;
    public override EntityContext Context => EntityContext.Gen4;
    public GroundTileAllowed GroundTile => ((EncounterArea4)Area).GroundTile;

    public byte StaticIndex { get; }
    public byte MagnetPullIndex { get; }
    public byte StaticCount { get; }
    public byte MagnetPullCount { get; }
    public SlotType Type => Area.Type;

    public byte SlotNumber { get; }
    public override Ball FixedBall => GetRequiredBallValue();
    public bool CanUseRadar => !GameVersion.HGSS.Contains(Version) && GroundTile.HasFlag(GroundTileAllowed.Grass);

    public EncounterSlot4(EncounterArea4 area, ushort species, byte form, byte min, byte max, byte slot, byte mpi, byte mpc, byte sti, byte stc) : base(area, species, form, min, max)
    {
        SlotNumber = slot;

        MagnetPullIndex = mpi;
        MagnetPullCount = mpc;

        StaticIndex = sti;
        StaticCount = stc;
    }

    protected override void SetFormatSpecificData(PKM pk) => ((PK4)pk).GroundTile = GroundTile.GetIndex();

    public override EncounterMatchRating GetMatchRating(PKM pk)
    {
        if ((pk.Ball == (int)Ball.Safari) != Locations.IsSafariZoneLocation4(Location))
            return EncounterMatchRating.PartialMatch;
        if ((pk.Ball == (int)Ball.Sport) != (Type == SlotType.BugContest))
        {
            // Nincada => Shedinja can wipe the ball back to Poke
            if (pk.Species != (int)Core.Species.Shedinja || pk.Ball != (int)Ball.Poke)
                return EncounterMatchRating.PartialMatch;
        }
        return base.GetMatchRating(pk);
    }

    private Ball GetRequiredBallValue()
    {
        if (Type is SlotType.BugContest)
            return Ball.Sport;
        return Locations.IsSafariZoneLocation4(Location) ? Ball.Safari : Ball.None;
    }
}