Changeset 5202e8d in pyramid

Timestamp:

10/11/13 14:30:12 (8 years ago)

Author:

Steve Castellotti <sc@…>

Branches:

master

Children:

85e090e

Parents:

f1b827c

Message:

• RGB processing in ADK mode

File:

• firmware/Puzzlebox_Pyramid/Puzzlebox_Pyramid.ino (modified) (11 diffs)

firmware/Puzzlebox_Pyramid/Puzzlebox_Pyramid.ino

@@ -65 +65 @@
 char _command;
 byte _attention_threshold = 67; //treshold for launching Orbit helicopter
+
+// EEG
+int eegSignal = 0;
+int eegAttention = 0;
+int eegMeditation = 0;
 
 // Operation mode
@@ -155 +160 @@
         Serial.begin(115200);
 
-
         // Android ADK
         delay(500);
         acc.powerOn();
-        delay(2000);
+
+
+        ////////Orbit Flight///////////
+        //initialize all timers except for 0, to save time keeping functions
+        InitTimersSafe(); 
+
+        //sets the IR_frequency for the specified pin
+        bool success = SetPinFrequencySafe(IR, IR_frequency);
+
+        //if the pin IR_frequency was set successfully, turn pin 13 on
+        if (success) {
+                Serial.println("INFO: PWM pin frequency set");  
+        }
+
+        /////////////RGB////////////////
+        Wire.begin(); // join i2c bus as master (address optional for master)
+
+
+        setStartScreen();
+
 
         if (acc.isConnected()) {
                 modeADK = true;
                 Serial.println("INFO: ADK connection detected, setting modeADK");
-        }
-
-        ////////Orbit Flight///////////
-        //initialize all timers except for 0, to save time keeping functions
-        InitTimersSafe(); 
-
-        //sets the IR_frequency for the specified pin
-        bool success = SetPinFrequencySafe(IR, IR_frequency);
-
-        //if the pin IR_frequency was set successfully, turn pin 13 on
-        if(success) {
-                Serial.println("Set PWM pin frequency SUCCESS");  
-        }
-
-        /////////////RGB////////////////
-        Wire.begin(); // join i2c bus as master (address optional for master)
-                
-        
+                setColorWheel(255, 0, 0); // red
+        }
+
+
         //////////Bluetooth///////////
         if (! modeADK) {
-                setupBlueToothConnection();
+                
+                delay(500);
+                
+//              setupBlueToothConnection();
                 
                 //wait 1s and flush the serial buffer
-                setColorWheel(0,0,255);
+                
                 delay(1000);
                 Serial.flush(); //delete?
@@ -283 +295 @@
         }
         
-        else if (Serial1.available()) {
-//              modeBluetooth = true;
-                Serial.println("INFO: Bluetooth input received, setting modeBluetooth");
-        }
-        
         else if (acc.isConnected()) {
                 modeADK = true;
+                Serial1.end();
                 Serial.println("INFO: ADK connection detected, setting modeADK");
+                setColorWheel(255, 0, 0); // red
+        }
+        
+        else if (Serial1.available()) {
+                modeBluetooth = true;
+                Serial.println("INFO: Bluetooth input received, setting modeBluetooth");
         }
         
@@ -369 +383 @@
                                         lastReceivedPacket = millis();
                                         Serial.print("\n");
+
                                         
+                                        if (poorQuality == 200)
+                                                setColorWheel(0,255,0); // Green
+
+
                                         //Update RGB Panel when received data from MindWave Mobile
                                         int att, med;
@@ -426 +445 @@
 
 void setupBlueToothConnection() {
+        
+        setColorWheel(0,0,255); // Blue
 
         Serial1.begin(38400); //Set Bluetooth Module BaudRate (for communicating to ADK) to default baud rate 38400
@@ -445 +466 @@
         if(digitalRead(BUTTON_PIN)==0){ //if needs pair
                 //update RGB Panel, Red Indicates it's pairing new headset
+                setColorWheel(0,0,0); // Black
                 RGB_Panel[5][0]=255;
                 RGB_Panel[5][1]=0;
@@ -508 +530 @@
                 Serial1.print("\r\n+STAUTO=1\r\n");// Permit Auto-connection 
                 //update bottom RGB LED to blue
+                setColorWheel(0,0,0); // Black
                 RGB_Panel[5][0]=0;
                 RGB_Panel[5][1]=0;
@@ -622 +645 @@
         
 } // sendFrame
+
+
+// ################################################################
+
+int calculateMeter(int value) {
+  
+        int result = 0;
+
+        if (value == 0)
+                result = 0;
+        else if (value <= 20)
+                result = 1;
+        else if (value <= 40)
+                result = 2;
+        else if (value <= 60)
+                result = 3;
+        else if (value <= 80)
+                result = 4;
+        else if (value <= 100)
+                result = 5;
+
+        return(result);
+  
+} // calculateMeter
+
+
+// ################################################################
+
+void updateFrameADK() {
+
+        // update RGB_Panel[][] here to set next Frame you want to send
+        // For Example:
+        // RGB_Panel[0][0]=0;
+        // RGB_Panel[0][1]=0;
+        // RGB_Panel[0][2]=255;
+        // will update the LED on 1 o'clock position to be blue.
+
+        // This following code simulates when you wear a headset.
+//      RGB_Panel[5][0]=0;
+//      RGB_Panel[5][1]=255;
+//      RGB_Panel[5][2]=0;
+
+//      int attention=random(0,7); //generates 0~6
+//      int meditation=random(0,7);
+
+
+        int attention = calculateMeter(eegAttention);
+        int meditation = calculateMeter(eegMeditation);
+
+
+        for(int i=6; i<=6+attention; i++){
+                RGB_Panel[i][0]=255;
+                RGB_Panel[i][1]=0;
+                RGB_Panel[i][2]=0;
+        }
+
+        if (meditation==6){
+                RGB_Panel[11][0]=0;
+                RGB_Panel[11][1]=0;
+                RGB_Panel[11][2]=255;
+        }
+                
+        for(int i=4; i>=4-meditation; i--){
+                RGB_Panel[i][0]=0;
+                RGB_Panel[i][1]=0;
+                RGB_Panel[i][2]=255;
+        }
+
+        if (attention==6) 
+                RGB_Panel[11][0]=255;
+                
+        for(int i=6+attention; i<=11; i++){
+                RGB_Panel[i][0]=0;
+                RGB_Panel[i][1]=0;
+                RGB_Panel[i][2]=0;
+        }
+
+        for(int i=4-meditation; i>=0; i--){
+                RGB_Panel[i][0]=0;
+                RGB_Panel[i][1]=0;
+                RGB_Panel[i][2]=0;
+        }
+
+} // updateFrame
 
 
@@ -872 +979 @@
         }
         
-        sendFrame(1000);
+        sendFrame(0);
+        
+}
+
+
+// ################################################################
+
+void setStartScreen() {
+        
+        // White
+        RGB_Panel[11][0]=255;
+        RGB_Panel[11][1]=255;
+        RGB_Panel[11][2]=255;
+        
+        RGB_Panel[3][0]=255;
+        RGB_Panel[3][1]=255;
+        RGB_Panel[3][2]=255;
+        
+        RGB_Panel[7][0]=255;
+        RGB_Panel[7][1]=255;
+        RGB_Panel[7][2]=255;
+        
+        // Red
+        for (int hour=0; hour < 3; hour++) {
+                RGB_Panel[hour][0]=255;
+                RGB_Panel[hour][1]=0;
+                RGB_Panel[hour][2]=0;
+        }
+        
+        // Green
+        for (int hour=4; hour < 7; hour++) {
+                RGB_Panel[hour][0]=0;
+                RGB_Panel[hour][1]=255;
+                RGB_Panel[hour][2]=0;
+        }
+        
+        // Blue
+        for (int hour=8; hour < 11; hour++) {
+                RGB_Panel[hour][0]=0;
+                RGB_Panel[hour][1]=0;
+                RGB_Panel[hour][2]=255;
+        }
+        
+        sendFrame(0);
         
 }
@@ -932 +1082 @@
         // Android ADK
         
-        Serial.println("parseADK()");
-        
         if (acc.isConnected()) {
                 
-                Serial.println("acc.isConnected()");
-                
                 int len = acc.read(msg, sizeof(msg), 1);
                 
-//              if (! modeADK)
-//                      Serial.println("INFO: ADK connection detected, setting modeADK");
-//                      modeADK = true;
-                
-                
-//              Serial.println("len: ", len);
-                
+                if (! modeADK) {
+                        Serial.println("INFO: ADK connection detected, setting modeADK");
+                        modeADK = true;
+                }
                 
                 // Action taken by Arduino is tied to the message it receives from Android
                 
-//              RGB_Panel[11][0]=255;
-//              RGB_Panel[11][1]=255;
-//              RGB_Panel[11][2]=128;
-//              
-//              RGB_Panel[0][0]=255;
-//              RGB_Panel[0][1]=255;
-//              RGB_Panel[0][2]=128;
-//              
-//              RGB_Panel[1][0]=255;
-//              RGB_Panel[1][1]=255;
-//              RGB_Panel[1][2]=128;
-                
-//              setColorWheel(255, 0, 255); // purple
-                setColorWheel(255, 0, 0); // red
-                
-                sendFrame(1000);
-        
+                if (len > 0) {
+                        if (msg[0] == 0x1) {
+                                // Reserved
+                        }
+                        else if(msg[0] == 0x2) {
+                                // Reserved
+                        }
+                        else if (msg[0] == 0x3) {
+                                // Reserved
+                        }
+                        
+                        // EEG Signal
+                        else if (msg[0] == 0x4) {
+                                eegSignal = 0;
+                                if (msg[1] == 0x1) {
+                                        eegSignal = 10;
+                                } else if (msg[1] == 0x2) {
+                                        eegSignal = 20;
+                                } else if (msg[1] == 0x3) {
+                                        eegSignal = 30;
+                                } else if (msg[1] == 0x4) {
+                                        eegSignal = 40;
+                                } else if (msg[1] == 0x5) {
+                                        eegSignal = 50;
+                                } else if (msg[1] == 0x6) {
+                                        eegSignal = 60;
+                                } else if (msg[1] == 0x7) {
+                                        eegSignal = 70;
+                                } else if (msg[1] == 0x8) {
+                                        eegSignal = 80;
+                                } else if (msg[1] == 0x9) {
+                                        eegSignal = 90;
+                                } else {
+                                        eegSignal = 100;
+                                }
+                        }
+                        
+                        
+                        // EEG Attention
+                        else if (msg[0] == 0x5) {
+                                eegAttention = 0;
+                                if (msg[1] == 0x1) {
+                                        eegAttention = 10;
+                                } else if (msg[1] == 0x2) {
+                                        eegAttention = 20;
+                                } else if (msg[1] == 0x3) {
+                                        eegAttention = 30;
+                                } else if (msg[1] == 0x4) {
+                                        eegAttention = 40;
+                                } else if (msg[1] == 0x5) {
+                                        eegAttention = 50;
+                                } else if (msg[1] == 0x6) {
+                                        eegAttention = 60;
+                                } else if (msg[1] == 0x7) {
+                                        eegAttention = 70;
+                                } else if (msg[1] == 0x8) {
+                                        eegAttention = 80;
+                                } else if (msg[1] == 0x9) {
+                                        eegAttention = 90;
+                                } else {
+                                        eegAttention = 100;
+                                }
+                        }
+                        
+                        
+                        // EEG Meditation
+                        else if (msg[0] == 0x6) {
+                                eegMeditation = 0;
+                                if (msg[1] == 0x1) {
+                                        eegMeditation = 10;
+                                } else if (msg[1] == 0x2) {
+                                        eegMeditation = 20;
+                                } else if (msg[1] == 0x3) {
+                                        eegMeditation = 30;
+                                } else if (msg[1] == 0x4) {
+                                        eegMeditation = 40;
+                                } else if (msg[1] == 0x5) {
+                                        eegMeditation = 50;
+                                } else if (msg[1] == 0x6) {
+                                        eegMeditation = 60;
+                                } else if (msg[1] == 0x7) {
+                                        eegMeditation = 70;
+                                } else if (msg[1] == 0x8) {
+                                        eegMeditation = 80;
+                                } else if (msg[1] == 0x9) {
+                                        eegMeditation = 90;
+                                } else {
+                                        eegMeditation = 100;
+                                }
+                        }
+                
+                
+//                      updateFrame(eegAttention, eegMeditation, eegSignal);
+                        updateFrameADK();
+                        sendFrame(1000);
+                
+                
+                } // len
+                else {
+//                      setColorWheel(255, 0, 0); // red
+                        delay(10); //cycle();
+                }
         } // if acc.isConnected()
 
@@ -983 +1211 @@
 
 
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+