ninjatrader/strategies/trin/TRINBot.cs

100 lines
3.1 KiB
C#

#region Using declarations
using NinjaTrader.Data;
using NinjaTrader.NinjaScript.Indicators;
using System.ComponentModel.DataAnnotations;
#endregion
namespace NinjaTrader.NinjaScript.Strategies
{
public class TRINBot : Strategy
{
private RSI rsi;
private SMA longTermTrend;
protected override void OnStateChange()
{
if (State == State.SetDefaults)
{
Name = "TRIN Bot";
Description = @"Taken from chapter 12 of Short Term Trading Strategies That Work";
Calculate = Calculate.OnBarClose;
EntriesPerDirection = 1;
LongTermTrendPeriod = 200;
RSIPeriod = 2;
RSISmoothing = 1;
RSIEntryThreshold = 50;
RSIExitThreshold = 65;
TRINEntryThreshold = 1.0;
DaysAboveThreshold = 3;
}
else if (State == State.Configure)
{
AddDataSeries("^TRIN", BarsPeriodType.Day, 1);
}
else if (State == State.DataLoaded)
{
rsi = RSI(RSIPeriod, RSISmoothing);
longTermTrend = SMA(LongTermTrendPeriod);
}
}
protected override void OnBarUpdate()
{
if (BarsInProgress != 0)
return;
if (CurrentBar < LongTermTrendPeriod || CurrentBars[1] < DaysAboveThreshold)
return;
if (Close[0] > longTermTrend[0] && rsi[0] < RSIEntryThreshold)
{
for (int i = 0; i < DaysAboveThreshold; i++)
{
if (Closes[1][i] <= TRINEntryThreshold)
return;
}
EnterLong();
}
else if (rsi[0] > RSIExitThreshold)
ExitLong();
}
public override string DisplayName
{
get { return Name; }
}
[NinjaScriptProperty]
[Display(Name = "Long-Term Trend Period", GroupName = "TRIN Bot", Order = 1)]
public int LongTermTrendPeriod { get; set; }
[NinjaScriptProperty]
[Display(Name = "RSI Period", GroupName = "TRIN Bot", Order = 2)]
public int RSIPeriod { get; set; }
[NinjaScriptProperty]
[Display(Name = "RSI Smoothing", GroupName = "TRIN Bot", Order = 3)]
public int RSISmoothing { get; set; }
[NinjaScriptProperty]
[Display(Name = "RSI Entry Threshold", GroupName = "TRIN Bot", Order = 4)]
public int RSIEntryThreshold { get; set; }
[NinjaScriptProperty]
[Display(Name = "RSI Exit Threshold", GroupName = "TRIN Bot", Order = 5)]
public int RSIExitThreshold { get; set; }
[NinjaScriptProperty]
[Display(Name = "TRIN Entry Threshold", GroupName = "TRIN Bot", Order = 6)]
public double TRINEntryThreshold { get; set; }
[NinjaScriptProperty]
[Display(Name = "Days Above Threshold", GroupName = "TRIN Bot", Order = 7)]
public int DaysAboveThreshold { get; set; }
}
}