From b0651046a1b5032f1f93fb3ff624c36743c471c2 Mon Sep 17 00:00:00 2001 From: Denis Tereshkin Date: Mon, 17 Dec 2018 13:52:43 +0700 Subject: [PATCH] Draw bar-based equity --- src/nailab/execution/executor.py | 3 ++- src/nailab/ui/strategywidget.py | 6 +++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/src/nailab/execution/executor.py b/src/nailab/execution/executor.py index a1632db..36079d2 100644 --- a/src/nailab/execution/executor.py +++ b/src/nailab/execution/executor.py @@ -30,6 +30,7 @@ class Executor: results = strategy.get_analyzer('stats').get_result() trades = strategy.get_analyzer('tradeslist').get_result() - return (results, trades) + equity = strategy.get_analyzer('equity').get_result() + return (results, trades, equity) diff --git a/src/nailab/ui/strategywidget.py b/src/nailab/ui/strategywidget.py index 4a9fa9f..800a3f7 100644 --- a/src/nailab/ui/strategywidget.py +++ b/src/nailab/ui/strategywidget.py @@ -174,13 +174,13 @@ class StrategyWidget(QtWidgets.QWidget): def update_equity_chart(self): - pnl = [x['pnl'] for x in self.result[1]] - cumpnl = np.cumsum(pnl) + #pnl = [x['pnl'] for x in self.result[1]] + #cumpnl = np.cumsum(pnl) if self.equity_widget is None: self.equity_widget = EquityChartWidget(self) self.ui.tabs.addTab(self.equity_widget, "Equity") - self.equity_widget.set_data(cumpnl) + self.equity_widget.set_data(self.result[2]) def update_trades_list(self): if self.trades_widget is None: