PKHeX/PKHeX.Core/Legality/Verifiers/IndividualValueVerifier.cs

using System;
using static PKHeX.Core.LegalityCheckStrings;

namespace PKHeX.Core;

/// <summary>
/// Verifies the <see cref="PKM.IVs"/>.
/// </summary>
public sealed class IndividualValueVerifier : Verifier
{
    protected override CheckIdentifier Identifier => CheckIdentifier.IVs;

    public override void Verify(LegalityAnalysis data)
    {
        switch (data.EncounterMatch)
        {
            case IPogoSlot s:
                VerifyIVsGoTransfer(data, s);
                break;
            case IFlawlessIVCount s:
                VerifyIVsFlawless(data, s);
                break;
            case EncounterSlot7:
                VerifyIVsGen7(data);
                break;
            case MysteryGift g:
                VerifyIVsMystery(data, g);
                break;
        }
        var pk = data.Entity;
        var hpiv = pk.IV_HP;
        if (hpiv < 30 && AllIVsEqual(pk, hpiv))
            data.AddLine(Get(string.Format(LIVAllEqual_0, hpiv), Severity.Fishy));
    }

    private static bool AllIVsEqual(PKM pk, int hpiv)
    {
        return (pk.IV_ATK == hpiv) && (pk.IV_DEF == hpiv) && (pk.IV_SPA == hpiv) && (pk.IV_SPD == hpiv) && (pk.IV_SPE == hpiv);
    }

    private void VerifyIVsMystery(LegalityAnalysis data, MysteryGift g)
    {
        if (!g.HasFixedIVs)
            return; // PID/IV style instead of fixed IVs.

        Span<int> IVs = stackalloc int[6];
        g.GetIVs(IVs);
        var ivflag = IVs.Find(static iv => (byte)(iv - 0xFC) < 3);
        if (ivflag == default) // Random IVs
        {
            bool valid = Legal.GetIsFixedIVSequenceValidSkipRand(IVs, data.Entity);
            if (!valid)
                data.AddLine(GetInvalid(LEncGiftIVMismatch));
        }
        else
        {
            int IVCount = ivflag - 0xFB;  // IV2/IV3
            VerifyIVsFlawless(data, IVCount);
        }
    }

    private void VerifyIVsGen7(LegalityAnalysis data)
    {
        var pk = data.Entity;
        if (pk.AbilityNumber == 4)
        {
            var abilities = (IPersonalAbility12H)pk.PersonalInfo;
            if (!AbilityVerifier.CanAbilityPatch(pk.Format, abilities, pk.Species))
                VerifyIVsFlawless(data, 2); // Chain of 10 yields 5% HA and 2 flawless IVs
        }
    }

    private void VerifyIVsFlawless(LegalityAnalysis data, IFlawlessIVCount s)
    {
        if (s.FlawlessIVCount != 0)
            VerifyIVsFlawless(data, s.FlawlessIVCount);
    }

    private void VerifyIVsFlawless(LegalityAnalysis data, int count)
    {
        if (data.Entity.FlawlessIVCount < count)
            data.AddLine(GetInvalid(string.Format(LIVF_COUNT0_31, count)));
    }

    private void VerifyIVsGoTransfer(LegalityAnalysis data, IPogoSlot g)
    {
        if (!g.GetIVsValid(data.Entity))
            data.AddLine(GetInvalid(LIVNotCorrect));
    }
}