#!/usr/bin/env python
from psychopy import core, visual, event
import numpy as np

# Main constants

brightness = .3 # brightness between 0 and 1
res = 1280, 960
nX = 44
nY = 33
bgOri = 0
targetOri = 45
distOri = 90
targetPos = range(33, 35), range(5, 15)
distPos = range(23, 24), range(2, 20)
imgPath = 'myStimulus.png'

w = res[0] / nX
h = res[0] / nY
dx = w/2
dy = h/2

x = np.linspace(0, 4*np.pi, 256) # Create a linspace as input for sin
tex = np.cos([x]) # A sine between 1 and -1
tex = tex * brightness - (1-brightness) # Adjust the texture brightness

myWin = visual.Window(res, units='pix', color=(-1+brightness))
bgStim = visual.GratingStim(myWin, size=(w,h), tex=tex, mask='gauss', ori=bgOri)
targetStim = visual.GratingStim(myWin, size=(w,h), tex=tex, mask='gauss', ori= \
    targetOri)
distStim = visual.GratingStim(myWin, size=(w,h), tex=tex, mask='gauss', ori= \
    distOri)

for i in range(nX):
    for j in range(nY):
        x = (w*i + dx) - res[0]/2
        y = res[1]/2 - (h*j + dy)
        if i in targetPos[0] and j in targetPos[1]:
            print 'Target at %d, %d / %d, %d' % (i, j, x, y)
            patch = targetStim
        elif i in distPos[0] and j in distPos[1]:
            print 'Distractor at %d, %d / %d, %d' % (i, j, x, y)
            patch = distStim
        else:
            patch = bgStim
        patch.setPos( (x, y) )
        patch.draw()

# Show the screen
myWin.flip()

# Save the image to disk
myWin.getMovieFrame()
myWin.movieFrames[0].save(imgPath)

print 'Done!'
raw_input()