def __init__(self, subscribers, publishers, port, baud, filter_data):
super(OpenBCISource, self).__init__(subscribers, publishers)
self.connector = OpenBCIConnector(port=port, baud=baud,
filter_data=filter_data)
_LOGGER.debug("Subscribers: %s" % self.subscribers)
_LOGGER.debug("Publishers: %s" % self.publishers)
class OpenBCISource(SourceInterface):
def __init__(self, subscribers, publishers,
port='/dev/tty.usbserial-DN0095VT',
baud=115200,
filter_data=True):
super(OpenBCISource, self).__init__(subscribers, publishers)
self.connector = OpenBCIConnector(port=port, baud=baud,
filter_data=filter_data)
_LOGGER.debug("Subscribers: %s" % self.subscribers)
_LOGGER.debug("Publishers: %s" % self.publishers)
def start(self):
# Callback functions to handle the sample for that metric.
# Each metric has a specific number of channels.
callback_functions = {}
for publisher in self.publishers:
metrics_to_num_channels = publisher.get_metrics_to_num_channels()
logging.debug("Metrics to channels mapping: %s" % metrics_to_num_channels)
for (metric_name, num_channels) in metrics_to_num_channels.items():
if metric_name not in callback_functions:
callback_functions[metric_name] = self.callback_factory(
metric_name, num_channels)
self.connector.start(callback_functions)
def callback_factory(self, metric_name, num_channels):
"""
Callback function generator for OpenBCI metrics
:return: callback function
"""
def callback(sample):
"""
Handle OpenBCI samples for that metric
:param sample: the sample to handle
"""
message = {}
for i in range(num_channels):
channel_value = "%.4f" % (
sample.channel_data[i] * 10 ** 9) # Nano volts
message["channel_%s" % i] = channel_value
message['timestamp'] = int(time.time() * 1000000) # micro seconds
for publisher in self.publishers:
logging.debug("Publishing on metric %s: %s" % (metric_name, message))
publisher.publish(metric_name, message)
return callback
class OpenBCISource(ModuleInterface):
def __init__(self, subscribers, publishers, port, baud, filter_data):
super(OpenBCISource, self).__init__(subscribers, publishers)
self.connector = OpenBCIConnector(port=port, baud=baud,
filter_data=filter_data)
_LOGGER.debug("Subscribers: %s" % self.subscribers)
_LOGGER.debug("Publishers: %s" % self.publishers)
def start(self):
# Callback functions to handle the sample for that metric.
# Each metric has a specific number of channels.
callback_functions = {}
for publisher in self.publishers:
metrics_to_num_channels = publisher.metrics_to_num_channels()
logging.debug("Metrics to channels mapping: %s" % metrics_to_num_channels)
for (metric_name, num_channels) in metrics_to_num_channels.items():
if metric_name not in callback_functions:
callback_functions[metric_name] = self.callback_factory(
metric_name, num_channels)
self.connector.start(callback_functions)
def callback_factory(self, metric_name, num_channels):
"""
Callback function generator for OpenBCI metrics
:return: callback function
"""
def callback(sample):
"""
Handle OpenBCI samples for that metric
:param sample: the sample to handle
"""
message = {}
for i in range(num_channels):
channel_value = "%.4f" % (
sample.channel_data[i] * 10 ** 9) # Nano volts
message["channel_%s" % i] = float(channel_value)
message['timestamp'] = sample.timestamp
for publisher in self.publishers:
logging.debug("Publishing on metric %s: %s" % (metric_name, message))
publisher.publish(metric_name, message)
return callback
def __init__(self, subscribers, publishers,
port='/dev/tty.usbserial-DN0095VT',
baud=115200,
filter_data=True):
super(OpenBCISource, self).__init__(subscribers, publishers)
self.connector = OpenBCIConnector(port=port, baud=baud,
filter_data=filter_data)
_LOGGER.debug("Subscribers: %s" % self.subscribers)
_LOGGER.debug("Publishers: %s" % self.publishers)
def test_OpenBCIConnector(self):
board = OpenBCIConnector()
callbacks = {self.metric_name: self.validate_start_method}
board.start(callbacks)
def test_OpenBCIConnector(self):
board = OpenBCIConnector()
callbacks = {self.metric_name: self.validate_start_method}
board.start(callbacks)
