pkNX/pkNX.WinForms/MainEditor/EditorPLA.cs
JelleInfinity 523a39e594
Fix evo editor (#249)
* Rename to Species

* Use Graphics to resize image

* Fix evo editor

Not the best/fastest method, but at least its functional now
2022-09-22 19:10:24 -07:00

459 lines
18 KiB
C#

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Windows.Forms;
using pkNX.Containers;
using pkNX.Game;
using pkNX.Randomization;
using pkNX.Structures;
using pkNX.Structures.FlatBuffers;
using pkNX.WinForms.Subforms;
using static pkNX.Structures.Species;
namespace pkNX.WinForms.Controls;
internal class EditorPLA : EditorBase
{
private GameData8a Data => ((GameManagerPLA)ROM).Data;
protected internal EditorPLA(GameManagerPLA rom) : base(rom) => CheckOodleDllPresence();
private static void CheckOodleDllPresence()
{
const string file = $"{Oodle.OodleLibraryPath}.dll";
var dir = Application.StartupPath;
var path = Path.Combine(dir, file);
if (!File.Exists(path))
WinFormsUtil.Alert($"{file} not found in the executable folder", "Some decompression functions may cause errors.");
}
public void EditCommon()
{
var text = ROM.GetFilteredFolder(GameFile.GameText, z => Path.GetExtension(z) == ".dat");
var config = new TextConfig(ROM.Game);
var tc = new TextContainer(text, config);
using var form = new TextEditor(tc, TextEditor.TextEditorMode.Common);
form.ShowDialog();
if (!form.Modified)
text.CancelEdits();
}
public void EditScript()
{
var text = ROM.GetFilteredFolder(GameFile.StoryText, z => Path.GetExtension(z) == ".dat");
var config = new TextConfig(ROM.Game);
var tc = new TextContainer(text, config);
using var form = new TextEditor(tc, TextEditor.TextEditorMode.Script);
form.ShowDialog();
if (!form.Modified)
text.CancelEdits();
}
public void EditTrainers()
{
var folder = ROM.GetFilteredFolder(GameFile.TrainerData);
var cache = new DataCache<TrData8a>(folder)
{
Create = FlatBufferConverter.DeserializeFrom<TrData8a>,
Write = FlatBufferConverter.SerializeFrom,
};
var names = folder.GetPaths().Select(Path.GetFileNameWithoutExtension).ToArray();
using var form = new GenericEditor<TrData8a>(cache, names, "Trainers", Randomize, canSave: true);
form.ShowDialog();
void Randomize()
{
var settings = EditUtil.Settings.Species;
var rand = new SpeciesRandomizer(ROM.Info, Data.PersonalData);
rand.Initialize(settings, GetSpeciesBanlist());
for (int i = 0; i < cache.Length; i++)
{
foreach (var t in cache[i].Team)
{
t.Species = rand.GetRandomSpecies(t.Species);
t.Form = GetRandomForm(t.Species);
t.Gender = (int)FixedGender.Random;
t.Nature = NatureType8a.Random;
t.Move_01.Move = t.Move_02.Move = t.Move_03.Move = t.Move_04.Move = 0;
t.Move_01.Mastered = t.Move_02.Mastered = t.Move_03.Mastered = t.Move_04.Mastered = true;
t.Shiny = Randomization.Util.Random.Next(0, 100 + 1) < 3;
t.IsOybn = Randomization.Util.Random.Next(0, 100 + 1) < 3;
}
}
}
if (!form.Modified)
cache.CancelEdits();
else
cache.Save();
}
public void EditMoveShop()
{
var names = ROM.GetStrings(TextName.MoveNames);
var data = ROM.GetFile(GameFile.MoveShop);
var obj = FlatBufferConverter.DeserializeFrom<MoveShopTable8a>(data[0]);
var result = PopFlat(obj.Table, "Move Shop Editor", z => names[z.Move]);
if (!result)
return;
data[0] = FlatBufferConverter.SerializeFrom(obj);
}
public void PopFlat<T1, T2>(GameFile file, string title, Func<T2, string> getName, Action? rand = null, bool canSave = true) where T1 : class, IFlatBufferArchive<T2> where T2 : class
{
var obj = ROM.GetFile(file);
var data = obj[0];
var root = FlatBufferConverter.DeserializeFrom<T1>(data);
var arr = root.Table;
if (!PopFlat(arr, title, getName, rand, canSave))
return;
obj[0] = FlatBufferConverter.SerializeFrom(root);
}
private static bool PopFlat<T2>(T2[] arr, string title, Func<T2, string> getName, Action? rand = null, bool canSave = true) where T2 : class
{
var names = arr.Select(getName).ToArray();
var cache = new DataCache<T2>(arr);
using var form = new GenericEditor<T2>(cache, names, title, randomize: rand, canSave: canSave);
form.ShowDialog();
return form.Modified;
}
public void EditThrowable_Param()
{
var itemNames = ROM.GetStrings(TextName.ItemNames);
PopFlat<ThrowableParamTable8a, ThrowableParam8a>(GameFile.ThrowableParam, "Throwable Param Editor", z => itemNames[z.ItemID]);
}
public void EditThrow_Param()
{
PopFlat<ThrowParamTable8a, ThrowParam8a>(GameFile.ThrowParam, "Throw Param Editor", z => z.Hash.ToString("X16"));
}
public void EditThrow_ResourceSet_Dictionary()
{
PopFlat<ThrowableResourceSetDictionary8a, ThrowableResourceSetEntry8a>(GameFile.ThrowableResourceSet, "Throwable ResourceSet Dictionary Editor", z => z.Hash_00.ToString("X16"));
}
public void EditThrow_Resource_Dictionary()
{
PopFlat<ThrowableResourceDictionary8a, ThrowableResourceEntry8a>(GameFile.ThrowableResource, "Throwable Resource Dictionary Editor", z => z.Hash_00.ToString("X16"));
}
public void EditThrow_PermissionSet_Param()
{
PopFlat<ThrowPermissionSetDictionary8a, ThrowPermissionSetEntry8a>(GameFile.ThrowPermissionSet, "Throw Permission Editor", z => z.Hash_00.ToString("X16"));
}
public void EditHa_Shop_Data()
{
var names = ROM.GetStrings(TextName.ItemNames);
var data = ROM.GetFile(GameFile.HaShop);
var obj = FlatBufferConverter.DeserializeFrom<HaShopTable8a>(data[0]);
var result = PopFlat(obj.Table, "ha_shop_data Editor", z => names[z.ItemID], Randomize);
if (!result)
return;
void Randomize()
{
foreach (var t in obj.Table)
{
if (Legal.Pouch_Recipe_LA.Contains((ushort)t.ItemID)) // preserve recipes
continue;
t.ItemID = Legal.Pouch_Items_LA[Randomization.Util.Random.Next(Legal.Pouch_Items_LA.Length)];
}
}
data[0] = FlatBufferConverter.SerializeFrom(obj);
}
public void EditApp_Config_List()
{
PopFlat<AppConfigList8a, AppconfigEntry8a>(GameFile.AppConfigList, "App Config List", z => z.OriginalPath);
}
public void EditArea_Weather()
{
var gfp = (GFPack)ROM.GetFile(GameFile.Resident);
var data = gfp[2065];
var obj = FlatBufferConverter.DeserializeFrom<AreaWeatherTable8a>(data);
var result = PopFlat(obj.Table, "Area Weather Editor", z => z.Hash.ToString("X16"));
if (!result)
return;
gfp[2065] = FlatBufferConverter.SerializeFrom(obj);
}
public void EditStatic()
{
var names = ROM.GetStrings(TextName.SpeciesNames);
var obj = ROM.GetFile(GameFile.EncounterStatic);
var data = obj[0];
var root = FlatBufferConverter.DeserializeFrom<EventEncount8aArchive>(data);
var entries = root.Table;
var result = PopFlat(entries, "Static Encounter Editor", z => $"{z.EncounterName} ({GetDetail(z, names)})", () => Randomize(entries));
if (result)
obj[0] = FlatBufferConverter.SerializeFrom(root);
static string GetDetail(EventEncount8a z, string[] names)
{
if (z.Table is not { Length: not 0 } x)
return "No Entries";
var s = x[0];
return $"{names[s.Species]}{(s.Form == 0 ? "" : $"-{s.Form}")} @ Lv. {s.Level}";
}
void Randomize(IEnumerable<EventEncount8a> arr)
{
var settings = EditUtil.Settings.Species;
var rand = new SpeciesRandomizer(ROM.Info, Data.PersonalData);
settings.Legends = false;
rand.Initialize(settings, GetSpeciesBanlist());
foreach (var entry in arr)
{
if (entry.Table is not { Length: > 0 } x)
continue;
foreach (var t in x)
{
bool isBoss = t.Species is (int)Kleavor or (int)Lilligant or (int)Arcanine or (int)Electrode or (int)Avalugg or (int)Arceus;
if (isBoss) // don't randomize boss battles
continue;
if (Legal.Legendary_8a.Contains(t.Species)) // don't randomize legendaries
continue;
t.Species = rand.GetRandomSpecies(t.Species);
t.Form = (byte)GetRandomForm(t.Species);
t.Nature = (int)Nature.Random;
t.Gender = (int)FixedGender.Random;
t.ShinyLock = ShinyType8a.Random;
t.Move1 = t.Move2 = t.Move3 = t.Move4 = 0;
t.Mastered1 = t.Mastered2 = t.Mastered3 = t.Mastered4 = true;
t.IV_HP = t.IV_ATK = t.IV_DEF = t.IV_SPA = t.IV_SPD = t.IV_SPE = 31;
t.GV_HP = t.GV_ATK = t.GV_DEF = t.GV_SPA = t.GV_SPD = t.GV_SPE = 10;
t.Height = t.Weight = -1;
}
}
}
}
public void EditGift()
{
var names = ROM.GetStrings(TextName.SpeciesNames);
var obj = ROM.GetFile(GameFile.EncounterGift);
var data = obj[0];
var root = FlatBufferConverter.DeserializeFrom<PokeAdd8aArchive>(data);
var entries = root.Table;
var result = PopFlat(entries, "Gift Encounter Editor", z => $"{names[z.Species]}{(z.Form == 0 ? "" : $"-{z.Form}")} @ Lv. {z.Level}", () => Randomize(entries));
if (result)
obj[0] = FlatBufferConverter.SerializeFrom(root);
void Randomize(IEnumerable<PokeAdd8a> arr)
{
var settings = EditUtil.Settings.Species;
var rand = new SpeciesRandomizer(ROM.Info, Data.PersonalData);
settings.Legends = false;
rand.Initialize(settings, GetSpeciesBanlist());
foreach (var t in arr)
{
t.Species = rand.GetRandomSpecies(t.Species);
t.Form = (byte)GetRandomForm(t.Species);
t.Nature = NatureType8a.Random;
t.Gender = (int)FixedGender.Random;
t.ShinyLock = ShinyType8a.Random;
t.Ball = Randomization.Util.Random.Next(27, 37); // [Strange, Origin]
t.Move1 = t.Move2 = t.Move3 = t.Move4 = 0;
t.Height = t.Weight = -1;
}
}
}
public int[] GetSpeciesBanlist()
{
var pt = Data.PersonalData;
var hasForm = new HashSet<int>();
var banned = new HashSet<int>();
foreach (var pi in pt.Table.Cast<IPersonalMisc_1>())
{
if (pi.IsPresentInGame)
{
banned.Remove(pi.Species);
hasForm.Add(pi.Species);
}
else if (!hasForm.Contains(pi.Species))
{
banned.Add(pi.Species);
}
}
return banned.ToArray();
}
public int GetRandomForm(int spec)
{
var pt = Data.PersonalData;
var formRand = pt.Table.Cast<IPersonalMisc_1>()
.Where(z => z.IsPresentInGame && !(Legal.BattleExclusiveForms.Contains(z.Species) || Legal.BattleFusions.Contains(z.Species)))
.GroupBy(z => z.Species)
.ToDictionary(z => z.Key, z => z.ToList());
if (!formRand.TryGetValue((ushort)spec, out var entries))
return 0;
var count = entries.Count;
return (Species)spec switch
{
Growlithe or Arcanine or Voltorb or Electrode or Typhlosion or Qwilfish or Samurott or Lilligant or Zorua or Zoroark or Braviary or Sliggoo or Goodra or Avalugg or Decidueye => 1,
Basculin => 2,
Kleavor => 0,
_ => Randomization.Util.Random.Next(0, count),
};
}
public void EditPersonal_Raw()
{
var names = ROM.GetStrings(TextName.SpeciesNames);
PopFlat<PersonalTableLAfb, PersonalInfoLAfb>(GameFile.PersonalStats, "Personal Info Editor (Raw)", z => $"{names[z.Species]}{(z.Form == 0 ? "" : $"-{z.Form}")}");
}
public void EditLearnset_Raw()
{
var names = ROM.GetStrings(TextName.SpeciesNames);
PopFlat<Learnset8a, Learnset8aMeta>(GameFile.Learnsets, "Learnset Editor (Raw)", z => $"{names[z.Species]}{(z.Form == 0 ? "" : $"-{z.Form}")}");
}
public void EditMiscSpeciesInfo()
{
var names = ROM.GetStrings(TextName.SpeciesNames);
PopFlat<PokeMiscTable8a, PokeMisc8a>(GameFile.PokeMisc, "Misc Species Info Editor", z => $"{names[z.Species]}{(z.Form == 0 ? "" : $"-{z.Form}")} ~ {z.Value}");
}
public void EditMap_viewer()
{
var resident = (GFPack)ROM.GetFile(GameFile.Resident);
using var form = new MapViewer8a((GameManagerPLA)ROM, resident);
form.ShowDialog();
}
public void EditAreas()
{
using var form = new AreaEditor8a((GameManagerPLA)ROM);
form.ShowDialog();
}
public void EditMoves()
{
var obj = ROM[GameFile.MoveStats]; // folder
var cache = new DataCache<Waza8a>(obj)
{
Create = FlatBufferConverter.DeserializeFrom<Waza8a>,
Write = FlatBufferConverter.SerializeFrom,
};
using var form = new GenericEditor<Waza8a>(cache, ROM.GetStrings(TextName.MoveNames), "Move Editor");
form.ShowDialog();
if (!form.Modified)
{
cache.CancelEdits();
return;
}
cache.Save();
Data.MoveData.ClearAll(); // force reload if used again
}
public void EditItems()
{
var obj = ROM[GameFile.ItemStats]; // mini
var data = obj[0];
var items = Item8a.GetArray(data);
var cache = new DataCache<Item8a>(items);
using var form = new GenericEditor<Item8a>(cache, ROM.GetStrings(TextName.ItemNames), "Item Editor");
form.ShowDialog();
if (!form.Modified)
{
cache.CancelEdits();
return;
}
obj[0] = Item8a.SetArray(items, data);
}
public void EditPokemon()
{
var editor = new PokeEditor8a
{
Evolve = Data.EvolutionData,
Learn = Data.LevelUpData,
Personal = Data.PersonalData
};
using var form = new PokeDataUI8a(editor, ROM, Data);
form.ShowDialog();
if (!form.Modified)
editor.CancelEdits();
else
editor.Save();
}
public void PopFlatConfig(GameFile file, string title)
{
var obj = ROM.GetFile(file); // flatbuffer
var data = obj[0];
var root = FlatBufferConverter.DeserializeFrom<ConfigureTable8a>(data);
var cache = new DataCache<Configure8aEntry>(root.Table);
var names = root.Table.Select(z => z.Name).ToArray();
using var form = new GenericEditor<Configure8aEntry>(cache, names, title);
form.ShowDialog();
if (!form.Modified)
return;
obj[0] = FlatBufferConverter.SerializeFrom(root);
}
public void EditShinyRate() => PopFlatConfig(GameFile.ShinyRolls, "Shiny Rate Editor");
public void EditWormholeRate() => PopFlatConfig(GameFile.WormholeConfig, "Wormhole Config Editor");
public void EditCapture_Config() => PopFlatConfig(GameFile.CaptureConfig, "CaptureConfig Editor");
public void EditBattle_Logic_Config() => PopFlatConfig(GameFile.BattleLogicConfig, "Battle Logic Config Editor");
public void EditEvent_Farm_Config() => PopFlatConfig(GameFile.EventFarmConfig, "Event Farm Config Editor");
public void EditPlayer_Config() => PopFlatConfig(GameFile.PlayerConfig, "Player Config Editor");
public void EditField_Landmark_Config() => PopFlatConfig(GameFile.FieldLandmarkConfig, "Field Landmark Config Editor");
public void EditBattle_View_Config() => PopFlatConfig(GameFile.BattleViewConfig, "Battle View Config Editor");
public void EditAICommon_Config() => PopFlatConfig(GameFile.AICommonConfig, "AI Common Config Editor");
public void EdiField_Spawner_Config() => PopFlatConfig(GameFile.FieldSpawnerConfig, "Field Spawner Config Editor");
public void EditOutbreak_Config() => PopFlatConfig(GameFile.OutbreakConfig, "Outbreak Config Editor");
public void EditEvolution_Config() => PopFlatConfig(GameFile.EvolutionConfig, "Evolution Config Editor");
public void EditBall_Throw_Config() => PopFlatConfig(GameFile.BallThrowConfig, "Ball Throw Config Editor");
public void EditSize_Scale_Config() => PopFlatConfig(GameFile.SizeScaleConfig, "Size Scale Config Editor");
public void EditEvolutions()
{
var names = ROM.GetStrings(TextName.SpeciesNames);
PopFlat<EvolutionTable8, EvolutionSet8a>(GameFile.Evolutions, "Evolution Editor",
z => $"{names[z.Species]}{(z.Form != 0 ? $"-{z.Form}" : "")}");
}
public void EditOutbreakDetail()
=> PopFlat<MassOutbreakTable8a, MassOutbreak8a>(GameFile.Outbreak, "Outbreak Proc Editor", z => z.WorkValueName);
public void EditNewOutbreak_Group()
=> PopFlat<NewHugeOutbreakGroupArchive8a, NewHugeOutbreakGroup8a>(GameFile.NewHugeGroup, "New Outbreak Group Editor", z => z.Group.ToString("X16"));
public void EditNewOutbreak_GroupLottery()
=> PopFlat<NewHugeOutbreakGroupLotteryArchive8a, NewHugeOutbreakGroupLottery8a>(GameFile.NewHugeGroupLottery, "New Outbreak Group Lottery Editor", z => z.LotteryGroup.ToString("X16"));
public void EditNewOutbreak_Lottery()
=> PopFlat<NewHugeOutbreakLotteryArchive8a, NewHugeOutbreakLottery8a>(GameFile.NewHugeLottery, "New Outbreak Lottery Editor", z => z.LotteryGroupString);
public void EditNewOutbreak_TimeLimit()
=> PopFlat<NewHugeOutbreakTimeLimitArchive8a, NewHugeOutbreakTimeLimit8a>(GameFile.NewHugeTimeLimit, "New Outbreak Time Limit Editor", z => z.Duration.ToString());
public void EditSymbolBehave()
{
var names = ROM.GetStrings(TextName.SpeciesNames);
PopFlat<PokeAIArchive8a, PokeAI8a>(GameFile.SymbolBehave, "Symbol Behavior Editor", z => $"{names[z.Species]}{(z.Form != 0 ? $"-{z.Form}" : "")}");
}
public void EditMasterDump()
{
using var md = new DumperPLA((GameManagerPLA)ROM);
md.ShowDialog();
}
}