ninjatrader/indicators/VolatilityRegime.cs

145 lines
4.8 KiB
C#

#region Using declarations
using System.Collections.Generic;
using System.ComponentModel;
using System.ComponentModel.DataAnnotations;
using System.Linq;
using System.Xml.Serialization;
using NinjaTrader.Gui.Chart;
using NinjaTrader.Data;
#endregion
//This namespace holds Indicators in this folder and is required. Do not change it.
namespace NinjaTrader.NinjaScript.Indicators
{
public class VolatilityRegime : Indicator
{
private const int VIXBars = 1;
protected override void OnStateChange()
{
if (State == State.SetDefaults)
{
Description = @"";
Name = "Volatility Regime";
Calculate = Calculate.OnBarClose;
IsOverlay = true;
DisplayInDataBox = true;
DrawOnPricePanel = true;
ScaleJustification = ScaleJustification.Right;
IsSuspendedWhileInactive = true;
RankPeriod = 252; // Approximately 1 year in trading days.
}
else if (State == State.Configure)
{
// TODO: Base number of days to request on a configurable date range.
AddDataSeries("^VIX", new BarsPeriod { BarsPeriodType = BarsPeriodType.Day, Value = 1 },
2000, "US Equities RTH", Bars.IsResetOnNewTradingDay);
}
else if (State == State.DataLoaded)
{
PercentRank = new Series<double>(this);
}
else if (State == State.Historical)
{
SetZOrder(-1); // Display behind bars on chart.
}
}
protected override void OnBarUpdate()
{
if (VIXBars != BarsInProgress)
return;
// Calculate percent rank of VIX closing prices over lookback period.
if (CurrentBar >= RankPeriod)
{
List<double> vixCloses = new List<double>();
for (int i = RankPeriod; i > 0; i--)
vixCloses.Add(Closes[VIXBars][i]);
double currentVixClose = Closes[VIXBars][0];
int rank = vixCloses.Count(x => x <= currentVixClose); // # of closes <= current price.
PercentRank[0] = (double)rank / RankPeriod;
}
else
{
PercentRank[0] = 0; // Insufficient data.
}
}
public override string DisplayName
{
get { return Name; }
}
[Browsable(false)]
[XmlIgnore]
public Series<double> PercentRank { get; set; }
#region Properties
[NinjaScriptProperty]
[Range(1, int.MaxValue)]
[Display(Name = "Rank Period", Description = "Period for the percent rank lookback", Order = 1, GroupName = "Volatility Regime")]
public int RankPeriod { get; set; }
#endregion
}
}
#region NinjaScript generated code. Neither change nor remove.
namespace NinjaTrader.NinjaScript.Indicators
{
public partial class Indicator : NinjaTrader.Gui.NinjaScript.IndicatorRenderBase
{
private VolatilityRegime[] cacheVolatilityRegime;
public VolatilityRegime VolatilityRegime(int rankPeriod)
{
return VolatilityRegime(Input, rankPeriod);
}
public VolatilityRegime VolatilityRegime(ISeries<double> input, int rankPeriod)
{
if (cacheVolatilityRegime != null)
for (int idx = 0; idx < cacheVolatilityRegime.Length; idx++)
if (cacheVolatilityRegime[idx] != null && cacheVolatilityRegime[idx].RankPeriod == rankPeriod && cacheVolatilityRegime[idx].EqualsInput(input))
return cacheVolatilityRegime[idx];
return CacheIndicator<VolatilityRegime>(new VolatilityRegime(){ RankPeriod = rankPeriod }, input, ref cacheVolatilityRegime);
}
}
}
namespace NinjaTrader.NinjaScript.MarketAnalyzerColumns
{
public partial class MarketAnalyzerColumn : MarketAnalyzerColumnBase
{
public Indicators.VolatilityRegime VolatilityRegime(int rankPeriod)
{
return indicator.VolatilityRegime(Input, rankPeriod);
}
public Indicators.VolatilityRegime VolatilityRegime(ISeries<double> input , int rankPeriod)
{
return indicator.VolatilityRegime(input, rankPeriod);
}
}
}
namespace NinjaTrader.NinjaScript.Strategies
{
public partial class Strategy : NinjaTrader.Gui.NinjaScript.StrategyRenderBase
{
public Indicators.VolatilityRegime VolatilityRegime(int rankPeriod)
{
return indicator.VolatilityRegime(Input, rankPeriod);
}
public Indicators.VolatilityRegime VolatilityRegime(ISeries<double> input , int rankPeriod)
{
return indicator.VolatilityRegime(input, rankPeriod);
}
}
}
#endregion