Changeset 343

Timestamp:

03/27/12 18:03:41 (8 years ago)

Author:

sc

Message:

main.cpp:

• new event loop

File:

• trunk/project/PuzzleboxSynapseEmotiv/PuzzleboxSynapseEmotiv/main.cpp (modified) (4 diffs)

trunk/project/PuzzleboxSynapseEmotiv/PuzzleboxSynapseEmotiv/main.cpp

@@ -14 +14 @@
 #define USE_COMPOSER = True;
 
-//void logEmotivUpdate(std::ostream& os, unsigned int userID, EmoStateHandle eState, bool withHeader = false);
-void logEmotivUpdate(unsigned int userID, EmoStateHandle eState, bool withHeader = false);
+bool print_to_console = 1;
+
+EmoEngineEventHandle eEvent             = EE_EmoEngineEventCreate();
+EmoStateHandle eState                           = EE_EmoStateCreate();
+unsigned int userID                                     = 0;
+
+void printHeaders();
+void processEmotivUpdate();
+//void logEmotivUpdate(unsigned int userID, EmoStateHandle eState, bool withHeader = false);
+void logEmotivUpdate(bool withHeader = false);
 
 int main(int argc, char** argv) {
 
-        EmoEngineEventHandle eEvent             = EE_EmoEngineEventCreate();
-        EmoStateHandle eState                           = EE_EmoStateCreate();
-        unsigned int userID                                     = 0;
+        //EmoEngineEventHandle eEvent           = EE_EmoEngineEventCreate();
+        //EmoStateHandle eState                         = EE_EmoStateCreate();
+        //unsigned int userID                           = 0;
         std::string emotivHost           = "127.0.0.1";
         const unsigned short composerPort       = 1726;
@@ -36 +44 @@
                 }
                 
-                std::cout << "Start receiving EmoState! Press any key to stop logging...\n" << std::endl;
-
-                //std::ofstream ofs(argv[1]);
-                bool writeHeader = true;
-                
-                while (!_kbhit()) {
-
-                        state = EE_EngineGetNextEvent(eEvent);
-
-                        // New event needs to be handled
-                        if (state == EDK_OK) {
-
-                                EE_Event_t eventType = EE_EmoEngineEventGetType(eEvent);
-                                EE_EmoEngineEventGetUserId(eEvent, &userID);
-
-                                // Log the EmoState if it has been updated
-                                if (eventType == EE_EmoStateUpdated) {
-
-                                        EE_EmoEngineEventGetEmoState(eEvent, eState);
-                                        const float timestamp = ES_GetTimeFromStart(eState);
-
-                                        //printf("%10.3fs : New EmoState from user %d ...\r", timestamp, userID);
-                                        
-                                        //logEmotivUpdate(ofs, userID, eState, writeHeader);
-                                        logEmotivUpdate(userID, eState, writeHeader);
-                                        writeHeader = false;
-                                }
-                        }
-                        else if (state != EDK_NO_EVENT) {
-                                std::cout << "Internal error in Emotiv Engine!" << std::endl;
-                                break;
-                        }
-
-                        Sleep(1);
+                printHeaders();
+
+                while (1) {
+                        processEmotivUpdate();
                 }
 
-                //ofs.close();
+
+                //std::cout << "Start receiving EmoState! Press any key to stop logging...\n" << std::endl;
+
+                //bool writeHeader = true;
+                //
+                //while (!_kbhit()) {
+
+                //      state = EE_EngineGetNextEvent(eEvent);
+
+                //      // New event needs to be handled
+                //      if (state == EDK_OK) {
+
+                //              EE_Event_t eventType = EE_EmoEngineEventGetType(eEvent);
+                //              EE_EmoEngineEventGetUserId(eEvent, &userID);
+
+                //              // Log the EmoState if it has been updated
+                //              if (eventType == EE_EmoStateUpdated) {
+
+                //                      EE_EmoEngineEventGetEmoState(eEvent, eState);
+                //                      const float timestamp = ES_GetTimeFromStart(eState);
+
+                //                      //printf("%10.3fs : New EmoState from user %d ...\r", timestamp, userID);
+                //                      
+                //                      //logEmotivUpdate(userID, eState, writeHeader);
+                //                      logEmotivUpdate(writeHeader);
+                //                      writeHeader = false;
+                //              }
+                //      }
+                //      else if (state != EDK_NO_EVENT) {
+                //              std::cout << "Internal error in Emotiv Engine!" << std::endl;
+                //              break;
+                //      }
+
+                //      Sleep(1);
+                //}
+
         }
         catch (const std::exception& e) {
@@ -88 +101 @@
 
 
+void printHeaders() {
+        std::cout << "Time,";
+        std::cout << "Emotiv UserID,";
+        std::cout << "Wireless Signal Status,";
+        std::cout << "Headset On,";
+
+        std::cout << "Short Term Excitement,";
+        std::cout << "Long Term Excitement,";
+        std::cout << "Meditation,";
+        std::cout << "Frustration,";
+        std::cout << "Engagement/Boredom,";
+        std::cout << "Cognitiv Action,";
+        std::cout << "Cognitiv Power";
+        std::cout << std::endl;
+}
+
+//void processEmotivUpdate(unsigned int userID, EmoStateHandle eState, bool withHeader) {
+void processEmotivUpdate() {
+
+        int state = EE_EngineGetNextEvent(eEvent);
+
+        // New event needs to be handled
+        if (state == EDK_OK) {
+
+                EE_Event_t eventType = EE_EmoEngineEventGetType(eEvent);
+                EE_EmoEngineEventGetUserId(eEvent, &userID);
+
+                // Log the EmoState if it has been updated
+                if (eventType == EE_EmoStateUpdated) {
+
+                        EE_EmoEngineEventGetEmoState(eEvent, eState);
+
+                        //if (print_to_console) {
+                        //      std::cout << "Time,";
+                        //      std::cout << "Emotiv UserID,";
+                        //      std::cout << "Wireless Signal Status,";
+                        //      std::cout << "Headset On,";
+
+                        //      std::cout << "Short Term Excitement,";
+                        //      std::cout << "Long Term Excitement,";
+                        //      std::cout << "Meditation,";
+                        //      std::cout << "Frustration,";
+                        //      std::cout << "Engagement/Boredom,";
+                        //      std::cout << "Cognitiv Action,";
+                        //      std::cout << "Cognitiv Power,";
+                        //      std::cout << std::endl;
+                        //} // DEBUG
+
+
+
+                        // Log the time stamp and user ID
+                        float time_from_start = ES_GetTimeFromStart(eState);
+                        //std::cout << userID << ",";
+                        int wireless_signal_status = static_cast<int>(ES_GetWirelessSignalStatus(eState));
+                        int headset_on = static_cast<int>(ES_GetHeadsetOn(eState));
+
+                        // Affectiv Suite results
+                        float excitement = ES_AffectivGetExcitementShortTermScore(eState);
+                        float excitement_long_term = ES_AffectivGetExcitementLongTermScore(eState);
+                        float meditation = ES_AffectivGetMeditationScore(eState);
+                        float frustration = ES_AffectivGetFrustrationScore(eState);
+                        float engagement_boredome = ES_AffectivGetEngagementBoredomScore(eState);
+
+                        // Cognitiv Suite results
+                        int cognitiv_action = static_cast<int>(ES_CognitivGetCurrentAction(eState));
+                        float cognitive_power = ES_CognitivGetCurrentActionPower(eState);
+
+                        if (print_to_console) {
+                                // Log the time stamp and user ID
+                                std::cout << time_from_start << ",";
+                                std::cout << userID << ",";
+                                std::cout << wireless_signal_status << ",";
+                                std::cout << headset_on << ",";
+                                std::cout << excitement << ",";
+                                std::cout << meditation << ",";
+                                std::cout << frustration << ",";
+                                std::cout << engagement_boredome << ",";
+                                std::cout << cognitiv_action << ",";
+                                std::cout << cognitive_power;
+                                std::cout << std::endl;
+                        } // DEBUG
+
+                }
+        }
+}
+
+
 //void logEmotivUpdate(std::ostream& os, unsigned int userID, EmoStateHandle eState, bool withHeader) {
-void logEmotivUpdate(unsigned int userID, EmoStateHandle eState, bool withHeader) {
+//void logEmotivUpdate(unsigned int userID, EmoStateHandle eState, bool withHeader) {
+void logEmotivUpdate(bool withHeader) {
 
         // Create the top header
         if (withHeader) {
                 std::cout << "Time,";
-                //std::cout << "UserID,";
+                std::cout << "Emotiv UserID,";
                 std::cout << "Wireless Signal Status,";
                 std::cout << "Headset On,";
@@ -123 +224 @@
         // Log the time stamp and user ID
         std::cout << ES_GetTimeFromStart(eState) << ",";
-        //std::cout << userID << ",";
+        std::cout << userID << ",";
         std::cout << static_cast<int>(ES_GetWirelessSignalStatus(eState)) << ",";
         std::cout << static_cast<int>(ES_GetHeadsetOn(eState)) << ",";