mirror of
https://github.com/kwsch/PKHeX.git
synced 2026-04-26 02:29:59 -05:00
5cd6f456f0
Fix ability index calc for generate & match Fix message for mystery gift fateful encounter flag should be false Add PA9 to GetBlank for anyone using the method via NuGet dll
315 lines
11 KiB
C#
315 lines
11 KiB
C#
using System;
|
|
using static PKHeX.Core.ShinyUtil;
|
|
|
|
namespace PKHeX.Core;
|
|
|
|
/// <summary>
|
|
/// Logic for Generation 9 Encounter RNG.
|
|
/// </summary>
|
|
public static class Encounter9RNG
|
|
{
|
|
/// <summary>
|
|
/// Sets the <see cref="pk"/> with random data based on the <see cref="enc"/> and <see cref="criteria"/>.
|
|
/// </summary>
|
|
/// <returns>True if the generated data matches the <see cref="criteria"/>.</returns>
|
|
public static bool TryApply32<TEnc>(this TEnc enc, PK9 pk, in ulong init, in GenerateParam9 param, in EncounterCriteria criteria)
|
|
where TEnc : IEncounterTemplate, ITeraRaid9
|
|
{
|
|
const int maxCtr = 100_000;
|
|
var rand = new Xoroshiro128Plus(init);
|
|
for (int ctr = 0; ctr < maxCtr; ctr++)
|
|
{
|
|
uint seed = (uint)rand.NextInt(uint.MaxValue);
|
|
if (!enc.CanBeEncountered(seed))
|
|
{
|
|
ctr--; // don't consider it an attempt
|
|
continue;
|
|
}
|
|
if (!GenerateData(pk, param, criteria, seed, param.IVs.IsSpecified))
|
|
continue;
|
|
|
|
var type = Tera9RNG.GetTeraType(seed, enc.TeraType, enc.Species, enc.Form);
|
|
pk.TeraTypeOriginal = (MoveType)type;
|
|
return true; // done.
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/// <inheritdoc cref="TryApply32{TEnc}(TEnc, PK9, in ulong, in GenerateParam9, in EncounterCriteria)"/>
|
|
public static bool TryApply64<TEnc>(this TEnc enc, PK9 pk, in ulong init, in GenerateParam9 param, in EncounterCriteria criteria)
|
|
where TEnc : ISpeciesForm, IGemType
|
|
{
|
|
var rand = new Xoroshiro128Plus(init);
|
|
const int maxCtr = 100_000;
|
|
for (int ctr = 0; ctr < maxCtr; ctr++)
|
|
{
|
|
ulong seed = rand.Next(); // fake cryptosecure
|
|
if (!GenerateData(pk, param, criteria, seed, param.IVs.IsSpecified))
|
|
continue;
|
|
|
|
var type = Tera9RNG.GetTeraType(seed, enc.TeraType, enc.Species, enc.Form);
|
|
pk.TeraTypeOriginal = (MoveType)type;
|
|
return true; // done.
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Fills out an entity with details from the provided encounter template.
|
|
/// </summary>
|
|
/// <returns>False if the seed cannot generate data matching the criteria.</returns>
|
|
public static bool GenerateData(PK9 pk, in GenerateParam9 enc, in EncounterCriteria criteria, in ulong seed, bool ignoreIVs = false)
|
|
{
|
|
var rand = new Xoroshiro128Plus(seed);
|
|
pk.EncryptionConstant = (uint)rand.NextInt(uint.MaxValue);
|
|
pk.PID = GetAdaptedPID(ref rand, pk, enc);
|
|
|
|
if (enc.Shiny is Shiny.Random && criteria.Shiny.IsShiny() != pk.IsShiny)
|
|
return false;
|
|
|
|
const int UNSET = -1;
|
|
const int MAX = 31;
|
|
Span<int> ivs = [UNSET, UNSET, UNSET, UNSET, UNSET, UNSET];
|
|
if (enc.IVs.IsSpecified)
|
|
{
|
|
enc.IVs.CopyToSpeedLast(ivs);
|
|
}
|
|
else
|
|
{
|
|
for (int i = 0; i < enc.FlawlessIVs; i++)
|
|
{
|
|
int index;
|
|
do { index = (int)rand.NextInt(6); }
|
|
while (ivs[index] != UNSET);
|
|
ivs[index] = MAX;
|
|
}
|
|
}
|
|
|
|
for (int i = 0; i < 6; i++)
|
|
{
|
|
if (ivs[i] == UNSET)
|
|
ivs[i] = (int)rand.NextInt(MAX + 1);
|
|
}
|
|
|
|
if (!ignoreIVs && !criteria.IsIVsCompatibleSpeedLast(ivs))
|
|
return false;
|
|
|
|
pk.IV32 = (uint)ivs[0] |
|
|
(uint)(ivs[1] << 05) |
|
|
(uint)(ivs[2] << 10) |
|
|
(uint)(ivs[5] << 15) | // speed is last in the array, but in the middle of the 32bit value
|
|
(uint)(ivs[3] << 20) |
|
|
(uint)(ivs[4] << 25);
|
|
|
|
int ability = enc.Ability switch
|
|
{
|
|
AbilityPermission.Any12H => 1 << (int)rand.NextInt(3),
|
|
AbilityPermission.Any12 => 1 << (int)rand.NextInt(2),
|
|
_ => (int)enc.Ability,
|
|
};
|
|
pk.RefreshAbility(ability >> 1);
|
|
|
|
var gender_ratio = enc.GenderRatio;
|
|
byte gender = gender_ratio switch
|
|
{
|
|
PersonalInfo.RatioMagicGenderless => 2,
|
|
PersonalInfo.RatioMagicFemale => 1,
|
|
PersonalInfo.RatioMagicMale => 0,
|
|
_ => GetGender(gender_ratio, rand.NextInt(100)),
|
|
};
|
|
if (criteria.IsSpecifiedGender() && !criteria.IsSatisfiedGender(gender))
|
|
return false;
|
|
pk.Gender = gender;
|
|
|
|
var nature = enc.Nature != Nature.Random ? enc.Nature : enc.Species == (int)Species.Toxtricity
|
|
? ToxtricityUtil.GetRandomNature(ref rand, pk.Form)
|
|
: (Nature)rand.NextInt(25);
|
|
|
|
// Compromise on Nature -- some are fixed, some are random. If the request wants a specific nature, just mint it.
|
|
if (criteria.IsSpecifiedNature() && !criteria.IsSatisfiedNature(nature))
|
|
return false;
|
|
pk.Nature = pk.StatNature = nature;
|
|
|
|
pk.HeightScalar = enc.Height != 0 ? enc.Height : (byte)(rand.NextInt(0x81) + rand.NextInt(0x80));
|
|
pk.WeightScalar = enc.Weight != 0 ? enc.Weight : (byte)(rand.NextInt(0x81) + rand.NextInt(0x80));
|
|
pk.Scale = enc.ScaleType.GetSizeValue(enc.Scale, ref rand);
|
|
return true;
|
|
}
|
|
|
|
public static bool IsMatch(PKM pk, in GenerateParam9 enc, in ulong seed)
|
|
{
|
|
// same as above method
|
|
var rand = new Xoroshiro128Plus(seed);
|
|
if (pk.EncryptionConstant != (uint)rand.NextInt(uint.MaxValue))
|
|
return false;
|
|
|
|
var pid = GetAdaptedPID(ref rand, pk, enc);
|
|
if (pk.PID != pid)
|
|
return false;
|
|
|
|
const int UNSET = -1;
|
|
Span<int> ivs = [UNSET, UNSET, UNSET, UNSET, UNSET, UNSET];
|
|
if (enc.IVs.IsSpecified)
|
|
{
|
|
enc.IVs.CopyToSpeedLast(ivs);
|
|
}
|
|
else
|
|
{
|
|
const int MAX = 31;
|
|
for (int i = 0; i < enc.FlawlessIVs; i++)
|
|
{
|
|
int index;
|
|
do { index = (int)rand.NextInt(6); }
|
|
while (ivs[index] != UNSET);
|
|
ivs[index] = MAX;
|
|
}
|
|
}
|
|
|
|
for (int i = 0; i < 6; i++)
|
|
{
|
|
if (ivs[i] == UNSET)
|
|
ivs[i] = (int)rand.NextInt(32);
|
|
}
|
|
|
|
if (pk.IV_HP != ivs[0])
|
|
return false;
|
|
if (pk.IV_ATK != ivs[1])
|
|
return false;
|
|
if (pk.IV_DEF != ivs[2])
|
|
return false;
|
|
if (pk.IV_SPA != ivs[3])
|
|
return false;
|
|
if (pk.IV_SPD != ivs[4])
|
|
return false;
|
|
if (pk.IV_SPE != ivs[5])
|
|
return false;
|
|
|
|
// Ability can be changed by Capsule/Patch.
|
|
// Defer this check to later.
|
|
// ReSharper disable once UnusedVariable
|
|
int ability = enc.Ability switch
|
|
{
|
|
AbilityPermission.Any12H => 1 << (int)rand.NextInt(3),
|
|
AbilityPermission.Any12 => 1 << (int)rand.NextInt(2),
|
|
_ => (int)enc.Ability,
|
|
};
|
|
|
|
var gender_ratio = enc.GenderRatio;
|
|
byte gender = gender_ratio switch
|
|
{
|
|
PersonalInfo.RatioMagicGenderless => 2,
|
|
PersonalInfo.RatioMagicFemale => 1,
|
|
PersonalInfo.RatioMagicMale => 0,
|
|
_ => GetGender(gender_ratio, rand.NextInt(100)),
|
|
};
|
|
if (pk.Gender != gender)
|
|
return false;
|
|
|
|
var nature = enc.Nature != Nature.Random ? enc.Nature : enc.Species == (int)Species.Toxtricity
|
|
? ToxtricityUtil.GetRandomNature(ref rand, pk.Form)
|
|
: (Nature)rand.NextInt(25);
|
|
if (pk.Nature != nature)
|
|
return false;
|
|
|
|
if (enc.Height == 0)
|
|
{
|
|
var value = (byte)(rand.NextInt(0x81) + rand.NextInt(0x80));
|
|
if (!IsHeightMatchSV(pk, value))
|
|
return false;
|
|
}
|
|
if (enc.Weight == 0)
|
|
{
|
|
var value = (byte)(rand.NextInt(0x81) + rand.NextInt(0x80));
|
|
if (pk is IScaledSize s && s.WeightScalar != value)
|
|
return false;
|
|
}
|
|
// Scale
|
|
{
|
|
var value = enc.ScaleType.GetSizeValue(enc.Scale, ref rand);
|
|
if (pk is IScaledSize3 s)
|
|
{
|
|
if (s.Scale != value)
|
|
return false;
|
|
}
|
|
else if (pk is IScaledSize s2)
|
|
{
|
|
if (s2.HeightScalar != value)
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
public static bool IsHeightMatchSV(PKM pk, byte value)
|
|
{
|
|
// HOME copies Scale to Height. Untouched by HOME must match the value.
|
|
// Viewing the save file in HOME will alter it too. Tracker definitely indicates it was viewed.
|
|
if (pk is not (IScaledSize s2 and IScaledSize3 s3))
|
|
return true;
|
|
|
|
// Viewed in HOME.
|
|
if (s2.HeightScalar == s3.Scale)
|
|
return true;
|
|
if (pk is IHomeTrack { HasTracker: true })
|
|
return false;
|
|
|
|
return s2.HeightScalar == value;
|
|
}
|
|
|
|
private static uint GetAdaptedPID(ref Xoroshiro128Plus rand, PKM pk, in GenerateParam9 enc)
|
|
{
|
|
var fakeTID = (uint)rand.NextInt();
|
|
uint pid = (uint)rand.NextInt();
|
|
if (enc.Shiny == Shiny.Random) // let's decide if it's shiny or not!
|
|
{
|
|
int i = 1;
|
|
bool isShiny;
|
|
uint xor;
|
|
while (true)
|
|
{
|
|
xor = GetShinyXor(pid, fakeTID);
|
|
isShiny = xor < 16;
|
|
if (isShiny)
|
|
{
|
|
if (xor != 0)
|
|
xor = 1;
|
|
break;
|
|
}
|
|
if (i >= enc.RollCount)
|
|
break;
|
|
pid = (uint)rand.NextInt();
|
|
i++;
|
|
}
|
|
ForceShinyState(isShiny, ref pid, pk.ID32, xor);
|
|
}
|
|
else if (enc.Shiny == Shiny.Always)
|
|
{
|
|
var tid = (ushort)fakeTID;
|
|
var sid = (ushort)(fakeTID >> 16);
|
|
if (!GetIsShiny6(fakeTID, pid)) // battled
|
|
pid = GetShinyPID(tid, sid, pid, 0);
|
|
if (!GetIsShiny6(pk.ID32, pid)) // captured
|
|
pid = GetShinyPID(pk.TID16, pk.SID16, pid, GetShinyXor(pid, fakeTID) == 0 ? 0u : 1u);
|
|
}
|
|
else // Never
|
|
{
|
|
if (GetIsShiny6(fakeTID, pid)) // battled
|
|
pid ^= 0x1000_0000;
|
|
if (GetIsShiny6(pk.ID32, pid)) // captured
|
|
pid ^= 0x1000_0000;
|
|
}
|
|
return pid;
|
|
}
|
|
|
|
public static byte GetGender(in byte ratio, in ulong rand100) => ratio switch
|
|
{
|
|
EntityGender.VM => rand100 < 12 ? (byte)1 : (byte)0, // 12.5%
|
|
EntityGender.MM => rand100 < 25 ? (byte)1 : (byte)0, // 25%
|
|
EntityGender.HH => rand100 < 50 ? (byte)1 : (byte)0, // 50%
|
|
EntityGender.MF => rand100 < 75 ? (byte)1 : (byte)0, // 75%
|
|
EntityGender.VF => rand100 < 89 ? (byte)1 : (byte)0, // 87.5%
|
|
|
|
_ => throw new ArgumentOutOfRangeException(nameof(ratio)),
|
|
};
|
|
}
|