using System;
using System.Collections.Generic;
using System.Linq;
namespace PKHeX.Core
{
///
/// Trade Encounter data
///
///
/// Trade data is fixed level in all cases except for the first few generations of games.
///
public abstract record EncounterTrade : IEncounterable, IMoveset, ILocation, IEncounterMatch
{
public int Species { get; init; }
public int Form { get; init; }
public int Level { get; init; }
public virtual int LevelMin => Level;
public int LevelMax => 100;
public IReadOnlyList Moves { get; init; } = Array.Empty();
public abstract int Generation { get; }
public GameVersion Version { get; }
public int CurrentLevel { get; init; } = -1;
public int Location { get; init; }
public int Ability { get; init; }
public int Gender { get; init; } = -1;
public Nature Nature { get; init; } = Nature.Random;
public Shiny Shiny { get; init; } = Shiny.Never;
public int Ball { get; init; } = 4;
public int TID { get; init; }
public int SID { get; init; }
public int OTGender { get; init; } = -1;
public IReadOnlyList IVs { get; init; } = Array.Empty();
public bool EggEncounter => false;
public int EggLocation { get; init; }
public bool EvolveOnTrade { get; init; }
public int TID7
{
init
{
TID = (ushort) value;
SID = value >> 16;
}
}
private const string _name = "In-game Trade";
public string Name => _name;
public string LongName => _name;
public bool IsNicknamed { get; init; } = true;
public bool IsShiny => Shiny.IsShiny();
public IReadOnlyList Nicknames { get; internal set; } = Array.Empty();
public IReadOnlyList TrainerNames { get; internal set; } = Array.Empty();
public string GetNickname(int language) => (uint)language < Nicknames.Count ? Nicknames[language] : string.Empty;
public string GetOT(int language) => (uint)language < TrainerNames.Count ? TrainerNames[language] : string.Empty;
public bool HasNickname => Nicknames.Count != 0 && IsNicknamed;
public bool HasTrainerName => TrainerNames.Count != 0;
protected EncounterTrade(GameVersion game) => Version = game;
private static int GetDefaultMetLocation(int generation) => generation switch
{
1 => 0,
2 => Locations.LinkTrade2NPC,
3 => Locations.LinkTrade3NPC,
4 => Locations.LinkTrade4NPC,
5 => Locations.LinkTrade5NPC,
6 => Locations.LinkTrade6NPC,
7 => Locations.LinkTrade6NPC, // 7 is same as 6
8 => Locations.LinkTrade6NPC, // 8 is same as 6
_ => throw new IndexOutOfRangeException(nameof(generation)),
};
public PKM ConvertToPKM(ITrainerInfo sav) => ConvertToPKM(sav, EncounterCriteria.Unrestricted);
public PKM ConvertToPKM(ITrainerInfo sav, EncounterCriteria criteria)
{
var pk = PKMConverter.GetBlank(Generation, Version);
sav.ApplyTo(pk);
ApplyDetails(sav, criteria, pk);
return pk;
}
protected virtual void ApplyDetails(ITrainerInfo sav, EncounterCriteria criteria, PKM pk)
{
var version = this.GetCompatibleVersion((GameVersion)sav.Game);
int lang = (int)Language.GetSafeLanguage(Generation, (LanguageID)sav.Language, version);
int level = CurrentLevel > 0 ? CurrentLevel : LevelMin;
if (level == 0)
level = Math.Max(1, LevelMin);
int species = Species;
if (EvolveOnTrade)
species++;
pk.EncryptionConstant = Util.Rand32();
pk.Species = species;
pk.Form = Form;
pk.Language = lang;
pk.OT_Name = pk.Format == 1 ? StringConverter12.G1TradeOTStr : HasTrainerName ? GetOT(lang) : sav.OT;
pk.OT_Gender = HasTrainerName ? Math.Max(0, OTGender) : sav.Gender;
pk.SetNickname(HasNickname ? GetNickname(lang) : string.Empty);
pk.CurrentLevel = level;
pk.Version = (int) version;
pk.TID = TID;
pk.SID = SID;
pk.Ball = Ball;
pk.OT_Friendship = pk.PersonalInfo.BaseFriendship;
SetPINGA(pk, criteria);
SetMoves(pk, version, level);
var time = DateTime.Now;
if (pk.Format != 2 || version == GameVersion.C)
{
var location = Location != 0 ? Location : GetDefaultMetLocation(Generation);
SetMetData(pk, level, location, time);
}
else
{
pk.OT_Gender = 0;
}
if (EggLocation != 0)
SetEggMetData(pk, time);
if (pk.Format < 6)
return;
sav.ApplyHandlingTrainerInfo(pk, force: true);
pk.SetRandomEC();
if (pk is PK6 pk6)
pk6.SetRandomMemory6();
}
protected virtual void SetPINGA(PKM pk, EncounterCriteria criteria)
{
var pi = pk.PersonalInfo;
int gender = criteria.GetGender(Gender, pi);
int nature = (int)criteria.GetNature(Nature);
int ability = criteria.GetAbilityFromNumber(Ability);
PIDGenerator.SetRandomWildPID(pk, Generation, nature, ability, gender);
pk.Nature = pk.StatNature = nature;
pk.Gender = gender;
pk.RefreshAbility(ability);
SetIVs(pk);
}
protected void SetIVs(PKM pk)
{
if (IVs.Count != 0)
pk.SetRandomIVs(IVs, 0);
else
pk.SetRandomIVs(flawless: 3);
}
private void SetMoves(PKM pk, GameVersion version, int level)
{
var moves = Moves.Count != 0 ? Moves : MoveLevelUp.GetEncounterMoves(pk, level, version);
if (pk.Format == 1 && moves.All(z => z == 0))
moves = ((PersonalInfoG1)PersonalTable.RB[Species]).Moves;
pk.SetMoves(moves);
pk.SetMaximumPPCurrent(moves);
}
private void SetEggMetData(PKM pk, DateTime time)
{
pk.Egg_Location = EggLocation;
pk.EggMetDate = time;
}
private static void SetMetData(PKM pk, int level, int location, DateTime time)
{
pk.Met_Level = level;
pk.Met_Location = location;
pk.MetDate = time;
}
public virtual bool IsMatchExact(PKM pkm, DexLevel evo)
{
if (IVs.Count != 0)
{
if (!Legal.GetIsFixedIVSequenceValidSkipRand(IVs, pkm))
return false;
}
if (!IsMatchNatureGenderShiny(pkm))
return false;
if (TID != pkm.TID)
return false;
if (SID != pkm.SID)
return false;
if (!IsMatchLevel(pkm, evo))
return false;
if (CurrentLevel != -1 && CurrentLevel > pkm.CurrentLevel)
return false;
if (Form != evo.Form && !FormInfo.IsFormChangeable(Species, Form, pkm.Form, pkm.Format))
return false;
if (OTGender != -1 && OTGender != pkm.OT_Gender)
return false;
if (EggLocation != pkm.Egg_Location)
return false;
// if (z.Ability == 4 ^ pkm.AbilityNumber == 4) // defer to Ability
// continue;
if (!Version.Contains((GameVersion)pkm.Version))
return false;
return true;
}
private bool IsMatchLevel(PKM pkm, DexLevel evo)
{
if (!pkm.HasOriginalMetLocation)
return evo.Level >= Level;
var loc = Location != 0 ? Location : GetDefaultMetLocation(Generation);
if (loc != pkm.Met_Location)
return false;
if (pkm.Format < 5)
return evo.Level >= Level;
return pkm.Met_Level == Level;
}
protected virtual bool IsMatchNatureGenderShiny(PKM pkm)
{
if (!Shiny.IsValid(pkm))
return false;
if (Gender != -1 && Gender != pkm.Gender)
return false;
if (Nature != Nature.Random && pkm.Nature != (int)Nature)
return false;
return true;
}
public EncounterMatchRating GetMatchRating(PKM pkm)
{
if (IsMatchPartial(pkm))
return EncounterMatchRating.PartialMatch;
if (IsMatchDeferred(pkm))
return EncounterMatchRating.Deferred;
return EncounterMatchRating.Match;
}
protected virtual bool IsMatchDeferred(PKM pkm) => false;
protected virtual bool IsMatchPartial(PKM pkm) => false;
}
}