PKHeX/PKHeX.Core/Editing/PKM/QR/QRPK7.cs

using System;
using static System.Buffers.Binary.BinaryPrimitives;

namespace PKHeX.Core;

/// <summary>
/// Generation 7 PGL QR Code encoded <see cref="PKM"/> entities.
/// </summary>
public sealed class QRPK7 : IEncounterInfo
{
    public GameVersion Version => (GameVersion)CassetteVersion;
    public bool EggEncounter => false;
    public byte LevelMin => Level;
    public byte LevelMax => Level;
    public int Generation => Version.GetGeneration();
    public EntityContext Context => EntityContext.Gen7;
    public bool IsShiny => false;

    private readonly byte[] Data;
    public const int SIZE = 0x30;
    public QRPK7(byte[] d) => Data = (byte[])d.Clone();

    public uint EncryptionConstant => ReadUInt32LittleEndian(Data.AsSpan(0));
    public byte HT_Flags => Data[4];
    public int Unk_5 => Data[5];
    public int Unk_6 => Data[6];
    public int Unk_7 => Data[7];
    public int Move1_PPUps => Data[8];
    public int Move2_PPUps => Data[9];
    public int Move3_PPUps => Data[0xA];
    public int Move4_PPUps => Data[0xB];
    public uint IV32 { get => ReadUInt32LittleEndian(Data.AsSpan(0xC)); set => WriteUInt32LittleEndian(Data.AsSpan(0xC), value); }
    public int IV_HP { get => (int)(IV32 >> 00) & 0x1F; set => IV32 = (uint)((IV32 & ~(0x1F << 00)) | (uint)((value > 31 ? 31 : value) << 00)); }
    public int IV_ATK { get => (int)(IV32 >> 05) & 0x1F; set => IV32 = (uint)((IV32 & ~(0x1F << 05)) | (uint)((value > 31 ? 31 : value) << 05)); }
    public int IV_DEF { get => (int)(IV32 >> 10) & 0x1F; set => IV32 = (uint)((IV32 & ~(0x1F << 10)) | (uint)((value > 31 ? 31 : value) << 10)); }
    public int IV_SPE { get => (int)(IV32 >> 15) & 0x1F; set => IV32 = (uint)((IV32 & ~(0x1F << 15)) | (uint)((value > 31 ? 31 : value) << 15)); }
    public int IV_SPA { get => (int)(IV32 >> 20) & 0x1F; set => IV32 = (uint)((IV32 & ~(0x1F << 20)) | (uint)((value > 31 ? 31 : value) << 20)); }
    public int IV_SPD { get => (int)(IV32 >> 25) & 0x1F; set => IV32 = (uint)((IV32 & ~(0x1F << 25)) | (uint)((value > 31 ? 31 : value) << 25)); }
    public uint PID => ReadUInt32LittleEndian(Data.AsSpan(0x10));
    public ushort Species => ReadUInt16LittleEndian(Data.AsSpan(0x14));
    public ushort HeldItem => ReadUInt16LittleEndian(Data.AsSpan(0x16));
    public ushort Move1 => ReadUInt16LittleEndian(Data.AsSpan(0x18));
    public ushort Move2 => ReadUInt16LittleEndian(Data.AsSpan(0x1A));
    public ushort Move3 => ReadUInt16LittleEndian(Data.AsSpan(0x1C));
    public ushort Move4 => ReadUInt16LittleEndian(Data.AsSpan(0x1E));
    public int Unk_20 => Data[0x20];
    public int AbilityIndex => Data[0x21];
    public int Nature => Data[0x22];
    public bool FatefulEncounter => (Data[0x23] & 1) == 1;
    public int Gender => (Data[0x23] >> 1) & 3;
    public byte Form => (byte)(Data[0x23] >> 3);
    public byte EV_HP => Data[0x24];
    public byte EV_ATK => Data[0x25];
    public byte EV_DEF => Data[0x26];
    public byte EV_SPE => Data[0x27];
    public byte EV_SPA => Data[0x28];
    public byte EV_SPD => Data[0x29];
    public int Unk_2A => Data[0x2A];
    public int Friendship => Data[0x2B];
    public int Ball => Data[0x2C];
    public byte Level => Data[0x2D];
    public int CassetteVersion => Data[0x2E];
    public int Language => Data[0x2F];

    /// <summary>
    /// Converts the <see cref="Data"/> to a rough PKM.
    /// </summary>
    public PKM ConvertToPKM(ITrainerInfo tr) => ConvertToPKM(tr, EncounterCriteria.Unrestricted);

    /// <summary>
    /// Converts the <see cref="Data"/> to a rough PKM.
    /// </summary>
    public PKM ConvertToPKM(ITrainerInfo tr, EncounterCriteria criteria)
    {
        var pk = new PK7
        {
            EncryptionConstant = EncryptionConstant,
            PID = PID,
            Language = Language,
            Species = Species,
            Gender = Gender,
            Nature = Nature,
            FatefulEncounter = FatefulEncounter,
            Form = Form,
            HyperTrainFlags = HT_Flags,
            IV_HP = IV_HP,
            IV_ATK = IV_ATK,
            IV_DEF = IV_DEF,
            IV_SPA = IV_SPA,
            IV_SPD = IV_SPD,
            IV_SPE = IV_SPE,
            EV_HP = EV_HP,
            EV_ATK = EV_ATK,
            EV_DEF = EV_DEF,
            EV_SPA = EV_SPA,
            EV_SPD = EV_SPD,
            EV_SPE = EV_SPE,
            Move1 = Move1,
            Move2 = Move2,
            Move3 = Move3,
            Move4 = Move4,
            Move1_PPUps = Move1_PPUps,
            Move2_PPUps = Move2_PPUps,
            Move3_PPUps = Move3_PPUps,
            Move4_PPUps = Move4_PPUps,
            HeldItem = HeldItem,
            HT_Friendship = Friendship,
            OT_Friendship = Friendship,
            Ball = Ball,
            Version = CassetteVersion,

            OT_Name = tr.OT,
            HT_Name = tr.OT,
            CurrentLevel = Level,
            Met_Level = Level,
            MetDate = DateOnly.FromDateTime(DateTime.Today),
        };
        RecentTrainerCache.SetConsoleRegionData3DS(pk, tr);

        pk.RefreshAbility(AbilityIndex >> 1);
        pk.ForcePartyData();

        pk.RefreshChecksum();
        return pk;
    }
}