chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,115 @@
|
||||
/*
|
||||
* 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 QuantConnect.Data.Market;
|
||||
using Python.Runtime;
|
||||
|
||||
namespace QuantConnect.Data.Consolidators
|
||||
{
|
||||
/// <summary>
|
||||
/// Type capable of consolidating trade bars from any base data instance
|
||||
/// </summary>
|
||||
public class BaseDataConsolidator : TradeBarConsolidatorBase<BaseData>
|
||||
{
|
||||
/// <summary>
|
||||
/// Create a new TickConsolidator for the desired resolution
|
||||
/// </summary>
|
||||
/// <param name="resolution">The resolution desired</param>
|
||||
/// <returns>A consolidator that produces data on the resolution interval</returns>
|
||||
public static BaseDataConsolidator FromResolution(Resolution resolution)
|
||||
{
|
||||
return new BaseDataConsolidator(resolution.ToTimeSpan());
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Creates a consolidator to produce a new 'TradeBar' representing the period
|
||||
/// </summary>
|
||||
/// <param name="period">The minimum span of time before emitting a consolidated bar</param>
|
||||
public BaseDataConsolidator(TimeSpan period)
|
||||
: base(period)
|
||||
{
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Creates a consolidator to produce a new 'TradeBar' representing the last count pieces of data
|
||||
/// </summary>
|
||||
/// <param name="maxCount">The number of pieces to accept before emitting a consolidated bar</param>
|
||||
public BaseDataConsolidator(int maxCount)
|
||||
: base(maxCount)
|
||||
{
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Creates a consolidator to produce a new 'TradeBar' representing the last count pieces of data or the period, whichever comes first
|
||||
/// </summary>
|
||||
/// <param name="maxCount">The number of pieces to accept before emitting a consolidated bar</param>
|
||||
/// <param name="period">The minimum span of time before emitting a consolidated bar</param>
|
||||
public BaseDataConsolidator(int maxCount, TimeSpan period)
|
||||
: base(maxCount, period)
|
||||
{
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Initializes a new instance of the <see cref="BaseDataConsolidator"/> class
|
||||
/// </summary>
|
||||
/// <param name="func">Func that defines the start time of a consolidated data</param>
|
||||
public BaseDataConsolidator(Func<DateTime, CalendarInfo> func)
|
||||
: base(func)
|
||||
{
|
||||
}
|
||||
|
||||
|
||||
/// <summary>
|
||||
/// Initializes a new instance of the <see cref="BaseDataConsolidator"/> class
|
||||
/// </summary>
|
||||
/// <param name="pyfuncobj">Func that defines the start time of a consolidated data</param>
|
||||
public BaseDataConsolidator(PyObject pyfuncobj)
|
||||
: base(pyfuncobj)
|
||||
{
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Aggregates the new 'data' into the 'workingBar'. The 'workingBar' will be
|
||||
/// null following the event firing
|
||||
/// </summary>
|
||||
/// <param name="workingBar">The bar we're building, null if the event was just fired and we're starting a new trade bar</param>
|
||||
/// <param name="data">The new data</param>
|
||||
protected override void AggregateBar(ref TradeBar workingBar, BaseData data)
|
||||
{
|
||||
if (workingBar == null)
|
||||
{
|
||||
workingBar = new TradeBar
|
||||
{
|
||||
Symbol = data.Symbol,
|
||||
Time = GetRoundedBarTime(data.Time),
|
||||
Close = data.Value,
|
||||
High = data.Value,
|
||||
Low = data.Value,
|
||||
Open = data.Value,
|
||||
DataType = data.DataType,
|
||||
Value = data.Value
|
||||
};
|
||||
}
|
||||
else
|
||||
{
|
||||
//Aggregate the working bar
|
||||
workingBar.Close = data.Value;
|
||||
if (data.Value < workingBar.Low) workingBar.Low = data.Value;
|
||||
if (data.Value > workingBar.High) workingBar.High = data.Value;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user