siege-song/Data/Assets/Malbers Animations/Common/Scripts/Behaviors/LookAtBehaviour.cs

using UnityEngine;

#if UNITY_EDITOR
using UnityEditor;
#endif


namespace MalbersAnimations.Utilities
{
    /// <summary> Enable Disable the LookAt Logic by Priority </summary>
    public interface ILookAtActivation
    {
        void EnableByPriority(int layer);
        void DisableByPriority(int layer);
        void ResetByPriority(int layer);
    }

    public class LookAtBehaviour : StateMachineBehaviour
    {
        public enum LookAtState { DoNothing, Enable, Disable , Reset}
        public enum EnterExit  { OnEnter, OnExit}

        private ILookAtActivation lookat;
        public EnterExit when = EnterExit.OnEnter;
        public LookAtState OnEnter = LookAtState.Enable;

        public override void OnStateEnter(Animator animator, AnimatorStateInfo stateInfo, int layerIndex)
        {
            if (lookat == null) lookat = animator.FindInterface<ILookAtActivation>();

            if (when == EnterExit.OnEnter) CheckLookAt(animator, layerIndex);
        }

        public override void OnStateExit(Animator animator, AnimatorStateInfo stateInfo, int layerIndex)
        {
            if (when == EnterExit.OnExit) CheckLookAt(animator, layerIndex);
        }


        private void CheckLookAt(Animator animator, int layerIndex)
        {
            if (animator.GetCurrentAnimatorStateInfo(layerIndex).fullPathHash == animator.GetNextAnimatorStateInfo(layerIndex).fullPathHash) return;

            if (lookat != null)
            {
                switch (OnEnter)
                {
                    case LookAtState.Reset: lookat.ResetByPriority(layerIndex + 1); break;
                    case LookAtState.Enable: lookat.EnableByPriority(layerIndex + 1); break;
                    case LookAtState.Disable: lookat.DisableByPriority(layerIndex + 1); break;
                    default: break;
                }
            }
        }

    }

   


#if UNITY_EDITOR
    [CustomEditor(typeof(LookAtBehaviour))]
    public class LookAtBehaviourED : Editor
    {
        SerializedProperty OnEnter, stateInfo;
        void OnEnable()
        {
            OnEnter = serializedObject.FindProperty("OnEnter");
            stateInfo = serializedObject.FindProperty("when");
        }

        public override void OnInspectorGUI()
        {
            serializedObject.Update();
            MalbersEditor.DrawDescription("Enable/Disable the Look At logic by layer priority");
            EditorGUILayout.PropertyField(stateInfo);
            EditorGUILayout.PropertyField(OnEnter, new GUIContent("Status"));
            serializedObject.ApplyModifiedProperties();
        }
    }
#endif
}