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;
}
}
}