Revize 4e32720c

Re: #8968 - second function plotting implemented

aswi2021vochomurka/view/main_view.py

@@ -4 +4 @@
 from PyQt5 import QtCore
 from PyQt5 import QtGui
 from PyQt5.QtCore import QSize, QThread, QObject, pyqtSignal, QSettings
-from PyQt5.QtWidgets import QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, QScrollArea, QGridLayout
+from PyQt5.QtWidgets import QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, QScrollArea, QGridLayout, QFileDialog
 from PyQt5.QtWidgets import QMenuBar, QAction, QPushButton
 from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
 
@@ -97 +97 @@
 
         self.chartsNum = 0
         self.dataDict = {}
+        self.dataDict2 = {}
         self.canvasDict = {}
         self.figureDict = {}
         self.widgetDict = {}
@@ -160 +161 @@
             figure.suptitle(message.topic)
             plt.plot(self.dataDict[message.topic])
 
+            if message.topic in self.dataDict2:
+                plt.plot(self.dataDict2[message.topic])
+
             self.canvasDict[message.topic].draw()
         else:
             # new topic
@@ -179 +183 @@
             self.widgetDict[message.topic] = widget
             self.widgetList.append(widget)
             widget.setLayout(self.layout)
-            button = QPushButton(':')
+            button = QPushButton('Load')
+            button.clicked.connect(lambda: self.getFile(message.topic))
             button.setFixedSize(QSize(40, 40))
+
             self.layout.addWidget(canvas)
             self.layout.addWidget(button)
             self.layout.setAlignment(button, QtCore.Qt.AlignTop)
@@ -190 +196 @@
 
             self.chartsNum += 1
 
+    def getFile(self, topic: str):
+        fname = QFileDialog.getOpenFileName(self, 'Open file',
+                                            'c:\\', "CSV files (*.csv)")
+        try:
+            figure = self.figureDict[topic]
+            figure.clear()
+
+            self.dataDict2[topic] = []
+
+            file1 = open(fname[0], 'r')
+            lines = file1.readlines()
+
+            count = 0
+            for line in lines:
+                count += 1
+                parts = line.split(';')
+                value = float(parts[3])
+                self.dataDict2[topic].append(value)
+
+            figure = plt.figure(figure.number)
+            figure.suptitle(topic)
+            plt.plot(self.dataDict[topic])
+            plt.plot(self.dataDict2[topic])
+
+            self.canvasDict[topic].draw()
+        except:
+            print("Loading error")
+
     def deletePlot(self, topic: str):
         """
         Deletes plot