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