Changeset 200 for trunk/synapse/Puzzlebox

Timestamp:

11/04/10 15:38:28 (11 years ago)

Author:

sc

Message:

synapse/Puzzlebox/Synapse/Interface.py:

• Export to CSV complete

synapse/Puzzlebox/Synapse/Protocol.py:

• fix to packet timestamps

Location:

trunk/synapse/Puzzlebox/Synapse

Files:

• Interface.py (modified) (2 diffs)
• Protocol.py (modified) (1 diff)

trunk/synapse/Puzzlebox/Synapse/Interface.py

@@ -10 +10 @@
 
 __changelog__ = """\
-Last Update: 2010.09.21
+Last Update: 2010.11.04
 
 """
@@ -736 +736 @@
         def exportData(self):
                 
-                textOutput = self.textEditDebugConsole.toPlainText()
-                
                 #output_file = 'text.synapse'
+                
+                (date, localtime) = self.parseTimeStamp(time.time())
+                
+                default_filename = '%s %s.csv' % (date, \
+                                      self.lineEditSessionProfile.text())
                 
                 output_file = QtGui.QFileDialog.getSaveFileName(self, \
                                  "Export Synapse Data to File", \
-                                 "", \
-                                 "Text File (*.txt)")
-                
+                                 default_filename, \
+                                 "CSV File (*.csv);;Text File (*.txt)")
+                
+                print output_file
+                
+                if str(output_file).endswith('.csv'):
+                        
+                        outputData = self.exportDataToCSV()
+                
+                
+                else:
+                        
+                        outputData = self.textEditDebugConsole.toPlainText()
+                        
+                        
                 file = open(output_file, 'w')
-                file.write(textOutput)
+                file.write(outputData)
                 file.close()
+        
+        
+        ##################################################################
+        
+        def exportDataToCSV(self):
+                
+                header = 'Date,Time,Delta,Theta,Low Alpha,High Alpha,Low Beta,High Beta,Low Gamma,Mid Gamma,Attention,Meditation,Signal Level\n'
+                
+                csv = {}
+                
+                for packet in self.packets['signals']:
+                        
+                        if 'rawEeg' in packet.keys():
+                                continue
+                        
+                        if packet['timestamp'] not in csv.keys():
+                                
+                                print packet
+                                timestamp = packet['timestamp']
+                                (date, localtime) = self.parseTimeStamp(timestamp)
+                                
+                                csv[timestamp] = {}
+                                csv[timestamp]['Date'] = date
+                                csv[timestamp]['Time'] = localtime
+                                csv[timestamp]['Delta'] = ''
+                                csv[timestamp]['Theta'] = ''
+                                csv[timestamp]['Low Alpha'] = ''
+                                csv[timestamp]['High Alpha'] = ''
+                                csv[timestamp]['Low Beta'] = ''
+                                csv[timestamp]['High Beta'] = ''
+                                csv[timestamp]['Low Gamma'] = ''
+                                csv[timestamp]['Mid Gamma'] = ''
+                                csv[timestamp]['Attention'] = ''
+                                csv[timestamp]['Meditation'] = ''
+                                csv[timestamp]['Signal Level'] = ''
+                        
+                        
+                        if 'eSense' in packet.keys():
+                                if 'attention' in packet['eSense'].keys():
+                                        csv[timestamp]['Attention'] = packet['eSense']['attention']
+                                if 'meditation' in packet['eSense'].keys():
+                                        csv[timestamp]['Meditation'] = packet['eSense']['meditation']
+                        
+                        if 'eegPower' in packet.keys():
+                                if 'delta' in packet['eegPower'].keys():
+                                        csv[timestamp]['Delta'] = packet['eegPower']['delta']
+                                if 'theta' in packet['eegPower'].keys():
+                                        csv[timestamp]['Theta'] = packet['eegPower']['theta']
+                                if 'lowAlpha' in packet['eegPower'].keys():
+                                        csv[timestamp]['Low Alpha'] = packet['eegPower']['lowAlpha']
+                                if 'highAlpha' in packet['eegPower'].keys():
+                                        csv[timestamp]['High Alpha'] = packet['eegPower']['highAlpha']
+                                if 'lowBeta' in packet['eegPower'].keys():
+                                        csv[timestamp]['Low Beta'] = packet['eegPower']['lowBeta']
+                                if 'highBeta' in packet['eegPower'].keys():
+                                        csv[timestamp]['High Beta'] = packet['eegPower']['highBeta']
+                                if 'lowGamma' in packet['eegPower'].keys():
+                                        csv[timestamp]['Low Gamma'] = packet['eegPower']['lowGamma']
+                                if 'highGamma' in packet['eegPower'].keys():
+                                        csv[timestamp]['Mid Gamma'] = packet['eegPower']['highGamma']
+                        
+                        if 'poorSignalLevel' in packet.keys():
+                                csv[timestamp]['Signal Level'] = packet['poorSignalLevel']
+                
+                
+                output = header
+                
+                csv_keys = csv.keys()
+                csv_keys.sort()
+                
+                for key in csv_keys:
+                        
+                        row = '%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s\n' % \
+                              (csv[key]['Date'], \
+                               csv[key]['Time'], \
+                               csv[key]['Delta'], \
+                               csv[key]['Theta'], \
+                               csv[key]['Low Alpha'], \
+                               csv[key]['High Alpha'], \
+                               csv[key]['Low Beta'], \
+                               csv[key]['High Beta'], \
+                               csv[key]['Low Gamma'], \
+                               csv[key]['Mid Gamma'], \
+                               csv[key]['Attention'], \
+                               csv[key]['Meditation'], \
+                               csv[key]['Signal Level'])
+                        
+                        
+                        output = output + row
+                
+                
+                return(output)
         
         

trunk/synapse/Puzzlebox/Synapse/Protocol.py

@@ -706 +706 @@
                 
                 
+                process_packet['timestamp'] = packet_update['timestamp']
+                
+                
                 if self.DEBUG > 3:
                         print self.data_packet