using System;
namespace PKHeX.Core;
///
/// Exposes information about the amount of Awakened stat boosts gained.
///
/// Used only in LGP/E.
public interface IAwakened
{
byte AV_HP { get; set; }
byte AV_ATK { get; set; }
byte AV_DEF { get; set; }
byte AV_SPE { get; set; }
byte AV_SPA { get; set; }
byte AV_SPD { get; set; }
}
public static class AwakeningUtil
{
///
/// Sums all values.
///
/// Data to sum with
public static int AwakeningSum(this IAwakened pk) => pk.AV_HP + pk.AV_ATK + pk.AV_DEF + pk.AV_SPE + pk.AV_SPA + pk.AV_SPD;
///
/// Clears all values.
///
/// Data to clear from
public static void AwakeningClear(this IAwakened pk) => pk.AwakeningSetAllTo(0);
///
/// Sets all values to the maximum value.
///
/// Data to set values for
public static void AwakeningMax(this IAwakened pk) => pk.AwakeningSetAllTo(Legal.AwakeningMax);
///
/// Sets all values to the specified value.
///
/// Data to set values for
/// Value to set all to
public static void AwakeningSetAllTo(this IAwakened pk, byte value) => pk.AV_HP = pk.AV_ATK = pk.AV_DEF = pk.AV_SPE = pk.AV_SPA = pk.AV_SPD = value;
///
/// Sets all values to the bare minimum legal value.
///
/// Data to set values for
public static void AwakeningMinimum(this IAwakened pk)
{
if (pk is not PB7 pb7)
return;
Span result = stackalloc byte[6];
SetExpectedMinimumAVs(result, pb7);
AwakeningSetVisual(pb7, result);
}
///
/// Sets all values to the specified value.
///
/// Data to set values for
/// Minimum value to set
/// Maximum value to set
public static void AwakeningSetRandom(this IAwakened pk, byte min = 0, int max = Legal.AwakeningMax)
{
if (pk is not PB7 pb7)
return;
Span result = stackalloc byte[6];
SetExpectedMinimumAVs(result, pb7);
var rnd = Util.Rand;
foreach (ref var av in result)
{
var realMin = Math.Max(min, av);
var realMax = Math.Max(av, max);
av = (byte)rnd.Next(realMin, realMax + 1);
}
AwakeningSetVisual(pb7, result);
}
///
/// Gets the awakening values according to their displayed order.
///
/// Data to set values for
/// Value storage result
public static void AwakeningGetVisual(IAwakened pk, Span value)
{
value[0] = pk.AV_HP;
value[1] = pk.AV_ATK;
value[2] = pk.AV_DEF;
value[3] = pk.AV_SPA;
value[4] = pk.AV_SPD;
value[5] = pk.AV_SPE;
}
///
/// Sets the awakening values according to their displayed order.
///
/// Data to set values for
/// Value storage to set from
public static void AwakeningSetVisual(IAwakened pk, ReadOnlySpan value)
{
pk.AV_HP = value[0];
pk.AV_ATK = value[1];
pk.AV_DEF = value[2];
pk.AV_SPA = value[3];
pk.AV_SPD = value[4];
pk.AV_SPE = value[5];
}
///
/// Gets if all values are within legal limits.
///
/// Data to check
public static bool AwakeningAllValid(this IAwakened pk)
{
if (pk.AV_HP > Legal.AwakeningMax)
return false;
if (pk.AV_ATK > Legal.AwakeningMax)
return false;
if (pk.AV_DEF > Legal.AwakeningMax)
return false;
if (pk.AV_SPE > Legal.AwakeningMax)
return false;
if (pk.AV_SPA > Legal.AwakeningMax)
return false;
if (pk.AV_SPD > Legal.AwakeningMax)
return false;
return true;
}
///
/// Sets one of the values based on its index within the array.
///
/// Pokémon to modify.
/// Index to set to
/// Value to set
public static byte SetAV(this IAwakened pk, int index, byte value) => index switch
{
0 => pk.AV_HP = value,
1 => pk.AV_ATK = value,
2 => pk.AV_DEF = value,
3 => pk.AV_SPE = value,
4 => pk.AV_SPA = value,
5 => pk.AV_SPD = value,
_ => throw new ArgumentOutOfRangeException(nameof(index)),
};
///
/// Sets one of the values based on its index within the array.
///
/// Pokémon to check.
/// Index to get
public static byte GetAV(this IAwakened pk, int index) => index switch
{
0 => pk.AV_HP,
1 => pk.AV_ATK,
2 => pk.AV_DEF,
3 => pk.AV_SPE,
4 => pk.AV_SPA,
5 => pk.AV_SPD,
_ => throw new ArgumentOutOfRangeException(nameof(index)),
};
///
/// Loads the values to the input span.
///
public static void GetAVs(this IAwakened pk, Span value)
{
if (value.Length != 6)
return;
value[0] = pk.AV_HP;
value[1] = pk.AV_ATK;
value[2] = pk.AV_DEF;
value[3] = pk.AV_SPE;
value[4] = pk.AV_SPA;
value[5] = pk.AV_SPD;
}
///
/// Sets the values based on the current .
///
/// Accessor for setting the values
/// Retriever for IVs
public static void SetSuggestedAwakenedValues(this IAwakened a, PKM pk)
{
Span result = stackalloc byte[6];
SetExpectedMinimumAVs(result, (PB7)a);
a.AV_HP = Legal.AwakeningMax;
a.AV_ATK = pk.IV_ATK == 0 ? result[1] : Legal.AwakeningMax;
a.AV_DEF = Legal.AwakeningMax;
a.AV_SPA = Legal.AwakeningMax;
a.AV_SPD = Legal.AwakeningMax;
a.AV_SPE = pk.IV_SPE == 0 ? result[5] : Legal.AwakeningMax;
}
public static bool IsAwakeningBelow(this IAwakened current, IAwakened initial) => !current.IsAwakeningAboveOrEqual(initial);
///
/// Checks if the has values greater or equal to the .
///
public static bool IsAwakeningAboveOrEqual(this IAwakened current, IAwakened initial)
{
if (current.AV_HP < initial.AV_HP)
return false;
if (current.AV_ATK < initial.AV_ATK)
return false;
if (current.AV_DEF < initial.AV_DEF)
return false;
if (current.AV_SPA < initial.AV_SPA)
return false;
if (current.AV_SPD < initial.AV_SPD)
return false;
if (current.AV_SPE < initial.AV_SPE)
return false;
return true;
}
///
/// Updates the span with the expected minimum values for each index.
///
/// Stat results
/// Entity to check
public static void SetExpectedMinimumAVs(Span result, PB7 pk)
{
// GO Park transfers start with 2 AVs for all stats.
// Every other encounter is either all 0, or can legally start at 0 (trades).
if (pk.Version == (int)GameVersion.GO)
result.Fill(GP1.InitialAV);
// Leveling up in-game adds 1 AV to a "random" index.
var start = pk.Met_Level;
var end = pk.CurrentLevel;
if (start == end)
return;
// Level up from met level to current level.
var nature = pk.Nature;
var character = pk.Characteristic;
var ec = pk.EncryptionConstant;
for (int i = start + 1; i <= end; i++)
{
var lm10 = i % 10;
var bits = (ec >> (3 * lm10)) & 7;
var index = PB7.GetRandomIndex((int)bits, character, nature);
++result[index];
}
}
}