temp_flicker = np.arange(0, n_frames, int(
realRR / 2)) # Get temp_flicker frames: every .5 s
delay = []
for i in np.arange(0, len(temp_flicker)):
rng.shuffle(jitter)
delay.append(jitter[0])
frame_flicker = temp_flicker + delay #
# load up the image stack. The images in img_buffer are in the sequential
# non-shuffled order
img = []
for im in imagelist:
img_buffer = np.array(Image.open(im), np.uint8) / 255.
if img_buffer.ndim == 2:
img_buffer = np.tile(img_buffer[:, :, np.newaxis], [1, 1, 3])
img.append(visual.RawImage(ec, img_buffer, scale=s))
ec.check_force_quit()
# make a blank image
blank = visual.RawImage(
ec, np.tile(bgcolor[0], np.multiply([s, s, 1], img_buffer.shape)))
bright = visual.RawImage(
ec, np.tile([1.], np.multiply([s, s, 1], img_buffer.shape)))
# Calculate stimulus size
d_pix = -np.diff(ec._convert_units([[3., 0.], [3., 0.]], 'deg', 'pix'),
axis=-1)
# do the drawing, then flip
ec.set_visible(True)
frametimes = []
buttons = []
def test_visuals(hide_window):
"""Test EC visual methods."""
with ExperimentController('test', **std_kwargs) as ec:
pytest.raises(TypeError, visual.Circle, ec, n_edges=3.5)
pytest.raises(ValueError, visual.Circle, ec, n_edges=3)
circ = visual.Circle(ec)
circ.draw()
pytest.raises(ValueError, circ.set_radius, [1, 2, 3])
pytest.raises(ValueError, circ.set_pos, [1])
pytest.raises(ValueError, visual.Triangle, ec, [5, 6])
tri = visual.Triangle(ec, [[-1, 0, 1], [-1, 1, -1]],
units='deg',
line_width=1.0)
tri.draw()
rect = visual.Rectangle(ec, [0, 0, 1, 1], line_width=1.0)
rect.draw()
diamond = visual.Diamond(ec, [0, 0, 1, 1], line_width=1.0)
diamond.draw()
pytest.raises(TypeError, visual.ConcentricCircles, ec, colors=dict())
pytest.raises(TypeError,
visual.ConcentricCircles,
ec,
colors=np.array([]))
pytest.raises(ValueError, visual.ConcentricCircles, ec, radii=[[1]])
pytest.raises(ValueError, visual.ConcentricCircles, ec, radii=[1])
fix = visual.ConcentricCircles(ec,
radii=[1, 2, 3],
colors=['w', 'k', 'y'])
fix.set_pos([0.5, 0.5])
fix.set_radius(0.1, 1)
fix.set_radii([0.1, 0.2, 0.3])
fix.set_color('w', 1)
fix.set_colors(['w', 'k', 'k'])
fix.set_colors(('w', 'k', 'k'))
pytest.raises(IndexError, fix.set_color, 'w', 3)
pytest.raises(ValueError, fix.set_colors, ['w', 'k'])
pytest.raises(ValueError, fix.set_colors, np.array(['w', 'k', 'k']))
pytest.raises(IndexError, fix.set_radius, 0.1, 3)
pytest.raises(ValueError, fix.set_radii, [0.1, 0.2])
fix.draw()
fix_2 = visual.FixationDot(ec)
fix_2.draw()
pytest.raises(ValueError, rect.set_pos, [0, 1, 2])
for shape in ((3, 3, 3), (3, 3, 4), (3, 3), (3, ), (3, ) * 4):
data = np.ones(shape)
if len(shape) not in (2, 3):
pytest.raises(RuntimeError, visual.RawImage, ec, data)
else:
img = visual.RawImage(ec, data)
print(img.bounds) # test bounds
assert_equal(img.scale, 1)
# test get_rect
imgrect = visual.Rectangle(ec, img.get_rect())
assert_equal(imgrect._points['fill'][(0, 2, 0, 1), (0, 0, 1, 1)],
img.bounds)
img.draw()
line = visual.Line(ec, [[0, 1], [1, 0]])
line.draw()
pytest.raises(ValueError, line.set_line_width, 100)
line.set_line_width(2)
line.draw()
pytest.raises(ValueError, line.set_coords, [0])
line.set_coords([0, 1])
ec.set_background_color('black')
text = visual.Text(ec,
'Hello {color (255 0 0 255)}Everybody!',
pos=[0, 0],
color=[1, 1, 1],
wrap=False)
text.draw()
text.set_color(None)
text.draw()
text = visual.Text(ec,
'Thank you, come again.',
pos=[0, 0],
color='white',
attr=False)
text.draw()
text.set_color('red')
text.draw()
bar = visual.ProgressBar(ec, [0, 0, 1, .2])
bar = visual.ProgressBar(ec, [0, 0, 1, 1], units='pix')
bar.update_bar(.5)
bar.draw()
pytest.raises(ValueError,
visual.ProgressBar,
ec, [0, 0, 1, .1],
units='deg')
pytest.raises(ValueError,
visual.ProgressBar,
ec, [0, 0, 1, .1],
colors=['w'])
pytest.raises(ValueError, bar.update_bar, 500)
def test_visuals():
"""Test EC visual methods
"""
with ExperimentController('test', **std_kwargs) as ec:
assert_raises(TypeError, visual.Circle, ec, n_edges=3.5)
assert_raises(ValueError, visual.Circle, ec, n_edges=3)
circ = visual.Circle(ec)
circ.draw()
assert_raises(ValueError, circ.set_radius, [1, 2, 3])
assert_raises(ValueError, circ.set_pos, [1])
assert_raises(ValueError, visual.Triangle, ec, [5, 6])
tri = visual.Triangle(ec, [[-1, 0, 1], [-1, 1, -1]],
units='deg',
line_width=1.0)
tri.draw()
rect = visual.Rectangle(ec, [0, 0, 1, 1], line_width=1.0)
rect.draw()
diamond = visual.Diamond(ec, [0, 0, 1, 1], line_width=1.0)
diamond.draw()
assert_raises(TypeError, visual.ConcentricCircles, ec, colors=dict())
assert_raises(TypeError,
visual.ConcentricCircles,
ec,
colors=np.array([]))
assert_raises(ValueError, visual.ConcentricCircles, ec, radii=[[1]])
assert_raises(ValueError, visual.ConcentricCircles, ec, radii=[1])
fix = visual.ConcentricCircles(ec,
radii=[1, 2, 3],
colors=['w', 'k', 'y'])
fix.set_pos([0.5, 0.5])
fix.set_radius(0.1, 1)
fix.set_radii([0.1, 0.2, 0.3])
fix.set_color('w', 1)
fix.set_colors(['w', 'k', 'k'])
fix.set_colors(('w', 'k', 'k'))
assert_raises(IndexError, fix.set_color, 'w', 3)
assert_raises(ValueError, fix.set_colors, ['w', 'k'])
assert_raises(ValueError, fix.set_colors, np.array(['w', 'k', 'k']))
assert_raises(IndexError, fix.set_radius, 0.1, 3)
assert_raises(ValueError, fix.set_radii, [0.1, 0.2])
fix.draw()
fix_2 = visual.FixationDot(ec)
fix_2.draw()
assert_raises(ValueError, rect.set_pos, [0, 1, 2])
img = visual.RawImage(ec, np.ones((3, 3, 4)))
print(img.bounds) # test bounds
assert_equal(img.scale, 1)
img.draw()
line = visual.Line(ec, [[0, 1], [1, 0]])
line.draw()
assert_raises(ValueError, line.set_line_width, 100)
line.set_line_width(2)
line.draw()
assert_raises(ValueError, line.set_coords, [0])
line.set_coords([0, 1])
ec.set_background_color('black')
text = visual.Text(ec,
'Hello {color (255 0 0 255)}Everybody!',
pos=[0, 0],
color=[1, 1, 1],
wrap=False)
text.draw()
text.set_color(None)
text.draw()
text = visual.Text(ec,
'Thank you, come again.',
pos=[0, 0],
color='white',
attr=False)
text.draw()
text.set_color('red')
text.draw()
# test video
std_kwargs.update(dict(enable_video=True, window_size=(640, 480)))
video_path = fetch_data_file('video/example-video.mp4')
with ExperimentController('test', **std_kwargs) as ec:
ec.load_video(video_path)
ec.video.play()
assert_raises(ValueError, ec.video.set_scale, 'foo')
assert_raises(ValueError, ec.video.set_scale, -1)
ec.video.set_scale('fill')
ec.video.set_scale('fit')
ec.video.set_scale(0.5)
ec.video.set_pos(pos=(0.1, 0), units='norm')
ec.wait_secs(0.2)
ec.video.pause()
ec.delete_video()
print(__doc__)
with ExperimentController('test', session='1', participant='2',
full_screen=False, window_size=[600, 600],
output_dir=None, version='dev') as ec:
ec.screen_text('hello')
# make an image with alpha the x-dimension (columns), RGB upward
img_buffer = np.zeros((120, 100, 4))
img_buffer[:, :50, 3] = 1.0
img_buffer[:, 50:, 3] = 0.5
img_buffer[0] = 1
for ii in range(3):
img_buffer[ii * 40:(ii + 1) * 40, :, ii] = 1.0
img = visual.RawImage(ec, img_buffer, scale=2.)
# make a line, rectangle, diamond, and circle
line = visual.Line(ec, [[-2, 2, 2, -2], [-2, 2, -2, -2]], units='deg',
line_color='w', line_width=2.0)
rect = visual.Rectangle(ec, [0, 0, 2, 2], units='deg', fill_color='y')
diamond = visual.Diamond(ec, [0, 0, 4, 4], units='deg', fill_color=None,
line_color='gray', line_width=2.0)
circle = visual.Circle(ec, 1, units='deg', line_color='w', fill_color='k',
line_width=2.0)
# do the drawing, then flip
for obj in [img, line, rect, diamond, circle]:
obj.draw()
screenshot = ec.screenshot() # must be called *before* the flip
def test_visuals():
"""Test EC visual methods."""
with ExperimentController('test', **std_kwargs) as ec:
assert_raises(TypeError, visual.Circle, ec, n_edges=3.5)
assert_raises(ValueError, visual.Circle, ec, n_edges=3)
circ = visual.Circle(ec)
circ.draw()
assert_raises(ValueError, circ.set_radius, [1, 2, 3])
assert_raises(ValueError, circ.set_pos, [1])
assert_raises(ValueError, visual.Triangle, ec, [5, 6])
tri = visual.Triangle(ec, [[-1, 0, 1], [-1, 1, -1]], units='deg',
line_width=1.0)
tri.draw()
rect = visual.Rectangle(ec, [0, 0, 1, 1], line_width=1.0)
rect.draw()
diamond = visual.Diamond(ec, [0, 0, 1, 1], line_width=1.0)
diamond.draw()
assert_raises(TypeError, visual.ConcentricCircles, ec, colors=dict())
assert_raises(TypeError, visual.ConcentricCircles, ec,
colors=np.array([]))
assert_raises(ValueError, visual.ConcentricCircles, ec, radii=[[1]])
assert_raises(ValueError, visual.ConcentricCircles, ec, radii=[1])
fix = visual.ConcentricCircles(ec, radii=[1, 2, 3],
colors=['w', 'k', 'y'])
fix.set_pos([0.5, 0.5])
fix.set_radius(0.1, 1)
fix.set_radii([0.1, 0.2, 0.3])
fix.set_color('w', 1)
fix.set_colors(['w', 'k', 'k'])
fix.set_colors(('w', 'k', 'k'))
assert_raises(IndexError, fix.set_color, 'w', 3)
assert_raises(ValueError, fix.set_colors, ['w', 'k'])
assert_raises(ValueError, fix.set_colors, np.array(['w', 'k', 'k']))
assert_raises(IndexError, fix.set_radius, 0.1, 3)
assert_raises(ValueError, fix.set_radii, [0.1, 0.2])
fix.draw()
fix_2 = visual.FixationDot(ec)
fix_2.draw()
assert_raises(ValueError, rect.set_pos, [0, 1, 2])
img = visual.RawImage(ec, np.ones((3, 3, 4)))
print(img.bounds) # test bounds
assert_equal(img.scale, 1)
img.draw()
line = visual.Line(ec, [[0, 1], [1, 0]])
line.draw()
assert_raises(ValueError, line.set_line_width, 100)
line.set_line_width(2)
line.draw()
assert_raises(ValueError, line.set_coords, [0])
line.set_coords([0, 1])
ec.set_background_color('black')
text = visual.Text(ec, 'Hello {color (255 0 0 255)}Everybody!',
pos=[0, 0], color=[1, 1, 1], wrap=False)
text.draw()
text.set_color(None)
text.draw()
text = visual.Text(ec, 'Thank you, come again.', pos=[0, 0],
color='white', attr=False)
text.draw()
text.set_color('red')
text.draw()
if ii <= 12:
eventnums[c] = [1, ii]
else:
eventnums[c] = [2, ii - 12]
# with ExperimentController('ReadingExperiment', full_screen=True) as ec:
with ExperimentController('ReadingExperiment',
full_screen=False,
window_size=[800, 800]) as ec:
# Set the background color to gray
ec.set_background_color([180 / 255., 180 / 255., 180 / 255., 1])
# load up the image stack
img = []
for im in imagelist:
img_buffer = np.array(Image.open(imagedir + im)) / 255.
img.append(visual.RawImage(ec, img_buffer, scale=2.))
# do the drawing, then flip
frametimes = []
buttons = []
ec.listen_presses()
last_flip = -1
# Initiate a counter
c = -1
for obj in img:
c = c + 1
# If this frame is the beginning of a new event type then we mark it in
# The MEG data file
if c in onsets:
ec.call_on_next_flip(ec.stamp_triggers,
eventnums[onsets.index(c)],
