public void ChangeState(AIStateId newState)
 {
     GetState(currentState)?.Exit(agent);
     currentState = newState;
     GetState(currentState)?.Enter(agent);
 }
    public AIState GetState(AIStateId stateId)
    {
        int index = (int)stateId;

        return(states[index]);
    }