PeriphNeuronStim/plot_widget.py at master · TheJRapp/PeriphNeuronStim · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
from PyQt5.uic import loadUiType
from PyQt5 import uic, QtGui, QtWidgets
from PyQt5.QtCore import pyqtSignal

import matplotlib
matplotlib.use('Qt5Agg')
from matplotlib.figure import Figure
from matplotlib.backends.backend_qt5agg import (
    FigureCanvasQTAgg as FigureCanvas,
    NavigationToolbar2QT as NavigationToolbar)
from matplotlib import pyplot as plt

Ui_PlotWidget, QWidget_Plot = uic.loadUiType("ui_plot_widget.ui")

class CustomFigure():
    def __init__(self, figure, label):
        self.fig = figure
        self.label = label


class PlotWidget(QWidget_Plot, Ui_PlotWidget):
    plot_requested = pyqtSignal()

    def __init__(self, parent = None):
        super(PlotWidget, self).__init__(parent)
        self.setupUi(self)

        self.figure_list = []
        self.plot_list_item_model = QtGui.QStandardItemModel(self.plot_list_view)
        self.show_plot_button.clicked.connect(self.plot_figure)

    def add_figure(self, figure, label):
        for i in range(len(self.figure_list)):
            if self.figure_list[i].label == label:
                self.figure_list[i].fig = figure
                self.update_plot_list()
                return
        self.figure_list.append(CustomFigure(figure, label))
        self.update_plot_list()

    def clear_figures(self):
        self.figure_list = []
        self.update_plot_list()

    def plot_figure(self):
        self.plot_requested.emit()

    def get_plot(self):
        current_index = self.plot_list_view.currentIndex().row()
        if self.figure_list:
            return self.figure_list[current_index].fig

    def update_plot_list(self):
        self.plot_list_item_model.clear()
        for custom_fig in self.figure_list:
            item = QtGui.QStandardItem(custom_fig.label)
            self.plot_list_item_model.appendRow(item)
        self.plot_list_view.setModel(self.plot_list_item_model)