ninjatrader/indicators/vwap/VWAP.cs

426 lines
14 KiB
C#

#region Using declarations
using NinjaTrader.Cbi;
using NinjaTrader.Data;
using NinjaTrader.Gui;
using NinjaTrader.NinjaScript.MarketAnalyzerColumns;
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.ComponentModel.DataAnnotations;
using System.Globalization;
using System.Windows.Media;
using System.Xml.Serialization;
#endregion
namespace NinjaTrader.NinjaScript.Indicators
{
public class VWAP : Indicator
{
private static int CHART_BARS = 0;
private static int TICK_BARS = 1;
private double TotalVolume;
private double VolumeWeightedPrice;
private double VolumeWeightedSquaredPrice;
private double TotalVolumeTick;
private double VolumeWeightedPriceTick;
private double VolumeWeightedSquaredPriceTick;
private DateTime MostRecentDate = DateTime.MinValue;
protected override void OnStateChange()
{
if (State == State.SetDefaults)
{
Description = "Volume Weighted Average Price (VWAP)";
Name = "VWAP";
Calculate = Calculate.OnBarClose;
IsOverlay = true;
DisplayInDataBox = true;
DrawOnPricePanel = true;
ScaleJustification = Gui.Chart.ScaleJustification.Right;
IsSuspendedWhileInactive = true;
BarsRequiredToPlot = 0;
Precision = VWAPPrecision.Chart;
Price = VWAPPrice.HLC; // Typical Price
StdDevMultiplier = 1.0;
AddPlot(new Stroke(Brushes.Yellow, DashStyleHelper.Solid, 3), PlotStyle.Line, "VWAP");
AddPlot(new Stroke(Brushes.Transparent, DashStyleHelper.Solid, 3), PlotStyle.Line, "Upper Standard Deviation");
AddPlot(new Stroke(Brushes.Transparent, DashStyleHelper.Solid, 3), PlotStyle.Line, "Lower Standard Deviation");
}
else if (State == State.Configure)
{
ClearOutputWindow();
if (Precision == VWAPPrecision.Tick)
{
AddDataSeries(Instrument.FullName, new BarsPeriod
{
BarsPeriodType = BarsPeriodType.Tick,
Value = 1
}, Bars.TradingHours.Name);
}
}
else if (State == State.DataLoaded)
{
ResetVWAP();
}
else if (State == State.Historical)
{
SetZOrder(-1); // Display behind bars on chart.
}
}
private double GetPrice(int bars, int index = 0)
{
switch (Price)
{
case VWAPPrice.Open:
return Opens[bars][index];
case VWAPPrice.High:
return Highs[bars][index];
case VWAPPrice.Low:
return Lows[bars][index];
case VWAPPrice.Close:
return Closes[bars][index];
case VWAPPrice.HL:
return (Highs[bars][index] + Lows[bars][index]) / 2;
case VWAPPrice.HLC:
return (Highs[bars][index] + Lows[bars][index] + Closes[bars][index]) / 3;
case VWAPPrice.OHLC:
default:
return (Opens[bars][index] + Highs[bars][index] + Lows[bars][index] + Closes[bars][index]) / 4;
}
}
private void ResetVWAP()
{
TotalVolume = 0;
VolumeWeightedPrice = 0;
VolumeWeightedSquaredPrice = 0;
TotalVolumeTick = 0;
VolumeWeightedPriceTick = 0;
VolumeWeightedSquaredPriceTick = 0;
}
protected override void OnBarUpdate()
{
if (Bars.IsFirstBarOfSession && IsFirstTickOfBar)
{
DateTime currentDate = Time[0].Date;
if (MostRecentDate != currentDate)
ResetVWAP();
if (CHART_BARS == BarsInProgress)
{
// Prevent visible jumps in VWAP when a new session begins.
PlotBrushes[0][0] = Brushes.Transparent;
PlotBrushes[1][0] = Brushes.Transparent;
PlotBrushes[2][0] = Brushes.Transparent;
}
MostRecentDate = currentDate;
}
if (CHART_BARS == BarsInProgress)
{
double price = GetPrice(CHART_BARS);
double vwap;
double variance;
double volumeWeightedPrice;
double volumeWeightedSquaredPrice;
double totalVolume;
if (Precision == VWAPPrecision.Tick)
{
volumeWeightedPrice = VolumeWeightedPriceTick;
volumeWeightedSquaredPrice = VolumeWeightedSquaredPriceTick;
totalVolume = TotalVolumeTick;
}
else if (Calculate == Calculate.OnBarClose)
{
VolumeWeightedPrice += Volume[0] * price;
VolumeWeightedSquaredPrice += Volume[0] * Math.Pow(price, 2);
TotalVolume += Volume[0];
volumeWeightedPrice = VolumeWeightedPrice;
volumeWeightedSquaredPrice = VolumeWeightedSquaredPrice;
totalVolume = TotalVolume;
}
else
{
if (IsFirstTickOfBar && !Bars.IsFirstBarOfSession)
{
int previousBar = 1;
double previousPrice = GetPrice(CHART_BARS, previousBar);
VolumeWeightedPrice += Volume[previousBar] * previousPrice;
VolumeWeightedSquaredPrice += Volume[previousBar] * Math.Pow(previousPrice, 2);
TotalVolume += Volume[previousBar];
}
volumeWeightedPrice = VolumeWeightedPrice + (Volume[0] * price);
volumeWeightedSquaredPrice = VolumeWeightedSquaredPrice + (Volume[0] * Math.Pow(price, 2));
totalVolume = TotalVolume + Volume[0];
}
vwap = volumeWeightedPrice / totalVolume;
variance = (volumeWeightedSquaredPrice / totalVolume) - Math.Pow(vwap, 2);
double stdDev = Math.Sqrt(variance);
Value[0] = vwap;
Values[1][0] = vwap + (StdDevMultiplier * stdDev);
Values[2][0] = vwap - (StdDevMultiplier * stdDev);
}
else if (TICK_BARS == BarsInProgress)
{
double price = GetPrice(TICK_BARS);
TotalVolumeTick += Volumes[TICK_BARS][0];
VolumeWeightedPriceTick += Volumes[TICK_BARS][0] * price;
VolumeWeightedSquaredPriceTick += Volumes[TICK_BARS][0] * Math.Pow(price, 2);
}
}
public override string DisplayName
{
get { return Name; }
}
[Browsable(false)]
[XmlIgnore]
public Series<double> UpperStdDev
{
get { return Values[1]; }
}
[Browsable(false)]
[XmlIgnore]
public Series<double> LowerStdDev
{
get { return Values[2]; }
}
#region Properties
[TypeConverter(typeof(VWAPPrecisionConverter))]
[PropertyEditor("NinjaTrader.Gui.Tools.StringStandardValuesEditorKey")]
[Display(Name = "Precision", Description = "Precision with which VWAP is calculated", GroupName = "VWAP", Order = 1)]
public VWAPPrecision Precision
{ get; set; }
[TypeConverter(typeof(VWAPPriceConverter))]
[PropertyEditor("NinjaTrader.Gui.Tools.StringStandardValuesEditorKey")]
[Display(Name = "Average Price", Description = "How the average price of the bar is calculated", GroupName = "VWAP", Order = 2)]
public VWAPPrice Price
{ get; set; }
[Range(0.1, double.MaxValue)]
[Display(Name = "Standard Deviation Multiplier", Description = "Multiplier for standard deviation bands", GroupName = "VWAP", Order = 3)]
public double StdDevMultiplier
{ get; set; }
#endregion
}
}
public enum VWAPPrecision
{
Chart,
Tick
}
public enum VWAPPrice
{
Open,
High,
Low,
Close,
HL,
HLC,
OHLC
}
public class VWAPPrecisionConverter : TypeConverter
{
private const string CHART = "Chart";
private const string TICK = "Tick";
public override StandardValuesCollection GetStandardValues(ITypeDescriptorContext context)
{
List<string> values = new List<string>() { CHART, TICK };
return new StandardValuesCollection(values);
}
public override object ConvertFrom(ITypeDescriptorContext context, CultureInfo culture, object value)
{
switch (value.ToString())
{
case CHART:
return VWAPPrecision.Chart;
case TICK:
return VWAPPrecision.Tick;
}
return VWAPPrecision.Chart;
}
public override object ConvertTo(ITypeDescriptorContext context, CultureInfo culture, object value, Type destinationType)
{
VWAPPrecision enumValue = (VWAPPrecision)Enum.Parse(typeof(VWAPPrecision), value.ToString());
switch (enumValue)
{
case VWAPPrecision.Chart:
return CHART;
case VWAPPrecision.Tick:
return TICK;
}
return CHART;
}
public override bool CanConvertFrom(ITypeDescriptorContext context, Type sourceType)
{ return true; }
public override bool CanConvertTo(ITypeDescriptorContext context, Type destinationType)
{ return true; }
public override bool GetStandardValuesExclusive(ITypeDescriptorContext context)
{ return true; }
public override bool GetStandardValuesSupported(ITypeDescriptorContext context)
{ return true; }
}
public class VWAPPriceConverter : TypeConverter
{
private const string OPEN = "Open";
private const string HIGH = "High";
private const string LOW = "Low";
private const string CLOSE = "Close";
private const string HIGH_LOW = "(High + Low) / 2";
private const string HIGH_LOW_CLOSE = "(High + Low + Close) / 3";
private const string OPEN_HIGH_LOW_CLOSE = "(Open + High + Low + Close) / 4";
public override StandardValuesCollection GetStandardValues(ITypeDescriptorContext context)
{
List<string> values = new List<string>() { OPEN, HIGH, LOW, CLOSE, HIGH_LOW, HIGH_LOW_CLOSE, OPEN_HIGH_LOW_CLOSE };
return new StandardValuesCollection(values);
}
public override object ConvertFrom(ITypeDescriptorContext context, CultureInfo culture, object value)
{
switch (value.ToString())
{
case OPEN:
return VWAPPrice.Open;
case HIGH:
return VWAPPrice.High;
case LOW:
return VWAPPrice.Low;
case CLOSE:
return VWAPPrice.Close;
case HIGH_LOW:
return VWAPPrice.HL;
case HIGH_LOW_CLOSE:
return VWAPPrice.HLC;
case OPEN_HIGH_LOW_CLOSE:
return VWAPPrice.OHLC;
}
return VWAPPrice.OHLC;
}
public override object ConvertTo(ITypeDescriptorContext context, CultureInfo culture, object value, Type destinationType)
{
VWAPPrice enumValue = (VWAPPrice)Enum.Parse(typeof(VWAPPrice), value.ToString());
switch (enumValue)
{
case VWAPPrice.Open:
return OPEN;
case VWAPPrice.High:
return HIGH;
case VWAPPrice.Low:
return LOW;
case VWAPPrice.Close:
return CLOSE;
case VWAPPrice.HL:
return HIGH_LOW;
case VWAPPrice.HLC:
return HIGH_LOW_CLOSE;
case VWAPPrice.OHLC:
return OPEN_HIGH_LOW_CLOSE;
}
return OPEN_HIGH_LOW_CLOSE;
}
public override bool CanConvertFrom(ITypeDescriptorContext context, Type sourceType)
{ return true; }
public override bool CanConvertTo(ITypeDescriptorContext context, Type destinationType)
{ return true; }
public override bool GetStandardValuesExclusive(ITypeDescriptorContext context)
{ return true; }
public override bool GetStandardValuesSupported(ITypeDescriptorContext context)
{ return true; }
}
#region NinjaScript generated code. Neither change nor remove.
namespace NinjaTrader.NinjaScript.Indicators
{
public partial class Indicator : NinjaTrader.Gui.NinjaScript.IndicatorRenderBase
{
private VWAP[] cacheVWAP;
public VWAP VWAP()
{
return VWAP(Input);
}
public VWAP VWAP(ISeries<double> input)
{
if (cacheVWAP != null)
for (int idx = 0; idx < cacheVWAP.Length; idx++)
if (cacheVWAP[idx] != null && cacheVWAP[idx].EqualsInput(input))
return cacheVWAP[idx];
return CacheIndicator<VWAP>(new VWAP(), input, ref cacheVWAP);
}
}
}
namespace NinjaTrader.NinjaScript.MarketAnalyzerColumns
{
public partial class MarketAnalyzerColumn : MarketAnalyzerColumnBase
{
public Indicators.VWAP VWAP()
{
return indicator.VWAP(Input);
}
public Indicators.VWAP VWAP(ISeries<double> input )
{
return indicator.VWAP(input);
}
}
}
namespace NinjaTrader.NinjaScript.Strategies
{
public partial class Strategy : NinjaTrader.Gui.NinjaScript.StrategyRenderBase
{
public Indicators.VWAP VWAP()
{
return indicator.VWAP(Input);
}
public Indicators.VWAP VWAP(ISeries<double> input )
{
return indicator.VWAP(input);
}
}
}
#endregion