def __init__(self, exp_info, file_name):
"""
Initialize experiment - read XML file, setup window, connect to netstation and tobii
exp_info - experiment information
file_name - name of XML file containing experiment definition
"""
self.exp_info = exp_info
self.name = None
self.type = None
self.num_blocks = 0
self.blocks = {}
self.block_order = []
# Window to use
wintype = 'pyglet' # use pyglet if possible, it's faster at event handling
# Add 14cm to distance - this is distance from eyetracker to monitor
mon = monitors.Monitor(exp_info['monitor'], distance=float(exp_info['monitor distance']))
self.win = Window(
[1280, 1024],
monitor=mon,
screen=SCREEN,
units="deg",
fullscr=True,
#fullscr=False,
color=[-1, -1, -1],
winType=wintype)
self.win.setMouseVisible(False)
event.clearEvents()
# Measure frame rate
self.mean_ms_per_frame, std_ms_per_frame, median_ms_per_frame = visual.getMsPerFrame(self.win, nFrames=60,
showVisual=True)
self.debug_sq=None
if exp_info['monitor']=='tobii':
self.debug_sq=psychopy.visual.Rect(self.win, width=30, height=30, units='pix')
self.debug_sq.setFillColor((1,1,1))
self.debug_sq.setPos((630,-500))
# Compute distractor duration in frames based on frame rate
distractor_duration_frames = int(2000.0/self.mean_ms_per_frame)
# Initialize set of distractors
self.distractor_set = DistractorSet(os.path.join(DATA_DIR, 'images', 'distractors', 'space'),
os.path.join(DATA_DIR, 'sounds', 'distractors'),
os.path.join(DATA_DIR, 'movies', 'distractors'),
os.path.join(DATA_DIR, 'images', 'distractors', 'star-cartoon.jpg'),
distractor_duration_frames, self.win)
# Connect to nestation
self.ns = None
if exp_info['eeg']:
# connect to netstation
self.ns = egi.Netstation()
ms_localtime = egi.ms_localtime
self.eye_tracker = None
mouse_visible = False
if exp_info['eyetracking source'] == 'tobii':
# Initialize eyetracker
self.eye_tracker = TobiiController(self.win)
self.eye_tracker.waitForFindEyeTracker()
self.eye_tracker.activate(EYETRACKER_NAME)
elif exp_info['eyetracking source'] == 'mouse':
mouse_visible = True
# Initialize mouse
self.mouse = event.Mouse(visible=mouse_visible, win=self.win)
self.gaze_debug=None
if self.exp_info['debug mode']:
self.gaze_debug=psychopy.visual.Circle(self.win, radius=1, fillColor=(1.0,-1.0,-1.0))
self.read_xml(file_name)
# Initialize netstation and eyetracker
self.initialize()
class Experiment:
"""
Base experiment class
"""
def __init__(self, exp_info, file_name):
"""
Initialize experiment - read XML file, setup window, connect to netstation and tobii
exp_info - experiment information
file_name - name of XML file containing experiment definition
"""
self.exp_info = exp_info
self.name = None
self.type = None
self.num_blocks = 0
self.blocks = {}
self.block_order = []
# Window to use
wintype = 'pyglet' # use pyglet if possible, it's faster at event handling
# Add 14cm to distance - this is distance from eyetracker to monitor
mon = monitors.Monitor(exp_info['monitor'], distance=float(exp_info['monitor distance']))
self.win = Window(
[1280, 1024],
monitor=mon,
screen=SCREEN,
units="deg",
fullscr=True,
#fullscr=False,
color=[-1, -1, -1],
winType=wintype)
self.win.setMouseVisible(False)
event.clearEvents()
# Measure frame rate
self.mean_ms_per_frame, std_ms_per_frame, median_ms_per_frame = visual.getMsPerFrame(self.win, nFrames=60,
showVisual=True)
self.debug_sq=None
if exp_info['monitor']=='tobii':
self.debug_sq=psychopy.visual.Rect(self.win, width=30, height=30, units='pix')
self.debug_sq.setFillColor((1,1,1))
self.debug_sq.setPos((630,-500))
# Compute distractor duration in frames based on frame rate
distractor_duration_frames = int(2000.0/self.mean_ms_per_frame)
# Initialize set of distractors
self.distractor_set = DistractorSet(os.path.join(DATA_DIR, 'images', 'distractors', 'space'),
os.path.join(DATA_DIR, 'sounds', 'distractors'),
os.path.join(DATA_DIR, 'movies', 'distractors'),
os.path.join(DATA_DIR, 'images', 'distractors', 'star-cartoon.jpg'),
distractor_duration_frames, self.win)
# Connect to nestation
self.ns = None
if exp_info['eeg']:
# connect to netstation
self.ns = egi.Netstation()
ms_localtime = egi.ms_localtime
self.eye_tracker = None
mouse_visible = False
if exp_info['eyetracking source'] == 'tobii':
# Initialize eyetracker
self.eye_tracker = TobiiController(self.win)
self.eye_tracker.waitForFindEyeTracker()
self.eye_tracker.activate(EYETRACKER_NAME)
elif exp_info['eyetracking source'] == 'mouse':
mouse_visible = True
# Initialize mouse
self.mouse = event.Mouse(visible=mouse_visible, win=self.win)
self.gaze_debug=None
if self.exp_info['debug mode']:
self.gaze_debug=psychopy.visual.Circle(self.win, radius=1, fillColor=(1.0,-1.0,-1.0))
self.read_xml(file_name)
# Initialize netstation and eyetracker
self.initialize()
def calibrate_eyetracker(self):
"""
Run eyetracker calibration routine
"""
retval = 'retry'
while retval == 'retry':
waitkey = True
retval = None
can_accept = self.eye_tracker.doCalibration(EYETRACKER_CALIBRATION_POINTS)
while waitkey:
for key in psychopy.event.getKeys():
if can_accept:
num_entered=True
try:
calib_idx=int(key)
can_accept = self.eye_tracker.doCalibration([EYETRACKER_CALIBRATION_POINTS[calib_idx-1]],
calib=self.eye_tracker.calib)
except:
num_entered=False
if not num_entered and key == 'a':
retval = 'accept'
waitkey = False
elif key == 'r':
retval = 'retry'
waitkey = False
elif key == 'escape':
retval = 'abort'
waitkey = False
self.eye_tracker.calresult.draw()
self.eye_tracker.calresultmsg.draw()
for point_label in self.eye_tracker.point_labels:
point_label.draw()
self.win.flip()
if retval == 'abort':
self.eye_tracker.closeDataFile()
self.eye_tracker.destroy()
self.win.close()
core.quit()
def initialize(self):
"""
Start netstation recording, calibrate eyetracker
"""
if self.ns is not None:
try:
self.ns.initialize(NETSTATION_IP, 55513)
self.ns.BeginSession()
self.ns.StartRecording()
except:
print('Could not connect with NetStation!')
# Initialize logging
logfile = os.path.join(DATA_DIR, 'logs', self.exp_info['experiment'], '%s_%s_%s.log' % (self.exp_info['child_id'],
self.exp_info['date'],
self.exp_info['session']))
if self.eye_tracker is not None:
self.eye_tracker.setDataFile(logfile, self.exp_info)
else:
datafile = open(logfile, 'w')
datafile.write('Recording date:\t' + datetime.datetime.now().strftime('%Y/%m/%d') + '\n')
datafile.write('Recording time:\t' + datetime.datetime.now().strftime('%H:%M:%S') + '\n')
datafile.write('Recording resolution\t%d x %d\n' % tuple(self.win.size))
for key, data in self.exp_info.iteritems():
datafile.write('%s:\t%s\n' % (key, data))
datafile.close()
# Create random block order
n_repeats = int(self.num_blocks/len(self.blocks.keys()))
for i in range(n_repeats):
subblock_order = copy.copy(self.blocks.keys())
np.random.shuffle(subblock_order)
self.block_order.extend(subblock_order)
# Synch with netstation in between trials
if self.ns is not None:
self.ns.sync()
if self.eye_tracker is not None:
self.eye_tracker.startTracking()
def close(self):
"""
Disconnect from eyetracker and netstation
"""
if self.eye_tracker is not None:
self.eye_tracker.stopTracking()
self.eye_tracker.closeDataFile()
# close netstation connection
if self.ns:
self.ns.StopRecording()
self.ns.EndSession()
self.ns.finalize()
self.win.close()
core.quit()
if self.eye_tracker is not None:
self.eye_tracker.destroy()
def run(self):
"""
Run task
ns - netstation connection
"""
pass
def read_xml(self, file_name):
"""
Read experiment definition file
:param file_name: file to read definition from
"""
pass
