109 lines
2.8 KiB
C#
109 lines
2.8 KiB
C#
using UnityEngine;
|
|
using UnityEngine.Rendering;
|
|
|
|
#if UNITY_EDITOR
|
|
using UnityEditor;
|
|
using NWH.NUI;
|
|
#endif
|
|
|
|
namespace NWH.Common.Vehicles
|
|
{
|
|
/// <summary>
|
|
/// Manages reflection probe for the vehicle and switches between probe types as needed.
|
|
/// </summary>
|
|
[RequireComponent(typeof(ReflectionProbe))]
|
|
[DefaultExecutionOrder(19)]
|
|
public partial class VehicleReflectionProbe : MonoBehaviour
|
|
{
|
|
public enum ProbeType
|
|
{
|
|
Baked,
|
|
Realtime,
|
|
}
|
|
|
|
public ProbeType awakeProbeType = ProbeType.Realtime;
|
|
public ProbeType asleepProbeType = ProbeType.Baked;
|
|
public bool bakeOnStart = true;
|
|
public bool bakeOnSleep = true;
|
|
|
|
private ReflectionProbe _reflectionProbe;
|
|
private Vehicle _vc;
|
|
|
|
|
|
private void OnEnable()
|
|
{
|
|
_vc = GetComponentInParent<Vehicle>();
|
|
if (_vc == null)
|
|
{
|
|
Debug.LogError("VehicleController not found.");
|
|
}
|
|
|
|
_reflectionProbe = GetComponent<ReflectionProbe>();
|
|
_vc.onEnable.AddListener(OnVehicleEnable);
|
|
_vc.onDisable.AddListener(OnVehicleDisable);
|
|
|
|
_reflectionProbe.refreshMode = ReflectionProbeRefreshMode.EveryFrame;
|
|
|
|
if (bakeOnStart)
|
|
{
|
|
_reflectionProbe.RenderProbe();
|
|
}
|
|
}
|
|
|
|
|
|
private void OnVehicleEnable()
|
|
{
|
|
_reflectionProbe.mode = awakeProbeType == ProbeType.Baked
|
|
? _reflectionProbe.mode = ReflectionProbeMode.Baked
|
|
: ReflectionProbeMode.Realtime;
|
|
}
|
|
|
|
|
|
private void OnVehicleDisable()
|
|
{
|
|
_reflectionProbe.mode = asleepProbeType == ProbeType.Baked
|
|
? _reflectionProbe.mode = ReflectionProbeMode.Baked
|
|
: ReflectionProbeMode.Realtime;
|
|
|
|
if (bakeOnSleep && _reflectionProbe.isActiveAndEnabled)
|
|
{
|
|
_reflectionProbe.RenderProbe();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
#if UNITY_EDITOR
|
|
namespace NWH.Common.Vehicles
|
|
{
|
|
[CustomEditor(typeof(VehicleReflectionProbe))]
|
|
[CanEditMultipleObjects]
|
|
public partial class VehicleReflectionProbeEditor : NUIEditor
|
|
{
|
|
public override bool OnInspectorNUI()
|
|
{
|
|
if (!base.OnInspectorNUI())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
drawer.Field("awakeProbeType");
|
|
drawer.Field("asleepProbeType");
|
|
drawer.Field("bakeOnStart");
|
|
drawer.Field("bakeOnSleep");
|
|
|
|
drawer.EndEditor(this);
|
|
return true;
|
|
}
|
|
|
|
|
|
public override bool UseDefaultMargins()
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
#endif
|