2
0
Fork 0
mirror of https://github.com/kwsch/PKHeX synced 2024-12-28 13:23:10 +00:00
PKHeX/PKHeX.Core/Saves/Substructures/OPower/OPower6.cs
2020-12-21 17:17:56 -08:00

93 lines
3.2 KiB
C#

using System;
using static PKHeX.Core.OPower6Type;
namespace PKHeX.Core
{
public sealed class OPower6 : SaveBlock
{
private static readonly OPowerFlagSet[] Mapping =
{
new(5, Hatching),
new(5, Bargain),
new(5, Prize_Money),
new(5, Exp_Point),
new(5, Capture),
new(3, Encounter),
new(3, Stealth),
new(3, HP_Restoring),
new(3, PP_Restoring),
new(1, Full_Recovery),
new(5, Befriending),
new(3, Attack),
new(3, Defense),
new(3, Sp_Attack),
new(3, Sp_Defense),
new(3, Speed),
new(3, Critical),
new(3, Accuracy),
};
static OPower6()
{
int index = 1; // Skip unused byte
foreach (var m in Mapping)
{
m.Offset = index;
index += m.Count;
}
}
public OPower6(SaveFile sav, int offset) : base(sav) => Offset = offset;
private static OPowerFlagSet Get(OPower6Type type) => Array.Find(Mapping, t => t.Identifier == type);
public static int GetOPowerCount(OPower6Type type) => Get(type).BaseCount;
public int GetOPowerLevel(OPower6Type type) => Get(type).GetOPowerLevel(Data, Offset);
public static bool GetHasOPowerS(OPower6Type type) => Get(type).HasOPowerS;
public static bool GetHasOPowerMAX(OPower6Type type) => Get(type).HasOPowerMAX;
public bool GetOPowerS(OPower6Type type) => Get(type).GetOPowerS(Data, Offset);
public bool GetOPowerMAX(OPower6Type type) => Get(type).GetOPowerMAX(Data, Offset);
public void SetOPowerLevel(OPower6Type type, int lvl) => Get(type).SetOPowerLevel(Data, Offset, lvl);
public void SetOPowerS(OPower6Type type, bool value) => Get(type).SetOPowerS(Data, Offset, value);
public void SetOPowerMAX(OPower6Type type, bool value) => Get(type).SetOPowerMAX(Data, Offset, value);
public bool MasterFlag
{
get => Data[Offset] == 1;
set => Data[Offset] = (byte) (value ? OPowerFlagState.Unlocked : OPowerFlagState.Locked);
}
public void UnlockAll() => ToggleFlags(allEvents: true);
public void UnlockRegular() => ToggleFlags();
public void ClearAll() => ToggleFlags(clearOnly: true);
private void ToggleFlags(bool allEvents = false, bool clearOnly = false)
{
foreach (var m in Mapping)
{
// Clear before applying new value
m.SetOPowerLevel(Data, Offset, 0);
m.SetOPowerS(Data, Offset, false);
m.SetOPowerMAX(Data, Offset, false);
if (clearOnly)
continue;
int lvl = allEvents ? m.BaseCount : (m.BaseCount != 1 ? 3 : 0); // Full_Recovery is ORAS/event only @ 1 level
m.SetOPowerLevel(Data, Offset, lvl);
if (!allEvents)
continue;
m.SetOPowerS(Data, Offset, true);
m.SetOPowerMAX(Data, Offset, true);
}
}
public byte[] Write() => Data;
}
}