Mirrors/PKHeX
2
0
Fork 0
mirror of https://github.com/kwsch/PKHeX synced 2024-11-23 20:43:07 +00:00
Code Issues Projects Releases Packages Wiki Activity
PKHeX/PKHeX.Core/Legality/RNG/RaidRNG.cs

311 lines
10 KiB
C#
Raw Normal View History

Verify full raid result with known seed
2020-01-23 08:11:07 +00:00
using System.Linq;
namespace PKHeX.Core
{
Add xmldoc, increase swish util flexibility
2021-01-07 23:34:26 +00:00
/// <summary>
/// Logic for Generating and Verifying Gen8 Raid Templates against PKM data.
/// </summary>
Verify full raid result with known seed
2020-01-23 08:11:07 +00:00
public static class RaidRNG
{
public static bool Verify<T>(this T raid, PK8 pk8, ulong seed) where T: EncounterStatic8Nest<T>
{
Unify concepts with different names AltForm & Form & Forme => Form GenNumber & Generation => Generation Extract out SpeciesForm interface, and re-add IGeneration For those using PKHeX as a dependency, this should be a pretty straightforward manual replacement... GenNumber and AltForm should be quick find-replace`s.
2020-12-11 04:42:30 +00:00
var pi = PersonalTable.SWSH.GetFormEntry(raid.Species, raid.Form);
Verify full raid result with known seed
2020-01-23 08:11:07 +00:00
var ratio = pi.Gender;
var abil = RemapAbilityToParam(raid.Ability);
var IVs = raid.IVs.Count == 0 ? GetBlankIVTemplate() : PKX.ReorderSpeedLast((int[])((int[])raid.IVs).Clone());
return Verify(pk8, seed, IVs, raid.FlawlessIVCount, abil, ratio);
}
public static void ApplyDetailsTo<T>(this T raid, PK8 pk8, ulong seed) where T : EncounterStatic8Nest<T>
{
// Ensure the species-form is set correctly (nature)
pk8.Species = raid.Species;
Unify concepts with different names AltForm & Form & Forme => Form GenNumber & Generation => Generation Extract out SpeciesForm interface, and re-add IGeneration For those using PKHeX as a dependency, this should be a pretty straightforward manual replacement... GenNumber and AltForm should be quick find-replace`s.
2020-12-11 04:42:30 +00:00
pk8.Form = raid.Form;
var pi = PersonalTable.SWSH.GetFormEntry(raid.Species, raid.Form);
Verify full raid result with known seed
2020-01-23 08:11:07 +00:00
var ratio = pi.Gender;
var abil = RemapAbilityToParam(raid.Ability);
var IVs = raid.IVs.Count == 0 ? GetBlankIVTemplate() : PKX.ReorderSpeedLast((int[])((int[])raid.IVs).Clone());
ApplyDetailsTo(pk8, seed, IVs, raid.FlawlessIVCount, abil, ratio);
}
Use more expression return style Reduces indentation & bracketing, a bit more concise
2021-01-02 01:08:49 +00:00
private static int RemapAbilityToParam(int a) => a switch
Verify full raid result with known seed
2020-01-23 08:11:07 +00:00
{
Use more expression return style Reduces indentation & bracketing, a bit more concise
2021-01-02 01:08:49 +00:00
-1 => 254,
0 => 255,
_ => (a >> 1)
};
Verify full raid result with known seed
2020-01-23 08:11:07 +00:00
private static int[] GetBlankIVTemplate() => new[] {-1, -1, -1, -1, -1, -1};
private static bool Verify(PKM pk, ulong seed, int[] ivs, int iv_count, int ability_param, int gender_ratio, sbyte nature_param = -1, Shiny shiny = Shiny.Random)
{
var rng = new Xoroshiro128Plus(seed);
var ec = (uint)rng.NextInt();
if (ec != pk.EncryptionConstant)
return false;
uint pid;
bool isShiny;
if (shiny == Shiny.Random) // let's decide if it's shiny or not!
{
var trID = (uint)rng.NextInt();
pid = (uint)rng.NextInt();
isShiny = GetShinyXor(pid, trID) < 16;
}
else
{
// no need to calculate a fake trainer
pid = (uint)rng.NextInt();
isShiny = shiny == Shiny.Always;
}
if (isShiny)
{
if (!GetIsShiny(pk.TID, pk.SID, pid))
pid = GetShinyPID(pk.TID, pk.SID, pid, 0);
}
else
{
if (GetIsShiny(pk.TID, pk.SID, pid))
pid ^= 0x1000_0000;
}
if (pk.PID != pid)
return false;
const int UNSET = -1;
const int MAX = 31;
for (int i = ivs.Count(z => z == MAX); i < iv_count; i++)
{
int index = (int)rng.NextInt(6);
while (ivs[index] != UNSET)
index = (int)rng.NextInt(6);
ivs[index] = MAX;
}
for (int i = 0; i < 6; i++)
{
if (ivs[i] == UNSET)
ivs[i] = (int)rng.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;
Minor clean Annotations (nullable), some switch cases for readability
2021-03-14 23:16:55 +00:00
int abil = ability_param switch
{
254 => (int)rng.NextInt(3),
255 => (int)rng.NextInt(2),
_ => ability_param
};
Verify full raid result with known seed
2020-01-23 08:11:07 +00:00
abil <<= 1; // 1/2/4
Handle Ability Patch changed abilities (#3024) Handles conditions when Ability Patch has been used to modify the PKM's ability => hidden ability. Per the description, it goes 1/2=>H, not the other way around. Verify ability bit first, as mystery gift case handling skips the bit check logic :P
2020-10-10 19:30:57 +00:00
var current = pk.AbilityNumber;
Fix logic to fix a broken raid test
2020-10-10 19:59:31 +00:00
if (abil == 4)
Handle Ability Patch changed abilities (#3024) Handles conditions when Ability Patch has been used to modify the PKM's ability => hidden ability. Per the description, it goes 1/2=>H, not the other way around. Verify ability bit first, as mystery gift case handling skips the bit check logic :P
2020-10-10 19:30:57 +00:00
{
Fix logic to fix a broken raid test
2020-10-10 19:59:31 +00:00
if (current != 4)
return false;
Handle Ability Patch changed abilities (#3024) Handles conditions when Ability Patch has been used to modify the PKM's ability => hidden ability. Per the description, it goes 1/2=>H, not the other way around. Verify ability bit first, as mystery gift case handling skips the bit check logic :P
2020-10-10 19:30:57 +00:00
}
Fix logic to fix a broken raid test
2020-10-10 19:59:31 +00:00
// else, for things that were made Hidden Ability, defer to Ability Checks (Ability Patch)
Verify full raid result with known seed
2020-01-23 08:11:07 +00:00
switch (gender_ratio)
{
case 255 when pk.Gender != 2:
if (pk.Gender != 2)
return false;
break;
case 254 when pk.Gender != 1:
if (pk.Gender != 1)
return false;
break;
case 000:
if (pk.Gender != 0)
return false;
break;
default:
var gender = (int)rng.NextInt(252) + 1 < gender_ratio ? 1 : 0;
if (pk.Gender != gender)
return false;
break;
}
if (nature_param == -1)
{
Unify concepts with different names AltForm & Form & Forme => Form GenNumber & Generation => Generation Extract out SpeciesForm interface, and re-add IGeneration For those using PKHeX as a dependency, this should be a pretty straightforward manual replacement... GenNumber and AltForm should be quick find-replace`s.
2020-12-11 04:42:30 +00:00
if (pk.Species == (int) Species.Toxtricity && pk.Form == 0)
Verify full raid result with known seed
2020-01-23 08:11:07 +00:00
{
var table = Nature0;
var choice = table[rng.NextInt((uint)table.Length)];
if (pk.Nature != choice)
return false;
}
Unify concepts with different names AltForm & Form & Forme => Form GenNumber & Generation => Generation Extract out SpeciesForm interface, and re-add IGeneration For those using PKHeX as a dependency, this should be a pretty straightforward manual replacement... GenNumber and AltForm should be quick find-replace`s.
2020-12-11 04:42:30 +00:00
else if (pk.Species == (int) Species.Toxtricity && pk.Form == 1)
Verify full raid result with known seed
2020-01-23 08:11:07 +00:00
{
var table = Nature1;
var choice = table[rng.NextInt((uint)table.Length)];
if (pk.Nature != choice)
return false;
}
else
{
var nature = (int)rng.NextInt(25);
if (pk.Nature != nature)
return false;
}
}
else
{
if (pk.Nature != nature_param)
return false;
}
if (pk is IScaledSize s)
{
var height = (int)rng.NextInt(0x81) + (int)rng.NextInt(0x80);
if (s.HeightScalar != height)
return false;
var weight = (int)rng.NextInt(0x81) + (int)rng.NextInt(0x80);
if (s.WeightScalar != weight)
return false;
}
return true;
}
private static bool ApplyDetailsTo(PKM pk, ulong seed, int[] ivs, int iv_count, int ability_param, int gender_ratio, sbyte nature_param = -1, Shiny shiny = Shiny.Random)
{
var rng = new Xoroshiro128Plus(seed);
pk.EncryptionConstant = (uint)rng.NextInt();
uint pid;
bool isShiny;
if (shiny == Shiny.Random) // let's decide if it's shiny or not!
{
var trID = (uint)rng.NextInt();
pid = (uint)rng.NextInt();
isShiny = GetShinyXor(pid, trID) < 16;
}
else
{
// no need to calculate a fake trainer
pid = (uint)rng.NextInt();
isShiny = shiny == Shiny.Always;
}
if (isShiny)
{
if (!GetIsShiny(pk.TID, pk.SID, pid))
pid = GetShinyPID(pk.TID, pk.SID, pid, 0);
}
else
{
if (GetIsShiny(pk.TID, pk.SID, pid))
pid ^= 0x1000_0000;
}
pk.PID = pid;
const int UNSET = -1;
const int MAX = 31;
for (int i = ivs.Count(z => z == MAX); i < iv_count; i++)
{
int index = (int)rng.NextInt(6);
while (ivs[index] != UNSET)
index = (int)rng.NextInt(6);
ivs[index] = MAX;
}
for (int i = 0; i < 6; i++)
{
if (ivs[i] == UNSET)
ivs[i] = (int)rng.NextInt(32);
}
pk.IV_HP = ivs[0];
pk.IV_ATK = ivs[1];
pk.IV_DEF = ivs[2];
pk.IV_SPA = ivs[3];
pk.IV_SPD = ivs[4];
pk.IV_SPE = ivs[5];
Add xmldoc, increase swish util flexibility
2021-01-07 23:34:26 +00:00
int abil = ability_param switch
{
254 => (int)rng.NextInt(3),
255 => (int)rng.NextInt(2),
_ => ability_param
};
Verify full raid result with known seed
2020-01-23 08:11:07 +00:00
pk.RefreshAbility(abil);
Condense some switch expressions Should be it for now; exhaustively searched for all that made sense to change
2020-12-25 18:58:33 +00:00
pk.Gender = gender_ratio switch
Verify full raid result with known seed
2020-01-23 08:11:07 +00:00
{
Condense some switch expressions Should be it for now; exhaustively searched for all that made sense to change
2020-12-25 18:58:33 +00:00
255 => 2,
254 => 1,
000 => 0,
_ => (int) rng.NextInt(252) + 1 < gender_ratio ? 1 : 0
};
Verify full raid result with known seed
2020-01-23 08:11:07 +00:00
int nature;
if (nature_param == -1)
{
Unify concepts with different names AltForm & Form & Forme => Form GenNumber & Generation => Generation Extract out SpeciesForm interface, and re-add IGeneration For those using PKHeX as a dependency, this should be a pretty straightforward manual replacement... GenNumber and AltForm should be quick find-replace`s.
2020-12-11 04:42:30 +00:00
if (pk.Species == (int)Species.Toxtricity && pk.Form == 0)
Verify full raid result with known seed
2020-01-23 08:11:07 +00:00
{
var table = Nature0;
nature = table[rng.NextInt((uint)table.Length)];
}
Unify concepts with different names AltForm & Form & Forme => Form GenNumber & Generation => Generation Extract out SpeciesForm interface, and re-add IGeneration For those using PKHeX as a dependency, this should be a pretty straightforward manual replacement... GenNumber and AltForm should be quick find-replace`s.
2020-12-11 04:42:30 +00:00
else if (pk.Species == (int)Species.Toxtricity && pk.Form == 1)
Verify full raid result with known seed
2020-01-23 08:11:07 +00:00
{
var table = Nature1;
nature = table[rng.NextInt((uint)table.Length)];
}
else
{
nature = (int)rng.NextInt(25);
}
}
else
{
nature = nature_param;
}
pk.StatNature = pk.Nature = nature;
if (pk is IScaledSize s)
{
var height = (int)rng.NextInt(0x81) + (int)rng.NextInt(0x80);
var weight = (int)rng.NextInt(0x81) + (int)rng.NextInt(0x80);
s.HeightScalar = height;
s.WeightScalar = weight;
}
return true;
}
private static uint GetShinyPID(int tid, int sid, uint pid, int type)
{
return (uint) (((tid ^ sid ^ (pid & 0xFFFF) ^ type) << 16) | (pid & 0xFFFF));
}
private static bool GetIsShiny(int tid, int sid, uint pid)
{
return GetShinyXor(pid, (uint) ((sid << 16) | tid)) < 16;
}
private static uint GetShinyXor(uint pid, uint oid)
{
var xor = pid ^ oid;
return (xor ^ (xor >> 16)) & 0xFFFF;
}
private static readonly int[] Nature0 = {3, 4, 2, 8, 9, 19, 22, 11, 13, 14, 0, 6, 24};
private static readonly int[] Nature1 = {1, 5, 7, 10, 12, 15, 16, 17, 18, 20, 21, 23};
}
}
Reference in a new issue Copy permalink