bmarechal / datalogger-gui

Download as
Browse Code ยป

Commit c8a8ff0f662bc584d3cdd9731eadb920ea78b743

Authored by bmarechal
1 parent 3a0af875bd
Exists in master

modify gui

Showing 1 changed file with 8 additions and 7 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.setMaximumWidth(120) 101 101 self.address.setMinimumWidth(140)
self.layout.addWidget(self.address, 99, 0) 102 102 self.address.setMaximumWidth(140)
103 self.layout.addWidget(self.address, 0, 1)
103 104
self.startButton = QtGui.QPushButton() 104 105 self.startButton = QtGui.QPushButton()
self.startButton.setText('Start log') 105 106 self.startButton.setText('Start log')
self.layout.addWidget(self.startButton, 99, 1) 106 107 self.layout.addWidget(self.startButton, 99, 0)
self.startButton.setEnabled(False) 107 108 self.startButton.setEnabled(False)
108 109
self.stopButton = QtGui.QPushButton() 109 110 self.stopButton = QtGui.QPushButton()
self.stopButton.setText('Stop log') 110 111 self.stopButton.setText('Stop log')
self.layout.addWidget(self.stopButton, 99, 2) 111 112 self.layout.addWidget(self.stopButton, 99, 1)
self.stopButton.setEnabled(False) 112 113 self.stopButton.setEnabled(False)
113 114
self.textDisplay = QtGui.QLabel() 114 115 self.textDisplay = QtGui.QLabel()
self.textDisplay.setText('>>') 115 116 self.textDisplay.setText('>>')
self.layout.addWidget(self.textDisplay, 100, 2) 116 117 self.layout.addWidget(self.textDisplay, 99, 2)
117 118
self.setComboInst() 118 119 self.setComboInst()
self.updateSignal() 119 120 self.updateSignal()
120 121
def setComboInst(self): 121 122 def setComboInst(self):
for name, obj in inspect.getmembers(instruments): 122 123 for name, obj in inspect.getmembers(instruments):
if inspect.ismodule(obj) and name.startswith('__') == False and name.startswith('abstract') == False: 123 124 if inspect.ismodule(obj) and name.startswith('__') == False and name.startswith('abstract') == False:
self.comboInst.addItem(name) 124 125 self.comboInst.addItem(name)
125 126
def setSignalsSlots(self): 126 127 def setSignalsSlots(self):
self.comboInst.currentIndexChanged.connect(self.updateSignal) 127 128 self.comboInst.currentIndexChanged.connect(self.updateSignal)
self.startButton.clicked.connect(self.startLog) 128 129 self.startButton.clicked.connect(self.startLog)
self.stopButton.clicked.connect(self.stopLog) 129 130 self.stopButton.clicked.connect(self.stopLog)
130 131
def runApp(self): 131 132 def runApp(self):
self.w.show() 132 133 self.w.show()
sys.exit(self.a.exec_()) 133 134 sys.exit(self.a.exec_())
134 135
@pyqtSlot() 135 136 @pyqtSlot()
def updateSignal(self): 136 137 def updateSignal(self):
for i in reversed(range(5, self.layout.count())): 137 138 for i in reversed(range(5, self.layout.count())):
self.layout.itemAt(i).widget().setParent(None) 138 139 self.layout.itemAt(i).widget().setParent(None)
139 140
defaultAddress = '' 140 141 defaultAddress = ''
channelsAviables = [] 141 142 channelsAviables = []
vtypesAviables = [] 142 143 vtypesAviables = []
143 144
exec('channelsAviables = instruments.%s.ALL_CHANNELS'%self.comboInst.currentText()) 144 145 exec('channelsAviables = instruments.%s.ALL_CHANNELS'%self.comboInst.currentText())
exec('vtypesAviables = instruments.%s.ALL_VAL_TYPE'%self.comboInst.currentText()) 145 146 exec('vtypesAviables = instruments.%s.ALL_VAL_TYPE'%self.comboInst.currentText())
exec('defaultAddress = instruments.%s.ADDRESS'%self.comboInst.currentText()) 146 147 exec('defaultAddress = instruments.%s.ADDRESS'%self.comboInst.currentText())
147 148
self.address.setText(defaultAddress) 148 149 self.address.setText(defaultAddress)
149 150
self.checkBoxChannels = [None]*len(channelsAviables) 150 151 self.checkBoxChannels = [None]*len(channelsAviables)
self.chListVtypes = [None]*len(self.checkBoxChannels) 151 152 self.chListVtypes = [None]*len(self.checkBoxChannels)
152 153
for i in range(len(self.checkBoxChannels)): 153 154 for i in range(len(self.checkBoxChannels)):
self.checkBoxChannels[i] = QtGui.QCheckBox() 154 155 self.checkBoxChannels[i] = QtGui.QCheckBox()
self.checkBoxChannels[i].setText(channelsAviables[i]) 155 156 self.checkBoxChannels[i].setText(channelsAviables[i])
self.checkBoxChannels[i].setChecked(False) 156 157 self.checkBoxChannels[i].setChecked(False)
self.chListVtypes[i] = QtGui.QListWidget() 157 158 self.chListVtypes[i] = QtGui.QListWidget()
for vtype in vtypesAviables: 158 159 for vtype in vtypesAviables:
self.chListVtypes[i].addItem(vtype) 159 160 self.chListVtypes[i].addItem(vtype)
self.chListVtypes[i].setCurrentRow(0) 160 161 self.chListVtypes[i].setCurrentRow(0)
self.layout.addWidget(self.checkBoxChannels[i], i, 1) 161 162 self.layout.addWidget(self.checkBoxChannels[i], i+3, 1)
self.layout.addWidget(self.chListVtypes[i], i, 2) 162 163 self.layout.addWidget(self.chListVtypes[i], i+3, 2)
self.checkBoxChannels[i].stateChanged.connect(self.infoSignal) 163 164 self.checkBoxChannels[i].stateChanged.connect(self.infoSignal)
self.chListVtypes[i].currentItemChanged.connect(self.infoSignal) 164 165 self.chListVtypes[i].currentItemChanged.connect(self.infoSignal)
165 166
self.address.textChanged.connect(self.infoSignal) 166 167 self.address.textChanged.connect(self.infoSignal)
167 168
self.infoSignal() 168 169 self.infoSignal()
169 170
@pyqtSlot() 170 171 @pyqtSlot()
def infoSignal(self): 171 172 def infoSignal(self):
self.instToLog = self.comboInst.currentText() 172 173 self.instToLog = self.comboInst.currentText()
self.addressToLog = self.address.text() 173 174 self.addressToLog = self.address.text()
self.chToLog = [] 174 175 self.chToLog = []
self.vTypeToLog = [] 175 176 self.vTypeToLog = []
176 177
for i in range(len(self.checkBoxChannels)): 177 178 for i in range(len(self.checkBoxChannels)):
if self.checkBoxChannels[i].isChecked(): 178 179 if self.checkBoxChannels[i].isChecked():
self.chListVtypes[i].setEnabled(True) 179 180 self.chListVtypes[i].setEnabled(True)
self.chToLog.append(str(self.checkBoxChannels[i].text())) 180 181 self.chToLog.append(str(self.checkBoxChannels[i].text()))
self.vTypeToLog.append(str(self.chListVtypes[i].currentItem().text())) 181 182 self.vTypeToLog.append(str(self.chListVtypes[i].currentItem().text()))
else: 182 183 else:
self.chListVtypes[i].setEnabled(False) 183 184 self.chListVtypes[i].setEnabled(False)
184 185
allChannelsUnchecked = False 185 186 allChannelsUnchecked = False
for i in self.checkBoxChannels: 186 187 for i in self.checkBoxChannels:
allChannelsUnchecked = allChannelsUnchecked or i.isChecked() 187 188 allChannelsUnchecked = allChannelsUnchecked or i.isChecked()
if allChannelsUnchecked == False: 188 189 if allChannelsUnchecked == False:
self.startButton.setEnabled(False) 189 190 self.startButton.setEnabled(False)
else: 190 191 else:
self.startButton.setEnabled(True) 191 192 self.startButton.setEnabled(True)
192 193