Commit b7ceca5e38b2b939900df4416316c8258a54b135
1 parent
050af22188
Exists in
master
datalogger-gui.py: add sampling time spinbox
Showing 1 changed file with 16 additions and 3 deletions Inline Diff
datalogger-gui.py
| #!/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, samplingtime, path = os.getcwd(), 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("+", "") | |
| meas = meas.replace("E", "e") | 64 | 64 | meas = meas.replace("E", "e") | |
| 65 | 65 | |||
| string = "%f\t%f\t%s" % (epoch, mjd, meas) | 66 | 66 | string = "%f\t%f\t%s" % (epoch, mjd, meas) | |
| self.data_file.write(string) # Write in a file | 67 | 67 | self.data_file.write(string) # Write in a file | |
| print(string) | 68 | 68 | print(string) | |
| 69 | 69 | |||
| self.thread = threading.Timer(self.samplingtime - (time.time() - tic), self.start) | 70 | 70 | self.thread = threading.Timer(self.samplingtime - (time.time() - tic), self.start) | |
| self.thread.start() | 71 | 71 | self.thread.start() | |
| 72 | 72 | |||
| def stop(self): | 73 | 73 | def stop(self): | |
| self.thread.cancel() | 74 | 74 | self.thread.cancel() | |
| self.instrument.disconnect() | 75 | 75 | self.instrument.disconnect() | |
| self.data_file.close() | 76 | 76 | self.data_file.close() | |
| 77 | 77 | |||
| #============================================================================== | 78 | 78 | #============================================================================== | |
| #============================================================================== | 79 | 79 | #============================================================================== | |
| 80 | 80 | |||
| class mainGui(): | 81 | 81 | class mainGui(): | |
| def __init__(self): | 82 | 82 | def __init__(self): | |
| self.setWindow() | 83 | 83 | self.setWindow() | |
| self.setSignalsSlots() | 84 | 84 | self.setSignalsSlots() | |
| self.runApp() | 85 | 85 | self.runApp() | |
| 86 | 86 | |||
| def setWindow(self): | 87 | 87 | def setWindow(self): | |
| self.a = QtGui.QApplication(sys.argv) | 88 | 88 | self.a = QtGui.QApplication(sys.argv) | |
| self.w = QtGui.QMainWindow() | 89 | 89 | self.w = QtGui.QMainWindow() | |
| self.w.resize(640, 480) | 90 | 90 | self.w.resize(640, 480) | |
| self.w.setWindowTitle('datalogger-gui') | 91 | 91 | self.w.setWindowTitle('datalogger-gui') | |
| 92 | 92 | |||
| self.wid = QtGui.QWidget() | 93 | 93 | self.wid = QtGui.QWidget() | |
| self.w.setCentralWidget(self.wid) | 94 | 94 | self.w.setCentralWidget(self.wid) | |
| self.layout = QtGui.QGridLayout() | 95 | 95 | self.layout = QtGui.QGridLayout() | |
| self.wid.setLayout(self.layout) | 96 | 96 | self.wid.setLayout(self.layout) | |
| 97 | 97 | |||
| self.comboInst = QtGui.QComboBox() | 98 | 98 | self.comboInst = QtGui.QComboBox() | |
| self.layout.addWidget(self.comboInst, 0, 0) | 99 | 99 | self.layout.addWidget(self.comboInst, 0, 0) | |
| 100 | 100 | |||
| self.address = QtGui.QLineEdit() | 101 | 101 | self.address = QtGui.QLineEdit() | |
| self.address.setMinimumWidth(140) | 102 | 102 | self.address.setMinimumWidth(140) | |
| self.address.setMaximumWidth(140) | 103 | 103 | self.address.setMaximumWidth(140) | |
| self.layout.addWidget(self.address, 0, 1) | 104 | 104 | self.layout.addWidget(self.address, 0, 1) | |
| 105 | 105 | |||
| 106 | self.samplingtime = QtGui.QDoubleSpinBox() | |||
| 107 | #self.samplingtime.setMinimumWidth(60) | |||
| 108 | #self.samplingtime.setMaximumWidth(60) | |||
| 109 | self.samplingtime.setMinimum(0.1) | |||
| 110 | self.samplingtime.setMaximum(1000) | |||
| 111 | self.samplingtime.setSingleStep(0.1) | |||
| 112 | self.samplingtime.setValue(1) | |||
| 113 | self.layout.addWidget(self.samplingtime, 0, 2) | |||
| 114 | ||||
| self.startButton = QtGui.QPushButton() | 106 | 115 | self.startButton = QtGui.QPushButton() | |
| self.startButton.setText('Start log') | 107 | 116 | self.startButton.setText('Start log') | |
| self.layout.addWidget(self.startButton, 99, 0) | 108 | 117 | self.layout.addWidget(self.startButton, 99, 0) | |
| self.startButton.setEnabled(False) | 109 | 118 | self.startButton.setEnabled(False) | |
| 110 | 119 | |||
| self.stopButton = QtGui.QPushButton() | 111 | 120 | self.stopButton = QtGui.QPushButton() | |
| self.stopButton.setText('Stop log') | 112 | 121 | self.stopButton.setText('Stop log') | |
| self.layout.addWidget(self.stopButton, 99, 1) | 113 | 122 | self.layout.addWidget(self.stopButton, 99, 1) | |
| self.stopButton.setEnabled(False) | 114 | 123 | self.stopButton.setEnabled(False) | |
| 115 | 124 | |||
| self.textDisplay = QtGui.QLabel() | 116 | 125 | self.textDisplay = QtGui.QLabel() | |
| self.textDisplay.setText('>>') | 117 | 126 | self.textDisplay.setText('>>') | |
| self.layout.addWidget(self.textDisplay, 99, 2) | 118 | 127 | self.layout.addWidget(self.textDisplay, 99, 2) | |
| 119 | 128 | |||
| self.setComboInst() | 120 | 129 | self.setComboInst() | |
| self.updateSignal() | 121 | 130 | self.updateSignal() | |
| 122 | 131 | |||
| def setComboInst(self): | 123 | 132 | def setComboInst(self): | |
| for name, obj in inspect.getmembers(instruments): | 124 | 133 | for name, obj in inspect.getmembers(instruments): | |
| if inspect.ismodule(obj) and name.startswith('__') == False and name.startswith('abstract') == False: | 125 | 134 | if inspect.ismodule(obj) and name.startswith('__') == False and name.startswith('abstract') == False: | |
| self.comboInst.addItem(name) | 126 | 135 | self.comboInst.addItem(name) | |
| 127 | 136 | |||
| def setSignalsSlots(self): | 128 | 137 | def setSignalsSlots(self): | |
| self.comboInst.currentIndexChanged.connect(self.updateSignal) | 129 | 138 | self.comboInst.currentIndexChanged.connect(self.updateSignal) | |
| self.startButton.clicked.connect(self.startLog) | 130 | 139 | self.startButton.clicked.connect(self.startLog) | |
| self.stopButton.clicked.connect(self.stopLog) | 131 | 140 | self.stopButton.clicked.connect(self.stopLog) | |
| 132 | 141 | |||
| def runApp(self): | 133 | 142 | def runApp(self): | |
| self.w.show() | 134 | 143 | self.w.show() | |
| self.a.aboutToQuit.connect(self.closeEvent) | 135 | 144 | self.a.aboutToQuit.connect(self.closeEvent) | |
| sys.exit(self.a.exec_()) | 136 | 145 | sys.exit(self.a.exec_()) | |
| 137 | 146 | |||
| def closeEvent(self): | 138 | 147 | def closeEvent(self): | |
| try: | 139 | 148 | try: | |
| self.stopLog() | 140 | 149 | self.stopLog() | |
| except: | 141 | 150 | except: | |
| pass | 142 | 151 | pass | |
| print('Done') | 143 | 152 | print('Done') | |
| 144 | 153 | |||
| @pyqtSlot() | 145 | 154 | @pyqtSlot() | |
| def updateSignal(self): | 146 | 155 | def updateSignal(self): | |
| for i in reversed(range(5, self.layout.count())): | 147 | 156 | for i in reversed(range(5, self.layout.count())): | |
| self.layout.itemAt(i).widget().setParent(None) | 148 | 157 | self.layout.itemAt(i).widget().setParent(None) | |
| 149 | 158 | |||
| defaultAddress = '' | 150 | 159 | defaultAddress = '' | |
| channelsAviables = [] | 151 | 160 | channelsAviables = [] | |
| vtypesAviables = [] | 152 | 161 | vtypesAviables = [] | |
| 153 | 162 | |||
| exec('channelsAviables = instruments.%s.ALL_CHANNELS'%self.comboInst.currentText()) | 154 | 163 | exec('channelsAviables = instruments.%s.ALL_CHANNELS'%self.comboInst.currentText()) | |
| exec('vtypesAviables = instruments.%s.ALL_VAL_TYPE'%self.comboInst.currentText()) | 155 | 164 | exec('vtypesAviables = instruments.%s.ALL_VAL_TYPE'%self.comboInst.currentText()) | |
| exec('defaultAddress = instruments.%s.ADDRESS'%self.comboInst.currentText()) | 156 | 165 | exec('defaultAddress = instruments.%s.ADDRESS'%self.comboInst.currentText()) | |
| 157 | 166 | |||
| self.address.setText(defaultAddress) | 158 | 167 | self.address.setText(defaultAddress) | |
| 159 | 168 | |||
| self.checkBoxChannels = [None]*len(channelsAviables) | 160 | 169 | self.checkBoxChannels = [None]*len(channelsAviables) | |
| self.chListVtypes = [None]*len(self.checkBoxChannels) | 161 | 170 | self.chListVtypes = [None]*len(self.checkBoxChannels) | |
| 162 | 171 | |||
| for i in range(len(self.checkBoxChannels)): | 163 | 172 | for i in range(len(self.checkBoxChannels)): | |
| self.checkBoxChannels[i] = QtGui.QCheckBox() | 164 | 173 | self.checkBoxChannels[i] = QtGui.QCheckBox() | |
| self.checkBoxChannels[i].setText(channelsAviables[i]) | 165 | 174 | self.checkBoxChannels[i].setText(channelsAviables[i]) | |
| self.checkBoxChannels[i].setChecked(False) | 166 | 175 | self.checkBoxChannels[i].setChecked(False) | |
| self.chListVtypes[i] = QtGui.QListWidget() | 167 | 176 | self.chListVtypes[i] = QtGui.QListWidget() | |
| for vtype in vtypesAviables: | 168 | 177 | for vtype in vtypesAviables: | |
| self.chListVtypes[i].addItem(vtype) | 169 | 178 | self.chListVtypes[i].addItem(vtype) | |
| self.chListVtypes[i].setCurrentRow(0) | 170 | 179 | self.chListVtypes[i].setCurrentRow(0) | |
| self.layout.addWidget(self.checkBoxChannels[i], i+3, 1) | 171 | 180 | self.layout.addWidget(self.checkBoxChannels[i], i+3, 1) | |
| self.layout.addWidget(self.chListVtypes[i], i+3, 2) | 172 | 181 | self.layout.addWidget(self.chListVtypes[i], i+3, 2) | |
| self.checkBoxChannels[i].stateChanged.connect(self.infoSignal) | 173 | 182 | self.checkBoxChannels[i].stateChanged.connect(self.infoSignal) | |
| self.chListVtypes[i].currentItemChanged.connect(self.infoSignal) | 174 | 183 | self.chListVtypes[i].currentItemChanged.connect(self.infoSignal) | |
| 175 | 184 | |||
| self.address.textChanged.connect(self.infoSignal) | 176 | 185 | self.address.textChanged.connect(self.infoSignal) | |
| 186 | self.samplingtime.valueChanged.connect(self.infoSignal) | |||
| 177 | 187 | |||
| self.infoSignal() | 178 | 188 | self.infoSignal() | |
| 179 | 189 | |||
| @pyqtSlot() | 180 | 190 | @pyqtSlot() | |
| def infoSignal(self): | 181 | 191 | def infoSignal(self): | |
| self.instToLog = self.comboInst.currentText() | 182 | 192 | self.instToLog = self.comboInst.currentText() | |
| self.addressToLog = self.address.text() | 183 | 193 | self.addressToLog = self.address.text() | |
| self.chToLog = [] | 184 | 194 | self.chToLog = [] | |
| self.vTypeToLog = [] | 185 | 195 | self.vTypeToLog = [] | |
| 196 | self.ts = self.samplingtime.value() | |||
| 186 | 197 | |||
| for i in range(len(self.checkBoxChannels)): | 187 | 198 | for i in range(len(self.checkBoxChannels)): | |
| if self.checkBoxChannels[i].isChecked(): | 188 | 199 | if self.checkBoxChannels[i].isChecked(): | |
| self.chListVtypes[i].setEnabled(True) | 189 | 200 | self.chListVtypes[i].setEnabled(True) | |
| self.chToLog.append(str(self.checkBoxChannels[i].text())) | 190 | 201 | self.chToLog.append(str(self.checkBoxChannels[i].text())) | |
| self.vTypeToLog.append(str(self.chListVtypes[i].currentItem().text())) | 191 | 202 | self.vTypeToLog.append(str(self.chListVtypes[i].currentItem().text())) | |
| else: | 192 | 203 | else: | |
| self.chListVtypes[i].setEnabled(False) | 193 | 204 | self.chListVtypes[i].setEnabled(False) | |
| 194 | 205 | |||
| allChannelsUnchecked = False | 195 | 206 | allChannelsUnchecked = False | |
| for i in self.checkBoxChannels: | 196 | 207 | for i in self.checkBoxChannels: | |
| allChannelsUnchecked = allChannelsUnchecked or i.isChecked() | 197 | 208 | allChannelsUnchecked = allChannelsUnchecked or i.isChecked() | |
| if allChannelsUnchecked == False: | 198 | 209 | if allChannelsUnchecked == False: | |
| self.startButton.setEnabled(False) | 199 | 210 | self.startButton.setEnabled(False) |