using System;

namespace UnityEngine.Perception.Randomization.Scenarios
{
    /// <summary>
    /// An example scenario where each scenario iteration runs for exactly one frame
    /// </summary>
    [AddComponentMenu("Perception/Randomization/Scenarios/Fixed Length Scenario")]
    public class FixedLengthScenario: Scenario<FixedLengthScenario.Constants>
    {
        [Serializable]
        public class Constants
        {
            public int framesPerIteration = 1;
            public int startingIteration;
            public int totalIterations = 1000;
        }

        public override bool isIterationComplete => currentIterationFrame >= constants.framesPerIteration;

        public override bool isScenarioComplete => currentIteration >= constants.totalIterations;

        public override void OnInitialize()
        {
            currentIteration = constants.startingIteration;
        }
    }
}