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

namespace cAlgo.Indicators
{
    [Indicator(IsOverlay = false, AccessRights = AccessRights.None)]
    public class AwesomeOscillator : Indicator
    {
        [Parameter(DefaultValue = 5)]
        public int periodFast { get; set; }

        [Parameter(DefaultValue = 34)]
        public int periodSlow { get; set; }

        [Parameter("Verbose", DefaultValue = true)]
        public bool OnVerbose { get; set; }

        [Output("Strong Bullish", PlotType = PlotType.Histogram, LineColor = "ForestGreen", Thickness = 4)]
        public IndicatorDataSeries StrongBullish { get; set; }

        [Output("Weak Bullish", PlotType = PlotType.Histogram, LineColor = "LawnGreen", Thickness = 4)]
        public IndicatorDataSeries WeakBullish { get; set; }

        [Output("Strong Bearish", PlotType = PlotType.Histogram, LineColor = "Red", Thickness = 4)]
        public IndicatorDataSeries StrongBearish { get; set; }

        [Output("Weak Bearish", PlotType = PlotType.Histogram, LineColor = "DarkSalmon", Thickness = 4)]
        public IndicatorDataSeries WeakBearish { get; set; }

        private SimpleMovingAverage smaSlow;
        private SimpleMovingAverage smaFast;
        private IndicatorDataSeries medianprice;

        protected override void Initialize()
        {
            medianprice = CreateDataSeries();
            smaSlow = Indicators.SimpleMovingAverage(medianprice, periodSlow);
            smaFast = Indicators.SimpleMovingAverage(medianprice, periodFast);
        }

        public override void Calculate(int index)
        {
            medianprice[index] = (Bars.HighPrices[index] + Bars.LowPrices[index]) / 2;
            double previousAO = smaFast.Result[index - 1] - smaSlow.Result[index - 1];
            double currentAO = smaFast.Result[index] - smaSlow.Result[index];

            if (currentAO > 0.0)
            {
                if (currentAO >= previousAO)
                {
                    StrongBullish[index] = smaFast.Result[index] - smaSlow.Result[index];
                }
                else
                {
                    WeakBullish[index] = smaFast.Result[index] - smaSlow.Result[index];
                }
            }
            else if (currentAO < 0.0)
            {
                if (currentAO <= previousAO)
                {
                    StrongBearish[index] = smaFast.Result[index] - smaSlow.Result[index];
                }
                else
                {
                    WeakBearish[index] = smaFast.Result[index] - smaSlow.Result[index];
                }

            }
        }
    }
}