From 2310ec42f05b7da7629e7397fdbf6007434cd96e Mon Sep 17 00:00:00 2001
From: Matt Nadareski <mnadareski@outlook.com>
Date: Wed, 22 Jun 2016 21:19:25 -0700
Subject: [PATCH] [RomTools] Add ability to remove from input

---
 SabreTools.Helper/Tools/RomTools.cs | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/SabreTools.Helper/Tools/RomTools.cs b/SabreTools.Helper/Tools/RomTools.cs
index 26a10459..e1c71a4e 100644
--- a/SabreTools.Helper/Tools/RomTools.cs
+++ b/SabreTools.Helper/Tools/RomTools.cs
@@ -2,6 +2,7 @@
 using System;
 using System.Collections.Generic;
 using System.IO;
+using System.Linq;
 using System.Security.Cryptography;
 using System.Text.RegularExpressions;
 
@@ -187,8 +188,9 @@ namespace SabreTools.Helper
 		/// </summary>
 		/// <param name="lastrom">Rom to use as a base</param>
 		/// <param name="datdata">DAT to match against</param>
+		/// <param name="remove">True to remove matched roms from the input, false otherwise (default)</param>
 		/// <returns>List of matched RomData objects</returns>
-		public static List<Rom> GetDuplicates(Rom lastrom, Dat datdata)
+		public static List<Rom> GetDuplicates(Rom lastrom, Dat datdata, bool remove = false)
 		{
 			List<Rom> output = new List<Rom>();
 
@@ -199,14 +201,27 @@ namespace SabreTools.Helper
 			}
 
 			// Try to find duplicates
-			foreach (List<Rom> roms in datdata.Roms.Values)
+			List<string> keys = datdata.Roms.Keys.ToList();
+			foreach (string key in keys)
 			{
+				List<Rom> roms = datdata.Roms[key];
+				List<Rom> left = new List<Rom>();
 				foreach (Rom rom in roms)
 				{
 					if (IsDuplicate(rom, lastrom))
 					{
 						output.Add(rom);
 					}
+					else
+					{
+						left.Add(rom);
+					}
+				}
+
+				// If we're in removal mode, replace the list with the new one
+				if (remove)
+				{
+					datdata.Roms[key] = left;
 				}
 			}