def button_3_callback(self):
if globls.Shocked:
globls.update_shock_flag(0)
globls.ManualShock = 0
globls.shock_off()
globls.custom_button[2].configure(text="Shock Off")
globls.dump_log('Shock Stopped')
data_str = '6 200 /'
data_recv_ts = time.time()
data = data_str + """{}""".format('q' *
(1024 - len(data_str.encode())))
DataFileObj.save_raw_eye_taskdata(data_recv_ts,
0,
0,
0,
0,
task_data=data)
else:
globls.update_shock_flag(1)
globls.ManualShock = 1
globls.shock_on()
globls.custom_button[2].configure(text="Shock On")
globls.dump_log('Shock Started')
data_str = '6 201 /'
data_recv_ts = time.time()
data = data_str + """{}""".format('q' *
(1024 - len(data_str.encode())))
DataFileObj.save_raw_eye_taskdata(data_recv_ts,
0,
0,
0,
0,
task_data=data)
return
def receiver(self):
'''
Receiver loop to check the incoming packets
:return:
'''
# globls.dump_log("Running Matlab thread")
globls.dump_log('Listening to 3rd party messages')
while globls.is_program_running:
try:
bin_data, addr = globls.arbitrator_sock.recvfrom(
globls.constant['data_size_length'])
data = bin_data.decode()
data_recv_ts = time.time()
# process the variables
if len(data):
# Check if there is any delimiter in the message '/' if yes then split
cmd_lists = data.split(
globls.constant['message_delimiter'])
# Data before / is valid data read and remove the space
cmd_list = cmd_lists[0].split()
if int(cmd_list[0]) == Constant.COMMAND_WORD_CONNECTION:
if int(cmd_list[1]) == Constant.CONNECTION_START:
# send the UDP packet to initiate the connection
data = '{} {}/{} {}/{} {}/{} {}/{} {}/'.format(
Constant.COMMAND_WORD_CONNECTION,
Constant.CONNECTION_ACK, const.SCREEN_HEIGHT,
globls.arbitrator_display['screen_height_mm'],
const.SCREEN_DISTANCE, globls.
arbitrator_display['screen_distance_mm'],
const.SCREEN_WIDTH,
globls.arbitrator_display['screen_width_mm'],
const.SCREEN_IOD,
globls.arbitrator_display['iod_mm'])
util.send_data(globls.arbitrator_sock,
globls.arbitrator_sock_addr, data,
'Connection Ack Message')
globls.update_load_subject_task_config_flag(True)
elif int(cmd_list[1]) == Constant.CONNECTION_ACK:
globls.connection_state = Constant.CONNECTION_ACK
# Update this flag to ensure that the user knows what is the next step after connection is up
globls.update_load_subject_task_config_flag(True)
# util.send_screen_parameters()
# print cmd_list
elif int(cmd_list[0]) == Constant.DISPLAY_COMMAND_WORD:
config = [int(cmd_list[1]), cmd_list[2]]
# print(config)
globls.update_displayable_config(config)
# Process the exit call
elif (int(cmd_list[0]) == Constant.COMMAND_WORD_CONTROL
and int(cmd_list[1]) == Constant.CONTROL_EXIT):
print('breaking code (arbitrator_server: line 99)')
break
elif int(cmd_list[0]) == Constant.EYE_IN_OUT_REQUEST:
eye_in_out = copy.copy(globls.eye_in_or_out)
# if (eye_in_out[0] != globls.last_sent_eye_in_out[0]
# or eye_in_out[1] != globls.last_sent_eye_in_out[1]):
# '{LEye_Status} {in/out}
# {LEye_X_gain} {LEye_X_offset}
# {LEye_Y_gain} {LEye_Y_offset}/...
# {REye_Status} {in/out}
# {REye_X_gain} {REye_X_offset}
# {REye_Y_gain} {REye_Y_offset}/'
data = '{} {} {} {} {} {}/{} {} {} {} {} {}/'.format(
Constant.LEFT_EYE_STATUS,
globls.eye_in_or_out[0],
globls.offset_gain_config[
ConfigLabels.LEFT_EYE_GAIN_X],
globls.offset_gain_config[
ConfigLabels.LEFT_EYE_OFFSET_X],
globls.offset_gain_config[
ConfigLabels.LEFT_EYE_GAIN_Y],
globls.offset_gain_config[
ConfigLabels.LEFT_EYE_OFFSET_Y],
Constant.RIGHT_EYE_STATUS,
globls.eye_in_or_out[1],
globls.offset_gain_config[
ConfigLabels.RIGHT_EYE_GAIN_X],
globls.offset_gain_config[
ConfigLabels.RIGHT_EYE_OFFSET_X],
globls.offset_gain_config[
ConfigLabels.RIGHT_EYE_GAIN_Y],
globls.offset_gain_config[
ConfigLabels.RIGHT_EYE_OFFSET_Y],
)
# if (eye_in_out[0] != globls.last_sent_eye_in_out[0]
# or eye_in_out[1] != globls.last_sent_eye_in_out[1]):
# data = '{} {}/{} {}/'.format(Constant.LEFT_EYE_STATUS, globls.eye_in_or_out[0],
# Constant.RIGHT_EYE_STATUS, globls.eye_in_or_out[1])
util.send_data(globls.arbitrator_sock_eye,
globls.arbitrator_sock_addr_eye, data)
# print "EYE in or out ", data
globls.update_last_sent_eye_in_out(eye_in_out)
elif int(cmd_list[0]
) == Constant.LATCHED_EYE_IN_OUT_REQUEST:
globls.latched_eye = 1
globls.caught = 0 # if just starting latched eye request, couldn't have caught eye out!
globls.dump_log('Initiating latched eye request')
elif int(cmd_list[0]) == Constant.END_LATCHED_EYE:
globls.latched_eye = 0
globls.caught = 0 # matlab finally caught your message about version violation
globls.dump_log('Ending latched eye request')
elif int(cmd_list[0]) == Constant.VERGENCE_REQUEST:
points = list(map(float, cmd_list[1:6]))
value = self.eye_check.vergence_error(points)
# if value != globls.last_vergence_error_sent:
data = '{} {}/'.format(Constant.VERGENCE_STATUS, value)
util.send_data(globls.arbitrator_sock_eye,
globls.arbitrator_sock_addr_eye, data)
# print "Vergence error ", data
globls.update_last_sent_vergence_error(value)
elif int(cmd_list[0]) == Constant.SCREEN_WINDOW_WIDTH:
globls.update_arbitrator_server_screen_resolution(
width=int(cmd_list[1]))
globls.dump_log(
'Received the screen width and scale factor to display on GUI: {}'
.format(
globls.arbitrator_display['x_scale_factor']))
elif int(cmd_list[0]) == Constant.SCREEN_WINDOW_HEIGHT:
globls.update_arbitrator_server_screen_resolution(
height=int(cmd_list[1]))
globls.dump_log(
'Received the screen Height and scale factor to display on GUI: {}'
.format(
globls.arbitrator_display['y_scale_factor']))
elif int(cmd_list[0]) == Constant.CENTRE_OF_WINDOWS:
# print cmd_list
# 100 number list of X Y Z diameter
# Maximum 30 points
no_of_points = int(cmd_list[1])
config = []
lst = cmd_list[2:]
# pair the data and give it to UI for updation
for i in range(0, len(lst), 6):
config.append([
float(lst[i]),
float(lst[i + 1]),
float(lst[i + 2]),
float(lst[i + 3]), lst[i + 4], lst[i + 5]
])
if len(config
) == Constant.MAXIMUM_CENTRE_OF_WINDOW:
break
globls.dump_log('Points to draw {}'.format(cmd_list))
globls.update_centre_point(no_of_points, config)
elif int(cmd_list[0]) == Constant.WINDOW_ON:
# print cmd_list
globls.update_centre_window_on_off(True)
globls.dump_log('Window On')
elif int(cmd_list[0]) == Constant.WINDOW_OFF:
# print cmd_list
globls.update_centre_window_on_off(False)
globls.dump_log('Window Off')
# Reset this to -1, -1 so that next time this is queried from
# Arbitrator server new values are sent
globls.update_last_sent_eye_in_out([-1, -1])
elif int(cmd_list[0]) == Constant.REWARD_ON:
# print cmd_list
self.DIO_obj.write_digital_bit(1)
#self.DIO_obj.reward_on()
globls.dump_log('Reward On')
elif int(cmd_list[0]) == Constant.REWARD_OFF:
# print cmd_list
self.DIO_obj.write_digital_bit(0)
#self.DIO_obj.reward_off()
globls.dump_log('Reward Off')
elif int(cmd_list[0]) == Constant.VERGENCE_ON:
# print cmd_list
globls.udpdate_vergence_display(True)
globls.dump_log('Vergence On')
elif int(cmd_list[0]) == Constant.VERGENCE_OFF:
# print cmd_list
globls.udpdate_vergence_display(False)
globls.dump_log('Vergence Off')
# Reset this to -1 so that next time this is queried from
# Arbitrator server new values are sent
globls.update_last_sent_vergence_error(-1)
elif int(cmd_list[0]) == Constant.DRAW_EVENTS:
tempnum = len(cmd_lists) - 1
globls.update_drawing_object_on(1)
globls.update_drawing_object_number(tempnum)
for i in range(0, tempnum):
templist = cmd_lists[i].split()
if i == 0:
tempx = float(templist[2])
tempy = float(templist[3])
if tempx == -1 and tempy == -1:
if not globls.Shocked:
globls.shock_on()
data_str = '6 201 /'
temp_data = data_str + """{}""".format(
'q' *
(1024 - len(data_str.encode())))
DataFileObj.save_raw_eye_taskdata(
data_recv_ts,
0,
0,
0,
0,
task_data=temp_data)
globls.update_shock_flag(1)
else:
# globls.prt.setData(0)
if not globls.ManualShock:
if globls.Shocked:
globls.shock_off()
data_str = '6 200 /'
temp_data = data_str + """{}""".format(
'q' * (1024 -
len(data_str.encode())))
globls.update_shock_flag(0)
DataFileObj.save_raw_eye_taskdata(
data_recv_ts,
0,
0,
0,
0,
task_data=temp_data)
else:
tempx = float(templist[1])
tempy = float(templist[2])
globls.update_drawing_object_pos(i, 0, tempx)
globls.update_drawing_object_pos(i, 1, tempy)
elif int(cmd_list[0]) == Constant.EVENTS:
globls.dump_log('event {}'.format(cmd_list))
if not globls.DUMP_DATA_CONTINUALLY:
if int(cmd_list[1]) == Constant.EVENT_START:
# Start the data save
DataFileObj.update_data_collect_flag(True)
# Dump auto offset data
globls.auto_offset_dump_data = 1
# Make sure primer and flag is still off
globls.auto_offset_update_primer = 0
globls.auto_offset_update_flag = 0
elif int(cmd_list[1]) == Constant.EVENT_STOP:
# This needs to be saved here as we dump the saved data to file
# And data collection falg will be reset
DataFileObj.save_raw_eye_taskdata(
data_recv_ts,
globls.last_raw_eye_data[0],
globls.last_raw_eye_data[1],
globls.last_raw_eye_data[2],
globls.last_raw_eye_data[3],
task_data=data)
# Stop saving the data and dump tp file
# Flag to stop saving the data is set inside this function
DataFileObj.dump_data_to_file()
globls.update_last_sent_vergence_error(-1)
globls.update_last_sent_eye_in_out([-1, -1])
# Dump auto offset data
globls.auto_offset_dump_data = 1
# Set update flag so offsets can now be changed
globls.auto_offset_update_flag = 1
# Check for fixation acquired and hold time acquired to start/stop auto eye offset
elif int(
cmd_list[1]) == Constant.FIXATION_ACQUIRED:
# Set "stop data collection" flag to false
globls.auto_offset_stop_collect = 0
# Set "start data collection" flag to true
globls.auto_offset_start_collect = 1
# Make sure primer is still off
globls.auto_offset_update_primer = 0
elif int(cmd_list[1]
) == Constant.FIXATION_HOLD_COMPLETE:
# Set "start data collection" flag to false
globls.auto_offset_start_collect = 0
# Set "stop data collection" flag to true
globls.auto_offset_stop_collect = 1
elif int(cmd_list[1]
) == Constant.SACCADE_TARGET_ACQUIRED:
# Set primer flag to update offsets after this trial is over
globls.auto_offset_update_primer = 1
# Save the data from 3rd party server
DataFileObj.save_raw_eye_taskdata(
data_recv_ts,
globls.last_raw_eye_data[0],
globls.last_raw_eye_data[1],
globls.last_raw_eye_data[2],
globls.last_raw_eye_data[3],
task_data=data)
# # Save the data from 3rd party server
# if globls.DUMP_DATA_CONTINUALLY:
# DataFileObj.dump_data_continually(data_recv_ts, data)
# except socket.timeout:
# pass
except Exception as e:
log.exception(e)
globls.arbitrator_sock.close()
print('globls.arbitrator_sock was closed (arbitrator_serve: line 290)')