bmarechal / datalogger-gui

Download as
Browse Code ยป

Commit 58765476717bc16d96db3c46c47854de36d09b26

Authored by bmarechal
1 parent a15fba68b0
Exists in master

datalogger-gui: stop properly when close with X button

Showing 1 changed file with 8 additions and 0 deletions Inline Diff

#!/usr/bin/env python 1 1 #!/usr/bin/env python
2 2
# -*- coding: utf-8 -*- 3 3 # -*- coding: utf-8 -*-
4 4
import time, os, instruments, inspect, sys, threading 5 5 import time, os, instruments, inspect, sys, threading
import PyQt4.QtGui as QtGui 6 6 import PyQt4.QtGui as QtGui
from PyQt4.QtCore import pyqtSlot 7 7 from PyQt4.QtCore import pyqtSlot
8 8
#============================================================================== 9 9 #==============================================================================
#============================================================================== 10 10 #==============================================================================
11 11
class acq_routine(): 12 12 class acq_routine():
def __init__(self, instrument, channels, vtypes, address, path = os.getcwd(), samplingtime = 1, fileduration = 24*3600): 13 13 def __init__(self, instrument, channels, vtypes, address, path = os.getcwd(), samplingtime = 1, fileduration = 24*3600):
exec('self.instrument = instruments.%s.%s(%s, %s, "%s")'%(instrument, instrument, channels, vtypes, address)) 14 14 exec('self.instrument = instruments.%s.%s(%s, %s, "%s")'%(instrument, instrument, channels, vtypes, address))
self.path = path 15 15 self.path = path
self.samplingtime = samplingtime 16 16 self.samplingtime = samplingtime
self.fileduration = fileduration 17 17 self.fileduration = fileduration
18 18
def makeTree(self): 19 19 def makeTree(self):
try: 20 20 try:
year = time.strftime("%Y", time.gmtime(self.t0)) 21 21 year = time.strftime("%Y", time.gmtime(self.t0))
month = time.strftime("%Y-%m", time.gmtime(self.t0)) 22 22 month = time.strftime("%Y-%m", time.gmtime(self.t0))
os.chdir(self.path + '/' + year + '/' + month) 23 23 os.chdir(self.path + '/' + year + '/' + month)
except: 24 24 except:
try: 25 25 try:
os.chdir(self.path + '/' + year) 26 26 os.chdir(self.path + '/' + year)
os.mkdir(month) 27 27 os.mkdir(month)
os.chdir(self.path + '/' + year + '/' + month) 28 28 os.chdir(self.path + '/' + year + '/' + month)
except: 29 29 except:
os.chdir(self.path) 30 30 os.chdir(self.path)
os.mkdir(year) 31 31 os.mkdir(year)
os.chdir(self.path + '/' + year) 32 32 os.chdir(self.path + '/' + year)
os.mkdir(month) 33 33 os.mkdir(month)
os.chdir(self.path + '/' + year + '/' + month) 34 34 os.chdir(self.path + '/' + year + '/' + month)
35 35
def connect(self): 36 36 def connect(self):
self.instrument.connect() 37 37 self.instrument.connect()
38 38
self.t0 = time.time() 39 39 self.t0 = time.time()
self.filename = time.strftime("%Y%m%d-%H%M%S", time.gmtime(self.t0)) + '-' + self.instrument.model() + '.dat' 40 40 self.filename = time.strftime("%Y%m%d-%H%M%S", time.gmtime(self.t0)) + '-' + self.instrument.model() + '.dat'
self.makeTree() 41 41 self.makeTree()
self.data_file = open(self.filename, 'wr', 0) 42 42 self.data_file = open(self.filename, 'wr', 0)
43 43
def start(self): 44 44 def start(self):
tic = time.time() 45 45 tic = time.time()
46 46
if (time.time() - self.t0 >= self.fileduration) & (self.fileduration >0 ): 47 47 if (time.time() - self.t0 >= self.fileduration) & (self.fileduration >0 ):
self.data_file.close() 48 48 self.data_file.close()
49 49
self.t0 = time.time() 50 50 self.t0 = time.time()
self.filename = time.strftime("%Y%m%d-%H%M%S", time.gmtime(self.t0)) + '-' + self.instrument.model() + '.dat' 51 51 self.filename = time.strftime("%Y%m%d-%H%M%S", time.gmtime(self.t0)) + '-' + self.instrument.model() + '.dat'
self.makeTree() 52 52 self.makeTree()
self.data_file = open(self.filename, 'wr', 0) 53 53 self.data_file = open(self.filename, 'wr', 0)
54 54
#epoch time 55 55 #epoch time
epoch = time.time() 56 56 epoch = time.time()
#MJD time 57 57 #MJD time
mjd = epoch / 86400.0 + 40587 58 58 mjd = epoch / 86400.0 + 40587
# Meas values 59 59 # Meas values
meas = self.instrument.getValue() 60 60 meas = self.instrument.getValue()
meas = meas.replace(",", "\t") 61 61 meas = meas.replace(",", "\t")
meas = meas.replace(";", "\t") 62 62 meas = meas.replace(";", "\t")
meas = meas.replace("+", "") 63 63 meas = meas.replace("+", "")
64 64
string = "%f\t%f\t%s" % (epoch, mjd, meas) 65 65 string = "%f\t%f\t%s" % (epoch, mjd, meas)
self.data_file.write(string) # Write in a file 66 66 self.data_file.write(string) # Write in a file
print(string) 67 67 print(string)
68 68
self.thread = threading.Timer(self.samplingtime - (time.time() - tic), self.start) 69 69 self.thread = threading.Timer(self.samplingtime - (time.time() - tic), self.start)
self.thread.start() 70 70 self.thread.start()
71 71
def stop(self): 72 72 def stop(self):
self.thread.cancel() 73 73 self.thread.cancel()
self.instrument.disconnect() 74 74 self.instrument.disconnect()
self.data_file.close() 75 75 self.data_file.close()
76 76
#============================================================================== 77 77 #==============================================================================
#============================================================================== 78 78 #==============================================================================
79 79
class mainGui(): 80 80 class mainGui():
def __init__(self): 81 81 def __init__(self):
self.setWindow() 82 82 self.setWindow()
self.setSignalsSlots() 83 83 self.setSignalsSlots()
self.runApp() 84 84 self.runApp()
85 85
def setWindow(self): 86 86 def setWindow(self):
self.a = QtGui.QApplication(sys.argv) 87 87 self.a = QtGui.QApplication(sys.argv)
self.w = QtGui.QMainWindow() 88 88 self.w = QtGui.QMainWindow()
self.w.resize(640, 480) 89 89 self.w.resize(640, 480)
self.w.setWindowTitle('datalogger-gui') 90 90 self.w.setWindowTitle('datalogger-gui')
91 91
self.wid = QtGui.QWidget() 92 92 self.wid = QtGui.QWidget()
self.w.setCentralWidget(self.wid) 93 93 self.w.setCentralWidget(self.wid)
self.layout = QtGui.QGridLayout() 94 94 self.layout = QtGui.QGridLayout()
self.wid.setLayout(self.layout) 95 95 self.wid.setLayout(self.layout)
96 96
self.comboInst = QtGui.QComboBox() 97 97 self.comboInst = QtGui.QComboBox()
self.layout.addWidget(self.comboInst, 0, 0) 98 98 self.layout.addWidget(self.comboInst, 0, 0)
99 99
self.address = QtGui.QLineEdit() 100 100 self.address = QtGui.QLineEdit()
self.address.setMinimumWidth(140) 101 101 self.address.setMinimumWidth(140)
self.address.setMaximumWidth(140) 102 102 self.address.setMaximumWidth(140)
self.layout.addWidget(self.address, 0, 1) 103 103 self.layout.addWidget(self.address, 0, 1)
104 104
self.startButton = QtGui.QPushButton() 105 105 self.startButton = QtGui.QPushButton()
self.startButton.setText('Start log') 106 106 self.startButton.setText('Start log')
self.layout.addWidget(self.startButton, 99, 0) 107 107 self.layout.addWidget(self.startButton, 99, 0)
self.startButton.setEnabled(False) 108 108 self.startButton.setEnabled(False)
109 109
self.stopButton = QtGui.QPushButton() 110 110 self.stopButton = QtGui.QPushButton()
self.stopButton.setText('Stop log') 111 111 self.stopButton.setText('Stop log')
self.layout.addWidget(self.stopButton, 99, 1) 112 112 self.layout.addWidget(self.stopButton, 99, 1)
self.stopButton.setEnabled(False) 113 113 self.stopButton.setEnabled(False)
114 114
self.textDisplay = QtGui.QLabel() 115 115 self.textDisplay = QtGui.QLabel()
self.textDisplay.setText('>>') 116 116 self.textDisplay.setText('>>')
self.layout.addWidget(self.textDisplay, 99, 2) 117 117 self.layout.addWidget(self.textDisplay, 99, 2)
118 118
self.setComboInst() 119 119 self.setComboInst()
self.updateSignal() 120 120 self.updateSignal()
121 121
def setComboInst(self): 122 122 def setComboInst(self):
for name, obj in inspect.getmembers(instruments): 123 123 for name, obj in inspect.getmembers(instruments):
if inspect.ismodule(obj) and name.startswith('__') == False and name.startswith('abstract') == False: 124 124 if inspect.ismodule(obj) and name.startswith('__') == False and name.startswith('abstract') == False:
self.comboInst.addItem(name) 125 125 self.comboInst.addItem(name)
126 126
def setSignalsSlots(self): 127 127 def setSignalsSlots(self):
self.comboInst.currentIndexChanged.connect(self.updateSignal) 128 128 self.comboInst.currentIndexChanged.connect(self.updateSignal)
self.startButton.clicked.connect(self.startLog) 129 129 self.startButton.clicked.connect(self.startLog)
self.stopButton.clicked.connect(self.stopLog) 130 130 self.stopButton.clicked.connect(self.stopLog)
131 131
def runApp(self): 132 132 def runApp(self):
self.w.show() 133 133 self.w.show()
134 self.a.aboutToQuit.connect(self.closeEvent)
sys.exit(self.a.exec_()) 134 135 sys.exit(self.a.exec_())
136
137 def closeEvent(self):
138 try:
139 self.stopLog()
140 except:
141 pass
142 print('Done')
135 143
@pyqtSlot() 136 144 @pyqtSlot()
def updateSignal(self): 137 145 def updateSignal(self):
for i in reversed(range(5, self.layout.count())): 138 146 for i in reversed(range(5, self.layout.count())):
self.layout.itemAt(i).widget().setParent(None) 139 147 self.layout.itemAt(i).widget().setParent(None)
140 148
defaultAddress = '' 141 149 defaultAddress = ''
channelsAviables = [] 142 150 channelsAviables = []
vtypesAviables = [] 143 151 vtypesAviables = []
144 152
exec('channelsAviables = instruments.%s.ALL_CHANNELS'%self.comboInst.currentText()) 145 153 exec('channelsAviables = instruments.%s.ALL_CHANNELS'%self.comboInst.currentText())
exec('vtypesAviables = instruments.%s.ALL_VAL_TYPE'%self.comboInst.currentText()) 146 154 exec('vtypesAviables = instruments.%s.ALL_VAL_TYPE'%self.comboInst.currentText())
exec('defaultAddress = instruments.%s.ADDRESS'%self.comboInst.currentText()) 147 155 exec('defaultAddress = instruments.%s.ADDRESS'%self.comboInst.currentText())
148 156
self.address.setText(defaultAddress) 149 157 self.address.setText(defaultAddress)
150 158
self.checkBoxChannels = [None]*len(channelsAviables) 151 159 self.checkBoxChannels = [None]*len(channelsAviables)
self.chListVtypes = [None]*len(self.checkBoxChannels) 152 160 self.chListVtypes = [None]*len(self.checkBoxChannels)
153 161
for i in range(len(self.checkBoxChannels)): 154 162 for i in range(len(self.checkBoxChannels)):
self.checkBoxChannels[i] = QtGui.QCheckBox() 155 163 self.checkBoxChannels[i] = QtGui.QCheckBox()
self.checkBoxChannels[i].setText(channelsAviables[i]) 156 164 self.checkBoxChannels[i].setText(channelsAviables[i])
self.checkBoxChannels[i].setChecked(False) 157 165 self.checkBoxChannels[i].setChecked(False)
self.chListVtypes[i] = QtGui.QListWidget() 158 166 self.chListVtypes[i] = QtGui.QListWidget()
for vtype in vtypesAviables: 159 167 for vtype in vtypesAviables:
self.chListVtypes[i].addItem(vtype) 160 168 self.chListVtypes[i].addItem(vtype)
self.chListVtypes[i].setCurrentRow(0) 161 169 self.chListVtypes[i].setCurrentRow(0)
self.layout.addWidget(self.checkBoxChannels[i], i+3, 1) 162 170 self.layout.addWidget(self.checkBoxChannels[i], i+3, 1)
self.layout.addWidget(self.chListVtypes[i], i+3, 2) 163 171 self.layout.addWidget(self.chListVtypes[i], i+3, 2)
self.checkBoxChannels[i].stateChanged.connect(self.infoSignal) 164 172 self.checkBoxChannels[i].stateChanged.connect(self.infoSignal)
self.chListVtypes[i].currentItemChanged.connect(self.infoSignal) 165 173 self.chListVtypes[i].currentItemChanged.connect(self.infoSignal)
166 174
self.address.textChanged.connect(self.infoSignal) 167 175 self.address.textChanged.connect(self.infoSignal)
168 176
self.infoSignal() 169 177 self.infoSignal()
170 178
@pyqtSlot() 171 179 @pyqtSlot()
def infoSignal(self): 172 180 def infoSignal(self):
self.instToLog = self.comboInst.currentText() 173 181 self.instToLog = self.comboInst.currentText()
self.addressToLog = self.address.text() 174 182 self.addressToLog = self.address.text()
self.chToLog = [] 175 183 self.chToLog = []
self.vTypeToLog = [] 176 184 self.vTypeToLog = []
177 185
for i in range(len(self.checkBoxChannels)): 178 186 for i in range(len(self.checkBoxChannels)):
if self.checkBoxChannels[i].isChecked(): 179 187 if self.checkBoxChannels[i].isChecked():
self.chListVtypes[i].setEnabled(True) 180 188 self.chListVtypes[i].setEnabled(True)
self.chToLog.append(str(self.checkBoxChannels[i].text())) 181 189 self.chToLog.append(str(self.checkBoxChannels[i].text()))
self.vTypeToLog.append(str(self.chListVtypes[i].currentItem().text())) 182 190 self.vTypeToLog.append(str(self.chListVtypes[i].currentItem().text()))
else: 183 191 else:
self.chListVtypes[i].setEnabled(False) 184 192 self.chListVtypes[i].setEnabled(False)
185 193
allChannelsUnchecked = False 186 194 allChannelsUnchecked = False