PKHeX/PKHeX.Core/Saves/Substructures/PokeDex/Zukan.cs

183 lines
7 KiB
C#
Raw Normal View History

using System;
using System.Linq;
namespace PKHeX.Core
{
public abstract class Zukan
{
protected SaveFile SAV { get; set; }
protected int PokeDex { get; set; }
protected int PokeDexLanguageFlags { get; set; }
protected abstract int OFS_SEEN { get; }
protected abstract int OFS_CAUGHT { get; }
protected abstract int BitSeenSize { get; }
protected abstract int DexLangFlagByteCount { get; }
protected abstract int DexLangIDCount { get; }
protected abstract int GetDexLangFlag(int lang);
2018-11-14 03:14:23 +00:00
public Func<int, int, int, int> DexFormIndexFetcher { get; protected set; }
protected abstract bool GetSaneFormsToIterate(int species, out int formStart, out int formEnd, int formIn);
2018-11-14 03:14:23 +00:00
protected virtual void SetSpindaDexData(PKM pkm, bool alreadySeen) { }
protected abstract void SetAllDexFlagsLanguage(int bit, int lang, bool value = true);
protected abstract void SetAllDexSeenFlags(int baseBit, int altform, int gender, bool isShiny, bool value = true);
2018-11-14 03:14:23 +00:00
protected bool GetFlag(int ofs, int bitIndex) => SAV.GetFlag(PokeDex + ofs + (bitIndex >> 3), bitIndex);
protected void SetFlag(int ofs, int bitIndex, bool value = true) => SAV.SetFlag(PokeDex + ofs + (bitIndex >> 3), bitIndex, value);
2018-11-14 03:14:23 +00:00
public virtual bool GetCaught(int species) => GetFlag(OFS_CAUGHT, species - 1);
public virtual void SetCaught(int species, bool value = true) => SetFlag(OFS_CAUGHT, species - 1, value);
public int SeenCount => Enumerable.Range(1, SAV.MaxSpeciesID).Count(GetSeen);
public int CaughtCount => Enumerable.Range(1, SAV.MaxSpeciesID).Count(GetCaught);
public decimal PercentSeen => (decimal)SeenCount / SAV.MaxSpeciesID;
public decimal PercentCaught => (decimal)CaughtCount / SAV.MaxSpeciesID;
2018-11-14 03:14:23 +00:00
public virtual bool GetSeen(int species)
{
// check all 4 seen flags (gender/shiny)
for (int i = 0; i < 4; i++)
{
2018-11-14 03:14:23 +00:00
if (GetSeen(species, i))
return true;
}
return false;
}
2018-11-14 03:14:23 +00:00
public bool GetSeen(int species, int i) => GetFlag(OFS_SEEN + (i * BitSeenSize), species - 1);
public void SetSeen(int species, int i, bool value) => SetFlag(OFS_SEEN + (i * BitSeenSize), species - 1, value);
public bool GetDisplayed(int bit, int i) => GetFlag(OFS_SEEN + ((i + 4) * BitSeenSize), bit);
public void SetDisplayed(int bit, int i, bool value) => SetFlag(OFS_SEEN + ((i + 4) * BitSeenSize), bit, value);
public bool GetLanguageFlag(int bit, int lang) => GetFlag(PokeDexLanguageFlags, (bit * DexLangIDCount) + lang);
public void SetLanguageFlag(int bit, int lang, bool value) => SetFlag(PokeDexLanguageFlags, (bit * DexLangIDCount) + lang, value);
public virtual void SetSeen(int species, bool value = true)
{
if (!value)
{
ClearSeen(species);
return;
}
// check all 4 seen flags (gender/shiny)
for (int i = 0; i < 4; i++)
{
if (GetFlag(OFS_SEEN + (i * BitSeenSize), species - 1))
return;
}
var gender = SAV.Personal[species].RandomGender() & 1;
SetAllDexSeenFlags(species - 1, 0, gender, false);
}
private void ClearSeen(int species)
{
SetCaught(species, false);
for (int i = 0; i < 4; i++)
SetFlag(OFS_SEEN + (i * BitSeenSize), species - 1, false);
}
public virtual void SetDex(PKM pkm)
{
if (PokeDex < 0 || SAV.Version == GameVersion.Invalid) // sanity
return;
if (pkm.Species == 0 || pkm.Species > SAV.MaxSpeciesID) // out of range
return;
if (pkm.IsEgg) // do not add
return;
2018-11-14 03:14:23 +00:00
int species = pkm.Species;
if (species == 327) // Spinda
SetSpindaDexData(pkm, GetSeen(species));
int bit = pkm.Species - 1;
2018-11-14 03:14:23 +00:00
int form = pkm.AltForm;
int gender = pkm.Gender & 1;
bool shiny = pkm.IsShiny;
int lang = pkm.Language;
SetDex(species, bit, form, gender, shiny, lang);
}
protected virtual void SetDex(int species, int bit, int form, int gender, bool shiny, int lang)
{
SetCaught(species); // Set the Owned Flag
SetAllDexSeenFlags(bit, form, gender, shiny); // genderless -> male
SetAllDexFlagsLanguage(bit, lang);
}
protected void SetDexFlags(int baseBit, int formBit, int gender, int shiny, bool value = true)
{
int shift = gender | (shiny << 1);
// Set the [Species/Gender/Shiny] Seen Flag
SetFlag(OFS_SEEN + (shift * BitSeenSize), baseBit, value);
// Set the Display flag if none are set
bool displayed = GetIsSpeciesFormAnyDisplayed(baseBit, formBit);
if (!displayed || !value)
SetFlag(OFS_SEEN + ((4 + shift) * BitSeenSize), formBit, value);
}
private bool GetIsSpeciesFormAnyDisplayed(int baseBit, int formBit)
{
// Check Displayed Status for base form
2018-11-14 03:14:23 +00:00
for (int i = 0; i < 4; i++)
{
2018-11-14 03:14:23 +00:00
if (GetDisplayed(baseBit, i))
return true;
}
if (baseBit == formBit)
return false;
// If form is not base form, check form too
2018-11-14 03:14:23 +00:00
for (int i = 0; i < 4; i++)
{
2018-11-14 03:14:23 +00:00
if (GetDisplayed(formBit, i))
return true;
}
return false;
}
// Bulk Manipulation
public void SetDexEntriesAll(bool value = true, int max = -1)
{
if (max <= 0)
max = SAV.MaxSpeciesID;
for (int i = 1; i <= max; i++)
SetDexEntriesSingle(i, value);
}
public void SetDexEntriesSingle(int species, bool value = true)
{
SetCaught(species, value);
SetSeen(species, value);
var entry = SAV.Personal[species];
int baseBit = species - 1;
int fc = entry.FormeCount;
for (int f = 0; f < fc; f++)
{
if (!entry.OnlyFemale)
{
SetAllDexSeenFlags(baseBit, f, 0, false, value);
SetAllDexSeenFlags(baseBit, f, 0, true, value);
}
if (!entry.OnlyMale && !entry.Genderless)
{
SetAllDexSeenFlags(baseBit, f, 1, false, value);
SetAllDexSeenFlags(baseBit, f, 1, true, value);
}
}
SetAllDexFlagsLanguage(baseBit, value);
}
protected void SetAllDexFlagsLanguage(int bit, bool value = true)
{
for (int i = 1; i <= DexLangIDCount + 1; i++)
SetAllDexFlagsLanguage(bit, i, value);
}
}
}