|
| 1 | +############################################################################# |
| 2 | +## BrainFlow + LSL ## |
| 3 | +## Use BrainFlow to read data from board send it as an LSL stream ## |
| 4 | +############################################################################# |
| 5 | + |
| 6 | +# Install dependencies with: |
| 7 | +# pip install --upgrade numpy brainflow pylsl |
| 8 | + |
| 9 | +# Here are example commands using Cyton and get_exg_channels()from BrainFlow. This has only been tested with Cyton + Dongle, for now. |
| 10 | + |
| 11 | +# Mac: |
| 12 | +# python3 Networking-Test-Kit/LSL/brainflow_lsl.py --board-id 2 --serial-port /dev/cu.usbserial-DM00D7TW --name test --data-type EXG --channel-names 1,2,3,4,5,6,7,8 --uid brainflow |
| 13 | + |
| 14 | +# Windows: |
| 15 | +# python3 Networking-Test-Kit/LSL/brainflow_lsl.py --board-id 2 --serial-port COM3 --name test --data-type EXG --channel-names 1,2,3,4,5,6,7,8 --uid brainflow |
| 16 | + |
| 17 | +import argparse |
| 18 | +import time |
| 19 | +import numpy as np |
| 20 | + |
| 21 | +from queue import Queue |
| 22 | + |
| 23 | +import brainflow |
| 24 | +from brainflow.board_shim import BoardShim, BrainFlowInputParams |
| 25 | +from brainflow.data_filter import DataFilter, FilterTypes, AggOperations |
| 26 | + |
| 27 | +from pylsl import StreamInfo, StreamOutlet, local_clock |
| 28 | + |
| 29 | +def channel_select(board, board_id, data_type): |
| 30 | + switcher = { |
| 31 | + 'EXG': board.get_exg_channels(board_id), |
| 32 | + # can add more |
| 33 | + } |
| 34 | + |
| 35 | + return switcher.get(data_type, "error") |
| 36 | + |
| 37 | +def main(): |
| 38 | + BoardShim.enable_dev_board_logger() |
| 39 | + |
| 40 | + parser = argparse.ArgumentParser() |
| 41 | + |
| 42 | + # brainflow params - use docs to check which parameters are required for specific board, e.g. for Cyton set serial port |
| 43 | + parser.add_argument('--timeout', type=int, help='timeout for device discovery or connection', required=False, default=0) |
| 44 | + parser.add_argument('--ip-address', type=str, help='ip address', required=False, default='') |
| 45 | + parser.add_argument('--board-id', type=int, help='board id, check docs to get a list of supported boards', required=True) |
| 46 | + parser.add_argument('--serial-port', type=str, help='serial port', required=False, default='') |
| 47 | + parser.add_argument('--streamer-params', type=str, help='streamer params', required=False, default='') |
| 48 | + |
| 49 | + # LSL params |
| 50 | + parser.add_argument('--name', type=str, help='name', required=True) |
| 51 | + parser.add_argument('--data-type', type=str, help='data type', required=True) |
| 52 | + parser.add_argument('--channel-names', type=str, help='channel names', required=True) |
| 53 | + parser.add_argument('--uid', type=str, help='uid', required=True) |
| 54 | + |
| 55 | + args = parser.parse_args() |
| 56 | + |
| 57 | + # brainflow initialization |
| 58 | + params = BrainFlowInputParams() |
| 59 | + params.serial_port = args.serial_port |
| 60 | + params.ip_address = args.ip_address |
| 61 | + board = BoardShim(args.board_id, params) |
| 62 | + |
| 63 | + # LSL initialization |
| 64 | + channel_names = args.channel_names.split(',') |
| 65 | + n_channels = len(channel_names) |
| 66 | + srate = board.get_sampling_rate(args.board_id) |
| 67 | + info = StreamInfo(args.name, args.data_type, n_channels, srate, 'double64', args.uid) |
| 68 | + outlet = StreamOutlet(info) |
| 69 | + fw_delay = 0 |
| 70 | + |
| 71 | + # prepare session |
| 72 | + board.prepare_session() |
| 73 | + |
| 74 | + # send commands to the board for every channel. Cyton has 8 Channels. Here, we turn off every channel except for 1 and 8. |
| 75 | + # This is here for testing purposes. |
| 76 | + #board.config_board("x1000110X") #Lower the gain to 1x on channel 1 |
| 77 | + #board.config_board("x1061000X") |
| 78 | + #board.config_board("x2161000X") |
| 79 | + #board.config_board("x3161000X") |
| 80 | + #board.config_board("x4161000X") |
| 81 | + #board.config_board("x5161000X") |
| 82 | + #board.config_board("x6161000X") |
| 83 | + #board.config_board("x7161000X") |
| 84 | + #board.config_board("x8060110X") |
| 85 | + |
| 86 | + # start stream |
| 87 | + board.start_stream(45000, args.streamer_params) |
| 88 | + time.sleep(1) |
| 89 | + start_time = local_clock() |
| 90 | + sent_samples = 0 |
| 91 | + queue = Queue(maxsize = 5*srate) |
| 92 | + chans = channel_select(board, args.board_id, args.data_type) |
| 93 | + |
| 94 | + # Vars for filters |
| 95 | + applyBandStop = True |
| 96 | + applyBandPass = True |
| 97 | + bandStopFrequency = 60.0 |
| 98 | + bp_lowerBound = 5.0 |
| 99 | + bp_upperBound = 50.0 |
| 100 | + bp_centerFreq = (bp_upperBound + bp_lowerBound) / 2.0; |
| 101 | + bp_bandWidth = bp_upperBound - bp_lowerBound |
| 102 | + |
| 103 | + |
| 104 | + # read data with brainflow and send it via LSL |
| 105 | + print("Now sending data...") |
| 106 | + while True: |
| 107 | + data = board.get_board_data()[chans] |
| 108 | + |
| 109 | + # It's best to apply filters on the receiving end, but this is here just for testing purposes. |
| 110 | + """ |
| 111 | + for chan in range(len(chans)): |
| 112 | + if applyBandStop: |
| 113 | + DataFilter.perform_bandstop(data[chan], |
| 114 | + BoardShim.get_sampling_rate(args.board_id), |
| 115 | + bandStopFrequency, |
| 116 | + 4.0, |
| 117 | + 2, |
| 118 | + FilterTypes.BUTTERWORTH.value, |
| 119 | + 0); |
| 120 | + if applyBandPass: |
| 121 | + DataFilter.perform_bandpass( |
| 122 | + data[chan], |
| 123 | + BoardShim.get_sampling_rate(args.board_id), |
| 124 | + bp_centerFreq, |
| 125 | + bp_bandWidth, |
| 126 | + 2, |
| 127 | + FilterTypes.BUTTERWORTH.value, |
| 128 | + 0); |
| 129 | + """ |
| 130 | + |
| 131 | + for i in range(len(data[0])): |
| 132 | + queue.put(data[:,i].tolist()) |
| 133 | + elapsed_time = local_clock() - start_time |
| 134 | + required_samples = int(srate * elapsed_time) - sent_samples |
| 135 | + if required_samples > 0 and queue.qsize() >= required_samples: |
| 136 | + mychunk = [] |
| 137 | + |
| 138 | + for i in range(required_samples): |
| 139 | + mychunk.append(queue.get()) |
| 140 | + stamp = local_clock() - fw_delay |
| 141 | + outlet.push_chunk(mychunk, stamp) |
| 142 | + sent_samples += required_samples |
| 143 | + time.sleep(1) |
| 144 | + |
| 145 | + |
| 146 | +if __name__ == "__main__": |
| 147 | + main() |
0 commit comments