using System; using System.Collections.Generic; namespace PKHeX.Core { // I wish I could replace this with raw pointers via Span :) public sealed class Zukan8 : ZukanBase { private readonly SCBlock Galar; private readonly SCBlock Rigel1; private readonly SCBlock Rigel2; public readonly IReadOnlyDictionary DexLookup; public Zukan8(SAV8SWSH sav, SCBlock galar, SCBlock rigel1, SCBlock rigel2) : base(sav, 0) { Galar = galar; Rigel1 = rigel1; Rigel2 = rigel2; var revision = GetRevision(); DexLookup = GetDexLookup(PersonalTable.SWSH, revision); } public int GetRevision() { if (Rigel1.Data.Length == 0) return 0; // No DLC1 data allocated if (Rigel2.Data.Length == 0) return 1; // No DLC2 data allocated return 2; } private byte[] GetDexBlock(Zukan8Type infoDexType) { return infoDexType switch { Zukan8Type.Galar => Galar.Data, Zukan8Type.Armor => Rigel1.Data, Zukan8Type.Crown => Rigel2.Data, _ => throw new ArgumentOutOfRangeException(nameof(infoDexType), infoDexType, null) }; } private static bool GetFlag(byte[] data, int offset, int bitIndex) => FlagUtil.GetFlag(data, offset + (bitIndex >> 3), bitIndex); private static void SetFlag(byte[] data, int offset, int bitIndex, bool value = true) => FlagUtil.SetFlag(data, offset + (bitIndex >> 3), bitIndex, value); private static Dictionary GetDexLookup(PersonalTable pt, int dexRevision) { var lookup = new Dictionary(); for (int i = 1; i <= pt.MaxSpeciesID; i++) { var p = (PersonalInfoSWSH) pt[i]; var index = p.PokeDexIndex; if (index != 0) { lookup.Add(i, new Zukan8Index(Zukan8Type.Galar, index)); continue; } if (dexRevision == 0) continue; var armor = p.ArmorDexIndex; if (armor != 0) { lookup.Add(i, new Zukan8Index(Zukan8Type.Armor, armor)); continue; } if (dexRevision == 1) continue; var crown = p.CrownDexIndex; if (crown != 0) { lookup.Add(i, new Zukan8Index(Zukan8Type.Crown, crown)); // continue; } } return lookup; } public static List GetRawIndexes(PersonalTable pt, int dexRevision) { var result = new List(); for (int i = 1; i <= pt.MaxSpeciesID; i++) { var p = (PersonalInfoSWSH)pt[i]; var index = p.PokeDexIndex; if (index != 0) result.Add(new Zukan8EntryInfo(i, new Zukan8Index(Zukan8Type.Galar, index))); } if (dexRevision == 0) return result; for (int i = 1; i <= pt.MaxSpeciesID; i++) { var p = (PersonalInfoSWSH)pt[i]; var index = p.ArmorDexIndex; if (index != 0) result.Add(new Zukan8EntryInfo(i, new Zukan8Index(Zukan8Type.Armor, index))); } if (dexRevision == 1) return result; for (int i = 1; i <= pt.MaxSpeciesID; i++) { var p = (PersonalInfoSWSH)pt[i]; var index = p.CrownDexIndex; if (index != 0) result.Add(new Zukan8EntryInfo(i, new Zukan8Index(Zukan8Type.Crown, index))); } return result; } private static int GetDexLangFlag(int lang) { if (lang > 10 || lang == 6 || lang <= 0) return -1; // invalid language if (lang >= 7) // skip over langID 6 (unused) lang--; lang--; // skip over langID 0 (unused) => [0-8] return lang; } public IList GetEntryNames(IReadOnlyList speciesNames) { var dex = new List(); foreach (var d in DexLookup) { var spec = d.Key; var entry = d.Value; var name = entry.GetEntryName(speciesNames, spec); dex.Add(name); } dex.Sort(); return dex; } #region Structure internal const int EntrySize = 0x30; // First 0x20 bytes are for seen flags, allocated as 4 QWORD values. private const int SeenRegionCount = 4; private const int SeenRegionSize = sizeof(ulong); // not_shiny_gender_0, // not_shiny_gender_1, // shiny_gender_0, // shiny_gender_1 // Each QWORD stores the following bits: // - FormsSeen[63], default form is index 0. // - Gigantamax:1 -- for Urshifu, they store a bit prior for the second Gigantamax form... // Next 4 bytes are for obtained info (u32) private const int OFS_CAUGHT = 0x20; // Owned:1 (posessed by player) // OwnedGigantamax:1 (Gigantamaxed by player in battle) // LanguagesObtained:2-14 (flags) // DisplayFormID:15-27 (value) // DisplayGigantamaxInstead:28 (flag) // DisplayGender:29/30 (m/f) // DisplayShiny:31 (flag) // Next 4 bytes are Battled Count (u32) private const int OFS_BATTLED = 0x24; // Next 4 bytes are unused/reserved for future updates. private const int OFS_UNK1 = 0x28; // Seen Gigantamax-1 Form:1 (Urshifu) // Next 4 bytes are Unused(?) private const int OFS_UNK2 = 0x2C; public bool GetEntry(int species, out Zukan8Index entry) => DexLookup.TryGetValue(species, out entry); public override bool GetSeen(int species) { if (!GetEntry(species, out var entry)) return false; return GetSeen(entry); } public bool GetSeen(Zukan8Index entry) { byte[] data = GetDexBlock(entry.DexType); int offset = entry.Offset; for (int i = 0; i < SeenRegionCount; i++) { var ofs = offset + (SeenRegionSize * i); if (BitConverter.ToUInt64(data, ofs) != 0) return true; } return false; } public bool GetSeenRegion(int species, int form, int region) { if (!GetEntry(species, out var entry)) return false; return GetSeenRegion(entry, form, region); } public bool GetSeenRegion(Zukan8Index entry, int form, int region) { if ((uint)region >= SeenRegionCount) throw new ArgumentException(nameof(region)); if ((uint)form > 63) return false; var dex = entry.DexType; var offset = entry.Offset; var data = GetDexBlock(dex); var ofs = SeenRegionSize * region; return GetFlag(data, offset + ofs, form); } public void SetSeenRegion(int species, int form, int region, bool value = true) { if (!GetEntry(species, out var entry)) return; SetSeenRegion(entry, form, region, value); } public void SetSeenRegion(Zukan8Index entry, int form, int region, bool value = true) { if ((uint) region >= SeenRegionCount) throw new ArgumentException(nameof(region)); if ((uint) form > 63) return; var data = GetDexBlock(entry.DexType); int index = entry.Offset; var ofs = SeenRegionSize * region; SetFlag(data, index + ofs, form, value); } public override bool GetCaught(int species) => GetCaughtFlagID(species, 0); public void SetCaught(int species, bool value = true) => SetCaughtFlagID(species, 0, value); public bool GetCaughtGigantamaxed(int species) => GetCaughtFlagID(species, 1); public void SetCaughtGigantamax(int species, bool value = true) => SetCaughtFlagID(species, 1, value); public bool GetIsLanguageIndexObtained(int species, int langIndex) => GetCaughtFlagID(species, 2 + langIndex); public void SetIsLanguageIndexObtained(int species, int langIndex, bool value = true) => SetCaughtFlagID(species, 2 + langIndex, value); public bool GetCaught(Zukan8Index entry) => GetCaughtFlagID(entry, 0); public void SetCaught(Zukan8Index entry, bool value = true) => SetCaughtFlagID(entry, 0, value); public bool GetCaughtGigantamaxed(Zukan8Index entry) => GetCaughtFlagID(entry, 1); public void SetCaughtGigantamax(Zukan8Index entry, bool value = true) => SetCaughtFlagID(entry, 1, value); public bool GetIsLanguageIndexObtained(Zukan8Index entry, int langIndex) => GetCaughtFlagID(entry, 2 + langIndex); public void SetIsLanguageIndexObtained(Zukan8Index entry, int langIndex, bool value = true) => SetCaughtFlagID(entry, 2 + langIndex, value); private bool GetCaughtFlagID(int species, int bit) { if (!GetEntry(species, out var entry)) return false; return GetCaughtFlagID(entry, bit); } private bool GetCaughtFlagID(Zukan8Index entry, int bit) { var data = GetDexBlock(entry.DexType); return GetFlag(data, entry.Offset + OFS_CAUGHT, bit); } public void SetCaughtFlagID(int species, int bit, bool value = true) { if (!GetEntry(species, out var entry)) return; SetCaughtFlagID(entry, bit, value); } public void SetCaughtFlagID(Zukan8Index entry, int bit, bool value = true) { var data = GetDexBlock(entry.DexType); SetFlag(data, entry.Offset + OFS_CAUGHT, bit, value); } public bool GetIsLanguageObtained(int species, int language) { int langIndex = GetDexLangFlag(language); if (langIndex < 0) return false; return GetIsLanguageIndexObtained(species, langIndex); } public void SetIsLanguageObtained(int species, int language, bool value = true) { int langIndex = GetDexLangFlag(language); if (langIndex < 0) return; SetIsLanguageIndexObtained(species, langIndex, value); } public uint GetFormDisplayed(int species) { if (!GetEntry(species, out var entry)) return 0; return GetFormDisplayed(entry); } public uint GetFormDisplayed(Zukan8Index entry) { var data = GetDexBlock(entry.DexType); var index = entry.Offset; var val = BitConverter.ToUInt32(data, index + OFS_CAUGHT); return (val >> 15) & 0x1FFF; // (0x1FFF is really overkill, GameFreak) } public void SetFormDisplayed(int species, uint value = 0) { if (!GetEntry(species, out var entry)) return; SetFormDisplayed(entry, value); } public void SetFormDisplayed(Zukan8Index entry, uint value = 0) { var data = GetDexBlock(entry.DexType); var index = entry.Offset; var val = BitConverter.ToUInt32(data, index + OFS_CAUGHT); uint nv = (val & ~(0x1FFFu << 15)) | ((value & 0x1FFF) << 15); BitConverter.GetBytes(nv).CopyTo(data, index + OFS_CAUGHT); } public uint GetGenderDisplayed(int species) { if (!GetEntry(species, out var entry)) return 0; return GetGenderDisplayed(entry); } public uint GetGenderDisplayed(Zukan8Index entry) { var data = GetDexBlock(entry.DexType); var index = entry.Offset; var val = BitConverter.ToUInt32(data, index + OFS_CAUGHT); return (val >> 29) & 3; } public void SetGenderDisplayed(int species, uint value = 0) { if (!GetEntry(species, out var entry)) return; SetGenderDisplayed(entry, value); } public void SetGenderDisplayed(Zukan8Index entry, uint value = 0) { var data = GetDexBlock(entry.DexType); var index = entry.Offset; var val = BitConverter.ToUInt32(data, index + OFS_CAUGHT); uint nv = (val & ~(3u << 29)) | ((value & 3) << 29); BitConverter.GetBytes(nv).CopyTo(data, index + OFS_CAUGHT); } public bool GetDisplayDynamaxInstead(int species) => GetCaughtFlagID(species, 28); public void SetDisplayDynamaxInstead(int species, bool value = true) => SetCaughtFlagID(species, 28, value); public bool GetDisplayShiny(int species) => GetCaughtFlagID(species, 31); public void SetDisplayShiny(int species, bool value = true) => SetCaughtFlagID(species, 31, value); public void SetCaughtFlags32(int species, uint value) => SetU32(species, value, OFS_CAUGHT); public uint GetBattledCount(int species) => GetU32(species, OFS_BATTLED); public void SetBattledCount(int species, uint value) => SetU32(species, value, OFS_BATTLED); public uint GetUnk1Count(int species) => GetU32(species, OFS_UNK1); public void SetUnk1Count(int species, uint value) => SetU32(species, value, OFS_UNK1); public uint GetUnk2Count(int species) => GetU32(species, OFS_UNK2); public void SetUnk2Count(int species, uint value) => SetU32(species, value, OFS_UNK2); public bool GetCaughtGigantamax1(int species) => GetFlag28(species, 0); public void SetCaughtGigantamax1(int species, bool value = true) => SetFlag28(species, 0, value); public bool GetCaughtGigantamax1(Zukan8Index entry) => GetFlag28(entry, 0); public void SetCaughtGigantamax1(Zukan8Index entry, bool value = true) => SetFlag28(entry, 0, value); private bool GetFlag28(int species, int bit) { if (!GetEntry(species, out var entry)) return false; return GetFlag28(entry, bit); } public void SetFlag28(int species, int bit, bool value = true) { if (!GetEntry(species, out var entry)) return; SetFlag28(entry, bit, value); } private bool GetFlag28(Zukan8Index entry, int bit) { var data = GetDexBlock(entry.DexType); return GetFlag(data, entry.Offset + OFS_UNK1, bit); } public void SetFlag28(Zukan8Index entry, int bit, bool value = true) { var data = GetDexBlock(entry.DexType); SetFlag(data, entry.Offset + OFS_UNK1, bit, value); } public bool GetDisplayDynamaxInstead(Zukan8Index entry) => GetCaughtFlagID(entry, 28); public void SetDisplayDynamaxInstead(Zukan8Index entry, bool value = true) => SetCaughtFlagID(entry, 28, value); public bool GetDisplayShiny(Zukan8Index entry) => GetCaughtFlagID(entry, 31); public void SetDisplayShiny(Zukan8Index entry, bool value = true) => SetCaughtFlagID(entry, 31, value); public void SetCaughtFlags32(Zukan8Index entry, uint value) => SetU32(entry, value, OFS_CAUGHT); public uint GetBattledCount(Zukan8Index entry) => GetU32(entry, OFS_BATTLED); public void SetBattledCount(Zukan8Index entry, uint value) => SetU32(entry, value, OFS_BATTLED); public uint GetUnk1Count(Zukan8Index entry) => GetU32(entry, OFS_UNK1); public void SetUnk1Count(Zukan8Index entry, uint value) => SetU32(entry, value, OFS_UNK1); public uint GetUnk2Count(Zukan8Index entry) => GetU32(entry, OFS_UNK2); public void SetUnk2Count(Zukan8Index entry, uint value) => SetU32(entry, value, OFS_UNK2); private uint GetU32(int species, int ofs) { if (!GetEntry(species, out var entry)) return 0; return GetU32(entry, ofs); } private uint GetU32(Zukan8Index entry, int ofs) { var dex = entry.DexType; var index = entry.Offset; var data = GetDexBlock(dex); return BitConverter.ToUInt32(data, index + ofs); } private void SetU32(int species, uint value, int ofs) { if (!GetEntry(species, out var entry)) return; SetU32(entry, value, ofs); } private void SetU32(Zukan8Index entry, uint value, int ofs) { var dex = entry.DexType; var index = entry.Offset; var data = GetDexBlock(dex); BitConverter.GetBytes(value).CopyTo(data, index + ofs); } #endregion #region Inherited public override void SetDex(PKM pkm) { int species = pkm.Species; if (!GetEntry(species, out _)) return; bool owned = GetCaught(species); var g = pkm.Gender == 1 ? 1 : 0; bool shiny = pkm.IsShiny; var s = shiny ? 2 : 0; int form = pkm.Form; if (species == (int)Species.Alcremie) { form *= 7; form += (int)((PK8)pkm).FormArgument; // alteration byte } else if (species == (int) Species.Eternatus && pkm.Form == 1) { form = 0; SetSeenRegion(species, 63, g | s); } SetSeenRegion(species, form, g | s); SetCaught(species); SetIsLanguageObtained(species, pkm.Language); if (!owned) { SetFormDisplayed(species, (byte)form); if (shiny) SetDisplayShiny(species); SetGenderDisplayed(species, (uint)g); } var count = GetBattledCount(species); if (count == 0) SetBattledCount(species, 1); } public override void SeenNone() { Array.Clear(Galar.Data, 0, Galar.Data.Length); Array.Clear(Rigel1.Data, 0, Rigel1.Data.Length); Array.Clear(Rigel2.Data, 0, Rigel2.Data.Length); } public override void CaughtNone() { foreach (var kvp in DexLookup) CaughtNone(kvp.Key); } private void CaughtNone(int species) { SetCaughtFlags32(species, 0); SetUnk1Count(species, 0); SetUnk2Count(species, 0); } public override void SeenAll(bool shinyToo = false) { SetAllSeen(true, shinyToo); } private void SeenAll(int species, int fc, bool shinyToo, bool value = true) { var pt = PersonalTable.SWSH; for (int form = 0; form < fc; form++) { var pi = pt.GetFormEntry(species, form); SeenAll(species, form, value, pi, shinyToo); } if (species == (int)Species.Slowbro) // Clear Mega Slowbro always SeenAll(species, 1, false, pt[species], shinyToo); if (!value) ClearGigantamaxFlags(species); } private void SeenAll(int species, int bitIndex, bool value, PersonalInfo pi, bool shinyToo) { if (pi.IsDualGender || !value) { SetSeenRegion(species, bitIndex, 0, value); SetSeenRegion(species, bitIndex, 1, value); if (!shinyToo && value) return; SetSeenRegion(species, bitIndex, 2, value); SetSeenRegion(species, bitIndex, 3, value); } else { var index = pi.OnlyFemale ? 1 : 0; SetSeenRegion(species, bitIndex, 0 + index); if (!shinyToo) return; SetSeenRegion(species, bitIndex, 2 + index); } } private void ClearGigantamaxFlags(int species) { SetSeenRegion(species, 63, 0, false); SetSeenRegion(species, 63, 1, false); SetSeenRegion(species, 63, 2, false); SetSeenRegion(species, 63, 3, false); } public override void CompleteDex(bool shinyToo = false) { foreach (var kvp in DexLookup) SetDexEntryAll(kvp.Key, shinyToo); } public override void CaughtAll(bool shinyToo = false) { SeenAll(shinyToo); foreach (var kvp in DexLookup) { var species = kvp.Key; SetAllCaught(species, true, shinyToo); } } private void SetAllCaught(int species, bool value = true, bool shinyToo = false) { SetCaught(species); for (int i = 0; i < 11; i++) SetIsLanguageObtained(species, i, value); if (value) { var pi = PersonalTable.SWSH[species]; if (shinyToo) SetDisplayShiny(species); SetGenderDisplayed(species, (uint)pi.RandomGender()); } else { SetDisplayShiny(species, false); SetDisplayDynamaxInstead(species, false); SetGenderDisplayed(species, 0); } } public override void SetAllSeen(bool value = true, bool shinyToo = false) { foreach (var kvp in DexLookup) { var species = kvp.Key; SetAllSeen(species, value, shinyToo); } } private void SetAllSeen(int species, bool value = true, bool shinyToo = false) { var pi = PersonalTable.SWSH[species]; var fc = pi.FormCount; if (species == (int) Species.Eternatus) fc = 1; // ignore gigantamax SeenAll(species, fc, shinyToo, value); if (species == (int) Species.Alcremie) { // Alcremie forms const int deco = 7; const int forms = 9; for (int i = 0; i < deco * forms; i++) // 0-62 SeenAll(species, i, value, pi, shinyToo); } if (SpeciesWithGigantamaxData.Contains(species)) { SeenAll(species, 63, value, pi, shinyToo); if (species == (int)Species.Urshifu) { SeenAll(species, 62, value, pi, shinyToo); SetCaughtGigantamax1(species); } SetCaughtGigantamax(species); } } public override void SetDexEntryAll(int species, bool shinyToo = false) { SetAllSeen(species, true, shinyToo); SetAllCaught(species, true); } public override void ClearDexEntryAll(int species) { if (!GetEntry(species, out var entry)) return; ClearDexEntryAll(entry); } private void ClearDexEntryAll(Zukan8Index entry) { var data = GetDexBlock(entry.DexType); Array.Clear(data, entry.Offset, EntrySize); } public void SetAllBattledCount(uint count = 500) { foreach (var kvp in DexLookup) { var species = kvp.Key; SetBattledCount(species, count); } } private static readonly HashSet SpeciesWithGigantamaxData = new() { (int)Species.Charizard, (int)Species.Butterfree, (int)Species.Pikachu, (int)Species.Meowth, (int)Species.Machamp, (int)Species.Gengar, (int)Species.Kingler, (int)Species.Lapras, (int)Species.Eevee, (int)Species.Snorlax, (int)Species.Garbodor, (int)Species.Corviknight, (int)Species.Orbeetle, (int)Species.Drednaw, (int)Species.Coalossal, (int)Species.Flapple, (int)Species.Appletun, (int)Species.Sandaconda, (int)Species.Toxtricity, (int)Species.Centiskorch, (int)Species.Hatterene, (int)Species.Grimmsnarl, (int)Species.Alcremie, (int)Species.Copperajah, (int)Species.Duraludon, (int)Species.Eternatus, // DLC 1 (int)Species.Rillaboom, (int)Species.Cinderace, (int)Species.Inteleon, (int)Species.Urshifu, }; #endregion } public readonly struct Zukan8Index { public readonly int Index; public readonly Zukan8Type DexType; public override string ToString() => $"{Index:000} - {DexType}"; public Zukan8Index(Zukan8Type dexType, int index) { DexType = dexType; Index = index; } private int GetSavedIndex() { var index = Index; if (index < 1) throw new IndexOutOfRangeException(); return index - 1; } public int Offset => GetSavedIndex() * Zukan8.EntrySize; private const int GalarCount = 400; private const int Rigel1Count = 211; private const int Rigel2Count = 2; // expects zero based indexes public static Zukan8Index GetFromRawIndex(int index) { if (index < 0) return new Zukan8Index(); if (index < GalarCount) return new Zukan8Index(Zukan8Type.Galar, index + 1); index -= GalarCount; if (index < Rigel1Count) return new Zukan8Index(Zukan8Type.Armor, index + 1); index -= Rigel1Count; if (index < Rigel2Count) return new Zukan8Index(Zukan8Type.Crown, index + 1); throw new ArgumentOutOfRangeException(nameof(index)); } public string DexPrefix => DexType switch { Zukan8Type.Galar => "O0", Zukan8Type.Armor => "R1", Zukan8Type.Crown => "R2", _ => throw new ArgumentOutOfRangeException() }; public int AbsoluteIndex => DexType switch { Zukan8Type.Galar => Index, Zukan8Type.Armor => Index + GalarCount, Zukan8Type.Crown => Index + GalarCount + Rigel1Count, _ => throw new ArgumentOutOfRangeException(nameof(DexType)), }; public string GetEntryName(IReadOnlyList speciesNames, int species) { return $"{DexPrefix}.{Index:000} - {speciesNames[species]}"; } public override bool Equals(object obj) => obj is Zukan8Index x && x.Equals(this); public bool Equals(Zukan8Index obj) => obj.Index == Index && obj.DexType == DexType; public override int GetHashCode() => (int)DexType ^ (Index << 3); public static bool operator ==(Zukan8Index left, Zukan8Index right) => left.Equals(right); public static bool operator !=(Zukan8Index left, Zukan8Index right) => !(left == right); } public sealed class Zukan8EntryInfo { public readonly int Species; public readonly Zukan8Index Entry; public Zukan8EntryInfo(int species, Zukan8Index entry) { Species = species; Entry = entry; } public string GetEntryName(IReadOnlyList speciesNames) => Entry.GetEntryName(speciesNames, Species); } public enum Zukan8Type : sbyte { None = 0, Galar, Armor, Crown, } }