mirror of
https://github.com/kwsch/PKHeX
synced 2024-11-27 14:30:56 +00:00
bced546c63
O(1) lookup for arbitrary species, some edge handling rules for specific game islands/forms. With HOME, there's only 3 islands of permissions. No allocation besides the singletons which aren't really necessary.
No longer need to peek within multiple hashsets, just fetch the "is possible" bit from the species listing and check if the bit is set.
Will be fun if ball shell swaps are added 🤞, get to set all bits for anything that can enter game with that feature (if ever added).
Add unit test for gen67 no-patch exclusion
95 lines
4.5 KiB
C#
95 lines
4.5 KiB
C#
using System.Diagnostics.CodeAnalysis;
|
|
using static PKHeX.Core.Ball;
|
|
|
|
namespace PKHeX.Core;
|
|
|
|
internal static class BallUseLegality
|
|
{
|
|
/// <summary>
|
|
/// In Sun/Moon, capturing with Heavy Ball is impossible in Sun/Moon for specific hard to catch species.
|
|
/// </summary>
|
|
/// <param name="species">Encounter species.</param>
|
|
/// <remarks>
|
|
/// Catch rate for these species is 3. Due to the heavy ball modifier adding [-20], the catch rate becomes 0.
|
|
/// </remarks>
|
|
/// <returns>True if it is impossible to capture in a <see cref="Heavy"/> ball.</returns>
|
|
public static bool IsAlolanCaptureNoHeavyBall(ushort species) => species is (int)Species.Beldum or (int)Species.TapuKoko or (int)Species.TapuLele or (int)Species.TapuBulu or (int)Species.TapuFini;
|
|
|
|
public static bool IsBallPermitted([ConstantExpected] ulong permit, int ball)
|
|
{
|
|
if ((uint)ball >= 64)
|
|
return false;
|
|
return (permit & (1ul << ball)) != 0;
|
|
}
|
|
|
|
public static ulong GetWildBalls(int generation, GameVersion game) => generation switch
|
|
{
|
|
1 => WildPokeBalls1,
|
|
2 => WildPokeBalls2,
|
|
3 => WildPokeBalls3,
|
|
4 => GameVersion.HGSS.Contains(game) ? WildPokeBalls4_HGSS : WildPokeBalls4_DPPt,
|
|
5 => WildPokeBalls5,
|
|
6 => WildPokeballs6,
|
|
7 => GameVersion.Gen7b.Contains(game) ? WildPokeballs7b : WildPokeballs7,
|
|
8 when GameVersion.BDSP.Contains(game) => WildPokeBalls4_HGSS,
|
|
8 when GameVersion.PLA == game => WildPokeBalls8a,
|
|
8 => GameVersion.GO == game ? WildPokeballs8g_WithRaid : WildPokeballs8,
|
|
9 => WildPokeballs9,
|
|
_ => default,
|
|
};
|
|
|
|
private const ulong WildPokeRegular = (1 << (int)Master)
|
|
| (1 << (int)Ultra)
|
|
| (1 << (int)Great)
|
|
| (1 << (int)Poke);
|
|
|
|
private const ulong WildPokeKurt4 = (1 << (int)Fast)
|
|
| (1 << (int)Level)
|
|
| (1 << (int)Lure)
|
|
| (1 << (int)Heavy)
|
|
| (1 << (int)Love)
|
|
| (1 << (int)Friend)
|
|
| (1 << (int)Moon);
|
|
|
|
private const ulong WildPokeEnhance3 = (1 << (int)Net)
|
|
| (1 << (int)Dive)
|
|
| (1 << (int)Nest)
|
|
| (1 << (int)Repeat)
|
|
| (1 << (int)Timer)
|
|
| (1 << (int)Luxury)
|
|
| (1 << (int)Premier);
|
|
|
|
private const ulong WildPokeEnhance4 = (1 << (int)Dusk) | (1 << (int)Heal) | (1 << (int)Quick);
|
|
private const ulong WildPokeEnhance5 = (1 << (int)Dream);
|
|
private const ulong WildPokeEnhance7 = (1 << (int)Beast);
|
|
private const ulong WildPokeEnhance8 = (1 << (int)Dream) | (1 << (int)Safari) | (1 << (int)Sport);
|
|
|
|
private const ulong WildPokeBalls8a =
|
|
(1ul << (int)LAPoke)
|
|
| (1ul << (int)LAGreat)
|
|
| (1ul << (int)LAUltra)
|
|
| (1ul << (int)LAFeather)
|
|
| (1ul << (int)LAWing)
|
|
| (1ul << (int)LAJet)
|
|
| (1ul << (int)LAHeavy)
|
|
| (1ul << (int)LALeaden)
|
|
| (1ul << (int)LAGigaton);
|
|
|
|
private const ulong WildPokeBalls1 = 1 << (int)Poke;
|
|
private const ulong WildPokeBalls2 = WildPokeBalls1;
|
|
private const ulong WildPokeBalls3 = WildPokeRegular | WildPokeEnhance3;
|
|
private const ulong WildPokeBalls4_DPPt = WildPokeBalls3 | WildPokeEnhance4;
|
|
public const ulong WildPokeBalls4_HGSS = WildPokeBalls4_DPPt | WildPokeKurt4;
|
|
private const ulong WildPokeBalls5 = WildPokeBalls4_DPPt;
|
|
|
|
public const ulong DreamWorldBalls = WildPokeBalls5 | WildPokeEnhance5;
|
|
private const ulong WildPokeballs6 = WildPokeBalls5; // Same as Gen5
|
|
private const ulong WildPokeballs7 = WildPokeBalls4_HGSS | WildPokeEnhance7; // Same as HGSS + Beast
|
|
private const ulong WildPokeballs8 = WildPokeballs7 | WildPokeEnhance8;
|
|
|
|
private const ulong WildPokeballs7b = WildPokeRegular | (1 << (int)Premier);
|
|
public const ulong WildPokeballs8g_WithRaid = WildPokeballs7b & ~(1ul << (int)Master); // Ultra Great Poke Premier, no Master
|
|
public const ulong WildPokeballs8g_WithoutRaid = WildPokeRegular & ~(1ul << (int)Master); // Ultra Great Poke, no Premier/Master
|
|
|
|
public const ulong WildPokeballs9 = WildPokeballs7 | WildPokeEnhance5; // Same as Gen7 + Dream
|
|
}
|