PKHeX/PKHeX.Core/PKM/HOME/GameDataPK8.cs
Kurt 95fbf66a6e
Refactor: Gen3/4 Lead Encounters, property fixing (#4193)
In addition to the Method 1 (and other sibling PIDIV types) correlation, an encounter can only be triggered if the calls prior land on the Method {1} seed. The RNG community has dubbed these patterns as "Method J" (D/P/Pt), "Method K" (HG/SS), and "Method H" (Gen3, coined by yours truly). The basic gist of these is that they are pre-requisites, like the Shadow locks of Colosseum/XD. 

Rename/re-type a bunch of properties to get the codebase more in line with correct property names & more obvious underlying types.
2024-02-22 21:20:54 -06:00

214 lines
9.3 KiB
C#

using System;
using static System.Buffers.Binary.BinaryPrimitives;
namespace PKHeX.Core;
/// <summary>
/// Side game data for <see cref="PK8"/> data transferred into HOME.
/// </summary>
public sealed class GameDataPK8 : HomeOptional1, IGameDataSide<PK8>, IGigantamax, IDynamaxLevel, ISociability, IGameDataSplitAbility, IPokerusStatus
{
private const HomeGameDataFormat ExpectFormat = HomeGameDataFormat.PK8;
private const int SIZE = HomeCrypto.SIZE_2GAME_PK8;
protected override HomeGameDataFormat Format => ExpectFormat;
public GameDataPK8() : base(SIZE) { }
public GameDataPK8(Memory<byte> buffer) : base(buffer) => EnsureSize(SIZE);
public GameDataPK8 Clone() => new(ToArray());
public int WriteTo(Span<byte> result) => WriteWithHeader(result);
#region Structure
public bool CanGigantamax { get => Data[0x00] != 0; set => Data[0x00] = (byte)(value ? 1 : 0); }
public uint Sociability { get => ReadUInt32LittleEndian(Data[0x01..]); set => WriteUInt32LittleEndian(Data[0x01..], value); }
public ushort Move1 { get => ReadUInt16LittleEndian(Data[0x05..]); set => WriteUInt16LittleEndian(Data[0x05..], value); }
public ushort Move2 { get => ReadUInt16LittleEndian(Data[0x07..]); set => WriteUInt16LittleEndian(Data[0x07..], value); }
public ushort Move3 { get => ReadUInt16LittleEndian(Data[0x09..]); set => WriteUInt16LittleEndian(Data[0x09..], value); }
public ushort Move4 { get => ReadUInt16LittleEndian(Data[0x0B..]); set => WriteUInt16LittleEndian(Data[0x0B..], value); }
public int Move1_PP { get => Data[0x0D]; set => Data[0x0D] = (byte)value; }
public int Move2_PP { get => Data[0x0E]; set => Data[0x0E] = (byte)value; }
public int Move3_PP { get => Data[0x0F]; set => Data[0x0F] = (byte)value; }
public int Move4_PP { get => Data[0x10]; set => Data[0x10] = (byte)value; }
public int Move1_PPUps { get => Data[0x11]; set => Data[0x11] = (byte)value; }
public int Move2_PPUps { get => Data[0x12]; set => Data[0x12] = (byte)value; }
public int Move3_PPUps { get => Data[0x13]; set => Data[0x13] = (byte)value; }
public int Move4_PPUps { get => Data[0x14]; set => Data[0x14] = (byte)value; }
public ushort RelearnMove1 { get => ReadUInt16LittleEndian(Data[0x15..]); set => WriteUInt16LittleEndian(Data[0x15..], value); }
public ushort RelearnMove2 { get => ReadUInt16LittleEndian(Data[0x17..]); set => WriteUInt16LittleEndian(Data[0x17..], value); }
public ushort RelearnMove3 { get => ReadUInt16LittleEndian(Data[0x19..]); set => WriteUInt16LittleEndian(Data[0x19..], value); }
public ushort RelearnMove4 { get => ReadUInt16LittleEndian(Data[0x1B..]); set => WriteUInt16LittleEndian(Data[0x1B..], value); }
public byte DynamaxLevel { get => Data[0x1D]; set => Data[0x1D] = value; }
private Span<byte> PokeJob => Data.Slice(0x1E, 14);
public bool GetPokeJobFlag(int index) => FlagUtil.GetFlag(PokeJob, index >> 3, index & 7);
public void SetPokeJobFlag(int index, bool value) => FlagUtil.SetFlag(PokeJob, index >> 3, index & 7, value);
public bool GetPokeJobFlagAny() => PokeJob.ContainsAnyExcept<byte>(0);
public void ClearPokeJobFlags() => PokeJob.Clear();
public byte Fullness { get => Data[0x2C]; set => Data[0x2C] = value; }
private Span<byte> RecordFlags => Data.Slice(0x2D, 14);
public bool GetMoveRecordFlag(int index) => FlagUtil.GetFlag(RecordFlags, index >> 3, index & 7);
public void SetMoveRecordFlag(int index, bool value) => FlagUtil.SetFlag(RecordFlags, index >> 3, index & 7, value);
public bool GetMoveRecordFlagAny() => RecordFlags.ContainsAnyExcept<byte>(0);
public void ClearMoveRecordFlags() => RecordFlags.Clear();
public int Palma { get => ReadInt32LittleEndian(Data[0x3B..]); set => WriteInt32LittleEndian(Data[0x3B..], value); }
public byte Ball { get => Data[0x3F]; set => Data[0x3F] = value; }
public ushort EggLocation { get => ReadUInt16LittleEndian(Data[0x40..]); set => WriteUInt16LittleEndian(Data[0x40..], value); }
public ushort MetLocation { get => ReadUInt16LittleEndian(Data[0x42..]); set => WriteUInt16LittleEndian(Data[0x42..], value); }
// Rev2 Additions
public byte PokerusState { get => Data[0x44]; set => Data[0x44] = value; }
public ushort Ability { get => ReadUInt16LittleEndian(Data[0x45..]); set => WriteUInt16LittleEndian(Data[0x45..], value); }
public byte AbilityNumber { get => Data[0x47]; set => Data[0x47] = value; }
#endregion
#region Conversion
public PersonalInfo GetPersonalInfo(ushort species, byte form) => PersonalTable.SWSH.GetFormEntry(species, form);
public void CopyTo(PK8 pk, PKH pkh)
{
this.CopyTo(pk);
pk.CanGigantamax = CanGigantamax;
pk.Sociability = Sociability;
pk.DynamaxLevel = DynamaxLevel;
pk.Fullness = Fullness;
pk.Palma = Palma;
PokeJob.CopyTo(pk.PokeJob);
RecordFlags.CopyTo(pk.RecordFlags);
pk.PokerusState = PokerusState;
pk.AbilityNumber = AbilityNumber;
pk.Ability = Ability;
if (!IsOriginallySWSH(pkh.Version, pk.MetLocation))
pk.Version = LocationsHOME.GetVersionSWSH(pkh.Version);
}
public void CopyFrom(PK8 pk, PKH pkh)
{
this.CopyFrom(pk);
CanGigantamax = pk.CanGigantamax;
Sociability = pk.Sociability;
DynamaxLevel = pk.DynamaxLevel;
Fullness = pk.Fullness;
Palma = pk.Palma;
pk.PokeJob.CopyTo(PokeJob);
pk.RecordFlags.CopyTo(RecordFlags);
PokerusState = pk.PokerusState;
AbilityNumber = (byte)pk.AbilityNumber;
Ability = (ushort)pk.Ability;
}
public void CopyFrom(PK7 pk, PKH pkh)
{
this.CopyFrom(pk);
PokerusState = pk.PokerusState;
AbilityNumber = (byte)pk.AbilityNumber;
Ability = (ushort)pk.Ability;
pkh.MarkingValue &= 0b1111_1111_1111;
if (!pk.IsNicknamed)
pkh.Nickname = SpeciesName.GetSpeciesNameGeneration(pk.Species, pk.Language, 8);
if (FormInfo.IsTotemForm(pk.Species, pk.Form))
pkh.Form = FormInfo.GetTotemBaseForm(pk.Species, pk.Form);
}
public PK8 ConvertToPKM(PKH pkh)
{
var pk = new PK8();
pkh.CopyTo(pk);
CopyTo(pk, pkh);
pk.ResetPartyStats();
pk.RefreshChecksum();
return pk;
}
#endregion
/// <summary> Reconstructive logic to best apply suggested values. </summary>
public static GameDataPK8? TryCreate(PKH pkh)
{
if (pkh.DataPB7 is { } x)
return CreateViaPB7(pkh, x);
var side = GetNearestNeighbor(pkh);
if (side is not null)
return Create(side, pkh);
return null;
}
// Ignores LGP/E, already preferred if exists.
private static IGameDataSide? GetNearestNeighbor(PKH pkh) => pkh.DataPK9 as IGameDataSide
?? pkh.DataPB8 as IGameDataSide
?? pkh.DataPA8;
private static GameDataPK8 CreateViaPB7(PKH pkh, GameDataPB7 x)
{
var result = new GameDataPK8();
x.CopyTo(result); // Moves are copied by default.
result.AbilityNumber = x.AbilityNumber;
result.PopulateFromCore(pkh);
return result;
}
private static GameDataPK8 Create(IGameDataSide side, PKH pkh)
{
var result = new GameDataPK8();
result.InitializeFrom(side, pkh);
result.ResetMoves(pkh.Species, pkh.Form, pkh.CurrentLevel, LearnSource8SWSH.Instance, EntityContext.Gen8);
return result;
}
public void InitializeFrom(IGameDataSide side, PKH pkh)
{
// BD/SP->SW/SH: Set the Met Location to the magic Location, set the Egg Location to 0 if -1, otherwise BDSPEgg
// (0 is a valid location, but no eggs can be EggMet there -- only hatched.)
// PLA->SW/SH: Set the Met Location to the magic Location, set the Egg Location to 0 (no eggs in game).
var version = pkh.Version;
var met = side.MetLocation;
var ball = GetBall(side.Ball);
var egg = GetEggLocation(side.EggLocation);
if (!IsOriginallySWSH(version, met))
RemapMetEgg(version, ref met, ref egg);
Ball = ball;
MetLocation = met;
EggLocation = egg;
if (side is IGameDataSplitAbility a)
AbilityNumber = a.AbilityNumber;
if (side is IPokerusStatus p)
PokerusState = p.PokerusState;
PopulateFromCore(pkh);
}
private void PopulateFromCore(PKH pkh)
{
var pi = PersonalTable.SWSH.GetFormEntry(pkh.Species, pkh.Form);
Ability = (ushort)pi.GetAbilityAtIndex(AbilityNumber >> 1);
}
private static void RemapMetEgg(GameVersion version, ref ushort met, ref ushort egg)
{
var remap = LocationsHOME.GetMetSWSH(met, version);
if (remap == met)
return;
met = remap;
egg = egg is not (0 or Locations.Default8bNone) ? LocationsHOME.SWSHEgg : (ushort)0;
}
private static bool IsOriginallySWSH(GameVersion version, ushort loc) => version is GameVersion.SW or GameVersion.SH && !IsFakeMetLocation(loc);
private static bool IsFakeMetLocation(ushort met) => LocationsHOME.IsLocationSWSH(met);
private static byte GetBall(byte ball) => ball > (byte)Core.Ball.Beast ? (byte)4 : ball;
private static ushort GetEggLocation(ushort egg) => egg != Locations.Default8bNone ? egg : (ushort)0;
}