PKHeX/PKHeX.Core/Game/Locations/Locations.cs

namespace PKHeX.Core;

/// <summary>
/// Decoration and logic for Met Location IDs
/// </summary>
public static class Locations
{
    public const int LinkTrade4 = 2002;
    public const int LinkTrade5 = 30003;
    public const int LinkTrade6 = 30002;

    public const int Daycare4 = 2000;
    public const int Daycare5 = 60002;
    public const int Daycare8b = 60010;

    public const int LinkTrade2NPC = 126;
    public const int LinkTrade3NPC = 254;
    public const int LinkTrade4NPC = 2001;
    public const int LinkTrade5NPC = 30002;
    public const int LinkTrade6NPC = 30001;

    public const int Breeder5 = 60003;
    public const int Breeder6 = 60004;

    public const int PokeWalker4 = 233;
    public const int Ranger4 = 3001;
    public const int Faraway4 = 3002;

    /// <summary> Goldenrod City in <see cref="GameVersion.C"/> </summary>
    public const int HatchLocationC = 16;

    /// <summary> Route 117 in <see cref="GameVersion.RSE"/> </summary>
    public const int HatchLocationRSE = 32;

    /// <summary> Four Island in <see cref="GameVersion.FRLG"/> </summary>
    public const int HatchLocationFRLG = 146;

    /// <summary> Solaceon Town in <see cref="GameVersion.DPPt"/> </summary>
    public const int HatchLocationDPPt = 4;

    /// <summary> Route 34 in <see cref="GameVersion.HGSS"/> </summary>
    public const int HatchLocationHGSS = 182;

    /// <summary> Skyarrow Bridge in <see cref="GameVersion.Gen5"/> </summary>
    public const int HatchLocation5 = 64;

    /// <summary> Route 7 in <see cref="GameVersion.XY"/> </summary>
    public const int HatchLocation6XY = 38;

    /// <summary> Battle Resort in <see cref="GameVersion.ORAS"/> </summary>
    public const int HatchLocation6AO = 318;

    /// <summary> Paniola Ranch in <see cref="GameVersion.Gen7"/> </summary>
    public const int HatchLocation7 = 78;

    /// <summary> Route 5 in <see cref="GameVersion.SWSH"/> </summary>
    public const int HatchLocation8 = 40;

    /// <summary> Solaceon Town in <see cref="GameVersion.BDSP"/> </summary>
    public const int HatchLocation8b = 446;

    /// <summary> Generation 1 -> Generation 7 Transfer Location (Kanto) </summary>
    public const int Transfer1 = 30013;

    /// <summary> Generation 2 -> Generation 7 Transfer Location (Johto) </summary>
    public const int Transfer2 = 30017;

    /// <summary> Generation 3 -> Generation 4 Transfer Location (Pal Park) </summary>
    public const int Transfer3 = 0x37;

    /// <summary> Generation 4 -> Generation 5 Transfer Location (Poké Transporter) </summary>
    public const int Transfer4 = 30001;

    /// <summary> Generation 4 -> Generation 5 Transfer Location (Crown Celebi - Event not activated in Gen 5) </summary>
    public const int Transfer4_CelebiUnused = 30010;

    /// <summary> Generation 4 -> Generation 5 Transfer Location (Crown Celebi - Event activated in Gen 5) </summary>
    public const int Transfer4_CelebiUsed = 30011;

    /// <summary> Generation 4 -> Generation 5 Transfer Location (Crown Beast - Event not activated in Gen 5) </summary>
    public const int Transfer4_CrownUnused = 30012;

    /// <summary> Generation 4 -> Generation 5 Transfer Location (Crown Beast - Event activated in Gen 5) </summary>
    public const int Transfer4_CrownUsed = 30013;

    /// <summary> Generation 6 Gift from Pokémon Link </summary>
    public const int LinkGift6 = 30011;

    /// <summary> Generation 7 Poké Pelago </summary>
    public const int Pelago7 = 30016;

    /// <summary> Generation 7 Transfer from GO to Pokémon LGP/E's GO Park </summary>
    public const int GO7 = 50;

    /// <summary> Generation 8 Transfer from GO to Pokémon HOME </summary>
    public const int GO8 = 30012;

    /// <summary> Generation 8 Gift from Pokémon HOME </summary>
    public const int HOME8 = 30018;

    public const int BugCatchingContest4 = 207;

    public const int HOME_SHSP = 59998; // SP traded to (SW)SH
    public const int HOME_SWBD = 59999; // BD traded to SW(SH)
    public const int HOME_SWLA = 60000; // PLA traded to SW(SH)
    public const int HOME_SWSHBDSPEgg = 65534; // -2 = 8bNone-1..
    public const int Default8bNone = 65535;

    public static int GetVersionSWSH(int ver) => (GameVersion)ver switch
    {
        GameVersion.PLA => (int)GameVersion.SW,
        GameVersion.BD => (int)GameVersion.SW,
        GameVersion.SP => (int)GameVersion.SH,
        _ => ver,
    };

    public static int GetMetSWSH(int loc, int ver) => (GameVersion)ver switch
    {
        GameVersion.PLA => HOME_SWLA,
        GameVersion.BD => HOME_SWBD,
        GameVersion.SP => HOME_SHSP,
        _ => loc,
    };

    public static bool IsValidMetBDSP(int loc, int ver) => loc switch
    {
        HOME_SHSP when ver == (int)GameVersion.SH => true,
        HOME_SWBD when ver == (int)GameVersion.SW => true,
        _ => false,
    };

    public static int TradedEggLocationNPC(int generation) => generation switch
    {
        1 => LinkTrade2NPC,
        2 => LinkTrade2NPC,
        3 => LinkTrade3NPC,
        4 => LinkTrade4NPC,
        5 => LinkTrade5NPC,
        _ => LinkTrade6NPC,
    };

    public static int TradedEggLocation(int generation, GameVersion ver) => generation switch
    {
        4 => LinkTrade4,
        5 => LinkTrade5,
        8 when GameVersion.BDSP.Contains(ver) => LinkTrade6NPC,
        _ => LinkTrade6,
    };

    public static bool IsPtHGSSLocation(int location) => location is > 111 and < 2000;
    public static bool IsPtHGSSLocationEgg(int location) => location is > 2010 and < 3000;
    public static bool IsEventLocation3(int location) => location is 255;
    public static bool IsEventLocation4(int location) => location is >= 3000 and <= 3076;
    public static bool IsEventLocation5(int location) => location is > 40000 and < 50000;

    private const int SafariLocation_RSE = 57;
    private const int SafariLocation_FRLG = 136;
    private const int SafariLocation_HGSS = 202;
    private const int MarshLocation_DPPt = 52;
    public static bool IsSafariZoneLocation3(int loc) => loc is SafariLocation_RSE or SafariLocation_FRLG;
    public static bool IsSafariZoneLocation4(int loc) => loc is MarshLocation_DPPt or SafariLocation_HGSS;
    public static bool IsSafariZoneLocation8b(int loc) => loc is (>= 219 and <= 224);

    public static bool IsEggLocationBred4(int loc, GameVersion ver)
    {
        if (loc is Daycare4 or LinkTrade4)
            return true;
        return loc == Faraway4 && ver is GameVersion.Pt or GameVersion.HG or GameVersion.SS;
    }

    public static bool IsEggLocationBred5(int loc) => loc is Daycare5 or LinkTrade5;
    public static bool IsEggLocationBred6(int loc) => loc is Daycare5 or LinkTrade6;
    public static bool IsEggLocationBred8b(int loc) => loc is Daycare8b or LinkTrade6NPC;

    public static int GetDaycareLocation(int generation, GameVersion version) => generation switch
    {
        1 or 2 or 3 => 0,
        4 => Daycare4,
        5 => Daycare5,
        8 when version is GameVersion.BD or GameVersion.SP => Daycare8b,
        _ => Daycare5,
    };
}