]>
Commit | Line | Data |
---|---|---|
1 | ## | |
2 | ## This file is part of the sigrok-meter project. | |
3 | ## | |
4 | ## Copyright (C) 2015 Jens Steinhauser <jens.steinhauser@gmail.com> | |
5 | ## | |
6 | ## This program is free software; you can redistribute it and/or modify | |
7 | ## it under the terms of the GNU General Public License as published by | |
8 | ## the Free Software Foundation; either version 2 of the License, or | |
9 | ## (at your option) any later version. | |
10 | ## | |
11 | ## This program is distributed in the hope that it will be useful, | |
12 | ## but WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
14 | ## GNU General Public License for more details. | |
15 | ## | |
16 | ## You should have received a copy of the GNU General Public License | |
17 | ## along with this program; if not, see <http://www.gnu.org/licenses/>. | |
18 | ## | |
19 | ||
20 | import qtcompat | |
21 | ||
22 | QtCore = qtcompat.QtCore | |
23 | QtGui = qtcompat.QtGui | |
24 | pyqtgraph = qtcompat.pyqtgraph | |
25 | ||
26 | # Black foreground on white background. | |
27 | pyqtgraph.setConfigOption('background', 'w') | |
28 | pyqtgraph.setConfigOption('foreground', 'k') | |
29 | ||
30 | class Plot(object): | |
31 | '''Helper class to keep all graphics items of a plot together.''' | |
32 | ||
33 | def __init__(self, view, xaxis, yaxis): | |
34 | self.view = view | |
35 | self.xaxis = xaxis | |
36 | self.yaxis = yaxis | |
37 | self.visible = False | |
38 | ||
39 | class MultiPlotItem(pyqtgraph.GraphicsWidget): | |
40 | ||
41 | # Emitted when a plot is shown. | |
42 | plotShown = QtCore.Signal() | |
43 | ||
44 | # Emitted when a plot is hidden by the user via the context menu. | |
45 | plotHidden = QtCore.Signal(Plot) | |
46 | ||
47 | def __init__(self, parent=None): | |
48 | pyqtgraph.GraphicsWidget.__init__(self, parent) | |
49 | ||
50 | self.setLayout(QtGui.QGraphicsGridLayout()) | |
51 | self.layout().setContentsMargins(10, 10, 10, 1) | |
52 | self.layout().setHorizontalSpacing(0) | |
53 | self.layout().setVerticalSpacing(0) | |
54 | ||
55 | for i in range(2): | |
56 | self.layout().setColumnPreferredWidth(i, 0) | |
57 | self.layout().setColumnMinimumWidth(i, 0) | |
58 | self.layout().setColumnSpacing(i, 0) | |
59 | ||
60 | self.layout().setColumnStretchFactor(0, 0) | |
61 | self.layout().setColumnStretchFactor(1, 100) | |
62 | ||
63 | # List of 'Plot' objects that are shown. | |
64 | self._plots = [] | |
65 | ||
66 | self._hideActions = {} | |
67 | ||
68 | def addPlot(self): | |
69 | '''Adds and returns a new plot.''' | |
70 | ||
71 | row = self.layout().rowCount() | |
72 | ||
73 | view = pyqtgraph.ViewBox(parent=self) | |
74 | ||
75 | # If this is not the first plot, link to the axis of the previous one. | |
76 | if self._plots: | |
77 | view.setXLink(self._plots[-1].view) | |
78 | ||
79 | yaxis = pyqtgraph.AxisItem(parent=self, orientation='left') | |
80 | yaxis.linkToView(view) | |
81 | yaxis.setGrid(255) | |
82 | ||
83 | xaxis = pyqtgraph.AxisItem(parent=self, orientation='bottom') | |
84 | xaxis.linkToView(view) | |
85 | xaxis.setGrid(255) | |
86 | ||
87 | plot = Plot(view, xaxis, yaxis) | |
88 | self._plots.append(plot) | |
89 | ||
90 | self.showPlot(plot) | |
91 | ||
92 | # Create a separate action object for each plots context menu, so that | |
93 | # we can later find out which plot should be hidden by looking at | |
94 | # 'self._hideActions'. | |
95 | hideAction = QtGui.QAction('Hide', self) | |
96 | hideAction.triggered.connect(self._onHideActionTriggered) | |
97 | self._hideActions[id(hideAction)] = plot | |
98 | view.menu.insertAction(view.menu.actions()[0], hideAction) | |
99 | ||
100 | return plot | |
101 | ||
102 | def _rowNumber(self, plot): | |
103 | '''Returns the number of the first row a plot occupies.''' | |
104 | ||
105 | # Every plot takes up two rows. | |
106 | return 2 * self._plots.index(plot) | |
107 | ||
108 | @QtCore.Slot() | |
109 | def _onHideActionTriggered(self, checked=False): | |
110 | # The plot that we want to hide. | |
111 | plot = self._hideActions[id(self.sender())] | |
112 | self.hidePlot(plot) | |
113 | ||
114 | def hidePlot(self, plot): | |
115 | '''Hides 'plot'.''' | |
116 | ||
117 | # Only hiding wouldn't give up the space occupied by the items, | |
118 | # we have to remove them from the layout. | |
119 | self.layout().removeItem(plot.view) | |
120 | self.layout().removeItem(plot.xaxis) | |
121 | self.layout().removeItem(plot.yaxis) | |
122 | ||
123 | plot.view.hide() | |
124 | plot.xaxis.hide() | |
125 | plot.yaxis.hide() | |
126 | ||
127 | row = self._rowNumber(plot) | |
128 | self.layout().setRowStretchFactor(row, 0) | |
129 | self.layout().setRowStretchFactor(row + 1, 0) | |
130 | ||
131 | plot.visible = False | |
132 | self.plotHidden.emit(plot) | |
133 | ||
134 | def showPlot(self, plot): | |
135 | '''Adds the items of the plot to the scene's layout and makes | |
136 | them visible.''' | |
137 | ||
138 | if plot.visible: | |
139 | return | |
140 | ||
141 | row = self._rowNumber(plot) | |
142 | self.layout().addItem(plot.yaxis, row, 0, QtCore.Qt.AlignRight) | |
143 | self.layout().addItem(plot.view, row, 1) | |
144 | self.layout().addItem(plot.xaxis, row + 1, 1) | |
145 | ||
146 | plot.view.show() | |
147 | plot.xaxis.show() | |
148 | plot.yaxis.show() | |
149 | ||
150 | for i in range(row, row + 2): | |
151 | self.layout().setRowPreferredHeight(i, 0) | |
152 | self.layout().setRowMinimumHeight(i, 0) | |
153 | self.layout().setRowSpacing(i, 0) | |
154 | ||
155 | self.layout().setRowStretchFactor(row, 100) | |
156 | self.layout().setRowStretchFactor(row + 1, 0) | |
157 | ||
158 | plot.visible = True | |
159 | self.plotShown.emit() | |
160 | ||
161 | class MultiPlotWidget(pyqtgraph.GraphicsView): | |
162 | '''Widget that aligns multiple plots on top of each other. | |
163 | ||
164 | (The built in classes fail at doing this correctly when the axis grow, | |
165 | just try zooming in the "GraphicsLayout" or the "Linked View" examples.)''' | |
166 | ||
167 | def __init__(self, parent=None): | |
168 | pyqtgraph.GraphicsView.__init__(self, parent) | |
169 | ||
170 | self.multiPlotItem = MultiPlotItem() | |
171 | self.setCentralItem(self.multiPlotItem) | |
172 | ||
173 | for m in [ | |
174 | 'addPlot', | |
175 | 'hidePlot', | |
176 | 'showPlot' | |
177 | ]: | |
178 | setattr(self, m, getattr(self.multiPlotItem, m)) | |
179 | ||
180 | self.multiPlotItem.plotShown.connect(self._on_plotShown) | |
181 | ||
182 | # Expose the signal of the plot item. | |
183 | self.plotHidden = self.multiPlotItem.plotHidden | |
184 | ||
185 | def _on_plotShown(self): | |
186 | # This call is needed if only one plot exists and it was hidden, | |
187 | # without it the layout would start acting weird and not make the | |
188 | # MultiPlotItem fill the view widget after showing the plot again. | |
189 | self.resizeEvent(None) |