import tkinter as tk from tkinter import * from tkinter import ttk from threading import Thread import time # For sleep, clock, time and perf_counter from datetime import datetime, timedelta import numpy as np import matplotlib.pyplot as plt from matplotlib.figure import Figure from matplotlib.backends.backend_tkagg import (FigureCanvasTkAgg, NavigationToolbar2Tk) import csv import ue9qol class SinGen: def __init__(self, ampl=1, offset=0, phase=0, sweeper=None): self.ampl = ampl self.phase = phase self.sweeper = sweeper self.offset = offset def getValue(self, sweeper=None): if sweeper is None and self.sweeper is None: print("Error: generator needs sweeper") return 0 if sweeper is None: sweeper = self.sweeper return self.ampl * np.sin( 2*np.pi*sweeper.getRelPos()) + self.offset class RampGen: # monotonically goes from start to stop, at final point move to start def __init__(self, start=0, stop=0, sweeper=None): self.start = start self.stop = stop self.sweeper = sweeper def getValue(self, sweeper=None): if sweeper is None and self.sweeper is None: print("Error: generator needs sweeper") return 0 if sweeper is None: sweeper = self.sweeper return self.start + sweeper.getRelPos()*(self.stop - self.start) class TriangleGen: # monotonically goes from start to stop, once reaches stop goes back to start def __init__(self, start=0, stop=0, sweeper=None): self.start = start self.stop = stop self.sweeper = sweeper def getValue(self, sweeper=None): if sweeper is None and self.sweeper is None: print("Error: generator needs sweeper") return 0 if sweeper is None: sweeper = self.sweeper if sweeper.getRelPos() < 0.5: return self.start + 2*sweeper.getRelPos()*(self.stop - self.start) return self.start + 2*(1-sweeper.getRelPos())*(self.stop - self.start) class PulseGen: # monotonically goes from start to stop, once reaches stop goes back to start def __init__(self, ampl=1, sweeper=None): self.ampl = ampl self.sweeper = sweeper def getValue(self, sweeper=None): if sweeper is None and self.sweeper is None: print("Error: generator needs sweeper") return 0 if sweeper is None: sweeper = self.sweeper if sweeper.getRelPos() < 0.5: return self.ampl return 0 class Sweeper: def __init__(self, widget, Npoints, SweepTime, onTicCallbacks=[]): # walk from start to stop with Npoints # cnt = 1 corresponds to start # cnt = Npoints corresponds to stop # variables like relVar are relative to the start of the period self.cnt = 0 # onTic will increase it right away self.widget = widget self.Npoints = Npoints self.start = 1 self.stop = self.Npoints self.SweepTime = SweepTime self.onTicCallbacks = onTicCallbacks self.isOn = False self.isRestart = True self.isTicRunning = False self.span = self.stop - self.start self.center = (self.stop + self.start)/2 self.dPos = self.span/(self.Npoints-1) self.dT = self.SweepTime/(self.Npoints-1) self.dTmS = round(self.dT*1000) # dT in milliseconds def reset(self): self.cnt = 0 # onTic will increase it right away self.isRestart = False self.startTime = datetime.now() def onTic(self): start = datetime.now() deadline = start + timedelta(milliseconds=self.dTmS) if not self.isOn: self.isTicRunning = False return self.isTicRunning = True if self.isRestart: self.reset() self.isRestart = False self.incr() for cb in self.onTicCallbacks: cb(self) stop = datetime.now() self.isTicRunning = False if stop > deadline: runTime = (stop-start).seconds + float((stop-start).microseconds)/1000000 print("Overrun: Callbacks took %s seconds instead of %s" % (runTime, self.dTmS/1000) ) self.widget.after(0, self.onTic) idleTime_mS = round((deadline-stop).seconds * 1000 + (deadline-stop).microseconds/1000) # print("Will idle for %s" % (idleTime_mS) ) self.widget.after(idleTime_mS, self.onTic) def cmdRestart(self): self.cnt = 0 if self.isOn: self.isRestart = True return self.reset() self.isOn = True self.onTic() def cmdStart(self): self.isOn = True self.onTic() def cmdStop(self): self.isOn = False def incr(self): self.cnt += 1 self.updPos() def updPos(self): self.relCnt = 1 + ((self.cnt-1) % self.Npoints) self.pos = self.start + self.dPos * (self.relCnt - 1) self.relPos = (self.pos-self.start)/self.span def getCnt(self): return self.cnt def getRelCnt(self): return self.relCnt def getPos(self): return self.pos def getRelPos(self): return self.relPos class Experiment: def __init__(self, root): self.root = root self.tic = 0 self.channelsNames2plot={'dac0', 'dac1', 'adc1', 'adc2', 'adc3', 'adc4'} self.xChannelName='dac0' # can be also 'tic' or any of above self.xlabel='Frequency (Hz)' self.lines2plot={} self.clearData() self.guiSetup(root) self.guiSweeper = Sweeper(self.root, Npoints=2, SweepTime=1, onTicCallbacks=[self.updatePlot]) self.guiSweeper.cmdStart() self.hardware = {} # self.hardwareSetup() self.sweeper = Sweeper(self.root, Npoints=100, SweepTime=1, onTicCallbacks=[self.onTic]) # self.funcGen = SinGen(2, 2, sweeper = self.sweeper) # self.funcGen = RampGen(0, 5, sweeper = self.sweeper) self.funcGen = TriangleGen(0, 5, sweeper = self.sweeper) def hardwareSetup(self): self.hardware['LabJack'] = ue9qol.UE9qol() def guiSetup(self, root): self.cntrlPannel=ttk.LabelFrame(root, text='controls') self.cntrlPannel.pack() self.bAutoZoom=Button(self.cntrlPannel,text="AutoZoom",command=self.autoZoom,font=('Arial','24')) self.bAutoZoom.pack(side='left') self.bRestart=Button(self.cntrlPannel,text="RESTART",command=self.restart,font=('Arial','24')) self.bRestart.pack(side='left') self.bStart=Button(self.cntrlPannel,text="START",command=self.start,font=('Arial','24')) self.bStart.pack(side='left') self.bStop=Button(self.cntrlPannel,text="STOP",command=self.stop,font=('Arial','24')) self.bStop.pack(side='left') self.bSave=Button(self.cntrlPannel,text="SAVE",command=self.saveCmd,font=('Arial','24')) self.bSave.pack(side='left') self.bExit=Button(self.cntrlPannel,text="EXIT",command=exit,font=('Arial','24')) self.bExit.pack(side='left') self.dataDisplay=ttk.LabelFrame(root, text='data') self.dataDisplay.pack() self.fig=plt.figure(figsize=[32, 24]) self.ax = self.fig.add_subplot(1,1,1) # self.ax.set_xlim([0,20]) # self.ax.set_ylim([0,20]) # self.ax.plot([i for i in range(10)],[i for i in range(10)]) self.line, = self.ax.plot(self.data['tic'], self.data['adc1'], '.') self.canvas = FigureCanvasTkAgg(self.fig, master = self.dataDisplay) self.canvas.draw() # placing the canvas on the Tkinter window # self.canvas.get_tk_widget().pack() # creating the Matplotlib toolbar self.toolbar = NavigationToolbar2Tk(self.canvas, self.dataDisplay) self.toolbar.update() # placing the toolbar on the Tkinter window self.canvas.get_tk_widget().pack() def clearData(self): self.data = {} self.data['tic'] = [] self.data['x'] = [] self.data['dac0'] = [] self.data['dac1'] = [] self.data['adc1'] = [] self.data['adc2'] = [] self.data['adc3'] = [] self.data['adc4'] = [] def stop(self): self.sweeper.cmdStop() def start(self): self.sweeper.cmdStart() def restart(self): self.clearData() self.sweeper.cmdRestart() def saveCmd(self): csv_file = 'data.csv' data = self.data try: with open(csv_file, 'w') as csvfile: writer = csv.writer(csvfile) writer.writerow(data.keys()) writer.writerows(zip(*data.values())) except IOError: print('I/O error') def onTic(self,swp=None): start = datetime.now() if swp is None: swp = self.sweeper # global tic counter tic = self.sweeper.getCnt() self.data['tic'].append(tic) # DAQ # daq0 = self.hardware['LabJack'] # dac0 dac0 = self.funcGen.getValue(swp) # daq0.setOutputCh(0, out0) self.data['dac0'].append(dac0) # dac1 dac1 = PulseGen(ampl=5, sweeper=swp).getValue() # daq0.setOutputCh(0, dac1) self.data['dac1'].append(dac1) # adc1 # adc1= daq0.getInputCh(1) adc1 = SinGen(ampl=1, sweeper=swp).getValue() self.data['adc1'].append( adc1 ) # adc2 # adc2= daq0.getInputCh(2) adc2 = SinGen(ampl=2, sweeper=swp).getValue() self.data['adc2'].append( adc2 ) # adc3 # adc3= daq0.getInputCh(3) adc3 = SinGen(ampl=3, sweeper=swp).getValue() self.data['adc3'].append( adc3 ) # adc4 # adc4= daq0.getInputCh(4) adc4 = SinGen(ampl=4, sweeper=swp).getValue() self.data['adc4'].append( adc4 ) # X-axis (i.e. independent variable) # self.data['x'].append(tic) self.data['x']=self.data[self.xChannelName] stop = datetime.now() runTime = (stop-start).seconds + float((stop-start).microseconds)/1000000 # print("onTic DAQ took %s seconds." % (runTime) ) def autoZoom(self): self.ax.cla() x = self.data['x'] for name in self.channelsNames2plot: if name not in self.data: continue y = self.data[name] self.lines2plot[name], = self.ax.plot(x, y, '.', label=name) self.ax.legend() plt.xlabel(self.xlabel) self.canvas.draw() def updatePlot(self,swp=None): start = datetime.now() # self.ax.cla() # self.line, = self.ax.plot(self.data['tic'], self.data['adc1'], '.') # t = Thread(target=self.canvas.draw) # self.line.set_data([.1, .2, .3], [.1, .2, .3]) x = self.data['x'] for name in self.channelsNames2plot: if name not in self.data: continue y = self.data[name] if name in self.lines2plot: ln = self.lines2plot[name] ln.set_data(x, y) self.ax.draw_artist(ln) # self.canvas.update() # self.canvas.draw() self.fig.canvas.flush_events() stop = datetime.now() runTime = (stop-start).seconds + float((stop-start).microseconds)/1000000 print("Replot took %s seconds to plot %s points." % (runTime, len(self.data['adc1'])) ) if __name__ == '__main__': root=Tk() experiment=Experiment(root) root.mainloop()