quantum-space-buddies/QSB/ShipSync/TransformSync/ShipTransformSync.cs

using QSB.Syncs.Sectored.Rigidbodies;
using QSB.Utility;
using UnityEngine;

namespace QSB.ShipSync.TransformSync;

public class ShipTransformSync : SectoredRigidbodySync
{
	public static ShipTransformSync LocalInstance { get; private set; }

	private float _lastSetPositionTime;
	private const float ForcePositionAfterTime = 1;

	protected override bool CheckReady() =>
		base.CheckReady() &&
		Locator.GetShipBody();

	public override void OnStartClient()
	{
		base.OnStartClient();
		LocalInstance = this;
	}

	protected override OWRigidbody InitAttachedRigidbody()
	{
		SectorDetector.Init(Locator.GetShipDetector().GetComponent<SectorDetector>());
		return Locator.GetShipBody();
	}

	/// Dont do base... this is a replacement!
	protected override void ApplyToAttached()
	{
		ApplyToSector();
		if (!ReferenceTransform)
		{
			return;
		}

		var targetPos = ReferenceTransform.FromRelPos(transform.position);

		if (Time.unscaledTime >= _lastSetPositionTime + ForcePositionAfterTime)
		{
			_lastSetPositionTime = Time.unscaledTime;

			var targetRot = ReferenceTransform.FromRelRot(transform.rotation);

			AttachedRigidbody.SetPosition(targetPos);
			AttachedRigidbody.SetRotation(targetRot);
		}

		var targetVelocity = ReferenceRigidbody.FromRelVel(Velocity, targetPos);
		var targetAngularVelocity = ReferenceRigidbody.FromRelAngVel(AngularVelocity);

		SetVelocity(AttachedRigidbody, targetVelocity);
		AttachedRigidbody.SetAngularVelocity(targetAngularVelocity);
	}

	/// use OWRigidbody version instead of ShipBody override
	private static void SetVelocity(OWRigidbody rigidbody, Vector3 newVelocity)
	{
		if (rigidbody.RunningKinematicSimulation())
		{
			rigidbody._kinematicRigidbody.velocity = newVelocity + Locator.GetCenterOfTheUniverse().GetStaticFrameVelocity_Internal();
		}
		else
		{
			rigidbody._rigidbody.velocity = newVelocity + Locator.GetCenterOfTheUniverse().GetStaticFrameVelocity_Internal();
		}

		rigidbody._lastVelocity = rigidbody._currentVelocity;
		rigidbody._currentVelocity = newVelocity;
	}

	protected override bool UseInterpolation => false;
}