Create some IDDB variants for splitting methods

This commit is contained in:
Matt Nadareski
2024-03-19 20:27:44 -04:00
parent e342c6e7d3
commit d6839e4c66
2 changed files with 169 additions and 2 deletions

View File

@@ -312,6 +312,7 @@ namespace SabreTools.Filtering
// Finally, strip out the parent tags
Splitter.RemoveTagsFromChild(datFile);
Splitter.RemoveTagsFromChildDB(datFile);
}
/// <summary>

View File

@@ -1,7 +1,6 @@
using System;
using System.Collections.Generic;
using System.Linq;
using SabreTools.Core;
using SabreTools.Core.Tools;
using SabreTools.DatFiles;
using SabreTools.DatItems;
@@ -114,6 +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);
}
/// <summary>
@@ -132,10 +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);
RemoveBiosRomsFromChild(datFile, true);
RemoveBiosRomsFromChildDB(datFile, true);
// Finally, remove the romof and cloneof tags so it's not picked up by the manager
RemoveTagsFromChild(datFile);
RemoveTagsFromChildDB(datFile);
}
/// <summary>
@@ -159,6 +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);
}
/// <summary>
@@ -177,10 +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);
RemoveBiosRomsFromChild(datFile, true);
RemoveBiosRomsFromChildDB(datFile, true);
// Finally, remove the romof and cloneof tags so it's not picked up by the manager
RemoveTagsFromChild(datFile);
RemoveTagsFromChildDB(datFile);
}
/// <summary>
@@ -199,10 +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);
RemoveBiosRomsFromChild(datFile, true);
RemoveBiosRomsFromChildDB(datFile, true);
// Finally, remove the romof and cloneof tags so it's not picked up by the manager
RemoveTagsFromChild(datFile);
RemoveTagsFromChildDB(datFile);
}
/// <summary>
@@ -218,13 +228,17 @@ namespace SabreTools.Filtering
// Now we want to loop through all of the games and set the correct information
RemoveRomsFromChild(datFile);
RemoveRomsFromChildDB(datFile);
// Now that we have looped through the cloneof tags, we loop through the romof tags
RemoveBiosRomsFromChild(datFile, false);
RemoveBiosRomsFromChildDB(datFile, false);
RemoveBiosRomsFromChild(datFile, true);
RemoveBiosRomsFromChildDB(datFile, true);
// Finally, remove the romof and cloneof tags so it's not picked up by the manager
RemoveTagsFromChild(datFile);
RemoveTagsFromChildDB(datFile);
}
/// <summary>
@@ -273,6 +287,8 @@ namespace SabreTools.Filtering
}
}
// TODO: Add AddRomsFromBiosDB
/// <summary>
/// Use device_ref and optionally slotoption tags to add roms to the children
/// </summary>
@@ -421,6 +437,8 @@ namespace SabreTools.Filtering
return foundnew;
}
// TODO: Add AddRomsFromDevicesDB
/// <summary>
/// Use cloneof tags to add roms to the children, setting the new romof tag in the process
/// </summary>
@@ -472,6 +490,8 @@ namespace SabreTools.Filtering
}
}
// TODO: Add AddRomsFromParentDB
/// <summary>
/// Use cloneof tags to add roms to the parents, removing the child sets in the process
/// </summary>
@@ -593,6 +613,8 @@ namespace SabreTools.Filtering
}
}
// TODO: Add AddRomsFromChildrenDB
/// <summary>
/// Remove all BIOS and device sets
/// </summary>
@@ -615,6 +637,31 @@ 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>
@@ -664,6 +711,52 @@ 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>
@@ -715,13 +808,64 @@ 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>
/// <param name="datFile">Current DatFile object to run operations on</param>
internal static void RemoveTagsFromChild(DatFile datFile)
{
List<string> games = datFile.Items.Keys.OrderBy(g => g).ToList();
List<string> games = [.. datFile.Items.Keys.OrderBy(g => g)];
foreach (string game in games)
{
var items = datFile.Items[game];
@@ -737,6 +881,28 @@ 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
}
}