using Fluctio.FluctioSim.Common.Configuration;
using Fluctio.FluctioSim.Core.MujocoExtensions;
using Fluctio.FluctioSim.Core.MujocoExtensions.States;
using Mujoco;
using UnityEngine;

namespace Fluctio.FluctioSim.Core.Components.MachineLearning.TriggerActions
{
	[AddComponentMenu(Config.PrefixedName + "/Machine Learning/Trigger Actions/Reset MuJoCo Chain", Config.ComponentMenuOrder + 80)]
	[DefaultExecutionOrder(10)] // should be after MjScene
	public class MujocoChainReseter : TriggerAction
	{
		[SerializeField] public MjBody body;
		private ChainState _initialState;

		private void Reset()
		{
			priority = -10; // before Randomizers
			body = GetComponent<MjBody>();
		}

		private void Start()
		{
			_initialState = body.GetChainState();
		}

		public override void OnExecute()
		{
			body.SetChainState(_initialState);
		}
	}
}