using System;
using System.Windows.Media;
using NinjaTrader.Data;

//This namespace holds strategies in this folder and is required. Do not change it.
namespace NinjaTrader.NinjaScript.Strategies.Barz.Sandbox.FuturesIO
{
    public class FIO_DetermineTimeFrame : Strategy
    {
        private TimeSpan[] _PeriodSpans;

        protected override void OnBarUpdate()
        {
            if (CurrentBar < BarsRequiredToTrade) { return; }

            // Conditionally populate plots based on whether the period is less
            // than or greater than the primary period.
            if (BarsInProgress == 0)
            {
                if (_PeriodSpans[1] > _PeriodSpans[0])
                {
                    Values[0][0] = Closes[1][0] + 20;
                }
                if (_PeriodSpans[2] < _PeriodSpans[0])
                {
                    Values[1][0] = Closes[2][0] - 20;
                }
            }

        }
        protected override void OnStateChange()
        {
            if (State == State.SetDefaults)
            {
                Name = "FIO_DetermineTimeFrame";

                // This creates a plot whose current value can be set in
                // OnBarUpdate using Values[0][0] = 25.1.
                AddPlot(Brushes.Yellow, "Larger Period");

                // This creates a plot whose current value can be set in
                // OnBarUpdate using Values[1][0] = 25.1.
                AddPlot(Brushes.Red, "Smaller Period");
            }
            else if (State == State.Configure)
            {
                // Ensure a 4 hour series has been added to your chart to serve
                // as the primary (base) series.

                // Add day chart here
                AddDataSeries(BarsPeriodType.Day, 1);

                // Add the 60 minute chart here
                AddDataSeries(BarsPeriodType.Minute, 60);
            }
            else if (State == State.DataLoaded)
            {
                _PeriodSpans = new TimeSpan[BarsArray.Length];

                for (int i = 0; i < BarsArray.Length; i++)
                {
                    BarsPeriod period = BarsArray[i].BarsPeriod;
                    Print(i + ": Period=" + period.BarsPeriodType.ToString()
                        + ", TimeSpan=" + period.ToTimeSpan());

                    // Capture the period for each series as a time span so we
                    // can easily do comparisons later.
                    _PeriodSpans[i] = period.ToTimeSpan();
                }

            }
        }

    }

    public static class BarsPeriodExtensions
    {
        public static TimeSpan ToTimeSpan(this BarsPeriod period)
        {
            switch (period.BarsPeriodType)
            {
                case BarsPeriodType.Second:
                    return TimeSpan.FromSeconds(period.Value);
                case BarsPeriodType.Minute:
                    return TimeSpan.FromMinutes(period.Value);
                case BarsPeriodType.Day:
                    return TimeSpan.FromDays(period.Value);
                case BarsPeriodType.Week:
                    return TimeSpan.FromDays(7 * period.Value);
                default:
                    throw new ArgumentException("BarsPeriodType not supported for conversion to TimeSpan: "
                        + period.BarsPeriodType);
            }
        }
    }
}