Пример #1
0
    def create_experiment_info(self):
        # create subject data directory
        data_dir = 'data/{}/'.format(self.subject_number.get())
        if not os.path.exists(data_dir):
            os.makedirs(data_dir)
        else:
            files = glob.glob(data_dir + '*')
            [os.remove(f) for f in files]

        # create config object
        config = Config(config_filename=self.config_filename.get())

        # create trial list and save to csv
        config.save_trial_list(path=data_dir)

        print "Trial list created as {}.".format(data_dir + 'trials.csv')

        # create experiment info and save to txt
        config.save_experiment_info(glove_hand=self.hand.get(), path=data_dir)

        print "Experiment info created as {}.".format(data_dir + 'exp_info.txt')
Пример #2
0
from config_parser import Config
import json
import viz
import vizact
import numpy as np
from vis_env import Room, BaseballGlove
from stimuli import Ball

config_filename = 'config/home.py'

config = Config(config_filename)
config.save_experiment_info(path='figures/')

# read experiment info (config dict)
with open('figures/exp_info.txt') as info_file:
    config = json.load(info_file)

room = Room(config)

#ball = Ball(room, size=0.15, position=[0,1.8,4], color=[1, 1, 1])
#ball.setContrastMode('constant_dark')
glove = viz.addChild('pole.wrl')


def rotate_glove():
    direction = glove.getMatrix().getForward()
    print direction
    print glove.getQuat()
    #direction[0] = -direction[0]
    #glove.lookAt(direction, viz.REL_LOCAL)
    #print glove.getMatrix().getForward()