it works

Signed-off-by: Andrey Parfenov <a1994ndrey@gmail.com>

Author: Andrey1994

app/build.gradle

@@ -31,8 +31,9 @@ dependencies {
     implementation 'androidx.lifecycle:lifecycle-extensions:2.1.0'
     implementation 'androidx.navigation:navigation-fragment:2.3.0'
     implementation 'androidx.navigation:navigation-ui:2.3.0'
+    implementation 'com.github.michael-rapp:android-material-preferences:5.3.2'
+    implementation 'com.jjoe64:graphview:4.2.2'
     testImplementation 'junit:junit:4.12'
     androidTestImplementation 'androidx.test.ext:junit:1.1.1'
     androidTestImplementation 'androidx.test.espresso:espresso-core:3.2.0'
-    implementation 'com.github.michael-rapp:android-material-preferences:5.3.2'
 }

app/src/main/AndroidManifest.xml

@@ -2,10 +2,8 @@
 <manifest xmlns:android="http://schemas.android.com/apk/res/android"
     package="com.example.brainflowplot">
 
-    # for devices which use network
     <uses-permission android:name="android.permission.INTERNET"></uses-permission>
     <uses-permission android:name="android.permission.ACCESS_NETWORK_STATE"></uses-permission>
-    # to use read/write file from/to SD card
     <uses-permission android:name="android.permission.READ_EXTERNAL_STORAGE"></uses-permission>
     <uses-permission android:name="android.permission.WRITE_EXTERNAL_STORAGE"></uses-permission>
 

app/src/main/java/com/example/brainflowplot/DataActivity.java

@@ -1,53 +1,48 @@
 package com.example.brainflowplot;
 
-import android.content.Context;
 import android.content.Intent;
 import android.content.SharedPreferences;
+import android.graphics.Color;
 import android.os.Bundle;
+import android.os.Handler;
 import android.util.Log;
-import android.view.View;
-import android.widget.Toast;
 
 import com.google.android.material.bottomnavigation.BottomNavigationView;
 
 import androidx.appcompat.app.AppCompatActivity;
 import androidx.navigation.NavController;
 import androidx.navigation.Navigation;
-import androidx.navigation.ui.AppBarConfiguration;
 import androidx.navigation.ui.NavigationUI;
 import androidx.preference.PreferenceManager;
 
+import brainflow.AggOperations;
 import brainflow.BoardShim;
 import brainflow.BrainFlowError;
 import brainflow.BrainFlowInputParams;
+import brainflow.DataFilter;
 
 public class DataActivity extends AppCompatActivity {
 
-    public BoardShim boardShim = null;
-    public int samplingRate = 0;
-    public int[] channels = null;
+    // to dont sync threads for dataplot, bandpower, psd, etc each thread will has its own data array  share boardShim object
+    public static BoardShim boardShim = null;
+    public static int samplingRate = 0;
+    public static int[] channels = null;
 
-    private boolean isTryingToConnect = false;
+    public static final int[] colors = {Color.BLUE, Color.YELLOW, Color.RED, Color.MAGENTA, Color.GREEN, Color.CYAN, Color.GRAY, Color.BLACK};
 
     @Override
     protected void onCreate(Bundle savedInstanceState) {
         super.onCreate(savedInstanceState);
         setContentView(R.layout.activity_data);
         BottomNavigationView navView = findViewById(R.id.nav_view);
-        // Passing each menu ID as a set of Ids because each
-        // menu should be considered as top level destinations.
 
-        // comment out these two methods for theme wo actionbar
-        /*
-        AppBarConfiguration appBarConfiguration = new AppBarConfiguration.Builder(
-
-                R.id.navigation_dataplot, R.id.navigation_psdplot, R.id.navigation_bandpowerplot)
-                .build();
-         */
         NavController navController = Navigation.findNavController(this, R.id.nav_host_fragment);
-        // NavigationUI.setupActionBarWithNavController(this, navController, appBarConfiguration);
         NavigationUI.setupWithNavController(navView, navController);
+    }
 
+    @Override
+    protected void onStart() {
+        super.onStart();
         // read settings
         SharedPreferences prefs = PreferenceManager.getDefaultSharedPreferences(this);
         int boardId = Integer.valueOf(prefs.getString(getString(R.string.board_id_key), "-1"));
@@ -60,41 +55,45 @@ protected void onCreate(Bundle savedInstanceState) {
         }
         String dataType = prefs.getString(getString(R.string.data_type_key), "");
 
-        boolean connected = false;
         try {
             BrainFlowInputParams params = new BrainFlowInputParams();
             params.ip_address = ipAddr;
             params.ip_port = ipPort;
             boardShim = new BoardShim(boardId, params);
+            // prepare_session is relatively long operation, doing it in UI thread lead to black window for a few seconds
             boardShim.prepare_session();
             boardShim.start_stream();
-            connected = true;
+            samplingRate = BoardShim.get_sampling_rate(boardId);
+            if (dataType.equals("EEG")) {
+                channels = BoardShim.get_eeg_channels(boardId);
+            } else {
+                if (dataType.equals("EMG")) {
+                    channels = BoardShim.get_emg_channels(boardId);
+                } else {
+                    channels = BoardShim.get_ecg_channels(boardId);
+                }
+            }
+            SettingsActivity.isPrevFailed = false;
         } catch (Exception e) {
-            Context context = getApplicationContext();
-            CharSequence text = "Error occurred, validate provided parameters and your board";
-            int duration = Toast.LENGTH_LONG;
-            Toast toast = Toast.makeText(context, text, duration);
-            toast.show();
             Log.e(getString(R.string.log_tag), e.getMessage());
-            connected = false;
+            SettingsActivity.isPrevFailed = true;
         }
-        if (!connected) {
-            // if failed to connect back to settings page
+        if (SettingsActivity.isPrevFailed) {
+            // if failed to connect go back to settings page
             Intent myIntent = new Intent(this, SettingsActivity.class);
             startActivity(myIntent);
         }
     }
 
     @Override
-    protected void onDestroy() {
+    protected void onStop() {
         try {
             if (boardShim != null) {
                 boardShim.release_session();
             }
         } catch (BrainFlowError e) {
             Log.e(getString(R.string.log_tag), e.getMessage());
         }
-        super.onDestroy();
+        super.onStop();
     }
-
 }

app/src/main/java/com/example/brainflowplot/SettingsActivity.java

@@ -1,15 +1,19 @@
 package com.example.brainflowplot;
 
+import android.content.Context;
 import android.content.Intent;
 import android.os.Bundle;
 import android.view.View;
+import android.widget.Toast;
 
 import androidx.appcompat.app.ActionBar;
 import androidx.appcompat.app.AppCompatActivity;
 import androidx.preference.PreferenceFragmentCompat;
 
 public class SettingsActivity extends AppCompatActivity {
 
+    public static boolean isPrevFailed = false;
+
     @Override
     protected void onCreate(Bundle savedInstanceState) {
         super.onCreate(savedInstanceState);
@@ -23,6 +27,14 @@ protected void onCreate(Bundle savedInstanceState) {
         if (actionBar != null) {
             actionBar.setDisplayHomeAsUpEnabled(true);
         }
+        // make toast in this actvity
+        if (isPrevFailed) {
+            Context context = getApplicationContext();
+            CharSequence text = "Error occurred, validate provided parameters and your board";
+            int duration = Toast.LENGTH_LONG;
+            Toast toast = Toast.makeText(context, text, duration);
+            toast.show();
+        }
     }
 
     public void prepareSession(View view) {

app/src/main/java/com/example/brainflowplot/ui/bandpowerplot/BandPowerPlotFragment.java

@@ -1,35 +1,135 @@
 package com.example.brainflowplot.ui.bandpowerplot;
 
+import android.graphics.Color;
 import android.os.Bundle;
+import android.os.Handler;
+import android.util.Log;
 import android.view.LayoutInflater;
 import android.view.View;
 import android.view.ViewGroup;
-import android.widget.TextView;
 
 import androidx.annotation.NonNull;
-import androidx.annotation.Nullable;
 import androidx.fragment.app.Fragment;
-import androidx.lifecycle.Observer;
-import androidx.lifecycle.ViewModelProviders;
 
+import com.example.brainflowplot.DataActivity;
 import com.example.brainflowplot.R;
+import com.jjoe64.graphview.GraphView;
+import com.jjoe64.graphview.GridLabelRenderer;
+import com.jjoe64.graphview.ValueDependentColor;
+import com.jjoe64.graphview.series.BarGraphSeries;
+import com.jjoe64.graphview.series.DataPoint;
+
+import org.apache.commons.lang3.tuple.ImmutablePair;
+import org.apache.commons.lang3.tuple.Pair;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import brainflow.BrainFlowError;
+import brainflow.DataFilter;
+import brainflow.WindowFunctions;
+
 
 public class BandPowerPlotFragment extends Fragment {
 
-    private BandPowerPlotViewModel bandPowerPlotViewModel;
+    public int windowSize = 2;
+    public int timeSleep = 100;
+    private GraphView graph = null;
+    private BarGraphSeries<DataPoint> series= null;
+
+    private Runnable worker = null;
+    private final Handler handler = new Handler();
+
+    private List<Pair<Integer, Integer>> bands = new ArrayList<Pair<Integer, Integer>>();
+    private final String[] names = {"Delta", "Theta", "Alpha", "Beta", "Gamma"};
 
     public View onCreateView(@NonNull LayoutInflater inflater,
                              ViewGroup container, Bundle savedInstanceState) {
-        bandPowerPlotViewModel =
-                ViewModelProviders.of(this).get(BandPowerPlotViewModel.class);
-        View root = inflater.inflate(R.layout.fragment_bandpowerplot, container, false);
-        final TextView textView = root.findViewById(R.id.text_notifications);
-        bandPowerPlotViewModel.getText().observe(getViewLifecycleOwner(), new Observer<String>() {
+        View rootView = inflater.inflate(R.layout.fragment_bandpowerplot, container, false);
+
+        graph = (GraphView) rootView.findViewById(R.id.bandpowergraph);
+        graph.setTitle("BandPower");
+        graph.getViewport().setXAxisBoundsManual(true);
+        graph.getViewport().setMinX(0);
+        graph.getViewport().setMaxX(5);
+        graph.getViewport().setYAxisBoundsManual(true);
+        graph.getViewport().setMinY(0);
+        graph.getViewport().setMaxY(1);
+        graph.getGridLabelRenderer().setVerticalLabelsVisible(false);
+        graph.getGridLabelRenderer().setHorizontalLabelsVisible(false);
+        graph.getGridLabelRenderer().setGridStyle(GridLabelRenderer.GridStyle.NONE);
+        series = new BarGraphSeries<DataPoint>();
+        series.setValueDependentColor(new ValueDependentColor<DataPoint>() {
             @Override
-            public void onChanged(@Nullable String s) {
-                textView.setText(s);
+            public int get(DataPoint data) {
+                return Color.rgb((int) data.getX() * 255 / 6, (int) Math.abs(data.getY() * 255 / 4), 100);
             }
         });
-        return root;
+        series.setTitle("Delta Theta Alpha Beta Gamma");
+
+        graph.addSeries(series);
+
+        bands.add(new ImmutablePair<Integer, Integer>(1, 4));
+        bands.add(new ImmutablePair<Integer, Integer>(4, 8));
+        bands.add(new ImmutablePair<Integer, Integer>(8, 13));
+        bands.add(new ImmutablePair<Integer, Integer>(13, 30));
+        bands.add(new ImmutablePair<Integer, Integer>(30, 70));
+
+        return rootView;
+    }
+
+    @Override
+    public void onResume() {
+        super.onResume();
+
+        worker = new Runnable() {
+            @Override
+            public void run() {
+                try {
+                    int numDataPoints = DataFilter.get_nearest_power_of_two(DataActivity.samplingRate * windowSize);
+                    double[][] tmpArray = DataActivity.boardShim.get_current_board_data(numDataPoints);
+                    if (tmpArray[0].length != numDataPoints) {
+                        // dont prepend, just wait for more data
+                        return;
+                    }
+                    // average across all channels
+                    DataPoint[] values = new DataPoint[bands.size()];
+                    for (int i = 0; i < values.length; i++) {
+                        values[i] = new DataPoint(i, 0);
+                    }
+                    for (int i = 0; i < DataActivity.channels.length; i++) {
+                        double[] channel = tmpArray[DataActivity.channels[i]];
+                        Pair<double[], double[]> psd = DataFilter.get_psd (channel, 0, channel.length, DataActivity.samplingRate, WindowFunctions.HAMMING.get_code ());
+                        for (int j = 0; j < bands.size(); j++) {
+                            double bandPower = DataFilter.get_band_power(psd, bands.get(j).getLeft(), bands.get(j).getRight());
+                            values[j] = new DataPoint(values[j].getX(), values[j].getY() + bandPower);
+                        }
+                    }
+                    // normalize for colors
+                    double max = 0;
+                    for (int j = 0; j < bands.size(); j++) {
+                        if (values[j].getY() > max) {
+                            max = values[j].getY();
+                        }
+                    }
+                    for (int j = 0; j < bands.size(); j++) {
+                        values[j] = new DataPoint(values[j].getX(), values[j].getY() / max);
+                    }
+                    series.resetData(values);
+                } catch (BrainFlowError e) {
+                    Log.e(getString(R.string.log_tag), e.getMessage());
+                }
+
+                handler.postDelayed(this, timeSleep);
+            }
+        };
+        handler.postDelayed(worker, timeSleep);
+    }
+
+    @Override
+    public void onPause() {
+        handler.removeCallbacks(worker);
+        graph.removeAllSeries();
+        super.onPause();
     }
 }