srf_archiver_time_plot/plot.py at main · slaclab/srf_archiver_time_plot · 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
59
60
61
62
63
64
65
66
67
68
69
from typing import Dict

from PyQt5.QtWidgets import QCheckBox
from pydm import Display

from lcls_tools.superconducting.sc_linac import MACHINE
from lcls_tools.superconducting.sc_linac_utils import ALL_CRYOMODULES
from pydm.widgets import PyDMTimePlot


class Plot(Display):
    def ui_filename(self):
        return "timeplot.ui"

    def __init__(self, parent=None, args=None):
        super().__init__(parent=parent, args=args, ui_filename="timeplot.ui")

        self.ui.cm_combobox.addItems(ALL_CRYOMODULES)
        self.ui.suffix_line_edit.returnPressed.connect(self.update)
        self.ui.cm_combobox.currentIndexChanged.connect(self.update)
        self.ui.second_spinbox.valueChanged.connect(self.update_time)
        self.ui.plot.setShowLegend(True)
        self.ui.autoscale_checkbox.stateChanged.connect(self.update)
        self.ui.ymin_spinbox.valueChanged.connect(self.update)
        self.ui.ymax_spinbox.valueChanged.connect(self.update)

        self.cavity_checkbox_map: Dict[QCheckBox, int] = {
            self.ui.c1_checkbox: 1,
            self.ui.c2_checkbox: 2,
            self.ui.c3_checkbox: 3,
            self.ui.c4_checkbox: 4,
            self.ui.c5_checkbox: 5,
            self.ui.c6_checkbox: 6,
            self.ui.c7_checkbox: 7,
            self.ui.c8_checkbox: 8,
        }

        for checkbox in self.cavity_checkbox_map.keys():
            checkbox.stateChanged.connect(self.update)

    @property
    def selected_cavities(self):
        selected_cavities = []

        for checkbox, cav_num in self.cavity_checkbox_map.items():
            if checkbox.isChecked():
                selected_cavities.append(cav_num)

        return selected_cavities

    def update_time(self):
        self.ui.plot.setTimeSpan(self.ui.second_spinbox.value())

    def update(self):
        plot: PyDMTimePlot = self.ui.plot
        plot.clearCurves()

        cm_obj = MACHINE.cryomodules[self.ui.cm_combobox.currentText()]

        for cav_num in self.selected_cavities:
            cavity = cm_obj.cavities[cav_num]
            self.ui.plot.addYChannel(
                y_channel=cavity.pv_addr(self.ui.suffix_line_edit.text()),
                useArchiveData=True,
            )

        self.ui.plot.setAutoRangeY(self.ui.autoscale_checkbox.isChecked())
        self.ui.plot.setMinYRange(self.ui.ymin_spinbox.value())
        self.ui.plot.setMaxYRange(self.ui.ymax_spinbox.value())