PgTools.set_cursor(screen, mid=True)
def check_stim(xCoord, yCoord):
for i in range(4):
if stims[i] and stims[i].collidepoint(xCoord, yCoord):
global last_stim
last_stim = i
return True
PgTools.write_ln(
filename="Sides_Task/results.csv",
data=[
"subject_name",
"trial",
"sides_num",
],
)
trialNum = 1
sideBools = [False, False, False, False]
last_stim = 0
passedTrials = 0
for i in range(sidesNum):
sideBools[i] = True
start_trial(stimWidth, sideBools)
# game loop
running = True
while running:
currentHeight += maxHeight / 4
currentHeight = int(currentHeight)
PgTools.set_cursor(screen, randCorner=True)
def check_stim(xCoord, yCoord):
for i in range(stimAmt):
if i != oddLocation and stimList[i].collidepoint(xCoord, yCoord):
return True
PgTools.write_ln(
filename="Oddity_Testing/results.csv",
data=[
"subject_name",
"trial",
"odd_stim_location",
"accuracy",
],
)
trialNum = 1
colorPair = PgTools.two_rand_color()
patColor = PgTools.rand_color()
randNums = [randint(0, 2), randint(0, 1)]
color = PgTools.rand_color()
patColor = PgTools.rand_color()
oddLocation = randint(0, stimAmt - 1)
stimList = []
maxLength = PgTools.SCREEN_SIZE[0] - stimLength
maxHeight = PgTools.SCREEN_SIZE[1] - stimHeight
screen.refresh()
continue
else:
if randShapes:
PgTools.rand_shape(screen.fg, (posX, posY,),(length, height), posSeed)
PgTools.rand_shape(screen.fg, (negX, negY,),(length, height), negSeed)
break
PgTools.write_ln(
filename="Two_Choice_Discrimination/results.csv",
data=[
"subject_name",
"set",
"trial",
"total_trial",
posStimName,
negStimName,
"input_coordinates",
"accuracy",
],
)
trialNum = 1
setNum = 1
passedTrials = 0
colorPair = PgTools.two_rand_color()
color = PgTools.rand_color()
totalTrial = 1
if randShapes:
posSeed = randint(0, 99999)
PgTools.set_cursor(screen, mid=True)
def check_stim(xCoord, yCoord):
for i in range(stimAmt):
if i != posLocation and stimList[i].collidepoint(
xCoord, yCoord) and screen.fg.get_at(
(xCoord, yCoord)) != (0, 0, 0):
return True
PgTools.write_ln(
filename="Delayed_Match_To_Sample/results.csv",
data=[
"subject_name",
"trial",
"comparison_stimuli",
"accuracy",
],
)
maxLength = PgTools.SCREEN_SIZE[0] - stimLength
maxHeight = PgTools.SCREEN_SIZE[1] - stimHeight
trialNum = 1
posColor = PgTools.rand_color()
patColor = PgTools.rand_color()
randNums = [randint(0, 2), randint(0, 1)]
stimList = []
seed = random.randint(0, 99999)
randNums = [randint(0, 2), randint(0, 1)]
randRI = random.choice(RILengths)
stimulus = pg.draw.circle(
screen.fg,
PgTools.GREEN,
(
int(PgTools.SCREEN_SIZE[0] / 2),
int(PgTools.SCREEN_SIZE[1] / 2),
),
radius,
)
PgTools.write_ln(
filename="Social_Stimuli_As_Rewards/results.csv",
data=[
"subject_name",
"trial",
"stimulus_size",
"image",
"image_duration",
],
)
path = os.path.join(
"/home",
"pi",
"Desktop",
"ChimpPygames",
"Social_Stimuli_As_Rewards",
"Social_Stimuli",
)
file = random.choice(os.listdir(path))
if (file.endswith((".mp4", ".mov"))):
stimulus = pg.draw.rect(
screen.fg,
PgTools.GREEN,
(
(x_mid, y_mid),
(length, height),
),
)
if randShapes:
PgTools.rand_shape(screen.fg, (x_mid, y_mid), (length, height))
PgTools.set_cursor(screen, randCorner=True)
PgTools.write_ln(
filename="Training_Task/resultsP1.csv",
data=["subject_name", "trial", "stimulus_size", "input_coordinates", "accuracy",],
)
trialNum = 1
passedTrials = 1
start_trial(stimLength, stimHeight)
on_bg = True
# game loop
running = True
while running:
for event in pg.event.get():
PgTools.quit_pg(event)
if event.type == PgTools.input_mode:
xCoord, yCoord = event.pos
if PgTools.touchscreen:
currentHeight = int(currentHeight)
PgTools.set_cursor(screen, randCorner=True)
def check_stim(xCoord, yCoord):
for i in range(stimAmt):
if i != SCSLocation and stimList[i].collidepoint(xCoord, yCoord):
return True
PgTools.write_ln(
filename="Delayed_Response_Task/results.csv",
data=[
"subject_name",
"number_of_possible_locations",
"trial",
SCSStimName,
"SCS_location",
"RI",
"accuracy",
],
)
trialNum = 1
color = PgTools.rand_color()
patColor = PgTools.rand_color()
randRI = random.choice(RILengths)
possibleLocations = random.sample(range(stimAmt), locationsAmt)
SCSLocation = random.choice(possibleLocations)
stimList = []
maxLength = PgTools.SCREEN_SIZE[0] - stimLength
maxHeight = PgTools.SCREEN_SIZE[1] - stimHeight
s = pg.Surface((length, height))
s.blit(images[random.randint(0, len(images) - 1)], (0, 0))
# s.blit(test_image, (0, 0)) # use to debug clipart
screen.fg.blit(s, (xCoord, yCoord))
stimulus = pg.draw.rect(screen.fg, Color('goldenrod'),
(xCoord, yCoord, length, height), 15)
# randInt = random.choice([0, 1, 2, 3, 4])
# if randShapes:
# PgTools.rand_shape(screen.fg, (xCoord, yCoord), (length, height), randInt)
PgTools.write_ln(
filename="Training_Task/resultsP2.csv",
data=[
"subject_name",
"trial",
"input_coordinates",
"accuracy",
],
)
trialNum = 1
passedTrials = 0
# randInt = random.randint(0, 99999) # seed
# load all our images
images = []
list_of_stimuli = glob.glob(
"/home/pi/Desktop/ChimpPygames/Images/samples/*.jpg")
for filepath in list_of_stimuli:
images.append(