quantum-space-buddies/QSB/Animation/Player/PlayerHeadRotationSync.cs

using OWML.Common;
using QSB.Utility;
using UnityEngine;

namespace QSB.Animation.Player;

public class PlayerHeadRotationSync : MonoBehaviour
{
	private Animator _attachedAnimator;
	private Transform _lookBase;
	private bool _isSetUp;

	public void Init(Transform lookBase)
	{
		_attachedAnimator = GetComponent<Animator>();
		_lookBase = lookBase;
		_isSetUp = true;
	}

	private void LateUpdate()
	{
		if (!_isSetUp)
		{
			return;
		}

		if (_attachedAnimator == null)
		{
			DebugLog.ToConsole($"Error - _attachedAnimator is null!", MessageType.Error);
			return;
		}

		if (_lookBase == null)
		{
			DebugLog.ToConsole($"Error - _lookBase is null!", MessageType.Error);
			return;
		}

		var bone = _attachedAnimator.GetBoneTransform(HumanBodyBones.Head);
		// Get the camera's local rotation with respect to the player body
		var lookLocalRotation = Quaternion.Inverse(_attachedAnimator.transform.rotation) * _lookBase.rotation;
		bone.localRotation = Quaternion.Euler(0f, 0f, lookLocalRotation.eulerAngles.x);
	}
}