Reset orders on chart after multiple strat runs

Skinok · Skinok · commit 09dedf0f8374 · 2021-12-01T14:45:35.000+01:00
diff --git a/finplotWindow.py b/finplotWindow.py
@@ -148,7 +148,7 @@ def drawOrders(self, orders = None):
                             elif order.executed.pnl < 0:
                                 color = "#FF3030"
 
-                            fplt.add_line(posOpen, posClose, color, 2, style="--" )
+                            fplt.add_line(posOpen, posClose, color, 2, ax=self.ax0, style="--" )
                             
                             pass
 
@@ -300,6 +300,13 @@ def setChartData(self, data):
         self.data = data
         pass
 
+    def resetChart(self):
+
+        # Remove all previous orders arrows and informations
+        self.orders = []
+
+        pass
+
     def updateChart(self):
 
         # Entirely reset graph
diff --git a/userInterface.py b/userInterface.py
@@ -392,7 +392,7 @@ def initialize(self):
 
         # Enable run button
         self.strategyTesterUI.runBacktestPB.setEnabled(False)
-        
+
         self.strategyTesterUI.initialize()
 
         pass
@@ -517,10 +517,8 @@ def dark_mode_toggle(self):
     # INDICATORS
     ##########
     def resetChart(self):
+        self.fpltWindow.resetChart()
         self.fpltWindow.updateChart()
-        #self.fpltWindow.resetPlots()
-        #self.fpltWindow.drawCandles()
-        #self.fpltWindow.refreshChart()
         pass
 
     def addIchimoku(self):
