mirror of
https://github.com/kwsch/PKHeX
synced 2024-11-27 22:40:22 +00:00
386 lines
No EOL
14 KiB
C#
386 lines
No EOL
14 KiB
C#
using System.Collections.Generic;
|
|
using System.Linq;
|
|
|
|
namespace PKHeX.Core
|
|
{
|
|
internal static class MoveTechnicalMachine
|
|
{
|
|
internal static GameVersion GetIsMachineMove(PKM pkm, int species, int form, int generation, int move, GameVersion ver = GameVersion.Any, bool RemoveTransfer = false)
|
|
{
|
|
if (pkm.IsMovesetRestricted())
|
|
ver = (GameVersion) pkm.Version;
|
|
switch (generation)
|
|
{
|
|
case 1: return GetIsMachine1(species, move);
|
|
case 2:
|
|
if (pkm.VC1 && move > Legal.MaxMoveID_1)
|
|
return Legal.NONE;
|
|
return GetIsMachine2(species, move);
|
|
case 3: return GetIsMachine3(species, move, pkm.Format, RemoveTransfer);
|
|
case 4: return GetIsMachine4(species, move, pkm.Format, RemoveTransfer, form);
|
|
case 5: return GetIsMachine5(species, move, form);
|
|
case 6: return GetIsMachine6(species, move, form, ver);
|
|
case 7: return GetIsMachine7(species, move, form, ver);
|
|
default:
|
|
return Legal.NONE;
|
|
}
|
|
}
|
|
|
|
private static GameVersion GetIsMachine1(int species, int move)
|
|
{
|
|
for (int i = 0; i < Legal.TMHM_RBY.Length; i++)
|
|
{
|
|
if (Legal.TMHM_RBY[i] != move)
|
|
continue;
|
|
if (PersonalTable.RB[species].TMHM[i])
|
|
return GameVersion.RB;
|
|
if (PersonalTable.Y[species].TMHM[i])
|
|
return GameVersion.Y;
|
|
}
|
|
|
|
return Legal.NONE;
|
|
}
|
|
|
|
private static GameVersion GetIsMachine2(int species, int move)
|
|
{
|
|
for (int i = 0; i < Legal.TMHM_GSC.Length; i++)
|
|
{
|
|
if (Legal.TMHM_GSC[i] == move && PersonalTable.C[species].TMHM[i])
|
|
return GameVersion.GS;
|
|
}
|
|
|
|
return Legal.NONE;
|
|
}
|
|
|
|
private static GameVersion GetIsMachine3(int species, int move, int format, bool RemoveTransfer)
|
|
{
|
|
for (int i = 0; i < Legal.TM_3.Length; i++)
|
|
{
|
|
if (Legal.TM_3[i] == move && PersonalTable.E[species].TMHM[i])
|
|
return GameVersion.Gen3;
|
|
}
|
|
|
|
if (!RemoveTransfer && format <= 3)
|
|
return GetIsMachine3HM(species, move);
|
|
|
|
return Legal.NONE;
|
|
}
|
|
|
|
private static GameVersion GetIsMachine3HM(int species, int move)
|
|
{
|
|
int x = 0;
|
|
foreach (var m in Legal.HM_3)
|
|
{
|
|
if (m == move && PersonalTable.E[species].TMHM[x + 50])
|
|
return GameVersion.Gen3;
|
|
x++;
|
|
}
|
|
return Legal.NONE;
|
|
}
|
|
|
|
private static GameVersion GetIsMachine4(int species, int move, int format, bool RemoveTransfer, int form)
|
|
{
|
|
for (int i = 0; i < Legal.TM_3.Length; i++)
|
|
{
|
|
if (Legal.TM_4[i] == move && PersonalTable.HGSS.GetFormeEntry(species, form).TMHM[i])
|
|
return GameVersion.Gen4;
|
|
}
|
|
|
|
if (RemoveTransfer && format > 4)
|
|
return GetIsMachine4HMTransfer(species, move, form);
|
|
|
|
return GetIsMachine4HM(species, move, form);
|
|
}
|
|
|
|
private static GameVersion GetIsMachine4HMTransfer(int species, int move, int form)
|
|
{
|
|
// The combination of both these moves is illegal, it should be checked that the pokemon only learn one
|
|
// except if it can learn any of these moves in gen 5 or later
|
|
switch (move)
|
|
{
|
|
case 250: // Whirlpool
|
|
if (PersonalTable.HGSS.GetFormeEntry(species, form).TMHM[96])
|
|
return GameVersion.HGSS;
|
|
break;
|
|
case 432: // Defog
|
|
if (PersonalTable.Pt.GetFormeEntry(species, form).TMHM[96])
|
|
return GameVersion.DPPt;
|
|
break;
|
|
}
|
|
return Legal.NONE;
|
|
}
|
|
|
|
private static GameVersion GetIsMachine4HM(int species, int move, int form)
|
|
{
|
|
{
|
|
int i = 0;
|
|
foreach (var m in Legal.HM_DPPt)
|
|
{
|
|
if (m == move)
|
|
{
|
|
if (PersonalTable.Pt.GetFormeEntry(species, form).TMHM[i + 92])
|
|
return GameVersion.DPPt;
|
|
break;
|
|
}
|
|
i++;
|
|
}
|
|
foreach (var m in Legal.HM_HGSS)
|
|
{
|
|
if (m == move)
|
|
{
|
|
if (PersonalTable.HGSS.GetFormeEntry(species, form).TMHM[i + 92])
|
|
return GameVersion.HGSS;
|
|
break;
|
|
}
|
|
i++;
|
|
}
|
|
}
|
|
return Legal.NONE;
|
|
}
|
|
|
|
private static GameVersion GetIsMachine5(int species, int move, int form)
|
|
{
|
|
for (int i = 0; i < Legal.TMHM_BW.Length; i++)
|
|
{
|
|
if (Legal.TMHM_BW[i] == move)
|
|
return PersonalTable.B2W2.GetFormeEntry(species, form).TMHM[i] ? GameVersion.Gen5 : Legal.NONE;
|
|
}
|
|
return Legal.NONE;
|
|
}
|
|
|
|
private static GameVersion GetIsMachine6(int species, int move, int form, GameVersion ver)
|
|
{
|
|
if (GameVersion.XY.Contains(ver))
|
|
{
|
|
for (int i = 0; i < Legal.TMHM_XY.Length; i++)
|
|
{
|
|
if (Legal.TMHM_XY[i] != move)
|
|
continue;
|
|
if (PersonalTable.XY.GetFormeEntry(species, form).TMHM[i])
|
|
return GameVersion.XY;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (GameVersion.ORAS.Contains(ver))
|
|
{
|
|
for (int i = 0; i < Legal.TMHM_AO.Length; i++)
|
|
{
|
|
if (Legal.TMHM_AO[i] != move)
|
|
continue;
|
|
if (PersonalTable.AO.GetFormeEntry(species, form).TMHM[i])
|
|
return GameVersion.ORAS;
|
|
break;
|
|
}
|
|
}
|
|
|
|
return Legal.NONE;
|
|
}
|
|
|
|
private static GameVersion GetIsMachine7(int species, int move, int form, GameVersion ver)
|
|
{
|
|
if (GameVersion.GG.Contains(ver))
|
|
{
|
|
for (int i = 0; i < Legal.TMHM_GG.Length; i++)
|
|
{
|
|
if (Legal.TMHM_GG[i] != move)
|
|
continue;
|
|
if (PersonalTable.GG.GetFormeEntry(species, form).TMHM[i])
|
|
return GameVersion.GG;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (GameVersion.SM.Contains(ver) && species <= Legal.MaxSpeciesID_7)
|
|
{
|
|
for (int i = 0; i < Legal.TMHM_SM.Length; i++)
|
|
{
|
|
if (Legal.TMHM_SM[i] != move)
|
|
continue;
|
|
if (PersonalTable.SM.GetFormeEntry(species, form).TMHM[i])
|
|
return GameVersion.SM;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (GameVersion.USUM.Contains(ver) && species <= Legal.MaxSpeciesID_7_USUM)
|
|
{
|
|
for (int i = 0; i < Legal.TMHM_SM.Length; i++)
|
|
{
|
|
if (Legal.TMHM_SM[i] != move)
|
|
continue;
|
|
if (PersonalTable.USUM.GetFormeEntry(species, form).TMHM[i])
|
|
return GameVersion.USUM;
|
|
break;
|
|
}
|
|
}
|
|
|
|
return Legal.NONE;
|
|
}
|
|
|
|
internal static IEnumerable<int> GetTMHM(PKM pkm, int species, int form, int generation, GameVersion ver = GameVersion.Any, bool RemoveTransfer = true)
|
|
{
|
|
var r = new List<int>();
|
|
if (pkm.IsMovesetRestricted())
|
|
ver = (GameVersion)pkm.Version;
|
|
|
|
switch (generation)
|
|
{
|
|
case 1: AddMachine1(r, species); break;
|
|
case 2: AddMachine2(r, species);
|
|
if (pkm.Format >= 7 && pkm.VC1)
|
|
r.RemoveAll(z => z > Legal.MaxMoveID_1);
|
|
break;
|
|
case 3: AddMachine3(r, species, pkm.Format, RemoveTransfer); break;
|
|
case 4: AddMachine4(r, species, pkm.Format, RemoveTransfer, form); break;
|
|
case 5: AddMachine5(r, species, form); break;
|
|
case 6: AddMachine6(r, species, form, ver); break;
|
|
case 7: AddMachine7(r, species, form, ver); break;
|
|
}
|
|
return r.Distinct();
|
|
}
|
|
|
|
private static void AddMachine1(List<int> r, int species)
|
|
{
|
|
int index = PersonalTable.RB.GetFormeIndex(species, 0);
|
|
if (index == 0)
|
|
return;
|
|
var pi_rb = (PersonalInfoG1)PersonalTable.RB[index];
|
|
var pi_y = (PersonalInfoG1)PersonalTable.Y[index];
|
|
r.AddRange(Legal.TMHM_RBY.Where((_, m) => pi_rb.TMHM[m]));
|
|
r.AddRange(Legal.TMHM_RBY.Where((_, m) => pi_y.TMHM[m]));
|
|
}
|
|
|
|
private static void AddMachine2(List<int> r, int species)
|
|
{
|
|
int index = PersonalTable.C.GetFormeIndex(species, 0);
|
|
if (index == 0)
|
|
return;
|
|
var pi_c = PersonalTable.C[index];
|
|
r.AddRange(Legal.TMHM_GSC.Where((_, m) => pi_c.TMHM[m]));
|
|
}
|
|
|
|
private static void AddMachine3(List<int> r, int species, int format, bool RemoveTransfer)
|
|
{
|
|
int index = PersonalTable.E.GetFormeIndex(species, 0);
|
|
if (index == 0)
|
|
return;
|
|
var pi_c = PersonalTable.E[index];
|
|
r.AddRange(Legal.TM_3.Where((_, m) => pi_c.TMHM[m]));
|
|
|
|
if (!RemoveTransfer || format == 3) // HM moves must be removed for 3->4, only give if current format.
|
|
r.AddRange(Legal.HM_3.Where((_, m) => pi_c.TMHM[m + 50]));
|
|
else if (format > 3) //Remove HM
|
|
r.AddRange(Legal.HM_3.Where((_, m) => pi_c.TMHM[m + 50]).Except(Legal.HM_3));
|
|
}
|
|
|
|
private static void AddMachine4(List<int> r, int species, int format, bool RemoveTransfer, int form)
|
|
{
|
|
var pi_hgss = PersonalTable.HGSS.GetFormeEntry(species, form);
|
|
var pi_dppt = PersonalTable.Pt.GetFormeEntry(species, form);
|
|
r.AddRange(Legal.TM_4.Where((_, m) => pi_hgss.TMHM[m]));
|
|
|
|
if (RemoveTransfer && format > 4)
|
|
{
|
|
// The combination of both these moves is illegal, it should be checked that the pokemon only learn one
|
|
// except if it can learn any of these moves in gen 5 or later
|
|
if (pi_hgss.TMHM[96])
|
|
r.Add(250); // Whirlpool
|
|
if (pi_dppt.TMHM[96])
|
|
r.Add(432); // Defog
|
|
}
|
|
else
|
|
{
|
|
r.AddRange(Legal.HM_DPPt.Where((_, m) => pi_dppt.TMHM[m + 92]));
|
|
r.AddRange(Legal.HM_HGSS.Where((_, m) => pi_hgss.TMHM[m + 92]));
|
|
}
|
|
}
|
|
|
|
private static void AddMachine5(List<int> r, int species, int form)
|
|
{
|
|
var pi = PersonalTable.B2W2.GetFormeEntry(species, form);
|
|
r.AddRange(Legal.TMHM_BW.Where((_, m) => pi.TMHM[m]));
|
|
}
|
|
|
|
private static void AddMachine6(List<int> r, int species, int form, GameVersion ver = GameVersion.Any)
|
|
{
|
|
switch (ver)
|
|
{
|
|
case GameVersion.Any: // Start at the top, hit every table
|
|
case GameVersion.X:
|
|
case GameVersion.Y:
|
|
case GameVersion.XY:
|
|
AddMachine6XY(r, species, form);
|
|
if (ver == GameVersion.Any) // Fall Through
|
|
AddMachine6AO(r, species, form);
|
|
break;
|
|
case GameVersion.AS:
|
|
case GameVersion.OR:
|
|
case GameVersion.ORAS:
|
|
AddMachine6AO(r, species, form);
|
|
break;
|
|
}
|
|
}
|
|
|
|
private static void AddMachine7(List<int> r, int species, int form, GameVersion ver = GameVersion.Any)
|
|
{
|
|
switch (ver)
|
|
{
|
|
case GameVersion.GP:
|
|
case GameVersion.GE:
|
|
case GameVersion.GG:
|
|
case GameVersion.GO:
|
|
AddMachineGG(r, species, form);
|
|
return;
|
|
case GameVersion.SN:
|
|
case GameVersion.MN:
|
|
case GameVersion.SM:
|
|
if (species <= Legal.MaxSpeciesID_7)
|
|
AddMachineSM(r, species, form);
|
|
return;
|
|
case GameVersion.Any:
|
|
case GameVersion.US:
|
|
case GameVersion.UM:
|
|
case GameVersion.USUM:
|
|
AddMachineUSUM(r, species, form);
|
|
if (ver == GameVersion.Any) // Fall Through
|
|
AddMachineSM(r, species, form);
|
|
return;
|
|
}
|
|
}
|
|
|
|
private static void AddMachine6XY(List<int> r, int species, int form)
|
|
{
|
|
var pi = PersonalTable.XY.GetFormeEntry(species, form);
|
|
r.AddRange(Legal.TMHM_XY.Where((_, m) => pi.TMHM[m]));
|
|
}
|
|
|
|
private static void AddMachine6AO(List<int> r, int species, int form)
|
|
{
|
|
var pi = PersonalTable.AO.GetFormeEntry(species, form);
|
|
r.AddRange(Legal.TMHM_AO.Where((_, m) => pi.TMHM[m]));
|
|
}
|
|
|
|
private static void AddMachineSM(List<int> r, int species, int form)
|
|
{
|
|
if (species > Legal.MaxSpeciesID_7)
|
|
return;
|
|
var pi = PersonalTable.SM.GetFormeEntry(species, form);
|
|
r.AddRange(Legal.TMHM_SM.Where((_, m) => pi.TMHM[m]));
|
|
}
|
|
|
|
private static void AddMachineUSUM(List<int> r, int species, int form)
|
|
{
|
|
var pi = PersonalTable.USUM.GetFormeEntry(species, form);
|
|
r.AddRange(Legal.TMHM_SM.Where((_, m) => pi.TMHM[m]));
|
|
}
|
|
|
|
private static void AddMachineGG(List<int> r, int species, int form)
|
|
{
|
|
if (species > Legal.MaxSpeciesID_7b)
|
|
return;
|
|
var pi = PersonalTable.GG.GetFormeEntry(species, form);
|
|
r.AddRange(Legal.TMHM_GG.Where((_, m) => pi.TMHM[m]));
|
|
}
|
|
}
|
|
} |