Merge pull request #159 from OpenBCI/development

Development

Author: AJ Keller

.gitignore

@@ -6,3 +6,4 @@ OpenBCI_GUI/temp.txt
 OpenBCI_GUI/t-temp.txt
 OpenBCI_GUI/build/source/*.java
 *.app
+OpenBCI_GUI/data/EEG_Data/*

CHANGELOG.md

@@ -1,3 +1,13 @@
+# 2.2.0
+
+### Bug Fixes
+* Fix #151 - Incorrect number of channels on playback caused index out of bounds errors.
+* Addresses #149 - Allows for proper scaling of channels with four thanks to #151 #157
+
+### New Features
+* Band power widget #153 (thanks @sunwangshu)
+* Closes #138 - Able to drag and drop the electrodes on the head map  (thanks @liqwid)
+
 # 2.1.2
 
 ### Bug Fixes

Icon

@@ -1475,4 +1475,4 @@ public void convertSDFile() {
       dataWriter.println();
     }
   }
-}
+}

OpenBCI_GUI/ControlPanel.pde

@@ -11,6 +11,14 @@ HashMap<String,float[][]> processed_file;
 HashMap<Integer,String> index_of_times;
 HashMap<String,Integer> index_of_times_rev;
 
+// indexs
+final int DELTA = 0; // 1-4 Hz
+final int THETA = 1; // 4-8 Hz
+final int ALPHA = 2; // 8-13 Hz
+final int BETA = 3; // 13-30 Hz
+final int GAMMA = 4; // 30-55 Hz
+
+
 //------------------------------------------------------------------------
 //                       Global Functions
 //------------------------------------------------------------------------
@@ -109,6 +117,8 @@ int getDataIfAvailable(int pointCounter) {
       //if (eegDataSource==DATASOURCE_PLAYBACKFILE) println("OpenBCI_GUI: getDataIfAvailable: currentTableRowIndex = " + currentTableRowIndex);
       //println("OpenBCI_GUI: getDataIfAvailable: pointCounter = " + pointCounter);
     } // close "has enough time passed"
+    else{
+    }
   }
   return pointCounter;
 }
@@ -154,6 +164,7 @@ void processNewData() {
   // w_openbionics.process();
 
   dataProcessing_user.process(yLittleBuff_uV, dataBuffY_uV, dataBuffY_filtY_uV, fftBuff);
+  dataProcessing.newDataToSend = true;
 
   //look to see if the latest data is railed so that we can notify the user on the GUI
   for (int Ichan=0; Ichan < nchan; Ichan++) is_railed[Ichan].update(dataPacketBuff[lastReadDataPacketInd].values[Ichan]);
@@ -330,12 +341,14 @@ int getPlaybackDataFromTable(Table datatable, int currentTableRowIndex, float sc
 
     if(!isRunning){
       try{
-        if(!isOldData) row.getString(nchan+4);
-        else row.getString(nchan+3);
+        row.getString(nchan+3);
 
-        nchan = 16;
+        // nchan = 16; AJK 5/31/17 see issue #151
+      }
+      catch (ArrayIndexOutOfBoundsException e){
+        println(e);
+        println("8 Channel");
       }
-      catch (ArrayIndexOutOfBoundsException e){ println("8 Channel");}
     }
 
   }
@@ -357,14 +370,26 @@ class DataProcessing {
   private int currentNotch_ind = 0;  // set to 0 to default to 60Hz, set to 1 to default to 50Hz
   float data_std_uV[];
   float polarity[];
-
+  boolean newDataToSend;
+  private String[] binNames;
+  final int[] processing_band_low_Hz = {
+    1, 4, 8, 13, 30
+  }; //lower bound for each frequency band of interest (2D classifier only)
+  final int[] processing_band_high_Hz = {
+    4, 8, 13, 30, 55
+  };  //upper bound for each frequency band of interest
+  float avgPowerInBins[][];
+  float headWidePower[];
+  int numBins;
 
   DataProcessing(int NCHAN, float sample_rate_Hz) {
     nchan = NCHAN;
     fs_Hz = sample_rate_Hz;
     data_std_uV = new float[nchan];
     polarity = new float[nchan];
-
+    newDataToSend = false;
+    avgPowerInBins = new float[nchan][processing_band_low_Hz.length];
+    headWidePower = new float[processing_band_low_Hz.length];
 
     //check to make sure the sample rate is acceptable and then define the filters
     if (abs(fs_Hz-250.0f) < 1.0) {
@@ -675,8 +700,25 @@ class DataProcessing {
         }
         fftBuff[Ichan].setBand(I, (float)foo); //put the smoothed data back into the fftBuff data holder for use by everyone else
       } //end loop over FFT bins
+      for (int i = 0; i < processing_band_low_Hz.length; i++) {
+        float sum = 0;
+        for (int j = processing_band_low_Hz[i]; j < processing_band_high_Hz[i]; j++) {
+          sum += fftBuff[Ichan].getBand(j);
+        }
+        avgPowerInBins[Ichan][i] = sum;
+      }
     } //end the loop over channels.
+    for (int i = 0; i < processing_band_low_Hz.length; i++) {
+      float sum = 0;
+
+      for (int j = 0; j < nchan; j++) {
+        sum += avgPowerInBins[j][i];
+      }
+      headWidePower[i] = sum/nchan;
+    }
 
+    //delta in channel 2 ... avgPowerInBins[1][DELTA];
+    //headwide beta ... headWidePower[BETA];
 
     //find strongest channel
     int refChanInd = findMax(data_std_uV);
@@ -696,5 +738,12 @@ class DataProcessing {
         polarity[Ichan]=-1.0;
       }
     }
+
+    // println("Brain Wide DELTA = " + headWidePower[DELTA]);
+    // println("Brain Wide THETA = " + headWidePower[THETA]);
+    // println("Brain Wide ALPHA = " + headWidePower[ALPHA]);
+    // println("Brain Wide BETA  = " + headWidePower[BETA]);
+    // println("Brain Wide GAMMA = " + headWidePower[GAMMA]);
+
   }
 }

OpenBCI_GUI/DataLogging.pde

@@ -98,14 +98,6 @@ void serialEvent(Serial port){
         numPacketsDropped = 0;
       }
 
-      //If networking enabled --> send data every sample if 8 channels or every other sample if 16 channels
-      if (networkType !=0) {
-        if (nchan==8) {
-          sendRawData_dataPacket(dataPacketBuff[curDataPacketInd], openBCI.get_scale_fac_uVolts_per_count(), openBCI.get_scale_fac_accel_G_per_count());
-        } else if ((nchan==16) && ((dataPacketBuff[curDataPacketInd].sampleIndex %2)!=1)) {
-          sendRawData_dataPacket(dataPacketBuff[curDataPacketInd], openBCI.get_scale_fac_uVolts_per_count(), openBCI.get_scale_fac_accel_G_per_count());
-        }
-      }
       switch (outputDataSource) {
       case OUTPUT_SOURCE_ODF:
         fileoutput_odf.writeRawData_dataPacket(dataPacketBuff[curDataPacketInd], openBCI.get_scale_fac_uVolts_per_count(), openBCI.get_scale_fac_accel_G_per_count());

OpenBCI_GUI/DataProcessing.pde

@@ -236,6 +236,7 @@ void parseKey(char val) {
     case 's':
       println("case s...");
       stopRunning();
+
       // stopButtonWasPressed();
       break;
     case 'b':
@@ -443,7 +444,16 @@ void parseKeycode(int val) {
   }
 }
 
+void mouseDragged() {
+
+  if (systemMode >= SYSTEMMODE_POSTINIT) {
 
+    //calling mouse dragged inly outside of Control Panel
+    if (controlPanel.isOpen == false) {
+      wm.mouseDragged();
+    }
+  }
+}
 //swtich yard if a click is detected
 void mousePressed() {