make discohawk a/b comparer run big batches faster

Dwedit · Jul 11, 2015 · 763c47c · 763c47c
1 parent 21c355b
commit 763c47c
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/BizHawk.Client.DiscoHawk/DiscoHawk.cs b/BizHawk.Client.DiscoHawk/DiscoHawk.cs
@@ -186,7 +186,7 @@ public struct CHANGEFILTERSTRUCT
 
 	class DiscoHawk
 	{
-		static List<string> FindExtensionsRecurse(string dir, string extUppercaseWithDot)
+		static List<string> FindCuesRecurse(string dir)
 		{
 			List<string> ret = new List<string>();
 			Queue<string> dpTodo = new Queue<string>();
@@ -197,12 +197,10 @@ static List<string> FindExtensionsRecurse(string dir, string extUppercaseWithDot
 				if (dpTodo.Count == 0)
 					break;
 				dpCurr = dpTodo.Dequeue();
-				Parallel.ForEach(new DirectoryInfo(dpCurr).GetFiles(), (fi) =>
+				foreach(var fi in new DirectoryInfo(dpCurr).GetFiles("*.cue"))
 				{
-					if (fi.Extension.ToUpperInvariant() == extUppercaseWithDot)
-						lock(ret)
-							ret.Add(fi.FullName);
-				});
+					ret.Add(fi.FullName);
+				}
 				Parallel.ForEach(new DirectoryInfo(dpCurr).GetDirectories(), (di) =>
 				{
 					lock(dpTodo)
@@ -264,7 +262,7 @@ public void Run(string[] args)
 			if (scanCues)
 			{
 				verbose = false;
-				var todo = FindExtensionsRecurse(dirArg, ".CUE");
+				var todo = FindCuesRecurse(dirArg);
 				var po = new ParallelOptions();
 				var cts = new CancellationTokenSource();
 				po.CancellationToken = cts.Token;