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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
from silx.gui.plot.PlotWindow import PlotWindow
from silx.gui.plot.PlotWidget import PlotWidget
from .. import qt
class RecordPlot(PlotWindow):
def __init__(self, parent=None, backend=None):
super(RecordPlot, self).__init__(parent=parent, backend=backend,
resetzoom=True, autoScale=True,
logScale=True, grid=True,
curveStyle=True, colormap=False,
aspectRatio=False, yInverted=False,
copy=True, save=True, print_=True,
control=True, position=True,
roi=True, mask=False, fit=True)
if parent is None:
self.setWindowTitle('RecordPlot')
self._axesSelectionToolBar = AxesSelectionToolBar(parent=self, plot=self)
self.addToolBar(qt.Qt.BottomToolBarArea, self._axesSelectionToolBar)
def setXAxisFieldName(self, value):
"""Set the current selected field for the X axis.
:param Union[str,None] value:
"""
label = '' if value is None else value
index = self._axesSelectionToolBar.getXAxisDropDown().findData(value)
if index >= 0:
self.getXAxis().setLabel(label)
self._axesSelectionToolBar.getXAxisDropDown().setCurrentIndex(index)
def getXAxisFieldName(self):
"""Returns currently selected field for the X axis or None.
rtype: Union[str,None]
"""
return self._axesSelectionToolBar.getXAxisDropDown().currentData()
def setYAxisFieldName(self, value):
self.getYAxis().setLabel(value)
index = self._axesSelectionToolBar.getYAxisDropDown().findText(value)
if index >= 0:
self._axesSelectionToolBar.getYAxisDropDown().setCurrentIndex(index)
def getYAxisFieldName(self):
return self._axesSelectionToolBar.getYAxisDropDown().currentText()
def setSelectableXAxisFieldNames(self, fieldNames):
"""Add list of field names to X axis
:param List[str] fieldNames:
"""
comboBox = self._axesSelectionToolBar.getXAxisDropDown()
comboBox.clear()
comboBox.addItem('-', None)
comboBox.insertSeparator(1)
for name in fieldNames:
comboBox.addItem(name, name)
def setSelectableYAxisFieldNames(self, fieldNames):
self._axesSelectionToolBar.getYAxisDropDown().clear()
self._axesSelectionToolBar.getYAxisDropDown().addItems(fieldNames)
def getAxesSelectionToolBar(self):
return self._axesSelectionToolBar
class AxesSelectionToolBar(qt.QToolBar):
def __init__(self, parent=None, plot=None, title='Plot Axes Selection'):
super(AxesSelectionToolBar, self).__init__(title, parent)
assert isinstance(plot, PlotWidget)
self.addWidget(qt.QLabel("Field selection: "))
self._labelXAxis = qt.QLabel(" X: ")
self.addWidget(self._labelXAxis)
self._selectXAxisDropDown = qt.QComboBox()
self.addWidget(self._selectXAxisDropDown)
self._labelYAxis = qt.QLabel(" Y: ")
self.addWidget(self._labelYAxis)
self._selectYAxisDropDown = qt.QComboBox()
self.addWidget(self._selectYAxisDropDown)
def getXAxisDropDown(self):
return self._selectXAxisDropDown
def getYAxisDropDown(self):
return self._selectYAxisDropDown
|
