TasMovie - save and load lag log to file

BizHawk.Client.Common/BinarySaveStates.cs

@@ -18,7 +18,11 @@ namespace BizHawk.Client.Common
 		Movieheader,
 		Comments,
 		Subtitles,
-		SyncSettings
+		SyncSettings,
+
+		// TasMovie
+		LagLog,
+		Greenzone
 	}
 
 	public class BinaryStateFileNames
@@ -48,6 +52,10 @@ namespace BizHawk.Client.Common
 			LumpNames[BinaryStateLump.Comments] = "Comments";
 			LumpNames[BinaryStateLump.Subtitles] = "Subtitles";
 			LumpNames[BinaryStateLump.SyncSettings] = "SyncSettings";
+
+			// TasMovie
+			LumpNames[BinaryStateLump.LagLog] = "LagLog";
+			LumpNames[BinaryStateLump.Greenzone] = "GreenZone";
 		}
 
 		public static string Get(BinaryStateLump lump)

BizHawk.Client.Common/movie/tasproj/TasMovie.IO.cs

@@ -1,7 +1,10 @@
 using System;
 using System.IO;
+using System.Linq;
 
 using BizHawk.Common;
+using BizHawk.Common.CollectionExtensions;
+using BizHawk.Common.IOExtensions;
 
 namespace BizHawk.Client.Common
 {
@@ -25,6 +28,8 @@ namespace BizHawk.Client.Common
 
 				bs.PutLump(BinaryStateLump.Input, tw => tw.WriteLine(RawInputLog()));
 
+				bs.PutLump(BinaryStateLump.LagLog, (BinaryWriter bw) => bw.Write(LagLog.ToByteArray()));
+
 				if (StartsFromSavestate)
 				{
 					if (TextSavestate != null)
@@ -118,6 +123,12 @@ namespace BizHawk.Client.Common
 					ExtractInputLog(tr, out errorMessage);
 				});
 
+				bl.GetLump(BinaryStateLump.LagLog, true, delegate(BinaryReader br)
+				{
+					var bytes = br.BaseStream.ReadAllBytes();
+					LagLog = bytes.ToBools().ToList();
+				});
+
 				if (StartsFromSavestate)
 				{
 					bl.GetCoreState(
@@ -136,7 +147,13 @@ namespace BizHawk.Client.Common
 
 		private void ClearTasprojExtrasBeforeLoad()
 		{
-			// TODO
+			LagLog.Clear();
+			StateManager.Clear();
+		}
+
+		private void RestoreLagLog(byte[] buffer)
+		{
+
 		}
 	}
 }

BizHawk.Client.Common/movie/tasproj/TasMovie.cs

@@ -11,7 +11,7 @@ namespace BizHawk.Client.Common
 {
 	public sealed partial class TasMovie : Bk2Movie
 	{
-		private readonly List<bool> LagLog = new List<bool>();
+		private List<bool> LagLog = new List<bool>();
 		private readonly TasStateManager StateManager = new TasStateManager();
 
 		public TasMovie(string path) : base(path) { }
@@ -37,7 +37,7 @@ namespace BizHawk.Client.Common
 				{
 					State = StateManager[index],
 					LogEntry = GetInput(index),
-					Lagged = LagLog[index]
+					Lagged = (index < LagLog.Count) ? LagLog[index] : false
 				};
 			}
 		}

BizHawk.Client.Common/movie/tasproj/TasStateManager.cs

@@ -63,5 +63,13 @@ namespace BizHawk.Client.Common
 				States.Remove(f);
 			}
 		}
+
+		/// <summary>
+		/// Clears all state information
+		/// </summary>
+		public void Clear()
+		{
+			States.Clear();
+		}
 	}
 }

BizHawk.Common/Extensions/CollectionExtensions.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Collections;
 using System.Collections.Generic;
 using System.Linq;
 
@@ -94,5 +95,27 @@ namespace BizHawk.Common.CollectionExtensions
 
 			throw new InvalidOperationException("Item not found");
 		}
+
+		public static byte[] ToByteArray(this IEnumerable<bool> list)
+		{
+			var bits = new BitArray(list.ToArray());
+			byte [] bytes = new byte[bits.Length / 8 + ( bits.Length % 8 == 0 ? 0 : 1 )];
+			bits.CopyTo(bytes, 0);
+			return bytes;
+		}
+
+		/// <summary>
+		/// Converts any byte array into a bit array represented as a list of bools
+		/// </summary>
+		public static IEnumerable<bool> ToBools(this byte[] bytes)
+		{
+			var bits = new BitArray(bytes);
+			var bools = new bool[bits.Length];
+			bits.CopyTo(bools, 0);
+
+			return bools;
+		}
+
+		
 	}
 }