From fcb5af62d61fd32b88db8d443367c1b73b881955 Mon Sep 17 00:00:00 2001
From: goyuken <goyuken@4bbf4f03-6f44-549e-7e59-35d4b0962838>
Date: Mon, 18 May 2015 01:30:30 +0000
Subject: [PATCH] speed up apple2 savestates, round 1

---
 .../Computers/AppleII/AppleII.IStatable.cs    | 96 ++++++++++++-------
 .../Computers/AppleII/AppleII.cs              |  1 +
 2 files changed, 63 insertions(+), 34 deletions(-)

diff --git a/BizHawk.Emulation.Cores/Computers/AppleII/AppleII.IStatable.cs b/BizHawk.Emulation.Cores/Computers/AppleII/AppleII.IStatable.cs
index 646658ac37..f0d5ec93dd 100644
--- a/BizHawk.Emulation.Cores/Computers/AppleII/AppleII.IStatable.cs
+++ b/BizHawk.Emulation.Cores/Computers/AppleII/AppleII.IStatable.cs
@@ -3,62 +3,90 @@ using System.IO;
 using System;
 using Newtonsoft.Json;
 using Newtonsoft.Json.Linq;
+using Jellyfish.Virtu;
 
 namespace BizHawk.Emulation.Cores.Computers.AppleII
 {
 	public partial class AppleII : IStatable
 	{
+		private class CoreConverter : JsonConverter
+		{
+			public override bool CanConvert(Type objectType)
+			{
+				return objectType == typeof(Machine);
+			}
+
+			public override bool CanRead { get { return true; } }
+			public override bool CanWrite { get { return false; } }
+
+			public override object ReadJson(JsonReader reader, Type objectType, object existingValue, JsonSerializer serializer)
+			{
+				// uses its own serialization context: intentional
+				return Machine.Deserialize(reader);
+			}
+
+			public override void WriteJson(JsonWriter writer, object value, JsonSerializer serializer)
+			{
+				throw new NotImplementedException();
+			}
+		}
+
 		public bool BinarySaveStatesPreferred { get { return false; } }
 
+		private void SerializeEverything(JsonWriter w)
+		{
+			// this is much faster than other possibilities for serialization
+			w.WriteStartObject();
+			w.WritePropertyName("Frame");
+			w.WriteValue(Frame);
+			w.WritePropertyName("LagCount");
+			w.WriteValue(LagCount);
+			w.WritePropertyName("IsLagFrame");
+			w.WriteValue(IsLagFrame);
+			w.WritePropertyName("CurrentDisk");
+			w.WriteValue(CurrentDisk);
+			w.WritePropertyName("Core");
+			_machine.Serialize(w);
+			w.WriteEndObject();
+		}
+
+		private void DeserializeEverything(JsonReader r)
+		{
+			var o = (OtherData)ser.Deserialize(r, typeof(OtherData));
+			Frame = o.Frame;
+			LagCount = o.LagCount;
+			IsLagFrame = o.IsLagFrame;
+			CurrentDisk = o.CurrentDisk;
+			_machine = o.Core;
+
+			// should not be needed.
+			// InitDisk();
+		}
+
 		private class OtherData
 		{
 			public int Frame;
 			public int LagCount;
 			public bool IsLagFrame;
 			public int CurrentDisk;
-			public JObject Core;
+			public Machine Core;
 		}
+
+		private void InitSaveStates()
+		{
+			ser.Converters.Add(new CoreConverter());
+		}
+
 		private JsonSerializer ser = new JsonSerializer();
 
-		[FeatureNotImplemented]
 		public void SaveStateText(TextWriter writer)
 		{
-			var w = new JTokenWriter();
-			_machine.Serialize(w);
-
-			var o = new OtherData
-			{
-				Frame = Frame,
-				LagCount = LagCount,
-				IsLagFrame = IsLagFrame,
-				CurrentDisk = CurrentDisk,
-				Core = (JObject)w.Token,
-			};
-
-			var jw = new JsonTextWriter(writer) { Formatting = Newtonsoft.Json.Formatting.Indented };
-			ser.Serialize(jw, o);
+			SerializeEverything(new JsonTextWriter(writer) { Formatting = Formatting.None });
 		}
 
 		public void LoadStateText(TextReader reader)
 		{
-			var o = (OtherData)ser.Deserialize(reader, typeof(OtherData));
-			Frame = o.Frame;
-			LagCount = o.LagCount;
-			IsLagFrame = o.IsLagFrame;
-			CurrentDisk = o.CurrentDisk;
-
-			var r = new JTokenReader(o.Core);
-			try
-			{
-				_machine = Jellyfish.Virtu.Machine.Deserialize(r);
-			}
-			catch (Exception e)
-			{
-				Console.WriteLine(e.ToString());
-				throw;
-			}
-			// should not be needed.
-			// InitDisk();
+			DeserializeEverything(new JsonTextReader(reader));
 		}
 
 		public void SaveStateBinary(BinaryWriter writer)
diff --git a/BizHawk.Emulation.Cores/Computers/AppleII/AppleII.cs b/BizHawk.Emulation.Cores/Computers/AppleII/AppleII.cs
index cd28019296..1c48ffba86 100644
--- a/BizHawk.Emulation.Cores/Computers/AppleII/AppleII.cs
+++ b/BizHawk.Emulation.Cores/Computers/AppleII/AppleII.cs
@@ -48,6 +48,7 @@ namespace BizHawk.Emulation.Cores.Computers.AppleII
 			//for junk.dsk the .dsk is important because it determines the format from that
 			InitDisk();
 
+			InitSaveStates();
 			SetupMemoryDomains();
 		}