def __load_bpod_ports(self):
# present error if no serial port is selected
if not self._serial_port.value:
self.warning("Please select a serial port before proceeding.",
"No serial port selected")
return
if "not connected" in self._serial_port.text:
self.warning(
"Please connect the device to the computer before proceeding.",
"Device not connected")
return
try:
bpod = Bpod(self._serial_port.value)
hw = bpod.hardware
# load the ports to the GUI ###############################
self._active_bnc.value = [
('BNC{0}'.format(j + 1), True)
for j, i in enumerate(hw.bnc_inputports_indexes)
]
self._active_wired.value = [
('Wire{0}'.format(j + 1), True)
for j, i in enumerate(hw.wired_inputports_indexes)
]
if len(self._active_behavior.value) == 0:
self._active_behavior.value = [
('Port{0}'.format(j + 1), True)
for j, i in enumerate(hw.behavior_inputports_indexes)
]
#############################################################
self._events.value = [["{0} ({1})".format(
x, i)] for i, x in enumerate(hw.channels.event_names)]
self._inputchannels.value = [
[x] for x in hw.channels.input_channel_names
]
self._outputchannels.value = [
[x] for x in hw.channels.output_channel_names
]
bpod.close()
except Exception as e:
self.critical(str(e), 'Error loading ports')
# !/usr/bin/python3
# -*- coding: utf-8 -*-
"""
Example adapted from Josh Sanders' original version on Sanworks Bpod repository
"""
from pybpodapi.bpod import Bpod
"""
Run this protocol now
"""
bpod = Bpod()
for m in bpod.modules:
print(m)
bpod.close()
output_actions=[('Valve', 2)])
sma.add_state(state_name='wait2',
state_timer=waittime,
state_change_conditions={EventName.Tup: 'Open Middle'},
output_actions=[])
sma.add_state(state_name='Open Middle',
state_timer=ValveOpenTime_M,
state_change_conditions={EventName.Tup: 'wait3'},
output_actions=[('Valve', 3)])
sma.add_state(state_name='wait3',
state_timer=waittime,
state_change_conditions={EventName.Tup: 'Close Valves'},
output_actions=[])
if i == 0:
sma.add_state(state_name='Close Valves',
state_timer=ValveCloseTime,
state_change_conditions={EventName.Tup: 'exit'},
output_actions=[(OutputChannel.PWM3, 255)]) #
else:
sma.add_state(state_name='Close Valves',
state_timer=ValveCloseTime,
state_change_conditions={EventName.Tup: 'exit'},
output_actions=[])
my_bpod.send_state_machine(
sma) # Send state machine description to Bpod device
my_bpod.run_state_machine(sma) # Run state machine
my_bpod.close()
def __init__(self, parent_win=None):
self.setup = parent_win
self.started_correctly = False
title = 'Emulator for setup: ' + self.setup.name
BaseWidget.__init__(self, title, parent_win=parent_win)
self.CHECKED_ICON = conf.EMULATOR_CHECKED_ICON
self.UNCHECKED_ICON = conf.EMULATOR_UNCHECKED_ICON
self._currentSetup = ControlLabel(self.setup.name)
self._selectedBoard = ControlLabel(self.setup.board.name)
self._selectedProtocol = ControlLabel(self.setup.task.name)
self._run_task_btn = ControlButton('Run protocol',
default=self.__run_protocol_btn_evt,
checkable=True,
helptext="When a task is running, you can skip all remaining trials by pressing this button. <br/> <b>NOTE:</b> This means that you will need to break the cycle in your task code when the run_state_machine method returns False.")
self._kill_task_btn = ControlButton('Kill',
default=self.__kill_btn_evt,
style="background-color:rgb(255,0,0);font-weight:bold;",
helptext="<b>NOTE:</b>This will exit the task process abruptly. The code you might have after the trial loop won't execute.")
self._kill_task_btn.enabled = False
self._stop_trial_btn = ControlButton('Skip trial',
default=self.__stop_trial_btn_evt,
enabled=False)
self._pause_btn = ControlButton('Pause',
default=self.__pause_btn_evt,
checkable=True,
enabled=False)
try:
bpod = Bpod(self.setup.board.serial_port)
except SerialException:
self.critical('No Bpod device connected, cannot continue until one is connected.', 'Bpod not connected')
return
except Exception:
# NOTE: try again in case of the first connection attempt where we always get the utf-8 exception
bpod = Bpod(self.setup.board.serial_port)
number_ports = bpod.hardware.inputs.count('P')
number_bnc = bpod.hardware.outputs.count('B')
number_wire_in = bpod.hardware.inputs.count('W')
number_wire_out = bpod.hardware.outputs.count('W')
self._valve_buttons = []
self._valve_label = ControlLabel("Valve")
self._led_buttons = []
self._led_label = ControlLabel("LED")
self._poke_buttons = []
self._poke_label = ControlLabel("Poke")
self._time_control = ControlText() # HH
for n in range(1, number_ports + 1):
btn_valve = ControlButton(str(n), icon=self.UNCHECKED_ICON, checkable=True)
btn_led = ControlButton(str(n), icon=self.UNCHECKED_ICON, checkable=True)
btn_poke = ControlButton(str(n), icon=self.UNCHECKED_ICON, checkable=True)
btn_valve.value = make_lambda_func(self.__button_on_click_evt, btn=btn_valve)
btn_led.value = make_lambda_func(self.__button_on_click_evt, btn=btn_led)
btn_poke.value = make_lambda_func(self.__button_on_click_evt, btn=btn_poke)
setattr(self, f'_btn_Valve{n}', btn_valve)
setattr(self, f'_btn_PWM{n}', btn_led)
setattr(self, f'_btn_Port{n}', btn_poke)
self._valve_buttons.append(btn_valve)
self._led_buttons.append(btn_led)
self._poke_buttons.append(btn_poke)
self._bnc_in_buttons = []
self._bnc_in_label = ControlLabel("BNC In")
self._bnc_out_buttons = []
self._bnc_out_label = ControlLabel("BNC Out")
for n in range(1, number_bnc + 1):
btn_bnc_in = ControlButton(str(n), icon=self.UNCHECKED_ICON, checkable=True)
btn_bnc_out = ControlButton(str(n), icon=self.UNCHECKED_ICON, checkable=True)
btn_bnc_in.value = make_lambda_func(self.__button_on_click_evt, btn=btn_bnc_in)
btn_bnc_out.value = make_lambda_func(self.__button_on_click_evt, btn=btn_bnc_out)
setattr(self, f'_btn_BNC_in{n}', btn_bnc_in)
setattr(self, f'_btn_BNC_out{n}', btn_bnc_out)
self._bnc_in_buttons.append(btn_bnc_in)
self._bnc_out_buttons.append(btn_bnc_out)
self._wire_in_buttons = []
self._wire_in_label = ControlLabel("Wire In")
self._wire_out_buttons = []
self._wire_out_label = ControlLabel("Wire Out")
for n in range(1, number_wire_in + 1):
btn_wire_in = ControlButton(str(n), icon=self.UNCHECKED_ICON, checkable=True)
btn_wire_in.value = make_lambda_func(self.__button_on_click_evt, btn=btn_wire_in)
setattr(self, f'_btn_Wire_in{n}', btn_wire_in)
self._wire_in_buttons.append(btn_wire_in)
for n in range(1, number_wire_out + 1):
btn_wire_out = ControlButton(str(n), icon=self.UNCHECKED_ICON, checkable=True)
btn_wire_out.value = make_lambda_func(self.__button_on_click_evt, btn=btn_wire_out)
setattr(self, f'_btn_Wire_out{n}', btn_wire_out)
self._wire_out_buttons.append(btn_wire_out)
self._modules_indexes_loaded = []
for idx, mod in enumerate(bpod.modules):
n = mod.serial_port
self._modules_indexes_loaded.append(n)
module_label = ControlLabel(f'{mod.name}')
control_text_bytes_msg = ControlText()
btn_send_msg_module = ControlButton(f'Send bytes')
btn_send_msg_module.value = make_lambda_func(self.__send_msg_btn_evt,
btn=btn_send_msg_module,
control_text=control_text_bytes_msg)
setattr(self, f'_module_label{n}', module_label)
setattr(self, f'_control_text_bytes_msg{n}', control_text_bytes_msg)
setattr(self, f'_btn_send_msg_module{n}', btn_send_msg_module)
if bpod:
bpod.close()
self.formset = [
([('Current setup:', '_currentSetup'),
('Selected board:', '_selectedBoard'),
('Selected protocol:', '_selectedProtocol')],
'',
['_run_task_btn', '_kill_task_btn', '_stop_trial_btn', '_pause_btn']),
'',
'Behaviour Ports',
('_valve_label', tuple([f'_btn_Valve{n.label}' for n in self._valve_buttons])),
('_led_label', tuple([f'_btn_PWM{n.label}' for n in self._led_buttons])),
('_poke_label', tuple([f'_btn_Port{n.label}' for n in self._poke_buttons])),
'',
('Time Control (sec)', '_time_control'),
'',
'BNC',
('_bnc_in_label',
tuple([f'_btn_BNC_in{n.label}' for n in self._bnc_in_buttons]),
'_bnc_out_label',
tuple([f'_btn_BNC_out{n.label}' for n in self._bnc_out_buttons])
),
'Wire' if number_wire_in != 0 else '',
('_wire_in_label' if number_wire_in != 0 else '',
tuple([f'_btn_Wire_in{n.label}' for n in self._wire_in_buttons]),
'_wire_out_label' if number_wire_out != 0 else '',
tuple([f'_btn_Wire_out{n.label}' for n in self._wire_out_buttons])
),
'',
'Send bytes to modules' if self._modules_indexes_loaded else '',
[(f'_module_label{n}', f'_control_text_bytes_msg{n}', f'_btn_send_msg_module{n}') for n in self._modules_indexes_loaded]
]
self.set_margin(10)
self.started_correctly = True