Pierre Karashchuk
95 linhas
1.8 KiB
Python
Arquivo Executável
95 linhas
1.8 KiB
Python
Arquivo Executável
#!/usr/bin/env python2
|
|
|
|
import pygame
|
|
import platform
|
|
import time
|
|
from pygame.locals import *
|
|
|
|
fullscreen_on = False
|
|
|
|
size = 1280,800
|
|
width, height = size
|
|
|
|
white = 255,255,255
|
|
black = 0,0,0
|
|
red = 255,0,0
|
|
green = 0,255,0
|
|
blue = 0,0,255
|
|
chartreuse = 127,255,0
|
|
light_green = 131,255,100
|
|
nice_red = 207,45,64
|
|
sky_blue = 100,244,255
|
|
nice_blue = 0,194,255
|
|
dark_gray = 20,20,20
|
|
|
|
|
|
space_x = 100
|
|
space_y = 100
|
|
|
|
rect_width = (width - space_x*3)/2.0
|
|
rect_height = (height - space_y *3)/2.0
|
|
rect_xs = [space_x, space_x*2+rect_width, space_x, space_x*2+rect_width]
|
|
rect_ys = [space_y, space_y, space_y*2+rect_height, space_y*2+rect_height]
|
|
|
|
freqs = [10, 14, 17, 20]
|
|
stimuli = [0, 0, 0, 0]
|
|
wait_times = [0,0,0,0]
|
|
|
|
|
|
|
|
if fullscreen_on:
|
|
screen = pygame.display.set_mode(size, FULLSCREEN)
|
|
else:
|
|
screen = pygame.display.set_mode(size)
|
|
|
|
if platform.system() == 'Windows':
|
|
wallclock = time.clock
|
|
else:
|
|
wallclock = time.time
|
|
|
|
pygame.mouse.set_visible(False)
|
|
|
|
|
|
def check_for_escape():
|
|
event = pygame.event.poll()
|
|
if event.type == 0:
|
|
return
|
|
elif event.dict.get('key', -1) == K_ESCAPE:
|
|
pygame.quit()
|
|
exit()
|
|
|
|
def draw_stimuli():
|
|
screen.fill(black)
|
|
|
|
for s, x, y in zip(stimuli, rect_xs, rect_ys):
|
|
if s == 0:
|
|
col = black
|
|
else:
|
|
col = white
|
|
|
|
rect = ((x, y), (rect_width, rect_height))
|
|
screen.fill(col, rect)
|
|
|
|
pygame.display.flip()
|
|
|
|
|
|
to_wait = 0
|
|
|
|
while True:
|
|
t = time.time()
|
|
check_for_escape()
|
|
|
|
draw_stimuli()
|
|
|
|
for i in range(len(stimuli)):
|
|
wait = wait_times[i]
|
|
if wait <= 1e-5:
|
|
stimuli[i] = 1 - stimuli[i]
|
|
wait_times[i] = 1.0/freqs[i]
|
|
|
|
to_wait = 0.005
|
|
time.sleep(to_wait)
|
|
for i in range(len(stimuli)):
|
|
wait_times[i] -= time.time() - t
|
|
|