rewind optimize memory

src/rewind/RewindFrame.hx

@@ -1,5 +1,9 @@
 package rewind;
 
+import haxe.io.BytesOutput;
+import haxe.io.Bytes;
+import haxe.io.BytesBuffer;
+import dif.io.BytesWriter;
 import h3d.Matrix;
 import mis.MissionElement.MissionElementBase;
 import triggers.CheckpointTrigger;
@@ -119,4 +123,284 @@ class RewindFrame {
 		c.modeState = modeState != null ? modeState.clone() : null;
 		return c;
 	}
+
+	public function serialize(rm:RewindManager) {
+		var bb = new BytesOutput();
+		var framesize = 0;
+		framesize += 32; // timeState
+		framesize += 24; // marblePosition
+		framesize += 32; // marbleOrientation
+		framesize += 24; // marbleVelocity
+		framesize += 24; // marbleAngularVelocity
+		framesize += 2; // marblePowerup
+		framesize += 8; // bonusTime
+		framesize += 2; // gemCount
+		framesize += 2 + gemStates.length * 1; // gemStates
+		framesize += 2 + powerupStates.length * 8; // powerupStates
+		framesize += 2 + landMineStates.length * 8; // landMineStates
+		framesize += 32; // activePowerupStates
+		framesize += 24; // currentUp
+		framesize += 24; // lastContactNormal
+		framesize += 2; // mpStates.length
+		for (s in mpStates) {
+			framesize += 8; // s.curState.currentTime
+			framesize += 8; // s.curState.targetTime
+			framesize += 24; // s.curState.velocity
+			framesize += 1; // s.stopped
+			framesize += 24; // s.position
+		}
+		framesize += 2; // trapdoorStates.length
+		for (s in trapdoorStates) {
+			framesize += 8; // s.lastContactTime
+			framesize += 1; // s.lastDirection
+			framesize += 8; // s.lastCompletion
+		}
+		framesize += 8; // blastAmt
+		if (oobState.oob)
+			framesize += 1; // oobState.oob
+		framesize += 32; // oobState.timeState
+		framesize += 1; // Null<checkpointState>
+		if (checkpointState != null) {
+			framesize += 4; // checkpointState.currentCheckpoint
+		}
+		framesize += 2; // checkpointState.currentCheckpointTrigger
+		framesize += 2; // checkpointState.checkpointCollectedGems.length
+		for (gem in checkpointState.checkpointCollectedGems.keys()) {
+			framesize += 2; // gem
+			framesize += 1; // checkpointState.checkpointCollectedGems[gem]
+		}
+		framesize += 2; // checkpointState.checkpointHeldPowerup
+		framesize += 1; // Null<checkpointState.checkpointUp>
+		if (checkpointState.checkpointUp != null)
+			framesize += 24; // checkpointState.checkpointUp
+		framesize += 8; // checkpointState.checkpointBlast
+		bb.prepare(framesize);
+		// Now actually write
+		bb.writeDouble(timeState.currentAttemptTime);
+		bb.writeDouble(timeState.timeSinceLoad);
+		bb.writeDouble(timeState.gameplayClock);
+		bb.writeDouble(timeState.dt);
+		bb.writeDouble(marblePosition.x);
+		bb.writeDouble(marblePosition.y);
+		bb.writeDouble(marblePosition.z);
+		bb.writeDouble(marbleOrientation.x);
+		bb.writeDouble(marbleOrientation.y);
+		bb.writeDouble(marbleOrientation.z);
+		bb.writeDouble(marbleOrientation.w);
+		bb.writeDouble(marbleVelocity.x);
+		bb.writeDouble(marbleVelocity.y);
+		bb.writeDouble(marbleVelocity.z);
+		bb.writeDouble(marbleAngularVelocity.x);
+		bb.writeDouble(marbleAngularVelocity.y);
+		bb.writeDouble(marbleAngularVelocity.z);
+		bb.writeInt16(rm.allocGO(marblePowerup));
+		bb.writeDouble(bonusTime);
+		bb.writeInt16(gemCount);
+		bb.writeInt16(gemStates.length);
+		for (s in gemStates) {
+			bb.writeByte(s ? 1 : 0);
+		}
+		bb.writeInt16(powerupStates.length);
+		for (s in powerupStates) {
+			bb.writeDouble(s);
+		}
+		bb.writeInt16(landMineStates.length);
+		for (s in landMineStates) {
+			bb.writeDouble(s);
+		}
+		bb.writeDouble(activePowerupStates[0]);
+		bb.writeDouble(activePowerupStates[1]);
+		bb.writeDouble(activePowerupStates[2]);
+		bb.writeDouble(activePowerupStates[3]);
+		bb.writeDouble(currentUp.x);
+		bb.writeDouble(currentUp.y);
+		bb.writeDouble(currentUp.z);
+		bb.writeDouble(lastContactNormal.x);
+		bb.writeDouble(lastContactNormal.y);
+		bb.writeDouble(lastContactNormal.z);
+		bb.writeInt16(mpStates.length);
+		for (s in mpStates) {
+			bb.writeDouble(s.curState.currentTime);
+			bb.writeDouble(s.curState.targetTime);
+			bb.writeDouble(s.curState.velocity.x);
+			bb.writeDouble(s.curState.velocity.y);
+			bb.writeDouble(s.curState.velocity.z);
+			bb.writeByte(s.stopped ? 1 : 0);
+			bb.writeDouble(s.position.x);
+			bb.writeDouble(s.position.y);
+			bb.writeDouble(s.position.z);
+		}
+		bb.writeInt16(trapdoorStates.length);
+		for (s in trapdoorStates) {
+			bb.writeDouble(s.lastContactTime);
+			bb.writeByte(s.lastDirection);
+			bb.writeDouble(s.lastCompletion);
+		}
+		bb.writeDouble(blastAmt);
+		bb.writeByte(oobState.oob ? 1 : 0);
+		if (oobState.oob) {
+			bb.writeDouble(oobState.timeState.currentAttemptTime);
+			bb.writeDouble(oobState.timeState.timeSinceLoad);
+			bb.writeDouble(oobState.timeState.gameplayClock);
+			bb.writeDouble(oobState.timeState.dt);
+		}
+		bb.writeByte(checkpointState.currentCheckpoint == null ? 0 : 1);
+		if (checkpointState.currentCheckpoint != null) {
+			bb.writeInt16(rm.allocGO(checkpointState.currentCheckpoint.obj));
+			bb.writeInt16(rm.allocME(checkpointState.currentCheckpoint.elem));
+		}
+		bb.writeInt16(rm.allocGO(checkpointState.currentCheckpointTrigger));
+		var chkgemcount = 0;
+		for (g in checkpointState.checkpointCollectedGems) {
+			chkgemcount++;
+		}
+		bb.writeInt16(chkgemcount);
+		for (gem in checkpointState.checkpointCollectedGems.keys()) {
+			bb.writeInt16(rm.allocGO(gem));
+			bb.writeByte(checkpointState.checkpointCollectedGems[gem] ? 1 : 0);
+		}
+		bb.writeInt16(rm.allocGO(checkpointState.checkpointHeldPowerup));
+		bb.writeByte(checkpointState.checkpointUp == null ? 0 : 1);
+		if (checkpointState.checkpointUp != null) {
+			bb.writeDouble(checkpointState.checkpointUp.x);
+			bb.writeDouble(checkpointState.checkpointUp.y);
+			bb.writeDouble(checkpointState.checkpointUp.z);
+		}
+		bb.writeDouble(checkpointState.checkpointBlast);
+		return bb.getBytes();
+	}
+
+	public function deserialize(rm:RewindManager, br:haxe.io.BytesInput) {
+		marblePosition = new Vector();
+		marbleOrientation = new Quat();
+		marbleVelocity = new Vector();
+		marbleAngularVelocity = new Vector();
+		currentUp = new Vector();
+		lastContactNormal = new Vector();
+		timeState = new TimeState();
+		timeState.currentAttemptTime = br.readDouble();
+		timeState.timeSinceLoad = br.readDouble();
+		timeState.gameplayClock = br.readDouble();
+		timeState.dt = br.readDouble();
+		marblePosition.x = br.readDouble();
+		marblePosition.y = br.readDouble();
+		marblePosition.z = br.readDouble();
+		marbleOrientation.x = br.readDouble();
+		marbleOrientation.y = br.readDouble();
+		marbleOrientation.z = br.readDouble();
+		marbleOrientation.w = br.readDouble();
+		marbleVelocity.x = br.readDouble();
+		marbleVelocity.y = br.readDouble();
+		marbleVelocity.z = br.readDouble();
+		marbleAngularVelocity.x = br.readDouble();
+		marbleAngularVelocity.y = br.readDouble();
+		marbleAngularVelocity.z = br.readDouble();
+		marblePowerup = cast rm.getGO(br.readInt16());
+		bonusTime = br.readDouble();
+		gemCount = br.readInt16();
+		gemStates = [];
+		var gemStates_len = br.readInt16();
+		for (i in 0...gemStates_len) {
+			gemStates.push(br.readByte() != 0);
+		}
+		powerupStates = [];
+		var powerupStates_len = br.readInt16();
+		for (i in 0...powerupStates_len) {
+			powerupStates.push(br.readDouble());
+		}
+		landMineStates = [];
+		var landMineStates_len = br.readInt16();
+		for (i in 0...landMineStates_len) {
+			landMineStates.push(br.readDouble());
+		}
+		activePowerupStates = [];
+		activePowerupStates.push(br.readDouble());
+		activePowerupStates.push(br.readDouble());
+		activePowerupStates.push(br.readDouble());
+		activePowerupStates.push(br.readDouble());
+		currentUp.x = br.readDouble();
+		currentUp.y = br.readDouble();
+		currentUp.z = br.readDouble();
+		lastContactNormal.x = br.readDouble();
+		lastContactNormal.y = br.readDouble();
+		lastContactNormal.z = br.readDouble();
+		mpStates = [];
+		var mpStates_len = br.readInt16();
+		for (i in 0...mpStates_len) {
+			var mpStates_item = {
+				curState: {
+					currentTime: 0.0,
+					targetTime: 0.0,
+					velocity: new Vector(),
+				},
+				stopped: false,
+				position: new Vector()
+			};
+			mpStates_item.curState.currentTime = br.readDouble();
+			mpStates_item.curState.targetTime = br.readDouble();
+			mpStates_item.curState.velocity.x = br.readDouble();
+			mpStates_item.curState.velocity.y = br.readDouble();
+			mpStates_item.curState.velocity.z = br.readDouble();
+			mpStates_item.stopped = br.readByte() != 0;
+			mpStates_item.position.x = br.readDouble();
+			mpStates_item.position.y = br.readDouble();
+			mpStates_item.position.z = br.readDouble();
+			mpStates.push(mpStates_item);
+		}
+		trapdoorStates = [];
+		var trapdoorStates_len = br.readInt16();
+		for (i in 0...trapdoorStates_len) {
+			var trapdoorStates_item = {
+				lastContactTime: 0.0,
+				lastDirection: 0,
+				lastCompletion: 0.0
+			};
+			trapdoorStates_item.lastContactTime = br.readDouble();
+			trapdoorStates_item.lastDirection = br.readByte();
+			trapdoorStates_item.lastCompletion = br.readDouble();
+			trapdoorStates.push(trapdoorStates_item);
+		}
+		blastAmt = br.readDouble();
+		oobState = {
+			oob: br.readByte() != 0,
+			timeState: null
+		};
+		if (oobState.oob) {
+			oobState.timeState = new TimeState();
+			oobState.timeState.currentAttemptTime = br.readDouble();
+			oobState.timeState.timeSinceLoad = br.readDouble();
+			oobState.timeState.gameplayClock = br.readDouble();
+			oobState.timeState.dt = br.readDouble();
+		}
+		var hasCheckpoint = br.readByte() != 0;
+		checkpointState = {
+			currentCheckpoint: null,
+			currentCheckpointTrigger: null,
+			checkpointCollectedGems: new Map<Gem, Bool>(),
+			checkpointHeldPowerup: null,
+			checkpointUp: null,
+			checkpointBlast: 0.0,
+		};
+		if (hasCheckpoint) {
+			var co = rm.getGO(br.readInt16());
+			var ce = rm.getME(br.readInt16());
+			checkpointState.currentCheckpoint = {obj: cast co, elem: ce};
+		}
+		checkpointState.currentCheckpointTrigger = cast rm.getGO(br.readInt16());
+		var checkpointState_checkpointCollectedGems_len = br.readInt16();
+		for (i in 0...checkpointState_checkpointCollectedGems_len) {
+			var gem = cast rm.getGO(br.readInt16());
+			var c = br.readByte() != 0;
+			checkpointState.checkpointCollectedGems.set(cast gem, c);
+		}
+		checkpointState.checkpointHeldPowerup = cast rm.getGO(br.readInt16());
+		var checkpointState_checkpointUp_has = br.readByte() != 0;
+		if (checkpointState_checkpointUp_has) {
+			checkpointState.checkpointUp = new Vector();
+			checkpointState.checkpointUp.x = br.readDouble();
+			checkpointState.checkpointUp.y = br.readDouble();
+			checkpointState.checkpointUp.z = br.readDouble();
+		}
+		checkpointState.checkpointBlast = br.readDouble();
+	}
 }

src/rewind/RewindManager.hx

@@ -1,5 +1,9 @@
 package rewind;
 
+import haxe.io.BytesInput;
+import haxe.io.BytesBuffer;
+import mis.MissionElement.MissionElementBase;
+import src.GameObject;
 import rewind.RewindFrame.RewindMPState;
 import shapes.AbstractBumper;
 import shapes.PowerUp;
@@ -9,8 +13,17 @@ import src.Util;
 import src.Settings;
 
 class RewindManager {
-	var frames:Array<RewindFrame> = [];
+	var frameData:BytesBuffer;
+	var frameElapsedTimes:Array<Float> = [];
+	var frameDataOffsets:Array<Int> = [];
+	var frameSizes:Array<Int> = [];
+	var allocObjMap:Map<GameObject, Int> = [];
+	var allocObjs:Array<GameObject> = [];
+	var allocMeMap:Map<MissionElementBase, Int> = [];
+	var allocMes:Array<MissionElementBase> = [];
 	var level:MarbleWorld;
+	var allocId = 0;
+	var allocMeId = 0;
 
 	public var timeScale:Float = 1;
 
@@ -20,6 +33,7 @@ class RewindManager {
 	public function new(level:MarbleWorld) {
 		this.level = level;
 		this.timeScale = Settings.optionsSettings.rewindTimescale;
+		this.frameData = new BytesBuffer();
 	}
 
 	public function recordFrame() {
@@ -87,7 +101,12 @@ class RewindManager {
 				checkpointHeldPowerup: @:privateAccess level.checkpointHeldPowerup,
 			};
 			rf.modeState = level.gameMode.getRewindState();
-			frames.push(rf);
+			frameElapsedTimes.push(level.timeState.currentAttemptTime);
+			frameDataOffsets.push(frameData.length);
+			var frameDataSerialized = rf.serialize(this);
+			frameSizes.push(frameDataSerialized.length);
+			frameData.addBytes(frameDataSerialized, 0, frameDataSerialized.length);
+			// frames.push(rf);
 		}
 	}
 
@@ -204,24 +223,89 @@ class RewindManager {
 	}
 
 	public function getNextRewindFrame(absTime:Float):RewindFrame {
-		if (frames.length == 0)
+		if (frameElapsedTimes.length == 0)
 			return null;
 
-		var topFrame = frames[frames.length - 1];
+		var topFrame = frameElapsedTimes[frameElapsedTimes.length - 1];
-		while (topFrame.timeState.currentAttemptTime > absTime) {
+		while (topFrame > absTime) {
-			if (frames.length == 1) {
+			if (frameElapsedTimes.length == 1) {
-				return frames[0];
+				return getFrameAtIndex(0);
 			}
-			frames.pop();
+			popFrame();
-			if (frames.length == 0)
+			if (frameElapsedTimes.length == 0)
 				return null;
-			topFrame = frames[frames.length - 1];
+			topFrame = frameElapsedTimes[frameElapsedTimes.length - 1];
 		}
-		return topFrame;
+		return getFrameAtIndex(frameElapsedTimes.length - 1);
+		// return topFrame;
+	}
+
+	function getFrameAtIndex(index:Int) {
+		var offset = frameDataOffsets[index];
+		var size = frameSizes[index];
+		#if sys
+		var frameBytes = @:privateAccess frameData.b.sub(offset, size);
+		var bi = new BytesInput(frameBytes.toBytes(size));
+		#end
+		#if js
+		var frameBytes = @:privateAccess frameData.buffer.slice(offset, offset + size);
+		var bi = new BytesInput(haxe.io.Bytes.ofData(frameBytes));
+		#end
+		var fr = new RewindFrame();
+		fr.deserialize(this, bi);
+		return fr;
+		return null;
+	}
+
+	function popFrame() {
+		frameElapsedTimes.pop();
+		var offset = frameDataOffsets[frameDataOffsets.length - 1];
+		@:privateAccess frameData.pos = offset;
+		frameDataOffsets.pop();
+		frameSizes.pop();
+	}
+
+	public function allocGO(go:GameObject) {
+		if (go == null)
+			return -1;
+		if (allocObjMap.exists(go))
+			return allocObjMap.get(go);
+		var newId = allocId++;
+		allocObjMap.set(go, newId);
+		allocObjs.push(go);
+		return newId;
+	}
+
+	public function getGO(id:Int):GameObject {
+		return allocObjs[id];
+	}
+
+	public function allocME(me:MissionElementBase) {
+		if (me == null)
+			return -1;
+		if (allocMeMap.exists(me))
+			return allocMeMap.get(me);
+		var newId = allocMeId++;
+		allocMeMap.set(me, newId);
+		allocMes.push(me);
+		return newId;
+	}
+
+	public function getME(id:Int):MissionElementBase {
+		return allocMes[id];
 	}
 
 	public function clear() {
-		frames = [];
+		frameData = new BytesBuffer(); // clear
+		frameDataOffsets = [];
+		frameElapsedTimes = [];
+		frameSizes = [];
+		allocObjs = [];
+		allocObjMap = [];
+		allocMes = [];
+		allocMeMap = [];
+		allocId = 0;
+		allocMeId = 0;
 		timeAccumulator = 0.0;
 	}
 }