()
| 6 | |
| 7 | |
| 8 | def main(): |
| 9 | BoardShim.enable_dev_board_logger() |
| 10 | |
| 11 | parser = argparse.ArgumentParser() |
| 12 | parser.add_argument('--id1', type=str, help='id for first emotibit', required=True) |
| 13 | parser.add_argument('--id2', type=str, help='id for second emotibit', required=True) |
| 14 | args = parser.parse_args() |
| 15 | |
| 16 | params1 = BrainFlowInputParams() |
| 17 | params1.serial_number = args.id1 |
| 18 | params2 = BrainFlowInputParams() |
| 19 | params2.serial_number = args.id2 |
| 20 | board_id = BoardIds.EMOTIBIT_BOARD.value |
| 21 | |
| 22 | presets = BoardShim.get_board_presets(board_id) |
| 23 | print (presets) |
| 24 | |
| 25 | # Init both boards |
| 26 | board1 = BoardShim(board_id, params1) |
| 27 | board2 = BoardShim(board_id, params2) |
| 28 | board1.prepare_session() |
| 29 | board2.prepare_session() |
| 30 | |
| 31 | # Start streaming for both |
| 32 | board1.start_stream() |
| 33 | board2.start_stream() |
| 34 | time.sleep(10) |
| 35 | |
| 36 | # Get data from both |
| 37 | data_default1 = board1.get_board_data(preset=BrainFlowPresets.DEFAULT_PRESET) |
| 38 | data_aux1 = board1.get_board_data(preset=BrainFlowPresets.AUXILIARY_PRESET) |
| 39 | data_anc1 = board1.get_board_data(preset=BrainFlowPresets.ANCILLARY_PRESET) |
| 40 | data_default2 = board2.get_board_data(preset=BrainFlowPresets.DEFAULT_PRESET) |
| 41 | data_aux2 = board2.get_board_data(preset=BrainFlowPresets.AUXILIARY_PRESET) |
| 42 | data_anc2 = board1.get_board_data(preset=BrainFlowPresets.ANCILLARY_PRESET) |
| 43 | |
| 44 | # Stop streaming for both |
| 45 | board1.stop_stream() |
| 46 | board2.stop_stream() |
| 47 | |
| 48 | # Release both boards |
| 49 | board1.release_session() |
| 50 | board2.release_session() |
| 51 | |
| 52 | # Write data from both |
| 53 | DataFilter.write_file(data_default1, 'default1.csv', 'w') |
| 54 | DataFilter.write_file(data_aux1, 'aux1.csv', 'w') |
| 55 | DataFilter.write_file(data_anc1, 'anc1.csv', 'w') |
| 56 | DataFilter.write_file(data_default2, 'default2.csv', 'w') |
| 57 | DataFilter.write_file(data_aux2, 'aux2.csv', 'w') |
| 58 | DataFilter.write_file(data_anc2, 'anc2.csv', 'w') |
| 59 | |
| 60 | |
| 61 | if __name__ == "__main__": |
no test coverage detected