def __init__(self, parent):
super(MakeNicerWidget, self).__init__(parent)
self.data = np.array([]) # array which holds data
# Time domain plot
self.tdWidget = DockablePlotWidget(self, CurveWidget)
self.tdWidget.calcFun.addFun('fs', lambda x: x, lambda x: x)
self.tdWidget.calcFun.addFun('µm', lambda x: x * fsDelay * 1e3,
lambda x: x / fsDelay * 1e-3)
self.tdWidget.calcFun.addFun('mm', lambda x: x * fsDelay,
lambda x: x / fsDelay)
tdPlot = self.tdWidget.get_plot()
self.tdSignal = SignalFT(self, plot=tdPlot)
self.tdFit = SignalFT(self, plot=tdPlot, col='r')
# Frequency domain plot
self.fdWidget = DockablePlotWidget(self, CurveWidget)
self.fdWidget.calcFun.addFun('PHz', lambda x: x, lambda x: x)
self.fdWidget.calcFun.addFun('THz', lambda x: x * 1e3,
lambda x: x * 1e-3)
self.fdWidget.calcFun.addFun('µm', lambda x: c0 / x * 1e-9,
lambda x: c0 / x * 1e-9)
self.fdWidget.calcFun.addFun('eV', lambda x: x, lambda x: x)
fdplot = self.fdWidget.get_plot()
self.fdSignal = SignalFT(self, plot=fdplot)
self.fdFit = SignalFT(self, plot=fdplot, col='r')
self.smoothNum = QSpinBox() # gives number of smoothin points
self.smoothNum.setMinimum(1)
self.smoothNum.setSingleStep(2)
# Put things together in layouts
buttonLayout = QGridLayout()
plotLayout = QVBoxLayout()
layout = QHBoxLayout()
plotLayout.addWidget(self.tdWidget)
plotLayout.addWidget(self.fdWidget)
buttonLayout.addWidget(QLabel('Fitting function'), 0, 0)
buttonLayout.addWidget(
QLineEdit("lambda x,A,f,phi: np.sin(2*np.pi*f*x+phi)"), 1, 0, 1, 2)
buttonLayout.addWidget(QLabel('Smooth'), 2, 0)
buttonLayout.addWidget(self.smoothNum, 2, 1)
buttonLayout.setRowStretch(3, 20)
layout.addLayout(buttonLayout)
layout.addLayout(plotLayout)
self.setLayout(layout)
# connect signals
self.updateTdPlot.connect(self.tdSignal.updatePlot)
self.updateTdFitPlot.connect(self.tdFit.updatePlot)
self.updateFdPlot.connect(lambda data: self.fdSignal.updatePlot(
self.fdSignal.computeFFT(data)))
self.updateFdFitPlot.connect(
lambda data: self.fdFit.updatePlot(self.fdFit.computeFFT(data)))
self.smoothNum.valueChanged.connect(self.smoothData)
self.setData()
class MakeNicerWidget(DockableTabWidget):
LOCATION = Qt.LeftDockWidgetArea
updateTdPlot = Signal(object)
updateTdFitPlot = Signal(object)
updateFdPlot = Signal(object)
updateFdFitPlot = Signal(object)
def __init__(self, parent):
super(MakeNicerWidget, self).__init__(parent)
self.data = np.array([]) # array which holds data
# Time domain plot
self.tdWidget = DockablePlotWidget(self, CurveWidget)
self.tdWidget.calcFun.addFun('fs', lambda x: x,
lambda x: x)
self.tdWidget.calcFun.addFun('µm', lambda x: x*fsDelay*1e3,
lambda x: x/fsDelay*1e-3)
self.tdWidget.calcFun.addFun('mm', lambda x: x*fsDelay,
lambda x: x/fsDelay)
tdPlot = self.tdWidget.get_plot()
self.tdSignal = SignalFT(self, plot=tdPlot)
self.tdFit = SignalFT(self, plot=tdPlot, col='r')
# Frequency domain plot
self.fdWidget = DockablePlotWidget(self, CurveWidget)
self.fdWidget.calcFun.addFun('PHz', lambda x: x,
lambda x: x)
self.fdWidget.calcFun.addFun('THz', lambda x: x*1e3,
lambda x: x*1e-3)
self.fdWidget.calcFun.addFun('µm', lambda x: c0/x*1e-9,
lambda x: c0/x*1e-9)
self.fdWidget.calcFun.addFun('eV', lambda x: x,
lambda x: x)
fdplot = self.fdWidget.get_plot()
self.fdSignal = SignalFT(self, plot=fdplot)
self.fdFit = SignalFT(self, plot=fdplot, col='r')
self.smoothNum = QSpinBox() # gives number of smoothin points
self.smoothNum.setMinimum(1)
self.smoothNum.setSingleStep(2)
# Put things together in layouts
buttonLayout = QGridLayout()
plotLayout = QVBoxLayout()
layout = QHBoxLayout()
plotLayout.addWidget(self.tdWidget)
plotLayout.addWidget(self.fdWidget)
buttonLayout.addWidget(QLabel('Fitting function'), 0, 0)
buttonLayout.addWidget(
QLineEdit("lambda x,A,f,phi: np.sin(2*np.pi*f*x+phi)"), 1, 0, 1, 2)
buttonLayout.addWidget(QLabel('Smooth'), 2, 0)
buttonLayout.addWidget(self.smoothNum, 2, 1)
buttonLayout.setRowStretch(3, 20)
layout.addLayout(buttonLayout)
layout.addLayout(plotLayout)
self.setLayout(layout)
# connect signals
self.updateTdPlot.connect(self.tdSignal.updatePlot)
self.updateTdFitPlot.connect(self.tdFit.updatePlot)
self.updateFdPlot.connect(lambda data:
self.fdSignal.updatePlot(self.fdSignal.computeFFT(data)))
self.updateFdFitPlot.connect(lambda data:
self.fdFit.updatePlot(self.fdFit.computeFFT(data)))
self.smoothNum.valueChanged.connect(self.smoothData)
self.setData()
def setData(self):
data = np.loadtxt('testData.txt', delimiter=',')
data[:,1] = (data[:,1]-data[:,1].min())/(data[:,1]-data[:,1].min()).max()
self.data = data
self.updateTdPlot.emit(data)
self.updateFdPlot.emit(data)
def smoothData(self, i):
x = self.data[:,0]
x = np.linspace(x[0], x[-1], x.shape[0]+i-1) # get x axis for smooth
y = self.data[:,1]
self.updateTdPlot.emit(np.column_stack((x, self.tdSignal.smooth(y, i))))
self.updateFdPlot.emit(self.fdSignal.computeFFT(
np.column_stack((x, self.fdSignal.smooth(y, i)))))
def runFitDialog(self):
x = self.plot4Curve.get_data()[0]
y = self.plot4Curve.get_data()[1]
def fit(x, params):
y0, a, x0, s, tau= params
return y0+a*0.5*(erf((x-x0)/(s/(2*np.sqrt(2*np.log(2)))))+1)*np.exp(-(x-x0)/tau)
y0 = FitParam("Offset", 0., -100., 100.)
a = FitParam("Amplitude", y.max(), 0., 10000.)
x0 = FitParam("x0", x[y.argmax()], -10., 10.)
s = FitParam("FWHM", 0.5, 0., 10.)
tau = FitParam("tau", 0.5, 0., 10.)
params = [y0, a, x0, s, tau]
values = guifit(x, y, fit, params, xlabel="Time (s)", ylabel="Power (a.u.)")
self.fitParam = values
def __init__(self):
QMainWindow.__init__(self)
self.stage = None
self.stopOsci = False
self.stopMeasure = False
self.setWindowTitle(APP_NAME)
###############
# Osci live
curveplot_toolbar = self.addToolBar(_("Curve Plotting Toolbar"))
self.osciCurveWidget = DockablePlotWidget(self, CurveWidget,
curveplot_toolbar)
self.osciCurveWidget.calcFun.addFun('s', lambda x: x,
lambda x: x)
self.osciCurveWidget.calcFun.addFun('ms', lambda x: x*1e3,
lambda x: x*1e-3)
self.osciCurveWidget.calcFun.addFun('µs', lambda x: x*1e6,
lambda x: x*1e-6)
osciPlot = self.osciCurveWidget.get_plot()
#osciPlot.set_axis_title('bottom', 'Time (s)')
#osciplot.add_item(make.legend("TR"))
self.osciSignal = SignalFT(self, plot=osciPlot)
self.osciSignal.addHCursor(0)
self.osciSignal.addBounds()
##############
# Time domain plot
self.tdWidget = DockablePlotWidget(self, CurveWidget,
curveplot_toolbar)
self.tdWidget.calcFun.addFun('fs', lambda x: x,
lambda x: x)
self.tdWidget.calcFun.addFun('µm', lambda x: x*fsDelay*1e3,
lambda x: x/fsDelay*1e-3)
self.tdWidget.calcFun.addFun('mm', lambda x: x*fsDelay,
lambda x: x/fsDelay)
tdPlot = self.tdWidget.get_plot()
#tdPlot.add_item(make.legend("TR"))
self.tdSignal = SignalFT(self, plot=tdPlot)
self.tdSignal.addVCursor(0)
##################
# Frequency domain plot
self.fdWidget = DockablePlotWidget(self, CurveWidget,
curveplot_toolbar)
self.fdWidget.calcFun.addFun('PHz', lambda x: x,
lambda x: x)
self.fdWidget.calcFun.addFun('THz', lambda x: x*1e3,
lambda x: x*1e-3)
# x = PHz -> 1e15, µm = 1e-6
self.fdWidget.calcFun.addFun('µm', lambda x: c0/x*1e-9,
lambda x: c0/x*1e-9)
self.fdWidget.calcFun.addFun('eV', lambda x: x,
lambda x: x)
fdplot = self.fdWidget.get_plot()
#fqdplot.add_item(make.legend("TR"))
self.fdSignal = SignalFT(self, plot=fdplot)
##############
# Main window widgets
self.tabwidget = DockableTabWidget(self)
#self.tabwidget.setMaximumWidth(500)
self.tiepieUi = TiePieUi(self)
self.piUi = PiStageUi(self)
#self.stage = self.piUi.stage
self.tabwidget.addTab(self.tiepieUi, QIcon('icons/Handyscope_HS4.png'),
_("Osci"))
self.tabwidget.addTab(self.piUi, QIcon('icons/piController.png'),
_("Stage"))
self.add_dockwidget(self.tabwidget, _("Inst. sett."))
# self.setCentralWidget(self.tabwidget)
self.osci_dock = self.add_dockwidget(self.osciCurveWidget,
title=_("Osciloscope"))
self.td_dock = self.add_dockwidget(self.tdWidget,
title=_("Time Domain"))
self.fd_dock = self.add_dockwidget(self.fdWidget,
title=_("Frequency Domain"))
################
# connect signals
self.piUi.startScanBtn.released.connect(self.startMeasureThr)
self.piUi.stopScanBtn.released.connect(self.stopMeasureThr)
self.piUi.xAxeChanged.connect(self.tdSignal.updateXAxe)
self.piUi.xAxeChanged.connect(self.fdSignal.updateXAxe)
self.piUi.niceBtn.released.connect(self.showMakeNicerWidget)
self.tiepieUi.scpConnected.connect(self.startOsciThr)
self.tiepieUi.xAxeChanged.connect(self.osciSignal.updateXAxe)
self.tiepieUi.yAxeChanged.connect(self.osciSignal.updateYAxe)
self.tiepieUi.triggLevelChanged.connect(self.osciSignal.setHCursor)
#self.piUi.centerBtn.released.connect(
# lambda x=None: self.piUi.setCenter(self.tdSignal.getVCursor()))
self.tdSignal.plot.SIG_MARKER_CHANGED.connect(
lambda x=None: self.piUi.newOffset(self.tdSignal.getVCursor()))
self.osciCurveWidget.calcFun.idxChanged.connect(self.osciSignal.funChanged)
self.tdWidget.calcFun.idxChanged.connect(self.tdSignal.funChanged)
self.fdWidget.calcFun.idxChanged.connect(self.fdSignal.funChanged)
self.updateOsciPlot.connect(self.osciSignal.updatePlot)
self.updateTdPlot.connect(self.tdSignal.updatePlot)
self.updateFdPlot.connect(lambda data:
self.fdSignal.updatePlot(self.fdSignal.computeFFT(data)))
################
# create threads
#self.osciThr = GenericThread(self.getOsciData)
self.osciThr = QThread()
self.osciThr.start()
self.osciWorker = GenericWorker(self.getOsciData)
self.osciWorker.moveToThread(self.osciThr)
#self.measureThr = GenericThread(self.getMeasureData)
self.measureThr = QThread()
self.measureThr.start()
self.measureWorker = GenericWorker(self.getMeasureData)
self.measureWorker.moveToThread(self.measureThr)
################
# File menu
file_menu = self.menuBar().addMenu(_("File"))
self.quit_action = create_action(self, _("Quit"), shortcut="Ctrl+Q",
icon=get_std_icon("DialogCloseButton"),
tip=_("Quit application"),
triggered=self.close)
saveData = create_action(self, _("Save"), shortcut="Ctrl+S",
icon=get_std_icon("DialogSaveButton"),
tip=_("Save data"),
triggered=self.saveData)
triggerTest_action = create_action(self, _("Stop Osci"),
shortcut="Ctrl+O",
icon=get_icon('fileopen.png'),
tip=_("Open an image"),
triggered=self.stopOsciThr)
add_actions(file_menu, (triggerTest_action, saveData, None, self.quit_action))
##############
# Eventually add an internal console (requires 'spyderlib')
self.sift_proxy = SiftProxy(self)
if DockableConsole is None:
self.console = None
else:
import time, scipy.signal as sps, scipy.ndimage as spi
ns = {'ftir': self.sift_proxy,
'np': np, 'sps': sps, 'spi': spi,
'os': os, 'sys': sys, 'osp': osp, 'time': time}
msg = "Example: ftir.s[0] returns signal object #0\n"\
"Modules imported at startup: "\
"os, sys, os.path as osp, time, "\
"numpy as np, scipy.signal as sps, scipy.ndimage as spi"
self.console = DockableConsole(self, namespace=ns, message=msg)
self.add_dockwidget(self.console, _("Console"))
'''
try:
self.console.interpreter.widget_proxy.sig_new_prompt.connect(
lambda txt: self.refresh_lists())
except AttributeError:
print('sift: spyderlib is outdated', file=sys.stderr)
'''
# Show main window and raise the signal plot panel
self.show()
class MainWindow(QMainWindow):
updateOsciPlot = Signal(object)
updateTdPlot = Signal(object)
updateFdPlot = Signal(object)
def __init__(self):
QMainWindow.__init__(self)
self.stage = None
self.stopOsci = False
self.stopMeasure = False
self.setWindowTitle(APP_NAME)
###############
# Osci live
curveplot_toolbar = self.addToolBar(_("Curve Plotting Toolbar"))
self.osciCurveWidget = DockablePlotWidget(self, CurveWidget,
curveplot_toolbar)
self.osciCurveWidget.calcFun.addFun('s', lambda x: x,
lambda x: x)
self.osciCurveWidget.calcFun.addFun('ms', lambda x: x*1e3,
lambda x: x*1e-3)
self.osciCurveWidget.calcFun.addFun('µs', lambda x: x*1e6,
lambda x: x*1e-6)
osciPlot = self.osciCurveWidget.get_plot()
#osciPlot.set_axis_title('bottom', 'Time (s)')
#osciplot.add_item(make.legend("TR"))
self.osciSignal = SignalFT(self, plot=osciPlot)
self.osciSignal.addHCursor(0)
self.osciSignal.addBounds()
##############
# Time domain plot
self.tdWidget = DockablePlotWidget(self, CurveWidget,
curveplot_toolbar)
self.tdWidget.calcFun.addFun('fs', lambda x: x,
lambda x: x)
self.tdWidget.calcFun.addFun('µm', lambda x: x*fsDelay*1e3,
lambda x: x/fsDelay*1e-3)
self.tdWidget.calcFun.addFun('mm', lambda x: x*fsDelay,
lambda x: x/fsDelay)
tdPlot = self.tdWidget.get_plot()
#tdPlot.add_item(make.legend("TR"))
self.tdSignal = SignalFT(self, plot=tdPlot)
self.tdSignal.addVCursor(0)
##################
# Frequency domain plot
self.fdWidget = DockablePlotWidget(self, CurveWidget,
curveplot_toolbar)
self.fdWidget.calcFun.addFun('PHz', lambda x: x,
lambda x: x)
self.fdWidget.calcFun.addFun('THz', lambda x: x*1e3,
lambda x: x*1e-3)
# x = PHz -> 1e15, µm = 1e-6
self.fdWidget.calcFun.addFun('µm', lambda x: c0/x*1e-9,
lambda x: c0/x*1e-9)
self.fdWidget.calcFun.addFun('eV', lambda x: x,
lambda x: x)
fdplot = self.fdWidget.get_plot()
#fqdplot.add_item(make.legend("TR"))
self.fdSignal = SignalFT(self, plot=fdplot)
##############
# Main window widgets
self.tabwidget = DockableTabWidget(self)
#self.tabwidget.setMaximumWidth(500)
self.tiepieUi = TiePieUi(self)
self.piUi = PiStageUi(self)
#self.stage = self.piUi.stage
self.tabwidget.addTab(self.tiepieUi, QIcon('icons/Handyscope_HS4.png'),
_("Osci"))
self.tabwidget.addTab(self.piUi, QIcon('icons/piController.png'),
_("Stage"))
self.add_dockwidget(self.tabwidget, _("Inst. sett."))
# self.setCentralWidget(self.tabwidget)
self.osci_dock = self.add_dockwidget(self.osciCurveWidget,
title=_("Osciloscope"))
self.td_dock = self.add_dockwidget(self.tdWidget,
title=_("Time Domain"))
self.fd_dock = self.add_dockwidget(self.fdWidget,
title=_("Frequency Domain"))
################
# connect signals
self.piUi.startScanBtn.released.connect(self.startMeasureThr)
self.piUi.stopScanBtn.released.connect(self.stopMeasureThr)
self.piUi.xAxeChanged.connect(self.tdSignal.updateXAxe)
self.piUi.xAxeChanged.connect(self.fdSignal.updateXAxe)
self.piUi.niceBtn.released.connect(self.showMakeNicerWidget)
self.tiepieUi.scpConnected.connect(self.startOsciThr)
self.tiepieUi.xAxeChanged.connect(self.osciSignal.updateXAxe)
self.tiepieUi.yAxeChanged.connect(self.osciSignal.updateYAxe)
self.tiepieUi.triggLevelChanged.connect(self.osciSignal.setHCursor)
#self.piUi.centerBtn.released.connect(
# lambda x=None: self.piUi.setCenter(self.tdSignal.getVCursor()))
self.tdSignal.plot.SIG_MARKER_CHANGED.connect(
lambda x=None: self.piUi.newOffset(self.tdSignal.getVCursor()))
self.osciCurveWidget.calcFun.idxChanged.connect(self.osciSignal.funChanged)
self.tdWidget.calcFun.idxChanged.connect(self.tdSignal.funChanged)
self.fdWidget.calcFun.idxChanged.connect(self.fdSignal.funChanged)
self.updateOsciPlot.connect(self.osciSignal.updatePlot)
self.updateTdPlot.connect(self.tdSignal.updatePlot)
self.updateFdPlot.connect(lambda data:
self.fdSignal.updatePlot(self.fdSignal.computeFFT(data)))
################
# create threads
#self.osciThr = GenericThread(self.getOsciData)
self.osciThr = QThread()
self.osciThr.start()
self.osciWorker = GenericWorker(self.getOsciData)
self.osciWorker.moveToThread(self.osciThr)
#self.measureThr = GenericThread(self.getMeasureData)
self.measureThr = QThread()
self.measureThr.start()
self.measureWorker = GenericWorker(self.getMeasureData)
self.measureWorker.moveToThread(self.measureThr)
################
# File menu
file_menu = self.menuBar().addMenu(_("File"))
self.quit_action = create_action(self, _("Quit"), shortcut="Ctrl+Q",
icon=get_std_icon("DialogCloseButton"),
tip=_("Quit application"),
triggered=self.close)
saveData = create_action(self, _("Save"), shortcut="Ctrl+S",
icon=get_std_icon("DialogSaveButton"),
tip=_("Save data"),
triggered=self.saveData)
triggerTest_action = create_action(self, _("Stop Osci"),
shortcut="Ctrl+O",
icon=get_icon('fileopen.png'),
tip=_("Open an image"),
triggered=self.stopOsciThr)
add_actions(file_menu, (triggerTest_action, saveData, None, self.quit_action))
##############
# Eventually add an internal console (requires 'spyderlib')
self.sift_proxy = SiftProxy(self)
if DockableConsole is None:
self.console = None
else:
import time, scipy.signal as sps, scipy.ndimage as spi
ns = {'ftir': self.sift_proxy,
'np': np, 'sps': sps, 'spi': spi,
'os': os, 'sys': sys, 'osp': osp, 'time': time}
msg = "Example: ftir.s[0] returns signal object #0\n"\
"Modules imported at startup: "\
"os, sys, os.path as osp, time, "\
"numpy as np, scipy.signal as sps, scipy.ndimage as spi"
self.console = DockableConsole(self, namespace=ns, message=msg)
self.add_dockwidget(self.console, _("Console"))
'''
try:
self.console.interpreter.widget_proxy.sig_new_prompt.connect(
lambda txt: self.refresh_lists())
except AttributeError:
print('sift: spyderlib is outdated', file=sys.stderr)
'''
# Show main window and raise the signal plot panel
self.show()
#------GUI refresh/setup
def add_dockwidget(self, child, title):
"""Add QDockWidget and toggleViewAction"""
dockwidget, location = child.create_dockwidget(title)
self.addDockWidget(location, dockwidget)
return dockwidget
def showMakeNicerWidget(self):
self.makeNicerWidget = MakeNicerWidget(self)
self.makeNicerDock = self.add_dockwidget(self.makeNicerWidget,
'Make FFT nicer')
#self.makeNicerDock.setFloating(True)
#self.fsBrowser = QDockWidget("4D Fermi Surface Browser", self)
#self.fsWidget = FermiSurface_Widget(self)
#self.fsBrowser.setWidget(self.fsWidget)
#self.fsBrowser.setFloating(True)
#self.addDockWidget(Qt.RightDockWidgetArea, self.fsBrowser)
def closeEvent(self, event):
if self.stage is not None:
self.stage.CloseConnection()
if self.console is not None:
self.console.exit_interpreter()
event.accept()
def saveData(self):
import datetime
now = datetime.datetime.now().strftime('%Y%m%d-%H%M%S')
# save time domain
foo = self.tdWidget.calcFun.functions
texts = [self.tdWidget.calcFun.itemText(i) for i in range(len(foo))]
tmp = ['td_x_{:s},td_y_{:s}'.format(i, i) for i in texts]
header = ','.join(tmp)
dataTd = np.zeros((self.tdSignal.getData(foo[0][0])[0].shape[0],
2*len(foo)))
for i, fun in enumerate(foo):
x, y = self.tdSignal.getData(fun[0])# [0]: fun, [1]: inverse fun
dataTd[:,2*i] = x
dataTd[:,2*i+1] = y
np.savetxt('data/{:s}_TD.txt'.format(now), dataTd, header=header)
self.tdSignal.plot.save_widget('data/{:s}_TD.png'.format(now))
# save frequency domain
foo = self.fdWidget.calcFun.functions
texts = [self.fdWidget.calcFun.itemText(i) for i in range(len(foo))]
tmp = ['fd_x_{:s},fd_y_{:s}'.format(i, i) for i in texts]
header += ','.join(tmp)
dataFd = np.zeros((self.fdSignal.getData(foo[0][0])[0].shape[0],
2*len(foo)))
for fun in foo:
x, y = self.fdSignal.getData(fun[0])
dataFd[:,2*i] = x
dataFd[:,2*i+1] = y
np.savetxt('data/{:s}_FD.txt'.format(now), dataFd, header=header)
self.fdSignal.plot.save_widget('data/{:s}_FD.png'.format(now))
# TODO: maybe put this in status bar
msg = QMessageBox()
msg.setIcon(QMessageBox.Information)
msg.setText('Data saved')
msg.exec_()
#def getStage(self, gcs):
# self.stage = gcs
# print(dir(self.stage))
# print(self.stage.qPOS())
def startOsciThr(self):
self.stopOsci = False
#self.osciThr.start()
self.osciWorker.start.emit()
def stopOsciThr(self):
self.stopOsci = True
# TODO: not quite shure how to do this with the worker solution.
# finished signal is emitted but this function should wait for worker to finish?!?
while(self.osciWorker.isRunning()):
time.sleep(0.03)
def getOsciData(self):
while not self.stopOsci:
data = self.tiepieUi.getData()
self.updateOsciPlot.emit(data)
time.sleep(0.5)
def startMeasureThr(self):
# stop osci thread and start measure thread
self.stopOsciThr()
# rescale tdPlot (updateXAxe)
self.piUi._xAxeChanged()
# set vCursor to 0
self.tdSignal.setVCursor(0)
self.piUi.setCenter()
# init x axe frequency domain plot to a min and max
delays = self.piUi.getDelays_fs()
data = np.column_stack((delays, np.zeros(len(delays))))
fdAxe = self.fdSignal.computeFFT(data)
self.fdSignal.updateXAxe(fdAxe[0,0], fdAxe[-1,0])
self.stopMeasure = False
#self.measureThr.start()
self.measureWorker.start.emit()
def stopMeasureThr(self):
self.stopMeasure = True
while(self.measureWorker.isRunning()):
time.sleep(0.03)
self.startOsciThr()
def getMeasureData(self):
delays = self.piUi.getDelays_fs()
data = np.column_stack((delays, np.zeros(len(delays))))
for i, delay in enumerate(delays):
if not self.stopMeasure:
self.piUi.gotoPos_fs(delay)
tmp = self.tiepieUi.getData()
self.updateOsciPlot.emit(tmp)
#print('measuring at', delay)
#y = dummyPulse(delay)
#data[i,1] = y
#data[i,1] = tmp[:,1].mean()
data[i,1] = self.osciSignal.computeSum()
#time.sleep(0.05)
self.updateTdPlot.emit(data)
self.updateFdPlot.emit(data)
else:
break
self.startOsciThr()
"""
def __init__(self):
QMainWindow.__init__(self)
self.stage = None
self.stopOsci = False
self.stopMeasure = False
self.setWindowTitle(APP_NAME)
###############
# Powermeter record
curveplot_toolbar = self.addToolBar(_("Curve Plotting Toolbar"))
self.curveWidget1 = DockablePlotWidget(self, CurveWidget,
curveplot_toolbar)
self.curveWidget1.calcFun.addFun('s', lambda x: x, lambda x: x)
self.curveWidget1.calcFun.addFun('min', lambda x: x / 60,
lambda x: x * 60)
self.curveWidget1.calcFun.addFun('hour', lambda x: x / 3600,
lambda x: x * 3600)
plot1 = self.curveWidget1.get_plot()
self.signal1 = SignalFT(self, plot=plot1)
##############
# Main window widgets
self.tabwidget = DockableTabWidget(self)
#self.tabwidget.setMaximumWidth(500)
self.maestroUi = MaestroUi(self)
self.fileUi = FileUi(self)
self.tabwidget.addTab(self.maestroUi,
QIcon('icons/Handyscope_HS4.png'), _("Maestro"))
self.tabwidget.addTab(self.fileUi, get_icon('filesave.png'), _('File'))
self.add_dockwidget(self.tabwidget, _("Inst. sett."))
# self.setCentralWidget(self.tabwidget)
self.dock1 = self.add_dockwidget(self.curveWidget1,
title=_("Powermeter"))
################
# connect signals
self.maestroUi.newPlotData.connect(self.signal1.updatePlot)
self.curveWidget1.calcFun.idxChanged.connect(self.signal1.funChanged)
self.fileUi.saveTxtBtn.released.connect(self.saveDataTxt)
self.fileUi.saveHdfBtn.released.connect(self.saveDataHDF5)
'''
self.piUi.startScanBtn.released.connect(self.startMeasureThr)
self.piUi.stopScanBtn.released.connect(self.stopMeasureThr)
self.piUi.xAxeChanged.connect(self.tdSignal.updateXAxe)
self.piUi.xAxeChanged.connect(self.fdSignal.updateXAxe)
self.piUi.niceBtn.released.connect(self.showMakeNicerWidget)
self.tiepieUi.scpConnected.connect(self.startOsciThr)
self.tiepieUi.xAxeChanged.connect(self.osciSignal.updateXAxe)
self.tiepieUi.yAxeChanged.connect(self.osciSignal.updateYAxe)
self.tiepieUi.triggLevelChanged.connect(self.osciSignal.setHCursor)
#self.piUi.centerBtn.released.connect(
# lambda x=None: self.piUi.setCenter(self.tdSignal.getVCursor()))
self.tdSignal.plot.SIG_MARKER_CHANGED.connect(
lambda x=None: self.piUi.newOffset(self.tdSignal.getVCursor()))
self.tdWidget.calcFun.idxChanged.connect(self.tdSignal.funChanged)
self.fdWidget.calcFun.idxChanged.connect(self.fdSignal.funChanged)
self.updateOsciPlot.connect(self.osciSignal.updatePlot)
self.updateTdPlot.connect(self.tdSignal.updatePlot)
self.updateFdPlot.connect(lambda data:
self.fdSignal.updatePlot(self.fdSignal.computeFFT(data)))
'''
################
# create threads
#self.osciThr = GenericThread(self.getOsciData)
'''
self.osciThr = QThread()
self.osciThr.start()
self.osciWorker = GenericWorker(self.getOsciData)
self.osciWorker.moveToThread(self.osciThr)
#self.measureThr = GenericThread(self.getMeasureData)
self.measureThr = QThread()
self.measureThr.start()
self.measureWorker = GenericWorker(self.getMeasureData)
self.measureWorker.moveToThread(self.measureThr)
'''
################
# File menu
file_menu = self.menuBar().addMenu(_("File"))
self.quit_action = create_action(
self,
_("Quit"),
shortcut="Ctrl+Q",
icon=get_std_icon("DialogCloseButton"),
tip=_("Quit application"),
triggered=self.close)
saveData = create_action(self,
_("Save"),
shortcut="Ctrl+S",
icon=get_std_icon("DialogSaveButton"),
tip=_("Save data"),
triggered=self.saveDataHDF5)
#add_actions(file_menu, (triggerTest_action, saveData, None, self.quit_action))
##############
# Eventually add an internal console (requires 'spyderlib')
self.sift_proxy = SiftProxy(self)
if DockableConsole is None:
self.console = None
else:
import time, scipy.signal as sps, scipy.ndimage as spi
ns = {
'ftir': self.sift_proxy,
'np': np,
'sps': sps,
'spi': spi,
'os': os,
'sys': sys,
'osp': osp,
'time': time
}
msg = "Example: ftir.s[0] returns signal object #0\n"\
"Modules imported at startup: "\
"os, sys, os.path as osp, time, "\
"numpy as np, scipy.signal as sps, scipy.ndimage as spi"
self.console = DockableConsole(self, namespace=ns, message=msg)
self.add_dockwidget(self.console, _("Console"))
'''
try:
self.console.interpreter.widget_proxy.sig_new_prompt.connect(
lambda txt: self.refresh_lists())
except AttributeError:
print('sift: spyderlib is outdated', file=sys.stderr)
'''
# Show main window and raise the signal plot panel
self.show()
class MainWindow(QMainWindow):
updateCameraPlot = Signal(object)
updateLineOutPlot = Signal(object)
def __init__(self):
QMainWindow.__init__(self)
self.stopAcqui = False
self.setWindowTitle(APP_NAME)
##############
# Camera image
self.image_toolbar = self.addToolBar(_("Image Processing Toolbar"))
imagevis_toolbar = self.addToolBar(_("Image Visualization Toolbar"))
self.imageWidget1 = DockablePlotWidget(self, ImageWidget,
imagevis_toolbar)
self.image1 = ImageFT(self, self.imageWidget1.get_plot())
self.image1.addHCursor(1)
self.image1.addRoi(0, 1, 2048, 1)
#self.image1.setup(self.image_toolbar)
###############
# camera line out
curveplot_toolbar = self.addToolBar(_("Curve Plotting Toolbar"))
self.curveWidget1 = DockablePlotWidget(self, CurveWidget,
curveplot_toolbar)
self.curveWidget1.calcFun.addFun('Pixels', lambda x: x, lambda x: x)
plot1 = self.curveWidget1.get_plot()
self.signal1 = SignalFT(self, plot=plot1)
##############
# Main window widgets
# status bar
self.status = self.statusBar()
# widgets
self.tabwidget = DockableTabWidget(self)
#self.tabwidget.setMaximumWidth(500)
self.greateyesUi = GreatEyesUi(self)
self.tabwidget.addTab(self.greateyesUi,
QIcon('icons/Handyscope_HS4.png'),
_("greateyes"))
#self.fileUi = FileUi(self)
#self.tabwidget.addTab(self.fileUi, get_icon('filesave.png'), _('File'))
self.add_dockwidget(self.tabwidget, _("Inst. sett."))
# self.setCentralWidget(self.tabwidget)
self.dock1 = self.add_dockwidget(self.imageWidget1, title=_("Camera"))
self.dock2 = self.add_dockwidget(self.curveWidget1, title=_("Lineout"))
################
# connect signals
self.greateyesUi.newPlotData.connect(self.newData)
self.greateyesUi.message.connect(self.updateStatus)
self.image1.plot.SIG_MARKER_CHANGED.connect(self.cursorMoved)
self.greateyesUi.loi.valueChanged.connect(self.cursorMoved)
self.greateyesUi.deltaPixels.valueChanged.connect(self.cursorMoved)
#self.curveWidget1.calcFun.idxChanged.connect(self.signal1.funChanged)
#self.fileUi.saveTxtBtn.released.connect(self.saveDataTxt)
#self.fileUi.saveHdfBtn.released.connect(self.saveDataHDF5)
'''
self.piUi.startScanBtn.released.connect(self.startMeasureThr)
self.piUi.stopScanBtn.released.connect(self.stopMeasureThr)
self.piUi.xAxeChanged.connect(self.tdSignal.updateXAxe)
self.piUi.xAxeChanged.connect(self.fdSignal.updateXAxe)
self.piUi.niceBtn.released.connect(self.showMakeNicerWidget)
self.tiepieUi.scpConnected.connect(self.startOsciThr)
self.tiepieUi.xAxeChanged.connect(self.osciSignal.updateXAxe)
self.tiepieUi.yAxeChanged.connect(self.osciSignal.updateYAxe)
self.tiepieUi.triggLevelChanged.connect(self.osciSignal.setHCursor)
#self.piUi.centerBtn.released.connect(
# lambda x=None: self.piUi.setCenter(self.tdSignal.getVCursor()))
self.tdSignal.plot.SIG_MARKER_CHANGED.connect(
lambda x=None: self.piUi.newOffset(self.tdSignal.getVCursor()))
self.tdWidget.calcFun.idxChanged.connect(self.tdSignal.funChanged)
self.fdWidget.calcFun.idxChanged.connect(self.fdSignal.funChanged)
self.updateOsciPlot.connect(self.osciSignal.updatePlot)
self.updateTdPlot.connect(self.tdSignal.updatePlot)
self.updateFdPlot.connect(lambda data:
self.fdSignal.updatePlot(self.fdSignal.computeFFT(data)))
'''
################
# File menu
'''
file_menu = self.menuBar().addMenu(_("File"))
self.quit_action = create_action(self, _("Quit"), shortcut="Ctrl+Q",
icon=get_std_icon("DialogCloseButton"),
tip=_("Quit application"),
triggered=self.close)
saveData = create_action(self, _("Save"), shortcut="Ctrl+S",
icon=get_std_icon("DialogSaveButton"),
tip=_("Save data"),
triggered=self.saveDataHDF5)
#add_actions(file_menu, (triggerTest_action, saveData, None, self.quit_action))
'''
##############
# Eventually add an internal console (requires 'spyderlib')
self.console = None
'''
self.sift_proxy = SiftProxy(self)
if DockableConsole is None:
self.console = None
else:
import time, scipy.signal as sps, scipy.ndimage as spi
ns = {'ftir': self.sift_proxy,
'np': np, 'sps': sps, 'spi': spi,
'os': os, 'sys': sys, 'osp': osp, 'time': time}
msg = "Example: ftir.s[0] returns signal object #0\n"\
"Modules imported at startup: "\
"os, sys, os.path as osp, time, "\
"numpy as np, scipy.signal as sps, scipy.ndimage as spi"
self.console = DockableConsole(self, namespace=ns, message=msg)
self.add_dockwidget(self.console, _("Console"))
'''
'''
try:
self.console.interpreter.widget_proxy.sig_new_prompt.connect(
lambda txt: self.refresh_lists())
except AttributeError:
print('sift: spyderlib is outdated', file=sys.stderr)
'''
# Show main window and raise the signal plot panel
self.show()
#------GUI refresh/setup
def add_dockwidget(self, child, title):
"""Add QDockWidget and toggleViewAction"""
dockwidget, location = child.create_dockwidget(title)
self.addDockWidget(location, dockwidget)
return dockwidget
def closeEvent(self, event):
self.greateyesUi.closeEvent(event)
if self.console is not None:
self.console.exit_interpreter()
event.accept()
def newData(self, image, timeStamp):
self.image1.updatePlot(
image,
timeStamp.isoformat() + ", " +
str(self.greateyesUi.cameraSettings['temperature']) + "°C")
self.updateLineOut(image)
self.saveDataHDF5(image, timeStamp)
def updateStatus(self, msg):
self.status.showMessage(msg, 10000)
def updateLineOut(self, image):
loi = self.greateyesUi.loi.value()
dLoi = self.greateyesUi.deltaPixels.value()
data = image[(loi - dLoi):(loi + dLoi + 1), :]
self.signal1.updatePlot(
np.column_stack((np.arange(0, data.shape[1]), data.sum(axis=0))))
self.image1.setHCursor(loi)
self.image1.setRoi(0, loi - dLoi, 2048, loi + dLoi)
def saveDataHDF5(self, image, timeStamp):
if not self.greateyesUi.autoSave.isChecked():
return
zeit = timeStamp.strftime('%Y%m%d-%H%M%S')
fileName = self.greateyesUi.directory + "/" + zeit
#fileName = QDir.toNativeSeparators(fileName)
print(fileName)
# save matlab file
savemat(
fileName, {
'image': image,
'comment': self.greateyesUi.comment.toPlainText(),
'camera_settings': self.greateyesUi.cameraSettings
})
# save images
self.image1.plot.save_widget(fileName + '_image.png')
self.signal1.plot.save_widget(fileName + '_lineout.png')
self.status.showMessage(fileName + " saved", 10000)
# TODO: maybe put this in status bar
'''
msg = QMessageBox()
msg.setIcon(QMessageBox.Information)
msg.setText('Data saved')
msg.exec_()
'''
def cursorMoved(self, mouse):
if not isinstance(mouse, int): # true if cursor moved by mouse
cursorVal = int(self.image1.getHCursor())
self.greateyesUi.loi.setValue(cursorVal)
else:
cursorVal = self.greateyesUi.loi.value()
self.image1.setHCursor(cursorVal)
roi = self.greateyesUi.deltaPixels.value()
self.image1.setRoi(0, cursorVal - roi, 2048, cursorVal + roi)