From 9f67d8e59b66be55e34a528ed6a05aaefcf41bfb Mon Sep 17 00:00:00 2001
From: adelikat <adelikat@tasvideos.org>
Date: Sun, 4 Oct 2015 10:43:36 -0400
Subject: [PATCH] TasBranchCollection - encapsulate unique identifier logic
 into TasBranchCollection

---
 .../movie/tasproj/TasBranch.cs                | 22 ++++++++++++-------
 .../movie/tasproj/TasMovie.cs                 |  4 ++--
 .../movie/tasproj/TasStateManager.cs          |  2 +-
 3 files changed, 17 insertions(+), 11 deletions(-)

diff --git a/BizHawk.Client.Common/movie/tasproj/TasBranch.cs b/BizHawk.Client.Common/movie/tasproj/TasBranch.cs
index fcb03c296b..f5a42191e6 100644
--- a/BizHawk.Client.Common/movie/tasproj/TasBranch.cs
+++ b/BizHawk.Client.Common/movie/tasproj/TasBranch.cs
@@ -2,6 +2,7 @@
 using System.Collections;
 using System.Collections.Generic;
 using System.IO;
+using System.Linq;
 using Newtonsoft.Json;
 using BizHawk.Bizware.BizwareGL;
 
@@ -9,14 +10,6 @@ namespace BizHawk.Client.Common
 {
 	public class TasBranch
 	{
-		public TasBranch()
-		{
-			do
-			{
-				UniqueIdentifier = Guid.NewGuid();
-			} while (TasMovie.BranchIndexByHash(UniqueIdentifier.GetHashCode()) != -1);
-		}
-
 		public int Frame { get; set; }
 		public byte[] CoreData { get; set; }
 		public List<string> InputLog { get; set; }
@@ -30,6 +23,19 @@ namespace BizHawk.Client.Common
 
 	public class TasBranchCollection : List<TasBranch>
 	{
+		public new void Add(TasBranch item)
+		{
+			var currentHashes = this.Select(b => b.UniqueIdentifier.GetHashCode()).ToList();
+
+			// TODO: loop until this is unique
+			if (currentHashes.Contains(item.UniqueIdentifier.GetHashCode()))
+			{
+				item.UniqueIdentifier = Guid.NewGuid();
+			}
+
+			base.Add(item);
+		}
+
 		public void Save(BinaryStateSaver bs)
 		{
 			var nheader = new IndexedStateLump(BinaryStateLump.BranchHeader);
diff --git a/BizHawk.Client.Common/movie/tasproj/TasMovie.cs b/BizHawk.Client.Common/movie/tasproj/TasMovie.cs
index b50eec82e2..a57ab51cd4 100644
--- a/BizHawk.Client.Common/movie/tasproj/TasMovie.cs
+++ b/BizHawk.Client.Common/movie/tasproj/TasMovie.cs
@@ -22,7 +22,7 @@ namespace BizHawk.Client.Common
 		private readonly Dictionary<int, IController> InputStateCache = new Dictionary<int, IController>();
 		public readonly List<string> VerificationLog = new List<string>(); // For movies that do not begin with power-on, this is the input required to get into the initial state
 
-		public static readonly TasBranchCollection Branches = new TasBranchCollection();
+		public readonly TasBranchCollection Branches = new TasBranchCollection();
 
 		private BackgroundWorker _progressReportWorker = null;
 		public void NewBGWorker(BackgroundWorker newWorker)
@@ -85,7 +85,7 @@ namespace BizHawk.Client.Common
 		public TasBranch GetBranch(int index) { return Branches[index]; }
 		public int BranchHashByIndex(int index) { return Branches[index].UniqueIdentifier.GetHashCode(); }
 
-		public static int BranchIndexByHash(int hash)
+		public int BranchIndexByHash(int hash)
 		{
 			TasBranch branch = Branches.Where(b => b.UniqueIdentifier.GetHashCode() == hash).SingleOrDefault();
 			if (branch == null)
diff --git a/BizHawk.Client.Common/movie/tasproj/TasStateManager.cs b/BizHawk.Client.Common/movie/tasproj/TasStateManager.cs
index d0a804b4cf..e40785fc73 100644
--- a/BizHawk.Client.Common/movie/tasproj/TasStateManager.cs
+++ b/BizHawk.Client.Common/movie/tasproj/TasStateManager.cs
@@ -710,7 +710,7 @@ namespace BizHawk.Client.Common
 			SortedList<int, StateManagerState> stateList = BranchStates[frame];
 			for (int i = 0; i < _movie.BranchCount; i++)
 			{
-				if (i == TasMovie.BranchIndexByHash(branchHash))
+				if (i == _movie.BranchIndexByHash(branchHash))
 					continue;
 
 				if (stateList != null && stateList.ContainsKey(i) && stateList[i] == stateToMatch)