Changeset 335

Timestamp:

03/25/12 10:59:39 (8 years ago)

Author:

sc

Message:

• PyEpoc? moved to separate module

File:

• trunk/Puzzlebox/Synapse/Emotiv/Protocol.py (modified) (4 diffs)

trunk/Puzzlebox/Synapse/Emotiv/Protocol.py

@@ -16 +16 @@
         try:
                 import PySide as PyQt4
-                from PySide import QtCore, QtGui, QtNetwork
+                from PySide import QtCore # , QtGui, QtNetwork
         except Exception, e:
                 print "ERROR: Exception importing PySide:",
                 print e
-                configuration.ENABLE_PYSIDE = False
+                #configuration.ENABLE_PYSIDE = False
+                ENABLE_PYSIDE = False
         else:
-                print "INFO: [Synapse:Interface] Using PySide module"
+                print "INFO: [Synapse:Emotiv:Protocol] Using PySide module"
 
-else:
-        print "INFO: [Synapse:Interface] Using PyQt4 module"
-        from PyQt4 import QtCore, QtGui, QtNetwork
+#if not configuration.ENABLE_PYSIDE:
+if not ENABLE_PYSIDE:
+        print "INFO: [Synapse:Emotiv:Protocol] Using PyQt4 module"
+        from PyQt4 import QtCore # , QtGui, QtNetwork
 
 
@@ -33 +35 @@
 
 
-from PyEpoc import *
-
-EmotivEngine = EpocHandler()
-choice = str()
-engineeventhandle = EmotivEngine.EE_EmoEngineEventCreate()
-statehandle = EmotivEngine.EE_EmoStateCreate()
-EmotivEngine.ES_Init(statehandle)
+#from PyEpoc.PyEpoc import *
+import PyEpoc.PyEpoc
 
 
@@ -75 +72 @@
                 
                 #self.device_id = device_id
-                #self.device_model = device_model
-                
-                #self.device = None
-                #self.buffer = ''
-                #self.payload_timestamp = time.time()
-                
-                #self.device = serial_device
-                #self.auto_connect_timestamp = time.time()
-                
-                #self.data_packet = {}
-                #self.data_packet['eegPower'] = {}
-                #self.data_packet['eSense'] = {}
-                
-                #self.packet_count = 0
-                #self.bad_packets = 0
-                #self.session_start_time = None
+                self.device_model = device_model        
+                self.device = serial_device
                 
                 #self.keep_running = True
-        
+
+                self.EmotivEngine = PyEpoc.PyEpoc.EpocHandler()
+                self.engineeventhandle = self.EmotivEngine.EE_EmoEngineEventCreate()
+                self.statehandle = self.EmotivEngine.EE_EmoStateCreate()
+                self.EmotivEngine.ES_Init(self.statehandle)
+
         
         ##################################################################
@@ -108 +96 @@
                 # Connect
                 if self.choice == "a":
-                        if EmotivEngine.EE_EngineRemoteConnect("127.0.0.1",1726) != ERRCODE['EDK_OK']:
+                        if self.EmotivEngine.EE_EngineRemoteConnect("127.0.0.1", 1726) != PyEpoc.PyEpoc.ERRCODE['EDK_OK']:
                                 print "Could not connect!"
                                 quit()
                         else: print "Connected to EmoComposer!"    
                 elif self.choice == "b":
-                        if EmotivEngine.EE_EngineConnect() != ERRCODE['EDK_OK']:
+                        if self.EmotivEngine.EE_EngineConnect() != PyEpoc.PyEpoc.ERRCODE['EDK_OK']:
                                 print "Could not connect!"
                                 quit()
                         else: print "Connected to Headset!"
                 
-                print """Engine Event handle at: %s""" %(engineeventhandle)
-                print """Emo State handle at: %s""" % (statehandle)
+                print """Engine Event handle at: %s""" %(self.engineeventhandle)
+                print """Emo State handle at: %s""" % (self.statehandle)
                 
                 # Start 
                 while True:
-                        state = EmotivEngine.EE_EngineGetNextEvent(engineeventhandle)
+                        state = self.EmotivEngine.EE_EngineGetNextEvent(self.engineeventhandle)
                         
-                        if state == ERRCODE['EDK_OK']:
-                                event = EmotivEngine.EE_EmoEngineEventGetType(engineeventhandle)
-                                user = EmotivEngine.EE_EmoEngineEventGetUserId(engineeventhandle)[1]
-                                if (event == EVENT['EE_EmoStateUpdated']):
-                                        EmotivEngine.EE_EmoEngineEventGetEmoState(engineeventhandle,statehandle)
+                        if state == PyEpoc.PyEpoc.ERRCODE['EDK_OK']:
+                                event = self.EmotivEngine.EE_EmoEngineEventGetType(self.engineeventhandle)
+                                user = self.EmotivEngine.EE_EmoEngineEventGetUserId(self.engineeventhandle)[1]
+                                if (event == PyEpoc.PyEpoc.EVENT['EE_EmoStateUpdated']):
+                                        self.EmotivEngine.EE_EmoEngineEventGetEmoState(self.engineeventhandle, self.statehandle)
                                         #status
-                                        battery = EmotivEngine.ES_GetBatteryChargeLevel(statehandle)
-                                        wireless = EmotivEngine.ES_GetWirelessSignalStatus(statehandle)
-                                        timestamp = EmotivEngine.ES_GetTimeFromStart(statehandle)
+                                        battery = self.EmotivEngine.ES_GetBatteryChargeLevel(self.statehandle)
+                                        wireless = self.EmotivEngine.ES_GetWirelessSignalStatus(self.statehandle)
+                                        timestamp = self.EmotivEngine.ES_GetTimeFromStart(self.statehandle)
                                         #emo
-                                        meditation = EmotivEngine.ES_AffectivGetMeditationScore(statehandle)
-                                        shortexcitement = EmotivEngine.ES_AffectivGetExcitementShortTermScore(statehandle)
-                                        longexcitement = EmotivEngine.ES_AffectivGetExcitementLongTermScore(statehandle)
-                                        frustration = EmotivEngine.ES_AffectivGetFrustrationScore(statehandle)
+                                        meditation = self.EmotivEngine.ES_AffectivGetMeditationScore(self.statehandle)
+                                        shortexcitement = self.EmotivEngine.ES_AffectivGetExcitementShortTermScore(self.statehandle)
+                                        longexcitement = self.EmotivEngine.ES_AffectivGetExcitementLongTermScore(self.statehandle)
+                                        frustration = self.EmotivEngine.ES_AffectivGetFrustrationScore(self.statehandle)
                                         #cognitiv
-                                        currentaction = EmotivEngine.ES_CognitivGetCurrentAction(statehandle)
-                                        currentactionpower = EmotivEngine.ES_CognitivGetCurrentActionPower(statehandle)
+                                        currentaction = self.EmotivEngine.ES_CognitivGetCurrentAction(self.statehandle)
+                                        currentactionpower = self.EmotivEngine.ES_CognitivGetCurrentActionPower(self.statehandle)
                                         print """--------------------------------------------------------------------------------"""
-                                        print """Timestamp: %s\tUser: %s\tEvent: %s\tW.Signal: %s\tBattery: %s""" % (timestamp,user,hex(event),wireless,battery)
-                                        print """S.Excite: %s\tL.Excite: %s\tMeditation: %s\tFrustr: %s""" % (shortexcitement,longexcitement,meditation,frustration)
-                                        print """Curr.Action: %s\tAction Pwr: %s""" % (hex(currentaction),currentactionpower)
+                                        print """Timestamp: %s\tUser: %s\tEvent: %s\tW.Signal: %s\tBattery: %s""" % (timestamp,user,hex(event), wireless, battery)
+                                        print """S.Excite: %s\tL.Excite: %s\tMeditation: %s\tFrustr: %s""" % (shortexcitement, longexcitement, meditation, frustration)
+                                        print """Curr.Action: %s\tAction Pwr: %s""" % (hex(currentaction), currentactionpower)