claunia/SabreTools
1
0
Fork 0
mirror of https://github.com/claunia/SabreTools.git synced 2025-12-16 19:14:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move IDDB splitting methods to IDDB

Browse Source
This commit is contained in:
Matt Nadareski
2024-03-19 20:41:39 -04:00
parent d6839e4c66
commit 440dcadfa1
2 changed files with 184 additions and 159 deletions
Download Patch File Download Diff File Expand all files Collapse all files

169
SabreTools.DatFiles/ItemDictionaryDB.cs
View File

@@ -1170,6 +1170,175 @@ namespace SabreTools.DatFiles
#endregion
#region Splitting
/// <summary>
/// Remove all BIOS and device sets
/// </summary>
public void RemoveBiosAndDeviceSets()
{
List<string> games = [.. SortedKeys];
foreach (string game in games)
{
var items = GetDatItemsForBucket(game);
if (items == null || items.Length == 0)
continue;
var machine = GetMachineForItem(items[0].Item1);
if (machine.Item2 == null)
continue;
if ((machine.Item2.GetBoolFieldValue(Models.Metadata.Machine.IsBiosKey) == true)
|| (machine.Item2.GetBoolFieldValue(Models.Metadata.Machine.IsDeviceKey) == true))
{
foreach (var item in items)
{
RemoveItem(item.Item1);
}
}
}
}
/// <summary>
/// Use romof tags to remove bios roms from children
/// </summary>
/// <param name="bios">True if only child Bios sets are touched, false for non-bios sets</param>
public void RemoveBiosRomsFromChild(bool bios)
{
// Loop through the romof tags
List<string> games = [.. SortedKeys];
foreach (string game in games)
{
// If the game has no items in it, we want to continue
var items = GetDatItemsForBucket(game);
if (items == null || items.Length == 0)
continue;
// Get the machine for the item
var machine = GetMachineForItem(items[0].Item1);
if (machine.Item2 == null)
continue;
// If the game (is/is not) a bios, we want to continue
if (bios ^ (machine.Item2.GetBoolFieldValue(Models.Metadata.Machine.IsBiosKey) == true))
continue;
// Determine if the game has a parent or not
string? parent = null;
if (!string.IsNullOrEmpty(machine.Item2.GetStringFieldValue(Models.Metadata.Machine.RomOfKey)))
parent = machine.Item2.GetStringFieldValue(Models.Metadata.Machine.RomOfKey);
// If the parent doesnt exist, we want to continue
if (string.IsNullOrEmpty(parent))
continue;
// If the parent doesn't have any items, we want to continue
var parentItems = GetDatItemsForBucket(parent!);
if (parentItems == null || parentItems.Length == 0)
continue;
// If the parent exists and has items, we remove the items that are in the parent from the current game
foreach ((long, DatItem) item in parentItems)
{
var matchedIndices = items.Where(i => i.Item2 == item.Item2).Select(i => i.Item1);
foreach (long index in matchedIndices)
{
RemoveItem(index);
}
}
}
}
/// <summary>
/// Use cloneof tags to remove roms from the children
/// </summary>
public void RemoveRomsFromChild()
{
List<string> games = [.. SortedKeys];
foreach (string game in games)
{
var items = GetDatItemsForBucket(game);
if (items == null)
continue;
// If the game has no items in it, we want to continue
if (items.Length == 0)
continue;
// Get the machine for the first item
var machine = GetMachineForItem(items[0].Item1);
if (machine.Item2 == null)
continue;
// Determine if the game has a parent or not
string? parent = null;
if (!string.IsNullOrEmpty(machine.Item2.GetStringFieldValue(Models.Metadata.Machine.CloneOfKey)))
parent = machine.Item2.GetStringFieldValue(Models.Metadata.Machine.CloneOfKey);
// If the parent doesnt exist, we want to continue
if (string.IsNullOrEmpty(parent))
continue;
// If the parent doesn't have any items, we want to continue
var parentItems = GetDatItemsForBucket(parent!);
if (parentItems == null || parentItems.Length == 0)
continue;
// If the parent exists and has items, we remove the parent items from the current game
foreach ((long, DatItem) item in parentItems)
{
var matchedIndices = items.Where(i => i.Item2 == item.Item2).Select(i => i.Item1);
foreach (long index in matchedIndices)
{
RemoveItem(index);
}
}
// Now we want to get the parent romof tag and put it in each of the remaining items
items = GetDatItemsForBucket(game);
machine = GetMachineForItem(GetDatItemsForBucket(parent!)![0].Item1);
if (machine.Item2 == null)
continue;
string? romof = machine.Item2.GetStringFieldValue(Models.Metadata.Machine.RomOfKey);
foreach ((long, DatItem) item in items!)
{
machine = GetMachineForItem(item.Item1);
if (machine.Item2 == null)
continue;
machine.Item2.SetFieldValue<string?>(Models.Metadata.Machine.RomOfKey, romof);
}
}
}
/// <summary>
/// Remove all romof and cloneof tags from all games
/// </summary>
public void RemoveTagsFromChild()
{
List<string> games = [.. SortedKeys];
foreach (string game in games)
{
var items = GetDatItemsForBucket(game);
if (items == null)
continue;
foreach ((long, DatItem) item in items)
{
var machine = GetMachineForItem(item.Item1);
if (machine.Item2 == null)
continue;
machine.Item2.SetFieldValue<string?>(Models.Metadata.Machine.CloneOfKey, null);
machine.Item2.SetFieldValue<string?>(Models.Metadata.Machine.RomOfKey, null);
machine.Item2.SetFieldValue<string?>(Models.Metadata.Machine.SampleOfKey, null);
}
}
}
#endregion
#region Statistics
/// <summary>

174
SabreTools.Filtering/Splitter.cs
View File

@@ -113,7 +113,7 @@ namespace SabreTools.Filtering
// Then, remove the romof and cloneof tags so it's not picked up by the manager
RemoveTagsFromChild(datFile);
RemoveTagsFromChildDB(datFile);
datFile.ItemsDB.RemoveTagsFromChild();
}
/// <summary>
@@ -132,13 +132,13 @@ namespace SabreTools.Filtering
// Now that we have looped through the cloneof tags, we loop through the romof tags
RemoveBiosRomsFromChild(datFile, false);
RemoveBiosRomsFromChildDB(datFile, false);
datFile.ItemsDB.RemoveBiosRomsFromChild(false);
RemoveBiosRomsFromChild(datFile, true);
RemoveBiosRomsFromChildDB(datFile, true);
datFile.ItemsDB.RemoveBiosRomsFromChild(true);
// Finally, remove the romof and cloneof tags so it's not picked up by the manager
RemoveTagsFromChild(datFile);
RemoveTagsFromChildDB(datFile);
datFile.ItemsDB.RemoveTagsFromChild();
}
/// <summary>
@@ -162,7 +162,7 @@ namespace SabreTools.Filtering
// Then, remove the romof and cloneof tags so it's not picked up by the manager
RemoveTagsFromChild(datFile);
RemoveTagsFromChildDB(datFile);
datFile.ItemsDB.RemoveTagsFromChild();
}
/// <summary>
@@ -181,13 +181,13 @@ namespace SabreTools.Filtering
// Now that we have looped through the cloneof tags, we loop through the romof tags
RemoveBiosRomsFromChild(datFile, false);
RemoveBiosRomsFromChildDB(datFile, false);
datFile.ItemsDB.RemoveBiosRomsFromChild(false);
RemoveBiosRomsFromChild(datFile, true);
RemoveBiosRomsFromChildDB(datFile, true);
datFile.ItemsDB.RemoveBiosRomsFromChild(true);
// Finally, remove the romof and cloneof tags so it's not picked up by the manager
RemoveTagsFromChild(datFile);
RemoveTagsFromChildDB(datFile);
datFile.ItemsDB.RemoveTagsFromChild();
}
/// <summary>
@@ -206,13 +206,13 @@ namespace SabreTools.Filtering
// Now that we have looped through the cloneof tags, we loop through the romof tags
RemoveBiosRomsFromChild(datFile, false);
RemoveBiosRomsFromChildDB(datFile, false);
datFile.ItemsDB.RemoveBiosRomsFromChild(false);
RemoveBiosRomsFromChild(datFile, true);
RemoveBiosRomsFromChildDB(datFile, true);
datFile.ItemsDB.RemoveBiosRomsFromChild(true);
// Finally, remove the romof and cloneof tags so it's not picked up by the manager
RemoveTagsFromChild(datFile);
RemoveTagsFromChildDB(datFile);
datFile.ItemsDB.RemoveTagsFromChild();
}
/// <summary>
@@ -228,17 +228,17 @@ namespace SabreTools.Filtering
// Now we want to loop through all of the games and set the correct information
RemoveRomsFromChild(datFile);
RemoveRomsFromChildDB(datFile);
datFile.ItemsDB.RemoveRomsFromChild();
// Now that we have looped through the cloneof tags, we loop through the romof tags
RemoveBiosRomsFromChild(datFile, false);
RemoveBiosRomsFromChildDB(datFile, false);
datFile.ItemsDB.RemoveBiosRomsFromChild(false);
RemoveBiosRomsFromChild(datFile, true);
RemoveBiosRomsFromChildDB(datFile, true);
datFile.ItemsDB.RemoveBiosRomsFromChild(true);
// Finally, remove the romof and cloneof tags so it's not picked up by the manager
RemoveTagsFromChild(datFile);
RemoveTagsFromChildDB(datFile);
datFile.ItemsDB.RemoveTagsFromChild();
}
/// <summary>
@@ -637,31 +637,6 @@ namespace SabreTools.Filtering
}
}
/// <summary>
/// Remove all BIOS and device sets
/// </summary>
/// <param name="datFile">Current DatFile object to run operations on</param>
internal static void RemoveBiosAndDeviceSetsDB(DatFile datFile)
{
List<string> games = [.. datFile.ItemsDB.SortedKeys];
foreach (string game in games)
{
var items = datFile.ItemsDB.GetDatItemsForBucket(game);
if (items == null)
continue;
if (items.Length > 0
&& ((items[0].Item2.GetFieldValue<Machine>(DatItem.MachineKey)!.GetBoolFieldValue(Models.Metadata.Machine.IsBiosKey) == true)
|| (items[0].Item2.GetFieldValue<Machine>(DatItem.MachineKey)!.GetBoolFieldValue(Models.Metadata.Machine.IsDeviceKey) == true)))
{
foreach (var item in items)
{
datFile.ItemsDB.RemoveItem(item.Item1);
}
}
}
}
/// <summary>
/// Use romof tags to remove bios roms from children
/// </summary>
@@ -711,52 +686,6 @@ namespace SabreTools.Filtering
}
}
/// <summary>
/// Use romof tags to remove bios roms from children
/// </summary>
/// <param name="datFile">Current DatFile object to run operations on</param>
/// <param name="bios">True if only child Bios sets are touched, false for non-bios sets (default)</param>
internal static void RemoveBiosRomsFromChildDB(DatFile datFile, bool bios = false)
{
// Loop through the romof tags
List<string> games = [.. datFile.ItemsDB.SortedKeys];
foreach (string game in games)
{
// If the game has no items in it, we want to continue
var items = datFile.ItemsDB.GetDatItemsForBucket(game);
if (items == null || items.Length == 0)
continue;
// If the game (is/is not) a bios, we want to continue
if (bios ^ (items[0].Item2.GetFieldValue<Machine>(DatItem.MachineKey)!.GetBoolFieldValue(Models.Metadata.Machine.IsBiosKey) == true))
continue;
// Determine if the game has a parent or not
string? parent = null;
if (!string.IsNullOrEmpty(items[0].Item2.GetFieldValue<Machine>(DatItem.MachineKey)!.GetStringFieldValue(Models.Metadata.Machine.RomOfKey)))
parent = items[0].Item2.GetFieldValue<Machine>(DatItem.MachineKey)!.GetStringFieldValue(Models.Metadata.Machine.RomOfKey);
// If the parent doesnt exist, we want to continue
if (string.IsNullOrEmpty(parent))
continue;
// If the parent doesn't have any items, we want to continue
var parentItems = datFile.ItemsDB.GetDatItemsForBucket(parent!);
if (parentItems == null || parentItems.Length == 0)
continue;
// If the parent exists and has items, we remove the items that are in the parent from the current game
foreach ((long, DatItem) item in parentItems)
{
var matchedIndices = items.Where(i => i.Item2 == item.Item2).Select(i => i.Item1);
foreach (long index in matchedIndices)
{
datFile.ItemsDB.RemoveItem(index);
}
}
}
}
/// <summary>
/// Use cloneof tags to remove roms from the children
/// </summary>
@@ -808,57 +737,6 @@ namespace SabreTools.Filtering
}
}
/// <summary>
/// Use cloneof tags to remove roms from the children
/// </summary>
/// <param name="datFile">Current DatFile object to run operations on</param>
internal static void RemoveRomsFromChildDB(DatFile datFile)
{
List<string> games = [.. datFile.ItemsDB.SortedKeys];
foreach (string game in games)
{
var items = datFile.ItemsDB.GetDatItemsForBucket(game);
if (items == null)
continue;
// If the game has no items in it, we want to continue
if (items.Length == 0)
continue;
// Determine if the game has a parent or not
string? parent = null;
if (!string.IsNullOrEmpty(items[0].Item2.GetFieldValue<Machine>(DatItem.MachineKey)!.GetStringFieldValue(Models.Metadata.Machine.CloneOfKey)))
parent = items[0].Item2.GetFieldValue<Machine>(DatItem.MachineKey)!.GetStringFieldValue(Models.Metadata.Machine.CloneOfKey);
// If the parent doesnt exist, we want to continue
if (string.IsNullOrEmpty(parent))
continue;
// If the parent doesn't have any items, we want to continue
var parentItems = datFile.ItemsDB.GetDatItemsForBucket(parent!);
if (parentItems == null || parentItems.Length == 0)
continue;
// If the parent exists and has items, we remove the parent items from the current game
foreach ((long, DatItem) item in parentItems)
{
var matchedIndices = items.Where(i => i.Item2 == item.Item2).Select(i => i.Item1);
foreach (long index in matchedIndices)
{
datFile.ItemsDB.RemoveItem(index);
}
}
// Now we want to get the parent romof tag and put it in each of the remaining items
items = datFile.ItemsDB.GetDatItemsForBucket(game);
string? romof = datFile.ItemsDB.GetDatItemsForBucket(parent)![0].Item2.GetFieldValue<Machine>(DatItem.MachineKey)!.GetStringFieldValue(Models.Metadata.Machine.RomOfKey);
foreach ((long, DatItem) item in items!)
{
item.Item2.GetFieldValue<Machine>(DatItem.MachineKey)!.SetFieldValue<string?>(Models.Metadata.Machine.RomOfKey, romof);
}
}
}
/// <summary>
/// Remove all romof and cloneof tags from all games
/// </summary>
@@ -881,28 +759,6 @@ namespace SabreTools.Filtering
}
}
/// <summary>
/// Remove all romof and cloneof tags from all games
/// </summary>
/// <param name="datFile">Current DatFile object to run operations on</param>
internal static void RemoveTagsFromChildDB(DatFile datFile)
{
List<string> games = [.. datFile.ItemsDB.SortedKeys];
foreach (string game in games)
{
var items = datFile.ItemsDB.GetDatItemsForBucket(game);
if (items == null)
continue;
foreach ((long, DatItem) item in items)
{
item.Item2.GetFieldValue<Machine>(DatItem.MachineKey)!.SetFieldValue<string?>(Models.Metadata.Machine.CloneOfKey, null);
item.Item2.GetFieldValue<Machine>(DatItem.MachineKey)!.SetFieldValue<string?>(Models.Metadata.Machine.RomOfKey, null);
item.Item2.GetFieldValue<Machine>(DatItem.MachineKey)!.SetFieldValue<string?>(Models.Metadata.Machine.SampleOfKey, null);
}
}
}
#endregion
}
}