def main():
trial = RedGreenTrial(name = TRIAL_NAME, dims = TRIAL_DIMS)
add_ball(trial)
add_targets(trial)
add_walls(trial)
add_obstacles(trial)
trial.save(flnm = TRIAL_NAME + ".ptr", fldir = SAVED_TRIAL_DIR) # save .ptr for viewing with simulator
trial.save(flnm = TRIAL_NAME + ".json", fldir = SAVED_TRIAL_DIR) # save .json for use in actual experiments
def write_file(filepath, filejson):
filename = filejson.name + ".json"
write_file = os.path.join(filepath, filename)
print("Writing data to: {}".format(write_file))
trial = RedGreenTrial(name=filejson.name, dims=filejson.dims)
trial.add_ball(initpos=filejson.ball[0],
initvel=filejson.ball[1],
rad=filejson.ball[2],
color=filejson.ball[3],
elast=filejson.ball[4])
for goal in filejson.goals:
trial.add_goal(upperleft=goal[0],
lowerright=goal[1],
onreturn=goal[2],
color=goal[3])
for wall in filejson.normwalls:
trial.add_wall(upperleft=wall[0],
lowerright=wall[1],
color=wall[2],
elast=wall[3])
trial.save(flnm=filejson.name + ".ptr",
fldir=filepath) # save .ptr for viewing with simulator
trial.save(flnm=filejson.name + ".json",
fldir=filepath) # save .json for use in actual experiments
