Merge pull request #1174 from OpenBCI/software-marker-widget

Add Software Marker Widget

Author: retiutut

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ### Improvements
 - Update repository to Processing 4.2 #1111
+- Add Software Marker Widget #1091
 
 # v5.2.1
 

Networking-Test-Kit/LSL/lslStreamTest.py

@@ -29,9 +29,11 @@ def testLSLSamplingRate():
             # print( len(chunk) )
             totalNumSamples += len(chunk)
             # print(chunk)
+            i = 0
             for sample in chunk:
-                print(sample)
+                print(sample, timestamp[i])
                 validSamples += 1
+                i += 1
 
     print( "Number of Chunks and Samples == {} , {}".format(numChunks, totalNumSamples) )
     print( "Valid Samples and Duration == {} / {}".format(validSamples, duration) )

Networking-Test-Kit/UDP/udp_receive_marker.py

@@ -0,0 +1,106 @@
+import socket
+import sys
+import time
+import argparse
+import signal
+import struct
+import os
+import json
+
+numSamples = 0
+
+# Print received message to console
+def print_message(*args):
+    try:
+        #print(args[0]) #added to see raw data 
+        obj = json.loads(args[0].decode())
+        print(obj.get('data'))
+        num_samples_packet = len(obj.get('data'))
+        global numSamples
+        print("NumSamplesInPacket_Marker == " + str(num_samples_packet))
+        numSamples += num_samples_packet
+        if obj:
+            return True
+        else:
+            return False
+    except BaseException as e:
+        print(e) 
+        return False
+
+# Clean exit from print mode
+def exit_print(signal, frame):
+    print("Closing listener")
+    sys.exit(0)
+
+# Record received message in text file
+def record_to_file(*args):
+    textfile.write(str(time.time()) + ",")
+    obj = json.loads(args[0].decode())
+    #print(obj.get('data'))
+    textfile.write(json.dumps(obj))
+    textfile.write("\n")
+
+# Save recording, clean exit from record mode
+def close_file(*args):
+    print("\nFILE SAVED")
+    textfile.close()
+    sys.exit(0)
+
+if __name__ == "__main__":
+  # Collect command line arguments
+  parser = argparse.ArgumentParser()
+  parser.add_argument("--ip",
+      default="127.0.0.1", help="The ip to listen on")
+  parser.add_argument("--port",
+      type=int, default=12345, help="The port to listen on")
+  parser.add_argument("--address",default="/openbci", help="address to listen to")
+  parser.add_argument("--option",default="print",help="Debugger option")
+  parser.add_argument("--len",default=9,help="Debugger option")
+  args = parser.parse_args()
+
+  # Set up necessary parameters from command line
+  length =  int(args.len)
+  if args.option=="print":
+      signal.signal(signal.SIGINT, exit_print)
+  elif args.option=="record":
+      i = 0
+      while os.path.exists("udp_test%s.txt" % i):
+        i += 1
+      filename = "udp_test%i.txt" % i
+      textfile = open(filename, "w")
+      textfile.write("time,address,messages\n")
+      textfile.write("-------------------------\n")
+      print("Recording to %s" % filename)
+      signal.signal(signal.SIGINT, close_file)
+
+  # Connect to socket
+  sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+  sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+  server_address = (args.ip, args.port)
+  sock.bind(server_address)
+
+  # Display socket attributes
+  print('--------------------')
+  print("-- UDP LISTENER -- ")
+  print('--------------------')
+  print("IP:", args.ip)
+  print("PORT:", args.port)
+  print('--------------------')
+  print("%s option selected" % args.option)
+
+  # Receive messages
+  print("Listening...")
+  start = time.time()
+  
+  duration = 10
+  while time.time() <= start + duration:
+    data, addr = sock.recvfrom(20000) # buffer size is 20000 bytes
+    if args.option=="print":
+        print_message(data)
+    elif args.option=="record":
+        record_to_file(data)
+        numSamples += 1
+
+print( "Samples == {}".format(numSamples) )
+print( "Duration == {}".format(duration) )
+print( "Avg Sampling Rate == {}".format(numSamples / duration) )

Networking-Test-Kit/UDP/udp_send_marker.py

@@ -0,0 +1,42 @@
+import socket
+import random
+import struct
+import time
+import argparse
+
+if __name__ == "__main__":
+
+    test_duration = 10
+
+    # Collect command line arguments
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--ip", default="127.0.0.1",
+      help="The IP of the UDP server")
+    parser.add_argument("--port", type=int, default=12350,
+      help="The port the UDP server is sending to")
+    args = parser.parse_args()
+
+    # Establish UDP socket
+    UDP_IP = args.ip
+    UDP_PORT = args.port
+    sock = socket.socket(socket.AF_INET, # Internet
+                          socket.SOCK_DGRAM) # UDP
+
+    # Display socket attributes
+    print('---------------------')
+    print("-- UDP MARKER SEND -- ")
+    print('---------------------')
+    print("IP:", args.ip)
+    print("PORT:", args.port)
+    print('--------------------=')
+
+    # Send test data
+    start = time.time()
+    while time.time() <= start + test_duration:
+      # generate random float
+      marker = random.uniform(0, 4)
+      # package as byte array
+      msg = struct.pack('!d', marker)
+      # send through socket
+      sock.sendto(msg, (UDP_IP, UDP_PORT))
+      time.sleep(.25)

OpenBCI_GUI/Board.pde

@@ -107,6 +107,10 @@ abstract class Board implements DataSource {
 
     public abstract Pair <Boolean, String> sendCommand(String command);
 
+    public abstract void insertMarker(int value);
+
+    public abstract void insertMarker(double value);
+
     // ***************************************
     // protected methods implemented by board
 

OpenBCI_GUI/BoardBrainFlowSynthetic.pde

@@ -96,9 +96,10 @@ class BoardBrainFlowSynthetic extends BoardBrainFlow implements AccelerometerCap
 
     @Override
     protected void addChannelNamesInternal(String[] channelNames) {
-        for (int i=0; i<getAccelerometerChannels().length; i++) {
+        for (int i = 0; i < getAccelerometerChannels().length; i++) {
             channelNames[getAccelerometerChannels()[i]] = "Accel Channel " + i;
         }
+        channelNames[getMarkerChannel()] = "Marker Channel";
     }
 
     @Override

OpenBCI_GUI/BoardBrainflow.pde

@@ -12,13 +12,16 @@ abstract class BoardBrainFlow extends Board {
     protected int sampleIndexChannelCache = -1;
     protected int timeStampChannelCache = -1;
     protected int totalChannelsCache = -1;
+    protected int markerChannelCache = -1;
     protected int[] exgChannelsCache = null;
     protected int[] otherChannelsCache = null;
 
     protected boolean streaming = false;
     protected double time_last_datapoint = -1.0;
     protected boolean data_popup_displayed = false;
 
+    private DateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSS");
+
     /* Abstract Functions.
      * Implement these in your board.
      */
@@ -300,4 +303,39 @@ abstract class BoardBrainFlow extends Board {
 
         return otherChannelsCache;
     }
+
+    @Override
+    public int getMarkerChannel() {
+        if (markerChannelCache < 0) {
+            try {
+                markerChannelCache = BoardShim.get_marker_channel(getBoardIdInt());
+            } catch (BrainFlowError e) {
+                e.printStackTrace();
+            }
+        }
+
+        return markerChannelCache;
+    }
+
+    @Override
+    public void insertMarker(double value) {
+        if (isConnected() && streaming) {
+            try {
+                boardShim.insert_marker(value);
+                String currentTimeString = dateFormat.format(new Date());
+                StringBuilder markerNotification = new StringBuilder("Inserted marker ");
+                markerNotification.append(value);
+                markerNotification.append(" at approximately: ");
+                markerNotification.append(currentTimeString);
+                println(markerNotification.toString());
+            } catch (BrainFlowError e) {
+                e.printStackTrace();
+            }
+        }
+    }
+
+    @Override
+    public void insertMarker(int value) {
+        insertMarker((double) value);
+    }
 };

OpenBCI_GUI/BoardCyton.pde

@@ -608,6 +608,7 @@ implements ImpedanceSettingsBoard, AccelerometerCapableBoard, AnalogCapableBoard
         for (int i=0; i<getAnalogChannels().length; i++) {
             channelNames[getAnalogChannels()[i]] = "Analog Channel " + i;
         }
+        channelNames[getMarkerChannel()] = "Marker Channel";
     }
 
     @Override

OpenBCI_GUI/BoardGanglion.pde

@@ -257,6 +257,7 @@ abstract class BoardGanglion extends BoardBrainFlow implements AccelerometerCapa
         for (int i=0; i<getAccelerometerChannels().length; i++) {
             channelNames[getAccelerometerChannels()[i]] = "Accel Channel " + i;
         }
+        channelNames[getMarkerChannel()] = "Marker Channel";
     }
 
     @Override

OpenBCI_GUI/BoardNull.pde

@@ -51,6 +51,21 @@ class BoardNull extends Board {
         return 0;
     }
 
+    @Override
+    public int getMarkerChannel() {
+        return 0;
+    }
+
+    @Override
+    public void insertMarker(int marker) {
+        // empty
+    }
+
+    @Override
+    public void insertMarker(double value) {
+        // empty
+    }
+
     @Override
     public void setEXGChannelActive(int channelIndex, boolean active) {
         // empty