using Fluctio.FluctioSim.Core.Components.MachineLearning.NumberProviders;
using UnityEngine;

namespace Fluctio.FluctioSim.Core.Components.Prefabs.Complex.RotaryPendulum
{
    [RequireComponent(typeof(PendulumBalanceTimer))]
    public class PendulumBalanceReward : NumberProvider
    {
        [SerializeField] protected float factor = 100;
        private PendulumBalanceTimer _balanceTimer;

        public void Start()
        {
            _balanceTimer = GetComponent<PendulumBalanceTimer>();
        }

        public override float GetNumber()
        {
            var balanceSeconds = _balanceTimer.GetNumber();
            return factor * balanceSeconds * balanceSeconds;
        }
    }
}