AIStateController.cs

Go to the documentation of this file.

using UnityEngine;

namespace HKI.Core.AI.StateMachine
{
    public class AIStateController : MonoBehaviour
    {
        // Public variables
        [SerializeField] AIStateControllerContainer AIStateControllerContainer = null;
        [SerializeField] AIState StartState = null;
        [SerializeField] AIState CurrentState = null;
        [SerializeField] AIState RemainInState = null;

        [HideInInspector] public float StateTimeElapsed = 0.0f;

        [HideInInspector] public Transform ThisTransform = null;

        // Private variables
        bool isAiActive = false;
        
        // Awake function
        void Awake()
        {
            ThisTransform = transform;
        }

        // OnEnable
        void OnEnable()
        {
            if(AIStateControllerContainer != null)
                AIStateControllerContainer.Add(this);
        }

        // OnDisable
        void OnDisable()
        {
            if(AIStateControllerContainer != null)
                AIStateControllerContainer.Remove(this);
        }

        // Update function
        public void UpdateState()
        {
            if (!isAiActive || CurrentState == null)
                return;

            CurrentState.UpdateState(this);
        }

        // Interface functions
        public void SetupAI(bool isAiActive)
        {
            this.isAiActive = isAiActive;
            
            CurrentState = StartState;
        }

        public void TransitionToState(AIState nextState)
        {
            if (nextState != RemainInState)
            {
                CurrentState = nextState;
                OnExitState();
            }
        }

        public bool CheckIfCountDownElapsed(float duration)
        {
            StateTimeElapsed += Time.deltaTime;

            return (StateTimeElapsed >= duration);
        }

        // Helper function
        void OnExitState()
        {
            StateTimeElapsed = 0.0f;
        }
    }
}