chore: import upstream snapshot with attribution
This commit is contained in:
wehub-resource-sync
@@ -0,0 +1,297 @@
|
||||
/*
|
||||
* QUANTCONNECT.COM - Democratizing Finance, Empowering Individuals.
|
||||
* Lean Algorithmic Trading Engine v2.0. Copyright 2014 QuantConnect Corporation.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Globalization;
|
||||
using System.Linq;
|
||||
using QuantConnect.Data;
|
||||
|
||||
namespace QuantConnect.Algorithm.CSharp
|
||||
{
|
||||
/// <summary>
|
||||
/// This demonstration imports indian NSE index "NIFTY" as a tradable security in addition to the USDINR currency pair. We move into the
|
||||
/// NSE market when the economy is performing well.s
|
||||
/// </summary>
|
||||
/// <meta name="tag" content="strategy example" />
|
||||
/// <meta name="tag" content="using data" />
|
||||
/// <meta name="tag" content="custom data" />
|
||||
public class CustomDataNiftyAlgorithm : QCAlgorithm
|
||||
{
|
||||
//Create variables for analyzing Nifty
|
||||
private CorrelationPair _today = new CorrelationPair();
|
||||
private readonly List<CorrelationPair> _prices = new List<CorrelationPair>();
|
||||
private const int _minimumCorrelationHistory = 50;
|
||||
|
||||
/// <summary>
|
||||
/// Initialise the data and resolution required, as well as the cash and start-end dates for your algorithm. All algorithms must initialized.
|
||||
/// </summary>
|
||||
public override void Initialize()
|
||||
{
|
||||
SetStartDate(2008, 1, 8);
|
||||
SetEndDate(2014, 7, 25);
|
||||
|
||||
//Set the cash for the strategy:
|
||||
SetCash(100000);
|
||||
|
||||
//Define the symbol and "type" of our generic data:
|
||||
var rupee = AddData<DollarRupee>("USDINR", Resolution.Daily).Symbol;
|
||||
var nifty = AddData<Nifty>("NIFTY", Resolution.Daily).Symbol;
|
||||
|
||||
Settings.AutomaticIndicatorWarmUp = true;
|
||||
var rupeeSma = SMA(rupee, 20);
|
||||
var niftySma = SMA(rupee, 20);
|
||||
Log($"SMA - Is ready? USDINR: {rupeeSma.IsReady} NIFTY: {niftySma.IsReady}");
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Event Handler for Nifty Data Events: These Nifty objects are created from our
|
||||
/// "Nifty" type below and fired into this event handler.
|
||||
/// </summary>
|
||||
/// <param name="data">One(1) Nifty Object, streamed into our algorithm synchronised in time with our other data streams</param>
|
||||
public override void OnData(Slice slice)
|
||||
{
|
||||
if (slice.ContainsKey("USDINR"))
|
||||
{
|
||||
_today = new CorrelationPair(Time) { CurrencyPrice = Convert.ToDouble(slice["USDINR"].Close) };
|
||||
}
|
||||
|
||||
if (!slice.ContainsKey("NIFTY"))
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
|
||||
_today.NiftyPrice = Convert.ToDouble(slice["NIFTY"].Close);
|
||||
if (_today.Date == slice["NIFTY"].Time)
|
||||
{
|
||||
_prices.Add(_today);
|
||||
|
||||
if (_prices.Count > _minimumCorrelationHistory)
|
||||
{
|
||||
_prices.RemoveAt(0);
|
||||
}
|
||||
}
|
||||
|
||||
//Strategy
|
||||
var quantity = (int)(Portfolio.MarginRemaining * 0.9m / slice["NIFTY"].Close);
|
||||
var highestNifty = (from pair in _prices select pair.NiftyPrice).Max();
|
||||
var lowestNifty = (from pair in _prices select pair.NiftyPrice).Min();
|
||||
|
||||
if (Time.DayOfWeek == DayOfWeek.Wednesday) //prices.Count >= minimumCorrelationHistory &&
|
||||
{
|
||||
//List<double> niftyPrices = (from pair in prices select pair.NiftyPrice).ToList();
|
||||
//List<double> currencyPrices = (from pair in prices select pair.CurrencyPrice).ToList();
|
||||
//double correlation = Correlation.Pearson(niftyPrices, currencyPrices);
|
||||
//double niftyFraction = (correlation)/2;
|
||||
|
||||
if (Convert.ToDouble(slice["NIFTY"].Open) >= highestNifty)
|
||||
{
|
||||
var code = Order("NIFTY", quantity - Portfolio["NIFTY"].Quantity);
|
||||
Debug("LONG " + code + " Time: " + Time.ToShortDateString() + " Quantity: " + quantity + " Portfolio:" + Portfolio["NIFTY"].Quantity + " Nifty: " + slice["NIFTY"].Close + " Buying Power: " + Portfolio.TotalPortfolioValue);
|
||||
}
|
||||
else if (Convert.ToDouble(slice["NIFTY"].Open) <= lowestNifty)
|
||||
{
|
||||
var code = Order("NIFTY", -quantity - Portfolio["NIFTY"].Quantity);
|
||||
Debug("SHORT " + code + " Time: " + Time.ToShortDateString() + " Quantity: " + quantity + " Portfolio:" + Portfolio["NIFTY"].Quantity + " Nifty: " + slice["NIFTY"].Close + " Buying Power: " + Portfolio.TotalPortfolioValue);
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (RegressionTestException err)
|
||||
{
|
||||
Debug("Error: " + err.Message);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// End of a trading day event handler. This method is called at the end of the algorithm day (or multiple times if trading multiple assets).
|
||||
/// </summary>
|
||||
/// <remarks>Method is called 10 minutes before closing to allow user to close out position.</remarks>
|
||||
public override void OnEndOfDay(Symbol symbol)
|
||||
{
|
||||
Plot("Nifty Closing Price", _today.NiftyPrice);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// NIFTY Custom Data Class
|
||||
/// </summary>
|
||||
public class Nifty : BaseData
|
||||
{
|
||||
/// <summary>
|
||||
/// Opening Price
|
||||
/// </summary>
|
||||
public decimal Open { get; set; }
|
||||
/// <summary>
|
||||
/// High Price
|
||||
/// </summary>
|
||||
public decimal High { get; set; }
|
||||
/// <summary>
|
||||
/// Low Price
|
||||
/// </summary>
|
||||
public decimal Low { get; set; }
|
||||
/// <summary>
|
||||
/// Closing Price
|
||||
/// </summary>
|
||||
public decimal Close { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// Default initializer for NIFTY.
|
||||
/// </summary>
|
||||
public Nifty()
|
||||
{
|
||||
Symbol = "NIFTY";
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Return the URL string source of the file. This will be converted to a stream
|
||||
/// </summary>
|
||||
public override SubscriptionDataSource GetSource(SubscriptionDataConfig config, DateTime date, bool isLiveMode)
|
||||
{
|
||||
return new SubscriptionDataSource("https://www.dropbox.com/s/rsmg44jr6wexn2h/CNXNIFTY.csv?dl=1", SubscriptionTransportMedium.RemoteFile);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Reader converts each line of the data source into BaseData objects. Each data type creates its own factory method, and returns a new instance of the object
|
||||
/// each time it is called.
|
||||
/// </summary>
|
||||
public override BaseData Reader(SubscriptionDataConfig config, string line, DateTime date, bool isLiveMode)
|
||||
{
|
||||
//New Nifty object
|
||||
var index = new Nifty();
|
||||
|
||||
try
|
||||
{
|
||||
//Example File Format:
|
||||
//Date, Open High Low Close Volume Turnover
|
||||
//2011-09-13 7792.9 7799.9 7722.65 7748.7 116534670 6107.78
|
||||
var data = line.Split(',');
|
||||
index.Time = DateTime.Parse(data[0], CultureInfo.InvariantCulture);
|
||||
index.EndTime = index.Time.AddDays(1);
|
||||
index.Open = Convert.ToDecimal(data[1], CultureInfo.InvariantCulture);
|
||||
index.High = Convert.ToDecimal(data[2], CultureInfo.InvariantCulture);
|
||||
index.Low = Convert.ToDecimal(data[3], CultureInfo.InvariantCulture);
|
||||
index.Close = Convert.ToDecimal(data[4], CultureInfo.InvariantCulture);
|
||||
index.Symbol = "NIFTY";
|
||||
index.Value = index.Close;
|
||||
}
|
||||
catch
|
||||
{
|
||||
}
|
||||
return index;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/// <summary>
|
||||
/// Dollar Rupe is a custom data type we create for this algorithm
|
||||
/// </summary>
|
||||
public class DollarRupee : BaseData
|
||||
{
|
||||
/// <summary>
|
||||
/// Open Price
|
||||
/// </summary>
|
||||
public decimal Open { get; set; } = 0;
|
||||
/// <summary>
|
||||
/// High Price
|
||||
/// </summary>
|
||||
public decimal High { get; set; } = 0;
|
||||
/// <summary>
|
||||
/// Low Price
|
||||
/// </summary>
|
||||
public decimal Low { get; set; } = 0;
|
||||
/// <summary>
|
||||
/// Closing Price
|
||||
/// </summary>
|
||||
public decimal Close { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// Default constructor for the custom data class.
|
||||
/// </summary>
|
||||
public DollarRupee()
|
||||
{
|
||||
Symbol = "USDINR";
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Return the URL string source of the file. This will be converted to a stream
|
||||
/// </summary>
|
||||
public override SubscriptionDataSource GetSource(SubscriptionDataConfig config, DateTime date, bool isLiveMode)
|
||||
{
|
||||
return new SubscriptionDataSource("https://www.dropbox.com/s/m6ecmkg9aijwzy2/USDINR.csv?dl=1", SubscriptionTransportMedium.RemoteFile);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Reader converts each line of the data source into BaseData objects. Each data type creates its own factory method, and returns a new instance of the object
|
||||
/// each time it is called.
|
||||
/// </summary>
|
||||
public override BaseData Reader(SubscriptionDataConfig config, string line, DateTime date, bool isLiveMode)
|
||||
{
|
||||
//New USDINR object
|
||||
var currency = new DollarRupee();
|
||||
|
||||
try
|
||||
{
|
||||
var data = line.Split(',');
|
||||
currency.Time = DateTime.Parse(data[0], CultureInfo.InvariantCulture);
|
||||
currency.EndTime = currency.Time.AddDays(1);
|
||||
currency.Close = Convert.ToDecimal(data[1], CultureInfo.InvariantCulture);
|
||||
currency.Symbol = "USDINR";
|
||||
currency.Value = currency.Close;
|
||||
}
|
||||
catch
|
||||
{
|
||||
}
|
||||
return currency;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Correlation Pair is a helper class to combine two data points which we'll use to perform the correlation.
|
||||
/// </summary>
|
||||
public class CorrelationPair
|
||||
{
|
||||
/// <summary>
|
||||
/// Date of the correlation pair
|
||||
/// </summary>
|
||||
public DateTime Date { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// Nifty price for this correlation pair
|
||||
/// </summary>
|
||||
public double NiftyPrice { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// Currency price for this correlation pair
|
||||
/// </summary>
|
||||
public double CurrencyPrice { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// Default initializer
|
||||
/// </summary>
|
||||
public CorrelationPair()
|
||||
{ }
|
||||
|
||||
/// <summary>
|
||||
/// Date based correlation pair initializer
|
||||
/// </summary>
|
||||
public CorrelationPair(DateTime date)
|
||||
{
|
||||
Date = date.Date;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user