updated utils and added README
This commit is contained in:
38
utils/trialGen.py
Normal file
38
utils/trialGen.py
Normal file
@@ -0,0 +1,38 @@
|
||||
import json
|
||||
from random import choice,shuffle
|
||||
direction = ["Up","Left","Right"]
|
||||
distance = ["Short","Medium","Long"]
|
||||
devices = ["cardboard","mouse","oculus"]
|
||||
options = []
|
||||
for x in direction:
|
||||
for y in distance:
|
||||
options.append(x+y)
|
||||
print(options)
|
||||
numDots = int(input("Number of Dots per Trial: "))
|
||||
numTrials = int(input("Number of Trials: "))
|
||||
totalNum = numDots * numTrials
|
||||
mod = totalNum % len(options)
|
||||
dots = []
|
||||
if mod is 0:
|
||||
c = totalNum//len(options)
|
||||
for option in options:
|
||||
dots = dots + ([option] * c)
|
||||
print(dots.count(option))
|
||||
shuffle(dots)
|
||||
print(dots)
|
||||
trials = []
|
||||
for x in range(numTrials):
|
||||
trial = []
|
||||
for y in range(numDots):
|
||||
trial.append(dots.pop())
|
||||
trials.append(trial)
|
||||
root = {}
|
||||
for d in devices:
|
||||
root['trials'] = trials
|
||||
root['inputMethod'] = d
|
||||
root['participantId'] = 0
|
||||
with open(d+'Trials.json','w') as f:
|
||||
json.dump(root,f,indent=4)
|
||||
|
||||
else:
|
||||
print("Please enter evenly divisible numbers")
|
||||
Reference in New Issue
Block a user