PKHeX/PKHeX.Core/Saves/Substructures/Gen3/SecretBase3Team.cs
Kurt 47071b41f3
Refactoring: Span-based value writes and method signatures (#3361)
Existing `get`/`set` logic is flawed in that it doesn't work on Big Endian operating systems, and it allocates heap objects when it doesn't need to.

`System.Buffers.Binary.BinaryPrimitives` in the `System.Memory` NuGet package provides both Little Endian and Big Endian methods to read and write data; all the `get`/`set` operations have been reworked to use this new API. This removes the need for PKHeX's manual `BigEndian` class, as all functions are already covered by the BinaryPrimitives API.

The `StringConverter` has now been rewritten to accept a Span to read from & write to, no longer requiring a temporary StringBuilder.

Other Fixes included:
- The Super Training UI for Gen6 has been reworked according to the latest block structure additions.
- Cloning a Stadium2 Save File now works correctly (opening from the Folder browser list).
- Checksum & Sanity properties removed from parent PKM class, and is now implemented via interface.
2022-01-02 21:35:59 -08:00

65 lines
No EOL
2.7 KiB
C#

using System;
using static System.Buffers.Binary.BinaryPrimitives;
namespace PKHeX.Core
{
public sealed class SecretBase3Team
{
private const int O_PID = 0;
private const int O_Moves = 0x18;
private const int O_Species = 0x24;
private const int O_Item = 0x30;
private const int O_Level = 0x3C;
private const int O_EV = 0x42;
private static int GetOffsetPID(int index) => O_PID + (index * 4);
private static int GetOffsetMove(int index, int move) => O_Moves + (index * 8) + (move * 2);
private static int GetOffsetSpecies(int index) => O_Species + (index * 2);
private static int GetOffsetItem(int index) => O_Item + (index * 2);
public readonly SecretBase3PKM[] Team;
private readonly byte[] Data;
public SecretBase3Team(byte[] data)
{
Team = new SecretBase3PKM[6];
for (int i = 0; i < Team.Length; i++)
Team[i] = GetPKM(i);
Data = data;
}
public byte[] Write()
{
for (int i = 0; i < Team.Length; i++)
SetPKM(i);
return Data;
}
private SecretBase3PKM GetPKM(int index) => new()
{
PID = ReadUInt32LittleEndian(Data.AsSpan(GetOffsetPID(index))),
Species = ReadUInt16LittleEndian(Data.AsSpan(GetOffsetSpecies(index))),
HeldItem = ReadUInt16LittleEndian(Data.AsSpan(GetOffsetItem(index))),
Move1 = ReadUInt16LittleEndian(Data.AsSpan(GetOffsetMove(index, 0))),
Move2 = ReadUInt16LittleEndian(Data.AsSpan(GetOffsetMove(index, 1))),
Move3 = ReadUInt16LittleEndian(Data.AsSpan(GetOffsetMove(index, 2))),
Move4 = ReadUInt16LittleEndian(Data.AsSpan(GetOffsetMove(index, 3))),
Level = Data[O_Level + index],
EVAll = Data[O_EV + index],
};
private void SetPKM(int index)
{
var pk = Team[index];
WriteUInt32LittleEndian(Data.AsSpan(GetOffsetPID(index)), pk.PID);
WriteUInt16LittleEndian(Data.AsSpan(GetOffsetSpecies(index)), (ushort)pk.Species);
WriteUInt16LittleEndian(Data.AsSpan(GetOffsetItem(index)), (ushort)pk.HeldItem);
WriteUInt16LittleEndian(Data.AsSpan(GetOffsetMove(index, 0)), (ushort)pk.Move1);
WriteUInt16LittleEndian(Data.AsSpan(GetOffsetMove(index, 1)), (ushort)pk.Move2);
WriteUInt16LittleEndian(Data.AsSpan(GetOffsetMove(index, 2)), (ushort)pk.Move3);
WriteUInt16LittleEndian(Data.AsSpan(GetOffsetMove(index, 3)), (ushort)pk.Move4);
Data[O_Level + index] = (byte)pk.Level;
Data[O_EV + index] = (byte)pk.EVAll;
}
}
}