bmarechal / datalogger-gui

Blame view

datalogger-gui.py 8.23 KB
edit raw normal view history
bb3889092   mer0m   Add files via upload 
1
2
3
4
5
6
7
8
9
10
11
12
 
  #!/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():
9058343c5   bmarechal   some minor fixes 
13
14
 
      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))
bb3889092   mer0m   Add files via upload 
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
 
          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)
9058343c5   bmarechal   some minor fixes 
99
100
101
 
          self.address = QtGui.QLineEdit()
          self.address.setMaximumWidth(120)
          self.layout.addWidget(self.address, 99, 0)
bb3889092   mer0m   Add files via upload 
102
103
104
105
 
  
          self.startButton = QtGui.QPushButton()
          self.startButton.setText('Start log')
          self.layout.addWidget(self.startButton, 99, 1)
7bc4035fc   bmarechal   enable startButto... 
106
 
          self.startButton.setEnabled(False)
bb3889092   mer0m   Add files via upload 
107
108
109
110
 
  
          self.stopButton = QtGui.QPushButton()
          self.stopButton.setText('Stop log')
          self.layout.addWidget(self.stopButton, 99, 2)
faa244fdd   bmarechal   Enable/disable st... 
111
 
          self.stopButton.setEnabled(False)
bb3889092   mer0m   Add files via upload 
112
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
 
  
          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)
9058343c5   bmarechal   some minor fixes 
138
 
          defaultAddress = ''
bb3889092   mer0m   Add files via upload 
139
140
141
142
143
 
          channelsAviables = []
          vtypesAviables = []
  
          exec('channelsAviables = instruments.%s.ALL_CHANNELS'%self.comboInst.currentText())
          exec('vtypesAviables = instruments.%s.ALL_VAL_TYPE'%self.comboInst.currentText())
9058343c5   bmarechal   some minor fixes 
144
 
          exec('defaultAddress = instruments.%s.ADDRESS'%self.comboInst.currentText())
bb3889092   mer0m   Add files via upload 
145
 








9058343c5
 
 bmarechal
 
some minor fixes




146




          self.address.setText(defaultAddress)









bb3889092
 
 mer0m
 
Add files via upload




147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162




  
          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)









9058343c5
 
 bmarechal
 
some minor fixes




163




          self.address.textChanged.connect(self.infoSignal)









faa244fdd
 
 bmarechal
 
Enable/disable st...




164












bb3889092
 
 mer0m
 
Add files via upload




165
166
167
168
169




          self.infoSignal()
  
      @pyqtSlot()
      def infoSignal(self):
          self.instToLog = self.comboInst.currentText()









9058343c5
 
 bmarechal
 
some minor fixes




170




          self.addressToLog = self.address.text()









bb3889092
 
 mer0m
 
Add files via upload




171
172
173
174
175




          self.chToLog = []
          self.vTypeToLog = []
  
          for i in range(len(self.checkBoxChannels)):
              if self.checkBoxChannels[i].isChecked():









8b41ae5b5
 
 bmarechal
 
enable/disable vt...




176




                  self.chListVtypes[i].setEnabled(True)









bb3889092
 
 mer0m
 
Add files via upload




177
178




                  self.chToLog.append(str(self.checkBoxChannels[i].text()))
                  self.vTypeToLog.append(str(self.chListVtypes[i].currentItem().text()))









8b41ae5b5
 
 bmarechal
 
enable/disable vt...




179
180




              else:
                  self.chListVtypes[i].setEnabled(False)









7bc4035fc
 
 bmarechal
 
enable startButto...




181
182
183
184
185
186
187




          allChannelsUnchecked = False
          for i in self.checkBoxChannels:
              allChannelsUnchecked = allChannelsUnchecked or i.isChecked()
          if allChannelsUnchecked == False:
              self.startButton.setEnabled(False)
          else:
              self.startButton.setEnabled(True)









9058343c5
 
 bmarechal
 
some minor fixes




188




          self.textDisplay.setText('>> %s@%s - %s - %s'%(self.instToLog, self.addressToLog, self.chToLog, self.vTypeToLog))









bb3889092
 
 mer0m
 
Add files via upload




189












9058343c5
 
 bmarechal
 
some minor fixes




190




          self.myLog = acq_routine(self.instToLog, self.chToLog, self.vTypeToLog, self.addressToLog)









bb3889092
 
 mer0m
 
Add files via upload




191
192
193




  
      @pyqtSlot()
      def startLog(self):









faa244fdd
 
 bmarechal
 
Enable/disable st...




194
195




          self.startButton.setEnabled(False)
          self.stopButton.setEnabled(True)









9058343c5
 
 bmarechal
 
some minor fixes




196




          self.address.setEnabled(False)









120da9952
 
 bmarechal
 
enable/disable co...




197




          self.comboInst.setEnabled(False)









f5cf6b2b6
 
 bmarechal
 
disable address a...




198
199




          for i in self.checkBoxChannels:
              i.setEnabled(False)









56312b9ab
 
 bmarechal
 
improve GUI




200
201




          for i in self.chListVtypes:
              i.setEnabled(False)









bb3889092
 
 mer0m
 
Add files via upload




202
203
204
205
206




          self.myLog.connect()
          self.myLog.start()
  
      @pyqtSlot()
      def stopLog(self):









faa244fdd
 
 bmarechal
 
Enable/disable st...




207
208




          self.startButton.setEnabled(True)
          self.stopButton.setEnabled(False)









9058343c5
 
 bmarechal
 
some minor fixes




209




          self.address.setEnabled(True)









120da9952
 
 bmarechal
 
enable/disable co...




210




          self.comboInst.setEnabled(True)









56312b9ab
 
 bmarechal
 
improve GUI




211
212
213
214
215
216
217




          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




218
219
220
221
222
223
224




          self.myLog.stop()
  
  #==============================================================================
  #==============================================================================
  
  if __name__ == "__main__":
      mainGui()