Merge pull request #3 from lucidcode/update-brainflow

IAmCoder · web-flow · commit 144d6b17d78e · 2024-10-01T14:58:16.000+10:00
Update BrainFlow to 5.14
diff --git a/BrainFlow/BrainFlow.csproj b/BrainFlow/BrainFlow.csproj
@@ -63,8 +63,9 @@
     <CodeAnalysisRuleSet>AllRules.ruleset</CodeAnalysisRuleSet>
   </PropertyGroup>
   <ItemGroup>
-    <Reference Include="brainflow, Version=1.0.0.0, Culture=neutral, processorArchitecture=MSIL">
-      <HintPath>..\packages\brainflow.5.2.0\lib\net45\brainflow.dll</HintPath>
+    <Reference Include="brainflow, Version=5.14.0.0, Culture=neutral, processorArchitecture=MSIL">
+      <HintPath>..\packages\brainflow.5.14.0\lib\netstandard2.0\brainflow.dll</HintPath>
+      <Private>True</Private>
     </Reference>
     <Reference Include="lucidcode.Controls, Version=1.0.0.1, Culture=neutral, processorArchitecture=MSIL">
       <SpecificVersion>False</SpecificVersion>
@@ -90,6 +91,8 @@
     <Reference Include="System.Xml" />
   </ItemGroup>
   <ItemGroup>
+    <Compile Include="Channels\Restfulness.cs" />
+    <Compile Include="Channels\Mindfulness.cs" />
     <Compile Include="UserSettings.cs" />
     <Compile Include="Board.cs" />
     <Compile Include="Channels\Channel16.cs" />
@@ -129,13 +132,6 @@
     <None Include="packages.config" />
   </ItemGroup>
   <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
-  <Import Project="..\packages\brainflow.5.2.0\build\brainflow.targets" Condition="Exists('..\packages\brainflow.5.2.0\build\brainflow.targets')" />
-  <Target Name="EnsureNuGetPackageBuildImports" BeforeTargets="PrepareForBuild">
-    <PropertyGroup>
-      <ErrorText>This project references NuGet package(s) that are missing on this computer. Use NuGet Package Restore to download them.  For more information, see http://go.microsoft.com/fwlink/?LinkID=322105. The missing file is {0}.</ErrorText>
-    </PropertyGroup>
-    <Error Condition="!Exists('..\packages\brainflow.5.2.0\build\brainflow.targets')" Text="$([System.String]::Format('$(ErrorText)', '..\packages\brainflow.5.2.0\build\brainflow.targets'))" />
-  </Target>
   <!-- To modify your build process, add your task inside one of the targets below and uncomment it. 
        Other similar extension points exist, see Microsoft.Common.targets.
   <Target Name="BeforeBuild">
diff --git a/BrainFlow/Channels/Mindfulness.cs b/BrainFlow/Channels/Mindfulness.cs
@@ -0,0 +1,37 @@
+﻿using System;
+
+namespace lucidcode.LucidScribe.Plugin.BrainFlow
+{
+    namespace Mindfulness
+    {
+        public class PluginHandler : Interface.LucidPluginBase
+        {
+            public override string Name { get { return "Mindfulness"; } }
+
+            public override bool Initialize()
+            {
+                try
+                {
+                    return Device.Initialize();
+                }
+                catch (Exception ex)
+                {
+                    throw (new Exception("The '" + Name + "' plugin failed to initialize: " + ex.Message));
+                }
+            }
+
+            public override double Value
+            {
+                get
+                {
+                    return Device.GetMetric(brainflow.BrainFlowMetrics.MINDFULNESS);
+                }
+            }
+
+            public override void Dispose()
+            {
+                Device.Dispose();
+            }
+        }
+    }
+}
diff --git a/BrainFlow/Channels/Restfulness.cs b/BrainFlow/Channels/Restfulness.cs
@@ -0,0 +1,37 @@
+﻿using System;
+
+namespace lucidcode.LucidScribe.Plugin.BrainFlow
+{
+    namespace Restfulness
+    {
+        public class PluginHandler : Interface.LucidPluginBase
+        {
+            public override string Name { get { return "Restfulness"; } }
+
+            public override bool Initialize()
+            {
+                try
+                {
+                    return Device.Initialize();
+                }
+                catch (Exception ex)
+                {
+                    throw (new Exception("The '" + Name + "' plugin failed to initialize: " + ex.Message));
+                }
+            }
+
+            public override double Value
+            {
+                get
+                {
+                    return Device.GetMetric(brainflow.BrainFlowMetrics.RESTFULNESS);
+                }
+            }
+
+            public override void Dispose()
+            {
+                Device.Dispose();
+            }
+        }
+    }
+}
diff --git a/BrainFlow/PluginHandler.cs b/BrainFlow/PluginHandler.cs
@@ -1,4 +1,5 @@
 ﻿using System;
+using System.Collections.Generic;
 using System.Threading;
 using System.Windows.Forms;
 using brainflow;
@@ -12,13 +13,24 @@ public static class Device
         static bool initError;
         static bool disposing;
 
+        static Thread boardThread;
+
         static BoardShim boardShim;
         static int[] eegChannels;
-        static Thread boardThread;
+        static int samplingRate;
 
         static double[] eegValues;
         static double[] eegTicks;
         static bool[] clearValues;
+        static bool readMetrics = false;
+        static bool hasMetrics = false;
+        static double[] featureVector;
+
+        static MLModel mindfulnessMlModel;
+        static BrainFlowModelParams mindfulnessModelParams;
+
+        static MLModel restfulnessMlModel;
+        static BrainFlowModelParams restfulnessModelParams;
 
         public static EventHandler<EventArgs> Channel1Changed;
         public static EventHandler<EventArgs> Channel2Changed;
@@ -81,24 +93,35 @@ public static Boolean Initialize()
             return true;
         }
 
-        static void GetBoardData(int board_id, BrainFlowInputParams input_params)
+        static void GetBoardData(int boardId, BrainFlowInputParams inputParams)
         {
-            boardShim = new BoardShim(board_id, input_params);
+            boardShim = new BoardShim(boardId, inputParams);
 
             boardShim.prepare_session();
             boardShim.start_stream();
 
-            eegChannels = BoardShim.get_eeg_channels(board_id);
+            samplingRate = BoardShim.get_sampling_rate(boardId);
+            eegChannels = BoardShim.get_eeg_channels(boardId);
             eegValues = new double[eegChannels.Length];
             eegTicks = new double[eegChannels.Length];
             clearValues = new bool[eegChannels.Length];
 
+            mindfulnessModelParams = new BrainFlowModelParams((int)BrainFlowMetrics.MINDFULNESS, (int)BrainFlowClassifiers.DEFAULT_CLASSIFIER);
+            mindfulnessMlModel = new MLModel(mindfulnessModelParams);
+            mindfulnessMlModel.prepare();
+
+            restfulnessModelParams = new BrainFlowModelParams((int)BrainFlowMetrics.RESTFULNESS, (int)BrainFlowClassifiers.DEFAULT_CLASSIFIER);
+            restfulnessMlModel = new MLModel(mindfulnessModelParams);
+            restfulnessMlModel.prepare();
+
+            var bandData = new List<double[,]>();
+
             do
             {
-                double[,] unprocessed_data = boardShim.get_board_data(20);
+                double[,] boardData = boardShim.get_board_data();
                 foreach (var index in eegChannels)
                 {
-                    double[] rows = unprocessed_data.GetRow(index);
+                    double[] rows = boardData.GetRow(index);
 
                     foreach (var row in rows)
                     {
@@ -130,6 +153,31 @@ static void GetBoardData(int board_id, BrainFlowInputParams input_params)
                     if (index == 15 && Channel15Changed != null) Channel15Changed(string.Join(",", rows), null);
                     if (index == 16 && Channel16Changed != null) Channel16Changed(string.Join(",", rows), null);
                 }
+
+                if (readMetrics)
+                {
+                    bandData.Add(boardData);
+
+                    if (bandData.Count > 512)
+                    {
+                        bandData.RemoveAt(0);
+
+                        double[,] data = new double[0, 0];
+
+                        foreach (var d in bandData)
+                        {
+                            data = mergeArrays(data, d, eegChannels.Length * 2);
+                        }
+
+                        if (data.Length > 16384)
+                        {
+                            Tuple<double[], double[]> bands = DataFilter.get_avg_band_powers(data, eegChannels, samplingRate, true);
+                            featureVector = bands.Item1;
+                            hasMetrics = true;
+                        }
+
+                    }
+                }
             }
             while (!disposing);
         }
@@ -154,5 +202,54 @@ public static double GetEEG(int index)
             clearValues[index - 1] = true;
             return average;
         }
+
+        public static double GetMetric(BrainFlowMetrics metric)
+        {
+            readMetrics = true;
+
+            if (!initialized) return 0;
+            if (!hasMetrics) return 0;
+
+            double prediction = 0;
+
+            if (metric == BrainFlowMetrics.MINDFULNESS)
+            {
+                prediction = mindfulnessMlModel.predict(featureVector)[0];
+            }
+            else if (metric == BrainFlowMetrics.RESTFULNESS)
+            {
+                prediction = restfulnessMlModel.predict(featureVector)[0];
+            }
+
+            return prediction;
+        }
+
+        static double[,] mergeArrays(double[,] data1, double[,] data2, int depth)
+        {
+            var data1Length = data1.Length / depth;
+            var data2Length = data2.Length / depth;
+
+            double[,] data = new double[depth, data1Length + data2Length]; ;
+
+            int i = 0;
+            for (; i < depth; i++)
+            {
+                for (int j = 0; j < data1Length; j++)
+                {
+                    data[i, j] = data1[i, j];
+                }
+            }
+            i = 0;
+            for (; i < depth; i++)
+            {
+                for (int j = 0; j < data2Length; j++)
+                {
+                    data[i, data1Length + j] = data2[i, j];
+                }
+
+            }
+
+            return data;
+        }
     }
 }
diff --git a/BrainFlow/Plugins/Mindfulness.lsd b/BrainFlow/Plugins/Mindfulness.lsd
@@ -0,0 +1,8 @@
+<LucidScribeData>
+  <Plugin>
+    <Enabled>False</Enabled>
+    <Baseline>
+    </Baseline>
+    <Color>-1</Color>
+  </Plugin>
+</LucidScribeData>
diff --git a/BrainFlow/Plugins/Restfulness.lsd b/BrainFlow/Plugins/Restfulness.lsd
@@ -0,0 +1,8 @@
+<LucidScribeData>
+  <Plugin>
+    <Enabled>False</Enabled>
+    <Baseline>
+    </Baseline>
+    <Color>-1</Color>
+  </Plugin>
+</LucidScribeData>
diff --git a/BrainFlow/Properties/AssemblyInfo.cs b/BrainFlow/Properties/AssemblyInfo.cs
@@ -10,7 +10,7 @@
 [assembly: AssemblyConfiguration("")]
 [assembly: AssemblyCompany("lucidcode")]
 [assembly: AssemblyProduct("Plugin.BrainFlow")]
-[assembly: AssemblyCopyright("Copyright © lucidcode 2022 - 2022")]
+[assembly: AssemblyCopyright("Copyright © lucidcode 2022 - 2024")]
 [assembly: AssemblyTrademark("")]
 [assembly: AssemblyCulture("")]
 
@@ -32,5 +32,5 @@
 // You can specify all the values or you can default the Build and Revision Numbers 
 // by using the '*' as shown below:
 // [assembly: AssemblyVersion("1.0.*")]
-[assembly: AssemblyVersion("1.0.1.0")]
-[assembly: AssemblyFileVersion("1.0.1.0")]
+[assembly: AssemblyVersion("1.0.3.0")]
+[assembly: AssemblyFileVersion("1.0.3.0")]
diff --git a/BrainFlow/packages.config b/BrainFlow/packages.config
@@ -1,5 +1,5 @@
 ﻿<?xml version="1.0" encoding="utf-8"?>
 <packages>
-  <package id="brainflow" version="5.2.0" targetFramework="net48" />
+  <package id="brainflow" version="5.14.0" targetFramework="net48" />
   <package id="Newtonsoft.Json" version="13.0.1" targetFramework="net48" />
 </packages>
diff --git a/Installer/Lucid Scribe BrainFlow.vdproj b/Installer/Lucid Scribe BrainFlow.vdproj
