PKHeX/PKHeX.Core/Saves/Substructures/Gen7/WB7Records.cs

using System;
using System.Linq;

namespace PKHeX.Core
{
    public class WB7Records : SaveBlock
    {
        public WB7Records(SaveFile sav) : base(sav)
        {
            Offset = ((SAV7b) sav).GetBlockOffset(BelugaBlockIndex.WB7Record);
        }

        private const int RecordMax = 10; // 0xE90 > (0x140 * 0xA = 0xC80), not sure what final 0x210 bytes are used for
        private const int FlagCountMax = 0x1C00; // (7168) end of the block?

        private int FlagStart => Offset + (RecordMax * WR7.Size);

        private int GetRecordOffset(int index)
        {
            if (index >= RecordMax)
                throw new ArgumentException(nameof(index));

            return Offset + (index * WR7.Size);
        }

        private int GetFlagOffset(int flag)
        {
            if (flag >= FlagCountMax)
                throw new ArgumentException(nameof(flag));
            return FlagStart + (flag / 8);
        }

        public WR7 GetRecord(int index)
        {
            int ofs = GetRecordOffset(index);
            byte[] data = new byte[WR7.Size];
            Array.Copy(Data, ofs, data, 0, WR7.Size);
            return new WR7(data);
        }

        public void SetRecord(WR7 record, int index)
        {
            int ofs = GetRecordOffset(index);
            record.Data.CopyTo(Data, ofs);
        }

        public WR7[] Records
        {
            get
            {
                var arr = new WR7[RecordMax];
                for (int i = 0; i < arr.Length; i++)
                    arr[i] = GetRecord(i);
                return arr;
            }
            set
            {
                for (int i = 0; i < value.Length; i++)
                    SetRecord(value[i], i);
            }
        }

        public bool GetFlag(int flag)
        {
            int ofs = GetFlagOffset(flag);
            var mask = 1 << (flag & 7);
            return (Data[ofs] & mask) != 0;
        }

        public void SetFlag(int flag, bool value)
        {
            int ofs = GetFlagOffset(flag);
            var mask = 1 << (flag & 7);
            if (value)
                Data[ofs] |= (byte)mask;
            else
                Data[ofs] &= (byte)mask;
        }

        public bool[] Flags
        {
            get => Enumerable.Range(0, FlagCountMax).Select(GetFlag).ToArray();
            set
            {
                for (int i = 0; i < value.Length; i++)
                    SetFlag(i, value[i]);
            }
        }
    }
}