quantum-space-buddies/QSB/Animation/Player/Thrusters/RemoteThrusterFlameController.cs

using QSB.Player;
using QSB.Utility;
using QSB.WorldSync;
using UnityEngine;

namespace QSB.Animation.Player.Thrusters;

[UsedInUnityProject]
internal class RemoteThrusterFlameController : MonoBehaviour
{
	[SerializeField]
	private Thruster _thruster;

	[SerializeField]
	private Light _light;

	[SerializeField]
	private AnimationCurve _scaleByThrust = AnimationCurve.Linear(0f, 0f, 1f, 1f);

	[SerializeField]
	private DampedSpring _scaleSpring = new();

	[SerializeField]
	private float _belowMaxThrustScalar = 1f;

	private MeshRenderer _thrusterRenderer;
	private Vector3 _thrusterFilter;
	private float _baseLightRadius;
	private float _currentScale;
	private RemotePlayerFluidDetector _fluidDetector;
	private bool _underwater;
	private bool _initialized;
	private PlayerInfo _attachedPlayer;

	// TODO : Make flames not appear underwater (Check original code!)

	public void Init(PlayerInfo player)
	{
		_attachedPlayer = player;

		_thrusterRenderer = GetComponent<MeshRenderer>();
		_thrusterFilter = OWUtilities.GetShipThrusterFilter(_thruster);
		_baseLightRadius = _light.range;
		_currentScale = 0f;
		_thrusterRenderer.enabled = false;
		_light.enabled = false;
		_light.shadows = LightShadows.Soft;
		_initialized = true;
		_underwater = false;
		_fluidDetector = player.FluidDetector;
		_fluidDetector.OnEnterFluidType += OnEnterExitFluidType;
		_fluidDetector.OnExitFluidType += OnEnterExitFluidType;
	}

	private void OnDestroy()
	{
		if (_fluidDetector != null)
		{
			_fluidDetector.OnEnterFluidType -= OnEnterExitFluidType;
			_fluidDetector.OnExitFluidType -= OnEnterExitFluidType;
		}
	}

	private void Update()
	{
		if (!_initialized)
		{
			return;
		}

		var num = _underwater
			? 0f
			: _scaleByThrust.Evaluate(GetThrustFraction());

		if (_belowMaxThrustScalar < 1f)
		{
			num *= _belowMaxThrustScalar;
		}

		_currentScale = _scaleSpring.Update(_currentScale, num, Time.deltaTime);
		if (_currentScale < 0f)
		{
			_currentScale = 0f;
			_scaleSpring.ResetVelocity();
		}

		if (_underwater && _currentScale <= 0.001f)
		{
			_currentScale = 0f;
			_scaleSpring.ResetVelocity();
		}

		transform.localScale = Vector3.one * _currentScale;
		_light.range = _baseLightRadius * _currentScale;
		_thrusterRenderer.enabled = _currentScale > 0f && _attachedPlayer.Visible;
		_light.enabled = _currentScale > 0f;
	}

	private void OnEnterExitFluidType(FluidVolume.Type type)
	{
		this._underwater = this._fluidDetector.InFluidType(FluidVolume.Type.WATER);
	}

	private float GetThrustFraction() => Vector3.Dot(_attachedPlayer.JetpackAcceleration.AccelerationVariableSyncer.Value, _thrusterFilter);

	private void OnRenderObject()
	{
		if (!QSBCore.DebugSettings.DrawLines || !QSBWorldSync.AllObjectsReady)
		{
			return;
		}

		Popcron.Gizmos.Sphere(_light.transform.position, 0.05f, Color.yellow, 4);
		Popcron.Gizmos.Line(_light.transform.position, _light.transform.parent.position, Color.yellow);
	}
}