mirror of
https://github.com/AdAstra-LD/DS-Pokemon-Rom-Editor.git
synced 2026-05-09 12:51:54 -05:00
302 lines
14 KiB
C#
302 lines
14 KiB
C#
using DSPRE.ROMFiles;
|
|
using System.Collections.Generic;
|
|
|
|
using System;
|
|
using System.Windows.Forms;
|
|
using static DSPRE.RomInfo;
|
|
using System.Reflection;
|
|
using System.Linq;
|
|
|
|
namespace DSPRE {
|
|
|
|
public partial class HeaderSearch : Form {
|
|
public static readonly Dictionary<MapHeader.SearchableFields, string> searchableHeaderFieldsDict = new Dictionary<MapHeader.SearchableFields, string>() {
|
|
[MapHeader.SearchableFields.AreaDataID] = "Area Data (ID)",
|
|
[MapHeader.SearchableFields.CameraAngleID] = "Camera Angle (ID)",
|
|
[MapHeader.SearchableFields.EventFileID] = "Event File (ID)",
|
|
[MapHeader.SearchableFields.InternalName] = "Internal Name",
|
|
[MapHeader.SearchableFields.LevelScriptID] = "Level Script (ID)",
|
|
[MapHeader.SearchableFields.MatrixID] = "Matrix (ID)",
|
|
[MapHeader.SearchableFields.MusicDayID] = "Music Day (ID)",
|
|
//[MapHeader.SearchableFields.MusicDayName] = "Music Day (Name)",
|
|
[MapHeader.SearchableFields.MusicNightID] = "Music Night (ID)",
|
|
//[MapHeader.SearchableFields.MusicNightName] = "Music Night (Name)",
|
|
[MapHeader.SearchableFields.ScriptFileID] = "Script File (ID)",
|
|
[MapHeader.SearchableFields.TextArchiveID] = "Text Archive (ID)",
|
|
[MapHeader.SearchableFields.WeatherID] = "Weather (ID)"
|
|
};
|
|
public enum NumOperators : byte {
|
|
//Order matters!
|
|
Equal,
|
|
Different,
|
|
Less,
|
|
Greater,
|
|
LessOrEqual,
|
|
GreaterOrEqual
|
|
};
|
|
|
|
public enum TextOperators : byte {
|
|
//Order matters!
|
|
Contains,
|
|
DoesNotContain,
|
|
IsExactly,
|
|
IsNot
|
|
};
|
|
|
|
private static readonly Dictionary<NumOperators, string> numOperatorsDict = new Dictionary<NumOperators, string>() {
|
|
//Order matters!
|
|
|
|
[NumOperators.Equal] = "Equals",
|
|
[NumOperators.Different] = "Is Different than",
|
|
[NumOperators.Less] = "Is Less than",
|
|
[NumOperators.Greater] = "Is Greater than",
|
|
[NumOperators.LessOrEqual] = "Is Less than or Equal to",
|
|
[NumOperators.GreaterOrEqual] = "Is Greater than or Equal to",
|
|
};
|
|
|
|
private static readonly Dictionary<TextOperators, string> textOperatorsDict = new Dictionary<TextOperators, string>() {
|
|
//Order matters!
|
|
|
|
[TextOperators.Contains] = "Contains",
|
|
[TextOperators.DoesNotContain] = "Does not contain",
|
|
[TextOperators.IsExactly] = "Is Exactly",
|
|
[TextOperators.IsNot] = "Is Not",
|
|
};
|
|
|
|
private List<string> intNames;
|
|
private ListBox headerListBox;
|
|
private ToolStripStatusLabel statusLabel;
|
|
|
|
public string status = "Ready";
|
|
|
|
public HeaderSearch(ref List<string> internalNames, ListBox headerListBox, ToolStripStatusLabel statusLabel) {
|
|
InitializeComponent();
|
|
|
|
intNames = internalNames;
|
|
this.headerListBox = headerListBox;
|
|
this.statusLabel = statusLabel;
|
|
|
|
foreach (string elem in searchableHeaderFieldsDict.Values) {
|
|
fieldToSearch1ComboBox.Items.Add(elem);
|
|
}
|
|
|
|
fieldToSearch1ComboBox.SelectedIndex = 0;
|
|
operator1ComboBox.SelectedIndex = 0;
|
|
}
|
|
|
|
#region Helper Methods
|
|
private void UpdateOperators(ComboBox operatorComboBox, ComboBox fieldToSearchComboBox) {
|
|
operatorComboBox.Items.Clear();
|
|
|
|
if (fieldToSearchComboBox.SelectedItem.ToString().Contains("ID")) {
|
|
foreach (string elem in numOperatorsDict.Values) {
|
|
operatorComboBox.Items.Add(elem);
|
|
}
|
|
valueTextBox.MaxLength = 5;
|
|
} else {
|
|
foreach (string elem in textOperatorsDict.Values) {
|
|
operatorComboBox.Items.Add(elem);
|
|
}
|
|
valueTextBox.MaxLength = 16;
|
|
}
|
|
|
|
operatorComboBox.SelectedIndex = 0;
|
|
}
|
|
#endregion
|
|
public static void ResetResults(ListBox headerListBox, List<string> intNames, bool prependNumbers) {
|
|
if (headerListBox.Items.Count < intNames.Count) {
|
|
|
|
headerListBox.Enabled = true;
|
|
headerListBox.Items.Clear();
|
|
|
|
if (prependNumbers) {
|
|
for (int i = 0; i < intNames.Count; i++) {
|
|
string name = intNames[i];
|
|
headerListBox.Items.Add(i.ToString("D3") + MapHeader.nameSeparator + name);
|
|
}
|
|
} else {
|
|
headerListBox.Items.AddRange(intNames.ToArray());
|
|
}
|
|
}
|
|
}
|
|
public static HashSet<string> AdvancedSearch(ushort startID, ushort finalID, List<string> intNames, int fieldToSearch, int oper, string valToSearch) {
|
|
if (fieldToSearch < 0 || oper < 0 || valToSearch == "") {
|
|
return null;
|
|
}
|
|
|
|
HashSet<string> result = new HashSet<string>();
|
|
|
|
switch (fieldToSearch) {
|
|
case (int)MapHeader.SearchableFields.InternalName:
|
|
for (ushort i = startID; i < finalID; i++) {
|
|
switch (oper) {
|
|
case (int)TextOperators.IsExactly:
|
|
if (intNames[i].Equals(valToSearch)) {
|
|
result.Add(i.ToString("D3") + MapHeader.nameSeparator + intNames[i]);
|
|
}
|
|
break;
|
|
case (int)TextOperators.IsNot:
|
|
if (!intNames[i].Equals(valToSearch)) {
|
|
result.Add(i.ToString("D3") + MapHeader.nameSeparator + intNames[i]);
|
|
}
|
|
break;
|
|
case (int)TextOperators.Contains:
|
|
if (intNames[i].IndexOf(valToSearch, StringComparison.InvariantCultureIgnoreCase) >= 0) {
|
|
result.Add(i.ToString("D3") + MapHeader.nameSeparator + intNames[i]);
|
|
}
|
|
break;
|
|
case (int)TextOperators.DoesNotContain:
|
|
if (intNames[i].IndexOf(valToSearch, StringComparison.InvariantCultureIgnoreCase) < 0) {
|
|
result.Add(i.ToString("D3") + MapHeader.nameSeparator + intNames[i]);
|
|
}
|
|
break;
|
|
default:
|
|
Console.WriteLine("Unrecognized operand!!!");
|
|
break;
|
|
}
|
|
}
|
|
break;
|
|
//case (int)MapHeader.SearchableFields.MusicDayName:
|
|
//Maybe in the future
|
|
//break;
|
|
//case (int)MapHeader.SearchableFields.MusicNightName:
|
|
//Maybe in the future
|
|
//break;
|
|
default:
|
|
string[] fieldSplit = searchableHeaderFieldsDict[(MapHeader.SearchableFields)fieldToSearch].Split();
|
|
|
|
fieldSplit[0] = fieldSplit[0].ToLower();
|
|
fieldSplit[fieldSplit.Length - 1] = fieldSplit[fieldSplit.Length - 1].Replace("(", "").Replace(")", ""); //Remove ( and ) from string
|
|
|
|
PropertyInfo property = typeof(MapHeader).GetProperty(String.Join("", fieldSplit));
|
|
ushort numToSearch;
|
|
|
|
try {
|
|
numToSearch = ushort.Parse(valToSearch);
|
|
} catch (OverflowException) {
|
|
MessageBox.Show("Your input exceeds the range of 16-bit integers (" + ushort.MinValue + " - " + ushort.MaxValue + ").", "Overflow Error", MessageBoxButtons.OK, MessageBoxIcon.Information);
|
|
return null;
|
|
}
|
|
|
|
for (ushort i = startID; i < finalID; i++) {
|
|
MapHeader h;
|
|
if (PatchToolboxDialog.flag_DynamicHeadersPatchApplied || PatchToolboxDialog.CheckFilesDynamicHeadersPatchApplied()) {
|
|
h = MapHeader.LoadFromFile(RomInfo.gameDirs[DirNames.dynamicHeaders].unpackedDir + "\\" + i.ToString("D4"), i, 0);
|
|
} else {
|
|
h = MapHeader.LoadFromARM9(i);
|
|
}
|
|
|
|
int headerField = int.Parse(property.GetValue(h, null).ToString());
|
|
|
|
switch (oper) {
|
|
case (int)NumOperators.Less:
|
|
if (headerField < numToSearch) {
|
|
result.Add(i.ToString("D3") + MapHeader.nameSeparator + intNames[i]);
|
|
}
|
|
break;
|
|
case (int)NumOperators.Equal:
|
|
if (headerField == numToSearch) {
|
|
result.Add(i.ToString("D3") + MapHeader.nameSeparator + intNames[i]);
|
|
}
|
|
break;
|
|
case (int)NumOperators.Greater:
|
|
if (headerField > numToSearch) {
|
|
result.Add(i.ToString("D3") + MapHeader.nameSeparator + intNames[i]);
|
|
}
|
|
break;
|
|
case (int)NumOperators.LessOrEqual:
|
|
if (headerField <= numToSearch) {
|
|
result.Add(i.ToString("D3") + MapHeader.nameSeparator + intNames[i]);
|
|
}
|
|
break;
|
|
case (int)NumOperators.GreaterOrEqual:
|
|
if (headerField >= numToSearch) {
|
|
result.Add(i.ToString("D3") + MapHeader.nameSeparator + intNames[i]);
|
|
}
|
|
break;
|
|
case (int)NumOperators.Different:
|
|
if (headerField != numToSearch) {
|
|
result.Add(i.ToString("D3") + MapHeader.nameSeparator + intNames[i]);
|
|
}
|
|
break;
|
|
default:
|
|
Console.WriteLine("Unrecognized operand!!!");
|
|
break;
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
return result;
|
|
}
|
|
private void startSearchButton_Click(object sender, EventArgs e) {
|
|
StartSearch(showDialog: true);
|
|
}
|
|
|
|
private void StartSearch(bool showDialog = true) {
|
|
if (valueTextBox.Text == "") {
|
|
//if (showDialog) {
|
|
// MessageBox.Show("Value to search is empty", "Can't search", MessageBoxButtons.OK, MessageBoxIcon.Exclamation);
|
|
//}
|
|
headerSearchResetButton_Click(null, null);
|
|
return;
|
|
}
|
|
|
|
HashSet<string> result;
|
|
headerListBox.Items.Clear();
|
|
|
|
try {
|
|
result = AdvancedSearch(0, (ushort)intNames.Count, intNames, fieldToSearch1ComboBox.SelectedIndex, operator1ComboBox.SelectedIndex, valueTextBox.Text);
|
|
} catch (FormatException) {
|
|
if (showDialog) {
|
|
MessageBox.Show("Make sure the value to search is correct.", "Format Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
|
|
}
|
|
valueTextBox.Clear();
|
|
headerListBox.Items.Add("Search parameters are invalid");
|
|
headerListBox.Enabled = false;
|
|
return;
|
|
}
|
|
|
|
string searchConfiguration = fieldToSearch1ComboBox.Text + " " + operator1ComboBox.Text.ToLower() + " " + '"' + valueTextBox.Text + '"';
|
|
if (result is null || result.Count <= 0) {
|
|
string res = "No header's " + searchConfiguration;
|
|
headerListBox.Items.Add(res);
|
|
headerListBox.Enabled = false;
|
|
statusLabel.Text = res;
|
|
} else {
|
|
string[] arr = new string[result.Count];
|
|
result.CopyTo(arr);
|
|
headerListBox.Items.AddRange(arr);
|
|
headerListBox.SelectedIndex = 0;
|
|
headerListBox.Enabled = true;
|
|
|
|
statusLabel.Text = "Showing headers whose " + searchConfiguration;
|
|
}
|
|
Update();
|
|
}
|
|
|
|
private void valueTextBox_KeyUp(object sender, KeyEventArgs e) {
|
|
if (autoSearchCB.Checked) {
|
|
StartSearch(showDialog: false);
|
|
} else if (e.KeyCode == Keys.Enter) {
|
|
StartSearch(showDialog: true);
|
|
}
|
|
}
|
|
private void headerSearchResetButton_Click(object sender, EventArgs e) {
|
|
ResetResults(headerListBox, intNames, prependNumbers: true);
|
|
valueTextBox.Clear();
|
|
statusLabel.Text = "Ready";
|
|
}
|
|
private void fieldToSearch1ComboBox_SelectedIndexChanged(object sender, EventArgs e) {
|
|
UpdateOperators(operator1ComboBox, fieldToSearch1ComboBox);
|
|
if (autoSearchCB.Checked) {
|
|
StartSearch(showDialog: false);
|
|
}
|
|
}
|
|
private void operator1ComboBox_SelectedIndexChanged(object sender, EventArgs e) {
|
|
if (autoSearchCB.Checked) {
|
|
StartSearch(showDialog: false);
|
|
}
|
|
}
|
|
}
|
|
}
|