MBHaxe/src/net/MarblePredictionStore.hx

package net;

import net.NetPacket.MarbleUpdatePacket;
import net.NetPacket.MarbleMovePacket;
import src.TimeState;
import src.Marble;
import h3d.Vector;

@:publicFields
class MarblePrediction {
	var tick:Int;
	var position:Vector;
	var velocity:Vector;
	var omega:Vector;

	public function new(marble:Marble, tick:Int) {
		this.tick = tick;
		position = @:privateAccess marble.newPos.clone();
		velocity = @:privateAccess marble.velocity.clone();
		omega = @:privateAccess marble.omega.clone();
	}

	public inline function getError(p:MarbleUpdatePacket) {
		var subs = position.sub(p.position).lengthSq() + velocity.sub(p.velocity).lengthSq() + omega.sub(p.omega).lengthSq();
		return subs;
	}
}

class MarblePredictionStore {
	var predictions:Map<Marble, Array<MarblePrediction>>;

	public function new() {
		predictions = [];
	}

	public function storeState(marble:Marble, tick:Int) {
		var state = new MarblePrediction(marble, tick);
		if (predictions.exists(marble)) {
			var arr = predictions[marble];
			while (arr.length != 0 && arr[0].tick >= tick)
				arr.shift();
			arr.push(state);
		} else {
			predictions.set(marble, [state]);
		}
	}

	public function retrieveState(marble:Marble, tick:Int) {
		if (predictions.exists(marble)) {
			var arr = predictions[marble];
			while (arr.length != 0 && arr[0].tick < tick)
				arr.shift();
			if (arr.length == 0)
				return null;
			var p = arr[0];
			if (p.tick == tick)
				return p;
			return null;
		}
		return null;
	}
}