using System;
using Fluctio.FluctioSim.Core.Components.MachineLearning.Agents;
using UnityEngine;

namespace Fluctio.FluctioSim.Core.Components.MachineLearning.NumberProviders
{
	public abstract class AgentNumberProvider : NumberProvider
	{
		[NonSerialized] protected Agent Agent;

		protected virtual void Awake()
		{
			Agent = GetComponentInParent<Agent>();
			if (Agent == null)
			{
				Debug.LogWarning("This component should be inside GameObject with Agent component");
			}
		}

		protected void CheckAgentExists()
		{
			if (Agent == null)
			{
				throw new ArgumentNullException(nameof(Agent));
			}
		}
	}
}