using System;
using System.Collections.Generic;
namespace PKHeX.Core
{
///
///
/// encounter area
///
public sealed record EncounterArea2 : EncounterArea
{
private static readonly byte[] BCC_SlotRates = { 20, 20, 10, 10, 05, 05, 10, 10, 05, 05 };
private static readonly byte[] RatesGrass = { 30, 30, 20, 10, 5, 4, 1 };
private static readonly byte[] RatesSurf = { 60, 30, 10 };
internal readonly EncounterTime Time;
public readonly byte Rate;
public readonly IReadOnlyList Rates;
public readonly EncounterSlot2[] Slots;
protected override IReadOnlyList Raw => Slots;
public static EncounterArea2[] GetAreas(BinLinkerAccessor input, GameVersion game)
{
var result = new EncounterArea2[input.Length];
for (int i = 0; i < result.Length; i++)
result[i] = new EncounterArea2(input[i], game);
return result;
}
private EncounterArea2(ReadOnlySpan data, GameVersion game) : base(game)
{
Location = data[0];
Time = (EncounterTime)data[1];
var type = (Type = (SlotType)data[2]) & (SlotType)0xF;
Rate = data[3];
var next = data[4..];
if (type is > SlotType.Surf and not SlotType.BugContest) // Not Grass/Surf
{
const int size = 5;
int count = next.Length / size;
Rates = next[..count].ToArray();
Slots = ReadSlots(next[count..], count);
}
else
{
const int size = 4;
int count = next.Length / size;
Rates = type switch
{
SlotType.BugContest => BCC_SlotRates,
SlotType.Grass => RatesGrass,
_ => RatesSurf,
};
Slots = ReadSlots(next, count);
}
}
private EncounterSlot2[] ReadSlots(ReadOnlySpan data, int count)
{
const int size = 4;
var slots = new EncounterSlot2[count];
for (int i = 0; i < slots.Length; i++)
{
var entry = data.Slice(i * size, size);
byte max = entry[3];
byte min = entry[2];
byte slotNum = entry[1];
byte species = entry[0];
slots[i] = new EncounterSlot2(this, species, min, max, slotNum);
}
return slots;
}
public override IEnumerable GetMatchingSlots(PKM pkm, IReadOnlyList chain)
{
if (pkm is not ICaughtData2 {CaughtData: not 0} pk2)
return GetSlotsFuzzy(chain);
if (pk2.Met_Location != Location)
return Array.Empty();
return GetSlotsSpecificLevelTime(chain, pk2.Met_TimeOfDay, pk2.Met_Level);
}
private IEnumerable GetSlotsSpecificLevelTime(IReadOnlyList chain, int time, int lvl)
{
foreach (var slot in Slots)
{
foreach (var evo in chain)
{
if (slot.Species != evo.Species)
continue;
if (slot.Form != evo.Form)
{
if (slot.Species != (int)Species.Unown || evo.Form >= 26) // Don't yield !? forms
break;
}
if (!slot.IsLevelWithinRange(lvl))
break;
if (!Time.Contains(time))
break;
yield return slot;
break;
}
}
}
private IEnumerable GetSlotsFuzzy(IReadOnlyList chain)
{
foreach (var slot in Slots)
{
foreach (var evo in chain)
{
if (slot.Species != evo.Species)
continue;
if (slot.Form != evo.Form)
{
if (slot.Species != (int) Species.Unown || evo.Form >= 26) // Don't yield !? forms
break;
}
if (slot.LevelMin > evo.Level)
break;
yield return slot;
break;
}
}
}
}
}