diff --git a/indicators/TwoB.cs b/indicators/TwoB.cs index 5fe7cbc..99429fe 100644 --- a/indicators/TwoB.cs +++ b/indicators/TwoB.cs @@ -17,6 +17,8 @@ namespace NinjaTrader.NinjaScript.Indicators public bool Is2BDown { get; set; } public int StartBarIndex { get; set; } public int EndBarIndex { get; set; } + public double PivotLevel { get; set; } + public int PivotBarIndex { get; set; } } public class TwoB : Indicator @@ -30,9 +32,13 @@ namespace NinjaTrader.NinjaScript.Indicators private double pivotHigh; private double pivotLow; + private int pivotHighBarIndex; + private int pivotLowBarIndex; private double mostRecentPivotHigh; private double mostRecentPivotLow; + private int mostRecentPivotHighBarIndex; + private int mostRecentPivotLowBarIndex; private double twoBUpLevel; private int twoBUpStartBarIndex; @@ -61,6 +67,7 @@ namespace NinjaTrader.NinjaScript.Indicators PivotPeriod = 10; TwoBUpStroke = new Stroke(Brushes.LimeGreen, 3); TwoBDownStroke = new Stroke(Brushes.Red, 3); + PivotStroke = new Stroke(Brushes.Yellow, 3); // For use during backtesting in order to limit the number of patterns processed. MaxPatterns = 0; @@ -88,14 +95,21 @@ namespace NinjaTrader.NinjaScript.Indicators if (pivot != null) { if (pivot.IsPivotHigh) + { pivotHigh = pivot.Level; + pivotHighBarIndex = pivot.StartBarIndex; + } else if (pivot.IsPivotLow) + { pivotLow = pivot.Level; + pivotLowBarIndex = pivot.StartBarIndex; + } } if (Open[0] < pivotHigh && High[0] > pivotHigh && mostRecentPivotHigh != pivotHigh) { mostRecentPivotHigh = pivotHigh; + mostRecentPivotHighBarIndex = pivotHighBarIndex; twoBDownLevel = Low[0]; twoBDownStartBarIndex = CurrentBar; @@ -103,6 +117,7 @@ namespace NinjaTrader.NinjaScript.Indicators else if (Open[0] > pivotLow && Low[0] < pivotLow && mostRecentPivotLow != pivotLow) { mostRecentPivotLow = pivotLow; + mostRecentPivotLowBarIndex = pivotLowBarIndex; twoBUpLevel = High[0]; twoBUpStartBarIndex = CurrentBar; @@ -116,7 +131,9 @@ namespace NinjaTrader.NinjaScript.Indicators Level = twoBUpLevel, Is2BUp = true, StartBarIndex = twoBUpStartBarIndex, - EndBarIndex = twoBUpEndBarIndex + EndBarIndex = twoBUpEndBarIndex, + PivotBarIndex = mostRecentPivotLowBarIndex, + PivotLevel = mostRecentPivotLow, }; twoBs.Add(twoBUp); CompletedPattern[0] = twoBUp; @@ -132,7 +149,9 @@ namespace NinjaTrader.NinjaScript.Indicators Level = twoBDownLevel, Is2BDown = true, StartBarIndex = twoBDownStartBarIndex, - EndBarIndex = twoBDownEndBarIndex + EndBarIndex = twoBDownEndBarIndex, + PivotBarIndex = mostRecentPivotHighBarIndex, + PivotLevel = mostRecentPivotHigh, }; twoBs.Add(twoBDown); CompletedPattern[0] = twoBDown; @@ -155,6 +174,9 @@ namespace NinjaTrader.NinjaScript.Indicators var stroke = twoB.Is2BUp ? TwoBUpStroke : TwoBDownStroke; DrawHorizontalLine(startX, endX, chartScale.GetYByValue(twoB.Level), stroke); + + int pivotX = chartControl.GetXByBarIndex(ChartBars, twoB.PivotBarIndex); + DrawHorizontalLine(pivotX, startX, chartScale.GetYByValue(twoB.PivotLevel), PivotStroke); } } @@ -186,6 +208,9 @@ namespace NinjaTrader.NinjaScript.Indicators [Display(Name = "2B Down Stroke", GroupName = "2B", Order = 3)] public Stroke TwoBDownStroke { get; set; } + [Display(Name = "Pivot Stroke", GroupName = "2B", Order = 4)] + public Stroke PivotStroke { get; set; } + [Browsable(false)] public int MaxPatterns { get; set; } }