Changeset 411 for trunk/Puzzlebox/Synapse/ThinkGear/Protocol.py

Timestamp:

01/27/16 01:10:35 (6 years ago)

Author:

sc

Message:

• exportTrainingDataToCSV() added
• collectProcessedData() added
• Training debug output added to Protocol

File:

• trunk/Puzzlebox/Synapse/ThinkGear/Protocol.py (modified) (10 diffs)

trunk/Puzzlebox/Synapse/ThinkGear/Protocol.py

@@ -86 +86 @@
 import signal
 import serial
+import copy
 
 if ((sys.platform != 'win32') and \
@@ -154 +155 @@
 DEBUG_PACKET_COUNT = 1024
 
+
+#CURRENT_SIGNAL = 0
+#CURRENT_ATTENTION = 0
+#CURRENT_MEDITATION = 0
+
 #####################################################################
 # Classes
@@ -185 +191 @@
                 self.data_packet['eegPower'] = {}
                 self.data_packet['eSense'] = {}
+                
+                self.current_signal = 200
+                self.current_attention = 0
+                self.current_meditaiton = 0
+                self.detection_threshold = 70
+                self.current_detection = 0
                 
                 #self.packet_count = 0
@@ -367 +379 @@
                 if (raw >= 32768):
                         raw = raw - 65536
+                
+                
+                
+                #print "%s,%s,%s,%s,%s" % (time.time(),
+                        #CURRENT_SIGNAL,
+                        #CURRENT_ATTENTION,
+                        #CURRENT_MEDITATION,
+                        #raw)
+                
+                if (self.current_signal == 0):
+                        print "%s,%i,%i,%i,%s,%i" % (time.time(),
+                                                self.current_signal,
+                                                self.current_attention,
+                                                self.current_meditaiton,
+                                                raw,
+                                                self.current_detection)
                 
                 
@@ -477 +505 @@
                                         print poor_signal_quality
                                 
+                                self.current_signal = copy.copy(poor_signal_quality)
+                                
                                 packet_update['poorSignalLevel'] = poor_signal_quality
+                                
+                                self.current_signal = poor_signal_quality
                         
                         
@@ -486 +518 @@
                                         print attention
                                 
+                                self.current_attention = copy.copy(attention)
+                                
+                                if (attention > self.detection_threshold):
+                                        self.current_detection = 1
+                                else:
+                                        self.current_detection = 0
+                                
                                 packet_update['eSense'] = {}
                                 packet_update['eSense']['attention'] = attention
@@ -495 +534 @@
                                         print "meditation:",
                                         print meditation
+                                
+                                self.current_meditaiton = copy.copy(meditation)
                                 
                                 packet_update['eSense'] = {}
@@ -521 +562 @@
                                 
                                 packet_update['rawEeg'] = raw_eeg_value
+                                
+                                #print "%s,%i,%i,%i,%s" % (time.time(),
+                                                                                #self.current_signal,
+                                                                                #self.current_attention,
+                                                                                #self.current_meditaiton,
+                                                                                #raw_eeg_value)
                         
                         
@@ -892 +939 @@
         def run(self):
                 
-                self.resetSession()
+                try:
+                        self.resetSession()
+                except Exception, e:
+                        if self.DEBUG:
+                                print "ERROR: self.resetSession():",
+                                print e
                 
                 if self.device != None and self.device.device != None:
@@ -1234 +1286 @@
                                         self.buffer += byte
                         
-                        except:
+                        except Exception, e:
                                 if self.DEBUG:
-                                        print "ERROR: failed to read from serial device"
+                                        print "ERROR: failed to read from serial device:",
+                                        print e
                                 break