def make_ephysCW_pc():
"""make_ephysCW_pc Makes positions, contrasts and block lengths for ephysCW
Generates ~2000 trias
:return: pc
:rtype: [type]
"""
contrasts = [1.0, 0.25, 0.125, 0.0625, 0.0]
len_block = [90]
pos = [-35] * int(len_block[0] / 2) + [35] * int(len_block[0] / 2)
cont = np.sort(contrasts * 10)[::-1][:-5].tolist()
prob = [0.5] * len_block[0]
pc = np.array([pos, cont + cont, prob]).T
np.random.shuffle(pc) # only shuffles on the first dimension
prob_left = 0.8 if blocks.draw_position([-35, 35], 0.5) < 0 else 0.2
while len(pc) < 2001:
len_block.append(blocks.get_block_len(60, min_=20, max_=100))
for x in range(len_block[-1]):
p = blocks.draw_position([-35, 35], prob_left)
c = misc.draw_contrast(contrasts, prob_type='uniform')
pc = np.append(pc, np.array([[p, c, prob_left]]), axis=0)
# do this in PC space
prob_left = np.round(np.abs(1 - prob_left), 1)
return pc, len_block
def make_pc():
contrasts = [1., 0.25, 0.125, 0.0625, 0.0]
len_block = [90]
pos = [-35] * int(len_block[0] / 2) + [35] * int(len_block[0] / 2)
cont = np.sort(contrasts * 10)[::-1][:-5].tolist()
pc = np.array([pos, cont + cont]).T
np.random.shuffle(pc) # only shuffles on the first dimension
prob_left = 0.8 if blocks.draw_position([-35, 35], 0.5) < 0 else 0.2
while len(pc) < 2001:
len_block.append(blocks.get_block_len(60, min_=20, max_=100))
for x in range(len_block[-1]):
p = blocks.draw_position([-35, 35], prob_left)
c = misc.draw_contrast(contrasts, prob_type='uniform')
pc = np.append(pc, np.array([[p, c]]), axis=0)
# do this in PC space
prob_left = np.round(np.abs(1 - prob_left), 1)
return pc, len_block
def next_trial(self):
# First trial exception
if self.trial_num == 0:
self.trial_num += 1
self.block_num += 1
self.block_trial_num += 1
# Send next trial info to Bonsai
bonsai.send_current_trial_info(self)
return
self.data_file = str(self.data_file)
# Increment trial number
self.trial_num += 1
# Update quiescent period
self.quiescent_period = self.quiescent_period_base + misc.texp()
# Update stimulus phase
self.stim_phase = random.uniform(0, 2 * math.pi)
# Update block
self = blocks.update_block_params(self)
# Update stim probability left + buffer
self.stim_probability_left = blocks.update_probability_left(self)
self.stim_probability_left_buffer.append(self.stim_probability_left)
# Update position + buffer
self.position = blocks.draw_position(self.position_set,
self.stim_probability_left)
self.position_buffer.append(self.position)
# Update contrast + buffer
self.contrast = misc.draw_contrast(
self.contrast_set, prob_type=self.contrast_set_probability_type)
self.contrast_buffer.append(self.contrast)
# Update signed_contrast + buffer (AFTER position update)
self.signed_contrast = self.contrast * np.sign(self.position)
self.signed_contrast_buffer.append(self.signed_contrast)
# Update state machine events
self.event_error = self.threshold_events_dict[self.position]
self.event_reward = self.threshold_events_dict[-self.position]
# Reset outcome variables for next trial
self.trial_correct = None
# Open the data file to append the next trial
self.data_file = open(self.data_file_path, "a")
# Send next trial info to Bonsai
bonsai.send_current_trial_info(self)
def __init__(self, sph):
# Constants from settings
self.session_start_delay_sec = sph.SESSION_START_DELAY_SEC
self.init_datetime = parser.parse(
sph.PYBPOD_SESSION) + datetime.timedelta(
0, self.session_start_delay_sec)
self.task_protocol = sph.PYBPOD_PROTOCOL
self.data_file_path = sph.DATA_FILE_PATH
self.data_file = open(self.data_file_path, "a")
self.position_set = sph.STIM_POSITIONS
self.contrast_set = sph.CONTRAST_SET
self.contrast_set_probability_type = sph.CONTRAST_SET_PROBABILITY_TYPE
self.repeat_on_error = sph.REPEAT_ON_ERROR
self.threshold_events_dict = sph.ROTARY_ENCODER.THRESHOLD_EVENTS
self.quiescent_period_base = sph.QUIESCENT_PERIOD
self.quiescent_period = self.quiescent_period_base + misc.texp()
self.response_window = sph.RESPONSE_WINDOW
self.interactive_delay = sph.INTERACTIVE_DELAY
self.iti_error = sph.ITI_ERROR
self.iti_correct_target = sph.ITI_CORRECT
self.osc_client = sph.OSC_CLIENT
self.stim_freq = sph.STIM_FREQ
self.stim_angle = sph.STIM_ANGLE
self.stim_gain = sph.STIM_GAIN
self.stim_sigma = sph.STIM_SIGMA
self.out_tone = sph.OUT_TONE
self.out_noise = sph.OUT_NOISE
self.out_stop_sound = sph.OUT_STOP_SOUND
self.poop_count = sph.POOP_COUNT
self.save_ambient_data = sph.RECORD_AMBIENT_SENSOR_DATA
self.as_data = {
"Temperature_C": -1,
"AirPressure_mb": -1,
"RelativeHumidity": -1,
}
# Reward amount
self.reward_amount = sph.REWARD_AMOUNT
self.reward_valve_time = sph.REWARD_VALVE_TIME
self.iti_correct = self.iti_correct_target - self.reward_valve_time
# Initialize parameters that may change every trial
self.trial_num = 0
self.stim_phase = 0.0
self.block_num = 0
self.block_trial_num = 0
self.block_len_factor = sph.BLOCK_LEN_FACTOR
self.block_len_min = sph.BLOCK_LEN_MIN
self.block_len_max = sph.BLOCK_LEN_MAX
self.block_probability_set = sph.BLOCK_PROBABILITY_SET
self.block_init_5050 = sph.BLOCK_INIT_5050
self.block_len = blocks.init_block_len(self)
# Position
self.stim_probability_left = blocks.init_probability_left(self)
self.stim_probability_left_buffer = [self.stim_probability_left]
self.position = blocks.draw_position(self.position_set,
self.stim_probability_left)
self.position_buffer = [self.position]
# Contrast
self.contrast = misc.draw_contrast(self.contrast_set)
self.contrast_buffer = [self.contrast]
self.signed_contrast = self.contrast * np.sign(self.position)
self.signed_contrast_buffer = [self.signed_contrast]
# RE event names
self.event_error = self.threshold_events_dict[self.position]
self.event_reward = self.threshold_events_dict[-self.position]
self.movement_left = self.threshold_events_dict[
sph.QUIESCENCE_THRESHOLDS[0]]
self.movement_right = self.threshold_events_dict[
sph.QUIESCENCE_THRESHOLDS[1]]
# Trial Completed params
self.elapsed_time = 0
self.behavior_data = []
self.response_time = None
self.response_time_buffer = []
self.response_side_buffer = []
self.trial_correct = None
self.trial_correct_buffer = []
self.ntrials_correct = 0
self.water_delivered = 0