bmarechal / datalogger-gui

Blame view

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