mirror of
https://github.com/kwsch/PKHeX
synced 2024-12-18 16:33:24 +00:00
633d4e040b
It might be wise to just ball up all the results (as decrypted) to speed up form loading (by just having to read one file instead of >1000 at a time). Maybe just scoop up all the files into the database and filter dupes, but no way to remove stuff....
521 lines
20 KiB
C#
521 lines
20 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Windows.Forms;
|
|
using PKHeX.Properties;
|
|
|
|
namespace PKHeX
|
|
{
|
|
public partial class SAV_Database : Form
|
|
{
|
|
public SAV_Database(Main f1)
|
|
{
|
|
m_parent = f1;
|
|
InitializeComponent();
|
|
PKXBOXES = new[]
|
|
{
|
|
bpkx1, bpkx2, bpkx3, bpkx4, bpkx5, bpkx6,
|
|
bpkx7, bpkx8, bpkx9, bpkx10,bpkx11,bpkx12,
|
|
bpkx13,bpkx14,bpkx15,bpkx16,bpkx17,bpkx18,
|
|
bpkx19,bpkx20,bpkx21,bpkx22,bpkx23,bpkx24,
|
|
bpkx25,bpkx26,bpkx27,bpkx28,bpkx29,bpkx30,
|
|
|
|
bpkx31,bpkx32,bpkx33,bpkx34,bpkx35,bpkx36,
|
|
bpkx37,bpkx38,bpkx39,bpkx40,bpkx41,bpkx42,
|
|
bpkx43,bpkx44,bpkx45,bpkx46,bpkx47,bpkx48,
|
|
bpkx49,bpkx50,bpkx51,bpkx52,bpkx53,bpkx54,
|
|
bpkx55,bpkx56,bpkx57,bpkx58,bpkx59,bpkx60,
|
|
bpkx61,bpkx62,bpkx63,bpkx64,bpkx65,bpkx66,
|
|
};
|
|
|
|
// Enable Scrolling when hovered over
|
|
var me = this;
|
|
PAN_Box.MouseEnter += (sender, args) =>
|
|
{
|
|
if (ActiveForm == me)
|
|
SCR_Box.Focus();
|
|
};
|
|
foreach (var slot in PKXBOXES)
|
|
{
|
|
slot.MouseEnter += (sender, args) =>
|
|
{
|
|
if (ActiveForm == me)
|
|
SCR_Box.Focus();
|
|
};
|
|
// Enable Click
|
|
slot.MouseClick += (sender, args) =>
|
|
{
|
|
if (ModifierKeys == Keys.Control)
|
|
clickView(sender, args);
|
|
};
|
|
}
|
|
|
|
Counter = L_Count.Text;
|
|
Viewed = L_Viewed.Text;
|
|
L_Viewed.Text = ""; // invis for now
|
|
populateComboBoxes();
|
|
|
|
ContextMenuStrip mnu = new ContextMenuStrip();
|
|
ToolStripMenuItem mnuView = new ToolStripMenuItem("View");
|
|
|
|
// Assign event handlers
|
|
mnuView.Click += clickView;
|
|
|
|
// Add to main context menu
|
|
mnu.Items.AddRange(new ToolStripItem[] { mnuView });
|
|
|
|
// Assign to datagridview
|
|
foreach (PictureBox p in PKXBOXES)
|
|
p.ContextMenuStrip = mnu;
|
|
|
|
// Load initial database
|
|
Database.Add(new DatabaseList
|
|
{
|
|
Version = 0,
|
|
Title = "Misc",
|
|
Description = "Individual pk6 files present in the db/sav.",
|
|
});
|
|
|
|
// Load databases
|
|
foreach (string file in Directory.GetFiles(DatabasePath, "*", SearchOption.AllDirectories))
|
|
{
|
|
if ((new FileInfo(file)).Extension == ".pk6")
|
|
Database[0].Slot.Add(new PK6(File.ReadAllBytes(file), file));
|
|
else
|
|
loadDatabase(File.ReadAllBytes(file));
|
|
}
|
|
// Fetch from save file
|
|
for (int i = 0; i < 930; i++)
|
|
{
|
|
string Identifier = String.Format("B{0}:{1}", (i / 30 + 1).ToString("00"), (i % 30 + 1).ToString("00"));
|
|
PK6 pk6 = new PK6(PKX.decryptArray(Main.savefile.Skip(Main.SaveGame.Box + 0xE8*i).Take(0xE8).ToArray()), Identifier);
|
|
if (pk6.Species != 0)
|
|
Database[0].Slot.Add(pk6);
|
|
}
|
|
|
|
// Prepare Database
|
|
prepareDBForSearch();
|
|
}
|
|
private Main m_parent;
|
|
private PictureBox[] PKXBOXES;
|
|
private const string DatabasePath = "db";
|
|
private List<DatabaseList> Database = new List<DatabaseList>();
|
|
private List<PK6> Results;
|
|
private List<PK6> RawDB;
|
|
private int slotSelected = -1; // = null;
|
|
private const int RES_MAX = 66;
|
|
private const int RES_MIN = 6;
|
|
private string Counter;
|
|
private string Viewed;
|
|
|
|
// Important Events
|
|
private void clickView(object sender, EventArgs e)
|
|
{
|
|
string name = (sender is ToolStripItem)
|
|
? ((sender as ToolStripItem).Owner as ContextMenuStrip).SourceControl.Name
|
|
: (sender as PictureBox).Name;
|
|
|
|
int index = Array.FindIndex(PKXBOXES, p => p.Name == name);
|
|
var dataArr = Results.Skip(SCR_Box.Value * RES_MIN).Take(RES_MAX).ToArray();
|
|
if (index >= dataArr.Length)
|
|
System.Media.SystemSounds.Exclamation.Play();
|
|
else
|
|
{
|
|
m_parent.populateFields(dataArr[index].Data, false);
|
|
slotSelected = index + SCR_Box.Value * RES_MIN;
|
|
FillPKXBoxes(SCR_Box.Value);
|
|
L_Viewed.Text = String.Format(Viewed, dataArr[index].Identifier);
|
|
}
|
|
}
|
|
private void populateComboBoxes()
|
|
{
|
|
// Set the Text
|
|
CB_HeldItem.DisplayMember =
|
|
CB_Species.DisplayMember =
|
|
CB_Ability.DisplayMember =
|
|
CB_Nature.DisplayMember =
|
|
CB_GameOrigin.DisplayMember =
|
|
CB_HPType.DisplayMember = "Text";
|
|
|
|
// Set the Value
|
|
CB_HeldItem.ValueMember =
|
|
CB_Species.ValueMember =
|
|
CB_Ability.ValueMember =
|
|
CB_Nature.ValueMember =
|
|
CB_GameOrigin.ValueMember =
|
|
CB_HPType.ValueMember = "Value";
|
|
|
|
var Any = new Util.cbItem {Text = "Any", Value = -1};
|
|
|
|
var DS_Species = new List<Util.cbItem>(Main.SpeciesDataSource);
|
|
DS_Species.RemoveAt(0); DS_Species.Insert(0, Any); CB_Species.DataSource = DS_Species;
|
|
|
|
var DS_Item = new List<Util.cbItem>(Main.ItemDataSource);
|
|
DS_Item.Insert(0, Any); CB_HeldItem.DataSource = DS_Item;
|
|
|
|
var DS_Nature = new List<Util.cbItem>(Main.NatureDataSource);
|
|
DS_Nature.Insert(0, Any); CB_Nature.DataSource = DS_Nature;
|
|
|
|
var DS_Ability = new List<Util.cbItem>(Main.AbilityDataSource);
|
|
DS_Ability.Insert(0, Any); CB_Ability.DataSource = DS_Ability;
|
|
|
|
var DS_Version = new List<Util.cbItem>(Main.VersionDataSource);
|
|
DS_Version.Insert(0, Any); CB_GameOrigin.DataSource = DS_Version;
|
|
|
|
string[] hptypes = new string[Main.types.Length - 2]; Array.Copy(Main.types, 1, hptypes, 0, hptypes.Length);
|
|
var DS_Type = Util.getCBList(hptypes, null);
|
|
DS_Type.Insert(0, Any); CB_HPType.DataSource = DS_Type;
|
|
|
|
// Set the Move ComboBoxes too..
|
|
var DS_Move = new List<Util.cbItem>(Main.MoveDataSource);
|
|
DS_Move.RemoveAt(0); DS_Move.Insert(0, Any);
|
|
{
|
|
foreach (ComboBox cb in new[] { CB_Move1, CB_Move2, CB_Move3, CB_Move4 })
|
|
{
|
|
cb.DisplayMember = "Text"; cb.ValueMember = "Value";
|
|
cb.DataSource = new BindingSource(DS_Move, null);
|
|
}
|
|
}
|
|
|
|
// Trigger a Reset
|
|
resetFilters(null, null);
|
|
}
|
|
private void resetFilters(object sender, EventArgs e)
|
|
{
|
|
CHK_Shiny.Checked = CHK_IsEgg.Checked = true;
|
|
CHK_Shiny.CheckState = CHK_IsEgg.CheckState = CheckState.Indeterminate;
|
|
MT_ESV.Text = "";
|
|
CB_HeldItem.SelectedIndex = 0;
|
|
CB_Species.SelectedIndex = 0;
|
|
CB_Ability.SelectedIndex = 0;
|
|
CB_Nature.SelectedIndex = 0;
|
|
CB_HPType.SelectedIndex = 0;
|
|
|
|
CB_Level.SelectedIndex = 0;
|
|
TB_Level.Text = "";
|
|
CB_EVTrain.SelectedIndex = 0;
|
|
CB_IV.SelectedIndex = 0;
|
|
|
|
CB_Move1.SelectedIndex = CB_Move2.SelectedIndex = CB_Move3.SelectedIndex = CB_Move4.SelectedIndex = 0;
|
|
|
|
CB_GameOrigin.SelectedIndex = 0;
|
|
CB_Generation.SelectedIndex = 0;
|
|
|
|
MT_ESV.Visible = L_ESV.Visible = false;
|
|
}
|
|
private void generateDBReport(object sender, EventArgs e)
|
|
{
|
|
if (Util.Prompt(MessageBoxButtons.YesNoCancel, "Generate a Report on all data?", "This may take a while...")
|
|
!= DialogResult.Yes)
|
|
return;
|
|
|
|
frmReport ReportForm = new frmReport();
|
|
ReportForm.Show();
|
|
ReportForm.PopulateData(Results.ToArray());
|
|
}
|
|
|
|
// IO Usage
|
|
private class DatabaseList
|
|
{
|
|
public List<PK6> Slot = new List<PK6>();
|
|
public int Version;
|
|
private bool Unicode;
|
|
public byte[] Unused;
|
|
public string Title;
|
|
public string Description;
|
|
|
|
public DatabaseList() { }
|
|
public DatabaseList(byte[] data)
|
|
{
|
|
if ((data.Length < 0x100 + 0xE8) || (data.Length - 0x100)%0xE8 != 0)
|
|
return;
|
|
|
|
Version = BitConverter.ToInt32(data, 0);
|
|
Unicode = data[0x5] == 1;
|
|
Unused = data.Skip(4).Take(0xB).ToArray();
|
|
|
|
if (Unicode)
|
|
{
|
|
Title = Encoding.Unicode.GetString(data, 0x10, 0x30).Trim();
|
|
Description = Encoding.Unicode.GetString(data, 0x40, 0x60).Trim();
|
|
}
|
|
else
|
|
{
|
|
Title = Encoding.ASCII.GetString(data, 0x10, 0x30).Trim();
|
|
Description = Encoding.ASCII.GetString(data, 0x40, 0x60).Trim();
|
|
}
|
|
|
|
int count = (data.Length - 0x100)/0xE8;
|
|
for (int i = 0; i < count; i++)
|
|
Slot.Add(new PK6(data.Skip(0x100 + i * 0xE8).Take(0xE8).ToArray()));
|
|
}
|
|
public byte[] Write()
|
|
{
|
|
using (MemoryStream ms = new MemoryStream())
|
|
using (BinaryWriter bw = new BinaryWriter(ms))
|
|
{
|
|
bw.Write(Version);
|
|
bw.Write(Unused);
|
|
|
|
byte[] title = Unicode ? Encoding.Unicode.GetBytes(Title) : Encoding.ASCII.GetBytes(Title);
|
|
Array.Resize(ref title, 0x30);
|
|
bw.Write(title);
|
|
|
|
byte[] desc = Unicode ? Encoding.Unicode.GetBytes(Description) : Encoding.ASCII.GetBytes(Description);
|
|
Array.Resize(ref title, 0x60);
|
|
bw.Write(desc);
|
|
|
|
foreach (var pk6 in Slot)
|
|
bw.Write(pk6.Data.Take(0xE8).ToArray());
|
|
|
|
return ms.ToArray();
|
|
}
|
|
}
|
|
}
|
|
private void loadDatabase(byte[] data)
|
|
{
|
|
var db = new DatabaseList(data);
|
|
if (db.Slot.Count > 0)
|
|
Database.Add(db);
|
|
}
|
|
private void prepareDBForSearch()
|
|
{
|
|
RawDB = new List<PK6>();
|
|
|
|
foreach (var db in Database)
|
|
RawDB.AddRange(db.Slot);
|
|
|
|
RawDB = new List<PK6>(RawDB.Where(pk => pk.Checksum == pk.CalculateChecksum()));
|
|
RawDB = new List<PK6>(RawDB.Distinct());
|
|
setResults(RawDB);
|
|
}
|
|
private void openDB(object sender, EventArgs e)
|
|
{
|
|
if (Directory.Exists(DatabasePath))
|
|
Process.Start("explorer.exe", @DatabasePath);
|
|
}
|
|
private void Menu_Export_Click(object sender, EventArgs e)
|
|
{
|
|
if (Results == null || Results.Count == 0)
|
|
{ Util.Alert("No results to export."); return; }
|
|
|
|
if (DialogResult.Yes != Util.Prompt(MessageBoxButtons.YesNo, "Export to a folder?"))
|
|
return;
|
|
|
|
FolderBrowserDialog fbd = new FolderBrowserDialog();
|
|
if (DialogResult.OK != fbd.ShowDialog())
|
|
return;
|
|
|
|
string path = fbd.SelectedPath;
|
|
if (!Directory.Exists(path)) // just in case...
|
|
Directory.CreateDirectory(path);
|
|
|
|
foreach (PK6 pk6 in Results)
|
|
File.WriteAllBytes(Path.Combine(path, Util.CleanFileName(pk6.FileName)),
|
|
pk6.Data.Take(0xE8).ToArray());
|
|
}
|
|
|
|
// View Updates
|
|
private void B_Search_Click(object sender, EventArgs e)
|
|
{
|
|
// Populate Search Query Result
|
|
IEnumerable<PK6> res = RawDB;
|
|
|
|
// Primary Searchables
|
|
int species = Util.getIndex(CB_Species);
|
|
int ability = Util.getIndex(CB_Ability);
|
|
int nature = Util.getIndex(CB_Nature);
|
|
int item = Util.getIndex(CB_HeldItem);
|
|
if (species != -1) res = res.Where(pk => pk.Species == species);
|
|
if (ability != -1) res = res.Where(pk => pk.Ability == ability);
|
|
if (nature != -1) res = res.Where(pk => pk.Nature == nature);
|
|
if (item != -1) res = res.Where(pk => pk.HeldItem == item);
|
|
|
|
// Secondary Searchables
|
|
int move1 = Util.getIndex(CB_Move1);
|
|
int move2 = Util.getIndex(CB_Move2);
|
|
int move3 = Util.getIndex(CB_Move3);
|
|
int move4 = Util.getIndex(CB_Move4);
|
|
if (move1 != -1) res = res.Where(pk => pk.Moves.Contains(move1));
|
|
if (move2 != -1) res = res.Where(pk => pk.Moves.Contains(move2));
|
|
if (move3 != -1) res = res.Where(pk => pk.Moves.Contains(move3));
|
|
if (move4 != -1) res = res.Where(pk => pk.Moves.Contains(move4));
|
|
int vers = Util.getIndex(CB_GameOrigin);
|
|
if (vers != -1) res = res.Where(pk => pk.Version == vers);
|
|
int hptype = Util.getIndex(CB_HPType);
|
|
if (hptype != -1) res = res.Where(pk => pk.HPType == hptype);
|
|
if (CHK_Shiny.CheckState == CheckState.Checked) res = res.Where(pk => pk.IsShiny);
|
|
if (CHK_Shiny.CheckState == CheckState.Unchecked) res = res.Where(pk => !pk.IsShiny);
|
|
if (CHK_IsEgg.CheckState == CheckState.Checked) res = res.Where(pk => pk.IsEgg);
|
|
if (CHK_IsEgg.CheckState == CheckState.Unchecked) res = res.Where(pk => !pk.IsEgg);
|
|
if (CHK_IsEgg.CheckState == CheckState.Checked && MT_ESV.Text != "")
|
|
res = res.Where(pk => pk.PSV == Convert.ToInt16(MT_ESV.Text));
|
|
|
|
// Tertiary Searchables
|
|
if (TB_Level.Text != "") // Level
|
|
{
|
|
int level = Convert.ToInt16(TB_Level.Text);
|
|
if (level <= 100)
|
|
switch (CB_Level.SelectedIndex)
|
|
{
|
|
case 0: break; // Any
|
|
case 1: // <=
|
|
res = res.Where(pk => pk.Stat_Level <= level);
|
|
break;
|
|
case 2: // ==
|
|
res = res.Where(pk => pk.Stat_Level == level);
|
|
break;
|
|
case 3: // >=
|
|
res = res.Where(pk => pk.Stat_Level >= level);
|
|
break;
|
|
}
|
|
}
|
|
switch (CB_IV.SelectedIndex)
|
|
{
|
|
case 0: break; // Do nothing
|
|
case 1: // <= 90
|
|
res = res.Where(pk => pk.IVs.Sum() <= 90);
|
|
break;
|
|
case 2: // 91-120
|
|
res = res.Where(pk => pk.IVs.Sum() > 90 && pk.IVs.Sum() <= 120);
|
|
break;
|
|
case 3: // 121-150
|
|
res = res.Where(pk => pk.IVs.Sum() > 120 && pk.IVs.Sum() <= 150);
|
|
break;
|
|
case 4: // 151-179
|
|
res = res.Where(pk => pk.IVs.Sum() > 150 && pk.IVs.Sum() < 180);
|
|
break;
|
|
case 5: // 180+
|
|
res = res.Where(pk => pk.IVs.Sum() >= 180);
|
|
break;
|
|
case 6: // == 186
|
|
res = res.Where(pk => pk.IVs.Sum() == 186);
|
|
break;
|
|
}
|
|
switch (CB_EVTrain.SelectedIndex)
|
|
{
|
|
case 0: break; // Do nothing
|
|
case 1: // None (0)
|
|
res = res.Where(pk => pk.EVs.Sum() == 0);
|
|
break;
|
|
case 2: // Some (127-0)
|
|
res = res.Where(pk => pk.EVs.Sum() < 128);
|
|
break;
|
|
case 3: // Half (128-507)
|
|
res = res.Where(pk => pk.EVs.Sum() >= 128 && pk.EVs.Sum() < 508);
|
|
break;
|
|
case 4: // Full (508+)
|
|
res = res.Where(pk => pk.EVs.Sum() >= 508);
|
|
break;
|
|
}
|
|
switch (CB_Generation.SelectedIndex)
|
|
{
|
|
case 0: break; // Do nothing
|
|
case 1: // Gen 6
|
|
res = res.Where(pk => pk.Gen6);
|
|
break;
|
|
case 2: // Gen 5
|
|
res = res.Where(pk => pk.Gen5);
|
|
break;
|
|
case 3: // Gen 4
|
|
res = res.Where(pk => pk.Gen4);
|
|
break;
|
|
case 4: // Gen 3
|
|
res = res.Where(pk => pk.Gen3);
|
|
break;
|
|
}
|
|
|
|
slotSelected = -1; // reset the slot last viewed
|
|
var results = res.ToArray();
|
|
if (results.Length == 0)
|
|
Util.Alert("No results found!");
|
|
|
|
setResults(new List<PK6>(results)); // updates Count Label as well.
|
|
System.Media.SystemSounds.Asterisk.Play();
|
|
}
|
|
private void updateScroll(object sender, ScrollEventArgs e)
|
|
{
|
|
if (e.OldValue != e.NewValue)
|
|
FillPKXBoxes(e.NewValue);
|
|
}
|
|
private void setResults(List<PK6> res)
|
|
{
|
|
Results = new List<PK6>(res);
|
|
|
|
SCR_Box.Maximum = (int)Math.Ceiling((decimal)Results.Count / RES_MIN);
|
|
if (SCR_Box.Maximum > 0) SCR_Box.Maximum -= 1;
|
|
|
|
SCR_Box.Value = 0;
|
|
FillPKXBoxes(0);
|
|
|
|
L_Count.Text = String.Format(Counter, Results.Count);
|
|
}
|
|
private void FillPKXBoxes(int start)
|
|
{
|
|
if (Results == null)
|
|
for (int i = 0; i < RES_MAX; i++)
|
|
PKXBOXES[i].Image = null;
|
|
PK6[] data = (Results).Skip(start * RES_MIN).Take(RES_MAX).ToArray();
|
|
for (int i = 0; i < data.Length; i++)
|
|
PKXBOXES[i].Image = data[i].Sprite;
|
|
for (int i = data.Length; i < RES_MAX; i++)
|
|
PKXBOXES[i].Image = null;
|
|
|
|
for (int i = 0; i < RES_MAX; i++)
|
|
PKXBOXES[i].BackgroundImage = Resources.slotTrans;
|
|
if (slotSelected != -1 && slotSelected >= RES_MIN * start && slotSelected < RES_MIN * start + RES_MAX)
|
|
PKXBOXES[slotSelected - start * RES_MIN].BackgroundImage = Resources.slotView;
|
|
}
|
|
|
|
// Misc Update Methods
|
|
private void toggleESV(object sender, EventArgs e)
|
|
{
|
|
L_ESV.Visible = MT_ESV.Visible = CHK_IsEgg.CheckState == CheckState.Checked;
|
|
}
|
|
private void changeLevel(object sender, EventArgs e)
|
|
{
|
|
if (CB_Level.SelectedIndex == 0)
|
|
TB_Level.Text = "";
|
|
}
|
|
private void changeGame(object sender, EventArgs e)
|
|
{
|
|
if (CB_GameOrigin.SelectedIndex != 0)
|
|
CB_Generation.SelectedIndex = 0;
|
|
}
|
|
private void changeGeneration(object sender, EventArgs e)
|
|
{
|
|
if (CB_Generation.SelectedIndex != 0)
|
|
CB_GameOrigin.SelectedIndex = 0;
|
|
}
|
|
|
|
// Debug
|
|
private void testQuery()
|
|
{
|
|
var query = from db in Database
|
|
select db.Slot.Where(p => p.Move1 == 1).ToArray();
|
|
|
|
var result = query.ToArray();
|
|
if (!result[0].Any())
|
|
return;
|
|
|
|
var any = result[0][0].Data;
|
|
m_parent.populateFields(any);
|
|
}
|
|
private void testUnique()
|
|
{
|
|
var query = from db in Database
|
|
select db.Slot.GroupBy(p => (p.Checksum + p.EncryptionConstant + p.Species)) // Unique criteria
|
|
.Select(grp => grp.First()).ToArray();
|
|
|
|
var result = query.ToArray();
|
|
if (!result[0].Any())
|
|
return;
|
|
|
|
var any = result[0][0].Data;
|
|
m_parent.populateFields(any);
|
|
}
|
|
}
|
|
}
|