mirror of https://github.com/m-labs/artiq
55 lines
1.5 KiB
Python
Executable File
55 lines
1.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import PyQt5 # make sure pyqtgraph imports Qt5
|
|
from PyQt5.QtCore import QTimer
|
|
import pyqtgraph
|
|
|
|
from artiq.applets.simple import TitleApplet
|
|
|
|
|
|
class HistogramPlot(pyqtgraph.PlotWidget):
|
|
def __init__(self, args, req):
|
|
pyqtgraph.PlotWidget.__init__(self)
|
|
self.args = args
|
|
self.timer = QTimer()
|
|
self.timer.setSingleShot(True)
|
|
self.timer.timeout.connect(self.length_warning)
|
|
|
|
def data_changed(self, value, metadata, persist, mods, title):
|
|
try:
|
|
y = value[self.args.y]
|
|
if self.args.x is None:
|
|
x = None
|
|
else:
|
|
x = value[self.args.x]
|
|
except KeyError:
|
|
return
|
|
if x is None:
|
|
x = list(range(len(y)+1))
|
|
|
|
if len(y) and len(x) == len(y) + 1:
|
|
self.timer.stop()
|
|
self.clear()
|
|
self.plot(x, y, stepMode=True, fillLevel=0,
|
|
brush=(0, 0, 255, 150))
|
|
self.setTitle(title)
|
|
else:
|
|
if not self.timer.isActive():
|
|
self.timer.start(1000)
|
|
|
|
def length_warning(self):
|
|
self.clear()
|
|
text = "⚠️ dataset lengths mismatch:\n"\
|
|
"There should be one more bin boundaries than there are Y values"
|
|
self.addItem(pyqtgraph.TextItem(text))
|
|
|
|
|
|
def main():
|
|
applet = TitleApplet(HistogramPlot)
|
|
applet.add_dataset("y", "Y values")
|
|
applet.add_dataset("x", "Bin boundaries", required=False)
|
|
applet.run()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|