Color MA


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

namespace cAlgo.Indicators
{
    [Indicator(IsOverlay = true, AccessRights = AccessRights.None)]
    public class ColorMA : Indicator
    {
        [Parameter("Periods", DefaultValue = 24)]
        public int period { get; set; }

        [Parameter("MA Type")]
        public MovingAverageType maType { get; set; }

        [Output("Main", Color = Colors.Cyan)]
        public IndicatorDataSeries Result { get; set; }

        private SimpleMovingAverage sma;
        private ExponentialMovingAverage ema;
        private TimeSeriesMovingAverage ts;
        private WeightedMovingAverage wma;
        private Vidya vid;
        private TriangularMovingAverage trn;
        private WellesWilderSmoothing wds;
        private int thc;

        protected override void Initialize()
        {
            Thickness();
            if (maType == MovingAverageType.Simple)
                sma = Indicators.SimpleMovingAverage(MarketSeries.Close, period);

            else if (maType == MovingAverageType.Exponential)
                ema = Indicators.ExponentialMovingAverage(MarketSeries.Close, period);

            else if (maType == MovingAverageType.TimeSeries)
                ts = Indicators.TimeSeriesMovingAverage(MarketSeries.Close, period);

            else if (maType == MovingAverageType.Weighted)
                wma = Indicators.WeightedMovingAverage(MarketSeries.Close, period);

            else if (maType == MovingAverageType.VIDYA)
                vid = Indicators.Vidya(MarketSeries.Close, period, 0.65);

            else if (maType == MovingAverageType.Triangular)
                trn = Indicators.TriangularMovingAverage(MarketSeries.Close, period);

            else if (maType == MovingAverageType.WilderSmoothing)
                wds = Indicators.WellesWilderSmoothing(MarketSeries.Close, period);

            Chart.ZoomChanged += OnChartZoomChanged;
        }

        void OnChartZoomChanged(ChartZoomEventArgs obj)
        {
            Thickness();
            for (int i = 0; i < Chart.BarsTotal; i++)
            {
                Chart.RemoveObject("smaColorBody" + i);
                Calculate(i);
            }
        }

        public override void Calculate(int index)
        {
            if (maType == MovingAverageType.Simple)
                Result[index] = sma.Result[index];
            else if (maType == MovingAverageType.Exponential)
                Result[index] = ema.Result[index];
            else if (maType == MovingAverageType.TimeSeries)
                Result[index] = ts.Result[index];
            else if (maType == MovingAverageType.Weighted)
                Result[index] = wma.Result[index];
            else if (maType == MovingAverageType.VIDYA)
                Result[index] = vid.Result[index];
            else if (maType == MovingAverageType.Triangular)
                Result[index] = trn.Result[index];
            else if (maType == MovingAverageType.WilderSmoothing)
                Result[index] = wds.Result[index];

            Chart.DrawTrendLine("smaColorHL" + index, index, MarketSeries.High[index], index, MarketSeries.Low[index], MarketSeries.High[index] < Result[index] ? Color.Red : MarketSeries.Low[index] > Result[index] ? Color.LawnGreen : Color.Orange);
            Chart.DrawTrendLine("smaColorBody" + index, index, MarketSeries.Close[index], index, MarketSeries.Open[index], MarketSeries.High[index] < Result[index] ? Color.Red : MarketSeries.Low[index] > Result[index] ? Color.LawnGreen : Color.Orange, thc);
        }

        private void Thickness()
        {
            if (Chart.Zoom == 0)
            {
                thc = 1;
            }
            if (Chart.Zoom == 1)
            {
                thc = 2;
            }
            if (Chart.Zoom == 2)
            {
                thc = 3;
            }
            if (Chart.Zoom == 3)
            {
                thc = 5;
            }
            if (Chart.Zoom == 4)
            {
                thc = 11;
            }
            if (Chart.Zoom == 5)
            {
                thc = 25;
            }
            return;
        }
    }
}