Draw bar-based equity

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)
 
 

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: