using System; using System.Collections.Generic; using System.Linq; namespace PKHeX.Core.Searching { /// /// search settings & searcher /// public sealed class SearchSettings { public int Format { private get; init; } public int Generation { private get; init; } public int Species { get; init; } = -1; public int Ability { private get; init; } = -1; public int Nature { private get; init; } = -1; public int Item { private get; init; } = -1; public int Version { private get; init; } = -1; public int HiddenPowerType { private get; init; } = -1; public SearchComparison SearchFormat { private get; init; } public SearchComparison SearchLevel { private get; init; } public bool? SearchShiny { get; set; } public bool? SearchLegal { private get; set; } public bool? SearchEgg { get; set; } public int? ESV { private get; set; } public int? Level { private get; init; } public int IVType { private get; init; } public int EVType { private get; init; } public CloneDetectionMethod SearchClones { private get; set; } public IList BatchInstructions { private get; init; } = Array.Empty(); private StringInstruction[] BatchFilters { get; set; } = Array.Empty(); public readonly List Moves = new(); // ReSharper disable once CollectionNeverUpdated.Global /// /// Extra Filters to be checked after all other filters have been checked. /// /// Collection is iterated right before clones are checked. public List> ExtraFilters { get; } = new(); /// /// Adds a move to the required move list. /// /// public void AddMove(int move) { if (move > 0 && !Moves.Contains(move)) Moves.Add(move); } /// /// Searches the input list, filtering out entries as specified by the settings. /// /// List of entries to search /// Search results that match all criteria public IEnumerable Search(IEnumerable list) { BatchFilters = StringInstruction.GetFilters(BatchInstructions).ToArray(); var result = SearchInner(list); if (SearchClones != CloneDetectionMethod.None) { var method = SearchUtil.GetCloneDetectMethod(SearchClones); result = SearchUtil.GetExtraClones(result, method); } return result; } /// /// Searches the input list, filtering out entries as specified by the settings. /// /// List of entries to search /// Search results that match all criteria public IEnumerable Search(IEnumerable list) { BatchFilters = StringInstruction.GetFilters(BatchInstructions).ToArray(); var result = SearchInner(list); if (SearchClones != CloneDetectionMethod.None) { var method = SearchUtil.GetCloneDetectMethod(SearchClones); string GetHash(SlotCache z) => method(z.Entity); result = SearchUtil.GetExtraClones(result, GetHash); } return result; } private IEnumerable SearchInner(IEnumerable list) { foreach (var pk in list) { if (IsSearchMatch(pk)) continue; yield return pk; } } private IEnumerable SearchInner(IEnumerable list) { foreach (var entry in list) { var pk = entry.Entity; if (IsSearchMatch(pk)) continue; yield return entry; } } private bool IsSearchMatch(PKM pk) { if (!SearchSimple(pk)) return false; if (!SearchIntermediate(pk)) return false; if (!SearchComplex(pk)) return false; foreach (var filter in ExtraFilters) { if (!filter(pk)) return false; } return true; } private bool SearchSimple(PKM pk) { if (Format > 0 && !SearchUtil.SatisfiesFilterFormat(pk, Format, SearchFormat)) return false; if (Species > -1 && pk.Species != Species) return false; if (Ability > -1 && pk.Ability != Ability) return false; if (Nature > -1 && pk.StatNature != Nature) return false; if (Item > -1 && pk.HeldItem != Item) return false; if (Version > -1 && pk.Version != Version) return false; return true; } private bool SearchIntermediate(PKM pk) { if (Generation > 0 && !SearchUtil.SatisfiesFilterGeneration(pk, Generation)) return false; if (Moves.Count > 0 && SearchUtil.SatisfiesFilterMoves(pk, Moves)) return false; if (HiddenPowerType > -1 && pk.HPType != HiddenPowerType) return false; if (SearchShiny != null && pk.IsShiny != SearchShiny) return false; if (IVType > 0 && SearchUtil.SatisfiesFilterIVs(pk, IVType)) return false; if (EVType > 0 && SearchUtil.SatisfiesFilterEVs(pk, EVType)) return false; return true; } private bool SearchComplex(PKM pk) { if (SearchEgg != null && !FilterResultEgg(pk)) return false; if (Level is not null or 0 && !SearchUtil.SatisfiesFilterLevel(pk, SearchLevel, (int) Level)) return false; if (SearchLegal != null && new LegalityAnalysis(pk).Valid != SearchLegal) return false; if (BatchFilters.Length != 0 && !SearchUtil.SatisfiesFilterBatchInstruction(pk, BatchFilters)) return false; return true; } private bool FilterResultEgg(PKM pk) { if (SearchEgg == false) return !pk.IsEgg; if (ESV != null) return pk.IsEgg && pk.PSV == ESV; return pk.IsEgg; } public IReadOnlyList GetVersions(SaveFile sav) => GetVersions(sav, GetFallbackVersion(sav)); public IReadOnlyList GetVersions(SaveFile sav, GameVersion fallback) { if (Version > 0) return new[] {(GameVersion) Version}; return Generation switch { 1 when !ParseSettings.AllowGen1Tradeback => new[] {GameVersion.RD, GameVersion.BU, GameVersion.GN, GameVersion.YW}, 2 when sav is SAV2 {Korean: true} => new[] {GameVersion.GD, GameVersion.SV}, 1 or 2 => new[] { GameVersion.RD, GameVersion.BU, GameVersion.GN, GameVersion.YW, GameVersion.GD, GameVersion.SV, GameVersion.C }, _ when fallback.GetGeneration() == Generation => GameUtil.GetVersionsWithinRange(sav, Generation).ToArray(), _ => GameUtil.GameVersions, }; } private static GameVersion GetFallbackVersion(ITrainerInfo sav) { var parent = GameUtil.GetMetLocationVersionGroup((GameVersion)sav.Game); if (parent == GameVersion.Invalid) parent = GameUtil.GetMetLocationVersionGroup(GameUtil.GetVersion(sav.Generation)); return parent; } } }