PKHeX/PKHeX.Core/Saves/Substructures/Gen8/TitleScreen8.cs

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

namespace PKHeX.Core
{
    /// <summary>
    /// Pokémon Team that shows up at the title screen.
    /// </summary>
    public sealed class TitleScreen8 : SaveBlock
    {
        public TitleScreen8(SAV8SWSH sav, SCBlock block) : base(sav, block.Data) { }

        /// <summary>
        /// Gets an object that exposes the data of the corresponding party <see cref="index"/>.
        /// </summary>
        public TitleScreen8Poke ViewPoke(int index)
        {
            if ((uint)index >= 6)
                throw new ArgumentOutOfRangeException(nameof(index));
            return new TitleScreen8Poke(Data, Offset + 0x00 + (index * TitleScreen8Poke.SIZE));
        }

        /// <summary>
        /// Applies the current <see cref="SaveFile.PartyData"/> to the block.
        /// </summary>
        public void SetPartyData() => LoadTeamData(SAV.PartyData);

        public void LoadTeamData(IList<PKM> party)
        {
            for (int i = 0; i < party.Count; i++)
                ViewPoke(i).LoadFrom(party[i]);
            for (int i = party.Count; i < 6; i++)
                ViewPoke(i).Clear();
        }
    }

    public sealed class TitleScreen8Poke : ISpeciesForm
    {
        public const int SIZE = 0x28;
        private readonly byte[] Data;
        private readonly int Offset;

        public TitleScreen8Poke(byte[] data, int offset)
        {
            Data = data;
            Offset = offset;
        }

        public int Species
        {
            get => ReadInt32LittleEndian(Data.AsSpan(Offset + 0x00));
            set => WriteInt32LittleEndian(Data.AsSpan(Offset + 0x00), value);
        }

        public int Form
        {
            get => ReadInt32LittleEndian(Data.AsSpan(Offset + 0x04));
            set => WriteInt32LittleEndian(Data.AsSpan(Offset + 0x04), value);
        }

        public int Gender
        {
            get => ReadInt32LittleEndian(Data.AsSpan(Offset + 0x08));
            set => WriteInt32LittleEndian(Data.AsSpan(Offset + 0x08), value);
        }

        public bool IsShiny
        {
            get => Data[Offset + 0xC] != 0;
            set => Data[Offset + 0xC] = value ? (byte)1 : (byte)0;
        }

        public uint EncryptionConstant
        {
            get => ReadUInt32LittleEndian(Data.AsSpan(Offset + 0x10));
            set => WriteUInt32LittleEndian(Data.AsSpan(Offset + 0x10), value);
        }

        public int FormArgument
        {
            get => ReadInt32LittleEndian(Data.AsSpan(Offset + 0x14));
            set => WriteInt32LittleEndian(Data.AsSpan(Offset + 0x14), value);
        }

        public uint Unknown18
        {
            get => ReadUInt32LittleEndian(Data.AsSpan(Offset + 0x18));
            set => WriteUInt32LittleEndian(Data.AsSpan(Offset + 0x18), value);
        }

        public uint Unknown1C
        {
            get => ReadUInt32LittleEndian(Data.AsSpan(Offset + 0x1C));
            set => WriteUInt32LittleEndian(Data.AsSpan(Offset + 0x1C), value);
        }

        public uint Unknown20
        {
            get => ReadUInt32LittleEndian(Data.AsSpan(Offset + 0x20));
            set => WriteUInt32LittleEndian(Data.AsSpan(Offset + 0x20), value);
        }

        public uint Unknown24
        {
            get => ReadUInt32LittleEndian(Data.AsSpan(Offset + 0x24));
            set => WriteUInt32LittleEndian(Data.AsSpan(Offset + 0x24), value);
        }

        public void Clear() => Array.Clear(Data, Offset, SIZE);

        public void LoadFrom(PKM pkm)
        {
            Species = pkm.Species;
            Form = pkm.Form;
            Gender = pkm.Gender;
            IsShiny = pkm.IsShiny;
            EncryptionConstant = pkm.EncryptionConstant;
            FormArgument = pkm is IFormArgument f && pkm.Species == (int)Core.Species.Alcremie ? (int)f.FormArgument : -1;
        }

        public void LoadFrom(TrainerCard8Poke pkm)
        {
            Species = pkm.Species;
            Form = pkm.Form;
            Gender = pkm.Gender;
            IsShiny = pkm.IsShiny;
            EncryptionConstant = pkm.EncryptionConstant;
            FormArgument = pkm.FormArgument;
        }
    }
}