eyepop_game.py

Ir a la documentación de este archivo.

# eyepop_game.py
import random
import constants
from pygaze import libscreen
from pygaze import libtime
from pygaze import liblog
from pygaze import libinput
from pygaze import eyetracker
 
# Configuración inicial
disp = libscreen.Display()
tracker = eyetracker.EyeTracker(disp)
keyboard = libinput.Keyboard(keylist=['space'], timeout=None)
log = liblog.Logfile()
log.write(["trialnr", "targetpos", "reaction_time", "success"])
 
# Pantallas
inscreen = libscreen.Screen()
inscreen.draw_text(text="¡Mira los círculos lo más rápido que puedas!\n(espacio para comenzar)", fontsize=24)
resultscreen = libscreen.Screen()
feedbackscreens = {
    1: libscreen.Screen(),
    0: libscreen.Screen()
}
feedbackscreens[1].draw_text(text="¡Buen reflejo!", colour=(0, 255, 0), fontsize=32)
feedbackscreens[0].draw_text(text="¡Muy lento!", colour=(255, 0, 0), fontsize=32)
 
# Calibración
tracker.calibrate()
 
# Mostrar instrucciones
disp.fill(inscreen)
disp.show()
keyboard.get_key()
 
# Juego: 15 rondas
score = 0
for trialnr in range(1, 16):
    # Generar posición del objetivo
    x = random.randint(200, constants.DISPSIZE[0] - 200)
    y = random.randint(150, constants.DISPSIZE[1] - 150)
    targetpos = (x, y)
 
    # Dibujar pantalla con objetivo
    targetscreen = libscreen.Screen()
    targetscreen.draw_circle(pos=targetpos, r=30, fill=True)
 
    # Corrección de deriva
    checked = False
    while not checked:
        disp.fill(libscreen.Screen())
        disp.show()
        checked = tracker.drift_correction()
 
    # Iniciar grabación
    tracker.start_recording()
    disp.fill(targetscreen)
    t0 = disp.show()
    tracker.log(f"round_{trialnr}_shown")
 
    # Esperar a que la mirada llegue al círculo
    success = 0
    reaction_time = -1
    timeout = 1000  # 1 segundo máximo
 
    start = libtime.get_time()
    while libtime.get_time() - start < timeout:
        gaze = tracker.sample()
        dx = gaze[0] - x
        dy = gaze[1] - y
        dist = (dx ** 2 + dy ** 2) ** 0.5
 
        if dist < 60:
            success = 1
            reaction_time = libtime.get_time() - t0
            score += 1
            break
 
    tracker.stop_recording()
 
    # Feedback
    disp.fill(feedbackscreens[success])
    disp.show()
    libtime.pause(600)
 
    # Registro
    log.write([trialnr, targetpos, reaction_time, success])
 
# Mostrar resultado final
resultscreen.draw_text(text=f"Juego terminado\nPuntuación: {score}/15", fontsize=32)
disp.fill(resultscreen)
disp.show()
libtime.pause(4000)
 
# Finalizar
log.close()
tracker.close()
disp.close()
libtime.expend()