From ecafe7268c75e1aa65e8e54681d63c29ad8029ea Mon Sep 17 00:00:00 2001
From: Matt Nadareski <mnadareski@outlook.com>
Date: Wed, 13 Mar 2024 20:51:14 -0400
Subject: [PATCH] Add ability to filter on retrieval in IDDB

---
 SabreTools.DatFiles/ItemDictionaryDB.cs | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/SabreTools.DatFiles/ItemDictionaryDB.cs b/SabreTools.DatFiles/ItemDictionaryDB.cs
index aeff2f93..1627c4fe 100644
--- a/SabreTools.DatFiles/ItemDictionaryDB.cs
+++ b/SabreTools.DatFiles/ItemDictionaryDB.cs
@@ -180,7 +180,7 @@ namespace SabreTools.DatFiles
         /// <summary>
         /// Get the items associated with a bucket name
         /// </summary>
-        public DatItem[]? GetDatItemsForBucket(string bucketName)
+        public DatItem[]? GetDatItemsForBucket(string bucketName, bool filter = false)
         {
             if (!_buckets.ContainsKey(bucketName))
                 return null;
@@ -190,7 +190,7 @@ namespace SabreTools.DatFiles
             var datItems = new List<DatItem>();
             foreach (long itemId in itemIds)
             {
-                if (_items.ContainsKey(itemId))
+                if (_items.ContainsKey(itemId) && (!filter || _items[itemId].GetBoolFieldValue(DatItem.RemoveKey) != true))
                     datItems.Add(_items[itemId]);
             }
 
@@ -200,7 +200,7 @@ namespace SabreTools.DatFiles
         /// <summary>
         /// Get the items associated with a machine index
         /// </summary>
-        public DatItem[]? GetDatItemsForMachine(long machineIndex)
+        public DatItem[]? GetDatItemsForMachine(long machineIndex, bool filter = false)
         {
             var itemIds = _itemToMachineMapping
                 .Where(mapping => mapping.Value == machineIndex)
@@ -209,7 +209,7 @@ namespace SabreTools.DatFiles
             var datItems = new List<DatItem>();
             foreach (long itemId in itemIds)
             {
-                if (_items.ContainsKey(itemId))
+                if (_items.ContainsKey(itemId) && (!filter || _items[itemId].GetBoolFieldValue(DatItem.RemoveKey) != true))
                     datItems.Add(_items[itemId]);
             }