chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,141 @@
|
||||
/*
|
||||
* 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 Python.Runtime;
|
||||
using QuantConnect.Algorithm.Framework.Alphas;
|
||||
using QuantConnect.Algorithm.Framework.Execution;
|
||||
using QuantConnect.Algorithm.Framework.Portfolio;
|
||||
using QuantConnect.Algorithm.Framework.Risk;
|
||||
using QuantConnect.Algorithm.Framework.Selection;
|
||||
using QuantConnect.Util;
|
||||
|
||||
namespace QuantConnect.Algorithm
|
||||
{
|
||||
public partial class QCAlgorithm
|
||||
{
|
||||
/// <summary>
|
||||
/// Sets the alpha model
|
||||
/// </summary>
|
||||
/// <param name="alpha">Model that generates alpha</param>
|
||||
[DocumentationAttribute(AlgorithmFramework)]
|
||||
public void SetAlpha(PyObject alpha)
|
||||
{
|
||||
Alpha = PythonUtil.CreateInstanceOrWrapper<IAlphaModel>(
|
||||
alpha,
|
||||
py => new AlphaModelPythonWrapper(py)
|
||||
);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Adds a new alpha model
|
||||
/// </summary>
|
||||
/// <param name="alpha">Model that generates alpha to add</param>
|
||||
[DocumentationAttribute(AlgorithmFramework)]
|
||||
public void AddAlpha(PyObject alpha)
|
||||
{
|
||||
var model = PythonUtil.CreateInstanceOrWrapper<IAlphaModel>(
|
||||
alpha,
|
||||
py => new AlphaModelPythonWrapper(py)
|
||||
);
|
||||
AddAlpha(model);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Sets the execution model
|
||||
/// </summary>
|
||||
/// <param name="execution">Model defining how to execute trades to reach a portfolio target</param>
|
||||
[DocumentationAttribute(AlgorithmFramework)]
|
||||
[DocumentationAttribute(TradingAndOrders)]
|
||||
public void SetExecution(PyObject execution)
|
||||
{
|
||||
Execution = PythonUtil.CreateInstanceOrWrapper<IExecutionModel>(
|
||||
execution,
|
||||
py => new ExecutionModelPythonWrapper(py)
|
||||
);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Sets the portfolio construction model
|
||||
/// </summary>
|
||||
/// <param name="portfolioConstruction">Model defining how to build a portfolio from alphas</param>
|
||||
[DocumentationAttribute(AlgorithmFramework)]
|
||||
[DocumentationAttribute(TradingAndOrders)]
|
||||
public void SetPortfolioConstruction(PyObject portfolioConstruction)
|
||||
{
|
||||
PortfolioConstruction = PythonUtil.CreateInstanceOrWrapper<IPortfolioConstructionModel>(
|
||||
portfolioConstruction,
|
||||
py => new PortfolioConstructionModelPythonWrapper(py)
|
||||
);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Sets the universe selection model
|
||||
/// </summary>
|
||||
/// <param name="universeSelection">Model defining universes for the algorithm</param>
|
||||
[DocumentationAttribute(AlgorithmFramework)]
|
||||
[DocumentationAttribute(Universes)]
|
||||
public void SetUniverseSelection(PyObject universeSelection)
|
||||
{
|
||||
UniverseSelection = PythonUtil.CreateInstanceOrWrapper<IUniverseSelectionModel>(
|
||||
universeSelection,
|
||||
py => new UniverseSelectionModelPythonWrapper(py)
|
||||
);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Adds a new universe selection model
|
||||
/// </summary>
|
||||
/// <param name="universeSelection">Model defining universes for the algorithm to add</param>
|
||||
[DocumentationAttribute(AlgorithmFramework)]
|
||||
[DocumentationAttribute(Universes)]
|
||||
public void AddUniverseSelection(PyObject universeSelection)
|
||||
{
|
||||
var model = PythonUtil.CreateInstanceOrWrapper<IUniverseSelectionModel>(
|
||||
universeSelection,
|
||||
py => new UniverseSelectionModelPythonWrapper(py)
|
||||
);
|
||||
AddUniverseSelection(model);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Sets the risk management model
|
||||
/// </summary>
|
||||
/// <param name="riskManagement">Model defining how risk is managed</param>
|
||||
[DocumentationAttribute(AlgorithmFramework)]
|
||||
[DocumentationAttribute(TradingAndOrders)]
|
||||
public void SetRiskManagement(PyObject riskManagement)
|
||||
{
|
||||
RiskManagement = PythonUtil.CreateInstanceOrWrapper<IRiskManagementModel>(
|
||||
riskManagement,
|
||||
py => new RiskManagementModelPythonWrapper(py)
|
||||
);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Adds a new risk management model
|
||||
/// </summary>
|
||||
/// <param name="riskManagement">Model defining how risk is managed to add</param>
|
||||
[DocumentationAttribute(AlgorithmFramework)]
|
||||
[DocumentationAttribute(TradingAndOrders)]
|
||||
public void AddRiskManagement(PyObject riskManagement)
|
||||
{
|
||||
var model = PythonUtil.CreateInstanceOrWrapper<IRiskManagementModel>(
|
||||
riskManagement,
|
||||
py => new RiskManagementModelPythonWrapper(py)
|
||||
);
|
||||
AddRiskManagement(model);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user