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