PKHeX/PKHeX.Core/Saves/Substructures/Gen5/FestaBlock5.cs

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

namespace PKHeX.Core
{
    public sealed class FestaBlock5 : SaveBlock<SAV5B2W2>
    {
        public FestaBlock5(SAV5B2W2 SAV, int offset) : base(SAV) => Offset = offset;

        public const ushort MaxScore = 9999;
        public const int FunfestFlag = 2438;

        public const int MaxMissionIndex = (int)Funfest5Mission.TheBerryHuntingAdventure;

        public ushort Hosted
        {
            get => ReadUInt16LittleEndian(Data.AsSpan(Offset + 0xF0));
            set => WriteUInt16LittleEndian(Data.AsSpan(Offset + 0xF0), Math.Min(MaxScore, value));
        }

        public ushort Participated
        {
            get => ReadUInt16LittleEndian(Data.AsSpan(Offset + 0xF2));
            set => WriteUInt16LittleEndian(Data.AsSpan(Offset + 0xF2), Math.Min(MaxScore, value));
        }

        public ushort Completed
        {
            get => ReadUInt16LittleEndian(Data.AsSpan(Offset + 0xF4));
            set => WriteUInt16LittleEndian(Data.AsSpan(Offset + 0xF4), Math.Min(MaxScore, value));
        }

        public ushort TopScores
        {
            get => ReadUInt16LittleEndian(Data.AsSpan(Offset + 0xF6));
            set => WriteUInt16LittleEndian(Data.AsSpan(Offset + 0xF6), Math.Min(MaxScore, value));
        }

        public byte WhiteEXP
        {
            get => Data[Offset + 0xF8];
            set => Data[Offset + 0xF8] = value;
        }

        public byte BlackEXP
        {
            get => Data[Offset + 0xF9];
            set => Data[Offset + 0xF9] = value;
        }

        public byte Participants
        {
            get => Data[Offset + 0xFA];
            set => Data[Offset + 0xFA] = value;
        }

        private static int GetMissionRecordOffset(int mission)
        {
            if ((uint)mission > MaxMissionIndex)
                throw new ArgumentOutOfRangeException(nameof(mission));
            return mission * sizeof(uint);
        }

        public Funfest5Score GetMissionRecord(int mission)
        {
            var raw = ReadUInt32LittleEndian(Data.AsSpan(Offset + GetMissionRecordOffset(mission)));
            return new Funfest5Score(raw);
        }

        public void SetMissionRecord(int mission, Funfest5Score score)
        {
            var value = score.RawValue;
            WriteUInt32LittleEndian(Data.AsSpan(Offset + GetMissionRecordOffset(mission)), value);
        }

        public bool IsFunfestMissionsUnlocked
        {
            get => SAV.GetEventFlag(FunfestFlag);
            set => SAV.SetEventFlag(FunfestFlag, value);
        }

        public bool IsFunfestMissionUnlocked(int mission)
        {
            if ((uint) mission > MaxMissionIndex)
                throw new ArgumentOutOfRangeException(nameof(mission));

            if (mission == 0)
                return !IsFunfestMissionsUnlocked;

            var req = FunfestMissionUnlockFlagRequired[mission];
            foreach (var f in req)
            {
                if (!SAV.GetEventFlag(f))
                    return false;
            }
            return true;
        }

        public void UnlockFunfestMission(int mission)
        {
            if ((uint)mission > MaxMissionIndex)
                throw new ArgumentOutOfRangeException(nameof(mission));

            IsFunfestMissionsUnlocked = true;
            var req = FunfestMissionUnlockFlagRequired[mission];
            foreach (var f in req)
                SAV.SetEventFlag(f, true);
        }

        public void UnlockAllFunfestMissions()
        {
            for (int i = 0; i < MaxMissionIndex; i++)
                UnlockFunfestMission(i);
        }

        private readonly int[][] FunfestMissionUnlockFlagRequired =
        {
            Array.Empty<int>(),   // 00
            Array.Empty<int>(),   // 01
            new[] { 2444 },       // 02
            Array.Empty<int>(),   // 03
            new[] { 2445 },       // 04
            Array.Empty<int>(),   // 05
            new[] { 2462 },       // 06
            new[] { 2452, 2476 }, // 07
            new[] { 2476, 2548 }, // 08
            new[] { 2447 }, new[] { 2447 }, // 09
            new[] { 2453 }, new[] { 2453 }, // 10
            new[] { 2504 },       // 11
            new[] { 2457, 2507 }, // 12
            new[] { 2458, 2478 }, // 13
            new[] { 2456, 2508 }, // 14
            new[] { 2448 }, new[] { 2448 }, // 15
            new[] { 2549 },       // 16
            new[] { 2449 },       // 17
            new[] { 2479, 2513 }, // 18
            new[] { 2479, 2550 }, // 19
            new[] { 2481 },       // 20
            new[] { 2459 },       // 21
            new[] { 2454 },       // 22
            new[] { 2551 },       // 23
            new[] { 2400 },       // 24
            new[] { 2400 },       // 25
            new[] { 2400 }, new[] { 2400 }, // 26
            new[] { 2400 }, new[] { 2400 }, // 27
            new[] { 2400 },       // 28
            new[] { 2400, 2460 }, // 29
            new[] { 2400 },       // 30
            new[] { 2400, 2461 }, new[] { 2400, 2461 }, // 31
            new[] { 2437 },       // 32
            new[] { 2450 },       // 33
            new[] { 2451 },       // 34
            new[] { 2455 },       // 35
            new[] { 0105 },       // 36
            new[] { 2400 },       // 37
            new[] { 2557 },       // 38
        };

        public static int GetExpNeededForLevelUp(int lv)
        {
            return lv > 8 ? 50 : (lv * 5) + 5;
        }

        public static int GetTotalEntreeExp(int lv)
        {
            if (lv < 9)
                return lv * (lv + 1) * 5 / 2;
            return ((lv - 9) * 50) + 225;
        }
    }

    public enum Funfest5Mission
    {
        TheFirstBerrySearch = 0,
        CollectBerries = 1,
        FindLostItems = 2,
        FindLostBoys = 3,
        EnjoyShopping = 4,
        FindAudino = 5,
        SearchFor3Pokemon = 6,
        TrainwithMartialArtists = 7,
        Sparringwith10Trainers = 8,
        B_GetRichQuick = 9,
        W_TreasureHunting = 10,
        B_ExcitingTrading = 11,
        W_ExhilaratingTrading = 12,
        FindEmolga = 13,
        WingsFallingontheDrawbridge = 14,
        FindTreasures = 15,
        MushroomsHideAndSeek = 16,
        B_FindMysteriousOres = 17,
        W_FindShiningOres = 18,
        The2LostTreasures = 19,
        BigHarvestofBerries = 20,
        RingtheBell = 21,
        TheBellthatRings3Times = 22,
        PathtoanAce = 23,
        ShockingShopping = 24,
        MemoryTraining = 25,
        PushtheLimitofYourMemory = 26,
        FindRustlingGrass = 27,
        FindShards = 28,
        B_ForgottenLostItems = 29,
        W_NotFoundLostItems = 30,
        B_WhatistheBestPrice = 31,
        W_WhatistheRealPrice = 32,
        GivemetheItem = 33,
        DoaGreatTradeUp = 34,
        SearchHiddenGrottes = 35,
        B_NoisyHiddenGrottes = 36,
        W_QuietHiddenGrottes = 37,
        FishingCompetition = 38,
        MulchCollector = 39,
        WhereareFlutteringHearts = 40,
        RockPaperScissorsCompetition = 41,
        TakeaWalkwithEggs = 42,
        FindSteelix = 43,
        TheBerryHuntingAdventure = 44,
    }

    public struct Funfest5Score
    {
        public uint RawValue;
        public Funfest5Score(uint raw) => RawValue = raw;

        public Funfest5Score(int total, int score, int level, bool isNew)
        {
            RawValue = 0;
            Total = total;
            Score = score;
            Level = level;
            IsNew = isNew;
        }

        // Structure - 32bits
        // u32 bestTotal:14
        // u32 bestScore:14
        // u32 level:3
        // u32 isNew:1

        public int Total
        {
            get => (int)(RawValue & 0x3FFFu);
            set => RawValue = (RawValue & ~0x3FFFu) | ((uint)value & 0x3FFFu);
        }

        public int Score
        {
            get => (int)((RawValue >> 14) & 0x3FFFu);
            set => RawValue = (RawValue & 0xF0003FFFu) | (((uint)value & 0x3FFFu) << 14);
        }

        public int Level
        {
            get => (int)((RawValue >> 28) & 0x7u);
            set => RawValue = (RawValue & 0x8FFFFFFFu) | (((uint)value & 0x7u) << 28);
        }

        public bool IsNew
        {
            get => RawValue >> 31 == 1;
            set => RawValue = (RawValue & 0x7FFFFFFFu) | ((value ? 1u : 0) << 31);
        }

        public override bool Equals(object obj) => obj is Funfest5Score f && f == this;
        public override int GetHashCode() => RawValue.GetHashCode();
        public static bool operator ==(Funfest5Score left, Funfest5Score right) => left.RawValue == right.RawValue;
        public static bool operator !=(Funfest5Score left, Funfest5Score right) => !(left == right);
    }
}