using System;
using cAlgo.API;
using cAlgo.API.Internals;
using cAlgo.API.Indicators;
using cAlgo.Indicators;

namespace cAlgo
{
    [Indicator(IsOverlay = false, TimeZone = TimeZones.UTC, AccessRights = AccessRights.None)]

    [Levels(20, 50, 80)]

    public class MultiTimeframeStochasticOscillator : Indicator
    {
        private StochasticOscillator _Stochastic;
        private Bars _Series;
        private int _TimeframeIndex;

        [Parameter("%K Periods", DefaultValue = 5, MinValue = 1)]
        public int _KPeriods { get; set; }

        [Parameter("%K Slowing", DefaultValue = 3, MinValue = 1)]
        public int _KSlowing { get; set; }

        [Parameter("%D Periods", DefaultValue = 3, MinValue = 0)]
        public int _DPeriods { get; set; }

        [Parameter("MA Type", DefaultValue = MovingAverageType.Simple)]
        public MovingAverageType _MAType { get; set; }

        [Parameter("Timeframe", DefaultValue = "hour1")]
        public TimeFrame _Timeframe { get; set; }

        [Parameter("Smoothing", DefaultValue = _Smoothing.No)]
        public _Smoothing _SmoothLine { get; set; }
        public enum _Smoothing
        {
            Yes,
            No
        }

        [Output("%D", LineColor = "Red", LineStyle = LineStyle.Lines)]
        public IndicatorDataSeries _PercentD { get; set; }

        [Output("%K", LineColor = "Green")]
        public IndicatorDataSeries _PercentK { get; set; }

        protected override void Initialize()
        {
            _Series = MarketData.GetBars(_Timeframe);
            _Stochastic = Indicators.StochasticOscillator(_Series, _KPeriods, _KSlowing, _DPeriods, _MAType);
        }

        public override void Calculate(int index)
        {
            switch (_SmoothLine)
            {
                case _Smoothing.No:
                    _TimeframeIndex = _Series.OpenTimes.GetIndexByTime(Bars.OpenTimes[index]);
                    break;
                default:
                    _TimeframeIndex = GetIndexByDate(_Series, Bars.OpenTimes[index]);
                    break;
            }

            _PercentD[index] = _Stochastic.PercentD[_TimeframeIndex];
            _PercentK[index] = _Stochastic.PercentK[_TimeframeIndex];
        }

        private int GetIndexByDate(Bars series, DateTime time)
        {
            for (int i = series.ClosePrices.Count - 1; i > 0; i--)
            {
                if (series.OpenTimes[i] == time)
                    return i;
            }
            return -1;
        }

    }
}