python_stimuli/SNRtest.py at master · AdvancedMRI/python_stimuli · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
#!/usr/bin/env python
# -*- coding: utf-8 -*-

# original stimulus written by the Silver lab
# psychopy version written by Alex Beckett
# code to try and draw the annulus + counterphase plaid stimuli from Silver 2008
# no targets or anything as yet

from __future__ import division

from psychopy import core, visual, event
from numpy import sin, pi

# Create a window to draw in
win = visual.Window((600, 600), allowGUI=False, monitor='testMonitor', units='deg',blendMode='avg',allowStencil=True)
nTrials = 10

# Initialize some stimuli
# create the mask we will draw the stimulus through - can only use one so draw one annulus and we will rotate it
mask = visual.RadialStim(win, tex='none', color=1, size=1,
    visibleWedge=[2, 178], radialCycles=4, angularCycles=8, interpolate=False,
    autoLog=False)  # this stim changes too much for autologging to be useful
# gratings for plaid stimulus
grating1 = visual.GratingStim(
    win, tex="sin", mask="circle", texRes=512,
    color='white', size=10, sf=2, ori=45, depth=0.5, opacity=0.5, autoLog=False)
grating2 = visual.GratingStim(
    win, tex="sin", mask="circle", texRes=512,
    color='white', size=10, sf=2, ori=-45, depth=0.5, opacity=0.5, autoLog=False)
# 0 contrats grating + gauss mask to act as target(s)
grating3 =visual.GratingStim(
    win, tex="sin", mask="gauss", texRes=128,pos=(2.5,0), contrast=0,
    color='white', size=4, sf=2, ori=45, opacity=1, autoLog=False)

# gray centre to add fixation to
central_grey = visual.PatchStim(win, tex=None, mask='raisedCos',
                             color=0,size=1)

message = visual.TextStim(
    win, text='Press q to quit',
    pos=(-0.95, -0.95), units='norm',
    anchorVert='bottom', anchorHoriz='left')

trialClock = core.Clock()
t = 0

# create aperture from mask
aperture = visual.Aperture(win, size=.5, shape=mask.verticesPix)  # try shape='square'

for x in range(nTrials):
    trialClock.reset()
    t = 0
    while  t < 10:  # quits after 20 secs
        t = trialClock.getTime()
        # set grating contrast
        grating1.contrast = sin(t * pi * 4)
        grating2.contrast = sin(t * pi * 4)

        # initally draw right side
        aperture.ori=0
        aperture.enabled = True  # enabled by default when created
        # draw plaid
        grating1.draw()
        grating2.draw()
        # draw target
        if t<5 and t>4:
            grating3.draw()

        # then rorate annulus to draw left
        aperture.ori=180
        # draw plaid
        grating1.draw()
        grating2.draw()

        # release aperture
        aperture.enabled = False  # enabled by default when created

        # draw fixation area
        central_grey.draw()

        message.draw()

        win.flip()
        for key in event.getKeys():
            keyTime=trialClock.getTime()
            if key in ['escape','q']:
                win.close()
                core.quit()
    trialClock.reset()
    t=0
    while t < 10:
        t = trialClock.getTime()
        message.draw()
        win.flip()
        for key in event.getKeys():
            keyTime=trialClock.getTime()
            if key in ['escape','q']:
                win.close()
                core.quit()

win.close()
core.quit()